C2F-W/Cell2Fire/Cell2Fire.h at af55d820c2d59f2b7ccaaa6d354870506a34e11d · fire2a/C2F-W · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
#ifndef CELL2FIRE
#define CELL2FIRE

// Headers
#include "Cells.h"
#include "DataGenerator.h"
#include "FuelModelKitral.h"
#include "FuelModelSpain.h"
#include "Lightning.h"
#include "ReadArgs.h"
#include "ReadCSV.h"
#include "Spotting.h"
#include "WriteCSV.h"

// Include libraries
#include <algorithm>
#include <boost/random.hpp>
#include <cmath>
#include <iostream>
#include <math.h>
#include <memory>
#include <random>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <vector>

using namespace std;
string C2FW_VERSION = "v0.0.0";
string sim_log_filename = "ignition_and_weather_log.csv";

class Cell2Fire
{
    // DFs
  private:
    CSVReader CSVWeather;
    CSVReader CSVForest;
    weatherDF* wdf_ptr;
    vector<weatherDF> wdf;

  public:
    // Main inputs
    arguments args;
    arguments* args_ptr;
    fuel_coefs* coef_ptr;
    fuel_coefs coefs[18];

    // Integers
    int sim;
    int rows;
    int cols;
    int widthSims;
    int weatherPeriod = 0;
    int year = 1;
    int gridNumber = 0;
    int weatherperiod = 0;
    long int nCells;
    int nIgnitions = 0;
    double xllcorner;
    double yllcorner;
    // Booleans
    bool noIgnition = true;  // None = -1
    bool activeCrown = false;
    bool messagesSent = false;
    bool repeatFire = false;
    bool done = false;
    bool noMessages = false;

    // Doubles
    double cellSide;
    double areaCells;
    double perimeterCells;
    double ROSRV{};

    // Strings
    string gridFolder;
    string messagesFolder;
    string rosFolder;
    string crownFolder;
    string surfaceIntensityFolder;
    string crownIntensityFolder;
    string sfbFolder;
    string cfbFolder;
    string surfaceFlameLengthFolder;
    string maxFlameLengthFolder;
    string crownFlameLengthFolder;
    string historyFolder;
    string ignitionsFolder;
    string statsFolder;

    // Vectors
    std::vector<int> fire_period;
    std::vector<std::vector<int>> coordCells;
    std::vector<std::unordered_map<std::string, int>> adjCells;
    std::vector<std::vector<std::string>> DF;
    std::vector<std::vector<std::string>> WDist;
    std::vector<float> ignProb;       // (long int&, int);
    std::vector<int> statusCells;     //(long int, int);
    std::vector<int> fTypeCells;      // (long int&, int);
    std::vector<string> fTypeCells2;  // (long int&, const char [9]);
    std::vector<std::vector<std::string>> WeatherData;
    std::vector<int> IgnitionPoints;
    vector<int> burnedOutList;
    std::vector<double> FSCell;
    std::vector<float> crownMetrics;
    std::vector<int> crownState;
    std::vector<float> surfaceIntensities;
    std::vector<float> crownIntensities;
    std::vector<float> crownFraction;
    std::vector<float> surfFraction;
    std::vector<float> RateOfSpreads;
    std::vector<float> surfaceFlameLengths;
    std::vector<float> crownFlameLengths;
    std::vector<float> maxFlameLengths;
    std::vector<std::vector<int>> IgnitionSets;
    // std::vector<int> IgnitionHistory;

    // Sets
    std::unordered_set<int> availCells;
    std::unordered_set<int> nonBurnableCells;
    std::unordered_set<int> burningCells;
    std::unordered_set<int> burntCells;
    std::unordered_set<int> harvestCells;

    // Cells Dictionary
    std::unordered_map<int, Cells> Cells_Obj;

    // Constructor
    Cell2Fire(arguments args);

    // Methods
    void InitCell(int id);
    string createOutputFolder(string metric);
    void reset(int rnumber, double rnumber2, int simExt);
    bool RunIgnition(boost::random::mt19937 generator, int ep);
    std::unordered_map<int, std::vector<int>> SendMessages();
    void GetMessages(std::unordered_map<int, std::vector<int>> sendMessageList);
    void Results();
    void outputGrid();
    void updateWeather();
    void Step(boost::random::mt19937 generator, int ep);
    void InitHarvested();

    // Utils
    std::vector<float> getROSMatrix();
    std::vector<float> getFireProgressMatrix();
    void chooseWeather(const string& weatherOpt, int rnumber, int simExt);
    int totalFirePeriods;
};

#endif