TMVAApply.C File Reference

#include "TMVA/Reader.h"
#include "TFile.h"
#include "TTree.h"
#include "TString.h"
#include "TRegexp.h"
#include "TEventList.h"
#include "TLeaf.h"
#include <iostream>
#include <map>
#include <vector>
#include <algorithm>
#include <utility>

Go to the source code of this file.

Macros
#define	MAX   1000

Typedefs
typedef std::map< int, int >	CountMap
typedef std::vector< pair < float, int > >	ValueMap

Functions
int	gettype (TTree *t, TString varname)
int	SplitString (TString s, TString delim, TString *toks, int maxtoks)
int	init (TTree *t, TString vars)
int	uid (int lev, int lrun, int lmode)
int	countEvents (TTree *t, TEventList &el)
TString	getFromCut (TString vars)
int	TMVAApply (TString fname, TString vars, TString precut="", TString wfile="", Float_t bglevel=0.0001)

Variables
Float_t	fbranch [MAX]
Float_t	freader [MAX]
Int_t	ibranch [MAX]
Int_t	types [MAX]
Bool_t	bbranch [MAX]
Int_t	ev
Int_t	run
Int_t	mode
Int_t	rec
Int_t	nbranch

Macro Definition Documentation

#define MAX   1000

Definition at line 15 of file TMVAApply.C.

Typedef Documentation

typedef std::map<int, int> CountMap

Definition at line 23 of file TMVAApply.C.

typedef std::vector<pair<float, int> > ValueMap

Definition at line 24 of file TMVAApply.C.

Function Documentation

int countEvents	(	TTree *	t,
		TEventList &	el
	)

Definition at line 94 of file TMVAApply.C.

References ev, evcnt, i, mode, run, and uid().

Referenced by TMVAApply().

{
        t->SetBranchStatus("*",0);
        t->SetBranchStatus("ev",1);
        t->SetBranchStatus("run",1);
        t->SetBranchStatus("mode",1);
        t->SetBranchStatus("recmode",1);
                
        CountMap evcnt;
        
        for (int i=0;i<el.GetN();++i)
        {
                t->GetEntry(el.GetEntry(i));
                evcnt[uid(ev,run,mode)]+=1;
        }
        t->SetBranchStatus("*",1);

        return evcnt.size();
}

TString getFromCut

(

TString

vars

)

Definition at line 116 of file TMVAApply.C.

References i, n, res, SplitString(), TString, and v.

Referenced by TMVAApply().

{
        TString toks[50];
        int n=SplitString(vars, "&&", toks, 50);
        TRegexp rvar("[_a-zA-Z][_a-zA-Z0-9]*");
        
        TString res=" ";
        
        for (int i=0;i<n;++i)
        {
                TString v = toks[i](rvar);
                if (v!="")
                {
                        if (v=="tag" || res.Contains(" "+v+" ")) continue;
                        res+=v+" ";
                }
        }
        
        res = res.Strip(TString::kBoth);
        
        return res;
}

int gettype	(	TTree *	t,
		TString	varname
	)

Definition at line 28 of file TMVAApply.C.

References TString.

Referenced by init().

{
        if (t->GetBranch(varname)==0) return -1;
        
        TString leaftype = t->GetLeaf(varname)->GetTypeName();
        
        if (leaftype=="Float_t")         return 0; 
        else if (leaftype=="Int_t")      return 1;  
        else if (leaftype=="Bool_t") return 2;
        
        return -1;
}

int init	(	TTree *	t,
		TString	vars
	)

Definition at line 59 of file TMVAApply.C.

References bbranch, ev, fbranch, gettype(), i, ibranch, mode, rec, run, SplitString(), TString, types, and v.

Referenced by TMVAApply().

{
        t->SetBranchAddress("ev",&ev);
        t->SetBranchAddress("run",&run);
        t->SetBranchAddress("mode",&mode);
        t->SetBranchAddress("recmode",&rec);
        
        TString toks[30];
        int N = SplitString(vars," ",toks,30);
        
        for (int i=0;i<N;++i)
        {
                TString v = toks[i];
                types[i] = gettype(t, v);
                switch (types[i])
                {
                        case 0: t->SetBranchAddress(v, &(fbranch[i])); break;
                        case 1: t->SetBranchAddress(v, &(ibranch[i])); break;
                        case 2: t->SetBranchAddress(v, &(bbranch[i])); break;
                }
                //cout <<toks[i]<<" of type "<<types[i]<<endl;
        }   
        
        return N;
}

int SplitString	(	TString	s,
		TString	delim,
		TString *	toks,
		int	maxtoks
	)

Definition at line 43 of file TMVAApply.C.

References i.

Referenced by getFromCut(), init(), and TMVAApply().

{
        TObjArray *tok = s.Tokenize(delim);
        int N = tok->GetEntries();      
        for (int i=0;i<N;++i) 
                if (i<maxtoks) 
                {
                        toks[i] = ((TObjString*)tok->At(i))->String();
                        toks[i].ReplaceAll("\t","");
                        toks[i] = toks[i].Strip(TString::kBoth);
                }
        return N;
}

int TMVAApply	(	TString	fname,
		TString	vars,
		TString	precut = "",
		TString	wfile = "",
		Float_t	bglevel = 0.0001
	)

Definition at line 140 of file TMVAApply.C.

References b, bbranch, countEvents(), cut, ev, f, fbranch, freader, getFromCut(), i, ibranch, init(), mode, nsig, printf(), run, s, SplitString(), t, TString, types, and uid().

