34 void initialize_sv_dumps_onFace(
limpet::MULTI_IF *pmiif, IMPregion_EMI* reg,
int id,
double t,
double dump_dt);
41 bool imp_region_emi_is_assigned(
const IMPregion_EMI& region,
int idx)
43 if (!(region.im && strlen(region.im) > 0))
return false;
44 if (idx < 2)
return true;
45 return region.num_IDs > 0;
50 int effective_num_imp_regions_emi()
52 const int configured_nreg = param_globals::num_imp_regions;
53 int nreg = configured_nreg;
54 while (nreg > 2 && !imp_region_emi_is_assigned(param_globals::imp_region_emi[nreg - 1], nreg - 1)) {
58 if (nreg < configured_nreg) {
59 static bool warned_once =
false;
61 MPI_Comm_rank(MPI_COMM_WORLD, &rank);
62 if (!warned_once && rank == 0) {
64 "Warning: num_imp_regions=%d but only %d imp_region_emi entries are assigned; ignoring trailing unassigned regions.\n",
65 configured_nreg, nreg);
75 void IonicsOnFace::compute_step()
80 miif->compute_ionic_current();
82 double comp_time =
timing(t2, t1);
83 this->compute_time += comp_time;
86 comp_stats.tot_time += comp_time;
92 void IonicsOnFace::destroy()
97 void IonicsOnFace::output_step()
100 void IonicsOnFace::initialize()
108 comp_stats.init_logger(
"ODE_stats.dat");
125 miif->name =
"myocardium";
126 miif->gdata[limpet::Vm] = Vmv;
127 miif->gdata[limpet::Iion] = IIon;
129 const int num_imp_regions = effective_num_imp_regions_emi();
133 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");
134 log_msg(NULL, 5,
ECHO,
"set a default ionic model and gapjuntion model in parameter file\\n");
142 rs[0].subregtags =
nullptr;
144 rs[1].subregtags =
nullptr;
147 for (
size_t i = 2; i < rs.size(); i++ ) {
148 rs[i].nsubregs = param_globals::imp_region_emi[i].num_IDs;
149 rs[i].subregtags =
new std::string[rs[i].nsubregs];
150 for (
int j = 0; j < rs[i].nsubregs;j++) {
151 std::string t1t2 = param_globals::imp_region_emi[i].ID[j];
152 size_t colon_pos = t1t2.find(
':');
153 if (colon_pos == std::string::npos) {
154 std::cerr <<
"Error: ':' the face tags are not defined properly in input file, it should be tag1:tag2 as a string" << std::endl;
158 rs[i].subregtags[j] = param_globals::imp_region_emi[i].ID[j];
163 region_mask_onFace(ion_domain, tags_data, line_face, tri_face, quad_face,
164 map_vertex_tag_to_dof, map_elem_uniqueFace_to_tags, intra_tags,
165 rs, reg_mask,
true,
"imp_region_emi");
167 for (
size_t i = 0; i < rs.size(); i++ ) {
168 delete[] rs[i].subregtags;
169 rs[i].subregtags =
nullptr;
173 tstart = setup_MIIF(loc_size, num_imp_regions, param_globals::imp_region_emi,
174 reg_mask.
data(), param_globals::start_statef, param_globals::num_adjustments,
175 param_globals::adjustment, param_globals::dt, purkfLen > 0);
177 miif->extUpdateVm = !param_globals::operator_splitting;
182 log_msg(logger, 0, 0,
"Changing simulation start time to %.2lf", tstart);
189 this->initialize_time +=
timing(t2, t1);
192 double IonicsOnFace::setup_MIIF(
limpet::node_count_t nnodes,
int nreg, IMPregion_EMI* impreg,
int* mask,
193 const char *start_fn,
int numadjust, IMPVariableAdjustment *adjust,
194 double time_step,
bool close)
199 miif->numNode = nnodes;
201 miif->numplugs = (
int*)calloc( miif->N_IIF,
sizeof(
int));
202 miif->plugtypes = std::vector<limpet::IonTypeList>(miif->N_IIF);
205 log_msg(logger,0,
ECHO,
"\nSetting up ionic models on EMI Face and plugins\n" \
206 "-----------------------------------\n\n" \
207 "Assigning IMPS to tagged regions:" );
209 for (
int i=0;i<miif->N_IIF;i++) {
213 miif->iontypes.push_back(*pT);
214 log_msg(logger, 0,
ECHO|
NONL,
"\tIonic model: %s to tag region(s)", impreg[i].im);
216 if(impreg[i].num_IDs > 0) {
217 for(
int j = 0; j < impreg[i].num_IDs; j++)
226 log_msg(NULL,5,
ECHO,
"Illegal IM specified: %s\n", impreg[i].im );
227 log_msg(NULL,5,
ECHO,
"Run bench --list-imps for a list of all available models.\n" );
232 if(impreg[i].plugins[0] !=
'\0') {
233 log_msg(logger,0,
ECHO|
NONL,
"\tPlug-in(s) : %s to tag region(s)", impreg[i].plugins);
235 for(
int j = 0; j < impreg[i].num_IDs; j++)
241 log_msg(NULL,5,
ECHO,
"Illegal plugin specified: %s\n", impreg[i].plugins);
242 log_msg(NULL,5,
ECHO,
"Run bench --list-imps for a list of all available plugins.\n" );
256 miif->initialize_MIIF();
258 for (
int i=0;i<miif->N_IIF;i++) {
260 remove_char(impreg[i].im_param, strlen(impreg[i].im_param),
' ');
261 miif->IIF[i]->tune(impreg[i].im_param, impreg[i].plugins, impreg[i].plug_param);
265 miif->initialize_currents(time_step, param_globals::ode_fac);
270 for (
int i=0;i<miif->N_IIF;i++) {
271 if (impreg[i].im_sv_init && strlen(impreg[i].im_sv_init) > 0)
272 if (
read_sv(miif, i, impreg[i].im_sv_init)) {
273 log_msg(NULL, 5,
ECHO|
FLUSH,
"State vector initialization failed for %s.\n", impreg[i].name);
278 if( !start_fn || strlen(start_fn)>0 )
279 tstart = (double) miif->restore_state(start_fn, ion_domain, close);
281 for (
int i=0; i<numadjust; i++)
287 bool restrict_to_algebraic =
true;
290 std::map<std::string,std::string> metadata;
294 if(metadata.count(
"grid") && metadata[
"grid"].compare(
"intra") == 0) {
297 read_indices_with_data(indices, values, adjust[i].file, imesh, nbr, restrict_to_algebraic, 1, PETSC_COMM_WORLD);
301 for(
size_t gi = 0; gi < indices.
size(); gi++)
302 indices[gi] = SF::local_nodal_to_local_petsc<mesh_int_t, mesh_real_t>(imesh, rank, indices[gi]);
308 adjPars->set(indices, values,
false,
true);
312 snprintf(fname,
sizeof fname,
"adj_%s_perm.dat", adjust[i].variable);
313 adjPars->write_ascii(fname,
false);
318 log_msg(0,3,0,
"%s warning: PETSC_TO_CANONICAL permutation needed registering!", __func__);
322 (*sc)(*adjPars,
true);
323 snprintf(fname,
sizeof fname,
"adj_%s_canonical.dat", adjust[i].variable);
324 adjPars->write_ascii(fname,
false);
327 int nc = miif->adjust_MIIF_variables(adjust[i].variable, indices, values);
328 log_msg(logger, 0, 0,
"Adjusted %d values for %s", nc, adjust[i].variable);
333 for (
int i=0;i<miif->N_IIF;i++)
334 initialize_sv_dumps_onFace(miif, impreg+i, i, tstart, param_globals::spacedt);
350 void initialize_sv_dumps_onFace(
limpet::MULTI_IF *pmiif, IMPregion_EMI* reg,
int id,
double t,
double dump_dt)
352 char svs[1024], plgs[1024], plgsvs[1024], fname[1024];
354 strcpy(svs, reg->im_sv_dumps ? reg->im_sv_dumps :
"");
355 strcpy(plgs, reg->plugins ? reg->plugins :
"");
356 strcpy(plgsvs, reg->plug_sv_dumps ? reg->plug_sv_dumps :
"");
358 if( !(strlen(svs)+strlen(plgsvs) ) )
367 strcpy(fname, param_globals::vofile);
370 log_msg(NULL, 5,
ECHO,
"%s: a region name must be specified\n", __func__ );
375 char* ext_start = NULL;
376 ext_start = strstr(fname,
"igb.gz");
377 if(ext_start == NULL) ext_start = strstr(fname,
"igb");
378 if(ext_start == NULL) ext_start = fname + strlen(fname);
379 strcpy(ext_start, reg->name);
395 const char* gridname,
const char* reglist)
397 bool AllTagsExist =
true;
400 tagset.
insert(tags_data.begin(), tags_data.end());
403 for (
size_t reg=0; reg<regspec.
size(); reg++)
405 for (
int k=0; k<regspec[reg].nsubregs; k++) {
408 size_t colon_pos = regspec[reg].subregtags[k].find(
':');
409 int tag1 = std::stoi(regspec[reg].subregtags[k].substr(0, colon_pos));
410 int tag2 = std::stoi(regspec[reg].subregtags[k].substr(colon_pos + 1));
413 std::string inverse_pair;
414 inverse_pair = std::to_string(tag2) +
":" + std::to_string(tag1);
415 if(tagset.
count(regspec[reg].subregtags[k]) || tagset.
count(inverse_pair)) {
424 "on face Region tag %s in %s[%d] not found in element list for %s grid.\n",
425 regspec[reg].subregtags[k].c_str(), reglist, reg, gridname);
426 AllTagsExist =
false;
431 log_msg(NULL, 4,
ECHO,
"Assigned wrong pair of tags on the face of EMI surface mesh!\n"
432 "Check region ID specs in input files!\n");
438 inline bool pair_is_gapjunction(
const std::pair<mesh_int_t, mesh_int_t>& tags,
441 return intra_tags.
find(tags.first) != intra_tags.
end() &&
442 intra_tags.
find(tags.second) != intra_tags.
end();
445 void region_mask_onFace(
mesh_t meshspec,
446 std::vector<std::string> & tags_data,
462 if(regspec.
size() == 1)
return;
468 size_t rIDsize = mask_elem ? mesh.l_numelem : mesh.l_numpts;
470 size_t nelem = mesh.l_numelem;
473 check_tags_in_elems_onFace(tags_data, regspec, mesh.name.c_str(), reglist);
475 regionIDs.
assign(rIDsize, 0);
477 int* rid = regionIDs.
data();
482 for (
size_t reg=0; reg < regspec.
size(); reg++) {
484 for (
int k=0; k < regspec[reg].nsubregs; k++) {
485 std::string curtag = regspec[reg].subregtags[k];
486 if(tag_to_reg.
count(curtag)) err++;
489 log_msg(0,4,0,
"%s warning: Tag idx %s is assigned to multiple regions!\n"
490 "Its final assignment will be to the highest assigned region ID!",
491 __func__, curtag.c_str());
493 tag_to_reg[curtag] = reg;
497 std::vector<int> mx_tag( rIDsize, -1 );
500 for(
size_t eidx=0; eidx<nelem; eidx++)
502 std::vector<int> elem_nodes;
504 std::string result_pair_orginal;
505 std::string result_pair_reverse;
506 std::pair<mesh_int_t,mesh_int_t> value = map_elem_uniqueFace_to_tags[eidx];
507 result_pair_orginal = std::to_string(value.first) +
":" + std::to_string(value.second);
508 result_pair_reverse = std::to_string(value.second) +
":" + std::to_string(value.first);
509 rid[eidx] = pair_is_gapjunction(value, intra_tags) ? 1 : 0;
511 if (tag_to_reg.
count(result_pair_orginal)) {
512 rid[eidx] = tag_to_reg[result_pair_orginal];
513 }
else if(tag_to_reg.
count(result_pair_reverse)){
514 rid[eidx] = tag_to_reg[result_pair_reverse];
516 if(tag!=value.first){
517 log_msg(NULL, 5,
ECHO,
"error in tags on surface mesh with the unique faces!!!.\\n");
525 double IonicsOnFace::timer_val(
const int timer_id)
527 double val = std::nan(
"NaN");
533 std::string IonicsOnFace::timer_unit(
const int timer_id)
541 if (impdata[limpet::Iion] != NULL)
542 impdata[limpet::Iion][n] = 0;
546 imp.
compute(n, n + 1, impdata);
561 void* find_SV_in_IMP_onFace(
limpet::MULTI_IF* miif,
const int idx,
const char *IMP,
const char *SV,
562 int* offset,
int* sz)
564 if(strcmp(IMP, miif->
iontypes[idx].get().get_name().c_str()) == 0) {
565 return (
void*) miif->
iontypes[idx].get().get_sv_offset(SV, offset, sz);
568 for(
int k=0; k<miif->
numplugs[idx]; k++ )
569 if(strcmp(IMP, miif->
plugtypes[idx][k].get().get_name().c_str()) == 0) {
570 return (
void*) miif->
plugtypes[idx][k].get().get_sv_offset(SV, offset, sz);
588 void alloc_gvec_data_onFace(
const int nGVcs,
const int nRegs,
const int nEmiRegs,
589 GVecs *prmGVecs, gvec_data_OnFace &glob_vecs)
591 glob_vecs.nRegs = nRegs;
594 glob_vecs.vecs.resize(nGVcs);
596 for (
size_t i = 0; i < glob_vecs.vecs.size(); i++) {
597 sv_data_onFace &gvec = glob_vecs.vecs[i];
599 gvec.name =
dupstr(prmGVecs[i].name);
601 gvec.bogus = prmGVecs[i].bogus;
603 gvec.imps = (
char**) calloc(nRegs,
sizeof(
char *));
604 gvec.svNames = (
char**) calloc(nRegs,
sizeof(
char *));
605 gvec.svSizes = (
int*) calloc(nRegs,
sizeof(
int));
606 gvec.svOff = (
int*) calloc(nRegs,
sizeof(
int));
609 for (
int j = 0; j < nRegs; j++) {
610 if (strlen(prmGVecs[i].imp)) gvec.imps[j] =
dupstr(prmGVecs[i].imp);
612 else if (j < nEmiRegs)
613 gvec.imps[j] =
dupstr(param_globals::imp_region_emi[j].im);
615 gvec.imps[j] =
dupstr(param_globals::PurkIon[j - nEmiRegs].im);
617 else if (j < nEmiRegs)
618 gvec.imps[j] =
dupstr(param_globals::imp_region_emi[j].im);
620 gvec.svNames[j] =
dupstr(prmGVecs[i].ID[j]);
637 igb_output_manager & output_manager)
641 int num_purk_regions = 0;
642 int nEmiRegs = effective_num_imp_regions_emi();
643 int nRegs = nEmiRegs + num_purk_regions;
645 alloc_gvec_data_onFace(param_globals::num_gvecs, nRegs, nEmiRegs, param_globals::gvec, GVs);
648 if (GVs->inclPS) sample_PS_ionSVs(purk);
651 for (
unsigned int i = 0; i < GVs.vecs.size(); i++) {
652 sv_data_onFace & gv = GVs.vecs[i];
655 for (
int j = 0; j < miif->
N_IIF; j++) {
656 gv.getsv[j] = find_SV_in_IMP_onFace(miif, j, gv.imps[j], gv.svNames[j],
657 gv.svOff + j, gv.svSizes + j);
659 if (gv.getsv[j] == NULL) {
660 log_msg(NULL, 3,
ECHO,
"\tWarning: SV(%s) not found in region %d\n", gv.svNames[j], j);
666 output_manager.register_output(gv.ordered,
intra_elec_msh, 1, gv.name, gv.units);
672 MULTI_IF* pmiif = &purk->ion;
673 for (
int j = miif->
N_IIF; j < nRegs; j++) {
674 gv.getsv[j] = find_SV_in_IMP_onFace(pmiif, j - miif->
N_IIF, gv.imps[j],
675 gv.svNames[j], gv.svOff + j, gv.svSizes + j);
677 if (gv.getsv[j] == NULL) {
678 LOG_MSG(NULL, 3,
ECHO,
"\tWarning: state variable \"%s\" not found in region %d\n", gv.svNames[j], j);
682 RVector_dup(purk->vm_pt, &gv.orderedPS_PS);
685 RVector_dup(purk->vm_pt_over, &gv.orderedPS);
686 initialize_grid_output(grid, NULL, tmo,
intra_elec_msh, GRID_WRITE, 0., 1., gv.units, gv.orderedPS,
687 1, gv.GVcName, -purk->npt, param_globals::output_level);
692 log_msg(NULL, 5,
ECHO,
"\tError: no state variables found for global vector %d\n", i);
693 log_msg(NULL, 5,
ECHO,
"Run bench --imp=YourModel --imp-info to get a list of all parameters.\\n");
708 void assemble_sv_gvec_onFace(gvec_data_OnFace & gvecs,
limpet::MULTI_IF *miif)
710 for(
size_t i=0; i<gvecs.vecs.size(); i++ ) {
711 sv_data_onFace & gv = gvecs.vecs[i];
714 gv.ordered->set(gv.bogus);
716 gv.ordered->get_ownership_range(start, stop);
718 for(
int n = 0; n<miif->
N_IIF; n++ ) {
719 if( !gv.getsv[n] )
continue;
725 indices[j] = miif->
NodeLists[n][j] + start;
730 gv.ordered->set(indices, data, add);
736 RVector_set( gv.orderedPS, gv.bogus );
738 if(IS_PURK_PROC(purk))
740 MULTI_IF *pmiif = &purk->ion;
741 Real *data =
new Real[purk->gvec_cab.nitems];
744 for(
int n = 0; n<pmiif->N_IIF; n++ ) {
745 int gvidx = n+miif->
N_IIF;
746 if( !gv.getsv[gvidx] )
continue;
747 for(
int j=0; j<purk->gvec_ion[n].nitems; j++ )
748 data[ci++] = ((
SVgetfcn)(gv.getsv[gvidx]))( pmiif->IIF+n, ((
int*)(purk->gvec_ion[n].data))[j],
751 RVector_setvals(gv.orderedPS, purk->gvec_cab.nitems, (
int*)purk->gvec_cab.data, data,
true);
754 RVector_sync( gv.orderedPS );
opencarp::local_index_t mesh_int_t
opencarp::global_index_t SF_int
Global algebraic index 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.
void compute(node_index_t start, node_index_t end, GlobalData_t **data)
Perform ionic model computation for 1 time step.
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.
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
node_count_t * N_Nodes
#nodes for each IMP
node_index_t ** 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.
LIMPET ionics and gap-junction models on the EMI unique-face interface mesh.
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)
IonType * get_ion_type(const std::string &name)
GlobalData_t(* SVgetfcn)(IonIfBase &, node_index_t, int)
void update_ts(ts *ptstp)
opencarp::local_index_t node_count_t
int read_sv(MULTI_IF *, int, const char *)
opencarp::local_index_t node_index_t
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.