growth_products.hpp

Go to the documentation of this file.

// inst/include/population_dynamics/growth/growth_products.hpp
#ifndef POPULATION_DYNAMICS_GROWTH_PRODUCTS_HPP
#define POPULATION_DYNAMICS_GROWTH_PRODUCTS_HPP
 
#include <cstddef>
#include <vector>
#include <stdexcept>
 
namespace fims_popdy {

template <typename Type>
struct GrowthProducts {
  std::size_t n_years = 0; 
  std::size_t n_ages = 0;  
  std::size_t n_sexes = 0; 
 
  std::vector<Type> mean_LAA; 
  std::vector<Type> sd_LAA;   
  std::vector<Type> mean_WAA; 

  GrowthProducts() = default;

  GrowthProducts(std::size_t years, std::size_t ages, std::size_t sexes) {
    Resize(years, ages, sexes);
  }

  void Resize(std::size_t years, std::size_t ages, std::size_t sexes) {
    n_years = years;
    n_ages = ages;
    n_sexes = sexes;
 
    const std::size_t n = n_years * n_ages * n_sexes;
 
    mean_LAA.assign(n, static_cast<Type>(0));
    sd_LAA.assign(n, static_cast<Type>(0));
    mean_WAA.assign(n, static_cast<Type>(0));
  }

  inline std::size_t Index(std::size_t y, std::size_t a, std::size_t s) const {
    // Cheap runtime safety. Keep it here, not sprinkled everywhere.
    if (y >= n_years || a >= n_ages || s >= n_sexes) {
      throw std::out_of_range("GrowthProducts index out of range");
    }
    return (y * n_ages + a) * n_sexes + s;
  }

  inline Type& MeanLAA(std::size_t y, std::size_t a, std::size_t s) {
    return mean_LAA[Index(y, a, s)];
  }

  inline Type& SdLAA(std::size_t y, std::size_t a, std::size_t s) {
    return sd_LAA[Index(y, a, s)];
  }

  inline Type& MeanWAA(std::size_t y, std::size_t a, std::size_t s) {
    return mean_WAA[Index(y, a, s)];
  }

  inline const Type& MeanLAA(std::size_t y, std::size_t a,
                             std::size_t s) const {
    return mean_LAA[Index(y, a, s)];
  }

  inline const Type& SdLAA(std::size_t y, std::size_t a, std::size_t s) const {
    return sd_LAA[Index(y, a, s)];
  }

  inline const Type& MeanWAA(std::size_t y, std::size_t a,
                             std::size_t s) const {
    return mean_WAA[Index(y, a, s)];
  }

  inline std::size_t Size() const { return mean_LAA.size(); }
};
 
}  // namespace fims_popdy
 
#endif /* POPULATION_DYNAMICS_GROWTH_PRODUCTS_HPP */