{
        if (vars.Contains("&&")) vars = getFromCut(vars);
        cout <<"Vars : "<<vars<<endl;
        
        TString sigcut = "tag&&mode%1000!=900";
        TString bkgcut = "tag&&mode%1000==900";

        if (precut!="")
        {
                cout <<"Precut : "<<precut<<endl;
                sigcut += "&&" + precut;
                bkgcut += "&&" + precut;
        }

        // determine ntp name and S0 and B0
        TRegexp rmode("M[0-9][0-9][0-9]");
        TRegexp rntp("n[0-9][0-9][0-9]");
        TRegexp regS("[0-9]+S");
        TRegexp regB("[0-9]+B");
        
        TString s  = fname(regS); s  = s(0,s.Length()-1);
        TString b  = fname(regB); b  = b(0,b.Length()-1);
        int n0s = s.Atoi()*10000;
        int n0b = b.Atoi()*10000;       
        TString treename = fname(rntp); 
        TString smode = fname(rmode);
        
        if (wfile=="") wfile="weights/M"+smode(1,3)+treename(1,3)+"_BDT.weights.xml";
        
        TFile *f = TFile::Open(fname);
        TTree *t =(TTree*) f->Get(treename);
        
        // prepare variables and branch mapping
        int Nbr = init(t,vars);
        TString varname[30];
        SplitString(vars," ",varname,30);
        
        // prepare TMVA reader
        TMVA::Reader *reader = new TMVA::Reader("Silent");
        for (int i=0;i<Nbr;++i) reader->AddVariable(varname[i], &freader[i]);
        reader->BookMVA("BDT",wfile);
        
        // initial number of S and B events
        TEventList elsall("elsall"), elball("elball");
        t->Draw(">>elsall",sigcut);
        t->Draw(">>elball",bkgcut);
        float nsig = countEvents(t,elsall);
        float nbkg = countEvents(t,elball);
        
        // apply tag cut
        TString tagcut="tag";
        if (precut!="") tagcut+="&&"+precut;
        t->Draw(">>el",tagcut);
        TEventList *el=(TEventList*)gDirectory->Get("el");
        
        int N = el->GetN();
        
        CountMap sigcnt, bkgcnt;
        ValueMap sigvals, bkgvals;

        // loop through TTree
        for (int i=0;i<N;++i)
        {
                t->GetEntry(el->GetEntry(i));
                
                if (i%10000==0) cout <<"#"<<flush;
                        
                for (int j=0;j<Nbr;++j) 
                {
                        Float_t var=0;
                        switch (types[j]) {case 2: var = (Float_t)bbranch[j]; break; case 1: var = (Float_t) ibranch[j]; break; default: var=fbranch[j]; }
                        freader[j] = var;
                }
                
                float mvares = reader->EvaluateMVA("BDT");
                
                if (mode%1000==900)     
                        bkgvals.push_back( std::make_pair(mvares, uid(ev,run,mode) ));
                else 
                        sigvals.push_back( std::make_pair(mvares, uid(ev,run,mode) ));
        }
        
        sort(bkgvals.begin(), bkgvals.end());
        
        int nbgmax = bglevel*n0b, i = bkgvals.size()-1;
        while ((int)bkgcnt.size()<nbgmax && i>=0) bkgcnt[bkgvals[i--].second]+=1;
        double cut = bkgvals[i].first;

        for (i=0;i<(int)sigvals.size();++i)
                if (sigvals[i].first>cut) sigcnt[sigvals[i].second]+=1;
        cout <<endl;
        printf("S : eff = %0.3f     N = %5d / %5d   N0 = %d\n", (double)sigcnt.size()/n0s, (int)sigcnt.size(), (int)nsig, n0s);
        printf("B : eff = %0.5f   N = %5d / %5d   N0 = %d\n",   (double)bkgcnt.size()/n0b, (int)bkgcnt.size(), (int)nbkg, n0b);
        cout <<"\nCUT = "<<cut<<endl;
        
        if (precut=="") precut="tag";
        else precut="tag&&"+precut;
        TString cfgline = smode(1,3)+treename(1,3)+" : "+precut+" : M"+smode(1,3)+treename(1,3)+"_BDT "+vars+" "+TString::Format("%f",cut);
        cout <<"cfgline -> "<<cfgline<<endl;
  return 0;
}

int uid	(	int	lev,
		int	lrun,
		int	lmode
	)

Definition at line 87 of file TMVAApply.C.

Referenced by countEvents(), and TMVAApply().

{
        return lev+10000*lrun+(((lmode/100)%10)*20+lmode%10)*100000;
}

Variable Documentation

Bool_t bbranch[MAX]

Definition at line 19 of file TMVAApply.C.

Referenced by init(), and TMVAApply().

Int_t ev

Definition at line 21 of file TMVAApply.C.

Referenced by countEvents(), init(), and TMVAApply().

Float_t fbranch[MAX]

Definition at line 17 of file TMVAApply.C.

Referenced by init(), and TMVAApply().

Float_t freader[MAX]

Definition at line 17 of file TMVAApply.C.

Referenced by TMVAApply().

Int_t ibranch[MAX]

Definition at line 18 of file TMVAApply.C.

Referenced by init(), and TMVAApply().

Int_t mode

Definition at line 21 of file TMVAApply.C.

Referenced by countEvents(), init(), and TMVAApply().

Int_t nbranch

Definition at line 21 of file TMVAApply.C.

Int_t rec

Definition at line 21 of file TMVAApply.C.

Referenced by init().

Int_t run

Definition at line 21 of file TMVAApply.C.

Referenced by countEvents(), init(), and TMVAApply().

Int_t types[MAX]

Definition at line 18 of file TMVAApply.C.

Referenced by init(), and TMVAApply().