main.cpp

#include "unistd.h"
#include "unistd.h"
#include "schedule.h"
#include "station.h"
#include "adjacent_station.h"
#include "time_interval.h"
#include "main_station.h"
#include "debug.h"
#include "defect_generator.h"
#include <sstream>

// In days
#define SMALL_DEFECT_FREQ 24
#define MEDIUM_DEFECT_FREQ 41
#define LARGE_DEFECT_FREQ 11

// In minutes
#define SMALL_DEFECT_DELAY 2
#define MEDIUM_DEFECT_DELAY 19
#define LARGE_DEFECT_DELAY 80

// Trails numbers in Zilina
#define TRANSPORT_RAILS_NUM 7
#define CARGO_RAILS_NUM 4

//Cargo trains through Zilina
#define CARGO_TRAINS_NUM 48

#define SIMULATION_DAYS 60

void generateCargoTrains(std::vector<CAdjacentStation*>& cargoTrainStations);

/**
 * Main simulation program.
 *
 * Simulation time unit is 60 days
 */

int main(int argc, char *argv[])
{
    DBG_LOG("SIMULATION START - DEBUG");

    RandomSeed(time(NULL));
    Init(0, CTimeInterval::TimeToMinutes(0,0, SIMULATION_DAYS));

    // Main station
    CMainStation& mainStation = CMainStation::GetInstance();
    mainStation.SetTitle("Zilina");
    mainStation.SetTransportRailsNumber(TRANSPORT_RAILS_NUM);
    mainStation.SetCargoRailsNumber(CARGO_RAILS_NUM);

    std::vector<CAdjacentStation*> cargoTrainStations;

    cargoTrainStations.push_back(mainStation.AddAdjacentStation("Bohumin"));
    mainStation.AddAdjacentStation("Brodno");
    mainStation.AddAdjacentStation("Bytca");
    mainStation.AddAdjacentStation("Cadca");
    mainStation.AddAdjacentStation("Cesky Tesin");
    mainStation.AddAdjacentStation("Horny Hricov");
    mainStation.AddAdjacentStation("Kralovany");
    mainStation.AddAdjacentStation("Kysucke Nove Mesto");
    mainStation.AddAdjacentStation("Povazska Bystrica");
    mainStation.AddAdjacentStation("Rudina");
    cargoTrainStations.push_back(mainStation.AddAdjacentStation("Ruzomberok"));
    cargoTrainStations.push_back(mainStation.AddAdjacentStation("Trencin"));
    mainStation.AddAdjacentStation("Vrutky");
    mainStation.AddAdjacentStation("Zilina Solinky");
    mainStation.AddAdjacentStation("Zilina Zariecie");


    // Track 127 - Zilina ------> Cadca
    mainStation.AddCoreTrack(mainStation.GetAdjacentStation("Bohumin"), 94)
        .AddNestedSegment(new CTrack(mainStation.GetAdjacentStation("Cesky Tesin"), 65))
        .AddNestedSegment(new CTrack(mainStation.GetAdjacentStation("Cadca"), 30))
        .AddNestedSegment(new CTrack(mainStation.GetAdjacentStation("Kysucke Nove Mesto"), 10))
        .AddNestedSegment(new CTrack(mainStation.GetAdjacentStation("Rudina"), 7))
        .AddNestedSegment(new CTrack(mainStation.GetAdjacentStation("Brodno"), 5));

    // Track 126 - Zilina ------> Rajec
    mainStation.AddCoreTrack(mainStation.GetAdjacentStation("Zilina Solinky"), 5)
        .AddNestedSegment(new CTrack(mainStation.GetAdjacentStation("Zilina Zariecie"), 2));

    // Track 120 - Bratislava ------> Zilina
    mainStation.AddCoreTrack(mainStation.GetAdjacentStation("Trencin"), 179)
        .AddNestedSegment(new CTrack(mainStation.GetAdjacentStation("Povazska Bystrica"), 32))
        .AddNestedSegment(new CTrack(mainStation.GetAdjacentStation("Bytca"), 17))
        .AddNestedSegment(new CTrack(mainStation.GetAdjacentStation("Horny Hricov"), 7));

    // Track 180 - Zilina ------> Kosice
    mainStation.AddCoreTrack(mainStation.GetAdjacentStation("Ruzomberok"), 57)
        .AddNestedSegment(new CTrack(mainStation.GetAdjacentStation("Kralovany"), 39))
        .AddNestedSegment(new CTrack(mainStation.GetAdjacentStation("Vrutky"), 21));

    // Parse schedule file
    CSchedule schedule;
    if(!schedule.OpenConfigFile() || !schedule.ParseAndPlan(mainStation))
        return -1;

    // Generate random cargo trains
    generateCargoTrains(cargoTrainStations);

    // Small Defects Generator

    CDefectGenerator smallDefects(CTimeInterval::TimeToMinutes(0,0, SMALL_DEFECT_FREQ),
      SMALL_DEFECT_DELAY);
    smallDefects.Activate(CTimeInterval::TimeToMinutes(0,0, SMALL_DEFECT_FREQ));

    // Medium Defects Generator
    CDefectGenerator mediumDefects(CTimeInterval::TimeToMinutes(0,0, MEDIUM_DEFECT_FREQ),
      MEDIUM_DEFECT_DELAY);
    mediumDefects.Activate(CTimeInterval::TimeToMinutes(0,0, MEDIUM_DEFECT_FREQ));

    // Large Defects Generator
    CDefectGenerator largeDefects(CTimeInterval::TimeToMinutes(0,0, LARGE_DEFECT_FREQ),
      LARGE_DEFECT_DELAY);
    largeDefects.Activate(CTimeInterval::TimeToMinutes(0,0, LARGE_DEFECT_FREQ));

    Run();

    mainStation.GetDelayHistogram().Output();
    mainStation.GetTransportRailsStore().Output();
    mainStation.GetCargoRailsStore().Output();
    mainStation.GetDefectsHistogram().Output();

    return 0;
}

void generateCargoTrains(std::vector<CAdjacentStation*>& cargoTrainStations)
{
    for(int i = 0; i < CARGO_TRAINS_NUM * SIMULATION_DAYS; i++ )
    {
        CMainStation* pMainStation = &CMainStation::GetInstance();
        CAdjacentStation* pAdjacentStation = cargoTrainStations[rand() % cargoTrainStations.size()];

        std::stringstream ssTrainName;
        ssTrainName << "Cargo Train No. " << i;
        unsigned startTime = rand() % CTimeInterval::TimeToMinutes(0,0,SIMULATION_DAYS);

        // coming to / leaving from main station
        CStation* pSrc;
        CStation* pDst;
        if(rand() % 2)
        {
            pSrc = pMainStation;
            pDst = pAdjacentStation;
        }
        else
        {
            pSrc = pAdjacentStation;
            pDst = pMainStation;
        }

        ssTrainName << pSrc->GetTitle()[0];

        // t = s / v
        unsigned duration = pMainStation->GetTrack(*pAdjacentStation).GetLength() / 100;
        // different speeds
        duration += Exponential(60);

        DBG_LOG_T("Cargo train from " << pSrc->GetTitle() << " to " << pDst->GetTitle());

        pSrc->AddCargoTrain(ssTrainName.str(), startTime, *pDst, CTrainGenerator::Frequency::FREQ_ONCE,
                startTime + duration, 5);
    }
}