/*
  This file is part of the FRED system.

  Copyright (c) 2010-2015, University of Pittsburgh, John Grefenstette,
  Shawn Brown, Roni Rosenfield, Alona Fyshe, David Galloway, Nathan
  Stone, Jay DePasse, Anuroop Sriram, and Donald Burke.

  Licensed under the BSD 3-Clause license.  See the file "LICENSE" for
  more information.
*/

#ifndef _FRED_MARKOV_EPIDEMIC_H
#define _FRED_MARKOV_EPIDEMIC_H

#include <string>
#include <vector>
using namespace std;

#include "Epidemic.h"
class Markov_Model;

typedef std::vector<Person*> person_vector_t;


class Markov_Epidemic : public Epidemic {

public:
  Markov_Epidemic(Disease * disease);

  ~Markov_Epidemic() {}

  void setup();

  void prepare();

  void update(int day);

  void markov_updates(int day);

  void process_transitions_to_state(int day, int state);

  void report_disease_specific_stats(int day);

  void end_of_run();

  void process_transition(int day, int i, int j, Person * person);

private:
  Markov_Model* markov_model;
  int number_of_states;
  person_vector_t * people_in_state;

  // event queues
  Events** transition_to_state_event_queue;
};

#endif // _FRED_MARKOV_EPIDEMIC_H