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");
126 miif->logger = logger;
128 miif->name =
"myocardium";
129 miif->gdata[limpet::Vm] = Vmv;
130 miif->gdata[limpet::Iion] = IIon;
132 const int num_imp_regions = effective_num_imp_regions_emi();
136 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");
137 log_msg(NULL, 5,
ECHO,
"set a default ionic model and gapjuntion model in parameter file\\n");
145 rs[0].subregtags =
nullptr;
147 rs[1].subregtags =
nullptr;
150 for (
size_t i = 2; i < rs.size(); i++ ) {
151 rs[i].nsubregs = param_globals::imp_region_emi[i].num_IDs;
152 rs[i].subregtags =
new std::string[rs[i].nsubregs];
153 for (
int j = 0; j < rs[i].nsubregs;j++) {
154 std::string t1t2 = param_globals::imp_region_emi[i].ID[j];
155 size_t colon_pos = t1t2.find(
':');
156 if (colon_pos == std::string::npos) {
157 std::cerr <<
"Error: ':' the face tags are not defined properly in input file, it should be tag1:tag2 as a string" << std::endl;
161 rs[i].subregtags[j] = param_globals::imp_region_emi[i].ID[j];
166 region_mask_onFace(ion_domain, tags_data, line_face, tri_face, quad_face,
167 map_vertex_tag_to_dof, map_elem_uniqueFace_to_tags, intra_tags,
168 rs, reg_mask,
true,
"imp_region_emi");
170 for (
size_t i = 0; i < rs.size(); i++ ) {
171 delete[] rs[i].subregtags;
172 rs[i].subregtags =
nullptr;
176 tstart = setup_MIIF(loc_size, num_imp_regions, param_globals::imp_region_emi,
177 reg_mask.
data(), param_globals::start_statef, param_globals::num_adjustments,
178 param_globals::adjustment, param_globals::dt, purkfLen > 0);
180 miif->extUpdateVm = !param_globals::operator_splitting;
185 log_msg(logger, 0, 0,
"Changing simulation start time to %.2lf", tstart);
192 this->initialize_time +=
timing(t2, t1);
195 double IonicsOnFace::setup_MIIF(
limpet::node_count_t nnodes,
int nreg, IMPregion_EMI* impreg,
int* mask,
196 const char *start_fn,
int numadjust, IMPVariableAdjustment *adjust,
197 double time_step,
bool close)
202 miif->numNode = nnodes;
204 miif->numplugs = (
int*)calloc( miif->N_IIF,
sizeof(
int));
205 miif->plugtypes = std::vector<limpet::IonTypeList>(miif->N_IIF);
208 log_msg(logger,0,
ECHO,
"\nSetting up ionic models on EMI Face and plugins\n" \
209 "-----------------------------------\n\n" \
210 "Assigning IMPS to tagged regions:" );
212 for (
int i=0;i<miif->N_IIF;i++) {
216 miif->iontypes.push_back(*pT);
217 log_msg(logger, 0,
ECHO|
NONL,
"\tIonic model: %s to tag region(s)", impreg[i].im);
219 if(impreg[i].num_IDs > 0) {
220 for(
int j = 0; j < impreg[i].num_IDs; j++)
229 log_msg(NULL,5,
ECHO,
"Illegal IM specified: %s\n", impreg[i].im );
230 log_msg(NULL,5,
ECHO,
"Run bench --list-imps for a list of all available models.\n" );
235 if(impreg[i].plugins[0] !=
'\0') {
236 log_msg(logger,0,
ECHO|
NONL,
"\tPlug-in(s) : %s to tag region(s)", impreg[i].plugins);
238 for(
int j = 0; j < impreg[i].num_IDs; j++)
244 log_msg(NULL,5,
ECHO,
"Illegal plugin specified: %s\n", impreg[i].plugins);
245 log_msg(NULL,5,
ECHO,
"Run bench --list-imps for a list of all available plugins.\n" );
259 miif->initialize_MIIF();
261 for (
int i=0;i<miif->N_IIF;i++) {
263 remove_char(impreg[i].im_param, strlen(impreg[i].im_param),
' ');
264 miif->IIF[i]->tune(impreg[i].im_param, impreg[i].plugins, impreg[i].plug_param);
268 miif->initialize_currents(time_step, param_globals::ode_fac);
273 for (
int i=0;i<miif->N_IIF;i++) {
274 if (impreg[i].im_sv_init && strlen(impreg[i].im_sv_init) > 0)
275 if (
read_sv(miif, i, impreg[i].im_sv_init)) {
276 log_msg(NULL, 5,
ECHO|
FLUSH,
"State vector initialization failed for %s.\n", impreg[i].name);
281 if( !start_fn || strlen(start_fn)>0 )
282 tstart = (double) miif->restore_state(start_fn, ion_domain, close);
284 for (
int i=0; i<numadjust; i++)
290 bool restrict_to_algebraic =
true;
293 std::map<std::string,std::string> metadata;
297 if(metadata.count(
"grid") && metadata[
"grid"].compare(
"intra") == 0) {
300 read_indices_with_data(indices, values, adjust[i].file, imesh, nbr, restrict_to_algebraic, 1, PETSC_COMM_WORLD);
304 for(
size_t gi = 0; gi < indices.
size(); gi++)
305 indices[gi] = SF::local_nodal_to_local_petsc<mesh_int_t, mesh_real_t>(imesh, rank, indices[gi]);
311 adjPars->set(indices, values,
false,
true);
315 snprintf(fname,
sizeof fname,
"adj_%s_perm.dat", adjust[i].variable);
316 adjPars->write_ascii(fname,
false);
321 log_msg(0,3,0,
"%s warning: PETSC_TO_CANONICAL permutation needed registering!", __func__);
325 (*sc)(*adjPars,
true);
326 snprintf(fname,
sizeof fname,
"adj_%s_canonical.dat", adjust[i].variable);
327 adjPars->write_ascii(fname,
false);
330 int nc = miif->adjust_MIIF_variables(adjust[i].variable, indices, values);
331 log_msg(logger, 0, 0,
"Adjusted %d values for %s", nc, adjust[i].variable);
336 for (
int i=0;i<miif->N_IIF;i++)
337 initialize_sv_dumps_onFace(miif, impreg+i, i, tstart, param_globals::spacedt);
353 void initialize_sv_dumps_onFace(
limpet::MULTI_IF *pmiif, IMPregion_EMI* reg,
int id,
double t,
double dump_dt)
355 char svs[1024], plgs[1024], plgsvs[1024], fname[1024];
357 strcpy(svs, reg->im_sv_dumps ? reg->im_sv_dumps :
"");
358 strcpy(plgs, reg->plugins ? reg->plugins :
"");
359 strcpy(plgsvs, reg->plug_sv_dumps ? reg->plug_sv_dumps :
"");
361 if( !(strlen(svs)+strlen(plgsvs) ) )
370 strcpy(fname, param_globals::vofile);
373 log_msg(NULL, 5,
ECHO,
"%s: a region name must be specified\n", __func__ );
378 size_t fname_len = strlen(fname);
379 char* ext_start = fname + fname_len;
380 if(fname_len >= 4 && strcmp(ext_start - 4,
".igb") == 0) ext_start -= 3;
381 strcpy(ext_start, reg->name);
397 const char* gridname,
const char* reglist)
399 bool AllTagsExist =
true;
402 tagset.
insert(tags_data.begin(), tags_data.end());
405 for (
size_t reg=0; reg<regspec.
size(); reg++)
407 for (
int k=0; k<regspec[reg].nsubregs; k++) {
410 size_t colon_pos = regspec[reg].subregtags[k].find(
':');
411 int tag1 = std::stoi(regspec[reg].subregtags[k].substr(0, colon_pos));
412 int tag2 = std::stoi(regspec[reg].subregtags[k].substr(colon_pos + 1));
415 std::string inverse_pair;
416 inverse_pair = std::to_string(tag2) +
":" + std::to_string(tag1);
417 if(tagset.
count(regspec[reg].subregtags[k]) || tagset.
count(inverse_pair)) {
426 "on face Region tag %s in %s[%d] not found in element list for %s grid.\n",
427 regspec[reg].subregtags[k].c_str(), reglist, reg, gridname);
428 AllTagsExist =
false;
433 log_msg(NULL, 4,
ECHO,
"Assigned wrong pair of tags on the face of EMI surface mesh!\n"
434 "Check region ID specs in input files!\n");
440 inline bool pair_is_gapjunction(
const std::pair<mesh_int_t, mesh_int_t>& tags,
443 return intra_tags.
find(tags.first) != intra_tags.
end() &&
444 intra_tags.
find(tags.second) != intra_tags.
end();
447 void region_mask_onFace(
mesh_t meshspec,
448 std::vector<std::string> & tags_data,
464 if(regspec.
size() == 1)
return;
470 size_t rIDsize = mask_elem ? mesh.l_numelem : mesh.l_numpts;
472 size_t nelem = mesh.l_numelem;
475 check_tags_in_elems_onFace(tags_data, regspec, mesh.name.c_str(), reglist);
477 regionIDs.
assign(rIDsize, 0);
479 int* rid = regionIDs.
data();
484 for (
size_t reg=0; reg < regspec.
size(); reg++) {
486 for (
int k=0; k < regspec[reg].nsubregs; k++) {
487 std::string curtag = regspec[reg].subregtags[k];
488 if(tag_to_reg.
count(curtag)) err++;
491 log_msg(0,4,0,
"%s warning: Tag idx %s is assigned to multiple regions!\n"
492 "Its final assignment will be to the highest assigned region ID!",
493 __func__, curtag.c_str());
495 tag_to_reg[curtag] = reg;
499 std::vector<int> mx_tag( rIDsize, -1 );
502 for(
size_t eidx=0; eidx<nelem; eidx++)
504 std::vector<int> elem_nodes;
506 std::string result_pair_orginal;
507 std::string result_pair_reverse;
508 std::pair<mesh_int_t,mesh_int_t> value = map_elem_uniqueFace_to_tags[eidx];
509 result_pair_orginal = std::to_string(value.first) +
":" + std::to_string(value.second);
510 result_pair_reverse = std::to_string(value.second) +
":" + std::to_string(value.first);
511 rid[eidx] = pair_is_gapjunction(value, intra_tags) ? 1 : 0;
513 if (tag_to_reg.
count(result_pair_orginal)) {
514 rid[eidx] = tag_to_reg[result_pair_orginal];
515 }
else if(tag_to_reg.
count(result_pair_reverse)){
516 rid[eidx] = tag_to_reg[result_pair_reverse];
518 if(tag!=value.first){
519 log_msg(NULL, 5,
ECHO,
"error in tags on surface mesh with the unique faces!!!.\\n");
527 double IonicsOnFace::timer_val(
const int timer_id)
529 double val = std::nan(
"NaN");
535 std::string IonicsOnFace::timer_unit(
const int timer_id)
543 if (impdata[limpet::Iion] != NULL)
544 impdata[limpet::Iion][n] = 0;
548 imp.
compute(n, n + 1, impdata);
563 void* find_SV_in_IMP_onFace(
limpet::MULTI_IF* miif,
const int idx,
const char *IMP,
const char *SV,
564 int* offset,
int* sz)
566 if(strcmp(IMP, miif->
iontypes[idx].get().get_name().c_str()) == 0) {
567 return (
void*) miif->
iontypes[idx].get().get_sv_offset(SV, offset, sz);
570 for(
int k=0; k<miif->
numplugs[idx]; k++ )
571 if(strcmp(IMP, miif->
plugtypes[idx][k].get().get_name().c_str()) == 0) {
572 return (
void*) miif->
plugtypes[idx][k].get().get_sv_offset(SV, offset, sz);
590 void alloc_gvec_data_onFace(
const int nGVcs,
const int nRegs,
const int nEmiRegs,
591 GVecs *prmGVecs, gvec_data_OnFace &glob_vecs)
593 glob_vecs.nRegs = nRegs;
596 glob_vecs.vecs.resize(nGVcs);
598 for (
size_t i = 0; i < glob_vecs.vecs.size(); i++) {
599 sv_data_onFace &gvec = glob_vecs.vecs[i];
601 gvec.name =
dupstr(prmGVecs[i].name);
603 gvec.bogus = prmGVecs[i].bogus;
605 gvec.imps = (
char**) calloc(nRegs,
sizeof(
char *));
606 gvec.svNames = (
char**) calloc(nRegs,
sizeof(
char *));
607 gvec.svSizes = (
int*) calloc(nRegs,
sizeof(
int));
608 gvec.svOff = (
int*) calloc(nRegs,
sizeof(
int));
611 for (
int j = 0; j < nRegs; j++) {
612 if (strlen(prmGVecs[i].imp)) gvec.imps[j] =
dupstr(prmGVecs[i].imp);
614 else if (j < nEmiRegs)
615 gvec.imps[j] =
dupstr(param_globals::imp_region_emi[j].im);
617 gvec.imps[j] =
dupstr(param_globals::PurkIon[j - nEmiRegs].im);
619 else if (j < nEmiRegs)
620 gvec.imps[j] =
dupstr(param_globals::imp_region_emi[j].im);
622 gvec.svNames[j] =
dupstr(prmGVecs[i].ID[j]);
639 igb_output_manager & output_manager)
643 int num_purk_regions = 0;
644 int nEmiRegs = effective_num_imp_regions_emi();
645 int nRegs = nEmiRegs + num_purk_regions;
647 alloc_gvec_data_onFace(param_globals::num_gvecs, nRegs, nEmiRegs, param_globals::gvec, GVs);
650 if (GVs->inclPS) sample_PS_ionSVs(purk);
653 for (
unsigned int i = 0; i < GVs.vecs.size(); i++) {
654 sv_data_onFace & gv = GVs.vecs[i];
657 for (
int j = 0; j < miif->
N_IIF; j++) {
658 gv.getsv[j] = find_SV_in_IMP_onFace(miif, j, gv.imps[j], gv.svNames[j],
659 gv.svOff + j, gv.svSizes + j);
661 if (gv.getsv[j] == NULL) {
662 log_msg(NULL, 3,
ECHO,
"\tWarning: SV(%s) not found in region %d\n", gv.svNames[j], j);
668 output_manager.register_output(gv.ordered,
intra_elec_msh, 1, gv.name, gv.units);
674 MULTI_IF* pmiif = &purk->ion;
675 for (
int j = miif->
N_IIF; j < nRegs; j++) {
676 gv.getsv[j] = find_SV_in_IMP_onFace(pmiif, j - miif->
N_IIF, gv.imps[j],
677 gv.svNames[j], gv.svOff + j, gv.svSizes + j);
679 if (gv.getsv[j] == NULL) {
680 LOG_MSG(NULL, 3,
ECHO,
"\tWarning: state variable \"%s\" not found in region %d\n", gv.svNames[j], j);
684 RVector_dup(purk->vm_pt, &gv.orderedPS_PS);
687 RVector_dup(purk->vm_pt_over, &gv.orderedPS);
688 initialize_grid_output(grid, NULL, tmo,
intra_elec_msh, GRID_WRITE, 0., 1., gv.units, gv.orderedPS,
689 1, gv.GVcName, -purk->npt, param_globals::output_level);
694 log_msg(NULL, 5,
ECHO,
"\tError: no state variables found for global vector %d\n", i);
695 log_msg(NULL, 5,
ECHO,
"Run bench --imp=YourModel --imp-info to get a list of all parameters.\\n");
710 void assemble_sv_gvec_onFace(gvec_data_OnFace & gvecs,
limpet::MULTI_IF *miif)
712 for(
size_t i=0; i<gvecs.vecs.size(); i++ ) {
713 sv_data_onFace & gv = gvecs.vecs[i];
716 gv.ordered->set(gv.bogus);
718 gv.ordered->get_ownership_range(start, stop);
720 for(
int n = 0; n<miif->
N_IIF; n++ ) {
721 if( !gv.getsv[n] )
continue;
727 indices[j] = miif->
NodeLists[n][j] + start;
732 gv.ordered->set(indices, data, add);
738 RVector_set( gv.orderedPS, gv.bogus );
740 if(IS_PURK_PROC(purk))
742 MULTI_IF *pmiif = &purk->ion;
743 Real *data =
new Real[purk->gvec_cab.nitems];
746 for(
int n = 0; n<pmiif->N_IIF; n++ ) {
747 int gvidx = n+miif->
N_IIF;
748 if( !gv.getsv[gvidx] )
continue;
749 for(
int j=0; j<purk->gvec_ion[n].nitems; j++ )
750 data[ci++] = ((
SVgetfcn)(gv.getsv[gvidx]))( pmiif->IIF+n, ((
int*)(purk->gvec_ion[n].data))[j],
753 RVector_setvals(gv.orderedPS, purk->gvec_cab.nitems, (
int*)purk->gvec_cab.data, data,
true);
756 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.