This tutorial is automatically generated from the file trunk/heart/test/TestRunningBidomainSimulationsTutorial.hpp at revision r7875. Note that the code is given in full at the bottom of the page.
An example showing how to run bidomain simulations (for monodomain, it is essentially the same)
Introduction
In this tutorial we show how Chaste is used to run a standard bidomain simulation. Note that monodomain simulations are run very similarly. The first thing that needs to be done, when writing any Chaste test, is to include the following header.
#include <cxxtest/TestSuite.h>
The main class to be used for running bidomain simulations is BidomainProblem.
#include "BidomainProblem.hpp"
The PlaneStimulusCellFactory is a useful class to include (see later).
#include "PlaneStimulusCellFactory.hpp"
LuoRudyIModel1991OdeSystem is the cell model which will be used in this simulation.
#include "LuoRudyIModel1991OdeSystem.hpp"
All tests which run cardiac simulations (which use Petsc) should include PetscSetupAndFinalize.hpp. This class ensures that PetscInitialise() is called with the appropriate arguments before any tests in the suite are run.
#include "PetscSetupAndFinalize.hpp"
Class used to to model a FEM mesh and helper class used to read it from file
#include "TetrahedralMesh.hpp" #include "TrianglesMeshReader.hpp"
Defining a cell factory
All mono/bidomain simulations need a cell factory as input. This is a class which tells the problem class what type of cardiac cells to create. The cell-factory class has to inherit from AbstractCardiacCellFactory<DIM>, which means it must implement the method CreateCardiacCellForTissueNode(unsigned nodeNum), which returns a pointer to an AbstractCardiacCell. Note, some concrete cell factories have been defined, such as the PlaneStimulusCellFactory, which could be used in the simulation, but for completeness we create our own cell factory in this test. For complicated problems with, say, heterogeneous cell types or particular stimuli, a new cell factory will have to be defined by the user for their particular problem.
This cell factory is a simple cell factory where every cell is a Luo-Rudy 91 cell, and only the cell at position (0,0) is given a non-zero stimulus.
class PointStimulus2dCellFactory : public AbstractCardiacCellFactory<2> {
Declare (smart) pointer to a SimpleStimulus for the cell which is stimulated. Note that AbstractCardiacCellFactory also has as protected members: mpZeroStimulus of type boost::shared_ptr<ZeroStimulus>; mpMesh, a pointer to the mesh used (the problem class will set this before it calls CreateCardiacCellForTissueNode, so it can be used in that method); mTimestep, a double (see below); and boost::shared_ptr<mpSolver> a forward euler ode solver (see below).
private: boost::shared_ptr<SimpleStimulus> mpStimulus; public:
Our contructor takes in nothing. It calls the constructor of AbstractCardiacCellFactory and we also initialise the stimulus to have magnitude 6000 (uA/cm3) and duration 0.5 (ms).
PointStimulus2dCellFactory() : AbstractCardiacCellFactory<2>(), mpStimulus(new SimpleStimulus(-6000.0, 0.5)) { }
Now we implement the pure method which needs to be implemented. We return a LR91 cell for each node, with the node at (0,0) given the non-zero stimulus, and all other nodes given the zero stimulus. Note that we use mpMesh, mTimestep, mpZeroStimulus and mpSolver which are all members of the base class. The timestep and solver are defined in the base class just so that the user doesn't have to create them here.
AbstractCardiacCell* CreateCardiacCellForTissueNode(unsigned nodeIndex) { double x = this->GetMesh()->GetNode(nodeIndex)->rGetLocation()[0]; double y = this->GetMesh()->GetNode(nodeIndex)->rGetLocation()[1]; if (fabs(x)+fabs(y)<1e-6) // ie if (x,y)==(0,0). An alternative would be if(norm_2(this->mpMesh->GetNode(nodeIndex)->rGetLocation())<1e-6) {
Even if running a bidomain simulation, only the intra-cellular stimulus should be given here. There is a separate Electrodes class for applying extra-cellular stimuli.
return new LuoRudyIModel1991OdeSystem(mpSolver, mpStimulus); } else {
The other cells have zero stimuli.
return new LuoRudyIModel1991OdeSystem(mpSolver, mpZeroStimulus); } }
We have no need for a destructor, since the problem class deals with deleting the cells.
};
Running a bidomain simulation
Now we can define the test class, which must inherit from CxxTest::TestSuite as usual.
class TestRunningBidomainSimulationsTutorial : public CxxTest::TestSuite {
Tests should be public...
public:
Define the test. Note the throw(Exception) - without this exception messages might not get printed out.
void TestSimpleSimulation() throw(Exception) {
The HeartConfig class is used to set various parameters. It gets the default values from ChasteDefaults.xml (in the base Chaste directory) (except the values in the 'Simulation' block of the XML file, which is only used by the Chaste executable). Parameters in this file can be re-set with HeartConfig if the user wishes, and other paramters such as end time must be set using HeartConfig. Let us begin by setting the end time (in ms), the mesh to use, and the output directory and filename-prefix.
HeartConfig::Instance()->SetSimulationDuration(1.0); //ms HeartConfig::Instance()->SetMeshFileName("mesh/test/data/square_128_elements"); HeartConfig::Instance()->SetOutputDirectory("BidomainTutorial"); HeartConfig::Instance()->SetOutputFilenamePrefix("results");
Next, we have to create a cell factory of the type we defined above.
PointStimulus2dCellFactory cell_factory;
Now we create a problem class using (a pointer to) the cell factory.
BidomainProblem<2> bidomain_problem( &cell_factory );
This is enough setup to run a simulation: we could now call Initialise() and Solve() to run...
// bidomain_problem.Initialise(); // bidomain_problem.Solve();
..However, instead we show how to set a few more parameters. To set the conductivity values in the principal fibre, sheet and normal directions do the following. Note that Create_c_vector is just a helper method for creating a c_vector<double,DIM> of the correct size (2, in this case). Make sure these methods are called before Initialise().
HeartConfig::Instance()->SetIntracellularConductivities(Create_c_vector(1.75, 0.19)); HeartConfig::Instance()->SetExtracellularConductivities(Create_c_vector(6.2, 2.4));
Let us also reset the surface-area-to-volume ratio and the capacitance
HeartConfig::Instance()->SetSurfaceAreaToVolumeRatio(1.0); // 1/cm HeartConfig::Instance()->SetCapacitance(1.0); // uF/cm^2
Now we call Initialise()...
bidomain_problem.Initialise();
The output will be written to /tmp/USER_NAME/testoutput/BidomainTutorial in hdf5 format. By default the output will be converted to meshalyzer format at the end of the simulation. To adjust this, or convert to Cmgui or VTK format instead, use methods in HeartConfig?, e.g.
HeartConfig::Instance()->SetVisualizeWithCmgui(true);
Now we call Solve() to run the simulation. Note that if you want to view the progress of longer simulations go to the the output directory and look at the file progress_status.txt, which will say the percentage of the simulation run. A useful linux command is therefore watch tail progress_status.txt which will repeatedly display the last few lines of this file.
bidomain_problem.Solve();
To now visualise the results, go to /tmp/USER_NAME/testoutput/BidomainTutorial/output, where you should find the mesh and output, and run meshalyzer.
The easiest way to look at the resultant voltage values (for the last timestep - the data for the previous timesteps is written to file but not retained) is to use a ReplicatableVector. bidomain_problem.GetSolution()) returns a PetSc vector of the form (V_0, phi_0, V_1, phi_e_1, ... V_n, phi_e_n), and we can create a ReplicatableVector for easy access to this PetSc vector's data. (This won't be very efficient with huge problems in parallel).
ReplicatableVector res_repl(bidomain_problem.GetSolution()); for(unsigned i=0; i<res_repl.GetSize(); i++) { // std::cout << res_repl[i] << "\n"; }
Alternatively, we show how to access the voltage values using the DistributedVector class, which can be used to only iterate over the values of the voltage owned by that process.
DistributedVector dist_bidomain_voltage = bidomain_problem.GetSolutionDistributedVector(); DistributedVector::Stripe bidomain_voltage(dist_bidomain_voltage, 0); DistributedVector::Stripe extracellular_potential(dist_bidomain_voltage, 1);
A loop over all the components owned by this process..
for (DistributedVector::Iterator index = dist_bidomain_voltage.Begin(); index != dist_bidomain_voltage.End(); ++index) {
.. and a simple test, that the 'last' node was stimulated:
if (index.Global==bidomain_problem.rGetMesh().GetNumNodes()-1) // ie if the last node { TS_ASSERT_LESS_THAN(0, bidomain_voltage[index]); } } }
Running a bidomain simulation with an external bath, and electrodes
Now, we illustrate how to run a simulation with an external bath and electrodes applying a boundary extracellular stimulus. Note that currently, bath problems can only be solved on rectangular/cuboid domains.
void TestWithBathAndElectrodes() throw (Exception) {
Important: we need to remember to reset the HeartConfig class, since it had reset various parameters in the previous test.
HeartConfig::Instance()->Reset();
First, set the end time and output info. In this simulation we'll explicitly read the mesh, alter it, then pass it to the problem class, so we don't set the mesh file name.
HeartConfig::Instance()->SetSimulationDuration(3.0); //ms HeartConfig::Instance()->SetOutputDirectory("BidomainTutorialWithBath"); HeartConfig::Instance()->SetOutputFilenamePrefix("results");
Bath problems seem to require decreased ODE timesteps.
HeartConfig::Instance()->SetOdeTimeStep(0.001); //ms
Next, use the PlaneStimulusCellFactory to define a set of Luo-Rudy cells. This factory normally sets the X=0 cells to be stimulated, but we don't want any intracellular stimulus in this test, so we pass in 0.0 as the stimulus magnitude.
PlaneStimulusCellFactory<LuoRudyIModel1991OdeSystem,2> cell_factory(0.0);
Now, we load up a rectangular mesh (in triangle/tetgen format), done as follows, using TrianglesMeshReader.
TrianglesMeshReader<2,2> reader("mesh/test/data/2D_0_to_1mm_400_elements"); TetrahedralMesh<2,2> mesh; mesh.ConstructFromMeshReader(reader);
In bath problems, each element has an attribute which must be set to 0 (cardiac tissue) or 1 (bath). This can be done by having an extra column in the element file (see for example mesh/test/data/1D_0_to_1_10_elements_with_two_attributes.ele, and note that the header in this file has 1 at the end to indicate that the file defines an attribute for each element. We have read in a mesh without this type of information set up, so we set it up manually, by looping over elements and setting those more than 2mm from the centre as bath elements (by default, the others are cardiac elements).
for(unsigned i=0; i<mesh.GetNumElements(); i++) { double x = mesh.GetElement(i)->CalculateCentroid()[0]; double y = mesh.GetElement(i)->CalculateCentroid()[1]; if( sqrt((x-0.05)*(x-0.05) + (y-0.05)*(y-0.05)) > 0.02 ) { mesh.GetElement(i)->SetRegion(HeartRegionCode::BATH); } }
Now we define the electrodes. First define the magnitude of the electrodes (ie the magnitude of the boundary extracellular stimulus), and the duration it lasts for. Currently, electrodes switch on at time 0 and have constant magnitude until they are switched off. (Note that this test has a small range of magnitudes that will work, perhaps because the electrodes are close to the tissue).
//-1e4 is under thershold, -1.4e4 too high - crashes the cell model double magnitude = -1.1e4; // uA/cm^2 double start_time = 0.0; double duration = 2; //ms
Electrodes work in two ways: the first electrode applies an input flux, and the opposite electrode can either be grounded or apply an equal and opposite flux (ie an output flux). The false here indicates the second electrode is not grounded, ie has an equal and opposite flux. The "0, 0.0, 0.1" indicates that the electrodes should be applied to the surfaces X=0.0 and X=0.1 (which must match the mesh provided) (so, for example, you should use "2, 0.0, 0.1" to apply electrodes to the surfaces Z=0.0 and Z=0.1, etc).
boost::shared_ptr<Electrodes<2> > p_electrodes( new Electrodes<2>(mesh, false, 0, 0.0, 0.1, magnitude, start_time, duration));
Now create the problem class, using the cell factory and passing in true as the second argument to indicate we are solving a bath problem..
BidomainProblem<2> bidomain_problem( &cell_factory, true );
..set the mesh and electrodes..
bidomain_problem.SetMesh(&mesh); bidomain_problem.SetElectrodes(p_electrodes);
..and solve as before.
bidomain_problem.Initialise(); bidomain_problem.Solve();
The results can be visualised as before. Note: The voltage is only defined at cardiac nodes (a node contained in any cardiac element), but for visualisation and computation a 'fake' value of ZERO is given for the voltage at bath nodes.
Finally, we can check that an AP was induced in any of the cardiac cells. We use a ReplicatableVector as before, and make sure we only check the voltage at cardiac cells.
Vec solution = bidomain_problem.GetSolution(); // the Vs and phi_e's, as a PetSc vector ReplicatableVector solution_repl(solution); bool ap_triggered = false; for(unsigned i=0; i<mesh.GetNumNodes(); i++) { if (mesh.GetNode(i)->GetRegion()==HeartRegionCode::TISSUE) { if (solution_repl[2*i] > 0.0) // 2*i, ie the voltage for this node (would be 2*i+1 for phi_e for this node) { ap_triggered = true; } } } TS_ASSERT(ap_triggered); } };
Code
The full code is given below
#include <cxxtest/TestSuite.h> #include "BidomainProblem.hpp" #include "PlaneStimulusCellFactory.hpp" #include "LuoRudyIModel1991OdeSystem.hpp" #include "PetscSetupAndFinalize.hpp" #include "TetrahedralMesh.hpp" #include "TrianglesMeshReader.hpp" class PointStimulus2dCellFactory : public AbstractCardiacCellFactory<2> { private: boost::shared_ptr<SimpleStimulus> mpStimulus; public: PointStimulus2dCellFactory() : AbstractCardiacCellFactory<2>(), mpStimulus(new SimpleStimulus(-6000.0, 0.5)) { } AbstractCardiacCell* CreateCardiacCellForTissueNode(unsigned nodeIndex) { double x = this->GetMesh()->GetNode(nodeIndex)->rGetLocation()[0]; double y = this->GetMesh()->GetNode(nodeIndex)->rGetLocation()[1]; if (fabs(x)+fabs(y)<1e-6) // ie if (x,y)==(0,0). An alternative would be if(norm_2(this->mpMesh->GetNode(nodeIndex)->rGetLocation())<1e-6) { return new LuoRudyIModel1991OdeSystem(mpSolver, mpStimulus); } else { return new LuoRudyIModel1991OdeSystem(mpSolver, mpZeroStimulus); } } }; class TestRunningBidomainSimulationsTutorial : public CxxTest::TestSuite { public: void TestSimpleSimulation() throw(Exception) { HeartConfig::Instance()->SetSimulationDuration(1.0); //ms HeartConfig::Instance()->SetMeshFileName("mesh/test/data/square_128_elements"); HeartConfig::Instance()->SetOutputDirectory("BidomainTutorial"); HeartConfig::Instance()->SetOutputFilenamePrefix("results"); PointStimulus2dCellFactory cell_factory; BidomainProblem<2> bidomain_problem( &cell_factory ); // bidomain_problem.Initialise(); // bidomain_problem.Solve(); HeartConfig::Instance()->SetIntracellularConductivities(Create_c_vector(1.75, 0.19)); HeartConfig::Instance()->SetExtracellularConductivities(Create_c_vector(6.2, 2.4)); HeartConfig::Instance()->SetSurfaceAreaToVolumeRatio(1.0); // 1/cm HeartConfig::Instance()->SetCapacitance(1.0); // uF/cm^2 bidomain_problem.Initialise(); HeartConfig::Instance()->SetVisualizeWithCmgui(true); bidomain_problem.Solve(); ReplicatableVector res_repl(bidomain_problem.GetSolution()); for(unsigned i=0; i<res_repl.GetSize(); i++) { // std::cout << res_repl[i] << "\n"; } DistributedVector dist_bidomain_voltage = bidomain_problem.GetSolutionDistributedVector(); DistributedVector::Stripe bidomain_voltage(dist_bidomain_voltage, 0); DistributedVector::Stripe extracellular_potential(dist_bidomain_voltage, 1); for (DistributedVector::Iterator index = dist_bidomain_voltage.Begin(); index != dist_bidomain_voltage.End(); ++index) { if (index.Global==bidomain_problem.rGetMesh().GetNumNodes()-1) // ie if the last node { TS_ASSERT_LESS_THAN(0, bidomain_voltage[index]); } } } void TestWithBathAndElectrodes() throw (Exception) { HeartConfig::Instance()->Reset(); HeartConfig::Instance()->SetSimulationDuration(3.0); //ms HeartConfig::Instance()->SetOutputDirectory("BidomainTutorialWithBath"); HeartConfig::Instance()->SetOutputFilenamePrefix("results"); HeartConfig::Instance()->SetOdeTimeStep(0.001); //ms PlaneStimulusCellFactory<LuoRudyIModel1991OdeSystem,2> cell_factory(0.0); TrianglesMeshReader<2,2> reader("mesh/test/data/2D_0_to_1mm_400_elements"); TetrahedralMesh<2,2> mesh; mesh.ConstructFromMeshReader(reader); for(unsigned i=0; i<mesh.GetNumElements(); i++) { double x = mesh.GetElement(i)->CalculateCentroid()[0]; double y = mesh.GetElement(i)->CalculateCentroid()[1]; if( sqrt((x-0.05)*(x-0.05) + (y-0.05)*(y-0.05)) > 0.02 ) { mesh.GetElement(i)->SetRegion(HeartRegionCode::BATH); } } //-1e4 is under thershold, -1.4e4 too high - crashes the cell model double magnitude = -1.1e4; // uA/cm^2 double start_time = 0.0; double duration = 2; //ms boost::shared_ptr<Electrodes<2> > p_electrodes( new Electrodes<2>(mesh, false, 0, 0.0, 0.1, magnitude, start_time, duration)); BidomainProblem<2> bidomain_problem( &cell_factory, true ); bidomain_problem.SetMesh(&mesh); bidomain_problem.SetElectrodes(p_electrodes); bidomain_problem.Initialise(); bidomain_problem.Solve(); Vec solution = bidomain_problem.GetSolution(); // the Vs and phi_e's, as a PetSc vector ReplicatableVector solution_repl(solution); bool ap_triggered = false; for(unsigned i=0; i<mesh.GetNumNodes(); i++) { if (mesh.GetNode(i)->GetRegion()==HeartRegionCode::TISSUE) { if (solution_repl[2*i] > 0.0) // 2*i, ie the voltage for this node (would be 2*i+1 for phi_e for this node) { ap_triggered = true; } } } TS_ASSERT(ap_triggered); } };