%%cpp -d
#include <cstdlib>
#include <iostream>
#include <map>
#include <string>

#include "TChain.h"
#include "TFile.h"
#include "TTree.h"
#include "TString.h"
#include "TObjString.h"
#include "TSystem.h"
#include "TROOT.h"

#include "TMVA/MethodCategory.h"
#include "TMVA/Factory.h"
#include "TMVA/DataLoader.h"
#include "TMVA/Tools.h"
#include "TMVA/TMVAGui.h"


Bool_t UseOffsetMethod = kTRUE;

 std::cout << std::endl << "==> Start TMVAClassificationCategory" << std::endl;

 TMVA::Tools::Instance();

 bool batchMode = false;

 TString outfileName( "TMVACC.root" );
 TFile* outputFile = TFile::Open( outfileName, "RECREATE" );

std::string factoryOptions( "!V:!Silent:Transformations=I;D;P;G,D" );
if (batchMode) factoryOptions += ":!Color:!DrawProgressBar";

 TMVA::Factory *factory = new TMVA::Factory( "TMVAClassificationCategory", outputFile, factoryOptions );

 TMVA::DataLoader *dataloader=new TMVA::DataLoader("dataset");

 dataloader->AddVariable( "var1", 'F' );
 dataloader->AddVariable( "var2", 'F' );
 dataloader->AddVariable( "var3", 'F' );
 dataloader->AddVariable( "var4", 'F' );

 dataloader->AddSpectator( "eta" );

 TFile *input(0);
 TString fname = gSystem->GetDirName(__FILE__) + "/data/";
 if (gSystem->AccessPathName( fname + "toy_sigbkg_categ_offset.root")) {
    // if directory data not found try using tutorials dir
    fname = gROOT->GetTutorialDir() + "/tmva/data/";
 }
 if (UseOffsetMethod) fname += "toy_sigbkg_categ_offset.root";
 else                 fname += "toy_sigbkg_categ_varoff.root";
 if (!gSystem->AccessPathName( fname )) {
    // first we try to find tmva_example.root in the local directory
    std::cout << "--- TMVAClassificationCategory: Accessing " << fname << std::endl;
    input = TFile::Open( fname );
 }

 if (!input) {
    std::cout << "ERROR: could not open data file: " << fname << std::endl;
    exit(1);
 }

 TTree *signalTree     = (TTree*)input->Get("TreeS");
 TTree *background = (TTree*)input->Get("TreeB");

 Double_t signalWeight     = 1.0;
 Double_t backgroundWeight = 1.0;

 dataloader->AddSignalTree    ( signalTree,     signalWeight     );
 dataloader->AddBackgroundTree( background, backgroundWeight );

 TCut mycuts = ""; // for example: TCut mycuts = "abs(var1)<0.5 && abs(var2-0.5)<1";
 TCut mycutb = ""; // for example: TCut mycutb = "abs(var1)<0.5";

 dataloader->PrepareTrainingAndTestTree( mycuts, mycutb,
                                      "nTrain_Signal=0:nTrain_Background=0:SplitMode=Random:NormMode=NumEvents:!V" );

 factory->BookMethod( dataloader, TMVA::Types::kFisher, "Fisher", "!H:!V:Fisher" );

 factory->BookMethod( dataloader, TMVA::Types::kLikelihood, "Likelihood",
                      "!H:!V:TransformOutput:PDFInterpol=Spline2:NSmoothSig[0]=20:NSmoothBkg[0]=20:NSmoothBkg[1]=10:NSmooth=1:NAvEvtPerBin=50" );

 TMVA::MethodCategory* mcat = 0;

 TString theCat1Vars = "var1:var2:var3:var4";
 TString theCat2Vars = (UseOffsetMethod ? "var1:var2:var3:var4" : "var1:var2:var3");

 TMVA::MethodBase* fiCat = factory->BookMethod( dataloader, TMVA::Types::kCategory, "FisherCat","" );
 mcat = dynamic_cast<TMVA::MethodCategory*>(fiCat);
 mcat->AddMethod( "abs(eta)<=1.3", theCat1Vars, TMVA::Types::kFisher, "Category_Fisher_1","!H:!V:Fisher" );
 mcat->AddMethod( "abs(eta)>1.3",  theCat2Vars, TMVA::Types::kFisher, "Category_Fisher_2","!H:!V:Fisher" );

 TMVA::MethodBase* liCat = factory->BookMethod( dataloader, TMVA::Types::kCategory, "LikelihoodCat","" );
 mcat = dynamic_cast<TMVA::MethodCategory*>(liCat);
 mcat->AddMethod( "abs(eta)<=1.3",theCat1Vars, TMVA::Types::kLikelihood,
                  "Category_Likelihood_1","!H:!V:TransformOutput:PDFInterpol=Spline2:NSmoothSig[0]=20:NSmoothBkg[0]=20:NSmoothBkg[1]=10:NSmooth=1:NAvEvtPerBin=50" );
 mcat->AddMethod( "abs(eta)>1.3", theCat2Vars, TMVA::Types::kLikelihood,
                  "Category_Likelihood_2","!H:!V:TransformOutput:PDFInterpol=Spline2:NSmoothSig[0]=20:NSmoothBkg[0]=20:NSmoothBkg[1]=10:NSmooth=1:NAvEvtPerBin=50" );

 factory->TrainAllMethods();

 factory->TestAllMethods();

 factory->EvaluateAllMethods();

 outputFile->Close();

 std::cout << "==> Wrote root file: " << outputFile->GetName() << std::endl;
 std::cout << "==> TMVAClassificationCategory is done!" << std::endl;

 delete factory;
 delete dataloader;

 if (!gROOT->IsBatch()) TMVA::TMVAGui( outfileName );