age_specific.hpp

Go to the documentation of this file.

#ifndef POPULATION_DYNAMICS_AGE_SPECIFIC_SELECTIVITY_HPP
#define POPULATION_DYNAMICS_AGE_SPECIFIC_SELECTIVITY_HPP
 
// #include "../../../interface/interface.hpp"
#include "../../../common/fims_math.hpp"
#include "../../../common/fims_vector.hpp"
#include "selectivity_base.hpp"
 
namespace fims_popdy {

template <typename Type>
struct AgeSpecificSelectivity : public SelectivityBase<Type> {
  fims::Vector<Type> logit_sel_at_age;

  size_t n_ages;

  size_t min_age;
 
  AgeSpecificSelectivity() : SelectivityBase<Type>() {}
 
  virtual ~AgeSpecificSelectivity() {}

  virtual const Type evaluate(const Type &x) {
    Type a = static_cast<Type>(0.0);
    Type b = static_cast<Type>(1.0);
    double x_temp = fims_math::Value(x);
    size_t x_final = static_cast<size_t>(x_temp);
    size_t i_age = x_final - min_age;
    return fims_math::inv_logit<Type>(a, b, logit_sel_at_age[i_age]);
  }

  virtual const Type evaluate(const Type &x, size_t pos) {
    Type a = static_cast<Type>(0.0);
    Type b = static_cast<Type>(1.0);
    double x_temp = fims_math::Value(x);
    size_t x_final = static_cast<size_t>(x_temp);
    size_t i_age = x_final - min_age;
    size_t n_values = logit_sel_at_age.size();
    if (n_values == 1) {
      return fims_math::inv_logit<Type>(a, b, logit_sel_at_age[0]);
    }
    if (n_ages == 0 || n_values % n_ages != 0) {
      throw std::invalid_argument(
          "AgeSpecificSelectivity: logit_sel_at_age has " +
          std::to_string(n_values) + " values, which is not 1 or a multiple " +
          "of n_ages (" + std::to_string(n_ages) + ").");
    }
    size_t i_value = (n_values == n_ages) ? i_age : pos * n_ages + i_age;
    return fims_math::inv_logit<Type>(a, b, logit_sel_at_age[i_value]);
  }
};
 
}  // namespace fims_popdy
 
#endif /* POPULATION_DYNAMICS_AGE_SPECIFIC_SELECTIVITY_HPP */