TPolynomialMultiDimFit.h

Go to the documentation of this file.

// Copyright (C) 2013-2024 CEA/DES
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published
// by the Free Software Foundation, either version 3 of the License, or 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 Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
// $Id$
// $Author$
// $Date$
// $Revision$
// $State$

#ifndef TPOLYNOMIALMULTIDIMFIT_H
#define TPOLYNOMIALMULTIDIMFIT_H

//ROOT
#include "TMultiDimFit.h"

// Uranie
#include "TModeler.h"
#include "TDataServer.h"
#include "TLinearRegression.h"

namespace URANIE
{
namespace Modeler
{
class TPolynomialMultiDimFit: public TModeler
{
    // Attributs
protected:
private:
    TMultiDimFit *_MultiDimFit;      
public:

    // Operations
private:
public:
    //---------------------------------------------

    TPolynomialMultiDimFit(URANIE::DataServer::TDataServer *tds,
            TString architecture, TMultiDimFit::EMDFPolyType type,
            Option_t *option = "");

    TPolynomialMultiDimFit(URANIE::DataServer::TDataServer *tds,
            const char *varexpinput, const char *varexpoutput,
            TMultiDimFit::EMDFPolyType type, Option_t * option = "");
    virtual ~TPolynomialMultiDimFit();

    //---------------------------------------------
    void setMinAngle(Double_t dangle = 1);
    void setMaxAngle(Double_t dangle = 0);
    void setMaxFunctions(Int_t n);
    void setMaxStudy(Int_t n);
    void setMaxTerms(Int_t n);
    void setMaxPowers(const Int_t *powers);
    void setMinRelativeError(Double_t derror);
    void setPowerLimit(Double_t limit = 1e-3);
    void setPowers(const Int_t *powers, Int_t terms);


    TMultiDimFit *getMultiDimFit()
    {
        return _MultiDimFit;
    }

    //---------------------------------------------
    void estimate(Option_t * option = "");

    void fillData(TMatrixD &Aw, TVectorD &yw);


    //---------------------------------------------

//     void exportFunction(const char* lang, const char* file = "",
//             const char* name = "", Option_t *option = "");

    void exportModelCplusplus(std::ofstream * sourcefile) const;

    void exportModelFortran(std::ofstream * sourcefile) const;
    

    void exportModelPMML(const char* file = "", 
                         const char* name = "",Option_t *option = "") const
    {
        TString soption = option;
        soption.ToLower();
        UNUSED(file,name);
        std::cerr
                << "   --- Method TPolynomialMultiDimFit::exportModelPMML(std::ofstream * sourcefile) not yet implemented"
                << endl;
        throw URANIE::Exceptions::UWarningExceptions(__FILE__, __LINE__,        
          "TPolynomialMultiDimFit::exportFunction\n The PMML format is not yet implemented.");
        
    };
    

    void exportModelPython(std::ofstream * sourcefile) const
    {
        UNUSED(sourcefile);
        std::cerr
                << "   --- Method TPolynomialMultiDimFit::exportModelPython(std::ofstream * sourcefile) not yet implemented"
                << endl;
        throw URANIE::Exceptions::UWarningExceptions(__FILE__, __LINE__,        
          "TPolynomialMultiDimFit::exportFunction\n The Python format is not yet implemented.");
    };

    //---------------------------------------------
    virtual void printLog(Option_t *option = "");

    ClassDef(URANIE::Modeler::TPolynomialMultiDimFit, ID_MODELER)
    //Classe de
};

} // Fin du namespace Modeler
} // Fin du namespace URANIE
#endif