123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803 |
- #include <math.h>
- #include <stdlib.h>
- #include <stdio.h>
- #include "nmie.h"
- #define round(x) ((x) >= 0 ? (int)((x) + 0.5):(int)((x) - 0.5))
- int Nstop(double xL) {
- int result;
- if (xL <= 8) {
- result = round(xL + 4*pow(xL, 1/3) + 1);
- } else if (xL <= 4200) {
- result = round(xL + 4.05*pow(xL, 1/3) + 2);
- } else {
- result = round(xL + 4*pow(xL, 1/3) + 2);
- }
- return result;
- }
- int Nmax(int L, int fl, int pl,
- std::vector<double> x,
- std::vector<std::complex<double> > m) {
- int i, result, ri, riM1;
- result = Nstop(x[L - 1]);
- for (i = fl; i < L; i++) {
- if (i > pl) {
- ri = round(std::abs(x[i]*m[i]));
- } else {
- ri = 0;
- }
- if (result < ri) {
- result = ri;
- }
- if ((i > fl) && ((i - 1) > pl)) {
- riM1 = round(std::abs(x[i - 1]* m[i]));
- } else {
- riM1 = 0;
- }
- if (result < riM1) {
- result = riM1;
- }
- }
- return result + 15;
- }
- void sphericalBessel(std::complex<double> r, int n_max, std::vector<std::complex<double> > &j, std::vector<std::complex<double> > &h) {
- int n;
- j[0] = sin(r)/r;
- j[1] = sin(r)/r/r - cos(r)/r;
- h[0] = -cos(r)/r;
- h[1] = -cos(r)/r/r - sin(r)/r;
- for (n = 2; n < n_max; n++) {
- j[n] = double(n + n + 1)*j[n - 1]/r - j[n - 2];
- h[n] = double(n + n + 1)*h[n - 1]/r - h[n - 2];
- }
- }
- void sphericalBessel(double r, int n_max, std::vector<double> &j, std::vector<double> &h) {
- int n;
- j[0] = sin(r)/r;
- j[1] = sin(r)/r/r - cos(r)/r;
- h[0] = -cos(r)/r;
- h[1] = -cos(r)/r/r - sin(r)/r;
- for (n = 2; n < n_max; n++) {
- j[n] = double(n + n + 1)*j[n - 1]/r - j[n - 2];
- h[n] = double(n + n + 1)*h[n - 1]/r - h[n - 2];
- }
- }
- std::complex<double> calc_an(int n, double XL, std::complex<double> Ha, std::complex<double> mL,
- std::complex<double> PsiXL, std::complex<double> ZetaXL,
- std::complex<double> PsiXLM1, std::complex<double> ZetaXLM1) {
- std::complex<double> Num = (Ha/mL + n/XL)*PsiXL - PsiXLM1;
- std::complex<double> Denom = (Ha/mL + n/XL)*ZetaXL - ZetaXLM1;
- return Num/Denom;
- }
- std::complex<double> calc_bn(int n, double XL, std::complex<double> Hb, std::complex<double> mL,
- std::complex<double> PsiXL, std::complex<double> ZetaXL,
- std::complex<double> PsiXLM1, std::complex<double> ZetaXLM1) {
- std::complex<double> Num = (mL*Hb + n/XL)*PsiXL - PsiXLM1;
- std::complex<double> Denom = (mL*Hb + n/XL)*ZetaXL - ZetaXLM1;
- return Num/Denom;
- }
- std::complex<double> calc_S1(int n, std::complex<double> an, std::complex<double> bn,
- double Pi, double Tau) {
- return double(n + n + 1)*(Pi*an + Tau*bn)/double(n*n + n);
- }
- std::complex<double> calc_S2(int n, std::complex<double> an, std::complex<double> bn,
- double Pi, double Tau) {
- return calc_S1(n, an, bn, Tau, Pi);
- }
- void calcPsiZeta(double x, int n_max,
- std::vector<std::complex<double> > D1,
- std::vector<std::complex<double> > D3,
- std::vector<std::complex<double> > &Psi,
- std::vector<std::complex<double> > &Zeta) {
- int n;
-
- Psi[0] = std::complex<double>(sin(x), 0);
- Zeta[0] = std::complex<double>(sin(x), -cos(x));
- for (n = 1; n <= n_max; n++) {
- Psi[n] = Psi[n - 1]*(n/x - D1[n - 1]);
- Zeta[n] = Zeta[n - 1]*(n/x - D3[n - 1]);
- }
- }
- void calcD1D3(std::complex<double> z, int n_max,
- std::vector<std::complex<double> > &D1,
- std::vector<std::complex<double> > &D3) {
- int n;
- std::vector<std::complex<double> > PsiZeta;
- PsiZeta.resize(n_max + 1);
-
- D1[n_max] = std::complex<double>(0.0, 0.0);
- for (n = n_max; n > 0; n--) {
- D1[n - 1] = double(n)/z - 1.0/(D1[n] + double(n)/z);
- }
-
- PsiZeta[0] = 0.5*(1.0 - std::complex<double>(cos(2.0*z.real()), sin(2.0*z.real()))*exp(-2.0*z.imag()));
- D3[0] = std::complex<double>(0.0, 1.0);
- for (n = 1; n <= n_max; n++) {
- PsiZeta[n] = PsiZeta[n - 1]*(double(n)/z - D1[n - 1])*(double(n)/z- D3[n - 1]);
- D3[n] = D1[n] + std::complex<double>(0.0, 1.0)/PsiZeta[n];
- }
- }
- void calcPiTau(int n_max, int nTheta, std::vector<double> Theta,
- std::vector< std::vector<double> > &Pi,
- std::vector< std::vector<double> > &Tau) {
- int n, t;
- for (n = 0; n < n_max; n++) {
-
-
-
- for (t = 0; t < nTheta; t++) {
- if (n == 0) {
-
- Pi[n][t] = 1.0;
- Tau[n][t] = (n + 1)*cos(Theta[t]);
- } else {
-
- Pi[n][t] = ((n == 1) ? ((n + n + 1)*cos(Theta[t])*Pi[n - 1][t]/n)
- : (((n + n + 1)*cos(Theta[t])*Pi[n - 1][t] - (n + 1)*Pi[n - 2][t])/n));
- Tau[n][t] = (n + 1)*cos(Theta[t])*Pi[n][t] - (n + 2)*Pi[n - 1][t];
- }
- }
- }
- }
- int ScattCoeffs(int L, int pl, std::vector<double> x, std::vector<std::complex<double> > m, int n_max,
- std::vector<std::complex<double> > &an, std::vector<std::complex<double> > &bn) {
-
-
-
-
-
- int fl = (pl > 0) ? pl : 0;
- if (n_max <= 0) {
- n_max = Nmax(L, fl, pl, x, m);
- }
- std::complex<double> z1, z2;
- std::complex<double> Num, Denom;
- std::complex<double> G1, G2;
- std::complex<double> Temp;
- int n, l;
-
-
-
-
-
-
-
-
- std::vector<std::vector<std::complex<double> > > D1_mlxl, D1_mlxlM1;
- D1_mlxl.resize(L);
- D1_mlxlM1.resize(L);
- std::vector<std::vector<std::complex<double> > > D3_mlxl, D3_mlxlM1;
- D3_mlxl.resize(L);
- D3_mlxlM1.resize(L);
- std::vector<std::vector<std::complex<double> > > Q;
- Q.resize(L);
- std::vector<std::vector<std::complex<double> > > Ha, Hb;
- Ha.resize(L);
- Hb.resize(L);
- for (l = 0; l < L; l++) {
- D1_mlxl[l].resize(n_max + 1);
- D1_mlxlM1[l].resize(n_max + 1);
- D3_mlxl[l].resize(n_max + 1);
- D3_mlxlM1[l].resize(n_max + 1);
- Q[l].resize(n_max + 1);
- Ha[l].resize(n_max);
- Hb[l].resize(n_max);
- }
- an.resize(n_max);
- bn.resize(n_max);
- std::vector<std::complex<double> > D1XL, D3XL;
- D1XL.resize(n_max + 1);
- D3XL.resize(n_max + 1);
- std::vector<std::complex<double> > PsiXL, ZetaXL;
- PsiXL.resize(n_max + 1);
- ZetaXL.resize(n_max + 1);
-
-
-
- if (fl == pl) {
- for (n = 0; n <= n_max; n++) {
- D1_mlxl[fl][n] = std::complex<double>(0.0, -1.0);
- D3_mlxl[fl][n] = std::complex<double>(0.0, 1.0);
- }
- } else {
- z1 = x[fl]* m[fl];
-
- calcD1D3(z1, n_max, D1_mlxl[fl], D3_mlxl[fl]);
- }
-
-
-
- for (n = 0; n < n_max; n++) {
- Ha[fl][n] = D1_mlxl[fl][n + 1];
- Hb[fl][n] = D1_mlxl[fl][n + 1];
- }
-
-
-
- for (l = fl + 1; l < L; l++) {
-
-
-
- z1 = x[l]*m[l];
- z2 = x[l - 1]*m[l];
-
- calcD1D3(z1, n_max, D1_mlxl[l], D3_mlxl[l]);
-
- calcD1D3(z2, n_max, D1_mlxlM1[l], D3_mlxlM1[l]);
-
-
-
-
- Num = exp(-2.0*(z1.imag() - z2.imag()))*std::complex<double>(cos(-2.0*z2.real()) - exp(-2.0*z2.imag()), sin(-2.0*z2.real()));
- Denom = std::complex<double>(cos(-2.0*z1.real()) - exp(-2.0*z1.imag()), sin(-2.0*z1.real()));
- Q[l][0] = Num/Denom;
- for (n = 1; n <= n_max; n++) {
- Num = (z1*D1_mlxl[l][n] + double(n))*(double(n) - z1*D3_mlxl[l][n - 1]);
- Denom = (z2*D1_mlxlM1[l][n] + double(n))*(double(n) - z2*D3_mlxlM1[l][n - 1]);
- Q[l][n] = (((x[l - 1]*x[l - 1])/(x[l]*x[l])* Q[l][n - 1])*Num)/Denom;
- }
-
- for (n = 1; n <= n_max; n++) {
-
- if ((l - 1) == pl) {
- G1 = -D1_mlxlM1[l][n];
- G2 = -D3_mlxlM1[l][n];
- } else {
- G1 = (m[l]*Ha[l - 1][n - 1]) - (m[l - 1]*D1_mlxlM1[l][n]);
- G2 = (m[l]*Ha[l - 1][n - 1]) - (m[l - 1]*D3_mlxlM1[l][n]);
- }
- Temp = Q[l][n]*G1;
- Num = (G2*D1_mlxl[l][n]) - (Temp*D3_mlxl[l][n]);
- Denom = G2 - Temp;
- Ha[l][n - 1] = Num/Denom;
-
- if ((l - 1) == pl) {
- G1 = Hb[l - 1][n - 1];
- G2 = Hb[l - 1][n - 1];
- } else {
- G1 = (m[l - 1]*Hb[l - 1][n - 1]) - (m[l]*D1_mlxlM1[l][n]);
- G2 = (m[l - 1]*Hb[l - 1][n - 1]) - (m[l]*D3_mlxlM1[l][n]);
- }
- Temp = Q[l][n]*G1;
- Num = (G2*D1_mlxl[l][n]) - (Temp* D3_mlxl[l][n]);
- Denom = (G2- Temp);
- Hb[l][n - 1] = (Num/ Denom);
- }
- }
-
-
-
-
- calcD1D3(x[L - 1], n_max, D1XL, D3XL);
-
- calcPsiZeta(x[L - 1], n_max, D1XL, D3XL, PsiXL, ZetaXL);
-
-
-
-
-
-
- for (n = 0; n < n_max; n++) {
-
-
-
-
- if (pl < (L - 1)) {
- 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]);
- 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]);
- } else {
- 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]);
- bn[n] = PsiXL[n + 1]/ZetaXL[n + 1];
- }
- }
- return n_max;
- }
- int nMie(int L, int pl, std::vector<double> x, std::vector<std::complex<double> > m,
- int nTheta, std::vector<double> Theta, int n_max,
- 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) {
- int i, n, t;
- std::vector<std::complex<double> > an, bn;
- std::complex<double> Qbktmp;
-
- n_max = ScattCoeffs(L, pl, x, m, n_max, an, bn);
- std::vector< std::vector<double> > Pi;
- Pi.resize(n_max);
- std::vector< std::vector<double> > Tau;
- Tau.resize(n_max);
- for (n = 0; n < n_max; n++) {
- Pi[n].resize(nTheta);
- Tau[n].resize(nTheta);
- }
- calcPiTau(n_max, nTheta, Theta, Pi, Tau);
- double x2 = x[L - 1]*x[L - 1];
-
- *Qext = 0;
- *Qsca = 0;
- *Qabs = 0;
- *Qbk = 0;
- Qbktmp = std::complex<double>(0.0, 0.0);
- *Qpr = 0;
- *g = 0;
- *Albedo = 0;
-
- for (t = 0; t < nTheta; t++) {
- S1[t] = std::complex<double>(0.0, 0.0);
- S2[t] = std::complex<double>(0.0, 0.0);
- }
-
-
-
- for (i = n_max - 2; i >= 0; i--) {
- n = i + 1;
-
- *Qext += (n + n + 1)*(an[i].real() + bn[i].real());
-
- *Qsca += (n + n + 1)*(an[i].real()*an[i].real() + an[i].imag()*an[i].imag() + bn[i].real()*bn[i].real() + bn[i].imag()*bn[i].imag());
-
-
- *Qpr += ((n*(n + 2)/(n + 1))*((an[i]*std::conj(an[n]) + bn[i]*std::conj(bn[n])).real()) + ((double)(n + n + 1)/(n*(n + 1)))*(an[i]*std::conj(bn[i])).real());
-
- Qbktmp = Qbktmp + (double)(n + n + 1)*(1 - 2*(n % 2))*(an[i]- bn[i]);
-
-
-
-
- for (t = 0; t < nTheta; t++) {
- S1[t] += calc_S1(n, an[i], bn[i], Pi[i][t], Tau[i][t]);
- S2[t] += calc_S2(n, an[i], bn[i], Pi[i][t], Tau[i][t]);
- }
- }
- *Qext = 2*(*Qext)/x2;
- *Qsca = 2*(*Qsca)/x2;
- *Qpr = *Qext - 4*(*Qpr)/x2;
- *Qabs = *Qext - *Qsca;
- *Albedo = *Qsca / *Qext;
- *g = (*Qext - *Qpr) / *Qsca;
- *Qbk = (Qbktmp.real()*Qbktmp.real() + Qbktmp.imag()*Qbktmp.imag())/x2;
- return n_max;
- }
- int nMie(int L, std::vector<double> x, std::vector<std::complex<double> > m,
- 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) {
- return nMie(L, -1, x, m, nTheta, Theta, -1, Qext, Qsca, Qabs, Qbk, Qpr, g, Albedo, S1, S2);
- }
- int nMie(int L, int pl, std::vector<double> x, std::vector<std::complex<double> > m,
- 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) {
- return nMie(L, pl, x, m, nTheta, Theta, -1, Qext, Qsca, Qabs, Qbk, Qpr, g, Albedo, S1, S2);
- }
- int nMie(int L, std::vector<double> x, std::vector<std::complex<double> > m,
- int nTheta, std::vector<double> Theta, int n_max,
- 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) {
- return nMie(L, -1, x, m, nTheta, Theta, n_max, Qext, Qsca, Qabs, Qbk, Qpr, g, Albedo, S1, S2);
- }
- int nField(int L, int pl, std::vector<double> x, std::vector<std::complex<double> > m, int n_max,
- int nCoords, std::vector<double> Xp, std::vector<double> Yp, std::vector<double> Zp,
- std::vector<std::complex<double> > &E, std::vector<std::complex<double> > &H) {
- int i, n, c;
-
-
-
-
-
-
- for (i = 1; i < (n_max - 1); i++) {
-
-
-
-
- }
- return n_max;
- }
|