34 void initialize_sv_dumps_onFace(
limpet::MULTI_IF *pmiif, IMPregion_EMI* reg,
int id,
double t,
double dump_dt);
38 bool imp_region_emi_is_assigned(
const IMPregion_EMI& region,
int idx)
40 if (!(region.im && strlen(region.im) > 0))
return false;
41 if (idx < 2)
return true;
42 return region.num_IDs > 0;
45 int effective_num_imp_regions_emi()
47 const int configured_nreg = param_globals::num_imp_regions;
48 int nreg = configured_nreg;
49 while (nreg > 2 && !imp_region_emi_is_assigned(param_globals::imp_region_emi[nreg - 1], nreg - 1)) {
53 if (nreg < configured_nreg) {
54 static bool warned_once =
false;
56 MPI_Comm_rank(MPI_COMM_WORLD, &rank);
57 if (!warned_once && rank == 0) {
59 "Warning: num_imp_regions=%d but only %d imp_region_emi entries are assigned; ignoring trailing unassigned regions.\n",
60 configured_nreg, nreg);
70 void IonicsOnFace::compute_step()
75 miif->compute_ionic_current();
77 double comp_time =
timing(t2, t1);
78 this->compute_time += comp_time;
81 comp_stats.tot_time += comp_time;
87 void IonicsOnFace::destroy()
92 void IonicsOnFace::output_step()
95 void IonicsOnFace::initialize()
103 comp_stats.init_logger(
"ODE_stats.dat");
107 int loc_size = mesh.l_numelem;
120 miif->name =
"myocardium";
121 miif->gdata[limpet::Vm] = Vmv;
122 miif->gdata[limpet::Iion] = IIon;
124 const int num_imp_regions = effective_num_imp_regions_emi();
128 log_msg(NULL, 5,
ECHO,
"\tError: num_imp_regions must be at least 2: the first region is reserved for the ionic model, and the second is designated for the gap junction.\n");
129 log_msg(NULL, 5,
ECHO,
"set a default ionic model and gapjuntion model in parameter file\\n");
137 rs[0].subregtags =
nullptr;
139 rs[1].subregtags =
nullptr;
142 for (
size_t i = 2; i < rs.size(); i++ ) {
143 rs[i].nsubregs = param_globals::imp_region_emi[i].num_IDs;
144 rs[i].subregtags =
new std::string[rs[i].nsubregs];
145 for (
int j = 0; j < rs[i].nsubregs;j++) {
146 std::string t1t2 = param_globals::imp_region_emi[i].ID[j];
147 size_t colon_pos = t1t2.find(
':');
148 if (colon_pos == std::string::npos) {
149 std::cerr <<
"Error: ':' the face tags are not defined properly in input file, it should be tag1:tag2 as a string" << std::endl;
153 rs[i].subregtags[j] = param_globals::imp_region_emi[i].ID[j];
158 region_mask_onFace(ion_domain, tags_data, line_face, tri_face, quad_face,
159 map_vertex_tag_to_dof, map_elem_uniqueFace_to_tags, intra_tags,
160 rs, reg_mask,
true,
"imp_region_emi");
162 for (
size_t i = 0; i < rs.size(); i++ ) {
163 delete[] rs[i].subregtags;
164 rs[i].subregtags =
nullptr;
168 tstart = setup_MIIF(loc_size, num_imp_regions, param_globals::imp_region_emi,
169 reg_mask.
data(), param_globals::start_statef, param_globals::num_adjustments,
170 param_globals::adjustment, param_globals::dt, purkfLen > 0);
172 miif->extUpdateVm = !param_globals::operator_splitting;
177 log_msg(logger, 0, 0,
"Changing simulation start time to %.2lf", tstart);
184 this->initialize_time +=
timing(t2, t1);
187 double IonicsOnFace::setup_MIIF(
int nnodes,
int nreg, IMPregion_EMI* impreg,
int* mask,
188 const char *start_fn,
int numadjust, IMPVariableAdjustment *adjust,
189 double time_step,
bool close)
194 miif->numNode = nnodes;
196 miif->numplugs = (
int*)calloc( miif->N_IIF,
sizeof(
int));
197 miif->plugtypes = std::vector<limpet::IonTypeList>(miif->N_IIF);
200 log_msg(logger,0,
ECHO,
"\nSetting up ionic models on EMI Face and plugins\n" \
201 "-----------------------------------\n\n" \
202 "Assigning IMPS to tagged regions:" );
204 for (
int i=0;i<miif->N_IIF;i++) {
208 miif->iontypes.push_back(*pT);
209 log_msg(logger, 0,
ECHO|
NONL,
"\tIonic model: %s to tag region(s)", impreg[i].im);
211 if(impreg[i].num_IDs > 0) {
212 for(
int j = 0; j < impreg[i].num_IDs; j++)
221 log_msg(NULL,5,
ECHO,
"Illegal IM specified: %s\n", impreg[i].im );
222 log_msg(NULL,5,
ECHO,
"Run bench --list-imps for a list of all available models.\n" );
227 if(impreg[i].plugins[0] !=
'\0') {
228 log_msg(logger,0,
ECHO|
NONL,
"\tPlug-in(s) : %s to tag region(s)", impreg[i].plugins);
230 for(
int j = 0; j < impreg[i].num_IDs; j++)
236 log_msg(NULL,5,
ECHO,
"Illegal plugin specified: %s\n", impreg[i].plugins);
237 log_msg(NULL,5,
ECHO,
"Run bench --list-imps for a list of all available plugins.\n" );
248 for (
int i=0; i<miif->numNode; i++)
252 miif->initialize_MIIF();
254 for (
int i=0;i<miif->N_IIF;i++) {
256 remove_char(impreg[i].im_param, strlen(impreg[i].im_param),
' ');
257 miif->IIF[i]->tune(impreg[i].im_param, impreg[i].plugins, impreg[i].plug_param);
261 miif->initialize_currents(time_step, param_globals::ode_fac);
266 for (
int i=0;i<miif->N_IIF;i++) {
267 if (impreg[i].im_sv_init && strlen(impreg[i].im_sv_init) > 0)
268 if (
read_sv(miif, i, impreg[i].im_sv_init)) {
269 log_msg(NULL, 5,
ECHO|
FLUSH,
"State vector initialization failed for %s.\n", impreg[i].name);
274 if( !start_fn || strlen(start_fn)>0 )
275 tstart = (double) miif->restore_state(start_fn, ion_domain, close);
277 for (
int i=0; i<numadjust; i++)
283 bool restrict_to_algebraic =
true;
286 std::map<std::string,std::string> metadata;
290 if(metadata.count(
"grid") && metadata[
"grid"].compare(
"intra") == 0) {
293 read_indices_with_data(indices, values, adjust[i].file, imesh, nbr, restrict_to_algebraic, 1, PETSC_COMM_WORLD);
297 for(
size_t gi = 0; gi < indices.
size(); gi++)
304 adjPars->set(indices, values);
308 snprintf(fname,
sizeof fname,
"adj_%s_perm.dat", adjust[i].variable);
309 adjPars->write_ascii(fname,
false);
314 log_msg(0,3,0,
"%s warning: PETSC_TO_CANONICAL permutation needed registering!", __func__);
318 (*sc)(*adjPars,
true);
319 snprintf(fname,
sizeof fname,
"adj_%s_canonical.dat", adjust[i].variable);
320 adjPars->write_ascii(fname,
false);
323 int nc = miif->adjust_MIIF_variables(adjust[i].variable, indices, values);
324 log_msg(logger, 0, 0,
"Adjusted %d values for %s", nc, adjust[i].variable);
329 for (
int i=0;i<miif->N_IIF;i++)
330 initialize_sv_dumps_onFace(miif, impreg+i, i, tstart, param_globals::spacedt);
344 void initialize_sv_dumps_onFace(
limpet::MULTI_IF *pmiif, IMPregion_EMI* reg,
int id,
double t,
double dump_dt)
346 char svs[1024], plgs[1024], plgsvs[1024], fname[1024];
348 strcpy(svs, reg->im_sv_dumps ? reg->im_sv_dumps :
"");
349 strcpy(plgs, reg->plugins ? reg->plugins :
"");
350 strcpy(plgsvs, reg->plug_sv_dumps ? reg->plug_sv_dumps :
"");
352 if( !(strlen(svs)+strlen(plgsvs) ) )
361 strcpy(fname, param_globals::vofile);
364 log_msg(NULL, 5,
ECHO,
"%s: a region name must be specified\n", __func__ );
369 char* ext_start = NULL;
370 ext_start = strstr(fname,
"igb.gz");
371 if(ext_start == NULL) ext_start = strstr(fname,
"igb");
372 if(ext_start == NULL) ext_start = fname + strlen(fname);
373 strcpy(ext_start, reg->name);
389 const char* gridname,
const char* reglist)
391 bool AllTagsExist =
true;
394 tagset.
insert(tags_data.begin(), tags_data.end());
397 for (
size_t reg=0; reg<regspec.
size(); reg++)
399 for (
int k=0; k<regspec[reg].nsubregs; k++) {
402 size_t colon_pos = regspec[reg].subregtags[k].find(
':');
403 int tag1 = std::stoi(regspec[reg].subregtags[k].substr(0, colon_pos));
404 int tag2 = std::stoi(regspec[reg].subregtags[k].substr(colon_pos + 1));
407 std::string inverse_pair;
408 inverse_pair = std::to_string(tag2) +
":" + std::to_string(tag1);
409 if(tagset.
count(regspec[reg].subregtags[k]) || tagset.
count(inverse_pair)) {
418 "on face Region tag %s in %s[%d] not found in element list for %s grid.\n",
419 regspec[reg].subregtags[k].c_str(), reglist, reg, gridname);
420 AllTagsExist =
false;
425 log_msg(NULL, 4,
ECHO,
"Assigned wrong pair of tags on the face of EMI surface mesh!\n"
426 "Check region ID specs in input files!\n");
432 inline bool pair_is_gapjunction(
const std::pair<mesh_int_t, mesh_int_t>& tags,
435 return intra_tags.
find(tags.first) != intra_tags.
end() &&
436 intra_tags.
find(tags.second) != intra_tags.
end();
439 void region_mask_onFace(
mesh_t meshspec,
440 std::vector<std::string> & tags_data,
456 if(regspec.
size() == 1)
return;
462 size_t rIDsize = mask_elem ? mesh.l_numelem : mesh.l_numpts;
464 size_t nelem = mesh.l_numelem;
467 check_tags_in_elems_onFace(tags_data, regspec, mesh.name.c_str(), reglist);
469 regionIDs.
assign(rIDsize, 0);
471 int* rid = regionIDs.
data();
476 for (
size_t reg=0; reg < regspec.
size(); reg++) {
478 for (
int k=0; k < regspec[reg].nsubregs; k++) {
479 std::string curtag = regspec[reg].subregtags[k];
480 if(tag_to_reg.
count(curtag)) err++;
483 log_msg(0,4,0,
"%s warning: Tag idx %s is assigned to multiple regions!\n"
484 "Its final assignment will be to the highest assigned region ID!",
485 __func__, curtag.c_str());
487 tag_to_reg[curtag] = reg;
491 std::vector<int> mx_tag( rIDsize, -1 );
494 for(
size_t eidx=0; eidx<nelem; eidx++)
496 std::vector<int> elem_nodes;
498 std::string result_pair_orginal;
499 std::string result_pair_reverse;
500 std::pair<mesh_int_t,mesh_int_t> value = map_elem_uniqueFace_to_tags[eidx];
501 result_pair_orginal = std::to_string(value.first) +
":" + std::to_string(value.second);
502 result_pair_reverse = std::to_string(value.second) +
":" + std::to_string(value.first);
503 rid[eidx] = pair_is_gapjunction(value, intra_tags) ? 1 : 0;
505 if (tag_to_reg.
count(result_pair_orginal)) {
506 rid[eidx] = tag_to_reg[result_pair_orginal];
507 }
else if(tag_to_reg.
count(result_pair_reverse)){
508 rid[eidx] = tag_to_reg[result_pair_reverse];
510 if(tag!=value.first){
511 log_msg(NULL, 5,
ECHO,
"error in tags on surface mesh with the unique faces!!!.\\n");
519 double IonicsOnFace::timer_val(
const int timer_id)
521 double val = std::nan(
"NaN");
527 std::string IonicsOnFace::timer_unit(
const int timer_id)
535 if (impdata[limpet::Iion] != NULL)
536 impdata[limpet::Iion][n] = 0;
540 imp.
compute(n, n + 1, impdata);
555 void* find_SV_in_IMP_onFace(
limpet::MULTI_IF* miif,
const int idx,
const char *IMP,
const char *SV,
556 int* offset,
int* sz)
558 if(strcmp(IMP, miif->
iontypes[idx].get().get_name().c_str()) == 0) {
559 return (
void*) miif->
iontypes[idx].get().get_sv_offset(SV, offset, sz);
562 for(
int k=0; k<miif->
numplugs[idx]; k++ )
563 if(strcmp(IMP, miif->
plugtypes[idx][k].get().get_name().c_str()) == 0) {
564 return (
void*) miif->
plugtypes[idx][k].get().get_sv_offset(SV, offset, sz);
581 void alloc_gvec_data_onFace(
const int nGVcs,
const int nRegs,
const int nEmiRegs,
582 GVecs *prmGVecs, gvec_data_OnFace &glob_vecs)
584 glob_vecs.nRegs = nRegs;
587 glob_vecs.vecs.resize(nGVcs);
589 for (
size_t i = 0; i < glob_vecs.vecs.size(); i++) {
590 sv_data_onFace &gvec = glob_vecs.vecs[i];
592 gvec.name =
dupstr(prmGVecs[i].name);
594 gvec.bogus = prmGVecs[i].bogus;
596 gvec.imps = (
char**) calloc(nRegs,
sizeof(
char *));
597 gvec.svNames = (
char**) calloc(nRegs,
sizeof(
char *));
598 gvec.svSizes = (
int*) calloc(nRegs,
sizeof(
int));
599 gvec.svOff = (
int*) calloc(nRegs,
sizeof(
int));
602 for (
int j = 0; j < nRegs; j++) {
603 if (strlen(prmGVecs[i].imp)) gvec.imps[j] =
dupstr(prmGVecs[i].imp);
605 else if (j < nEmiRegs)
606 gvec.imps[j] =
dupstr(param_globals::imp_region_emi[j].im);
608 gvec.imps[j] =
dupstr(param_globals::PurkIon[j - nEmiRegs].im);
610 else if (j < nEmiRegs)
611 gvec.imps[j] =
dupstr(param_globals::imp_region_emi[j].im);
613 gvec.svNames[j] =
dupstr(prmGVecs[i].ID[j]);
632 igb_output_manager & output_manager)
636 int num_purk_regions = 0;
637 int nEmiRegs = effective_num_imp_regions_emi();
638 int nRegs = nEmiRegs + num_purk_regions;
640 alloc_gvec_data_onFace(param_globals::num_gvecs, nRegs, nEmiRegs, param_globals::gvec, GVs);
643 if (GVs->inclPS) sample_PS_ionSVs(purk);
646 for (
unsigned int i = 0; i < GVs.vecs.size(); i++) {
647 sv_data_onFace & gv = GVs.vecs[i];
650 for (
int j = 0; j < miif->
N_IIF; j++) {
651 gv.getsv[j] = find_SV_in_IMP_onFace(miif, j, gv.imps[j], gv.svNames[j],
652 gv.svOff + j, gv.svSizes + j);
654 if (gv.getsv[j] == NULL) {
655 log_msg(NULL, 3,
ECHO,
"\tWarning: SV(%s) not found in region %d\n", gv.svNames[j], j);
661 output_manager.register_output(gv.ordered,
intra_elec_msh, 1, gv.name, gv.units);
667 MULTI_IF* pmiif = &purk->ion;
668 for (
int j = miif->
N_IIF; j < nRegs; j++) {
669 gv.getsv[j] = find_SV_in_IMP_onFace(pmiif, j - miif->
N_IIF, gv.imps[j],
670 gv.svNames[j], gv.svOff + j, gv.svSizes + j);
672 if (gv.getsv[j] == NULL) {
673 LOG_MSG(NULL, 3,
ECHO,
"\tWarning: state variable \"%s\" not found in region %d\n", gv.svNames[j], j);
677 RVector_dup(purk->vm_pt, &gv.orderedPS_PS);
680 RVector_dup(purk->vm_pt_over, &gv.orderedPS);
681 initialize_grid_output(grid, NULL, tmo,
intra_elec_msh, GRID_WRITE, 0., 1., gv.units, gv.orderedPS,
682 1, gv.GVcName, -purk->npt, param_globals::output_level);
687 log_msg(NULL, 5,
ECHO,
"\tError: no state variables found for global vector %d\n", i);
688 log_msg(NULL, 5,
ECHO,
"Run bench --imp=YourModel --imp-info to get a list of all parameters.\\n");
704 void assemble_sv_gvec_onFace(gvec_data_OnFace & gvecs,
limpet::MULTI_IF *miif)
706 for(
size_t i=0; i<gvecs.vecs.size(); i++ ) {
707 sv_data_onFace & gv = gvecs.vecs[i];
710 gv.ordered->set(gv.bogus);
712 gv.ordered->get_ownership_range(start, stop);
714 for(
int n = 0; n<miif->
N_IIF; n++ ) {
715 if( !gv.getsv[n] )
continue;
720 for(
int j=0; j<miif->
N_Nodes[n]; j++ ) {
721 indices[j] = miif->
NodeLists[n][j] + start;
726 gv.ordered->set(indices, data, add);
732 RVector_set( gv.orderedPS, gv.bogus );
734 if(IS_PURK_PROC(purk))
736 MULTI_IF *pmiif = &purk->ion;
737 Real *data =
new Real[purk->gvec_cab.nitems];
740 for(
int n = 0; n<pmiif->N_IIF; n++ ) {
741 int gvidx = n+miif->
N_IIF;
742 if( !gv.getsv[gvidx] )
continue;
743 for(
int j=0; j<purk->gvec_ion[n].nitems; j++ )
744 data[ci++] = ((
SVgetfcn)(gv.getsv[gvidx]))( pmiif->IIF+n, ((
int*)(purk->gvec_ion[n].data))[j],
747 RVector_setvals(gv.orderedPS, purk->gvec_cab.nitems, (
int*)purk->gvec_cab.data, data,
true);
750 RVector_sync( gv.orderedPS );
std::int32_t SF_int
Use the general std::int32_t as int type.
Comfort class. Provides getter functions to access the mesh member variables more comfortably.
Container for a PETSc VecScatter.
A vector storing arbitrary data.
size_t size() const
The current size of the vector.
void assign(InputIterator s, InputIterator e)
Assign a memory range.
T * data()
Pointer to the vector's start.
hm_int count(const K &key) const
Check if key exists.
Custom unordered_set implementation.
iterator find(const K &key)
hm_int count(const K &key) const
void insert(InputIterator first, InputIterator last)
Represents the ionic model and plug-in (IMP) data structure.
ts & get_tstp()
Gets the time stepper.
void for_each(const std::function< void(IonIfBase &)> &consumer)
Executes the consumer functions on this IMP and each of its plugins.
void compute(int start, int end, GlobalData_t **data)
Perform ionic model computation for 1 time step.
std::vector< IonIfBase * > IIF
array of IIF's
void sv_dump_add_by_name_list(int, char *, char *, char *, char *, char *, double, double)
int * numplugs
number of plugins for each region
std::vector< IonTypeList > plugtypes
plugins types for each region
IonTypeList iontypes
type for each region
int N_IIF
how many different IIF's
int * N_Nodes
#nodes for each IMP
int ** NodeLists
local partitioned node lists for each IMP stored
void setup(double inp_dt, double inp_start, double inp_end)
Initialize the timer_manager.
void reset_timers()
Reset time in timer_manager and then reset registered timers.
Electrical ionics functions (gap junction + ionic current) on the face of interface mesh for EMI mesh...
T local_nodal_to_local_petsc(const meshdata< T, S > &mesh, int rank, T local_nodal)
void init_vector(SF::abstract_vector< T, S > **vec)
SF_nbr
Enumeration encoding the different supported numberings.
@ NBR_PETSC
PETSc numbering of nodes.
@ NBR_REF
The nodal numbering of the reference mesh (the one stored on HD).
@ NBR_SUBMESH
Submesh nodal numbering: The globally ascending sorted reference indices are reindexed.
int get_plug_flag(char *plgstr, int *out_num_plugins, IonTypeList &out_plugins)
GlobalData_t(* SVgetfcn)(IonIfBase &, int, int)
IonType * get_ion_type(const std::string &name)
void update_ts(ts *ptstp)
int read_sv(MULTI_IF *, int, const char *)
std::map< int, std::string > units
timer_manager * tm_manager
a manager for the various physics timers
sf_mesh & get_mesh(const mesh_t gt)
Get a mesh by specifying the gridID.
SF::scattering * get_permutation(const int mesh_id, const int perm_id, const int dpn)
Get the PETSC to canonical permutation scattering for a given mesh and number of dpn.
SF::meshdata< mesh_int_t, mesh_real_t > sf_mesh
void read_metadata(const std::string filename, std::map< std::string, std::string > &metadata, MPI_Comm comm)
Read metadata from the header.
int get_rank(MPI_Comm comm=PETSC_COMM_WORLD)
T get_global(T in, MPI_Op OP, MPI_Comm comm=PETSC_COMM_WORLD)
Do a global reduction on a variable.
SF::scattering * register_permutation(const int mesh_id, const int perm_id, const int dpn)
Register a permutation between two orderings for a mesh.
void register_data(sf_vec *dat, datavec_t d)
Register a data vector in the global registry.
char * dupstr(const char *old_str)
void log_msg(FILE_SPEC out, int level, unsigned char flag, const char *fmt,...)
mesh_t
The enum identifying the different meshes we might want to load.
void get_time(double &tm)
SF::abstract_vector< SF_int, SF_real > sf_vec
void remove_char(char *buff, const int buffsize, const char c)
void read_indices_with_data(SF::vector< T > &idx, SF::vector< S > &dat, const std::string filename, const hashmap::unordered_map< mesh_int_t, mesh_int_t > &dd_map, const int dpn, MPI_Comm comm)
like read_indices, but with associated data for each index
V timing(V &t2, const V &t1)
#define PETSC_TO_CANONICAL
Permute algebraic data from PETSC to canonical ordering.