step_d.hpp

来自「麻省理工的计算光子晶体的程序」· HPP 代码 · 共 360 行 · 第 1/2 页

/* This file was automatically generated --- DO NOT EDIT */

if (have_m) {
  if (have_p) {
    if (have_m_pml) {
      if (have_p_pml) {
        if (n3==1) {
          if (s2==1) {
            for (int i1=0; i1<n1; i1++)
              for (int i2=0, ind=i1*s1; i2<n2; i2++,ind+=1) {
                const double em = the_f_m_pml[ind];
                const double ep = the_f_p_pml[ind];
                const double m_deriv_m = f_m[ind]-f_m[ind+stride_m];
                const double deriv_p = f_p[ind+stride_p]-f_p[ind];
                const double m_change = decay_m[ind]*(Courant*m_deriv_m - C_m[ind]*em);
                const double p_change = decay_p[ind]*(Courant*deriv_p - C_p[ind]*ep);
                the_f_p_pml[ind] += p_change;
                the_f_m_pml[ind] += m_change;
                the_f[ind] += m_change + p_change;
              }
          } else { // not s2==1
            for (int i1=0; i1<n1; i1++)
              for (int i2=0, ind=i1*s1; i2<n2; i2++,ind+=s2) {
                const double em = the_f_m_pml[ind];
                const double ep = the_f_p_pml[ind];
                const double m_deriv_m = f_m[ind]-f_m[ind+stride_m];
                const double deriv_p = f_p[ind+stride_p]-f_p[ind];
                const double m_change = decay_m[ind]*(Courant*m_deriv_m - C_m[ind]*em);
                const double p_change = decay_p[ind]*(Courant*deriv_p - C_p[ind]*ep);
                the_f_p_pml[ind] += p_change;
                the_f_m_pml[ind] += m_change;
                the_f[ind] += m_change + p_change;
              }
          }
        } else { // not n3==1
          if (s3==1) {
            for (int i1=0; i1<n1; i1++)
              for (int i2=0; i2<n2; i2++)
                for (int i3=0, ind=i1*s1+i2*s2; i3<n3; i3++, ind+=1) {
                  const double em = the_f_m_pml[ind];
                  const double ep = the_f_p_pml[ind];
                  const double m_deriv_m = f_m[ind]-f_m[ind+stride_m];
                  const double deriv_p = f_p[ind+stride_p]-f_p[ind];
                  const double m_change = decay_m[ind]*(Courant*m_deriv_m - C_m[ind]*em);
                  const double p_change = decay_p[ind]*(Courant*deriv_p - C_p[ind]*ep);
                  the_f_p_pml[ind] += p_change;
                  the_f_m_pml[ind] += m_change;
                  the_f[ind] += m_change + p_change;
                }
          } else { // not s3==1
            for (int i1=0; i1<n1; i1++)
              for (int i2=0; i2<n2; i2++)
                for (int i3=0, ind=i1*s1+i2*s2; i3<n3; i3++, ind+=s3) {
                  const double em = the_f_m_pml[ind];
                  const double ep = the_f_p_pml[ind];
                  const double m_deriv_m = f_m[ind]-f_m[ind+stride_m];
                  const double deriv_p = f_p[ind+stride_p]-f_p[ind];
                  const double m_change = decay_m[ind]*(Courant*m_deriv_m - C_m[ind]*em);
                  const double p_change = decay_p[ind]*(Courant*deriv_p - C_p[ind]*ep);
                  the_f_p_pml[ind] += p_change;
                  the_f_m_pml[ind] += m_change;
                  the_f[ind] += m_change + p_change;
                }
          }
        }
      } else { // not have_p_pml
        if (n3==1) {
          if (s2==1) {
            for (int i1=0; i1<n1; i1++)
              for (int i2=0, ind=i1*s1; i2<n2; i2++,ind+=1) {
                const double em = the_f_m_pml[ind];
                const double m_deriv_m = f_m[ind]-f_m[ind+stride_m];
                const double deriv_p = f_p[ind+stride_p]-f_p[ind];
                const double m_change = decay_m[ind]*(Courant*m_deriv_m - C_m[ind]*em);
                const double p_change = Courant*deriv_p;
                the_f_m_pml[ind] += m_change;
                the_f[ind] += m_change + p_change;
              }
          } else { // not s2==1
            for (int i1=0; i1<n1; i1++)
              for (int i2=0, ind=i1*s1; i2<n2; i2++,ind+=s2) {
                const double em = the_f_m_pml[ind];
                const double m_deriv_m = f_m[ind]-f_m[ind+stride_m];
                const double deriv_p = f_p[ind+stride_p]-f_p[ind];
                const double m_change = decay_m[ind]*(Courant*m_deriv_m - C_m[ind]*em);
                const double p_change = Courant*deriv_p;
                the_f_m_pml[ind] += m_change;
                the_f[ind] += m_change + p_change;
              }
          }
        } else { // not n3==1
          if (s3==1) {
            for (int i1=0; i1<n1; i1++)
              for (int i2=0; i2<n2; i2++)
                for (int i3=0, ind=i1*s1+i2*s2; i3<n3; i3++, ind+=1) {
                  const double em = the_f_m_pml[ind];
                  const double m_deriv_m = f_m[ind]-f_m[ind+stride_m];
                  const double deriv_p = f_p[ind+stride_p]-f_p[ind];
                  const double m_change = decay_m[ind]*(Courant*m_deriv_m - C_m[ind]*em);
                  const double p_change = Courant*deriv_p;
                  the_f_m_pml[ind] += m_change;
                  the_f[ind] += m_change + p_change;
                }
          } else { // not s3==1
            for (int i1=0; i1<n1; i1++)
              for (int i2=0; i2<n2; i2++)
                for (int i3=0, ind=i1*s1+i2*s2; i3<n3; i3++, ind+=s3) {
                  const double em = the_f_m_pml[ind];
                  const double m_deriv_m = f_m[ind]-f_m[ind+stride_m];
                  const double deriv_p = f_p[ind+stride_p]-f_p[ind];
                  const double m_change = decay_m[ind]*(Courant*m_deriv_m - C_m[ind]*em);
                  const double p_change = Courant*deriv_p;
                  the_f_m_pml[ind] += m_change;
                  the_f[ind] += m_change + p_change;
                }
          }
        }
      }
    } else { // not have_m_pml
      if (have_p_pml) {
        if (n3==1) {
          if (s2==1) {
            for (int i1=0; i1<n1; i1++)
              for (int i2=0, ind=i1*s1; i2<n2; i2++,ind+=1) {
                const double ep = the_f_p_pml[ind];
                const double m_deriv_m = f_m[ind]-f_m[ind+stride_m];
                const double deriv_p = f_p[ind+stride_p]-f_p[ind];
                const double m_change = Courant*m_deriv_m;
                const double p_change = decay_p[ind]*(Courant*deriv_p - C_p[ind]*ep);
                the_f_p_pml[ind] += p_change;
                the_f[ind] += m_change + p_change;
              }
          } else { // not s2==1
            for (int i1=0; i1<n1; i1++)
              for (int i2=0, ind=i1*s1; i2<n2; i2++,ind+=s2) {
                const double ep = the_f_p_pml[ind];
                const double m_deriv_m = f_m[ind]-f_m[ind+stride_m];
                const double deriv_p = f_p[ind+stride_p]-f_p[ind];
                const double m_change = Courant*m_deriv_m;
                const double p_change = decay_p[ind]*(Courant*deriv_p - C_p[ind]*ep);
                the_f_p_pml[ind] += p_change;
                the_f[ind] += m_change + p_change;
              }
          }
        } else { // not n3==1
          if (s3==1) {
            for (int i1=0; i1<n1; i1++)
              for (int i2=0; i2<n2; i2++)
                for (int i3=0, ind=i1*s1+i2*s2; i3<n3; i3++, ind+=1) {
                  const double ep = the_f_p_pml[ind];
                  const double m_deriv_m = f_m[ind]-f_m[ind+stride_m];
                  const double deriv_p = f_p[ind+stride_p]-f_p[ind];
                  const double m_change = Courant*m_deriv_m;
                  const double p_change = decay_p[ind]*(Courant*deriv_p - C_p[ind]*ep);
                  the_f_p_pml[ind] += p_change;
                  the_f[ind] += m_change + p_change;
                }
          } else { // not s3==1
            for (int i1=0; i1<n1; i1++)
              for (int i2=0; i2<n2; i2++)
                for (int i3=0, ind=i1*s1+i2*s2; i3<n3; i3++, ind+=s3) {
                  const double ep = the_f_p_pml[ind];
                  const double m_deriv_m = f_m[ind]-f_m[ind+stride_m];
                  const double deriv_p = f_p[ind+stride_p]-f_p[ind];
                  const double m_change = Courant*m_deriv_m;
                  const double p_change = decay_p[ind]*(Courant*deriv_p - C_p[ind]*ep);
                  the_f_p_pml[ind] += p_change;
                  the_f[ind] += m_change + p_change;
                }
          }
        }
      } else { // not have_p_pml
        if (n3==1) {
          if (s2==1) {
            for (int i1=0; i1<n1; i1++)
              for (int i2=0, ind=i1*s1; i2<n2; i2++,ind+=1) {
                const double m_deriv_m = f_m[ind]-f_m[ind+stride_m];
                const double deriv_p = f_p[ind+stride_p]-f_p[ind];
                the_f[ind] += Courant*(m_deriv_m + deriv_p);
              }