nmie.cc 71 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395
  1. //**********************************************************************************//
  2. // Copyright (C) 2009-2015 Ovidio Pena <ovidio@bytesfall.com> //
  3. // Copyright (C) 2013-2015 Konstantin Ladutenko <kostyfisik@gmail.com> //
  4. // //
  5. // This file is part of scattnlay //
  6. // //
  7. // This program is free software: you can redistribute it and/or modify //
  8. // it under the terms of the GNU General Public License as published by //
  9. // the Free Software Foundation, either version 3 of the License, or //
  10. // (at your option) any later version. //
  11. // //
  12. // This program is distributed in the hope that it will be useful, //
  13. // but WITHOUT ANY WARRANTY; without even the implied warranty of //
  14. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the //
  15. // GNU General Public License for more details. //
  16. // //
  17. // The only additional remark is that we expect that all publications //
  18. // describing work using this software, or all commercial products //
  19. // using it, cite the following reference: //
  20. // [1] O. Pena and U. Pal, "Scattering of electromagnetic radiation by //
  21. // a multilayered sphere," Computer Physics Communications, //
  22. // vol. 180, Nov. 2009, pp. 2348-2354. //
  23. // //
  24. // You should have received a copy of the GNU General Public License //
  25. // along with this program. If not, see <http://www.gnu.org/licenses/>. //
  26. //**********************************************************************************//
  27. //**********************************************************************************//
  28. // This class implements the algorithm for a multilayered sphere described by: //
  29. // [1] W. Yang, "Improved recursive algorithm for light scattering by a //
  30. // multilayered sphere,” Applied Optics, vol. 42, Mar. 2003, pp. 1710-1720. //
  31. // //
  32. // You can find the description of all the used equations in: //
  33. // [2] O. Pena and U. Pal, "Scattering of electromagnetic radiation by //
  34. // a multilayered sphere," Computer Physics Communications, //
  35. // vol. 180, Nov. 2009, pp. 2348-2354. //
  36. // //
  37. // Hereinafter all equations numbers refer to [2] //
  38. //**********************************************************************************//
  39. #include "nmie.h"
  40. #include <array>
  41. #include <algorithm>
  42. #include <cstdio>
  43. #include <cstdlib>
  44. #include <stdexcept>
  45. #include <vector>
  46. namespace nmie {
  47. //helpers
  48. template<class T> inline T pow2(const T value) {return value*value;}
  49. int round(double x) {
  50. return x >= 0 ? (int)(x + 0.5):(int)(x - 0.5);
  51. }
  52. //**********************************************************************************//
  53. // This function emulates a C call to calculate the actual scattering parameters //
  54. // and amplitudes. //
  55. // //
  56. // Input parameters: //
  57. // L: Number of layers //
  58. // pl: Index of PEC layer. If there is none just send -1 //
  59. // x: Array containing the size parameters of the layers [0..L-1] //
  60. // m: Array containing the relative refractive indexes of the layers [0..L-1] //
  61. // nTheta: Number of scattering angles //
  62. // Theta: Array containing all the scattering angles where the scattering //
  63. // amplitudes will be calculated //
  64. // nmax: Maximum number of multipolar expansion terms to be used for the //
  65. // calculations. Only use it if you know what you are doing, otherwise //
  66. // set this parameter to -1 and the function will calculate it //
  67. // //
  68. // Output parameters: //
  69. // Qext: Efficiency factor for extinction //
  70. // Qsca: Efficiency factor for scattering //
  71. // Qabs: Efficiency factor for absorption (Qabs = Qext - Qsca) //
  72. // Qbk: Efficiency factor for backscattering //
  73. // Qpr: Efficiency factor for the radiation pressure //
  74. // g: Asymmetry factor (g = (Qext-Qpr)/Qsca) //
  75. // Albedo: Single scattering albedo (Albedo = Qsca/Qext) //
  76. // S1, S2: Complex scattering amplitudes //
  77. // //
  78. // Return value: //
  79. // Number of multipolar expansion terms used for the calculations //
  80. //**********************************************************************************//
  81. int nMie(const int L, const int pl, std::vector<double>& x, std::vector<std::complex<double> >& m, const int nTheta, std::vector<double>& Theta, const int nmax, double *Qext, double *Qsca, double *Qabs, double *Qbk, double *Qpr, double *g, double *Albedo, std::vector<std::complex<double> >& S1, std::vector<std::complex<double> >& S2) {
  82. if (x.size() != L || m.size() != L)
  83. throw std::invalid_argument("Declared number of layers do not fit x and m!");
  84. if (Theta.size() != nTheta)
  85. throw std::invalid_argument("Declared number of sample for Theta is not correct!");
  86. try {
  87. MultiLayerMie multi_layer_mie;
  88. multi_layer_mie.SetLayersSize(x);
  89. multi_layer_mie.SetLayersIndex(m);
  90. multi_layer_mie.SetAngles(Theta);
  91. multi_layer_mie.SetPECLayer(pl);
  92. multi_layer_mie.SetMaxTerms(nmax);
  93. multi_layer_mie.RunMieCalculation();
  94. *Qext = multi_layer_mie.GetQext();
  95. *Qsca = multi_layer_mie.GetQsca();
  96. *Qabs = multi_layer_mie.GetQabs();
  97. *Qbk = multi_layer_mie.GetQbk();
  98. *Qpr = multi_layer_mie.GetQpr();
  99. *g = multi_layer_mie.GetAsymmetryFactor();
  100. *Albedo = multi_layer_mie.GetAlbedo();
  101. S1 = multi_layer_mie.GetS1();
  102. S2 = multi_layer_mie.GetS2();
  103. } catch(const std::invalid_argument& ia) {
  104. // Will catch if multi_layer_mie fails or other errors.
  105. std::cerr << "Invalid argument: " << ia.what() << std::endl;
  106. throw std::invalid_argument(ia);
  107. return -1;
  108. }
  109. return 0;
  110. }
  111. //**********************************************************************************//
  112. // This function is just a wrapper to call the full 'nMie' function with fewer //
  113. // parameters, it is here mainly for compatibility with older versions of the //
  114. // program. Also, you can use it if you neither have a PEC layer nor want to define //
  115. // any limit for the maximum number of terms. //
  116. // //
  117. // Input parameters: //
  118. // L: Number of layers //
  119. // x: Array containing the size parameters of the layers [0..L-1] //
  120. // m: Array containing the relative refractive indexes of the layers [0..L-1] //
  121. // nTheta: Number of scattering angles //
  122. // Theta: Array containing all the scattering angles where the scattering //
  123. // amplitudes will be calculated //
  124. // //
  125. // Output parameters: //
  126. // Qext: Efficiency factor for extinction //
  127. // Qsca: Efficiency factor for scattering //
  128. // Qabs: Efficiency factor for absorption (Qabs = Qext - Qsca) //
  129. // Qbk: Efficiency factor for backscattering //
  130. // Qpr: Efficiency factor for the radiation pressure //
  131. // g: Asymmetry factor (g = (Qext-Qpr)/Qsca) //
  132. // Albedo: Single scattering albedo (Albedo = Qsca/Qext) //
  133. // S1, S2: Complex scattering amplitudes //
  134. // //
  135. // Return value: //
  136. // Number of multipolar expansion terms used for the calculations //
  137. //**********************************************************************************//
  138. int nMie(const int L, std::vector<double>& x, std::vector<std::complex<double> >& m, const int nTheta, std::vector<double>& Theta, double *Qext, double *Qsca, double *Qabs, double *Qbk, double *Qpr, double *g, double *Albedo, std::vector<std::complex<double> >& S1, std::vector<std::complex<double> >& S2) {
  139. return nmie::nMie(L, -1, x, m, nTheta, Theta, -1, Qext, Qsca, Qabs, Qbk, Qpr, g, Albedo, S1, S2);
  140. }
  141. //**********************************************************************************//
  142. // This function is just a wrapper to call the full 'nMie' function with fewer //
  143. // parameters, it is useful if you want to include a PEC layer but not a limit //
  144. // for the maximum number of terms. //
  145. // //
  146. // Input parameters: //
  147. // L: Number of layers //
  148. // pl: Index of PEC layer. If there is none just send -1 //
  149. // x: Array containing the size parameters of the layers [0..L-1] //
  150. // m: Array containing the relative refractive indexes of the layers [0..L-1] //
  151. // nTheta: Number of scattering angles //
  152. // Theta: Array containing all the scattering angles where the scattering //
  153. // amplitudes will be calculated //
  154. // //
  155. // Output parameters: //
  156. // Qext: Efficiency factor for extinction //
  157. // Qsca: Efficiency factor for scattering //
  158. // Qabs: Efficiency factor for absorption (Qabs = Qext - Qsca) //
  159. // Qbk: Efficiency factor for backscattering //
  160. // Qpr: Efficiency factor for the radiation pressure //
  161. // g: Asymmetry factor (g = (Qext-Qpr)/Qsca) //
  162. // Albedo: Single scattering albedo (Albedo = Qsca/Qext) //
  163. // S1, S2: Complex scattering amplitudes //
  164. // //
  165. // Return value: //
  166. // Number of multipolar expansion terms used for the calculations //
  167. //**********************************************************************************//
  168. int nMie(const int L, const int pl, std::vector<double>& x, std::vector<std::complex<double> >& m, const int nTheta, std::vector<double>& Theta, double *Qext, double *Qsca, double *Qabs, double *Qbk, double *Qpr, double *g, double *Albedo, std::vector<std::complex<double> >& S1, std::vector<std::complex<double> >& S2) {
  169. return nmie::nMie(L, pl, x, m, nTheta, Theta, -1, Qext, Qsca, Qabs, Qbk, Qpr, g, Albedo, S1, S2);
  170. }
  171. //**********************************************************************************//
  172. // This function is just a wrapper to call the full 'nMie' function with fewer //
  173. // parameters, it is useful if you want to include a limit for the maximum number //
  174. // of terms but not a PEC layer. //
  175. // //
  176. // Input parameters: //
  177. // L: Number of layers //
  178. // x: Array containing the size parameters of the layers [0..L-1] //
  179. // m: Array containing the relative refractive indexes of the layers [0..L-1] //
  180. // nTheta: Number of scattering angles //
  181. // Theta: Array containing all the scattering angles where the scattering //
  182. // amplitudes will be calculated //
  183. // nmax: Maximum number of multipolar expansion terms to be used for the //
  184. // calculations. Only use it if you know what you are doing, otherwise //
  185. // set this parameter to -1 and the function will calculate it //
  186. // //
  187. // Output parameters: //
  188. // Qext: Efficiency factor for extinction //
  189. // Qsca: Efficiency factor for scattering //
  190. // Qabs: Efficiency factor for absorption (Qabs = Qext - Qsca) //
  191. // Qbk: Efficiency factor for backscattering //
  192. // Qpr: Efficiency factor for the radiation pressure //
  193. // g: Asymmetry factor (g = (Qext-Qpr)/Qsca) //
  194. // Albedo: Single scattering albedo (Albedo = Qsca/Qext) //
  195. // S1, S2: Complex scattering amplitudes //
  196. // //
  197. // Return value: //
  198. // Number of multipolar expansion terms used for the calculations //
  199. //**********************************************************************************//
  200. int nMie(const int L, std::vector<double>& x, std::vector<std::complex<double> >& m, const int nTheta, std::vector<double>& Theta, const int nmax, double *Qext, double *Qsca, double *Qabs, double *Qbk, double *Qpr, double *g, double *Albedo, std::vector<std::complex<double> >& S1, std::vector<std::complex<double> >& S2) {
  201. return nmie::nMie(L, -1, x, m, nTheta, Theta, nmax, Qext, Qsca, Qabs, Qbk, Qpr, g, Albedo, S1, S2);
  202. }
  203. //**********************************************************************************//
  204. // This function emulates a C call to calculate complex electric and magnetic field //
  205. // in the surroundings and inside (TODO) the particle. //
  206. // //
  207. // Input parameters: //
  208. // L: Number of layers //
  209. // pl: Index of PEC layer. If there is none just send 0 (zero) //
  210. // x: Array containing the size parameters of the layers [0..L-1] //
  211. // m: Array containing the relative refractive indexes of the layers [0..L-1] //
  212. // nmax: Maximum number of multipolar expansion terms to be used for the //
  213. // calculations. Only use it if you know what you are doing, otherwise //
  214. // set this parameter to 0 (zero) and the function will calculate it. //
  215. // ncoord: Number of coordinate points //
  216. // Coords: Array containing all coordinates where the complex electric and //
  217. // magnetic fields will be calculated //
  218. // //
  219. // Output parameters: //
  220. // E, H: Complex electric and magnetic field at the provided coordinates //
  221. // //
  222. // Return value: //
  223. // Number of multipolar expansion terms used for the calculations //
  224. //**********************************************************************************//
  225. int nField(const int L, const int pl, const std::vector<double>& x, const std::vector<std::complex<double> >& m, const int nmax, const int ncoord, const std::vector<double>& Xp_vec, const std::vector<double>& Yp_vec, const std::vector<double>& Zp_vec, std::vector<std::vector<std::complex<double> > >& E, std::vector<std::vector<std::complex<double> > >& H) {
  226. if (x.size() != L || m.size() != L)
  227. throw std::invalid_argument("Declared number of layers do not fit x and m!");
  228. if (Xp_vec.size() != ncoord || Yp_vec.size() != ncoord || Zp_vec.size() != ncoord
  229. || E.size() != ncoord || H.size() != ncoord)
  230. throw std::invalid_argument("Declared number of coords do not fit Xp, Yp, Zp, E, or H!");
  231. for (auto f:E)
  232. if (f.size() != 3)
  233. throw std::invalid_argument("Field E is not 3D!");
  234. for (auto f:H)
  235. if (f.size() != 3)
  236. throw std::invalid_argument("Field H is not 3D!");
  237. try {
  238. MultiLayerMie multi_layer_mie;
  239. //multi_layer_mie.SetPECLayer(pl);
  240. multi_layer_mie.SetLayersSize(x);
  241. multi_layer_mie.SetLayersIndex(m);
  242. multi_layer_mie.SetFieldCoords({Xp_vec, Yp_vec, Zp_vec});
  243. multi_layer_mie.RunFieldCalculation();
  244. E = multi_layer_mie.GetFieldE();
  245. H = multi_layer_mie.GetFieldH();
  246. //multi_layer_mie.GetFailed();
  247. } catch(const std::invalid_argument& ia) {
  248. // Will catch if multi_layer_mie fails or other errors.
  249. std::cerr << "Invalid argument: " << ia.what() << std::endl;
  250. throw std::invalid_argument(ia);
  251. return - 1;
  252. }
  253. return 0;
  254. }
  255. // ********************************************************************** //
  256. // Returns previously calculated Qext //
  257. // ********************************************************************** //
  258. double MultiLayerMie::GetQext() {
  259. if (!isMieCalculated_)
  260. throw std::invalid_argument("You should run calculations before result request!");
  261. return Qext_;
  262. }
  263. // ********************************************************************** //
  264. // Returns previously calculated Qabs //
  265. // ********************************************************************** //
  266. double MultiLayerMie::GetQabs() {
  267. if (!isMieCalculated_)
  268. throw std::invalid_argument("You should run calculations before result request!");
  269. return Qabs_;
  270. }
  271. // ********************************************************************** //
  272. // Returns previously calculated Qsca //
  273. // ********************************************************************** //
  274. double MultiLayerMie::GetQsca() {
  275. if (!isMieCalculated_)
  276. throw std::invalid_argument("You should run calculations before result request!");
  277. return Qsca_;
  278. }
  279. // ********************************************************************** //
  280. // Returns previously calculated Qbk //
  281. // ********************************************************************** //
  282. double MultiLayerMie::GetQbk() {
  283. if (!isMieCalculated_)
  284. throw std::invalid_argument("You should run calculations before result request!");
  285. return Qbk_;
  286. }
  287. // ********************************************************************** //
  288. // Returns previously calculated Qpr //
  289. // ********************************************************************** //
  290. double MultiLayerMie::GetQpr() {
  291. if (!isMieCalculated_)
  292. throw std::invalid_argument("You should run calculations before result request!");
  293. return Qpr_;
  294. }
  295. // ********************************************************************** //
  296. // Returns previously calculated assymetry factor //
  297. // ********************************************************************** //
  298. double MultiLayerMie::GetAsymmetryFactor() {
  299. if (!isMieCalculated_)
  300. throw std::invalid_argument("You should run calculations before result request!");
  301. return asymmetry_factor_;
  302. }
  303. // ********************************************************************** //
  304. // Returns previously calculated Albedo //
  305. // ********************************************************************** //
  306. double MultiLayerMie::GetAlbedo() {
  307. if (!isMieCalculated_)
  308. throw std::invalid_argument("You should run calculations before result request!");
  309. return albedo_;
  310. }
  311. // ********************************************************************** //
  312. // Returns previously calculated S1 //
  313. // ********************************************************************** //
  314. std::vector<std::complex<double> > MultiLayerMie::GetS1() {
  315. if (!isMieCalculated_)
  316. throw std::invalid_argument("You should run calculations before result request!");
  317. return S1_;
  318. }
  319. // ********************************************************************** //
  320. // Returns previously calculated S2 //
  321. // ********************************************************************** //
  322. std::vector<std::complex<double> > MultiLayerMie::GetS2() {
  323. if (!isMieCalculated_)
  324. throw std::invalid_argument("You should run calculations before result request!");
  325. return S2_;
  326. }
  327. // ********************************************************************** //
  328. // Modify scattering (theta) angles //
  329. // ********************************************************************** //
  330. void MultiLayerMie::SetAngles(const std::vector<double>& angles) {
  331. isIntCoeffsCalc_ = false;
  332. isExtCoeffsCalc_ = false;
  333. isMieCalculated_ = false;
  334. theta_ = angles;
  335. }
  336. // ********************************************************************** //
  337. // Modify size of all layers //
  338. // ********************************************************************** //
  339. void MultiLayerMie::SetLayersSize(const std::vector<double>& layer_size) {
  340. isIntCoeffsCalc_ = false;
  341. isExtCoeffsCalc_ = false;
  342. isMieCalculated_ = false;
  343. size_param_.clear();
  344. double prev_layer_size = 0.0;
  345. for (auto curr_layer_size : layer_size) {
  346. if (curr_layer_size <= 0.0)
  347. throw std::invalid_argument("Size parameter should be positive!");
  348. if (prev_layer_size > curr_layer_size)
  349. throw std::invalid_argument
  350. ("Size parameter for next layer should be larger than the previous one!");
  351. prev_layer_size = curr_layer_size;
  352. size_param_.push_back(curr_layer_size);
  353. }
  354. }
  355. // ********************************************************************** //
  356. // Modify refractive index of all layers //
  357. // ********************************************************************** //
  358. void MultiLayerMie::SetLayersIndex(const std::vector< std::complex<double> >& index) {
  359. isIntCoeffsCalc_ = false;
  360. isExtCoeffsCalc_ = false;
  361. isMieCalculated_ = false;
  362. refr_index_ = index;
  363. }
  364. // ********************************************************************** //
  365. // Modify coordinates for field calculation //
  366. // ********************************************************************** //
  367. void MultiLayerMie::SetFieldCoords(const std::vector< std::vector<double> >& coords) {
  368. if (coords.size() != 3)
  369. throw std::invalid_argument("Error! Wrong dimension of field monitor points!");
  370. if (coords[0].size() != coords[1].size() || coords[0].size() != coords[2].size())
  371. throw std::invalid_argument("Error! Missing coordinates for field monitor points!");
  372. coords_ = coords;
  373. }
  374. // ********************************************************************** //
  375. // ********************************************************************** //
  376. // ********************************************************************** //
  377. void MultiLayerMie::SetPECLayer(int layer_position) {
  378. isIntCoeffsCalc_ = false;
  379. isExtCoeffsCalc_ = false;
  380. isMieCalculated_ = false;
  381. if (layer_position < 0)
  382. throw std::invalid_argument("Error! Layers are numbered from 0!");
  383. PEC_layer_position_ = layer_position;
  384. }
  385. // ********************************************************************** //
  386. // Set maximun number of terms to be used //
  387. // ********************************************************************** //
  388. void MultiLayerMie::SetMaxTerms(int nmax) {
  389. isIntCoeffsCalc_ = false;
  390. isExtCoeffsCalc_ = false;
  391. isMieCalculated_ = false;
  392. nmax_preset_ = nmax;
  393. }
  394. // ********************************************************************** //
  395. // ********************************************************************** //
  396. // ********************************************************************** //
  397. double MultiLayerMie::GetSizeParameter() {
  398. if (size_param_.size() > 0)
  399. return size_param_.back();
  400. else
  401. return 0;
  402. }
  403. // ********************************************************************** //
  404. // Clear layer information //
  405. // ********************************************************************** //
  406. void MultiLayerMie::ClearLayers() {
  407. isIntCoeffsCalc_ = false;
  408. isExtCoeffsCalc_ = false;
  409. isMieCalculated_ = false;
  410. size_param_.clear();
  411. refr_index_.clear();
  412. }
  413. // ********************************************************************** //
  414. // ********************************************************************** //
  415. // ********************************************************************** //
  416. // Computational core
  417. // ********************************************************************** //
  418. // ********************************************************************** //
  419. // ********************************************************************** //
  420. // ********************************************************************** //
  421. // Calculate calcNstop - equation (17) //
  422. // ********************************************************************** //
  423. void MultiLayerMie::calcNstop() {
  424. const double& xL = size_param_.back();
  425. if (xL <= 8) {
  426. nmax_ = round(xL + 4.0*pow(xL, 1.0/3.0) + 1);
  427. } else if (xL <= 4200) {
  428. nmax_ = round(xL + 4.05*pow(xL, 1.0/3.0) + 2);
  429. } else {
  430. nmax_ = round(xL + 4.0*pow(xL, 1.0/3.0) + 2);
  431. }
  432. }
  433. // ********************************************************************** //
  434. // Maximum number of terms required for the calculation //
  435. // ********************************************************************** //
  436. void MultiLayerMie::calcNmax(int first_layer) {
  437. int ri, riM1;
  438. const std::vector<double>& x = size_param_;
  439. const std::vector<std::complex<double> >& m = refr_index_;
  440. calcNstop(); // Set initial nmax_ value
  441. for (int i = first_layer; i < x.size(); i++) {
  442. if (i > PEC_layer_position_)
  443. ri = round(std::abs(x[i]*m[i]));
  444. else
  445. ri = 0;
  446. nmax_ = std::max(nmax_, ri);
  447. // first layer is pec, if pec is present
  448. if ((i > first_layer) && ((i - 1) > PEC_layer_position_))
  449. riM1 = round(std::abs(x[i - 1]* m[i]));
  450. else
  451. riM1 = 0;
  452. nmax_ = std::max(nmax_, riM1);
  453. }
  454. nmax_ += 15; // Final nmax_ value
  455. }
  456. // ********************************************************************** //
  457. // Calculate an - equation (5) //
  458. // ********************************************************************** //
  459. std::complex<double> MultiLayerMie::calc_an(int n, double XL, std::complex<double> Ha, std::complex<double> mL,
  460. std::complex<double> PsiXL, std::complex<double> ZetaXL,
  461. std::complex<double> PsiXLM1, std::complex<double> ZetaXLM1) {
  462. std::complex<double> Num = (Ha/mL + n/XL)*PsiXL - PsiXLM1;
  463. std::complex<double> Denom = (Ha/mL + n/XL)*ZetaXL - ZetaXLM1;
  464. return Num/Denom;
  465. }
  466. // ********************************************************************** //
  467. // Calculate bn - equation (6) //
  468. // ********************************************************************** //
  469. std::complex<double> MultiLayerMie::calc_bn(int n, double XL, std::complex<double> Hb, std::complex<double> mL,
  470. std::complex<double> PsiXL, std::complex<double> ZetaXL,
  471. std::complex<double> PsiXLM1, std::complex<double> ZetaXLM1) {
  472. std::complex<double> Num = (mL*Hb + n/XL)*PsiXL - PsiXLM1;
  473. std::complex<double> Denom = (mL*Hb + n/XL)*ZetaXL - ZetaXLM1;
  474. return Num/Denom;
  475. }
  476. // ********************************************************************** //
  477. // Calculates S1 - equation (25a) //
  478. // ********************************************************************** //
  479. std::complex<double> MultiLayerMie::calc_S1(int n, std::complex<double> an, std::complex<double> bn,
  480. double Pi, double Tau) {
  481. return double(n + n + 1)*(Pi*an + Tau*bn)/double(n*n + n);
  482. }
  483. // ********************************************************************** //
  484. // Calculates S2 - equation (25b) (it's the same as (25a), just switches //
  485. // Pi and Tau) //
  486. // ********************************************************************** //
  487. std::complex<double> MultiLayerMie::calc_S2(int n, std::complex<double> an, std::complex<double> bn,
  488. double Pi, double Tau) {
  489. return calc_S1(n, an, bn, Tau, Pi);
  490. }
  491. //**********************************************************************************//
  492. // This function calculates the logarithmic derivatives of the Riccati-Bessel //
  493. // functions (D1 and D3) for a complex argument (z). //
  494. // Equations (16a), (16b) and (18a) - (18d) //
  495. // //
  496. // Input parameters: //
  497. // z: Complex argument to evaluate D1 and D3 //
  498. // nmax_: Maximum number of terms to calculate D1 and D3 //
  499. // //
  500. // Output parameters: //
  501. // D1, D3: Logarithmic derivatives of the Riccati-Bessel functions //
  502. //**********************************************************************************//
  503. void MultiLayerMie::calcD1D3(const std::complex<double> z,
  504. std::vector<std::complex<double> >& D1,
  505. std::vector<std::complex<double> >& D3) {
  506. // Downward recurrence for D1 - equations (16a) and (16b)
  507. D1[nmax_] = std::complex<double>(0.0, 0.0);
  508. const std::complex<double> zinv = std::complex<double>(1.0, 0.0)/z;
  509. for (int n = nmax_; n > 0; n--) {
  510. D1[n - 1] = double(n)*zinv - 1.0/(D1[n] + double(n)*zinv);
  511. }
  512. if (std::abs(D1[0]) > 100000.0)
  513. throw std::invalid_argument("Unstable D1! Please, try to change input parameters!\n");
  514. // Upward recurrence for PsiZeta and D3 - equations (18a) - (18d)
  515. PsiZeta_[0] = 0.5*(1.0 - std::complex<double>(std::cos(2.0*z.real()), std::sin(2.0*z.real()))
  516. *std::exp(-2.0*z.imag()));
  517. D3[0] = std::complex<double>(0.0, 1.0);
  518. for (int n = 1; n <= nmax_; n++) {
  519. PsiZeta_[n] = PsiZeta_[n - 1]*(static_cast<double>(n)*zinv - D1[n - 1])
  520. *(static_cast<double>(n)*zinv- D3[n - 1]);
  521. D3[n] = D1[n] + std::complex<double>(0.0, 1.0)/PsiZeta_[n];
  522. }
  523. }
  524. //**********************************************************************************//
  525. // This function calculates the Riccati-Bessel functions (Psi and Zeta) for a //
  526. // complex argument (z). //
  527. // Equations (20a) - (21b) //
  528. // //
  529. // Input parameters: //
  530. // z: Complex argument to evaluate Psi and Zeta //
  531. // nmax: Maximum number of terms to calculate Psi and Zeta //
  532. // //
  533. // Output parameters: //
  534. // Psi, Zeta: Riccati-Bessel functions //
  535. //**********************************************************************************//
  536. void MultiLayerMie::calcPsiZeta(std::complex<double> z,
  537. std::vector<std::complex<double> >& Psi,
  538. std::vector<std::complex<double> >& Zeta) {
  539. std::vector<std::complex<double> > D1(nmax_ + 1), D3(nmax_ + 1);
  540. // First, calculate the logarithmic derivatives
  541. calcD1D3(z, D1, D3);
  542. // Now, use the upward recurrence to calculate Psi and Zeta - equations (20a) - (21b)
  543. std::complex<double> c_i(0.0, 1.0);
  544. Psi[0] = std::sin(z);
  545. Zeta[0] = std::sin(z) - c_i*std::cos(z);
  546. for (int n = 1; n <= nmax_; n++) {
  547. Psi[n] = Psi[n - 1]*(static_cast<double>(n)/z - D1[n - 1]);
  548. Zeta[n] = Zeta[n - 1]*(static_cast<double>(n)/z - D3[n - 1]);
  549. }
  550. }
  551. //**********************************************************************************//
  552. // This function calculates the spherical Bessel (jn) and Hankel (h1n) functions //
  553. // and their derivatives for a given complex value z. See pag. 87 B&H. //
  554. // //
  555. // Input parameters: //
  556. // z: Complex argument to evaluate jn and h1n //
  557. // nmax_: Maximum number of terms to calculate jn and h1n //
  558. // //
  559. // Output parameters: //
  560. // jn, h1n: Spherical Bessel and Hankel functions //
  561. // jnp, h1np: Derivatives of the spherical Bessel and Hankel functions //
  562. // //
  563. // What we actually calculate are the Ricatti-Bessel fucntions and then simply //
  564. // evaluate the spherical Bessel and Hankel functions and their derivatives //
  565. // using the relations: //
  566. // //
  567. // j[n] = Psi[n]/z //
  568. // j'[n] = 0.5*(Psi[n-1]-Psi[n+1]-jn[n])/z //
  569. // h1[n] = Zeta[n]/z //
  570. // h1'[n] = 0.5*(Zeta[n-1]-Zeta[n+1]-h1n[n])/z //
  571. // //
  572. //**********************************************************************************//
  573. void MultiLayerMie::sbesjh(std::complex<double> z,
  574. std::vector<std::complex<double> >& jn, std::vector<std::complex<double> >& jnp,
  575. std::vector<std::complex<double> >& h1n, std::vector<std::complex<double> >& h1np) {
  576. std::vector<std::complex<double> > Psi(nmax_ + 1), Zeta(nmax_ + 1);
  577. // First, calculate the Riccati-Bessel functions
  578. calcPsiZeta(z, Psi, Zeta);
  579. // Now, calculate Spherical Bessel and Hankel functions and their derivatives
  580. for (int n = 0; n < nmax_; n++) {
  581. jn[n] = Psi[n]/z;
  582. h1n[n] = Zeta[n]/z;
  583. if (n == 0) {
  584. jnp[n] = -Psi[1]/z - 0.5*jn[n]/z;
  585. h1np[n] = -Zeta[1]/z - 0.5*h1n[n]/z;
  586. } else {
  587. jnp[n] = 0.5*(Psi[n - 1] - Psi[n + 1] - jn[n])/z;
  588. h1np[n] = 0.5*(Zeta[n - 1] - Zeta[n + 1] - h1n[n])/z;
  589. }
  590. }
  591. }
  592. //**********************************************************************************//
  593. // This function calculates Pi and Tau for a given value of cos(Theta). //
  594. // Equations (26a) - (26c) //
  595. // //
  596. // Input parameters: //
  597. // nmax_: Maximum number of terms to calculate Pi and Tau //
  598. // nTheta: Number of scattering angles //
  599. // Theta: Array containing all the scattering angles where the scattering //
  600. // amplitudes will be calculated //
  601. // //
  602. // Output parameters: //
  603. // Pi, Tau: Angular functions Pi and Tau, as defined in equations (26a) - (26c) //
  604. //**********************************************************************************//
  605. void MultiLayerMie::calcPiTau(const double& costheta,
  606. std::vector<double>& Pi, std::vector<double>& Tau) {
  607. int n;
  608. //****************************************************//
  609. // Equations (26a) - (26c) //
  610. //****************************************************//
  611. // Initialize Pi and Tau
  612. Pi[0] = 1.0;
  613. Tau[0] = costheta;
  614. // Calculate the actual values
  615. if (nmax_ > 1) {
  616. Pi[1] = 3*costheta*Pi[0];
  617. Tau[1] = 2*costheta*Pi[1] - 3*Pi[0];
  618. for (n = 2; n < nmax_; n++) {
  619. Pi[n] = ((n + n + 1)*costheta*Pi[n - 1] - (n + 1)*Pi[n - 2])/n;
  620. Tau[n] = (n + 1)*costheta*Pi[n] - (n + 2)*Pi[n - 1];
  621. }
  622. }
  623. } // end of MultiLayerMie::calcPiTau(...)
  624. void MultiLayerMie::calcSpherHarm(const double Rho, const double Phi, const double Theta,
  625. const std::complex<double>& zn, const std::complex<double>& dzn,
  626. const double& Pi, const double& Tau, const double& n,
  627. std::vector<std::complex<double> >& Mo1n, std::vector<std::complex<double> >& Me1n,
  628. std::vector<std::complex<double> >& No1n, std::vector<std::complex<double> >& Ne1n) {
  629. // using eq 4.50 in BH
  630. std::complex<double> c_zero(0.0, 0.0);
  631. std::complex<double> deriv = Rho*dzn + zn;
  632. using std::sin;
  633. using std::cos;
  634. Mo1n[0] = c_zero;
  635. Mo1n[1] = cos(Phi)*Pi*zn;
  636. Mo1n[2] = -sin(Phi)*Tau*zn;
  637. Me1n[0] = c_zero;
  638. Me1n[1] = -sin(Phi)*Pi*zn;
  639. Me1n[2] = -cos(Phi)*Tau*zn;
  640. No1n[0] = sin(Phi)*(n*n + n)*sin(Theta)*Pi*zn/Rho;
  641. No1n[1] = sin(Phi)*Tau*deriv/Rho;
  642. No1n[2] = cos(Phi)*Pi*deriv/Rho;
  643. Ne1n[0] = cos(Phi)*(n*n + n)*sin(Theta)*Pi*zn/Rho;
  644. Ne1n[1] = cos(Phi)*Tau*deriv/Rho;
  645. Ne1n[2] = -sin(Phi)*Pi*deriv/Rho;
  646. } // end of MultiLayerMie::calcSpherHarm(...)
  647. //**********************************************************************************//
  648. // This function calculates the scattering coefficients required to calculate //
  649. // both the near- and far-field parameters. //
  650. // //
  651. // Input parameters: //
  652. // L: Number of layers //
  653. // pl: Index of PEC layer. If there is none just send -1 //
  654. // x: Array containing the size parameters of the layers [0..L-1] //
  655. // m: Array containing the relative refractive indexes of the layers [0..L-1] //
  656. // nmax: Maximum number of multipolar expansion terms to be used for the //
  657. // calculations. Only use it if you know what you are doing, otherwise //
  658. // set this parameter to -1 and the function will calculate it. //
  659. // //
  660. // Output parameters: //
  661. // an, bn: Complex scattering amplitudes //
  662. // //
  663. // Return value: //
  664. // Number of multipolar expansion terms used for the calculations //
  665. //**********************************************************************************//
  666. void MultiLayerMie::ExtScattCoeffs() {
  667. isExtCoeffsCalc_ = false;
  668. const std::vector<double>& x = size_param_;
  669. const std::vector<std::complex<double> >& m = refr_index_;
  670. const int& pl = PEC_layer_position_;
  671. const int L = refr_index_.size();
  672. //************************************************************************//
  673. // Calculate the index of the first layer. It can be either 0 (default) //
  674. // or the index of the outermost PEC layer. In the latter case all layers //
  675. // below the PEC are discarded. //
  676. // ***********************************************************************//
  677. // TODO, is it possible for PEC to have a zero index? If yes than
  678. // is should be:
  679. // int fl = (pl > - 1) ? pl : 0;
  680. // This will give the same result, however, it corresponds the
  681. // logic - if there is PEC, than first layer is PEC.
  682. // Well, I followed the logic: First layer is always zero unless it has
  683. // an upper PEC layer.
  684. int fl = (pl > 0) ? pl : 0;
  685. if (nmax_preset_ <= 0) calcNmax(fl);
  686. else nmax_ = nmax_preset_;
  687. std::complex<double> z1, z2;
  688. //**************************************************************************//
  689. // Note that since Fri, Nov 14, 2014 all arrays start from 0 (zero), which //
  690. // means that index = layer number - 1 or index = n - 1. The only exception //
  691. // are the arrays for representing D1, D3 and Q because they need a value //
  692. // for the index 0 (zero), hence it is important to consider this shift //
  693. // between different arrays. The change was done to optimize memory usage. //
  694. //**************************************************************************//
  695. // Allocate memory to the arrays
  696. std::vector<std::complex<double> > D1_mlxl(nmax_ + 1), D1_mlxlM1(nmax_ + 1),
  697. D3_mlxl(nmax_ + 1), D3_mlxlM1(nmax_ + 1);
  698. std::vector<std::vector<std::complex<double> > > Q(L), Ha(L), Hb(L);
  699. for (int l = 0; l < L; l++) {
  700. Q[l].resize(nmax_ + 1);
  701. Ha[l].resize(nmax_);
  702. Hb[l].resize(nmax_);
  703. }
  704. an_.resize(nmax_);
  705. bn_.resize(nmax_);
  706. PsiZeta_.resize(nmax_ + 1);
  707. std::vector<std::complex<double> > PsiXL(nmax_ + 1), ZetaXL(nmax_ + 1);
  708. //*************************************************//
  709. // Calculate D1 and D3 for z1 in the first layer //
  710. //*************************************************//
  711. if (fl == pl) { // PEC layer
  712. for (int n = 0; n <= nmax_; n++) {
  713. D1_mlxl[n] = std::complex<double>(0.0, - 1.0);
  714. D3_mlxl[n] = std::complex<double>(0.0, 1.0);
  715. }
  716. } else { // Regular layer
  717. z1 = x[fl]* m[fl];
  718. // Calculate D1 and D3
  719. calcD1D3(z1, D1_mlxl, D3_mlxl);
  720. }
  721. //******************************************************************//
  722. // Calculate Ha and Hb in the first layer - equations (7a) and (8a) //
  723. //******************************************************************//
  724. for (int n = 0; n < nmax_; n++) {
  725. Ha[fl][n] = D1_mlxl[n + 1];
  726. Hb[fl][n] = D1_mlxl[n + 1];
  727. }
  728. //*****************************************************//
  729. // Iteration from the second layer to the last one (L) //
  730. //*****************************************************//
  731. std::complex<double> Temp, Num, Denom;
  732. std::complex<double> G1, G2;
  733. for (int l = fl + 1; l < L; l++) {
  734. //************************************************************//
  735. //Calculate D1 and D3 for z1 and z2 in the layers fl + 1..L //
  736. //************************************************************//
  737. z1 = x[l]*m[l];
  738. z2 = x[l - 1]*m[l];
  739. //Calculate D1 and D3 for z1
  740. calcD1D3(z1, D1_mlxl, D3_mlxl);
  741. //Calculate D1 and D3 for z2
  742. calcD1D3(z2, D1_mlxlM1, D3_mlxlM1);
  743. //*********************************************//
  744. //Calculate Q, Ha and Hb in the layers fl + 1..L //
  745. //*********************************************//
  746. // Upward recurrence for Q - equations (19a) and (19b)
  747. Num = std::exp(-2.0*(z1.imag() - z2.imag()))
  748. *std::complex<double>(std::cos(-2.0*z2.real()) - std::exp(-2.0*z2.imag()), std::sin(-2.0*z2.real()));
  749. Denom = std::complex<double>(std::cos(-2.0*z1.real()) - std::exp(-2.0*z1.imag()), std::sin(-2.0*z1.real()));
  750. Q[l][0] = Num/Denom;
  751. for (int n = 1; n <= nmax_; n++) {
  752. Num = (z1*D1_mlxl[n] + double(n))*(double(n) - z1*D3_mlxl[n - 1]);
  753. Denom = (z2*D1_mlxlM1[n] + double(n))*(double(n) - z2*D3_mlxlM1[n - 1]);
  754. Q[l][n] = ((pow2(x[l - 1]/x[l])* Q[l][n - 1])*Num)/Denom;
  755. }
  756. // Upward recurrence for Ha and Hb - equations (7b), (8b) and (12) - (15)
  757. for (int n = 1; n <= nmax_; n++) {
  758. //Ha
  759. if ((l - 1) == pl) { // The layer below the current one is a PEC layer
  760. G1 = -D1_mlxlM1[n];
  761. G2 = -D3_mlxlM1[n];
  762. } else {
  763. G1 = (m[l]*Ha[l - 1][n - 1]) - (m[l - 1]*D1_mlxlM1[n]);
  764. G2 = (m[l]*Ha[l - 1][n - 1]) - (m[l - 1]*D3_mlxlM1[n]);
  765. } // end of if PEC
  766. Temp = Q[l][n]*G1;
  767. Num = (G2*D1_mlxl[n]) - (Temp*D3_mlxl[n]);
  768. Denom = G2 - Temp;
  769. Ha[l][n - 1] = Num/Denom;
  770. //Hb
  771. if ((l - 1) == pl) { // The layer below the current one is a PEC layer
  772. G1 = Hb[l - 1][n - 1];
  773. G2 = Hb[l - 1][n - 1];
  774. } else {
  775. G1 = (m[l - 1]*Hb[l - 1][n - 1]) - (m[l]*D1_mlxlM1[n]);
  776. G2 = (m[l - 1]*Hb[l - 1][n - 1]) - (m[l]*D3_mlxlM1[n]);
  777. } // end of if PEC
  778. Temp = Q[l][n]*G1;
  779. Num = (G2*D1_mlxl[n]) - (Temp* D3_mlxl[n]);
  780. Denom = (G2- Temp);
  781. Hb[l][n - 1] = (Num/ Denom);
  782. } // end of for Ha and Hb terms
  783. } // end of for layers iteration
  784. //**************************************//
  785. //Calculate Psi and Zeta for XL //
  786. //**************************************//
  787. // Calculate PsiXL and ZetaXL
  788. calcPsiZeta(x[L - 1], PsiXL, ZetaXL);
  789. //*********************************************************************//
  790. // Finally, we calculate the scattering coefficients (an and bn) and //
  791. // the angular functions (Pi and Tau). Note that for these arrays the //
  792. // first layer is 0 (zero), in future versions all arrays will follow //
  793. // this convention to save memory. (13 Nov, 2014) //
  794. //*********************************************************************//
  795. for (int n = 0; n < nmax_; n++) {
  796. //********************************************************************//
  797. //Expressions for calculating an and bn coefficients are not valid if //
  798. //there is only one PEC layer (ie, for a simple PEC sphere). //
  799. //********************************************************************//
  800. if (pl < (L - 1)) {
  801. an_[n] = calc_an(n + 1, x[L - 1], Ha[L - 1][n], m[L - 1], PsiXL[n + 1], ZetaXL[n + 1], PsiXL[n], ZetaXL[n]);
  802. bn_[n] = calc_bn(n + 1, x[L - 1], Hb[L - 1][n], m[L - 1], PsiXL[n + 1], ZetaXL[n + 1], PsiXL[n], ZetaXL[n]);
  803. } else {
  804. an_[n] = calc_an(n + 1, x[L - 1], std::complex<double>(0.0, 0.0), std::complex<double>(1.0, 0.0), PsiXL[n + 1], ZetaXL[n + 1], PsiXL[n], ZetaXL[n]);
  805. bn_[n] = PsiXL[n + 1]/ZetaXL[n + 1];
  806. }
  807. } // end of for an and bn terms
  808. isExtCoeffsCalc_ = true;
  809. } // end of MultiLayerMie::ExtScattCoeffs(...)
  810. //**********************************************************************************//
  811. // This function calculates the actual scattering parameters and amplitudes //
  812. // //
  813. // Input parameters: //
  814. // L: Number of layers //
  815. // pl: Index of PEC layer. If there is none just send -1 //
  816. // x: Array containing the size parameters of the layers [0..L-1] //
  817. // m: Array containing the relative refractive indexes of the layers [0..L-1] //
  818. // nTheta: Number of scattering angles //
  819. // Theta: Array containing all the scattering angles where the scattering //
  820. // amplitudes will be calculated //
  821. // nmax_: Maximum number of multipolar expansion terms to be used for the //
  822. // calculations. Only use it if you know what you are doing, otherwise //
  823. // set this parameter to -1 and the function will calculate it //
  824. // //
  825. // Output parameters: //
  826. // Qext: Efficiency factor for extinction //
  827. // Qsca: Efficiency factor for scattering //
  828. // Qabs: Efficiency factor for absorption (Qabs = Qext - Qsca) //
  829. // Qbk: Efficiency factor for backscattering //
  830. // Qpr: Efficiency factor for the radiation pressure //
  831. // g: Asymmetry factor (g = (Qext-Qpr)/Qsca) //
  832. // Albedo: Single scattering albedo (Albedo = Qsca/Qext) //
  833. // S1, S2: Complex scattering amplitudes //
  834. // //
  835. // Return value: //
  836. // Number of multipolar expansion terms used for the calculations //
  837. //**********************************************************************************//
  838. void MultiLayerMie::RunMieCalculation() {
  839. if (size_param_.size() != refr_index_.size())
  840. throw std::invalid_argument("Each size parameter should have only one index!");
  841. if (size_param_.size() == 0)
  842. throw std::invalid_argument("Initialize model first!");
  843. const std::vector<double>& x = size_param_;
  844. isIntCoeffsCalc_ = false;
  845. isExtCoeffsCalc_ = false;
  846. isMieCalculated_ = false;
  847. // Calculate scattering coefficients
  848. ExtScattCoeffs();
  849. if (!isExtCoeffsCalc_) // TODO seems to be unreachable
  850. throw std::invalid_argument("Calculation of scattering coefficients failed!");
  851. // Initialize the scattering parameters
  852. Qext_ = 0;
  853. Qsca_ = 0;
  854. Qabs_ = 0;
  855. Qbk_ = 0;
  856. Qpr_ = 0;
  857. asymmetry_factor_ = 0;
  858. albedo_ = 0;
  859. Qsca_ch_.clear();
  860. Qext_ch_.clear();
  861. Qabs_ch_.clear();
  862. Qbk_ch_.clear();
  863. Qpr_ch_.clear();
  864. Qsca_ch_.resize(nmax_ - 1);
  865. Qext_ch_.resize(nmax_ - 1);
  866. Qabs_ch_.resize(nmax_ - 1);
  867. Qbk_ch_.resize(nmax_ - 1);
  868. Qpr_ch_.resize(nmax_ - 1);
  869. Qsca_ch_norm_.resize(nmax_ - 1);
  870. Qext_ch_norm_.resize(nmax_ - 1);
  871. Qabs_ch_norm_.resize(nmax_ - 1);
  872. Qbk_ch_norm_.resize(nmax_ - 1);
  873. Qpr_ch_norm_.resize(nmax_ - 1);
  874. // Initialize the scattering amplitudes
  875. std::vector<std::complex<double> > tmp1(theta_.size(),std::complex<double>(0.0, 0.0));
  876. S1_.swap(tmp1);
  877. S2_ = S1_;
  878. std::vector<double> Pi(nmax_), Tau(nmax_);
  879. std::complex<double> Qbktmp(0.0, 0.0);
  880. std::vector< std::complex<double> > Qbktmp_ch(nmax_ - 1, Qbktmp);
  881. // By using downward recurrence we avoid loss of precision due to float rounding errors
  882. // See: https://docs.oracle.com/cd/E19957-01/806-3568/ncg_goldberg.html
  883. // http://en.wikipedia.org/wiki/Loss_of_significance
  884. for (int i = nmax_ - 2; i >= 0; i--) {
  885. const int n = i + 1;
  886. // Equation (27)
  887. Qext_ch_norm_[i] = (an_[i].real() + bn_[i].real());
  888. Qext_ch_[i] = (n + n + 1.0)*Qext_ch_norm_[i];
  889. //Qext_ch_[i] = (n + n + 1)*(an_[i].real() + bn_[i].real());
  890. Qext_ += Qext_ch_[i];
  891. // Equation (28)
  892. Qsca_ch_norm_[i] = (an_[i].real()*an_[i].real() + an_[i].imag()*an_[i].imag()
  893. + bn_[i].real()*bn_[i].real() + bn_[i].imag()*bn_[i].imag());
  894. Qsca_ch_[i] = (n + n + 1.0)*Qsca_ch_norm_[i];
  895. Qsca_ += Qsca_ch_[i];
  896. // Qsca_ch_[i] += (n + n + 1)*(an_[i].real()*an_[i].real() + an_[i].imag()*an_[i].imag()
  897. // + bn_[i].real()*bn_[i].real() + bn_[i].imag()*bn_[i].imag());
  898. // Equation (29) TODO We must check carefully this equation. If we
  899. // remove the typecast to double then the result changes. Which is
  900. // the correct one??? Ovidio (2014/12/10) With cast ratio will
  901. // give double, without cast (n + n + 1)/(n*(n + 1)) will be
  902. // rounded to integer. Tig (2015/02/24)
  903. Qpr_ch_[i]=((n*(n + 2)/(n + 1))*((an_[i]*std::conj(an_[n]) + bn_[i]*std::conj(bn_[n])).real())
  904. + ((double)(n + n + 1)/(n*(n + 1)))*(an_[i]*std::conj(bn_[i])).real());
  905. Qpr_ += Qpr_ch_[i];
  906. // Equation (33)
  907. Qbktmp_ch[i] = (double)(n + n + 1)*(1 - 2*(n % 2))*(an_[i]- bn_[i]);
  908. Qbktmp += Qbktmp_ch[i];
  909. // Calculate the scattering amplitudes (S1 and S2) //
  910. // Equations (25a) - (25b) //
  911. for (int t = 0; t < theta_.size(); t++) {
  912. calcPiTau(std::cos(theta_[t]), Pi, Tau);
  913. S1_[t] += calc_S1(n, an_[i], bn_[i], Pi[i], Tau[i]);
  914. S2_[t] += calc_S2(n, an_[i], bn_[i], Pi[i], Tau[i]);
  915. }
  916. }
  917. double x2 = pow2(x.back());
  918. Qext_ = 2.0*(Qext_)/x2; // Equation (27)
  919. for (double& Q : Qext_ch_) Q = 2.0*Q/x2;
  920. Qsca_ = 2.0*(Qsca_)/x2; // Equation (28)
  921. for (double& Q : Qsca_ch_) Q = 2.0*Q/x2;
  922. //for (double& Q : Qsca_ch_norm_) Q = 2.0*Q/x2;
  923. Qpr_ = Qext_ - 4.0*(Qpr_)/x2; // Equation (29)
  924. for (int i = 0; i < nmax_ - 1; ++i) Qpr_ch_[i] = Qext_ch_[i] - 4.0*Qpr_ch_[i]/x2;
  925. Qabs_ = Qext_ - Qsca_; // Equation (30)
  926. for (int i = 0; i < nmax_ - 1; ++i) {
  927. Qabs_ch_[i] = Qext_ch_[i] - Qsca_ch_[i];
  928. Qabs_ch_norm_[i] = Qext_ch_norm_[i] - Qsca_ch_norm_[i];
  929. }
  930. albedo_ = Qsca_/Qext_; // Equation (31)
  931. asymmetry_factor_ = (Qext_ - Qpr_)/Qsca_; // Equation (32)
  932. Qbk_ = (Qbktmp.real()*Qbktmp.real() + Qbktmp.imag()*Qbktmp.imag())/x2; // Equation (33)
  933. isMieCalculated_ = true;
  934. }
  935. // ********************************************************************** //
  936. // ********************************************************************** //
  937. // ********************************************************************** //
  938. void MultiLayerMie::IntScattCoeffs() {
  939. if (!isExtCoeffsCalc_)
  940. throw std::invalid_argument("(IntScattCoeffs) You should calculate external coefficients first!");
  941. isIntCoeffsCalc_ = false;
  942. std::complex<double> c_one(1.0, 0.0);
  943. std::complex<double> c_zero(0.0, 0.0);
  944. const int L = refr_index_.size();
  945. // we need to fill
  946. // std::vector< std::vector<std::complex<double> > > aln_, bln_, cnl_, dln_;
  947. // for n = [0..nmax_) and for l=[L..0)
  948. // TODO: to decrease cache miss outer loop is with n and inner with reversed l
  949. // at the moment outer is forward l and inner in n
  950. aln_.resize(L + 1);
  951. bln_.resize(L + 1);
  952. cln_.resize(L + 1);
  953. dln_.resize(L + 1);
  954. for (auto& element:aln_) element.resize(nmax_);
  955. for (auto& element:bln_) element.resize(nmax_);
  956. for (auto& element:cln_) element.resize(nmax_);
  957. for (auto& element:dln_) element.resize(nmax_);
  958. // Yang, paragraph under eq. A3
  959. // a^(L + 1)_n = a_n, d^(L + 1) = 1 ...
  960. for (int i = 0; i < nmax_; ++i) {
  961. aln_[L][i] = an_[i];
  962. bln_[L][i] = bn_[i];
  963. cln_[L][i] = c_one;
  964. dln_[L][i] = c_one;
  965. }
  966. std::vector<std::complex<double> > z(L), z1(L);
  967. for (int i = 0; i < L - 1; ++i) {
  968. z[i] = size_param_[i]*refr_index_[i];
  969. z1[i] = size_param_[i]*refr_index_[i + 1];
  970. }
  971. z[L - 1] = size_param_[L - 1]*refr_index_[L - 1];
  972. z1[L - 1] = size_param_[L - 1];
  973. std::vector< std::vector<std::complex<double> > > D1z(L), D1z1(L), D3z(L), D3z1(L);
  974. std::vector< std::vector<std::complex<double> > > Psiz(L), Psiz1(L), Zetaz(L), Zetaz1(L);
  975. for (int l = 0; l < L; ++l) {
  976. D1z[l].resize(nmax_ + 1);
  977. D1z1[l].resize(nmax_ + 1);
  978. D3z[l].resize(nmax_ + 1);
  979. D3z1[l].resize(nmax_ + 1);
  980. Psiz[l].resize(nmax_ + 1);
  981. Psiz1[l].resize(nmax_ + 1);
  982. Zetaz[l].resize(nmax_ + 1);
  983. Zetaz1[l].resize(nmax_ + 1);
  984. }
  985. for (int l = 0; l < L; ++l) {
  986. calcD1D3(z[l], D1z[l], D3z[l]);
  987. calcD1D3(z1[l], D1z1[l], D3z1[l]);
  988. calcPsiZeta(z[l], Psiz[l], Zetaz[l]);
  989. calcPsiZeta(z1[l], Psiz1[l], Zetaz1[l]);
  990. }
  991. auto& m = refr_index_;
  992. std::vector< std::complex<double> > m1(L);
  993. for (int l = 0; l < L - 1; ++l) m1[l] = m[l + 1];
  994. m1[L - 1] = std::complex<double> (1.0, 0.0);
  995. for (int l = L - 1; l >= 0; l--) {
  996. for (int n = nmax_ - 2; n >= 0; n--) {
  997. auto denomZeta = m1[l]*Zetaz[l][n + 1]*(D1z[l][n + 1] - D3z[l][n + 1]);
  998. auto denomPsi = m1[l]*Psiz[l][n + 1]*(D1z[l][n + 1] - D3z[l][n + 1]);
  999. auto T1 = aln_[l + 1][n]*Zetaz1[l][n + 1] - dln_[l + 1][n]*Psiz1[l][n + 1];
  1000. auto T2 = bln_[l + 1][n]*Zetaz1[l][n + 1] - cln_[l + 1][n]*Psiz1[l][n + 1];
  1001. auto T3 = -D1z1[l][n + 1]*dln_[l + 1][n]*Psiz1[l][n + 1] + D3z1[l][n + 1]*aln_[l + 1][n]*Zetaz1[l][n + 1];
  1002. auto T4 = -D1z1[l][n + 1]*cln_[l + 1][n]*Psiz1[l][n + 1] + D3z1[l][n + 1]*bln_[l + 1][n]*Zetaz1[l][n + 1];
  1003. // anl
  1004. aln_[l][n] = (D1z[l][n + 1]*m1[l]*T1 - m[l]*T3)/denomZeta;
  1005. // bnl
  1006. bln_[l][n] = (D1z[l][n + 1]*m[l]*T2 - m1[l]*T4)/denomZeta;
  1007. // cnl
  1008. cln_[l][n] = (D3z[l][n + 1]*m[l]*T2 - m1[l]*T4)/denomPsi;
  1009. // dnl
  1010. dln_[l][n] = (D3z[l][n + 1]*m1[l]*T1 - m[l]*T3)/denomPsi;
  1011. } // end of all n
  1012. } // end of all l
  1013. // Check the result and change aln_[0][n] and aln_[0][n] for exact zero
  1014. for (int n = 0; n < nmax_; ++n) {
  1015. if (std::abs(aln_[0][n]) < 1e-10) aln_[0][n] = 0.0;
  1016. else throw std::invalid_argument("Unstable calculation of aln_[0][n]!");
  1017. if (std::abs(bln_[0][n]) < 1e-10) bln_[0][n] = 0.0;
  1018. else throw std::invalid_argument("Unstable calculation of aln_[0][n]!");
  1019. }
  1020. isIntCoeffsCalc_ = true;
  1021. }
  1022. // ********************************************************************** //
  1023. // ********************************************************************** //
  1024. // ********************************************************************** //
  1025. // external scattering field = incident + scattered //
  1026. // BH p.92 (4.37), 94 (4.45), 95 (4.50) //
  1027. // assume: medium is non-absorbing; refim = 0; Uabs = 0 //
  1028. // ********************************************************************** //
  1029. void MultiLayerMie::fieldExt(const double Rho, const double Phi, const double Theta,
  1030. std::vector<std::complex<double> >& E, std::vector<std::complex<double> >& H) {
  1031. std::complex<double> c_zero(0.0, 0.0), c_i(0.0, 1.0), c_one(1.0, 0.0);
  1032. std::vector<std::complex<double> > ipow = {c_one, c_i, -c_one, -c_i}; // Vector containing precomputed integer powers of i to avoid computation
  1033. std::vector<std::complex<double> > M3o1n(3), M3e1n(3), N3o1n(3), N3e1n(3);
  1034. std::vector<std::complex<double> > Ei(3, c_zero), Hi(3, c_zero), Es(3, c_zero), Hs(3, c_zero);
  1035. std::vector<std::complex<double> > jn(nmax_ + 1), jnp(nmax_ + 1), h1n(nmax_ + 1), h1np(nmax_ + 1);
  1036. std::vector<double> Pi(nmax_), Tau(nmax_);
  1037. // Calculate spherical Bessel and Hankel functions
  1038. sbesjh(Rho, jn, jnp, h1n, h1np);
  1039. // Calculate angular functions Pi and Tau
  1040. calcPiTau(std::cos(Theta), Pi, Tau);
  1041. for (int n = 0; n < nmax_; n++) {
  1042. int n1 = n + 1;
  1043. double rn = static_cast<double>(n1);
  1044. // using BH 4.12 and 4.50
  1045. calcSpherHarm(Rho, Phi, Theta, h1n[n1], h1np[n1], Pi[n], Tau[n], rn, M3o1n, M3e1n, N3o1n, N3e1n);
  1046. // scattered field: BH p.94 (4.45)
  1047. std::complex<double> En = ipow[n1 % 4]*(rn + rn + 1.0)/(rn*rn + rn);
  1048. for (int i = 0; i < 3; i++) {
  1049. Es[i] = Es[i] + En*(c_i*an_[n]*N3e1n[i] - bn_[n]*M3o1n[i]);
  1050. Hs[i] = Hs[i] + En*(c_i*bn_[n]*N3o1n[i] + an_[n]*M3e1n[i]);
  1051. }
  1052. }
  1053. // incident E field: BH p.89 (4.21); cf. p.92 (4.37), p.93 (4.38)
  1054. // basis unit vectors = er, etheta, ephi
  1055. std::complex<double> eifac = std::exp(std::complex<double>(0.0, Rho*std::cos(Theta)));
  1056. {
  1057. using std::sin;
  1058. using std::cos;
  1059. Ei[0] = eifac*sin(Theta)*cos(Phi);
  1060. Ei[1] = eifac*cos(Theta)*cos(Phi);
  1061. Ei[2] = -eifac*sin(Phi);
  1062. }
  1063. // magnetic field
  1064. double hffact = 1.0/(cc_*mu_);
  1065. for (int i = 0; i < 3; i++) {
  1066. Hs[i] = hffact*Hs[i];
  1067. }
  1068. // incident H field: BH p.26 (2.43), p.89 (4.21)
  1069. std::complex<double> hffacta = hffact;
  1070. std::complex<double> hifac = eifac*hffacta;
  1071. {
  1072. using std::sin;
  1073. using std::cos;
  1074. Hi[0] = hifac*sin(Theta)*sin(Phi);
  1075. Hi[1] = hifac*cos(Theta)*sin(Phi);
  1076. Hi[2] = hifac*cos(Phi);
  1077. }
  1078. for (int i = 0; i < 3; i++) {
  1079. // electric field E [V m - 1] = EF*E0
  1080. E[i] = Ei[i] + Es[i];
  1081. H[i] = Hi[i] + Hs[i];
  1082. }
  1083. } // end of MultiLayerMie::fieldExt(...)
  1084. // ********************************************************************** //
  1085. // ********************************************************************** //
  1086. // ********************************************************************** //
  1087. void MultiLayerMie::fieldInt(const double Rho, const double Phi, const double Theta,
  1088. std::vector<std::complex<double> >& E, std::vector<std::complex<double> >& H) {
  1089. std::complex<double> c_zero(0.0, 0.0), c_i(0.0, 1.0), c_one(1.0, 0.0);
  1090. std::vector<std::complex<double> > ipow = {c_one, c_i, -c_one, -c_i}; // Vector containing precomputed integer powers of i to avoid computation
  1091. std::vector<std::complex<double> > M3o1n(3), M3e1n(3), N3o1n(3), N3e1n(3);
  1092. std::vector<std::complex<double> > M1o1n(3), M1e1n(3), N1o1n(3), N1e1n(3);
  1093. std::vector<std::complex<double> > El(3, c_zero), Hl(3, c_zero);
  1094. std::vector<std::complex<double> > jn(nmax_ + 1), jnp(nmax_ + 1), h1n(nmax_ + 1), h1np(nmax_ + 1);
  1095. std::vector<double> Pi(nmax_), Tau(nmax_);
  1096. int l = 0; // Layer number
  1097. std::complex<double> ml;
  1098. if (Rho > size_param_.back()) {
  1099. l = size_param_.size();
  1100. ml = c_one;
  1101. } else {
  1102. for (int i = size_param_.size() - 1; i >= 0 ; i--) {
  1103. if (Rho <= size_param_[i]) {
  1104. l = i;
  1105. }
  1106. }
  1107. ml = refr_index_[l];
  1108. }
  1109. // Calculate spherical Bessel and Hankel functions
  1110. sbesjh(Rho*ml, jn, jnp, h1n, h1np);
  1111. // Calculate angular functions Pi and Tau
  1112. calcPiTau(std::cos(Theta), Pi, Tau);
  1113. for (int n = nmax_ - 1; n >= 0; n--) {
  1114. int n1 = n + 1;
  1115. double rn = static_cast<double>(n1);
  1116. // using BH 4.12 and 4.50
  1117. calcSpherHarm(Rho, Phi, Theta, jn[n1], jnp[n1], Pi[n], Tau[n], rn, M1o1n, M1e1n, N1o1n, N1e1n);
  1118. calcSpherHarm(Rho, Phi, Theta, h1n[n1], h1np[n1], Pi[n], Tau[n], rn, M3o1n, M3e1n, N3o1n, N3e1n);
  1119. // Total field in the lth layer: eqs. (1) and (2) in Yang, Appl. Opt., 42 (2003) 1710-1720
  1120. std::complex<double> En = ipow[n1 % 4]*(rn + rn + 1.0)/(rn*rn + rn);
  1121. for (int i = 0; i < 3; i++) {
  1122. El[i] = El[i] + En*(cln_[l][n]*M1o1n[i] - c_i*dln_[l][n]*N1e1n[i]
  1123. + c_i*aln_[l][n]*N3e1n[i] - bln_[l][n]*M3o1n[i]);
  1124. Hl[i] = Hl[i] + En*(-dln_[l][n]*M1e1n[i] - c_i*cln_[l][n]*N1o1n[i]
  1125. + c_i*bln_[l][n]*N3o1n[i] + aln_[l][n]*M3e1n[i]);
  1126. }
  1127. } // end of for all n
  1128. // magnetic field
  1129. double hffact = 1.0/(cc_*mu_);
  1130. for (int i = 0; i < 3; i++) {
  1131. Hl[i] = hffact*Hl[i];
  1132. }
  1133. for (int i = 0; i < 3; i++) {
  1134. // electric field E [V m - 1] = EF*E0
  1135. E[i] = El[i];
  1136. H[i] = Hl[i];
  1137. }
  1138. } // end of MultiLayerMie::fieldInt(...)
  1139. //**********************************************************************************//
  1140. // This function calculates complex electric and magnetic field in the surroundings //
  1141. // and inside (TODO) the particle. //
  1142. // //
  1143. // Input parameters: //
  1144. // L: Number of layers //
  1145. // pl: Index of PEC layer. If there is none just send 0 (zero) //
  1146. // x: Array containing the size parameters of the layers [0..L-1] //
  1147. // m: Array containing the relative refractive indexes of the layers [0..L-1] //
  1148. // nmax: Maximum number of multipolar expansion terms to be used for the //
  1149. // calculations. Only use it if you know what you are doing, otherwise //
  1150. // set this parameter to 0 (zero) and the function will calculate it. //
  1151. // ncoord: Number of coordinate points //
  1152. // Coords: Array containing all coordinates where the complex electric and //
  1153. // magnetic fields will be calculated //
  1154. // //
  1155. // Output parameters: //
  1156. // E, H: Complex electric and magnetic field at the provided coordinates //
  1157. // //
  1158. // Return value: //
  1159. // Number of multipolar expansion terms used for the calculations //
  1160. //**********************************************************************************//
  1161. void MultiLayerMie::RunFieldCalculation() {
  1162. // Calculate external scattering coefficients an_ and bn_
  1163. ExtScattCoeffs();
  1164. // Calculate internal scattering coefficients aln_ and bln_
  1165. IntScattCoeffs();
  1166. long total_points = coords_[0].size();
  1167. E_.resize(total_points);
  1168. H_.resize(total_points);
  1169. for (auto& f : E_) f.resize(3);
  1170. for (auto& f : H_) f.resize(3);
  1171. for (int point = 0; point < total_points; point++) {
  1172. const double& Xp = coords_[0][point];
  1173. const double& Yp = coords_[1][point];
  1174. const double& Zp = coords_[2][point];
  1175. // Convert to spherical coordinates
  1176. double Rho = std::sqrt(pow2(Xp) + pow2(Yp) + pow2(Zp));
  1177. // If Rho=0 then Theta is undefined. Just set it to zero to avoid problems
  1178. double Theta = (Rho > 0.0) ? std::acos(Zp/Rho) : 0.0;
  1179. // If Xp=Yp=0 then Phi is undefined. Just set it to zero to avoid problems
  1180. double Phi = (Xp != 0.0 || Yp != 0.0) ? std::acos(Xp/std::sqrt(pow2(Xp) + pow2(Yp))) : 0.0;
  1181. // Avoid convergence problems due to Rho too small
  1182. if (Rho < 1e-5) Rho = 1e-5;
  1183. //*******************************************************//
  1184. // external scattering field = incident + scattered //
  1185. // BH p.92 (4.37), 94 (4.45), 95 (4.50) //
  1186. // assume: medium is non-absorbing; refim = 0; Uabs = 0 //
  1187. //*******************************************************//
  1188. // This array contains the fields in spherical coordinates
  1189. std::vector<std::complex<double> > Es(3), Hs(3);
  1190. // Firstly the easiest case: the field outside the particle
  1191. if (Rho >= GetSizeParameter()) {
  1192. fieldInt(Rho, Phi, Theta, Es, Hs);
  1193. // fieldExt(Rho, Phi, Theta, Es, Hs);
  1194. //printf("\nFin E ext: %g,%g,%g Rho=%g\n", std::abs(Es[0]), std::abs(Es[1]),std::abs(Es[2]), Rho);
  1195. } else {
  1196. fieldInt(Rho, Phi, Theta, Es, Hs);
  1197. // printf("\nFin E int: %g,%g,%g Rho=%g\n", std::abs(Es[0]), std::abs(Es[1]),std::abs(Es[2]), Rho);
  1198. }
  1199. { //Now, convert the fields back to cartesian coordinates
  1200. using std::sin;
  1201. using std::cos;
  1202. E_[point][0] = sin(Theta)*cos(Phi)*Es[0] + cos(Theta)*cos(Phi)*Es[1] - sin(Phi)*Es[2];
  1203. E_[point][1] = sin(Theta)*sin(Phi)*Es[0] + cos(Theta)*sin(Phi)*Es[1] + cos(Phi)*Es[2];
  1204. E_[point][2] = cos(Theta)*Es[0] - sin(Theta)*Es[1];
  1205. H_[point][0] = sin(Theta)*cos(Phi)*Hs[0] + cos(Theta)*cos(Phi)*Hs[1] - sin(Phi)*Hs[2];
  1206. H_[point][1] = sin(Theta)*sin(Phi)*Hs[0] + cos(Theta)*sin(Phi)*Hs[1] + cos(Phi)*Hs[2];
  1207. H_[point][2] = cos(Theta)*Hs[0] - sin(Theta)*Hs[1];
  1208. }
  1209. //printf("Cart E: %g,%g,%g Rho=%g\n", std::abs(Ex), std::abs(Ey),std::abs(Ez), Rho);
  1210. } // end of for all field coordinates
  1211. } // end of MultiLayerMie::RunFieldCalculation()
  1212. } // end of namespace nmie