All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
EpidemicSimulationSEIR.h
Go to the documentation of this file.
1 /*
2  * EpidemicSimulationSEIR.h
3  *
4  * Created on: 20.11.2015
5  * Author: Christian Staudt
6  */
7 
8 #ifndef EPIDEMICSIMULATIONSEIR_H_
9 #define EPIDEMICSIMULATIONSEIR_H_
10 
11 #include "../base/Algorithm.h"
12 #include "../graph/Graph.h"
13 
14 namespace NetworKit {
15 
22 public:
30 
31  void run() override;
32 
33 
34  std::vector<std::vector<count>> getData();
35 
36 protected:
37 
38  const Graph& G;
40  double transP;
44  enum class State {S, E, I, R, U}; // Susceptible, Exposed, Infectious, Removed, Undefined
45  std::vector<State> state;
46  std::vector<index> timestamp;
47  std::vector<std::vector<count>> stats;
49 
50 };
51 
52 } /* namespace NetworKit */
53 
54 #endif /* DEPIDEMICSIMULATIONSEIR_H_ */
std::vector< std::vector< count > > stats
Definition: EpidemicSimulationSEIR.h:47
count eTime
Definition: EpidemicSimulationSEIR.h:41
bool randStartNode
Definition: EpidemicSimulationSEIR.h:48
std::vector< State > state
Definition: EpidemicSimulationSEIR.h:45
EpidemicSimulationSEIR(const Graph &G, count tMax, double transP, count eTime, count iTime, node zero)
Definition: EpidemicSimulationSEIR.cpp:13
count tMax
Definition: EpidemicSimulationSEIR.h:39
double transP
Definition: EpidemicSimulationSEIR.h:40
void run() override
The generic run method which calls runImpl() and takes care of setting to the appropriate value...
Definition: EpidemicSimulationSEIR.cpp:16
node zero
Definition: EpidemicSimulationSEIR.h:43
count iTime
Definition: EpidemicSimulationSEIR.h:42
std::vector< index > timestamp
Definition: EpidemicSimulationSEIR.h:46
std::vector< std::vector< count > > getData()
Definition: EpidemicSimulationSEIR.cpp:103
State
Definition: EpidemicSimulationSEIR.h:44
uint64_t count
Definition: Globals.h:21
index node
Definition: Globals.h:23
const Graph & G
Definition: EpidemicSimulationSEIR.h:38
Definition: Algorithm.h:9
A graph (with optional weights) and parallel iterator methods.
Definition: Graph.h:79
Node centrality index which ranks nodes by their degree.
Definition: EpidemicSimulationSEIR.h:21