Modelling_Distribution.cpp

Go to the documentation of this file.

/********************************************************************
 *  Copyright (C) 2021 by Federico Marulli and Giorgio Lesci        *
 *  federico.marulli3@unibo.it                                      *
 *                                                                  *
 *  This program is free software; you can redistribute it and/or   * 
 *  modify it under the terms of the GNU General Public License as  *
 *  published by the Free Software Foundation; either version 2 of  *
 *  the License, or (at your option) any later version.             *
 *                                                                  *
 *  This program is distributed in the hope that it will be useful, *
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of  *
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the   *
 *  GNU General Public License for more details.                    *
 *                                                                  *
 *  You should have received a copy of the GNU General Public       *
 *  License along with this program; if not, write to the Free      *
 *  Software Foundation, Inc.,                                      *
 *  59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.       *
 ********************************************************************/
 
#include "Modelling_Distribution.h"
 
using namespace std;
 
using namespace cbl;
 
 
// ============================================================================================
 
 
void cbl::modelling::distribution::Modelling_Distribution::set_model_Distribution (const statistics::PriorDistribution mean_prior, const statistics::PriorDistribution std_prior, const std::string mean_name, const std::string std_name)
{
  const int nParams = 2;
  vector<statistics::ParameterType> Par_type(nParams, statistics::ParameterType::_Base_);
  vector<string> Par_string(nParams);
  std::vector<statistics::PriorDistribution> param_prior (nParams);
 
  // Set the parameter names and priors
  Par_string[0] = mean_name;
  Par_string[1] = std_name;
  param_prior[0] = mean_prior;
  param_prior[1] = std_prior;
  
  // set prior
  m_set_prior(param_prior);
 
  // construct the model
  m_model = make_shared<statistics::Model1D>(statistics::Model1D(&model_gaussian, nParams, Par_type, Par_string, NULL));
}
 
 
// ============================================================================================
 
 
void cbl::modelling::distribution::Modelling_Distribution::set_model_Distribution (const double k, const statistics::PriorDistribution mean_prior, const statistics::PriorDistribution std0_prior, const std::string mean_name, const std::string std0_name, const std::string std_name)
{
  m_data_model.k = k;
  
  const int nParams = 3;
  const int nParams_derived = 1;
  
  vector<statistics::ParameterType> Par_type(nParams, statistics::ParameterType::_Base_);
  Par_type[2] = statistics::ParameterType::_Derived_;
  
  vector<string> Par_string(nParams);
  std::vector<statistics::PriorDistribution> param_prior (nParams-nParams_derived);
 
  // Set the parameter names and priors
  Par_string[0] = mean_name;
  Par_string[1] = std0_name;
  Par_string[2] = std_name;
  
  param_prior[0] = mean_prior;
  param_prior[1] = std0_prior;
  
  // set prior
  m_set_prior(param_prior);
 
  // set fixed parameters
  auto inputs = make_shared<STR_Distr_model>(m_data_model);
 
  // construct the model
  m_model = make_shared<statistics::Model1D>(statistics::Model1D(&model_gaussian2, nParams, Par_type, Par_string, inputs));
}
 
 
// ============================================================================================
 
 
std::vector<double> cbl::modelling::distribution::model_gaussian (const std::vector<double> x, const std::shared_ptr<void> inputs, std::vector<double> &parameter)
{
  (void)inputs;
  
  std::shared_ptr<void> ptr;
  std::vector<double> values(x.size());
 
  for (size_t i=0; i<x.size(); i++)
    values[i] = cbl::gaussian(x[i], ptr, {parameter[0], parameter[1]});
 
  return values;
}
 
 
// ============================================================================================
 
 
std::vector<double> cbl::modelling::distribution::model_gaussian2 (const std::vector<double> x, const std::shared_ptr<void> inputs, std::vector<double> &parameter)
{
  shared_ptr<STR_Distr_model> pp = static_pointer_cast<STR_Distr_model>(inputs);
  
  std::shared_ptr<void> ptr;
  std::vector<double> values(x.size());
  parameter[2] = parameter[1]*pp->k;
  
  for (size_t i=0; i<x.size(); i++)
    values[i] = cbl::gaussian(x[i], ptr, {parameter[0], parameter[2]});
 
  return values;
}