#include "parameters.hpp"
#include "aux.hpp"

using namespace std;

namespace STD {

GaussianProcessParameters::GaussianProcessParameters(size_t first_iter,
                                                     bool mean, bool center_)
    : first_iteration(first_iter), free_mean(mean), center(center_) {}

double Hyperparameters::get_param(Coefficient::Type distribution,
                                  size_t idx) const {
  switch (distribution) {
    case Coefficient::Type::gamma:
      if (idx == 0)
        return gamma_1;
      else
        return gamma_2;
    case Coefficient::Type::beta:
      if (idx == 0)
        return beta_1;
      else
        return beta_2;
    case Coefficient::Type::beta_prime:
      if (idx == 0)
        return beta_prime_1;
      else
        return beta_prime_2;
    case Coefficient::Type::normal:
      if (idx == 0)
        return normal_1;
      else
        return normal_2;
    default:
      throw std::runtime_error(
          "Error: no default hyper parameters defined for chosen "
          "distribution.");
      break;
  }
}

std::ostream &operator<<(std::ostream &os, const Hyperparameters &hyperparams) {
  os << "gamma_1 = " << hyperparams.gamma_1 << endl;
  os << "gamma_2 = " << hyperparams.gamma_2 << endl;
  os << "beta_1 = " << hyperparams.beta_1 << endl;
  os << "beta_2 = " << hyperparams.beta_2 << endl;
  os << "beta_prime_1 = " << hyperparams.beta_prime_1 << endl;
  os << "beta_prime_2 = " << hyperparams.beta_prime_2 << endl;
  os << "normal_1 = " << hyperparams.normal_1 << endl;
  os << "normal_2 = " << hyperparams.normal_2 << endl;
  return os;
}
}  // namespace STD