nmie.cc 85 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643
  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 "bessel.h"
  40. #include "nmie.h"
  41. #include <array>
  42. #include <algorithm>
  43. #include <cstdio>
  44. #include <cstdlib>
  45. #include <stdexcept>
  46. #include <vector>
  47. namespace nmie {
  48. //helpers
  49. template<class T> inline T pow2(const T value) {return value*value;}
  50. int round(double x) {
  51. return x >= 0 ? (int)(x + 0.5):(int)(x - 0.5);
  52. }
  53. //**********************************************************************************//
  54. // This function emulates a C call to calculate the actual scattering parameters //
  55. // and amplitudes. //
  56. // //
  57. // Input parameters: //
  58. // L: Number of layers //
  59. // pl: Index of PEC layer. If there is none just send -1 //
  60. // x: Array containing the size parameters of the layers [0..L-1] //
  61. // m: Array containing the relative refractive indexes of the layers [0..L-1] //
  62. // nTheta: Number of scattering angles //
  63. // Theta: Array containing all the scattering angles where the scattering //
  64. // amplitudes will be calculated //
  65. // nmax: Maximum number of multipolar expansion terms to be used for the //
  66. // calculations. Only use it if you know what you are doing, otherwise //
  67. // set this parameter to -1 and the function will calculate it //
  68. // //
  69. // Output parameters: //
  70. // Qext: Efficiency factor for extinction //
  71. // Qsca: Efficiency factor for scattering //
  72. // Qabs: Efficiency factor for absorption (Qabs = Qext - Qsca) //
  73. // Qbk: Efficiency factor for backscattering //
  74. // Qpr: Efficiency factor for the radiation pressure //
  75. // g: Asymmetry factor (g = (Qext-Qpr)/Qsca) //
  76. // Albedo: Single scattering albedo (Albedo = Qsca/Qext) //
  77. // S1, S2: Complex scattering amplitudes //
  78. // //
  79. // Return value: //
  80. // Number of multipolar expansion terms used for the calculations //
  81. //**********************************************************************************//
  82. int nMie(const unsigned int L, const int pl, std::vector<double>& x, std::vector<std::complex<double> >& m, const unsigned 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) {
  83. if (x.size() != L || m.size() != L)
  84. throw std::invalid_argument("Declared number of layers do not fit x and m!");
  85. if (Theta.size() != nTheta)
  86. throw std::invalid_argument("Declared number of sample for Theta is not correct!");
  87. try {
  88. MultiLayerMie multi_layer_mie;
  89. multi_layer_mie.SetLayersSize(x);
  90. multi_layer_mie.SetLayersIndex(m);
  91. multi_layer_mie.SetAngles(Theta);
  92. multi_layer_mie.SetPECLayer(pl);
  93. multi_layer_mie.SetMaxTerms(nmax);
  94. multi_layer_mie.RunMieCalculation();
  95. *Qext = multi_layer_mie.GetQext();
  96. *Qsca = multi_layer_mie.GetQsca();
  97. *Qabs = multi_layer_mie.GetQabs();
  98. *Qbk = multi_layer_mie.GetQbk();
  99. *Qpr = multi_layer_mie.GetQpr();
  100. *g = multi_layer_mie.GetAsymmetryFactor();
  101. *Albedo = multi_layer_mie.GetAlbedo();
  102. S1 = multi_layer_mie.GetS1();
  103. S2 = multi_layer_mie.GetS2();
  104. } catch(const std::invalid_argument& ia) {
  105. // Will catch if multi_layer_mie fails or other errors.
  106. std::cerr << "Invalid argument: " << ia.what() << std::endl;
  107. throw std::invalid_argument(ia);
  108. return -1;
  109. }
  110. return 0;
  111. }
  112. //**********************************************************************************//
  113. // This function is just a wrapper to call the full 'nMie' function with fewer //
  114. // parameters, it is here mainly for compatibility with older versions of the //
  115. // program. Also, you can use it if you neither have a PEC layer nor want to define //
  116. // any limit for the maximum number of terms. //
  117. // //
  118. // Input parameters: //
  119. // L: Number of layers //
  120. // x: Array containing the size parameters of the layers [0..L-1] //
  121. // m: Array containing the relative refractive indexes of the layers [0..L-1] //
  122. // nTheta: Number of scattering angles //
  123. // Theta: Array containing all the scattering angles where the scattering //
  124. // amplitudes will be calculated //
  125. // //
  126. // Output parameters: //
  127. // Qext: Efficiency factor for extinction //
  128. // Qsca: Efficiency factor for scattering //
  129. // Qabs: Efficiency factor for absorption (Qabs = Qext - Qsca) //
  130. // Qbk: Efficiency factor for backscattering //
  131. // Qpr: Efficiency factor for the radiation pressure //
  132. // g: Asymmetry factor (g = (Qext-Qpr)/Qsca) //
  133. // Albedo: Single scattering albedo (Albedo = Qsca/Qext) //
  134. // S1, S2: Complex scattering amplitudes //
  135. // //
  136. // Return value: //
  137. // Number of multipolar expansion terms used for the calculations //
  138. //**********************************************************************************//
  139. int nMie(const unsigned int L, std::vector<double>& x, std::vector<std::complex<double> >& m, const unsigned 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) {
  140. return nmie::nMie(L, -1, x, m, nTheta, Theta, -1, Qext, Qsca, Qabs, Qbk, Qpr, g, Albedo, S1, S2);
  141. }
  142. //**********************************************************************************//
  143. // This function is just a wrapper to call the full 'nMie' function with fewer //
  144. // parameters, it is useful if you want to include a PEC layer but not a limit //
  145. // for the maximum number of terms. //
  146. // //
  147. // Input parameters: //
  148. // L: Number of layers //
  149. // pl: Index of PEC layer. If there is none just send -1 //
  150. // x: Array containing the size parameters of the layers [0..L-1] //
  151. // m: Array containing the relative refractive indexes of the layers [0..L-1] //
  152. // nTheta: Number of scattering angles //
  153. // Theta: Array containing all the scattering angles where the scattering //
  154. // amplitudes will be calculated //
  155. // //
  156. // Output parameters: //
  157. // Qext: Efficiency factor for extinction //
  158. // Qsca: Efficiency factor for scattering //
  159. // Qabs: Efficiency factor for absorption (Qabs = Qext - Qsca) //
  160. // Qbk: Efficiency factor for backscattering //
  161. // Qpr: Efficiency factor for the radiation pressure //
  162. // g: Asymmetry factor (g = (Qext-Qpr)/Qsca) //
  163. // Albedo: Single scattering albedo (Albedo = Qsca/Qext) //
  164. // S1, S2: Complex scattering amplitudes //
  165. // //
  166. // Return value: //
  167. // Number of multipolar expansion terms used for the calculations //
  168. //**********************************************************************************//
  169. int nMie(const unsigned int L, const int pl, std::vector<double>& x, std::vector<std::complex<double> >& m, const unsigned 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) {
  170. return nmie::nMie(L, pl, x, m, nTheta, Theta, -1, Qext, Qsca, Qabs, Qbk, Qpr, g, Albedo, S1, S2);
  171. }
  172. //**********************************************************************************//
  173. // This function is just a wrapper to call the full 'nMie' function with fewer //
  174. // parameters, it is useful if you want to include a limit for the maximum number //
  175. // of terms but not a PEC layer. //
  176. // //
  177. // Input parameters: //
  178. // L: Number of layers //
  179. // x: Array containing the size parameters of the layers [0..L-1] //
  180. // m: Array containing the relative refractive indexes of the layers [0..L-1] //
  181. // nTheta: Number of scattering angles //
  182. // Theta: Array containing all the scattering angles where the scattering //
  183. // amplitudes will be calculated //
  184. // nmax: Maximum number of multipolar expansion terms to be used for the //
  185. // calculations. Only use it if you know what you are doing, otherwise //
  186. // set this parameter to -1 and the function will calculate it //
  187. // //
  188. // Output parameters: //
  189. // Qext: Efficiency factor for extinction //
  190. // Qsca: Efficiency factor for scattering //
  191. // Qabs: Efficiency factor for absorption (Qabs = Qext - Qsca) //
  192. // Qbk: Efficiency factor for backscattering //
  193. // Qpr: Efficiency factor for the radiation pressure //
  194. // g: Asymmetry factor (g = (Qext-Qpr)/Qsca) //
  195. // Albedo: Single scattering albedo (Albedo = Qsca/Qext) //
  196. // S1, S2: Complex scattering amplitudes //
  197. // //
  198. // Return value: //
  199. // Number of multipolar expansion terms used for the calculations //
  200. //**********************************************************************************//
  201. int nMie(const unsigned int L, std::vector<double>& x, std::vector<std::complex<double> >& m, const unsigned 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) {
  202. return nmie::nMie(L, -1, x, m, nTheta, Theta, nmax, Qext, Qsca, Qabs, Qbk, Qpr, g, Albedo, S1, S2);
  203. }
  204. //**********************************************************************************//
  205. // This function emulates a C call to calculate complex electric and magnetic field //
  206. // in the surroundings and inside (TODO) the particle. //
  207. // //
  208. // Input parameters: //
  209. // L: Number of layers //
  210. // pl: Index of PEC layer. If there is none just send 0 (zero) //
  211. // x: Array containing the size parameters of the layers [0..L-1] //
  212. // m: Array containing the relative refractive indexes of the layers [0..L-1] //
  213. // nmax: Maximum number of multipolar expansion terms to be used for the //
  214. // calculations. Only use it if you know what you are doing, otherwise //
  215. // set this parameter to 0 (zero) and the function will calculate it. //
  216. // ncoord: Number of coordinate points //
  217. // Coords: Array containing all coordinates where the complex electric and //
  218. // magnetic fields will be calculated //
  219. // //
  220. // Output parameters: //
  221. // E, H: Complex electric and magnetic field at the provided coordinates //
  222. // //
  223. // Return value: //
  224. // Number of multipolar expansion terms used for the calculations //
  225. //**********************************************************************************//
  226. int nField(const unsigned int L, const int pl, const std::vector<double>& x, const std::vector<std::complex<double> >& m, const int nmax, const unsigned 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) {
  227. if (x.size() != L || m.size() != L)
  228. throw std::invalid_argument("Declared number of layers do not fit x and m!");
  229. if (Xp_vec.size() != ncoord || Yp_vec.size() != ncoord || Zp_vec.size() != ncoord
  230. || E.size() != ncoord || H.size() != ncoord)
  231. throw std::invalid_argument("Declared number of coords do not fit Xp, Yp, Zp, E, or H!");
  232. for (auto f:E)
  233. if (f.size() != 3)
  234. throw std::invalid_argument("Field E is not 3D!");
  235. for (auto f:H)
  236. if (f.size() != 3)
  237. throw std::invalid_argument("Field H is not 3D!");
  238. try {
  239. MultiLayerMie multi_layer_mie;
  240. //multi_layer_mie.SetPECLayer(pl); // TODO add PEC layer to field plotting
  241. multi_layer_mie.SetLayersSize(x);
  242. multi_layer_mie.SetLayersIndex(m);
  243. multi_layer_mie.SetFieldCoords({Xp_vec, Yp_vec, Zp_vec});
  244. multi_layer_mie.RunFieldCalculation();
  245. E = multi_layer_mie.GetFieldE();
  246. H = multi_layer_mie.GetFieldH();
  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. isExpCoeffsCalc_ = false;
  332. isScaCoeffsCalc_ = 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. isExpCoeffsCalc_ = false;
  341. isScaCoeffsCalc_ = 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. isExpCoeffsCalc_ = false;
  360. isScaCoeffsCalc_ = false;
  361. isMieCalculated_ = false;
  362. refractive_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. isExpCoeffsCalc_ = false;
  379. isScaCoeffsCalc_ = false;
  380. isMieCalculated_ = false;
  381. if (layer_position < 0 && layer_position != -1)
  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. isExpCoeffsCalc_ = false;
  390. isScaCoeffsCalc_ = 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. isExpCoeffsCalc_ = false;
  408. isScaCoeffsCalc_ = false;
  409. isMieCalculated_ = false;
  410. size_param_.clear();
  411. refractive_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(unsigned int first_layer) {
  437. int ri, riM1;
  438. const std::vector<double>& x = size_param_;
  439. const std::vector<std::complex<double> >& m = refractive_index_;
  440. calcNstop(); // Set initial nmax_ value
  441. for (unsigned int i = first_layer; i < x.size(); i++) {
  442. if (static_cast<int>(i) > PEC_layer_position_) // static_cast used to avoid warning
  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) && (static_cast<int>(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. // Calculate an and bn for bulk sphere size x and index m //
  478. // equation (4.56) and (4.57) BH //
  479. // ********************************************************************** //
  480. void MultiLayerMie::calc_an_bn_bulk(std::vector<std::complex<double> >& an,
  481. std::vector<std::complex<double> >& bn,
  482. double x, std::complex<double> m) {
  483. //printf("==========\n m = %g,%g, x= %g\n", std::real(m), std::imag(m), x);
  484. std::vector<std::complex<double> > PsiX(nmax_ + 1), ZetaX(nmax_ + 1);
  485. std::vector<std::complex<double> > PsiMX(nmax_ + 1), ZetaMX(nmax_ + 1);
  486. // First, calculate the Riccati-Bessel functions
  487. calcPsiZeta(x, PsiX, ZetaX);
  488. calcPsiZeta(m*x, PsiMX, ZetaMX);
  489. std::vector<std::complex<double> > D1X(nmax_ + 1), D3X(nmax_ + 1);
  490. std::vector<std::complex<double> > D1MX(nmax_ + 1), D3MX(nmax_ + 1);
  491. // Calculate the logarithmic derivatives
  492. calcD1D3(x, D1X, D3X);
  493. calcD1D3(m*x, D1MX, D3MX);
  494. std::vector<std::complex<double> > dPsiX(nmax_ + 1), dZetaX(nmax_ + 1);
  495. std::vector<std::complex<double> > dPsiMX(nmax_ + 1);
  496. for (int i = 0; i < nmax_ + 1; ++i) {
  497. dPsiX[i] = D1X[i]*PsiX[i];
  498. dPsiMX[i] = D1MX[i]*PsiMX[i];
  499. //dZetaX[i] = D3X[i]*ZetaX[i];
  500. }
  501. bessel::calcZeta(nmax_, x, ZetaX, dZetaX);
  502. an.resize(nmax_);
  503. bn.resize(nmax_);
  504. for (int i = 0; i < nmax_; i++) {
  505. int n = i+1;
  506. std::complex<double> Num = m*PsiMX[n]*dPsiX[n] - PsiX[n]*dPsiMX[n];
  507. std::complex<double> Denom = m*PsiMX[n]*dZetaX[n] - ZetaX[n]*dPsiMX[n];
  508. an[i] = Num/Denom;
  509. Num = PsiMX[n]*dPsiX[n] - m*PsiX[n]*dPsiMX[n];
  510. Denom = PsiMX[n]*dZetaX[n] - m*ZetaX[n]*dPsiMX[n];
  511. bn[i] = Num/Denom;
  512. }
  513. // printf("dPsiX\n");
  514. // for (auto a: dPsiX) printf("%11.4er%+10.5ei ",std::real(a), std::imag(a));
  515. // printf("\ndPsiMX\n");
  516. // for (auto a: dPsiMX) printf("%11.4er%+10.5ei ",std::real(a), std::imag(a));
  517. // printf("\nPsiX\n");
  518. // for (auto a: PsiX) printf("%11.4er%+10.5ei ",std::real(a), std::imag(a));
  519. // printf("\nPsiMX\n");
  520. // for (auto a: PsiMX) printf("%11.4er%+10.5ei ",std::real(a), std::imag(a));
  521. // printf("\nZetaX\n");
  522. // for (auto a: ZetaX) printf("%11.4er%+10.5ei ",std::real(a), std::imag(a));
  523. // printf("\ndZetaX\n");
  524. // for (auto a: dZetaX) printf("%11.4er%+10.5ei ",std::real(a), std::imag(a));
  525. // printf("\nsize param: %g\n", x);
  526. }
  527. // ********************************************************************** //
  528. // Calculates S1 - equation (25a) //
  529. // ********************************************************************** //
  530. std::complex<double> MultiLayerMie::calc_S1(int n, std::complex<double> an, std::complex<double> bn,
  531. double Pi, double Tau) {
  532. return double(n + n + 1)*(Pi*an + Tau*bn)/double(n*n + n);
  533. }
  534. // ********************************************************************** //
  535. // Calculates S2 - equation (25b) (it's the same as (25a), just switches //
  536. // Pi and Tau) //
  537. // ********************************************************************** //
  538. std::complex<double> MultiLayerMie::calc_S2(int n, std::complex<double> an, std::complex<double> bn,
  539. double Pi, double Tau) {
  540. return calc_S1(n, an, bn, Tau, Pi);
  541. }
  542. //**********************************************************************************//
  543. // This function calculates the logarithmic derivatives of the Riccati-Bessel //
  544. // functions (D1 and D3) for a complex argument (z). //
  545. // Equations (16a), (16b) and (18a) - (18d) //
  546. // //
  547. // Input parameters: //
  548. // z: Complex argument to evaluate D1 and D3 //
  549. // nmax_: Maximum number of terms to calculate D1 and D3 //
  550. // //
  551. // Output parameters: //
  552. // D1, D3: Logarithmic derivatives of the Riccati-Bessel functions //
  553. //**********************************************************************************//
  554. void MultiLayerMie::calcD1D3(const std::complex<double> z,
  555. std::vector<std::complex<double> >& D1,
  556. std::vector<std::complex<double> >& D3) {
  557. // Downward recurrence for D1 - equations (16a) and (16b)
  558. D1[nmax_] = std::complex<double>(0.0, 0.0);
  559. const std::complex<double> zinv = std::complex<double>(1.0, 0.0)/z;
  560. for (int n = nmax_; n > 0; n--) {
  561. D1[n - 1] = static_cast<double>(n)*zinv - 1.0/(D1[n] + static_cast<double>(n)*zinv);
  562. }
  563. if (std::abs(D1[0]) > 100000.0)
  564. throw std::invalid_argument("Unstable D1! Please, try to change input parameters!\n");
  565. // Upward recurrence for PsiZeta and D3 - equations (18a) - (18d)
  566. PsiZeta_[0] = 0.5*(1.0 - std::complex<double>(std::cos(2.0*z.real()), std::sin(2.0*z.real()))
  567. *std::exp(-2.0*z.imag()));
  568. D3[0] = std::complex<double>(0.0, 1.0);
  569. for (int n = 1; n <= nmax_; n++) {
  570. PsiZeta_[n] = PsiZeta_[n - 1]*(static_cast<double>(n)*zinv - D1[n - 1])
  571. *(static_cast<double>(n)*zinv - D3[n - 1]);
  572. D3[n] = D1[n] + std::complex<double>(0.0, 1.0)/PsiZeta_[n];
  573. }
  574. }
  575. //**********************************************************************************//
  576. // This function calculates the Riccati-Bessel functions (Psi and Zeta) for a //
  577. // complex argument (z). //
  578. // Equations (20a) - (21b) //
  579. // //
  580. // Input parameters: //
  581. // z: Complex argument to evaluate Psi and Zeta //
  582. // nmax: Maximum number of terms to calculate Psi and Zeta //
  583. // //
  584. // Output parameters: //
  585. // Psi, Zeta: Riccati-Bessel functions //
  586. //**********************************************************************************//
  587. void MultiLayerMie::calcPsiZeta(std::complex<double> z,
  588. std::vector<std::complex<double> >& Psi,
  589. std::vector<std::complex<double> >& Zeta) {
  590. std::complex<double> c_i(0.0, 1.0);
  591. std::vector<std::complex<double> > D1(nmax_ + 1), D3(nmax_ + 1);
  592. // First, calculate the logarithmic derivatives
  593. calcD1D3(z, D1, D3);
  594. // Now, use the upward recurrence to calculate Psi and Zeta - equations (20a) - (21b)
  595. Psi[0] = std::sin(z);
  596. Zeta[0] = std::sin(z) - c_i*std::cos(z);
  597. for (int n = 1; n <= nmax_; n++) {
  598. Psi[n] = Psi[n - 1]*(static_cast<double>(n)/z - D1[n - 1]);
  599. Zeta[n] = Zeta[n - 1]*(static_cast<double>(n)/z - D3[n - 1]);
  600. }
  601. }
  602. //**********************************************************************************//
  603. // This function calculates the spherical Bessel (jn) and Hankel (h1n) functions //
  604. // and their derivatives for a given complex value z. See pag. 87 B&H. //
  605. // //
  606. // Input parameters: //
  607. // z: Complex argument to evaluate jn and h1n //
  608. // nmax_: Maximum number of terms to calculate jn and h1n //
  609. // //
  610. // Output parameters: //
  611. // jn, h1n: Spherical Bessel and Hankel functions //
  612. // jnp, h1np: Derivatives of the spherical Bessel and Hankel functions //
  613. // //
  614. // What we actually calculate are the Ricatti-Bessel fucntions and then simply //
  615. // evaluate the spherical Bessel and Hankel functions and their derivatives //
  616. // using the relations: //
  617. // //
  618. // j[n] = Psi[n]/z //
  619. // j'[n] = j[n-1]-(n+1)*jn[n])/z //
  620. // h1[n] = Zeta[n]/z //
  621. // h1'[n] = h1[n-1]-(n+1)*h1n[n]/z //
  622. // //
  623. //**********************************************************************************//
  624. void MultiLayerMie::sbesjh(std::complex<double> z,
  625. std::vector<std::complex<double> >& jn, std::vector<std::complex<double> >& jnp,
  626. std::vector<std::complex<double> >& h1n, std::vector<std::complex<double> >& h1np) {
  627. // std::vector<std::complex<double> > Psi(nmax_ + 1), Zeta(nmax_ + 1);
  628. // // First, calculate the Riccati-Bessel functions
  629. // calcPsiZeta(z, Psi, Zeta);
  630. // // Now, calculate Spherical Bessel and Hankel functions and their derivatives
  631. // for (int n = 0; n <= nmax_; n++) {
  632. // jn[n] = Psi[n]/z;
  633. // h1n[n] = Zeta[n]/z;
  634. // if (n == 0) {
  635. // jnp[0] = -Psi[1]/z - jn[0]/z;
  636. // h1np[0] = -Zeta[1]/z - h1n[0]/z;
  637. // } else {
  638. // jnp[n] = jn[n - 1] - static_cast<double>(n + 1)*jn[n]/z;
  639. // h1np[n] = h1n[n - 1] - static_cast<double>(n + 1)*h1n[n]/z;
  640. // }
  641. // }
  642. std::vector< std::complex<double> > yn, ynp;
  643. int nm;
  644. bessel::csphjy (nmax_, z, nm, jn, jnp, yn, ynp );
  645. auto c_i = std::complex<double>(0.0,1.0);
  646. h1n.resize(nmax_+1);
  647. h1np.resize(nmax_+1);
  648. for (int i = 0; i < nmax_+1; ++i) {
  649. h1n[i] = jn[i] + c_i*yn[i];
  650. h1np[i] = jnp[i] + c_i*ynp[i];
  651. }
  652. }
  653. //**********************************************************************************//
  654. // This function calculates Pi and Tau for a given value of cos(Theta). //
  655. // Equations (26a) - (26c) //
  656. // //
  657. // Input parameters: //
  658. // nmax_: Maximum number of terms to calculate Pi and Tau //
  659. // nTheta: Number of scattering angles //
  660. // Theta: Array containing all the scattering angles where the scattering //
  661. // amplitudes will be calculated //
  662. // //
  663. // Output parameters: //
  664. // Pi, Tau: Angular functions Pi and Tau, as defined in equations (26a) - (26c) //
  665. //**********************************************************************************//
  666. void MultiLayerMie::calcPiTau(const double& costheta,
  667. std::vector<double>& Pi, std::vector<double>& Tau) {
  668. int i;
  669. //****************************************************//
  670. // Equations (26a) - (26c) //
  671. //****************************************************//
  672. // Initialize Pi and Tau
  673. Pi[0] = 1.0; // n=1
  674. Tau[0] = costheta;
  675. // Calculate the actual values
  676. if (nmax_ > 1) {
  677. Pi[1] = 3*costheta*Pi[0]; //n=2
  678. Tau[1] = 2*costheta*Pi[1] - 3*Pi[0];
  679. for (i = 2; i < nmax_; i++) { //n=[3..nmax_]
  680. Pi[i] = ((i + i + 1)*costheta*Pi[i - 1] - (i + 1)*Pi[i - 2])/i;
  681. Tau[i] = (i + 1)*costheta*Pi[i] - (i + 2)*Pi[i - 1];
  682. }
  683. }
  684. } // end of MultiLayerMie::calcPiTau(...)
  685. //**********************************************************************************//
  686. // This function calculates vector spherical harmonics (eq. 4.50, p. 95 BH), //
  687. // required to calculate the near-field parameters. //
  688. // //
  689. // Input parameters: //
  690. // Rho: Radial distance //
  691. // Phi: Azimuthal angle //
  692. // Theta: Polar angle //
  693. // rn: Either the spherical Ricatti-Bessel function of first or third kind //
  694. // Dn: Logarithmic derivative of rn //
  695. // Pi, Tau: Angular functions Pi and Tau //
  696. // n: Order of vector spherical harmonics //
  697. // //
  698. // Output parameters: //
  699. // Mo1n, Me1n, No1n, Ne1n: Complex vector spherical harmonics //
  700. //**********************************************************************************//
  701. void MultiLayerMie::calcSpherHarm(const double Rho, const double Theta, const double Phi,
  702. const std::complex<double>& rn, const std::complex<double>& Dn,
  703. const double& Pi, const double& Tau, const double& n,
  704. std::vector<std::complex<double> >& Mo1n, std::vector<std::complex<double> >& Me1n,
  705. std::vector<std::complex<double> >& No1n, std::vector<std::complex<double> >& Ne1n) {
  706. // using eq 4.50 in BH
  707. std::complex<double> c_zero(0.0, 0.0);
  708. <<<<<<< HEAD
  709. =======
  710. std::complex<double> deriv = Rho*dzn + zn;
  711. //std::complex<double> deriv (6.6667e-04, 0.0);
  712. //printf("n=%g, Phi=%g Pi=%g Tau=%g zn=%+er%+ei deriv=%+er%+ei\n", n, Phi, Pi, Tau, zn.real(), zn.imag(), deriv.real(), deriv.imag());
  713. >>>>>>> 645549af67fbee6f0583798534e030f27a500a44
  714. using std::sin;
  715. using std::cos;
  716. Mo1n[0] = c_zero;
  717. Mo1n[1] = cos(Phi)*Pi*rn/Rho;
  718. Mo1n[2] = -sin(Phi)*Tau*rn/Rho;
  719. Me1n[0] = c_zero;
  720. Me1n[1] = -sin(Phi)*Pi*rn/Rho;
  721. Me1n[2] = -cos(Phi)*Tau*rn/Rho;
  722. No1n[0] = sin(Phi)*(n*n + n)*sin(Theta)*Pi*rn/Rho/Rho;
  723. No1n[1] = sin(Phi)*Tau*Dn*rn/Rho;
  724. No1n[2] = cos(Phi)*Pi*Dn*rn/Rho;
  725. Ne1n[0] = cos(Phi)*(n*n + n)*sin(Theta)*Pi*rn/Rho/Rho;
  726. Ne1n[1] = cos(Phi)*Tau*Dn*rn/Rho;
  727. Ne1n[2] = -sin(Phi)*Pi*Dn*rn/Rho;
  728. } // end of MultiLayerMie::calcSpherHarm(...)
  729. //**********************************************************************************//
  730. // This function calculates the scattering coefficients required to calculate //
  731. // both the near- and far-field parameters. //
  732. // //
  733. // Input parameters: //
  734. // L: Number of layers //
  735. // pl: Index of PEC layer. If there is none just send -1 //
  736. // x: Array containing the size parameters of the layers [0..L-1] //
  737. // m: Array containing the relative refractive indexes of the layers [0..L-1] //
  738. // nmax: Maximum number of multipolar expansion terms to be used for the //
  739. // calculations. Only use it if you know what you are doing, otherwise //
  740. // set this parameter to -1 and the function will calculate it. //
  741. // //
  742. // Output parameters: //
  743. // an, bn: Complex scattering amplitudes //
  744. // //
  745. // Return value: //
  746. // Number of multipolar expansion terms used for the calculations //
  747. //**********************************************************************************//
  748. void MultiLayerMie::ScattCoeffs() {
  749. isScaCoeffsCalc_ = false;
  750. const std::vector<double>& x = size_param_;
  751. const std::vector<std::complex<double> >& m = refractive_index_;
  752. const int& pl = PEC_layer_position_;
  753. const int L = refractive_index_.size();
  754. //************************************************************************//
  755. // Calculate the index of the first layer. It can be either 0 (default) //
  756. // or the index of the outermost PEC layer. In the latter case all layers //
  757. // below the PEC are discarded. //
  758. // ***********************************************************************//
  759. int fl = (pl > 0) ? pl : 0;
  760. if (nmax_preset_ <= 0) calcNmax(fl);
  761. else nmax_ = nmax_preset_;
  762. std::complex<double> z1, z2;
  763. //**************************************************************************//
  764. // Note that since Fri, Nov 14, 2014 all arrays start from 0 (zero), which //
  765. // means that index = layer number - 1 or index = n - 1. The only exception //
  766. // are the arrays for representing D1, D3 and Q because they need a value //
  767. // for the index 0 (zero), hence it is important to consider this shift //
  768. // between different arrays. The change was done to optimize memory usage. //
  769. //**************************************************************************//
  770. // Allocate memory to the arrays
  771. std::vector<std::complex<double> > D1_mlxl(nmax_ + 1), D1_mlxlM1(nmax_ + 1),
  772. D3_mlxl(nmax_ + 1), D3_mlxlM1(nmax_ + 1);
  773. std::vector<std::vector<std::complex<double> > > Q(L), Ha(L), Hb(L);
  774. for (int l = 0; l < L; l++) {
  775. Q[l].resize(nmax_ + 1);
  776. Ha[l].resize(nmax_);
  777. Hb[l].resize(nmax_);
  778. }
  779. an_.resize(nmax_);
  780. bn_.resize(nmax_);
  781. PsiZeta_.resize(nmax_ + 1);
  782. std::vector<std::complex<double> > PsiXL(nmax_ + 1), ZetaXL(nmax_ + 1);
  783. //*************************************************//
  784. // Calculate D1 and D3 for z1 in the first layer //
  785. //*************************************************//
  786. if (fl == pl) { // PEC layer
  787. for (int n = 0; n <= nmax_; n++) {
  788. D1_mlxl[n] = std::complex<double>(0.0, - 1.0);
  789. D3_mlxl[n] = std::complex<double>(0.0, 1.0);
  790. }
  791. } else { // Regular layer
  792. z1 = x[fl]* m[fl];
  793. // Calculate D1 and D3
  794. calcD1D3(z1, D1_mlxl, D3_mlxl);
  795. }
  796. //******************************************************************//
  797. // Calculate Ha and Hb in the first layer - equations (7a) and (8a) //
  798. //******************************************************************//
  799. for (int n = 0; n < nmax_; n++) {
  800. Ha[fl][n] = D1_mlxl[n + 1];
  801. Hb[fl][n] = D1_mlxl[n + 1];
  802. }
  803. //*****************************************************//
  804. // Iteration from the second layer to the last one (L) //
  805. //*****************************************************//
  806. std::complex<double> Temp, Num, Denom;
  807. std::complex<double> G1, G2;
  808. for (int l = fl + 1; l < L; l++) {
  809. //************************************************************//
  810. //Calculate D1 and D3 for z1 and z2 in the layers fl + 1..L //
  811. //************************************************************//
  812. z1 = x[l]*m[l];
  813. z2 = x[l - 1]*m[l];
  814. //Calculate D1 and D3 for z1
  815. calcD1D3(z1, D1_mlxl, D3_mlxl);
  816. //Calculate D1 and D3 for z2
  817. calcD1D3(z2, D1_mlxlM1, D3_mlxlM1);
  818. //*************************************************//
  819. //Calculate Q, Ha and Hb in the layers fl + 1..L //
  820. //*************************************************//
  821. // Upward recurrence for Q - equations (19a) and (19b)
  822. Num = std::exp(-2.0*(z1.imag() - z2.imag()))
  823. *std::complex<double>(std::cos(-2.0*z2.real()) - std::exp(-2.0*z2.imag()), std::sin(-2.0*z2.real()));
  824. Denom = std::complex<double>(std::cos(-2.0*z1.real()) - std::exp(-2.0*z1.imag()), std::sin(-2.0*z1.real()));
  825. Q[l][0] = Num/Denom;
  826. for (int n = 1; n <= nmax_; n++) {
  827. Num = (z1*D1_mlxl[n] + double(n))*(double(n) - z1*D3_mlxl[n - 1]);
  828. Denom = (z2*D1_mlxlM1[n] + double(n))*(double(n) - z2*D3_mlxlM1[n - 1]);
  829. Q[l][n] = ((pow2(x[l - 1]/x[l])* Q[l][n - 1])*Num)/Denom;
  830. }
  831. // Upward recurrence for Ha and Hb - equations (7b), (8b) and (12) - (15)
  832. for (int n = 1; n <= nmax_; n++) {
  833. //Ha
  834. if ((l - 1) == pl) { // The layer below the current one is a PEC layer
  835. G1 = -D1_mlxlM1[n];
  836. G2 = -D3_mlxlM1[n];
  837. } else {
  838. G1 = (m[l]*Ha[l - 1][n - 1]) - (m[l - 1]*D1_mlxlM1[n]);
  839. G2 = (m[l]*Ha[l - 1][n - 1]) - (m[l - 1]*D3_mlxlM1[n]);
  840. } // end of if PEC
  841. Temp = Q[l][n]*G1;
  842. Num = (G2*D1_mlxl[n]) - (Temp*D3_mlxl[n]);
  843. Denom = G2 - Temp;
  844. Ha[l][n - 1] = Num/Denom;
  845. //Hb
  846. if ((l - 1) == pl) { // The layer below the current one is a PEC layer
  847. G1 = Hb[l - 1][n - 1];
  848. G2 = Hb[l - 1][n - 1];
  849. } else {
  850. G1 = (m[l - 1]*Hb[l - 1][n - 1]) - (m[l]*D1_mlxlM1[n]);
  851. G2 = (m[l - 1]*Hb[l - 1][n - 1]) - (m[l]*D3_mlxlM1[n]);
  852. } // end of if PEC
  853. Temp = Q[l][n]*G1;
  854. Num = (G2*D1_mlxl[n]) - (Temp* D3_mlxl[n]);
  855. Denom = (G2- Temp);
  856. Hb[l][n - 1] = (Num/ Denom);
  857. } // end of for Ha and Hb terms
  858. } // end of for layers iteration
  859. //**************************************//
  860. //Calculate Psi and Zeta for XL //
  861. //**************************************//
  862. // Calculate PsiXL and ZetaXL
  863. calcPsiZeta(x[L - 1], PsiXL, ZetaXL);
  864. //*********************************************************************//
  865. // Finally, we calculate the scattering coefficients (an and bn) and //
  866. // the angular functions (Pi and Tau). Note that for these arrays the //
  867. // first layer is 0 (zero), in future versions all arrays will follow //
  868. // this convention to save memory. (13 Nov, 2014) //
  869. //*********************************************************************//
  870. for (int n = 0; n < nmax_; n++) {
  871. //********************************************************************//
  872. //Expressions for calculating an and bn coefficients are not valid if //
  873. //there is only one PEC layer (ie, for a simple PEC sphere). //
  874. //********************************************************************//
  875. if (pl < (L - 1)) {
  876. 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]);
  877. 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]);
  878. } else {
  879. 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]);
  880. bn_[n] = PsiXL[n + 1]/ZetaXL[n + 1];
  881. }
  882. } // end of for an and bn terms
  883. isScaCoeffsCalc_ = true;
  884. } // end of MultiLayerMie::ScattCoeffs(...)
  885. //**********************************************************************************//
  886. // This function calculates the actual scattering parameters and amplitudes //
  887. // //
  888. // Input parameters: //
  889. // L: Number of layers //
  890. // pl: Index of PEC layer. If there is none just send -1 //
  891. // x: Array containing the size parameters of the layers [0..L-1] //
  892. // m: Array containing the relative refractive indexes of the layers [0..L-1] //
  893. // nTheta: Number of scattering angles //
  894. // Theta: Array containing all the scattering angles where the scattering //
  895. // amplitudes will be calculated //
  896. // nmax_: Maximum number of multipolar expansion terms to be used for the //
  897. // calculations. Only use it if you know what you are doing, otherwise //
  898. // set this parameter to -1 and the function will calculate it //
  899. // //
  900. // Output parameters: //
  901. // Qext: Efficiency factor for extinction //
  902. // Qsca: Efficiency factor for scattering //
  903. // Qabs: Efficiency factor for absorption (Qabs = Qext - Qsca) //
  904. // Qbk: Efficiency factor for backscattering //
  905. // Qpr: Efficiency factor for the radiation pressure //
  906. // g: Asymmetry factor (g = (Qext-Qpr)/Qsca) //
  907. // Albedo: Single scattering albedo (Albedo = Qsca/Qext) //
  908. // S1, S2: Complex scattering amplitudes //
  909. // //
  910. // Return value: //
  911. // Number of multipolar expansion terms used for the calculations //
  912. //**********************************************************************************//
  913. void MultiLayerMie::RunMieCalculation() {
  914. if (size_param_.size() != refractive_index_.size())
  915. throw std::invalid_argument("Each size parameter should have only one index!");
  916. if (size_param_.size() == 0)
  917. throw std::invalid_argument("Initialize model first!");
  918. const std::vector<double>& x = size_param_;
  919. isExpCoeffsCalc_ = false;
  920. isScaCoeffsCalc_ = false;
  921. isMieCalculated_ = false;
  922. // Calculate scattering coefficients
  923. ScattCoeffs();
  924. if (!isScaCoeffsCalc_) // TODO seems to be unreachable
  925. throw std::invalid_argument("Calculation of scattering coefficients failed!");
  926. // Initialize the scattering parameters
  927. Qext_ = 0.0;
  928. Qsca_ = 0.0;
  929. Qabs_ = 0.0;
  930. Qbk_ = 0.0;
  931. Qpr_ = 0.0;
  932. asymmetry_factor_ = 0.0;
  933. albedo_ = 0.0;
  934. // Initialize the scattering amplitudes
  935. std::vector<std::complex<double> > tmp1(theta_.size(),std::complex<double>(0.0, 0.0));
  936. S1_.swap(tmp1);
  937. S2_ = S1_;
  938. std::vector<double> Pi(nmax_), Tau(nmax_);
  939. std::complex<double> Qbktmp(0.0, 0.0);
  940. std::vector< std::complex<double> > Qbktmp_ch(nmax_ - 1, Qbktmp);
  941. // By using downward recurrence we avoid loss of precision due to float rounding errors
  942. // See: https://docs.oracle.com/cd/E19957-01/806-3568/ncg_goldberg.html
  943. // http://en.wikipedia.org/wiki/Loss_of_significance
  944. for (int i = nmax_ - 2; i >= 0; i--) {
  945. const int n = i + 1;
  946. // Equation (27)
  947. Qext_ += (n + n + 1.0)*(an_[i].real() + bn_[i].real());
  948. // Equation (28)
  949. Qsca_ += (n + n + 1.0)*(an_[i].real()*an_[i].real() + an_[i].imag()*an_[i].imag()
  950. + bn_[i].real()*bn_[i].real() + bn_[i].imag()*bn_[i].imag());
  951. // Equation (29)
  952. Qpr_ += ((n*(n + 2)/(n + 1))*((an_[i]*std::conj(an_[n]) + bn_[i]*std::conj(bn_[n])).real())
  953. + ((double)(n + n + 1)/(n*(n + 1)))*(an_[i]*std::conj(bn_[i])).real());
  954. // Equation (33)
  955. Qbktmp += (double)(n + n + 1)*(1 - 2*(n % 2))*(an_[i]- bn_[i]);
  956. // Calculate the scattering amplitudes (S1 and S2) //
  957. // Equations (25a) - (25b) //
  958. for (unsigned int t = 0; t < theta_.size(); t++) {
  959. calcPiTau(std::cos(theta_[t]), Pi, Tau);
  960. S1_[t] += calc_S1(n, an_[i], bn_[i], Pi[i], Tau[i]);
  961. S2_[t] += calc_S2(n, an_[i], bn_[i], Pi[i], Tau[i]);
  962. }
  963. }
  964. double x2 = pow2(x.back());
  965. Qext_ = 2.0*(Qext_)/x2; // Equation (27)
  966. Qsca_ = 2.0*(Qsca_)/x2; // Equation (28)
  967. Qpr_ = Qext_ - 4.0*(Qpr_)/x2; // Equation (29)
  968. Qabs_ = Qext_ - Qsca_; // Equation (30)
  969. albedo_ = Qsca_/Qext_; // Equation (31)
  970. asymmetry_factor_ = (Qext_ - Qpr_)/Qsca_; // Equation (32)
  971. Qbk_ = (Qbktmp.real()*Qbktmp.real() + Qbktmp.imag()*Qbktmp.imag())/x2; // Equation (33)
  972. isMieCalculated_ = true;
  973. }
  974. //**********************************************************************************//
  975. // This function calculates the expansion coefficients inside the particle, //
  976. // required to calculate the near-field parameters. //
  977. // //
  978. // Input parameters: //
  979. // L: Number of layers //
  980. // pl: Index of PEC layer. If there is none just send -1 //
  981. // x: Array containing the size parameters of the layers [0..L-1] //
  982. // m: Array containing the relative refractive indexes of the layers [0..L-1] //
  983. // nmax: Maximum number of multipolar expansion terms to be used for the //
  984. // calculations. Only use it if you know what you are doing, otherwise //
  985. // set this parameter to -1 and the function will calculate it. //
  986. // //
  987. // Output parameters: //
  988. // aln, bln, cln, dln: Complex scattering amplitudes inside the particle //
  989. // //
  990. // Return value: //
  991. // Number of multipolar expansion terms used for the calculations //
  992. //**********************************************************************************//
  993. void MultiLayerMie::ExpanCoeffs() {
  994. if (!isScaCoeffsCalc_)
  995. throw std::invalid_argument("(ExpanCoeffs) You should calculate external coefficients first!");
  996. isExpCoeffsCalc_ = false;
  997. std::complex<double> c_one(1.0, 0.0), c_zero(0.0, 0.0);
  998. const int L = refractive_index_.size();
  999. aln_.resize(L + 1);
  1000. bln_.resize(L + 1);
  1001. cln_.resize(L + 1);
  1002. dln_.resize(L + 1);
  1003. for (int l = 0; l <= L; l++) {
  1004. aln_[l].resize(nmax_);
  1005. bln_[l].resize(nmax_);
  1006. cln_[l].resize(nmax_);
  1007. dln_[l].resize(nmax_);
  1008. }
  1009. // Yang, paragraph under eq. A3
  1010. // a^(L + 1)_n = a_n, d^(L + 1) = 1 ...
  1011. for (int n = 0; n < nmax_; n++) {
  1012. aln_[L][n] = an_[n];
  1013. bln_[L][n] = bn_[n];
  1014. cln_[L][n] = c_one;
  1015. dln_[L][n] = c_one;
  1016. printf("aln_[%02i, %02i] = %g,%g; bln_[%02i, %02i] = %g,%g; cln_[%02i, %02i] = %g,%g; dln_[%02i, %02i] = %g,%g\n", L, n, std::real(aln_[L][n]), std::imag(aln_[L][n]), L, n, std::real(bln_[L][n]), std::imag(bln_[L][n]), L, n, std::real(cln_[L][n]), std::imag(cln_[L][n]), L, n, std::real(dln_[L][n]), std::imag(dln_[L][n]));
  1017. }
  1018. std::vector<std::complex<double> > D1z(nmax_ + 1), D1z1(nmax_ + 1), D3z(nmax_ + 1), D3z1(nmax_ + 1);
  1019. std::vector<std::complex<double> > Psiz(nmax_ + 1), Psiz1(nmax_ + 1), Zetaz(nmax_ + 1), Zetaz1(nmax_ + 1);
  1020. std::complex<double> denomZeta, denomPsi, T1, T2, T3, T4;
  1021. auto& m = refractive_index_;
  1022. std::vector< std::complex<double> > m1(L);
  1023. for (int l = 0; l < L - 1; l++) m1[l] = m[l + 1];
  1024. m1[L - 1] = std::complex<double> (1.0, 0.0);
  1025. std::complex<double> z, z1;
  1026. for (int l = L - 1; l >= 0; l--) {
  1027. z = size_param_[l]*m[l];
  1028. z1 = size_param_[l]*m1[l];
  1029. calcD1D3(z, D1z, D3z);
  1030. calcD1D3(z1, D1z1, D3z1);
  1031. calcPsiZeta(z, Psiz, Zetaz);
  1032. calcPsiZeta(z1, Psiz1, Zetaz1);
  1033. for (int n = 0; n < nmax_; n++) {
  1034. int n1 = n + 1;
  1035. denomZeta = m1[l]*Zetaz[n1]*(D1z[n1] - D3z[n1]);
  1036. denomPsi = m1[l]*Psiz[n1]*(D1z[n1] - D3z[n1]);
  1037. T1 = aln_[l + 1][n]*Zetaz1[n1] - dln_[l + 1][n]*Psiz1[n1];
  1038. T2 = bln_[l + 1][n]*Zetaz1[n1] - cln_[l + 1][n]*Psiz1[n1];
  1039. T3 = D1z1[n1]*dln_[l + 1][n]*Psiz1[n1] - D3z1[n1]*aln_[l + 1][n]*Zetaz1[n1];
  1040. T4 = D1z1[n1]*cln_[l + 1][n]*Psiz1[n1] - D3z1[n1]*bln_[l + 1][n]*Zetaz1[n1];
  1041. // aln
  1042. aln_[l][n] = (D1z[n1]*m1[l]*T1 + m[l]*T3)/denomZeta;
  1043. // bln
  1044. bln_[l][n] = (D1z[n1]*m[l]*T2 + m1[l]*T4)/denomZeta;
  1045. // cln
  1046. cln_[l][n] = (D3z[n1]*m[l]*T2 + m1[l]*T4)/denomPsi;
  1047. // dln
  1048. dln_[l][n] = (D3z[n1]*m1[l]*T1 + m[l]*T3)/denomPsi;
  1049. printf("aln_[%02i, %02i] = %g,%g; bln_[%02i, %02i] = %g,%g; cln_[%02i, %02i] = %g,%g; dln_[%02i, %02i] = %g,%g\n", l, n, real(aln_[l][n]), imag(aln_[l][n]), l, n, real(bln_[l][n]), imag(bln_[l][n]), l, n, real(cln_[l][n]), imag(cln_[l][n]), l, n, real(dln_[l][n]), imag(dln_[l][n]));
  1050. } // end of all n
  1051. } // end of all l
  1052. // Check the result and change aln_[0][n] and aln_[0][n] for exact zero
  1053. for (int n = 0; n < nmax_; ++n) {
  1054. // printf("n=%d, aln_=%g,%g, bln_=%g,%g \n", n, real(aln_[0][n]), imag(aln_[0][n]),
  1055. // real(bln_[0][n]), imag(bln_[0][n]));
  1056. if (std::abs(aln_[0][n]) < 1e-10) aln_[0][n] = 0.0;
  1057. else throw std::invalid_argument("Unstable calculation of aln_[0][n]!");
  1058. if (std::abs(bln_[0][n]) < 1e-10) bln_[0][n] = 0.0;
  1059. else throw std::invalid_argument("Unstable calculation of bln_[0][n]!");
  1060. }
  1061. isExpCoeffsCalc_ = true;
  1062. } // end of void MultiLayerMie::ExpanCoeffs()
  1063. //**********************************************************************************//
  1064. // This function calculates the expansion coefficients inside the particle, //
  1065. // required to calculate the near-field parameters. //
  1066. // //
  1067. // Input parameters: //
  1068. // L: Number of layers //
  1069. // pl: Index of PEC layer. If there is none just send -1 //
  1070. // x: Array containing the size parameters of the layers [0..L-1] //
  1071. // m: Array containing the relative refractive indexes of the layers [0..L-1] //
  1072. // nmax: Maximum number of multipolar expansion terms to be used for the //
  1073. // calculations. Only use it if you know what you are doing, otherwise //
  1074. // set this parameter to -1 and the function will calculate it. //
  1075. // //
  1076. // Output parameters: //
  1077. // aln, bln, cln, dln: Complex scattering amplitudes inside the particle //
  1078. // //
  1079. // Return value: //
  1080. // Number of multipolar expansion terms used for the calculations //
  1081. //**********************************************************************************//
  1082. void MultiLayerMie::ExpanCoeffsV2() {
  1083. if (!isScaCoeffsCalc_)
  1084. throw std::invalid_argument("(ExpanCoeffs) You should calculate external coefficients first!");
  1085. isExpCoeffsCalc_ = false;
  1086. std::complex<double> c_one(1.0, 0.0), c_zero(0.0, 0.0);
  1087. const int L = refractive_index_.size();
  1088. aln_.resize(L + 1);
  1089. bln_.resize(L + 1);
  1090. cln_.resize(L + 1);
  1091. dln_.resize(L + 1);
  1092. for (int l = 0; l <= L; l++) {
  1093. aln_[l].resize(nmax_);
  1094. bln_[l].resize(nmax_);
  1095. cln_[l].resize(nmax_);
  1096. dln_[l].resize(nmax_);
  1097. }
  1098. // Yang, paragraph under eq. A3
  1099. // a^(L + 1)_n = a_n, d^(L + 1) = 1 ...
  1100. for (int n = 0; n < nmax_; n++) {
  1101. aln_[L][n] = an_[n];
  1102. bln_[L][n] = bn_[n];
  1103. cln_[L][n] = c_one;
  1104. dln_[L][n] = c_one;
  1105. printf("aln_[%02i, %02i] = %g,%g; bln_[%02i, %02i] = %g,%g; cln_[%02i, %02i] = %g,%g; dln_[%02i, %02i] = %g,%g\n", L, n, std::real(aln_[L][n]), std::imag(aln_[L][n]), L, n, std::real(bln_[L][n]), std::imag(bln_[L][n]), L, n, std::real(cln_[L][n]), std::imag(cln_[L][n]), L, n, real(dln_[L][n]), std::imag(dln_[L][n]));
  1106. }
  1107. std::vector<std::complex<double> > D1z(nmax_ + 1), D1z1(nmax_ + 1), D3z(nmax_ + 1), D3z1(nmax_ + 1);
  1108. std::vector<std::complex<double> > Psiz(nmax_ + 1), Psiz1(nmax_ + 1), Zetaz(nmax_ + 1), Zetaz1(nmax_ + 1);
  1109. std::complex<double> denomZeta, denomPsi, T1, T2, T3, T4;
  1110. std::vector<std::vector<std::complex<double> > > a(2);
  1111. a[0].resize(3);
  1112. a[1].resize(3);
  1113. auto& m = refractive_index_;
  1114. std::vector< std::complex<double> > m1(L);
  1115. for (int l = 0; l < L - 1; l++) m1[l] = m[l + 1];
  1116. m1[L - 1] = std::complex<double> (1.0, 0.0);
  1117. std::complex<double> z, z1;
  1118. for (int l = L - 1; l >= 0; l--) {
  1119. z = size_param_[l]*m[l];
  1120. z1 = size_param_[l]*m1[l];
  1121. calcD1D3(z, D1z, D3z);
  1122. calcD1D3(z1, D1z1, D3z1);
  1123. calcPsiZeta(z, Psiz, Zetaz);
  1124. calcPsiZeta(z1, Psiz1, Zetaz1);
  1125. for (int n = 0; n < nmax_; n++) {
  1126. int n1 = n + 1;
  1127. a[0][0] = m1[l]*D3z[n1]*Zetaz[n1];
  1128. a[0][1] = -m1[l]*D1z[n1]*Psiz[n1];
  1129. a[0][2] = aln_[l + 1][n]*m[l]*D3z1[n1]*Zetaz1[n1];
  1130. a[0][2] -= dln_[l + 1][n]*m[l]*D1z1[n1]*Psiz1[n1];
  1131. a[1][0] = Zetaz[n1];
  1132. a[1][1] = -Psiz[n1];
  1133. a[1][2] = aln_[l + 1][n]*Zetaz1[n1] - dln_[l + 1][n]*Psiz1[n1];
  1134. // aln
  1135. aln_[l][n] = (a[0][2]*a[1][1] - a[0][1]*a[1][2])/(a[0][0]*a[1][1] - a[0][1]*a[1][0]);
  1136. // dln
  1137. dln_[l][n] = (a[0][2]*a[1][0] - a[0][0]*a[1][2])/(a[0][1]*a[1][0] - a[0][0]*a[0][1]);
  1138. /*for (int i = 0; i < 2; i++) {
  1139. for (int j = 0; j < 3; j++) {
  1140. printf("a[%i, %i] = %g,%g ", i, j, real(a[i][j]), imag(a[i][j]));
  1141. }
  1142. printf("\n");
  1143. }
  1144. printf("aln_[%i, %i] = %g,%g; dln_[%i, %i] = %g,%g\n\n", l, n, real(aln_[l][n]), imag(aln_[l][n]), l, n, real(dln_[l][n]), imag(dln_[l][n]));*/
  1145. a[0][0] = D3z[n1]*Zetaz[n1];
  1146. a[0][1] = -D1z[n1]*Psiz[n1];
  1147. a[0][2] = bln_[l + 1][n]*D3z1[n1]*Zetaz1[n1];
  1148. a[0][2] -= cln_[l + 1][n]*D1z1[n1]*Psiz1[n1];
  1149. a[1][0] = m1[l]*Zetaz[n1];
  1150. a[1][1] = -m1[l]*Psiz[n1];
  1151. a[1][2] = bln_[l + 1][n]*m[l]*Zetaz1[n1] - cln_[l + 1][n]*m[l]*Psiz1[n1];
  1152. // bln
  1153. bln_[l][n] = (a[0][2]*a[1][1] - a[0][1]*a[1][2])/(a[0][0]*a[1][1] - a[0][1]*a[1][0]);
  1154. // cln
  1155. cln_[l][n] = (a[0][2]*a[1][0] - a[0][0]*a[1][2])/(a[0][1]*a[1][0] - a[0][0]*a[0][1]);
  1156. printf("aln_[%02i, %02i] = %g,%g; bln_[%02i, %02i] = %g,%g; cln_[%02i, %02i] = %g,%g; dln_[%02i, %02i] = %g,%g\n", l, n, real(aln_[l][n]), imag(aln_[l][n]), l, n, real(bln_[l][n]), imag(bln_[l][n]), l, n, real(cln_[l][n]), imag(cln_[l][n]), l, n, real(dln_[l][n]), imag(dln_[l][n]));
  1157. } // end of all n
  1158. } // end of all l
  1159. // Check the result and change aln_[0][n] and aln_[0][n] for exact zero
  1160. for (int n = 0; n < nmax_; ++n) {
  1161. // printf("n=%d, aln_=%g,%g, bln_=%g,%g \n", n, real(aln_[0][n]), imag(aln_[0][n]),
  1162. // real(bln_[0][n]), imag(bln_[0][n]));
  1163. if (std::abs(aln_[0][n]) < 1e-1) aln_[0][n] = 0.0;
  1164. else throw std::invalid_argument("Unstable calculation of aln_[0][n]!");
  1165. if (std::abs(bln_[0][n]) < 1e-1) bln_[0][n] = 0.0;
  1166. else throw std::invalid_argument("Unstable calculation of bln_[0][n]!");
  1167. }
  1168. isExpCoeffsCalc_ = true;
  1169. } // end of void MultiLayerMie::ExpanCoeffs()
  1170. // ********************************************************************** //
  1171. // external scattering field = incident + scattered //
  1172. // BH p.92 (4.37), 94 (4.45), 95 (4.50) //
  1173. // assume: medium is non-absorbing; refim = 0; Uabs = 0 //
  1174. // ********************************************************************** //
  1175. void MultiLayerMie::fieldExt(const double Rho, const double Theta, const double Phi,
  1176. std::vector<std::complex<double> >& E, std::vector<std::complex<double> >& H) {
  1177. std::complex<double> c_zero(0.0, 0.0), c_i(0.0, 1.0), c_one(1.0, 0.0);
  1178. std::vector<std::complex<double> > ipow = {c_one, c_i, -c_one, -c_i}; // Vector containing precomputed integer powers of i to avoid computation
  1179. std::vector<std::complex<double> > M3o1n(3), M3e1n(3), N3o1n(3), N3e1n(3);
  1180. std::vector<std::complex<double> > Ei(3, c_zero), Hi(3, c_zero), Es(3, c_zero), Hs(3, c_zero);
  1181. std::vector<std::complex<double> > Psi(nmax_ + 1), D1n(nmax_ + 1), Zeta(nmax_ + 1), D3n(nmax_ + 1);
  1182. std::vector<double> Pi(nmax_), Tau(nmax_);
  1183. // Avoid calculation inside the particle
  1184. if (Rho < size_param_.back()) {
  1185. for (int i = 0; i < 3; i++) {
  1186. E[i] = c_zero;
  1187. H[i] = c_zero;
  1188. }
  1189. return;
  1190. }
  1191. calcD1D3(Rho, D1n, D3n);
  1192. calcPsiZeta(Rho, Psi, Zeta);
  1193. // Calculate spherical Bessel and Hankel functions
  1194. //sbesjh(Rho, jn, jnp, h1n, h1np);
  1195. // Calculate angular functions Pi and Tau
  1196. calcPiTau(std::cos(Theta), Pi, Tau);
  1197. for (int n = 0; n < nmax_; n++) {
  1198. int n1 = n + 1;
  1199. double rn = static_cast<double>(n1);
  1200. // using BH 4.12 and 4.50
  1201. calcSpherHarm(Rho, Theta, Phi, Zeta[n1], D3n[n1], Pi[n], Tau[n], rn, M3o1n, M3e1n, N3o1n, N3e1n);
  1202. // scattered field: BH p.94 (4.45)
  1203. std::complex<double> En = ipow[n1 % 4]*(rn + rn + 1.0)/(rn*rn + rn);
  1204. for (int i = 0; i < 3; i++) {
  1205. Es[i] = Es[i] + En*(c_i*an_[n]*N3e1n[i] - bn_[n]*M3o1n[i]);
  1206. Hs[i] = Hs[i] + En*(c_i*bn_[n]*N3o1n[i] + an_[n]*M3e1n[i]);
  1207. }
  1208. }
  1209. // incident E field: BH p.89 (4.21); cf. p.92 (4.37), p.93 (4.38)
  1210. // basis unit vectors = er, etheta, ephi
  1211. std::complex<double> eifac = std::exp(std::complex<double>(0.0, Rho*std::cos(Theta)));
  1212. {
  1213. using std::sin;
  1214. using std::cos;
  1215. Ei[0] = eifac*sin(Theta)*cos(Phi);
  1216. Ei[1] = eifac*cos(Theta)*cos(Phi);
  1217. Ei[2] = -eifac*sin(Phi);
  1218. }
  1219. // magnetic field
  1220. double hffact = 1.0/(cc_*mu_);
  1221. for (int i = 0; i < 3; i++) {
  1222. Hs[i] = hffact*Hs[i];
  1223. }
  1224. // incident H field: BH p.26 (2.43), p.89 (4.21)
  1225. std::complex<double> hffacta = hffact;
  1226. std::complex<double> hifac = eifac*hffacta;
  1227. {
  1228. using std::sin;
  1229. using std::cos;
  1230. Hi[0] = hifac*sin(Theta)*sin(Phi);
  1231. Hi[1] = hifac*cos(Theta)*sin(Phi);
  1232. Hi[2] = hifac*cos(Phi);
  1233. }
  1234. for (int i = 0; i < 3; i++) {
  1235. // electric field E [V m - 1] = EF*E0
  1236. E[i] = Ei[i] + Es[i];
  1237. H[i] = Hi[i] + Hs[i];
  1238. }
  1239. } // end of MultiLayerMie::fieldExt(...)
  1240. //**********************************************************************************//
  1241. // This function calculates the electric (E) and magnetic (H) fields inside and //
  1242. // around the particle. //
  1243. // //
  1244. // Input parameters (coordinates of the point): //
  1245. // Rho: Radial distance //
  1246. // Phi: Azimuthal angle //
  1247. // Theta: Polar angle //
  1248. // //
  1249. // Output parameters: //
  1250. // E, H: Complex electric and magnetic fields //
  1251. //**********************************************************************************//
  1252. void MultiLayerMie::calcField(const double Rho, const double Theta, const double Phi,
  1253. std::vector<std::complex<double> >& E, std::vector<std::complex<double> >& H) {
  1254. std::complex<double> c_zero(0.0, 0.0), c_i(0.0, 1.0), c_one(1.0, 0.0);
  1255. std::vector<std::complex<double> > ipow = {c_one, c_i, -c_one, -c_i}; // Vector containing precomputed integer powers of i to avoid computation
  1256. std::vector<std::complex<double> > M3o1n(3), M3e1n(3), N3o1n(3), N3e1n(3);
  1257. std::vector<std::complex<double> > M1o1n(3), M1e1n(3), N1o1n(3), N1e1n(3);
  1258. std::vector<std::complex<double> > Psi(nmax_ + 1), D1n(nmax_ + 1), Zeta(nmax_ + 1), D3n(nmax_ + 1);
  1259. std::vector<double> Pi(nmax_), Tau(nmax_);
  1260. std::vector<std::complex<double> > Ei(3), Hi(3);
  1261. int l = 0; // Layer number
  1262. std::complex<double> ml;
  1263. // Initialize E and H
  1264. for (int i = 0; i < 3; i++) {
  1265. E[i] = c_zero;
  1266. H[i] = c_zero;
  1267. }
  1268. if (Rho > size_param_.back()) {
  1269. l = size_param_.size();
  1270. ml = c_one;
  1271. } else {
  1272. for (int i = size_param_.size() - 1; i >= 0 ; i--) {
  1273. if (Rho <= size_param_[i]) {
  1274. l = i;
  1275. break;
  1276. }
  1277. }
  1278. <<<<<<< HEAD
  1279. ml = refractive_index_[l];
  1280. }
  1281. calcD1D3(Rho, D1n, D3n);
  1282. calcPsiZeta(Rho, Psi, Zeta);
  1283. // Calculate spherical Bessel and Hankel functions and their derivatives
  1284. //sbesjh(Rho*ml, jn, jnp, h1n, h1np);
  1285. =======
  1286. ml = refractive_index_[l]; // TODO check the correct layer is used
  1287. }
  1288. // Calculate spherical Bessel and Hankel functions and their derivatives
  1289. sbesjh(2.0*PI_*Rho*ml, jn, jnp, h1n, h1np);
  1290. printf("2.0*PI*Rho*ml = %10.5er%+10.5ei\n",std::real(2.0*PI_*Rho*ml), std::imag(2.0*PI_*Rho*ml));
  1291. >>>>>>> 645549af67fbee6f0583798534e030f27a500a44
  1292. // Calculate angular functions Pi and Tau
  1293. calcPiTau(std::cos(Theta), Pi, Tau);
  1294. printf("Thetd = %g, cos(Theta) = %g\n", Theta, std::cos(Theta));
  1295. printf("jn:\n");
  1296. for (auto p : jn) printf("%+11.4er%+11.4ei\n",p.real(), p.imag());
  1297. printf("Pi:\n");
  1298. for (auto p : Pi) printf("%11.4e\n",p);
  1299. printf("Tau:\n");
  1300. for (auto p : Tau) printf("%11.4e\n",p);
  1301. for (int n = nmax_ - 2; n >= 0; n--) {
  1302. int n1 = n + 1;
  1303. double rn = static_cast<double>(n1);
  1304. // using BH 4.12 and 4.50
  1305. <<<<<<< HEAD
  1306. calcSpherHarm(Rho, Theta, Phi, Psi[n1], D1n[n1], Pi[n], Tau[n], rn, M1o1n, M1e1n, N1o1n, N1e1n);
  1307. calcSpherHarm(Rho, Theta, Phi, Zeta[n1], D3n[n1], Pi[n], Tau[n], rn, M3o1n, M3e1n, N3o1n, N3e1n);
  1308. =======
  1309. calcSpherHarm(Rho, Theta, Phi, jn[n1], jnp[n1], Pi[n], Tau[n], rn, M1o1n, M1e1n, N1o1n, N1e1n);
  1310. calcSpherHarm(Rho, Theta, Phi, h1n[n1], h1np[n1], Pi[n], Tau[n], rn, M3o1n, M3e1n, N3o1n, N3e1n); //TODO uncomment line
  1311. auto deriv1 = -rn*jn[n1]+Rho*jn[n1-1];
  1312. auto deriv2 = Rho*jnp[n1] + jn[n1];
  1313. printf("n=%d deriv1: %+11.4e deriv2: %+11.4ei\n",n1, deriv1.real(), deriv2.real());
  1314. // printf("N1e1n[%d]: ", n1);
  1315. // for (auto p : N1e1n) printf("%+11.4er%+11.4ei\t",p.real(), p.imag());
  1316. // printf("\n");
  1317. >>>>>>> 645549af67fbee6f0583798534e030f27a500a44
  1318. // Total field in the lth layer: eqs. (1) and (2) in Yang, Appl. Opt., 42 (2003) 1710-1720
  1319. std::complex<double> En = ipow[n1 % 4]*(rn + rn + 1.0)/(rn*rn + rn);
  1320. for (int i = 0; i < 3; i++) {
  1321. // electric field E [V m - 1] = EF*E0
  1322. E[i] += En*(cln_[l][n]*M1o1n[i] - c_i*dln_[l][n]*N1e1n[i]
  1323. + c_i*aln_[l][n]*N3e1n[i] - bln_[l][n]*M3o1n[i]);
  1324. H[i] += En*(-dln_[l][n]*M1e1n[i] - c_i*cln_[l][n]*N1o1n[i]
  1325. + c_i*bln_[l][n]*N3o1n[i] + aln_[l][n]*M3e1n[i]);
  1326. Ei[i] += En*(M1o1n[i] - c_i*N1e1n[i]);
  1327. Hi[i] += En*(-M1e1n[i] - c_i*N1o1n[i]);
  1328. }
  1329. } // end of for all n
  1330. //printf("rho = %11.4e; phi = %11.4eº; theta = %11.4eº; x[%i] = %11.4e; m[%i] = %11.4er%+10.5ei\n", Rho, Phi*180./PI_, Theta*180./PI_, l, size_param_[l], l, std::real(ml), std::imag(ml));
  1331. // magnetic field
  1332. double hffact = 1.0/(cc_*mu_);
  1333. for (int i = 0; i < 3; i++) {
  1334. H[i] = hffact*H[i];
  1335. Hi[i] *= hffact;
  1336. printf("E[%i] = %10.5er%+10.5ei; Ei[%i] = %10.5er%+10.5ei; H[%i] = %10.5er%+10.5ei; Hi[%i] = %10.5er%+10.5ei\n", i, std::real(E[i]), std::imag(E[i]), i, std::real(Ei[i]), std::imag(Ei[i]), i, std::real(H[i]), std::imag(H[i]), i, std::real(Hi[i]), std::imag(Hi[i]));
  1337. }
  1338. } // end of MultiLayerMie::calcField(...)
  1339. //**********************************************************************************//
  1340. // This function calculates complex electric and magnetic field in the surroundings //
  1341. // and inside the particle. //
  1342. // //
  1343. // Input parameters: //
  1344. // L: Number of layers //
  1345. // pl: Index of PEC layer. If there is none just send 0 (zero) //
  1346. // x: Array containing the size parameters of the layers [0..L-1] //
  1347. // m: Array containing the relative refractive indexes of the layers [0..L-1] //
  1348. // nmax: Maximum number of multipolar expansion terms to be used for the //
  1349. // calculations. Only use it if you know what you are doing, otherwise //
  1350. // set this parameter to 0 (zero) and the function will calculate it. //
  1351. // ncoord: Number of coordinate points //
  1352. // Coords: Array containing all coordinates where the complex electric and //
  1353. // magnetic fields will be calculated //
  1354. // //
  1355. // Output parameters: //
  1356. // E, H: Complex electric and magnetic field at the provided coordinates //
  1357. // //
  1358. // Return value: //
  1359. // Number of multipolar expansion terms used for the calculations //
  1360. //**********************************************************************************//
  1361. void MultiLayerMie::RunFieldCalculation() {
  1362. double Rho, Theta, Phi;
  1363. // Calculate scattering coefficients an_ and bn_
  1364. ScattCoeffs();
  1365. // std::vector<std::complex<double> > an1(nmax_), bn1(nmax_);
  1366. // calc_an_bn_bulk(an1, bn1, size_param_.back(), refractive_index_.back());
  1367. // for (int n = 0; n < nmax_; n++) {
  1368. // printf("an_[%i] = %11.4er%+10.5ei; an_bulk_[%i] = %11.4er%+10.5ei\n", n, std::real(an_[n]), std::imag(an_[n]), n, std::real(an1[n]), std::imag(an1[n]));
  1369. // printf("bn_[%i] = %11.4er%+10.5ei; bn_bulk_[%i] = %11.4er%+10.5ei\n", n, std::real(bn_[n]), std::imag(bn_[n]), n, std::real(bn1[n]), std::imag(bn1[n]));
  1370. // }
  1371. // Calculate expansion coefficients aln_, bln_, cln_, and dln_
  1372. ExpanCoeffs();
  1373. // for (int i = 0; i < nmax_; ++i) {
  1374. // printf("cln_[%i] = %11.4er%+10.5ei; dln_[%i] = %11.4er%+10.5ei\n", i, std::real(cln_[0][i]), std::imag(cln_[0][i]),
  1375. // i, std::real(dln_[0][i]), std::imag(dln_[0][i]));
  1376. // }
  1377. long total_points = coords_[0].size();
  1378. E_.resize(total_points);
  1379. H_.resize(total_points);
  1380. for (auto& f : E_) f.resize(3);
  1381. for (auto& f : H_) f.resize(3);
  1382. for (int point = 0; point < total_points; point++) {
  1383. const double& Xp = coords_[0][point];
  1384. const double& Yp = coords_[1][point];
  1385. const double& Zp = coords_[2][point];
  1386. // Convert to spherical coordinates
  1387. Rho = std::sqrt(pow2(Xp) + pow2(Yp) + pow2(Zp));
  1388. // If Rho=0 then Theta is undefined. Just set it to zero to avoid problems
  1389. Theta = (Rho > 0.0) ? std::acos(Zp/Rho) : 0.0;
  1390. // If Xp=Yp=0 then Phi is undefined. Just set it to zero to avoid problems
  1391. if (Xp == 0.0)
  1392. Phi = (Yp != 0.0) ? std::asin(Yp/std::sqrt(pow2(Xp) + pow2(Yp))) : 0.0;
  1393. else
  1394. Phi = std::acos(Xp/std::sqrt(pow2(Xp) + pow2(Yp)));
  1395. // Avoid convergence problems due to Rho too small
  1396. if (Rho < 1e-5) Rho = 1e-5;
  1397. //printf("X = %g; Y = %g; Z = %g; pho = %g; phi = %g; theta = %g\n", Xp, Yp, Zp, Rho, Phi*180./PI_, Theta*180./PI_);
  1398. //*******************************************************//
  1399. // external scattering field = incident + scattered //
  1400. // BH p.92 (4.37), 94 (4.45), 95 (4.50) //
  1401. // assume: medium is non-absorbing; refim = 0; Uabs = 0 //
  1402. //*******************************************************//
  1403. // This array contains the fields in spherical coordinates
  1404. std::vector<std::complex<double> > Es(3), Hs(3);
  1405. // Firstly the easiest case: the field outside the particle
  1406. // if (Rho >= GetSizeParameter()) {
  1407. // fieldExt(Rho, Theta, Phi, Es, Hs);
  1408. // } else {
  1409. calcField(Rho, Theta, Phi, Es, Hs); //Should work fine both: inside and outside the particle
  1410. //}
  1411. { //Now, convert the fields back to cartesian coordinates
  1412. using std::sin;
  1413. using std::cos;
  1414. E_[point][0] = sin(Theta)*cos(Phi)*Es[0] + cos(Theta)*cos(Phi)*Es[1] - sin(Phi)*Es[2];
  1415. E_[point][1] = sin(Theta)*sin(Phi)*Es[0] + cos(Theta)*sin(Phi)*Es[1] + cos(Phi)*Es[2];
  1416. E_[point][2] = cos(Theta)*Es[0] - sin(Theta)*Es[1];
  1417. H_[point][0] = sin(Theta)*cos(Phi)*Hs[0] + cos(Theta)*cos(Phi)*Hs[1] - sin(Phi)*Hs[2];
  1418. H_[point][1] = sin(Theta)*sin(Phi)*Hs[0] + cos(Theta)*sin(Phi)*Hs[1] + cos(Phi)*Hs[2];
  1419. H_[point][2] = cos(Theta)*Hs[0] - sin(Theta)*Hs[1];
  1420. }
  1421. } // end of for all field coordinates
  1422. } // end of MultiLayerMie::RunFieldCalculation()
  1423. } // end of namespace nmie