/*
  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.
*/

//
// File: Markov_Infection.h
//
#ifndef _FRED_MARKOV_INFECTION_H
#define _FRED_MARKOV_INFECTION_H

#include "Infection.h"

class Disease;
class Person;
class Mixing_Group;

class Markov_Infection : public Infection {

 public:

  // if primary infection, infector and place are null.
  Markov_Infection(Disease* disease, Person* infector, Person* host, Mixing_Group* mixing_group, int day);

  ~Markov_Infection() {}

  void update(int day);

  void setup();

  double get_infectivity(int day);

  double get_symptoms(int day);

  bool is_fatal(int day);

  int get_state() {
    return this->state;
  }

  void set_state(int _state) {
    this->state = _state;
  }

 private:
  int state;
};

#endif // _FRED_MARKOV_INFECTION_H