source: trunk/MagicSoft/Mars/mranforest/MRanForestGrow.h@ 7217

#ifndef MARS_MRanForestGrow
#define MARS_MRanForestGrow

#ifndef MARS_MRead
#include "MRead.h"
#endif

class MHMatrix;
class MParList;
class MRanForest;
class MRanTree;

class MRanForestGrow : public MRead
{
private:
    MRanTree   *fRanTree;
    MRanForest *fRanForest;
    MHMatrix   *fMGammas;   //! Gammas describing matrix
    MHMatrix   *fMHadrons;  //! Hadrons (non gammas) describing matrix

    Int_t fNumTrees;
    Int_t fNumTry;
    Int_t fNdSize;

    Int_t PreProcess(MParList *pList);
    Int_t Process();
    Int_t PostProcess();

    UInt_t  GetEntries()            { return fNumTrees; }
    TString GetFileName() const     { return "MRanForestGrow"; }
    TString GetFullFileName() const { return "MRanForestGrow"; }

public:
    MRanForestGrow(const char *name=NULL, const char *title=NULL);

    void SetNumTrees(Int_t n=-1) { fNumTrees=n>0?n:100; }
    void SetNumTry(Int_t   n=-1) { fNumTry  =n>0?n:  3; }
    void SetNdSize(Int_t   n=-1) { fNdSize  =n>0?n:  1; }

    ClassDef(MRanForestGrow, 0) // Task to grow a random forest
};

#endif