main.cpp

// Corona Simulation - basic simulation of a human transmissable virus
// Copyright (C) 2020  wbrinksma

// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

#include "simulation.h"
#include <iostream>
#include <random>
#include <math.h>
#include "html_canvas.h"
#include "ChartJS_handler.h"

//Constants to control the simulation
const int SUBJECT_COUNT = 100;
const int SIM_WIDTH = 800;
const int SIM_HEIGHT = 500;
const int SUBJECT_RADIUS = 2;

// B3 assignment
// constants which defines the time for how long the subject will be infected and is immune to the virus
const int INFECTION_TIME = 500;
const int IMMUNE_TIME = 500;

int main() {
    corsim::Simulation s(SIM_WIDTH,SIM_HEIGHT, INFECTION_TIME, IMMUNE_TIME, std::make_unique<corsim::HTMLCanvas>(30,150,SIM_WIDTH,SIM_HEIGHT),
        std::make_unique<corsim::ChartJSHandler>());

    //Code to randomly generate certain numbers, which is done by using certain distributions
    std::random_device rd;
    std::mt19937 mt(rd());
    std::uniform_real_distribution<double> dist_w(1.0, SIM_WIDTH);
    std::uniform_real_distribution<double> dist_h(1.0, SIM_HEIGHT);
    std::uniform_real_distribution<double> dist_dx(-1.0, 1.0);
    std::uniform_real_distribution<double> dist_dy(-1.0, 1.0);

    for (int i = 0; i<SUBJECT_COUNT; ++i)
    {
        double x = dist_w(mt); //Randomly generate x position
        double y = dist_h(mt); //Randomly generate y position
        
        corsim::Subject su(x,y,SUBJECT_RADIUS,false, false);

        su.set_dx(dist_dx(mt));
        su.set_dy(dist_dy(mt));

        if(i == SUBJECT_COUNT-1)
        {
            su.infect();
        }

        s.add_subject(std::move(su));
    }  

    s.run();
}