Model1D.h

Go to the documentation of this file.

/********************************************************************
 *  Copyright (C) 2014 by Federico Marulli and Alfonso Veropalumbo  *
 *  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.       *
 ********************************************************************/
 
#ifndef __MODEL1D__
#define __MODEL1D__
 
#include "Model.h"
 
 
// ===================================================================================================
 
 
namespace cbl {
  
  namespace statistics {
 
    class Model1D : public Model {
 
    public:
 
 
      Model1D () : Model() {m_dimension=Dim::_1D_;}
 
      Model1D (const model_function_1D function, const size_t nparameters, std::vector<ParameterType> parameterTypes={}, std::vector<std::string> parameterNames={}, const std::shared_ptr<void> inputs=NULL) : Model(nparameters, parameterTypes, parameterNames, inputs)
      { set_function(function); m_dimension=Dim::_1D_; }
 
      Model1D (const std::vector<double> (*function)(const std::vector<double> xx, std::vector<double> &val), const size_t nparameters, std::vector<ParameterType> parameterTypes={}, std::vector<std::string> parameterNames={}, const std::shared_ptr<void> inputs=NULL) : Model(nparameters, parameterTypes, parameterNames, inputs)
      { set_function(function); m_dimension=Dim::_1D_; }
 
      ~Model1D () = default;
 
 
      void set_function (const model_function_1D function);
 
      void set_function (const std::vector<double> (*function)(const std::vector<double> xx, std::vector<double> &val));
 
      double operator () (const double xx, std::vector<double> &parameters) const override
      {
    std::vector<std::vector<double>> xvec(1, std::vector<double>(1, xx));
    return Model::operator()(xvec, parameters)[0][0];
      }
 
      std::vector<double> operator () (const std::vector<double> xx, std::vector<double> &parameters) const override
      { 
    std::vector<std::vector<double>> xvec(1, xx);
    return Model::operator()(xvec, parameters)[0];
      }
 
      void stats_from_chains (const std::vector<double> xx, std::vector<double> &median_model, std::vector<double> &low_model, std::vector<double> &up_model, const int start=0, const int thin=1) override;
 
      void write (const std::string output_dir, const std::string output_file, const std::vector<double> xx, const std::vector<double> parameters) override;
 
      void write_at_bestfit (const std::string output_dir, const std::string output_file, const std::vector<double> xx) override;
 
      void write_from_chains (const std::string output_dir, const std::string output_file, const std::vector<double> xx, const int start=0, const int thin=1) override;
    
    };
  }
}
 
#endif