19 void initialize_sv_dumps_onFace(
limpet::MULTI_IF *pmiif, IMPregion_EMI* reg,
int id,
double t,
double dump_dt);
26 bool imp_region_emi_is_assigned(
const IMPregion_EMI& region,
int idx)
28 if (!(region.im && strlen(region.im) > 0))
return false;
29 if (idx < 2)
return true;
30 return region.num_IDs > 0;
35 int effective_num_imp_regions_emi()
37 const int configured_nreg = param_globals::num_imp_regions;
38 int nreg = configured_nreg;
39 while (nreg > 2 && !imp_region_emi_is_assigned(param_globals::imp_region_emi[nreg - 1], nreg - 1)) {
43 if (nreg < configured_nreg) {
44 static bool warned_once =
false;
46 MPI_Comm_rank(MPI_COMM_WORLD, &rank);
47 if (!warned_once && rank == 0) {
49 "Warning: num_imp_regions=%d but only %d imp_region_emi entries are assigned; ignoring trailing unassigned regions.\n",
50 configured_nreg, nreg);
60 void IonicsOnFace::compute_step()
65 miif->compute_ionic_current();
67 double comp_time =
timing(t2, t1);
68 this->compute_time += comp_time;
71 comp_stats.tot_time += comp_time;
77 void IonicsOnFace::destroy()
82 void IonicsOnFace::output_step()
85 void IonicsOnFace::initialize()
93 comp_stats.init_logger(
"ODE_stats.dat");
111 miif->logger = logger;
113 miif->name =
"myocardium";
114 miif->gdata[limpet::Vm] = Vmv;
115 miif->gdata[limpet::Iion] = IIon;
117 const int num_imp_regions = effective_num_imp_regions_emi();
121 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");
122 log_msg(NULL, 5,
ECHO,
"set a default ionic model and gapjuntion model in parameter file\\n");
130 rs[0].subregtags =
nullptr;
132 rs[1].subregtags =
nullptr;
135 for (
size_t i = 2; i < rs.size(); i++ ) {
136 rs[i].nsubregs = param_globals::imp_region_emi[i].num_IDs;
137 rs[i].subregtags =
new std::string[rs[i].nsubregs];
138 for (
int j = 0; j < rs[i].nsubregs;j++) {
139 std::string t1t2 = param_globals::imp_region_emi[i].ID[j];
140 size_t colon_pos = t1t2.find(
':');
141 if (colon_pos == std::string::npos) {
142 std::cerr <<
"Error: ':' the face tags are not defined properly in input file, it should be tag1:tag2 as a string" << std::endl;
146 rs[i].subregtags[j] = param_globals::imp_region_emi[i].ID[j];
151 region_mask_onFace(ion_domain, tags_data, line_face, tri_face, quad_face,
152 map_vertex_tag_to_dof, map_elem_uniqueFace_to_tags, intra_tags,
153 rs, reg_mask,
true,
"imp_region_emi");
155 for (
size_t i = 0; i < rs.size(); i++ ) {
156 delete[] rs[i].subregtags;
157 rs[i].subregtags =
nullptr;
161 tstart = setup_MIIF(loc_size, num_imp_regions, param_globals::imp_region_emi,
162 reg_mask.
data(), param_globals::start_statef, param_globals::num_adjustments,
163 param_globals::adjustment, param_globals::dt, purkfLen > 0);
165 miif->extUpdateVm = !param_globals::operator_splitting;
170 log_msg(logger, 0, 0,
"Changing simulation start time to %.2lf", tstart);
177 this->initialize_time +=
timing(t2, t1);
180 double IonicsOnFace::setup_MIIF(
limpet::node_count_t nnodes,
int nreg, IMPregion_EMI* impreg,
int* mask,
181 const char *start_fn,
int numadjust, IMPVariableAdjustment *adjust,
182 double time_step,
bool close)
187 miif->numNode = nnodes;
189 miif->numplugs = (
int*)calloc( miif->N_IIF,
sizeof(
int));
190 miif->plugtypes = std::vector<limpet::IonTypeList>(miif->N_IIF);
193 log_msg(logger,0,
ECHO,
"\nSetting up ionic models on EMI Face and plugins\n" \
194 "-----------------------------------\n\n" \
195 "Assigning IMPS to tagged regions:" );
197 for (
int i=0;i<miif->N_IIF;i++) {
201 miif->iontypes.push_back(*pT);
202 log_msg(logger, 0,
ECHO|
NONL,
"\tIonic model: %s to tag region(s)", impreg[i].im);
204 if(impreg[i].num_IDs > 0) {
205 for(
int j = 0; j < impreg[i].num_IDs; j++)
214 log_msg(NULL,5,
ECHO,
"Illegal IM specified: %s\n", impreg[i].im );
215 log_msg(NULL,5,
ECHO,
"Run bench --list-imps for a list of all available models.\n" );
220 if(impreg[i].plugins[0] !=
'\0') {
221 log_msg(logger,0,
ECHO|
NONL,
"\tPlug-in(s) : %s to tag region(s)", impreg[i].plugins);
223 for(
int j = 0; j < impreg[i].num_IDs; j++)
229 log_msg(NULL,5,
ECHO,
"Illegal plugin specified: %s\n", impreg[i].plugins);
230 log_msg(NULL,5,
ECHO,
"Run bench --list-imps for a list of all available plugins.\n" );
244 miif->initialize_MIIF();
246 for (
int i=0;i<miif->N_IIF;i++) {
248 remove_char(impreg[i].im_param, strlen(impreg[i].im_param),
' ');
249 miif->IIF[i]->tune(impreg[i].im_param, impreg[i].plugins, impreg[i].plug_param);
253 miif->initialize_currents(time_step, param_globals::ode_fac);
258 for (
int i=0;i<miif->N_IIF;i++) {
259 if (impreg[i].im_sv_init && strlen(impreg[i].im_sv_init) > 0)
260 if (
read_sv(miif, i, impreg[i].im_sv_init)) {
261 log_msg(NULL, 5,
ECHO|
FLUSH,
"State vector initialization failed for %s.\n", impreg[i].name);
266 if( !start_fn || strlen(start_fn)>0 )
267 tstart = (double) miif->restore_state(start_fn, ion_domain, close);
269 for (
int i=0; i<numadjust; i++)
275 bool restrict_to_algebraic =
true;
278 std::map<std::string,std::string> metadata;
282 if(metadata.count(
"grid") && metadata[
"grid"].compare(
"intra") == 0) {
285 read_indices_with_data(indices, values, adjust[i].file, imesh, nbr, restrict_to_algebraic, 1, PETSC_COMM_WORLD);
289 for(
size_t gi = 0; gi < indices.
size(); gi++)
290 indices[gi] = SF::local_nodal_to_local_petsc<mesh_int_t, mesh_real_t>(imesh, rank, indices[gi]);
296 adjPars->set(indices, values,
false,
true);
300 snprintf(fname,
sizeof fname,
"adj_%s_perm.dat", adjust[i].variable);
301 adjPars->write_ascii(fname,
false);
306 log_msg(0,3,0,
"%s warning: PETSC_TO_CANONICAL permutation needed registering!", __func__);
310 (*sc)(*adjPars,
true);
311 snprintf(fname,
sizeof fname,
"adj_%s_canonical.dat", adjust[i].variable);
312 adjPars->write_ascii(fname,
false);
315 int nc = miif->adjust_MIIF_variables(adjust[i].variable, indices, values);
316 log_msg(logger, 0, 0,
"Adjusted %d values for %s", nc, adjust[i].variable);
321 for (
int i=0;i<miif->N_IIF;i++)
322 initialize_sv_dumps_onFace(miif, impreg+i, i, tstart, param_globals::spacedt);
338 void initialize_sv_dumps_onFace(
limpet::MULTI_IF *pmiif, IMPregion_EMI* reg,
int id,
double t,
double dump_dt)
340 char svs[1024], plgs[1024], plgsvs[1024], fname[1024];
342 strcpy(svs, reg->im_sv_dumps ? reg->im_sv_dumps :
"");
343 strcpy(plgs, reg->plugins ? reg->plugins :
"");
344 strcpy(plgsvs, reg->plug_sv_dumps ? reg->plug_sv_dumps :
"");
346 if( !(strlen(svs)+strlen(plgsvs) ) )
355 strcpy(fname, param_globals::vofile);
358 log_msg(NULL, 5,
ECHO,
"%s: a region name must be specified\n", __func__ );
363 size_t fname_len = strlen(fname);
364 char* ext_start = fname + fname_len;
365 if(fname_len >= 4 && strcmp(ext_start - 4,
".igb") == 0) ext_start -= 3;
366 strcpy(ext_start, reg->name);
382 const char* gridname,
const char* reglist)
384 bool AllTagsExist =
true;
387 tagset.
insert(tags_data.begin(), tags_data.end());
390 for (
size_t reg=0; reg<regspec.
size(); reg++)
392 for (
int k=0; k<regspec[reg].nsubregs; k++) {
395 size_t colon_pos = regspec[reg].subregtags[k].find(
':');
396 int tag1 = std::stoi(regspec[reg].subregtags[k].substr(0, colon_pos));
397 int tag2 = std::stoi(regspec[reg].subregtags[k].substr(colon_pos + 1));
400 std::string inverse_pair;
401 inverse_pair = std::to_string(tag2) +
":" + std::to_string(tag1);
402 if(tagset.
count(regspec[reg].subregtags[k]) || tagset.
count(inverse_pair)) {
411 "on face Region tag %s in %s[%d] not found in element list for %s grid.\n",
412 regspec[reg].subregtags[k].c_str(), reglist, reg, gridname);
413 AllTagsExist =
false;
418 log_msg(NULL, 4,
ECHO,
"Assigned wrong pair of tags on the face of EMI surface mesh!\n"
419 "Check region ID specs in input files!\n");
425 inline bool pair_is_gapjunction(
const std::pair<mesh_int_t, mesh_int_t>& tags,
428 return intra_tags.
find(tags.first) != intra_tags.
end() &&
429 intra_tags.
find(tags.second) != intra_tags.
end();
432 void region_mask_onFace(
mesh_t meshspec,
433 std::vector<std::string> & tags_data,
449 if(regspec.
size() == 1)
return;
455 size_t rIDsize = mask_elem ? mesh.l_numelem : mesh.l_numpts;
457 size_t nelem = mesh.l_numelem;
460 check_tags_in_elems_onFace(tags_data, regspec, mesh.name.c_str(), reglist);
462 regionIDs.
assign(rIDsize, 0);
464 int* rid = regionIDs.
data();
469 for (
size_t reg=0; reg < regspec.
size(); reg++) {
471 for (
int k=0; k < regspec[reg].nsubregs; k++) {
472 std::string curtag = regspec[reg].subregtags[k];
473 if(tag_to_reg.
count(curtag)) err++;
476 log_msg(0,4,0,
"%s warning: Tag idx %s is assigned to multiple regions!\n"
477 "Its final assignment will be to the highest assigned region ID!",
478 __func__, curtag.c_str());
480 tag_to_reg[curtag] = reg;
484 std::vector<int> mx_tag( rIDsize, -1 );
487 for(
size_t eidx=0; eidx<nelem; eidx++)
489 std::vector<int> elem_nodes;
491 std::string result_pair_orginal;
492 std::string result_pair_reverse;
493 std::pair<mesh_int_t,mesh_int_t> value = map_elem_uniqueFace_to_tags[eidx];
494 result_pair_orginal = std::to_string(value.first) +
":" + std::to_string(value.second);
495 result_pair_reverse = std::to_string(value.second) +
":" + std::to_string(value.first);
496 rid[eidx] = pair_is_gapjunction(value, intra_tags) ? 1 : 0;
498 if (tag_to_reg.
count(result_pair_orginal)) {
499 rid[eidx] = tag_to_reg[result_pair_orginal];
500 }
else if(tag_to_reg.
count(result_pair_reverse)){
501 rid[eidx] = tag_to_reg[result_pair_reverse];
503 if(tag!=value.first){
504 log_msg(NULL, 5,
ECHO,
"error in tags on surface mesh with the unique faces!!!.\\n");
512 double IonicsOnFace::timer_val(
const int timer_id)
514 double val = std::nan(
"NaN");
520 std::string IonicsOnFace::timer_unit(
const int timer_id)
528 if (impdata[limpet::Iion] != NULL)
529 impdata[limpet::Iion][n] = 0;
533 imp.
compute(n, n + 1, impdata);
548 void* find_SV_in_IMP_onFace(
limpet::MULTI_IF* miif,
const int idx,
const char *IMP,
const char *SV,
549 int* offset,
int* sz)
551 if(strcmp(IMP, miif->
iontypes[idx].get().get_name().c_str()) == 0) {
552 return (
void*) miif->
iontypes[idx].get().get_sv_offset(SV, offset, sz);
555 for(
int k=0; k<miif->
numplugs[idx]; k++ )
556 if(strcmp(IMP, miif->
plugtypes[idx][k].get().get_name().c_str()) == 0) {
557 return (
void*) miif->
plugtypes[idx][k].get().get_sv_offset(SV, offset, sz);
575 void alloc_gvec_data_onFace(
const int nGVcs,
const int nRegs,
const int nEmiRegs,
576 GVecs *prmGVecs, gvec_data_OnFace &glob_vecs)
578 glob_vecs.nRegs = nRegs;
581 glob_vecs.vecs.resize(nGVcs);
583 for (
size_t i = 0; i < glob_vecs.vecs.size(); i++) {
584 sv_data_onFace &gvec = glob_vecs.vecs[i];
586 gvec.name =
dupstr(prmGVecs[i].name);
588 gvec.bogus = prmGVecs[i].bogus;
590 gvec.imps = (
char**) calloc(nRegs,
sizeof(
char *));
591 gvec.svNames = (
char**) calloc(nRegs,
sizeof(
char *));
592 gvec.svSizes = (
int*) calloc(nRegs,
sizeof(
int));
593 gvec.svOff = (
int*) calloc(nRegs,
sizeof(
int));
596 for (
int j = 0; j < nRegs; j++) {
597 if (strlen(prmGVecs[i].imp)) gvec.imps[j] =
dupstr(prmGVecs[i].imp);
599 else if (j < nEmiRegs)
600 gvec.imps[j] =
dupstr(param_globals::imp_region_emi[j].im);
602 gvec.imps[j] =
dupstr(param_globals::PurkIon[j - nEmiRegs].im);
604 else if (j < nEmiRegs)
605 gvec.imps[j] =
dupstr(param_globals::imp_region_emi[j].im);
607 gvec.svNames[j] =
dupstr(prmGVecs[i].ID[j]);
624 igb_output_manager & output_manager)
628 int num_purk_regions = 0;
629 int nEmiRegs = effective_num_imp_regions_emi();
630 int nRegs = nEmiRegs + num_purk_regions;
632 alloc_gvec_data_onFace(param_globals::num_gvecs, nRegs, nEmiRegs, param_globals::gvec, GVs);
635 if (GVs->inclPS) sample_PS_ionSVs(purk);
638 for (
unsigned int i = 0; i < GVs.vecs.size(); i++) {
639 sv_data_onFace & gv = GVs.vecs[i];
642 for (
int j = 0; j < miif->
N_IIF; j++) {
643 gv.getsv[j] = find_SV_in_IMP_onFace(miif, j, gv.imps[j], gv.svNames[j],
644 gv.svOff + j, gv.svSizes + j);
646 if (gv.getsv[j] == NULL) {
647 log_msg(NULL, 3,
ECHO,
"\tWarning: SV(%s) not found in region %d\n", gv.svNames[j], j);
653 output_manager.register_output(gv.ordered,
intra_elec_msh, 1, gv.name, gv.units);
659 MULTI_IF* pmiif = &purk->ion;
660 for (
int j = miif->
N_IIF; j < nRegs; j++) {
661 gv.getsv[j] = find_SV_in_IMP_onFace(pmiif, j - miif->
N_IIF, gv.imps[j],
662 gv.svNames[j], gv.svOff + j, gv.svSizes + j);
664 if (gv.getsv[j] == NULL) {
665 LOG_MSG(NULL, 3,
ECHO,
"\tWarning: state variable \"%s\" not found in region %d\n", gv.svNames[j], j);
669 RVector_dup(purk->vm_pt, &gv.orderedPS_PS);
672 RVector_dup(purk->vm_pt_over, &gv.orderedPS);
673 initialize_grid_output(grid, NULL, tmo,
intra_elec_msh, GRID_WRITE, 0., 1., gv.units, gv.orderedPS,
674 1, gv.GVcName, -purk->npt, param_globals::output_level);
679 log_msg(NULL, 5,
ECHO,
"\tError: no state variables found for global vector %d\n", i);
680 log_msg(NULL, 5,
ECHO,
"Run bench --imp=YourModel --imp-info to get a list of all parameters.\\n");
695 void assemble_sv_gvec_onFace(gvec_data_OnFace & gvecs,
limpet::MULTI_IF *miif)
697 for(
size_t i=0; i<gvecs.vecs.size(); i++ ) {
698 sv_data_onFace & gv = gvecs.vecs[i];
701 gv.ordered->set(gv.bogus);
703 gv.ordered->get_ownership_range(start, stop);
705 for(
int n = 0; n<miif->
N_IIF; n++ ) {
706 if( !gv.getsv[n] )
continue;
712 indices[j] = miif->
NodeLists[n][j] + start;
717 gv.ordered->set(indices, data, add);
723 RVector_set( gv.orderedPS, gv.bogus );
725 if(IS_PURK_PROC(purk))
727 MULTI_IF *pmiif = &purk->ion;
728 Real *data =
new Real[purk->gvec_cab.nitems];
731 for(
int n = 0; n<pmiif->N_IIF; n++ ) {
732 int gvidx = n+miif->
N_IIF;
733 if( !gv.getsv[gvidx] )
continue;
734 for(
int j=0; j<purk->gvec_ion[n].nitems; j++ )
735 data[ci++] = ((
SVgetfcn)(gv.getsv[gvidx]))( pmiif->IIF+n, ((
int*)(purk->gvec_ion[n].data))[j],
738 RVector_setvals(gv.orderedPS, purk->gvec_cab.nitems, (
int*)purk->gvec_cab.data, data,
true);
741 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.