JRegressor.hh

Go to the documentation of this file.

#ifndef __JFIT__JREGRESSOR__
#define __JFIT__JREGRESSOR__

#include "Jeep/JMessage.hh"


/**
 * \file
 * General purpose data regression method.
 * \author mdejong
 */
namespace JFIT {}
namespace JPP { using namespace JFIT; }

namespace JFIT {

  using JEEP::JMessage;
  
  
  /**
   * Abstract minimiser.
   * This "minimiser" can be used to determine the chi2 of the data for a given model value.
   */
  template<class JModel_t>
  struct JAbstractMinimiser :
    public JMessage< JAbstractMinimiser<JModel_t> >
  {
    /**
     * Default constructor.
     */
    JAbstractMinimiser() 
    {}


    /**
     * Get chi2.
     *
     * \param  fit            fit function
     * \param  __begin        begin of data
     * \param  __end          end   of data
     * \return                chi2
     */
    template<class JFunction_t, class T>
    double operator()(const JFunction_t& fit, T __begin, T __end) 
    {
      double chi2 = 0.0;

      for (T i = __begin; i != __end; ++i) {
        chi2 += fit(value, *i);
      }

      return chi2;
    }

    JModel_t value;    //!< model value
  };


  /**
   * Template definition of a data regressor of given model.
   *
   * The first template argument refers to the model that should be fitted to the data and
   * the second to the type of minimiser.
   */
  template<class JModel_t, template<class> class JMinimiser_t = JAbstractMinimiser>
  struct JRegressor;


  /**
   * Abstract class for global fit method.
   */
  template<class JModel_t, template<class> class JMinimiser_t = JAbstractMinimiser>
  struct JAbstractRegressor :
    public JMinimiser_t<JModel_t>
  {
    typedef JMinimiser_t<JModel_t>                      minimiser_type;
    typedef JRegressor<JModel_t, JMinimiser_t>          regressor_type;


    /**
     * Global fit.
     *
     * \param  value              start value
     * \param  __begin            begin of data set
     * \param  __end              end   of data set
     * \return                    chi2
     */
    template<class T>
    double operator()(const JModel_t& value, T __begin, T __end)
    {
      static_cast<minimiser_type&>(*this).value = value;

      return static_cast<minimiser_type&>(*this)(static_cast<regressor_type&>(*this), __begin, __end);
    }


    /**
     * Global fit.
     *
     * \param  value              start value
     * \param  __begin1           begin of first  data set
     * \param  __end1             end   of first  data set
     * \param  __begin2           begin of second data set
     * \param  __end2             end   of second data set
     * \return                    chi2
     */
    template<class T1, class T2>
    double operator()(const JModel_t& value, 
                      T1 __begin1, T1 __end1,
                      T2 __begin2, T2 __end2)
    {
      static_cast<minimiser_type&>(*this).value = value;

      return static_cast<minimiser_type&>(*this)(static_cast<regressor_type&>(*this), __begin1, __end1, __begin2, __end2);
    }
  };
}

#endif