TSamplingMixture.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 TSAMPLINGMIXTURE_H
#define TSAMPLINGMIXTURE_H

#include <iostream>
using std::cout;
using std::endl;

#include "TVectorD.h"

#include "simplex.h"

#include "TSamplerDeterminist.h"

namespace URANIE
{
namespace Sampler
{
class TSamplingMixture: public TSamplerDeterminist
{

    // Attributes
public:
    enum EType
    {
        kSummit = 10,
        kCenterEdge,
        kThirdEdge,
        kCenterFace,
        kCenterSegment,
        kCentroide
    };
private:
    enum ERelation
    {
        kLower, kUpper
    };

    Double_t _dThreshold;               
    vector<vector<double> > _planes;              
    Simplex *_simplex;                        

    Bool_t _bSummits;                                  
    Bool_t _bCenterEdges;                          
    Bool_t _bThirdEdges;                            
    Bool_t _bCenterFaces;                          
    Bool_t _bCentroide;                              
    Bool_t _bCenterSegments;                    

    Bool_t _bSaveCandidates;              

    TString _sAttNameTypeOfPoints; 
    TString _sAttNameCandidate; 

    Int_t _nGenerateSample; 
    Bool_t _bTrace;                    
    TString _sTraceFile;                         

    // Operations
    void fillListOfCandidatesPoints(list<vector<double> > lst, EType typePoint);
    void setListOfCandidatesPoints();
public:
    //---------------------------------------------

    TSamplingMixture(URANIE::DataServer::TDataServer *tds, Int_t ns,
            Double_t dThreshold = 1.0, Option_t *option = "");

    virtual ~TSamplingMixture();

    void init();

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

    const char* getNameInternalAttributeCandidatesPoints()
    {
        return _sAttNameCandidate.Data();
    }


    const char* getNameInternalAttributeTypePoints()
    {
        return _sAttNameTypeOfPoints.Data();
    }


    Int_t getNGeneratedSample()
    {
        return _nGenerateSample;
    }

    void setTrace(TString straceFile = "");
    void unsetTrace()
    {
        _bTrace = kFALSE;
    }
    Bool_t getTrace()
    {
        return _bTrace;
    }
    TString getTraceFile();

    void setSummits()
    {
        _bSummits = kTRUE;
    }
    void unsetSummits()
    {
        _bSummits = kFALSE;
    }
    void setCenterEdges()
    {
        _bCenterEdges = kTRUE;
    }
    void unsetCenterEdges()
    {
        _bCenterEdges = kFALSE;
    }

    void setThirdEdges()
    {
        _bThirdEdges = kTRUE;
    }
    void unsetThirdEdges()
    {
        _bThirdEdges = kFALSE;
    }

    void setCenterFaces()
    {
        _bCenterFaces = kTRUE;
    }
    void unsetCenterFaces()
    {
        _bCenterFaces = kFALSE;
    }

    void setCentroide()
    {
        _bCentroide = kTRUE;
    }
    void unsetCentroide()
    {
        _bCentroide = kFALSE;
    }

    void setCenterSegments()
    {
        _bCenterSegments = kTRUE;
    }
    void unsetCenterSegments()
    {
        _bCenterSegments = kFALSE;
    }

    void setSaveCandidates()
    {
        _bSaveCandidates = kTRUE;
    }
    void unsetSaveCandidates()
    {
        _bSaveCandidates = kFALSE;
    }
    Bool_t isSaveCandidates()
    {
        return _bSaveCandidates;
    }

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

    void addConstraint(TVectorD vecConstraint);

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

    virtual void generateSample(Option_t *option = "MD");

    void generateCandidates(Option_t *option);

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

    ClassDef(URANIE::Sampler::TSamplingMixture, ID_SAMPLER)
    // Classe to generate mixture sampling
};
} // Fin du namespace Sampler
} // Fin du namespace URANIE

#endif
// fin du fichier $RCSfile$.