openCARP
Doxygen code documentation for the open cardiac electrophysiology simulator openCARP
ionicsOnFace.cc
Go to the documentation of this file.
1 // ----------------------------------------------------------------------------
2 // openCARP is an open cardiac electrophysiology simulator.
3 //
4 // Copyright (C) 2020 openCARP project
5 //
6 // This program is licensed under the openCARP Academic Public License (APL)
7 // v1.0: You can use and redistribute it and/or modify it in non-commercial
8 // academic environments under the terms of APL as published by the openCARP
9 // project v1.0, or (at your option) any later version. Commercial use requires
10 // a commercial license (info@opencarp.org).
11 //
12 // This program is distributed without any warranty; see the openCARP APL for
13 // more details.
14 //
15 // You should have received a copy of the openCARP APL along with this program
16 // and can find it online: http://www.opencarp.org/license
17 // ----------------------------------------------------------------------------
18 
27 #if WITH_EMI_MODEL
28 #include "ionicsOnFace.h"
29 
30 #include "SF_init.h"
31 
32 namespace opencarp {
33 
34 void initialize_sv_dumps_onFace(limpet::MULTI_IF *pmiif, IMPregion_EMI* reg, int id, double t, double dump_dt);
35 
36 namespace {
37 
38 // Region 0 and 1 are default EMI face regions and are valid when an ionic
39 // model is assigned. User-defined regions 2+ must additionally provide at
40 // least one face-tag pair.
41 bool imp_region_emi_is_assigned(const IMPregion_EMI& region, int idx)
42 {
43  if (!(region.im && strlen(region.im) > 0)) return false;
44  if (idx < 2) return true;
45  return region.num_IDs > 0;
46 }
47 
48 // Ignore trailing unassigned imp_region_emi entries. This lets users keep the
49 // generated default parameter count while only assigning the regions they use.
50 int effective_num_imp_regions_emi()
51 {
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)) {
55  --nreg;
56  }
57 
58  if (nreg < configured_nreg) {
59  static bool warned_once = false;
60  int rank = 0;
61  MPI_Comm_rank(MPI_COMM_WORLD, &rank);
62  if (!warned_once && rank == 0) {
63  log_msg(NULL, 1, ECHO,
64  "Warning: num_imp_regions=%d but only %d imp_region_emi entries are assigned; ignoring trailing unassigned regions.\n",
65  configured_nreg, nreg);
66  warned_once = true;
67  }
68  }
69 
70  return nreg;
71 }
72 
73 } // namespace
74 
75 void IonicsOnFace::compute_step()
76 {
77  double t1, t2;
78  get_time(t1);
79 
80  miif->compute_ionic_current();
81 
82  double comp_time = timing(t2, t1);
83  this->compute_time += comp_time;
84 
85  comp_stats.calls++;
86  comp_stats.tot_time += comp_time;
87 
89  comp_stats.log_stats(user_globals::tm_manager->time, false);
90 }
91 
92 void IonicsOnFace::destroy()
93 {
94  miif->free_MIIF();
95 }
96 
97 void IonicsOnFace::output_step()
98 {}
99 
100 void IonicsOnFace::initialize()
101 {
102  double t1, t2;
103  get_time(t1);
104 
105  set_dir(OUTPUT);
106 
107  // initialize generic logger for ODE timings per time_dt
108  comp_stats.init_logger("ODE_stats.dat");
109 
110  double tstart = 0.;
111  sf_mesh & mesh = get_mesh(ion_domain);
112  limpet::node_count_t loc_size = mesh.l_numelem;
113 
114  // create ionic current vector and register it
115  sf_vec *IIon;
116  sf_vec *Vmv;
117  SF::init_vector(&IIon, mesh, 1, sf_vec::elemwise);
118  SF::init_vector(&Vmv, mesh, 1, sf_vec::elemwise);
121 
122  // setup miif
123  miif = new limpet::MULTI_IF();
124  // store IIF_IDs and Plugins in arrays
125  miif->name = "myocardium";
126  miif->gdata[limpet::Vm] = Vmv;
127  miif->gdata[limpet::Iion] = IIon;
128 
129  const int num_imp_regions = effective_num_imp_regions_emi();
130  SF::vector<RegionSpecs_EMI> rs(num_imp_regions);
131 
132  if (rs.size() <2) {
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");
135  exit(1);
136  }
137 
138  // Region 0 and 1 are implicit defaults:
139  // - region 0: default membrane faces
140  // - region 1: default gap-junction faces
141  rs[0].nsubregs = 0;
142  rs[0].subregtags = nullptr;
143  rs[1].nsubregs = 0;
144  rs[1].subregtags = nullptr;
145 
146  // imp_region_emi[2+] assign models to specific face tag pairs like "t1:t2"
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]; // Allocate memory for the string array
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;
155  exit(1);
156  }
157 
158  rs[i].subregtags[j] = param_globals::imp_region_emi[i].ID[j]; // Convert char* to std::string.
159  }
160  }
161 
162  SF::vector<int> reg_mask;
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");
166 
167  for (size_t i = 0; i < rs.size(); i++ ) {
168  delete[] rs[i].subregtags;
169  rs[i].subregtags = nullptr;
170  }
171 
172  int purkfLen = 1;
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);
176 
177  miif->extUpdateVm = !param_globals::operator_splitting;
178 
179  // if we start at a non-zero time (i.e. we have restored a state), we notify the
180  // timer manager
181  if(tstart > 0.) {
182  log_msg(logger, 0, 0, "Changing simulation start time to %.2lf", tstart);
183  user_globals::tm_manager->setup(param_globals::dt, tstart, param_globals::tend);
185  }
186 
187  set_dir(INPUT);
188 
189  this->initialize_time += timing(t2, t1);
190 }
191 
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)
195 {
196  double tstart = 0;
197 
198  miif->N_IIF = nreg;
199  miif->numNode = nnodes;
200  miif->iontypes = {};
201  miif->numplugs = (int*)calloc( miif->N_IIF, sizeof(int));
202  miif->plugtypes = std::vector<limpet::IonTypeList>(miif->N_IIF);
203  miif->targets = std::vector<limpet::Target>(miif->N_IIF, limpet::Target::AUTO);
204 
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:" );
208 
209  for (int i=0;i<miif->N_IIF;i++) {
210  auto pT = limpet::get_ion_type(std::string(impreg[i].im));
211  if (pT != NULL)
212  {
213  miif->iontypes.push_back(*pT);
214  log_msg(logger, 0, ECHO|NONL, "\tIonic model: %s to tag region(s)", impreg[i].im);
215 
216  if(impreg[i].num_IDs > 0) {
217  for(int j = 0; j < impreg[i].num_IDs; j++)
218  log_msg(logger,0,ECHO|NONL, " [%s],", impreg[i].ID[j]);
219  log_msg(logger,0,ECHO,"\b.");
220  }
221  else {
222  log_msg(logger,0,ECHO, " [0] (implicitely)");
223  }
224  }
225  else {
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" );
228  EXIT(1);
229  }
230  if (limpet::get_plug_flag( impreg[i].plugins, &miif->numplugs[i], miif->plugtypes[i]))
231  {
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);
234 
235  for(int j = 0; j < impreg[i].num_IDs; j++)
236  log_msg(logger,0,ECHO|NONL, " [%d],", impreg[i].ID[j]);
237  log_msg(logger,0,ECHO,"\b.");
238  }
239  }
240  else {
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" );
243  EXIT(1);
244  }
245  }
246 
247  miif->IIFmask = (limpet::IIF_Mask_t*)calloc(miif->numNode, sizeof(limpet::IIF_Mask_t));
248 
249  // The mask is element-wise on the EMI unique-face mesh: one ionic/gap-junction
250  // region id per local unique-face element.
251  if (mask) {
252  for (limpet::node_index_t i=0; i<miif->numNode; i++)
253  miif->IIFmask[i] = (limpet::IIF_Mask_t) mask[i];
254  }
255 
256  miif->initialize_MIIF();
257 
258  for (int i=0;i<miif->N_IIF;i++) {
259  // the IMP tuning does not handle spaces well, thus we remove them here
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);
262  }
263 
264  set_dir(INPUT);
265  miif->initialize_currents(time_step, param_globals::ode_fac);
266 
267  // overriding initial values goes here
268  // read in single cell state vector and spread it out over the entire region
269  set_dir(INPUT);
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);
274  EXIT(-1);
275  }
276  }
277 
278  if( !start_fn || strlen(start_fn)>0 )
279  tstart = (double) miif->restore_state(start_fn, ion_domain, close);
280 
281  for (int i=0; i<numadjust; i++)
282  {
283  set_dir(INPUT);
284 
285  SF::vector<SF_int> indices;
286  SF::vector<double> values;
287  bool restrict_to_algebraic = true;
288 
289  sf_mesh & imesh = get_mesh(ion_domain);
290  std::map<std::string,std::string> metadata;
291  read_metadata(adjust[i].file, metadata, PETSC_COMM_WORLD);
292 
293  SF::SF_nbr nbr = SF::NBR_REF;
294  if(metadata.count("grid") && metadata["grid"].compare("intra") == 0) {
295  nbr = SF::NBR_SUBMESH;
296  }
297  read_indices_with_data(indices, values, adjust[i].file, imesh, nbr, restrict_to_algebraic, 1, PETSC_COMM_WORLD);
298 
299  int rank = get_rank();
300 
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]);
303 
304  // debug, output parameters on global intracellular vector
305  if(adjust[i].dump) {
306  sf_vec* adjPars;
307  SF::init_vector(&adjPars, imesh, 1, sf_vec::algebraic);
308  adjPars->set(indices, values, false, true);
309 
310  set_dir(OUTPUT);
311  char fname[2085];
312  snprintf(fname, sizeof fname, "adj_%s_perm.dat", adjust[i].variable);
313  adjPars->write_ascii(fname, false);
314 
315  // get the scattering to the canonical permutation
316  SF::scattering* sc = get_permutation(ion_domain, PETSC_TO_CANONICAL, 1);
317  if(sc == NULL) {
318  log_msg(0,3,0, "%s warning: PETSC_TO_CANONICAL permutation needed registering!", __func__);
319  sc = register_permutation(ion_domain, PETSC_TO_CANONICAL, 1);
320  }
321 
322  (*sc)(*adjPars, true);
323  snprintf(fname, sizeof fname, "adj_%s_canonical.dat", adjust[i].variable);
324  adjPars->write_ascii(fname, false);
325  }
326 
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);
329  }
330 
331  set_dir(OUTPUT);
332 
333  for (int i=0;i<miif->N_IIF;i++)
334  initialize_sv_dumps_onFace(miif, impreg+i, i, tstart, param_globals::spacedt);
335 
336  return tstart;
337 }
338 
350 void initialize_sv_dumps_onFace(limpet::MULTI_IF *pmiif, IMPregion_EMI* reg, int id, double t, double dump_dt)
351 {
352  char svs[1024], plgs[1024], plgsvs[1024], fname[1024];
353 
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 : "");
357 
358  if( !(strlen(svs)+strlen(plgsvs) ) )
359  return;
360 
361  /* The string passed to the "reg_name" argument (#4) of the sv_dump_add
362  * function is supposed to be "region name". It's only purpose is to
363  * provide the base name for the SV dump file, eg: Purkinje.Ca_i.bin.
364  * Thus, we pass: [vofile].[reg name], Otherwise, dumping SVs in
365  * batched runs would be extremely tedious.
366  */
367  strcpy(fname, param_globals::vofile); // [vofile].igb.gz
368 
369  if( !reg->name ) {
370  log_msg(NULL, 5, ECHO, "%s: a region name must be specified\n", __func__ );
371  exit(0);
372  }
373 
374  // We want to convert vofile.igb or vofile.igb.gz to vofile.regname
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);
380 
381  pmiif->sv_dump_add_by_name_list(id, reg->im, fname, svs, plgs, plgsvs, t, dump_dt);
382 }
383 
394 bool check_tags_in_elems_onFace(std::vector<std::string> & tags_data, SF::vector<RegionSpecs_EMI> & regspec,
395  const char* gridname, const char* reglist)
396 {
397  bool AllTagsExist = true;
399 
400  tagset.insert(tags_data.begin(), tags_data.end());
401 
402  // cycle through all user-specified regions
403  for (size_t reg=0; reg<regspec.size(); reg++)
404  // cycle through all tags which belong to the region
405  for (int k=0; k<regspec[reg].nsubregs; k++) {
406  // check whether this tag exists in element list
407  int n = 0;
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));
411 
412  // considered the other combinations of tags
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)) {
416  n++;
417  }
418 
419  // globalize n
420  int N = get_global(n, MPI_SUM);
421 
422  if (N==0) {
423  log_msg(NULL, 3, ECHO,
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;
427  }
428  }
429 
430  if (!AllTagsExist) {
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");
433  }
434 
435  return AllTagsExist;
436 }
437 
438 inline bool pair_is_gapjunction(const std::pair<mesh_int_t, mesh_int_t>& tags,
439  const hashmap::unordered_set<int>& intra_tags)
440 {
441  return intra_tags.find(tags.first) != intra_tags.end() &&
442  intra_tags.find(tags.second) != intra_tags.end();
443 }
444 
445 void region_mask_onFace(mesh_t meshspec,
446  std::vector<std::string> & tags_data,
448  std::pair<SF::emi_face<mesh_int_t,SF::tuple<mesh_int_t>>,
449  SF::emi_face<mesh_int_t,SF::tuple<mesh_int_t>>>> & line_face,
451  std::pair<SF::emi_face<mesh_int_t,SF::triple<mesh_int_t>>,
452  SF::emi_face<mesh_int_t,SF::triple<mesh_int_t>>>> & tri_face,
454  std::pair<SF::emi_face<mesh_int_t,SF::quadruple<mesh_int_t>>,
455  SF::emi_face<mesh_int_t,SF::quadruple<mesh_int_t>>>> & quad_face,
456  hashmap::unordered_map<std::pair<mesh_int_t,mesh_int_t>, mesh_int_t> & map_vertex_tag_to_dof,
457  hashmap::unordered_map<mesh_int_t, std::pair<mesh_int_t, mesh_int_t>> & map_elem_uniqueFace_to_tags,
458  const hashmap::unordered_set<int> & intra_tags,
459  SF::vector<RegionSpecs_EMI> & regspec,
460  SF::vector<int> & regionIDs, bool mask_elem, const char* reglist)
461 {
462  if(regspec.size() == 1) return;
463 
464  sf_mesh & mesh = get_mesh(meshspec);
466 
467  // Initialize the list with the default membrane region id, 0.
468  size_t rIDsize = mask_elem ? mesh.l_numelem : mesh.l_numpts;
469 
470  size_t nelem = mesh.l_numelem;
471 
472  // Check whether all specified face-tag pairs exist in the unique-face list.
473  check_tags_in_elems_onFace(tags_data, regspec, mesh.name.c_str(), reglist);
474 
475  regionIDs.assign(rIDsize, 0);
476 
477  int* rid = regionIDs.data();
479 
480  // Generate a map from face-tag pair strings to region IDs. This also lets us
481  // detect duplicate assignments and simplifies the regionIDs fill loop.
482  for (size_t reg=0; reg < regspec.size(); reg++) {
483  int err = 0;
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++;
487 
488  if(get_global(err, MPI_SUM))
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());
492 
493  tag_to_reg[curtag] = reg;
494  }
495  }
496 
497  std::vector<int> mx_tag( rIDsize, -1 );
498 
499  // Cycle through unique-face elements and assign default or user-defined regions.
500  for(size_t eidx=0; eidx<nelem; eidx++)
501  {
502  std::vector<int> elem_nodes;
503  mesh_int_t tag = mesh.tag[eidx];
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;
510 
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];
515  }
516  if(tag!=value.first){
517  log_msg(NULL, 5, ECHO, "error in tags on surface mesh with the unique faces!!!.\\n");
518  exit(1);
519  }
520  }
521 }
522 
525 double IonicsOnFace::timer_val(const int timer_id)
526 {
527  double val = std::nan("NaN");
528  return val;
529 }
530 
533 std::string IonicsOnFace::timer_unit(const int timer_id)
534 {
535  std::string s_unit;
536  return s_unit;
537 }
538 
539 void compute_IIF_OnFace(limpet::IonIfBase& pIF, limpet::GlobalData_t** impdata, limpet::node_index_t n)
540 {
541  if (impdata[limpet::Iion] != NULL)
542  impdata[limpet::Iion][n] = 0;
543 
544  pIF.for_each([&](limpet::IonIfBase& imp) {
545  update_ts(&imp.get_tstp());
546  imp.compute(n, n + 1, impdata);
547  });
548 }
549 
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)
563 {
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);
566  }
567  else {
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);
571  }
572  }
573 
574  return NULL;
575 }
576 
577 
588 void alloc_gvec_data_onFace(const int nGVcs, const int nRegs, const int nEmiRegs,
589  GVecs *prmGVecs, gvec_data_OnFace &glob_vecs)
590 {
591  glob_vecs.nRegs = nRegs;
592 
593  if (nGVcs) {
594  glob_vecs.vecs.resize(nGVcs);
595 
596  for (size_t i = 0; i < glob_vecs.vecs.size(); i++) {
597  sv_data_onFace &gvec = glob_vecs.vecs[i];
598 
599  gvec.name = dupstr(prmGVecs[i].name);
600  gvec.units = dupstr(prmGVecs[i].units);
601  gvec.bogus = prmGVecs[i].bogus;
602 
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));
607  gvec.getsv = (void**) calloc(nRegs, sizeof(limpet::SVgetfcn));
608 
609  for (int j = 0; j < nRegs; j++) {
610  if (strlen(prmGVecs[i].imp)) gvec.imps[j] = dupstr(prmGVecs[i].imp);
611 #ifdef WITH_PURK
612  else if (j < nEmiRegs)
613  gvec.imps[j] = dupstr(param_globals::imp_region_emi[j].im);
614  else
615  gvec.imps[j] = dupstr(param_globals::PurkIon[j - nEmiRegs].im);
616 #else
617  else if (j < nEmiRegs)
618  gvec.imps[j] = dupstr(param_globals::imp_region_emi[j].im);
619 #endif
620  gvec.svNames[j] = dupstr(prmGVecs[i].ID[j]);
621  }
622  }
623  }
624 }
625 
636 void init_sv_gvec_onFace(gvec_data_OnFace& GVs, limpet::MULTI_IF* miif, sf_vec & tmpl,
637  igb_output_manager & output_manager)
638 {
639  GVs.inclPS = false;
640  // int num_purk_regions = GVs->inclPS ? purk->ion.N_IIF : 0;
641  int num_purk_regions = 0;
642  int nEmiRegs = effective_num_imp_regions_emi();
643  int nRegs = nEmiRegs + num_purk_regions;
644 
645  alloc_gvec_data_onFace(param_globals::num_gvecs, nRegs, nEmiRegs, param_globals::gvec, GVs);
646 
647 #ifdef WITH_PURK
648  if (GVs->inclPS) sample_PS_ionSVs(purk);
649 #endif
650 
651  for (unsigned int i = 0; i < GVs.vecs.size(); i++) {
652  sv_data_onFace & gv = GVs.vecs[i];
653  int noSV = 0;
654 
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);
658 
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);
661  noSV++;
662  }
663  }
664 
665  SF::init_vector(&gv.ordered, &tmpl);
666  output_manager.register_output(gv.ordered, intra_elec_msh, 1, gv.name, gv.units);
667 
668 #ifdef WITH_PURK
669  // same procedure for Purkinje
670  if (GVs->inclPS) {
671  IF_PURK_PROC(purk) {
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);
676 
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);
679  noSV++;
680  }
681  }
682  RVector_dup(purk->vm_pt, &gv.orderedPS_PS);
683  MYO_COMM(purk);
684  }
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);
688  }
689 #endif
690 
691  if (noSV == nRegs) {
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");
694  exit(1);
695  }
696  }
697 }
698 
708 void assemble_sv_gvec_onFace(gvec_data_OnFace & gvecs, limpet::MULTI_IF *miif)
709 {
710  for(size_t i=0; i<gvecs.vecs.size(); i++ ) {
711  sv_data_onFace & gv = gvecs.vecs[i];
712 
713  // Set to the default value.
714  gv.ordered->set(gv.bogus);
715  SF_int start, stop;
716  gv.ordered->get_ownership_range(start, stop);
717 
718  for( int n = 0; n<miif->N_IIF; n++ ) {
719  if( !gv.getsv[n] ) continue;
720 
721  SF::vector<SF_real> data (miif->N_Nodes[n]);
722  SF::vector<SF_int> indices(miif->N_Nodes[n]);
723 
724  for( limpet::node_index_t j=0; j<miif->N_Nodes[n]; j++ ) {
725  indices[j] = miif->NodeLists[n][j] + start;
726  data[j] = ((limpet::SVgetfcn)(gv.getsv[n]))( *miif->IIF[n], j, gv.svOff[n]);
727  }
728 
729  bool add = false;
730  gv.ordered->set(indices, data, add);
731  }
732 
733 #ifdef WITH_PURK
734  // include purkinje in sv dump, if exists
735  if(gvecs->inclPS) {
736  RVector_set( gv.orderedPS, gv.bogus ); // set this to the default value
737 
738  if(IS_PURK_PROC(purk))
739  {
740  MULTI_IF *pmiif = &purk->ion;
741  Real *data = new Real[purk->gvec_cab.nitems];
742 
743  int ci=0;
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],
749  gv.svOff[gvidx]);
750  }
751  RVector_setvals(gv.orderedPS, purk->gvec_cab.nitems, (int*)purk->gvec_cab.data, data, true);
752  delete [] data;
753  }
754  RVector_sync( gv.orderedPS );
755  }
756 #endif
757  }
758 }
759 
760 
761 } // namespace opencarp
762 #endif
double Real
Definition: DataTypes.h:14
opencarp::local_index_t mesh_int_t
Definition: SF_container.h:46
opencarp::global_index_t SF_int
Global algebraic index type.
Definition: SF_globals.h:32
#define FLUSH
Definition: basics.h:319
#define ECHO
Definition: basics.h:316
#define NONL
Definition: basics.h:320
Comfort class. Provides getter functions to access the mesh member variables more comfortably.
Definition: SF_fem_utils.h:704
Container for a PETSc VecScatter.
A vector storing arbitrary data.
Definition: SF_vector.h:43
size_t size() const
The current size of the vector.
Definition: SF_vector.h:104
void assign(InputIterator s, InputIterator e)
Assign a memory range.
Definition: SF_vector.h:161
T * data()
Pointer to the vector's start.
Definition: SF_vector.h:91
hm_int count(const K &key) const
Check if key exists.
Definition: hashmap.hpp:627
Custom unordered_set implementation.
Definition: hashmap.hpp:754
iterator find(const K &key)
Definition: hashmap.hpp:1096
hm_int count(const K &key) const
Definition: hashmap.hpp:1082
void insert(InputIterator first, InputIterator last)
Definition: hashmap.hpp:1052
Represents the ionic model and plug-in (IMP) data structure.
Definition: ION_IF.h:139
void compute(node_index_t start, node_index_t end, GlobalData_t **data)
Perform ionic model computation for 1 time step.
Definition: ION_IF.cc:270
ts & get_tstp()
Gets the time stepper.
Definition: ION_IF.cc:208
void for_each(const std::function< void(IonIfBase &)> &consumer)
Executes the consumer functions on this IMP and each of its plugins.
Definition: ION_IF.cc:417
std::vector< IonIfBase * > IIF
array of IIF's
Definition: MULTI_ION_IF.h:202
void sv_dump_add_by_name_list(int, char *, char *, char *, char *, char *, double, double)
int * numplugs
number of plugins for each region
Definition: MULTI_ION_IF.h:209
std::vector< IonTypeList > plugtypes
plugins types for each region
Definition: MULTI_ION_IF.h:215
IonTypeList iontypes
type for each region
Definition: MULTI_ION_IF.h:212
int N_IIF
how many different IIF's
Definition: MULTI_ION_IF.h:211
node_count_t * N_Nodes
#nodes for each IMP
Definition: MULTI_ION_IF.h:200
node_index_t ** NodeLists
local partitioned node lists for each IMP stored
Definition: MULTI_ION_IF.h:201
void setup(double inp_dt, double inp_start, double inp_end)
Initialize the timer_manager.
Definition: timer_utils.cc:36
void reset_timers()
Reset time in timer_manager and then reset registered timers.
Definition: timer_utils.h:115
LIMPET ionics and gap-junction models on the EMI unique-face interface mesh.
void init_vector(SF::abstract_vector< T, S > **vec)
Definition: SF_init.h:107
SF_nbr
Enumeration encoding the different supported numberings.
Definition: SF_container.h:200
@ NBR_PETSC
PETSc numbering of nodes.
Definition: SF_container.h:203
@ NBR_REF
The nodal numbering of the reference mesh (the one stored on HD).
Definition: SF_container.h:201
@ NBR_SUBMESH
Submesh nodal numbering: The globally ascending sorted reference indices are reindexed.
Definition: SF_container.h:202
int get_plug_flag(char *plgstr, int *out_num_plugins, IonTypeList &out_plugins)
@ AUTO
Definition: target.h:46
IonType * get_ion_type(const std::string &name)
SF_real GlobalData_t
Definition: limpet_types.h:27
GlobalData_t(* SVgetfcn)(IonIfBase &, node_index_t, int)
Definition: ion_type.h:48
void update_ts(ts *ptstp)
Definition: ION_IF.cc:466
opencarp::local_index_t node_count_t
Definition: limpet_types.h:29
int read_sv(MULTI_IF *, int, const char *)
char IIF_Mask_t
Definition: ion_type.h:50
opencarp::local_index_t node_index_t
Definition: limpet_types.h:28
std::map< int, std::string > units
Definition: stimulate.cc:41
timer_manager * tm_manager
a manager for the various physics timers
Definition: main.cc:55
@ iotm_console
Definition: timer_utils.h:44
sf_mesh & get_mesh(const mesh_t gt)
Get a mesh by specifying the gridID.
Definition: sf_interface.cc:33
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
Definition: sf_interface.h:48
int set_dir(IO_t dest)
Definition: sim_utils.cc:1347
void read_metadata(const std::string filename, std::map< std::string, std::string > &metadata, MPI_Comm comm)
Read metadata from the header.
Definition: fem_utils.cc:72
int get_rank(MPI_Comm comm=PETSC_COMM_WORLD)
Definition: basics.h:284
T get_global(T in, MPI_Op OP, MPI_Comm comm=PETSC_COMM_WORLD)
Do a global reduction on a variable.
Definition: basics.h:233
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.
Definition: sim_utils.cc:1810
@ OUTPUT
Definition: sim_utils.h:54
char * dupstr(const char *old_str)
Definition: basics.cc:44
void log_msg(FILE_SPEC out, int level, unsigned char flag, const char *fmt,...)
Definition: basics.cc:72
mesh_t
The enum identifying the different meshes we might want to load.
Definition: sf_interface.h:59
@ intra_elec_msh
Definition: sf_interface.h:60
void get_time(double &tm)
Definition: basics.h:444
SF::abstract_vector< SF_int, SF_real > sf_vec
Definition: sf_interface.h:50
void remove_char(char *buff, const int buffsize, const char c)
Definition: basics.h:364
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
Definition: fem_utils.h:269
V timing(V &t2, const V &t1)
Definition: basics.h:456
#define PETSC_TO_CANONICAL
Permute algebraic data from PETSC to canonical ordering.
Definition: sf_interface.h:79