Skip to content

implement lumped capacitor and resistor #141

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 5 commits into
base: development
Choose a base branch
from
Open

implement lumped capacitor and resistor #141

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
f78a8b4
implement lumped capacitor and resistor
jackieyao0114 Sep 3, 2025
4ca9117
merge new upstream
jackieyao0114 Sep 4, 2025
5f61553
fix lumped element definition
jackieyao0114 Sep 11, 2025
7a63dc2
Merge branch 'development' of https://github.com/AMReX-Microelectroni…
jackieyao0114 Sep 11, 2025
67a2354
fix parsing
jackieyao0114 Sep 14, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
137 changes: 119 additions & 18 deletions Source/FieldSolver/FiniteDifferenceSolver/MacroscopicEvolveE.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -133,8 +133,36 @@ void FiniteDifferenceSolver::MacroscopicEvolveECartesian (
amrex::ignore_unused(edge_lengths);
#endif

auto &warpx = WarpX::GetInstance();

const int lev = 0;
const amrex::GpuArray<amrex::Real, AMREX_SPACEDIM> dx = warpx.Geom(lev).CellSizeArray();

int use_lumped_resistor = warpx.use_lumped_resistor;
int use_lumped_capacitor = warpx.use_lumped_capacitor;

amrex::MultiFab& sigma_mf = macroscopic_properties->getsigma_mf();
amrex::MultiFab& epsilon_mf = macroscopic_properties->getepsilon_mf();

amrex::MultiFab* lumped_resistor_x_mf = nullptr;
amrex::MultiFab* lumped_resistor_y_mf = nullptr;
amrex::MultiFab* lumped_resistor_z_mf = nullptr;
amrex::MultiFab* lumped_capacitor_x_mf = nullptr;
amrex::MultiFab* lumped_capacitor_y_mf = nullptr;
amrex::MultiFab* lumped_capacitor_z_mf = nullptr;

if (use_lumped_resistor){
lumped_resistor_x_mf = &macroscopic_properties->getlumped_resistor_x_mf();
lumped_resistor_y_mf = &macroscopic_properties->getlumped_resistor_y_mf();
lumped_resistor_z_mf = &macroscopic_properties->getlumped_resistor_z_mf();
}

if (use_lumped_capacitor){
lumped_capacitor_x_mf = &macroscopic_properties->getlumped_capacitor_x_mf();
lumped_capacitor_y_mf = &macroscopic_properties->getlumped_capacitor_y_mf();
lumped_capacitor_z_mf = &macroscopic_properties->getlumped_capacitor_z_mf();
}

#ifndef WARPX_MAG_LLG
amrex::MultiFab& mu_mf = macroscopic_properties->getmu_mf();
#endif
Expand All @@ -144,6 +172,19 @@ void FiniteDifferenceSolver::MacroscopicEvolveECartesian (
amrex::GpuArray<int, 3> const& sigma_stag = macroscopic_properties->sigma_IndexType;
amrex::GpuArray<int, 3> const& epsilon_stag = macroscopic_properties->epsilon_IndexType;
amrex::GpuArray<int, 3> const& macro_cr = macroscopic_properties->macro_cr_ratio;

if (use_lumped_resistor){
amrex::GpuArray<int, 3> const& lumped_resistor_x_stag = macroscopic_properties->lumped_resistor_x_IndexType;
amrex::GpuArray<int, 3> const& lumped_resistor_y_stag = macroscopic_properties->lumped_resistor_y_IndexType;
amrex::GpuArray<int, 3> const& lumped_resistor_z_stag = macroscopic_properties->lumped_resistor_z_IndexType;
}

if (use_lumped_capacitor){
amrex::GpuArray<int, 3> const& lumped_capacitor_x_stag = macroscopic_properties->lumped_capacitor_x_IndexType;
amrex::GpuArray<int, 3> const& lumped_capacitor_y_stag = macroscopic_properties->lumped_capacitor_y_IndexType;
amrex::GpuArray<int, 3> const& lumped_capacitor_z_stag = macroscopic_properties->lumped_capacitor_z_IndexType;
}

amrex::GpuArray<int, 3> const& Ex_stag = macroscopic_properties->Ex_IndexType;
amrex::GpuArray<int, 3> const& Ey_stag = macroscopic_properties->Ey_IndexType;
amrex::GpuArray<int, 3> const& Ez_stag = macroscopic_properties->Ez_IndexType;
Expand Down Expand Up @@ -176,6 +217,24 @@ void FiniteDifferenceSolver::MacroscopicEvolveECartesian (
// material prop //
amrex::Array4<amrex::Real> const& sigma_arr = sigma_mf.array(mfi);
amrex::Array4<amrex::Real> const& eps_arr = epsilon_mf.array(mfi);

// Lumped element arrays (default-constructed; set only if feature enabled)
amrex::Array4<amrex::Real> resistor_x_arr, resistor_y_arr, resistor_z_arr;
amrex::Array4<amrex::Real> capacitor_x_arr, capacitor_y_arr, capacitor_z_arr;

if (use_lumped_resistor) {
// These produce lightweight views; safe to copy and capture by value
resistor_x_arr = lumped_resistor_x_mf->array(mfi);
resistor_y_arr = lumped_resistor_y_mf->array(mfi);
resistor_z_arr = lumped_resistor_z_mf->array(mfi);
}
if (use_lumped_capacitor) {
capacitor_x_arr = lumped_capacitor_x_mf->array(mfi);
capacitor_y_arr = lumped_capacitor_y_mf->array(mfi);
capacitor_z_arr = lumped_capacitor_z_mf->array(mfi);
}


#ifndef WARPX_MAG_LLG
amrex::Array4<amrex::Real> const& mu_arr = mu_mf.array(mfi);
#endif
Expand Down Expand Up @@ -220,12 +279,25 @@ void FiniteDifferenceSolver::MacroscopicEvolveECartesian (
// Interpolated permittivity, epsilon, to Ex position on the grid
amrex::Real const epsilon_interp = ablastr::coarsen::sample::Interp(eps_arr, epsilon_stag,
Ex_stag, macro_cr, i, j, k, scomp);
amrex::Real alpha = T_MacroAlgo::alpha( sigma_interp, epsilon_interp, dt);
amrex::Real beta = T_MacroAlgo::beta( sigma_interp, epsilon_interp, dt);
Ex(i, j, k) = alpha * Ex(i, j, k)
+ beta * ( - T_Algo::DownwardDz(Hy, coefs_z, n_coefs_z, i, j, k,0)
+ T_Algo::DownwardDy(Hz, coefs_y, n_coefs_y, i, j, k,0)
) - beta * jx(i, j, k);

// Extra conductance term from a lumped resistor on Ex edges
amrex::Real const fac1 = (dt/epsilon_interp) *
( sigma_interp + ((use_lumped_resistor && resistor_x_arr(i,j,k)!=0._rt)
? dx[0] / (dx[1]*dx[2]*resistor_x_arr(i,j,k))
: 0._rt) );

// Extra term from a lumped capacitor on Ex edges
amrex::Real const fac2 = (use_lumped_capacitor && capacitor_x_arr(i,j,k)!=0._rt)
? capacitor_x_arr(i,j,k) * dx[0] / (dx[1]*dx[2]*epsilon_interp)
: 0._rt;

amrex::Real alpha_compact = T_MacroAlgo::alpha_compact(fac1, fac2);
amrex::Real beta_compact = T_MacroAlgo::beta_compact(fac1, fac2);

Ex(i, j, k) = alpha_compact * Ex(i, j, k)
+ dt/epsilon_interp * beta_compact * ( - T_Algo::DownwardDz(Hy, coefs_z, n_coefs_z, i, j, k,0)
+ T_Algo::DownwardDy(Hz, coefs_y, n_coefs_y, i, j, k,0))
- dt/epsilon_interp * beta_compact * jx(i, j, k);
},

[=] AMREX_GPU_DEVICE (int i, int j, int k){
Expand All @@ -244,12 +316,28 @@ void FiniteDifferenceSolver::MacroscopicEvolveECartesian (
// Interpolated permittivity, epsilon, to Ey position on the grid
amrex::Real const epsilon_interp = ablastr::coarsen::sample::Interp(eps_arr, epsilon_stag,
Ey_stag, macro_cr, i, j, k, scomp);
amrex::Real alpha = T_MacroAlgo::alpha( sigma_interp, epsilon_interp, dt);
amrex::Real beta = T_MacroAlgo::beta( sigma_interp, epsilon_interp, dt);
Ey(i, j, k) = alpha * Ey(i, j, k)
+ beta * ( - T_Algo::DownwardDx(Hz, coefs_x, n_coefs_x, i, j, k,0)
+ T_Algo::DownwardDz(Hx, coefs_z, n_coefs_z, i, j, k,0)
) - beta * jy(i, j, k);

// Extra conductance term from a lumped resistor on Ey edges
amrex::Real const fac1 = (dt/epsilon_interp) *
( sigma_interp + ((use_lumped_resistor && resistor_y_arr(i,j,k)!=0._rt)
? dx[1] / (dx[0]*dx[2]*resistor_y_arr(i,j,k))
: 0._rt) );
// if (use_lumped_resistor && resistor_y_arr(i,j,k)!=0._rt){
// printf("fac1=%f\n", fac1);
// }

// Extra term from a lumped capacitor on Ey edges
amrex::Real const fac2 = (use_lumped_capacitor && capacitor_y_arr(i,j,k)!=0._rt)
? capacitor_y_arr(i,j,k) * dx[1] / (dx[0]*dx[2]*epsilon_interp)
: 0._rt;

amrex::Real alpha_compact = T_MacroAlgo::alpha_compact(fac1, fac2);
amrex::Real beta_compact = T_MacroAlgo::beta_compact(fac1, fac2);

Ey(i, j, k) = alpha_compact * Ey(i, j, k)
+ dt/epsilon_interp * beta_compact * ( - T_Algo::DownwardDx(Hz, coefs_x, n_coefs_x, i, j, k,0)
+ T_Algo::DownwardDz(Hx, coefs_z, n_coefs_z, i, j, k,0))
- dt/epsilon_interp * beta_compact * jy(i, j, k);
},

[=] AMREX_GPU_DEVICE (int i, int j, int k){
Expand All @@ -263,12 +351,25 @@ void FiniteDifferenceSolver::MacroscopicEvolveECartesian (
// Interpolated permittivity, epsilon, to Ez position on the grid
amrex::Real const epsilon_interp = ablastr::coarsen::sample::Interp(eps_arr, epsilon_stag,
Ez_stag, macro_cr, i, j, k, scomp);
amrex::Real alpha = T_MacroAlgo::alpha( sigma_interp, epsilon_interp, dt);
amrex::Real beta = T_MacroAlgo::beta( sigma_interp, epsilon_interp, dt);
Ez(i, j, k) = alpha * Ez(i, j, k)
+ beta * ( - T_Algo::DownwardDy(Hx, coefs_y, n_coefs_y, i, j, k,0)
+ T_Algo::DownwardDx(Hy, coefs_x, n_coefs_x, i, j, k,0)
) - beta * jz(i, j, k);

// Extra conductance term from a lumped resistor on Ez edges
amrex::Real const fac1 = (dt/epsilon_interp) *
( sigma_interp + ((use_lumped_resistor && resistor_z_arr(i,j,k)!=0._rt)
? dx[2] / (dx[0]*dx[1]*resistor_z_arr(i,j,k))
: 0._rt) );

// Extra term from a lumped capacitor on Ez edges
amrex::Real const fac2 = (use_lumped_capacitor && capacitor_z_arr(i,j,k)!=0._rt)
? capacitor_z_arr(i,j,k) * dx[2] / (dx[0]*dx[1]*epsilon_interp)
: 0._rt;

amrex::Real alpha_compact = T_MacroAlgo::alpha_compact(fac1, fac2);
amrex::Real beta_compact = T_MacroAlgo::beta_compact(fac1, fac2);

Ez(i, j, k) = alpha_compact * Ez(i, j, k)
+ dt/epsilon_interp * beta_compact * ( - T_Algo::DownwardDy(Hx, coefs_y, n_coefs_y, i, j, k,0)
+ T_Algo::DownwardDx(Hy, coefs_x, n_coefs_x, i, j, k,0))
- dt/epsilon_interp * beta_compact * jz(i, j, k);
}
);
}
Expand Down
93 changes: 91 additions & 2 deletions Source/FieldSolver/FiniteDifferenceSolver/MacroscopicProperties/MacroscopicProperties.H
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,20 @@ public:
amrex::MultiFab& getmu_mf () {return (*m_mu_mf);}
amrex::MultiFab * get_pointer_mu () {return m_mu_mf.get();}

amrex::MultiFab& getlumped_resistor_x_mf () {return (*m_lumped_resistor_x_mf);}
amrex::MultiFab * get_pointer_lumped_resistor_x () {return m_lumped_resistor_x_mf.get();}
amrex::MultiFab& getlumped_resistor_y_mf () {return (*m_lumped_resistor_y_mf);}
amrex::MultiFab * get_pointer_lumped_resistor_y () {return m_lumped_resistor_y_mf.get();}
amrex::MultiFab& getlumped_resistor_z_mf () {return (*m_lumped_resistor_z_mf);}
amrex::MultiFab * get_pointer_lumped_resistor_z () {return m_lumped_resistor_z_mf.get();}

amrex::MultiFab& getlumped_capacitor_x_mf () {return (*m_lumped_capacitor_x_mf);}
amrex::MultiFab * get_pointer_lumped_capacitor_x () {return m_lumped_capacitor_x_mf.get();}
amrex::MultiFab& getlumped_capacitor_y_mf () {return (*m_lumped_capacitor_y_mf);}
amrex::MultiFab * get_pointer_lumped_capacitor_y () {return m_lumped_capacitor_y_mf.get();}
amrex::MultiFab& getlumped_capacitor_z_mf () {return (*m_lumped_capacitor_z_mf);}
amrex::MultiFab * get_pointer_lumped_capacitor_z () {return m_lumped_capacitor_z_mf.get();}

/** Gpu Vector with index type of coarsening ratio with default value (1,1,1) */
amrex::GpuArray<int, 3> macro_cr_ratio;
/** Initializes the Multifabs storing macroscopic properties
Expand All @@ -73,6 +87,17 @@ public:
amrex::GpuArray<int, 3> epsilon_IndexType;
/** Gpu Vector with index type of the permeability multifab */
amrex::GpuArray<int, 3> mu_IndexType;

/** Gpu Vector with index type of the lumped resistor multifab */
amrex::GpuArray<int, 3> lumped_resistor_x_IndexType;
amrex::GpuArray<int, 3> lumped_resistor_y_IndexType;
amrex::GpuArray<int, 3> lumped_resistor_z_IndexType;

/** Gpu Vector with index type of the lumped capacitor multifab */
amrex::GpuArray<int, 3> lumped_capacitor_x_IndexType;
amrex::GpuArray<int, 3> lumped_capacitor_y_IndexType;
amrex::GpuArray<int, 3> lumped_capacitor_z_IndexType;

/** Gpu Vector with index type of the Ex multifab */
amrex::GpuArray<int, 3> Ex_IndexType;
/** Gpu Vector with index type of the Ey multifab */
Expand All @@ -85,6 +110,12 @@ public:
amrex::GpuArray<int, 3> By_IndexType;
/** Gpu Vector with index type of the Bz multifab */
amrex::GpuArray<int, 3> Bz_IndexType;
/** Gpu Vector with index type of the jx multifab */
amrex::GpuArray<int, 3> jx_IndexType;
/** Gpu Vector with index type of the jy multifab */
amrex::GpuArray<int, 3> jy_IndexType;
/** Gpu Vector with index type of the jz multifab */
amrex::GpuArray<int, 3> jz_IndexType;

/** Stores initialization type for conductivity : constant or parser */
std::string m_sigma_s = "constant";
Expand Down Expand Up @@ -117,6 +148,15 @@ public:
std::unique_ptr<amrex::Parser> m_epsilon_parser;
std::unique_ptr<amrex::Parser> m_mu_parser;

std::unique_ptr<amrex::Parser> m_lumped_resistor_x_parser;
std::unique_ptr<amrex::Parser> m_lumped_resistor_y_parser;
std::unique_ptr<amrex::Parser> m_lumped_resistor_z_parser;

std::unique_ptr<amrex::Parser> m_lumped_capacitor_x_parser;
std::unique_ptr<amrex::Parser> m_lumped_capacitor_y_parser;
std::unique_ptr<amrex::Parser> m_lumped_capacitor_z_parser;


#ifdef WARPX_MAG_LLG
/** Gpu Vector with index type of the Hx multifab */
amrex::GpuArray<int, 3> Hx_IndexType;
Expand Down Expand Up @@ -324,7 +364,24 @@ private:
/** Multifab for m_mu */
std::unique_ptr<amrex::MultiFab> m_mu_mf;


/** Multifab for lumped resistor */
std::unique_ptr<amrex::MultiFab> m_lumped_resistor_x_mf;
std::unique_ptr<amrex::MultiFab> m_lumped_resistor_y_mf;
std::unique_ptr<amrex::MultiFab> m_lumped_resistor_z_mf;
/** string for storing parser function */
std::string m_str_lumped_resistor_x_function;
std::string m_str_lumped_resistor_y_function;
std::string m_str_lumped_resistor_z_function;

/** Multifab for lumped capacitor */
std::unique_ptr<amrex::MultiFab> m_lumped_capacitor_x_mf;
std::unique_ptr<amrex::MultiFab> m_lumped_capacitor_y_mf;
std::unique_ptr<amrex::MultiFab> m_lumped_capacitor_z_mf;
/** string for storing parser function */
std::string m_str_lumped_capacitor_x_function;
std::string m_str_lumped_capacitor_y_function;
std::string m_str_lumped_capacitor_z_function;

/** string for storing parser function */
std::string m_str_sigma_function;
std::string m_str_epsilon_function;
Expand Down Expand Up @@ -362,6 +419,22 @@ struct LaxWendroffAlgo {
return beta;
}

AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
static amrex::Real alpha_compact (amrex::Real const fac1,
amrex::Real const fac2) {
using namespace amrex;
amrex::Real alpha_compact = (1._rt - 0.5_rt * fac1 + fac2)/(1._rt + 0.5_rt * fac1 + fac2);
return alpha_compact;
}

AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
static amrex::Real beta_compact (amrex::Real const fac1,
amrex::Real const fac2) {
using namespace amrex;
amrex::Real beta_compact = 1._rt / (1._rt + 0.5_rt * fac1 + fac2);
return beta_compact;
}

};

/**
Expand Down Expand Up @@ -394,6 +467,22 @@ struct BackwardEulerAlgo {
return beta;
}

AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
static amrex::Real alpha_compact (amrex::Real const fac1,
amrex::Real const fac2) {
using namespace amrex;
amrex::Real alpha_compact = (1._rt + fac2)/(1._rt + fac1 + fac2);
return alpha_compact;
}

AMREX_GPU_HOST_DEVICE AMREX_FORCE_INLINE
static amrex::Real beta_compact (amrex::Real const fac1,
amrex::Real const fac2) {
using namespace amrex;
amrex::Real beta_compact = 1._rt / (1._rt + fac1 + fac2);
return beta_compact;
}

};

#endif // WARPX_MACROSCOPIC_PROPERTIES_H_
#endif // WARPX_MACROSCOPIC_PROPERTIES_H_
Loading
Loading