THJMA.h

Go to the documentation of this file.

/* --- Ansi C --- */
// Copyright (C) 2019 CEA/DEN
// 
// 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/>.

#ifndef __HJMA__
#define __HJMA__

// ROOT
#include "Rtypes.h"
#include "TRandom3.h"

// URANIE
#include "TOptimShare.h"

#include "TDataServer.h"
#include "TAttribute.h"
#include "TRun.h"
#include "TEval.h"
#include "TOptimFit.h"
#include "TFitGetter.h"

//#include "TEGO.h"
#include "MetaModelOptim.h"

namespace URANIE {
namespace MetaModelOptim {

struct hjma_domain {
    double mini, maxi, it;
};
struct hjma_pt {
    double *xy;
    vector<URANIE::DataServer::UEntry*> *codi, *codo;
};
struct hjma_work {
    struct hjma_pt *best, *cur, one, two;
};
struct hjma_mem {
    double crit;
    vector<URANIE::DataServer::UEntry*> data;
};


class THjmaSolver : public URANIE::Reoptimizer::TOptimSolver
{
private:
    int _Neval, _Nfirst, _Nstep;
    
public:
    THjmaSolver();
    virtual ~THjmaSolver();
    
    // parameters
    void setSize(int start, int step);
    void setMaximumEval(int eval);
    
    // master
    virtual URANIE::Reoptimizer::TOptimShare *getMaster(
        URANIE::DataServer::TDataServer *tds,
        URANIE::Relauncher::TRun *run);
    
    ClassDef(URANIE::MetaModelOptim::THjmaSolver, ID_METAMODELOPTIM)
};


class THJMA : public URANIE::Reoptimizer::TOptimShare
{
private:
    inline double evaluate_pt(struct hjma_pt &pt);
    inline void evaluate_best(struct hjma_work &self);
    inline bool evaluate_current(struct hjma_work &self);


protected:
    int _Ncall, _Ncut, _Nfirst, _Nstep;
    int _first, _update;
    URANIE::DataServer::TAttribute *_y;
    URANIE::Reoptimizer::TFitGetter *_crit, *_valid;
    TRandom3 _rnd;
    double **_global;
    std::vector<double *> *_mode;
    struct hjma_domain *_dom;

    void build(); 
    void globalsearch(struct hjma_work &self);
    int hjonestep(struct hjma_work &self, double *dep, double step);
    void hjtwostep(struct hjma_work &self, double *dep);
    int hjsearch(struct hjma_work &self, double prox, int loop, int dup);
    int duplicate(struct hjma_work &self, double it);
    void addSolution(struct hjma_work &self);
    int getBest(struct hjma_work &self);

    
public:
    THJMA(URANIE::DataServer::TDataServer *tds, URANIE::Relauncher::TRun *run);
    virtual ~THJMA();
    
    // initialisation
    virtual void addObjective(URANIE::DataServer::TAttribute *,
            URANIE::Reoptimizer::TOptimFit *f=NULL);
    void setUsable(URANIE::Reoptimizer::TOptimFit *);
    void setSize(int first, int step);
    virtual void setMaximumEval(int max);

    // solve
    void fitnessChanged();
    virtual void solverLoop();
    
    // post
    double *item(int);
    
    ClassDef(URANIE::MetaModelOptim::THJMA, ID_METAMODELOPTIM)
};

} } //namespace

#endif