ProbabilityMap.h

/* Copyright (c) Queen's Printer for Ontario, 2020. */
/* Copyright (c) His Majesty the King in Right of Canada as represented by the Minister of Natural Resources, 2021-2025. */
 
/* SPDX-License-Identifier: AGPL-3.0-or-later */
 
#pragma once
#include <string>
#include <vector>
#include "GridMap.h"
#include "Statistics.h"
#include "Perimeter.h"
namespace fs
{
namespace sim
{
class Model;
class IntensityMap;
class ProbabilityMap
{
public:
  ProbabilityMap() = delete;
  ~ProbabilityMap() = default;
  ProbabilityMap(const ProbabilityMap& rhs) noexcept = delete;
  ProbabilityMap(ProbabilityMap&& rhs) noexcept = delete;
  ProbabilityMap& operator=(const ProbabilityMap& rhs) noexcept = delete;
  ProbabilityMap& operator=(ProbabilityMap&& rhs) noexcept = delete;
  ProbabilityMap(const string dir_out,
                 DurationSize time,
                 DurationSize start_time,
                 int min_value,
                 int low_max,
                 int med_max,
                 int max_value,
                 const data::GridBase& grid_info);
  ProbabilityMap* copyEmpty() const;
  void setPerimeter(const topo::Perimeter* const perimeter);
  void addProbabilities(const ProbabilityMap& rhs);
  void addProbability(const IntensityMap& for_time);
  [[nodiscard]] vector<MathSize> getSizes() const;
  [[nodiscard]] util::Statistics getStatistics() const;
  [[nodiscard]] size_t numSizes() const noexcept;
  void show() const;
  void saveSizes(const string& base_name) const;
  void saveAll(const tm& start_time,
               DurationSize time,
               const bool is_interim) const;
  void saveTotal(const string& base_name, const bool is_interim) const;
  void saveTotalCount(const string& base_name) const;
  void saveHigh(const string& base_name) const;
  void saveModerate(const string& base_name) const;
  void saveLow(const string& base_name) const;
  void reset();
  static void deleteInterim();
private:
  bool record_if_interim(const char* filename) const;
  template <class R>
  string saveToProbabilityFile(const data::GridMap<size_t>& grid,
                               const string& dir,
                               const string& base_name,
                               const R divisor) const
  {
    const string filename = grid.saveToProbabilityFile(dir, base_name, divisor);
    record_if_interim(filename.c_str());
    return filename;
  };
  const string dir_out_;
  data::GridMap<size_t> all_;
  data::GridMap<size_t> high_;
  data::GridMap<size_t> med_;
  data::GridMap<size_t> low_;
  vector<MathSize> sizes_{};
  const DurationSize time_;
  const DurationSize start_time_;
  mutable mutex mutex_;
  IntensitySize min_value_;
  IntensitySize max_value_;
  const IntensitySize low_max_;
  const IntensitySize med_max_;
  const topo::Perimeter* perimeter_;
};
}
}