inspectmode.C

Go to the documentation of this file.

#include <vector>
#include <fstream>
#include <iostream>

#include "TChain.h"
#include "TCanvas.h"
#include "TH2F.h"
#include "TString.h"
#include "TRegexp.h"
#include "TLine.h"
#include "TArrow.h"

using std::cout;
using std::endl;

typedef std::vector<TString> StrArr;
typedef std::map<TString,int> VarHash;
typedef std::map<TString, std::vector<double> > VarCutMap;
typedef std::map<TString, std::vector<double> >::iterator VarCutMapIt;

StrArr varlist;
VarHash vhash;

VarCutMap lecutmap, gecutmap;

int SplitString(TString s, TString delim, StrArr &toks)
{
        toks.clear();
        
        TObjArray *tok = s.Tokenize(delim);
        int N = tok->GetEntries();      

        for (int i=0;i<N;++i) 
        {
                TString token = ((TObjString*)tok->At(i))->String();
                token.ReplaceAll("\t","");
                token = token.Strip(TString::kBoth);
                toks.push_back(token);
        }
        return N;
}

bool readConfig(TString fname, int mode)
{
        ifstream file(fname.Data());
        if (!file.is_open()) return false;

        varlist.clear();
        vhash.clear();
        lecutmap.clear();
        gecutmap.clear();

        char line[1500];

        StrArr sar1, sar2;
        
        // loop through file line by line
        while (!file.eof())
        {
                file.getline(line,1499);
                TString sline(line);
                
                // remove tabs everywhere
                sline.ReplaceAll("\t","");
                // if lines contains a '#' somewhere, cut string from that position (=comment)
                if (sline.Contains("#")) sline = sline(0,sline.Index("#")-1);
                // remove whitespace at begin and end
                sline = sline.Strip(TString::kBoth);
                
                if (sline=="") continue;
                
                SplitString(sline,":",sar1);
                
                int cmode  = (sar1[0].Atoi())%1000;
                double sqs = (sar1[0].Atof())/100000.;
                sqs = int(sqs*100)/100.;

                if (cmode != mode) continue;
                
                cout <<sline<<endl;
                
                SplitString(sar1[1],"&&",sar2);
                
                for (int i=0;i<sar2.size();++i)
                {
                        TString var=sar2[i], cut=sar2[i];
                        cut.ReplaceAll("=","");
                        double gecut=-99999., lecut=-99999.;
                        
                        if (sar2[i].Contains(">")) 
                        {
                                var=sar2[i](0,sar2[i].Index(">"));
                                gecut = TString(cut(cut.Index(">")+1,1000)).Atof();
                        }
                        
                        else if (sar2[i].Contains("<")) 
                        {
                                var=sar2[i](0,sar2[i].Index("<"));
                                lecut = TString(cut(cut.Index("<")+1,1000)).Atof();
                        }

                        //cout <<sqs<<" "<<"cut:"<<cut<<" le:"<<lecut<<" ge:"<<gecut<<endl;     
                        
                        if ( lecutmap.find(var)==lecutmap.end() && var!="tag" ) 
                        {
                                varlist.push_back(var); 
                                //vhash[var]=1;
                                std::vector<double> t;
                                lecutmap[var] = t;
                                gecutmap[var] = t;
                        }
                        
                        if (lecut>-99999) {lecutmap[var].push_back(sqs); lecutmap[var].push_back(lecut);}
                        if (gecut>-99999) {gecutmap[var].push_back(sqs); gecutmap[var].push_back(gecut);}
                }
        }
        
        file.close();
        
        return true;
}

int inspectmode(int mode, TString opt="cont", TString selcfg = "../../softrig/selection_fsim_dec2014.cfg")
{
        TString path="/data.local1/kgoetzen/data/softrig/fsim/merge4/shr/"; 
        
        TChain n(TString::Format("n%3d",mode));
        n.Add(TString::Format("%sM*_n%3d_*root",path.Data(),mode));

        TCanvas *c1=new TCanvas("c1","c1",10,10,800,600);

        if (!readConfig(selcfg,mode)) return;
        
        cout <<"vars("<<mode<<")"<<endl;
        
        TLine l1;
        l1.SetLineWidth(2);
        l1.SetLineColor(2);

        TArrow ar1;
        ar1.SetAngle(40);
        ar1.SetLineWidth(2);
        ar1.SetLineColor(2);
        ar1.SetFillColor(2);
        
        for (int i=0;i<varlist.size();++i) 
        {
                TString var = varlist[i];
                cout <<var<<""<<endl;
                
                double min = n.GetMinimum(var), max = n.GetMaximum(var), d=max-min;
                
                min-=0.05*d;
                max+=0.05*d;
                
                TH2F *h=new TH2F("h",var+" (mode "+TString::Format("%d",mode)+")",360,2.2,5.8,200,min,max);
                h->SetStats(0);
                h->SetXTitle("E_{cm} [GeV]");
                h->SetYTitle(var);
                
                n.Draw(var+":mode/1e8-0.13+recmode*0.01>>h","xmct");
                n.Draw(var+":mode/1e5+0.05>>+h","mode%1000==900");
                n.Draw(var+":mode/1e5+0.03>>+h","mode%1000==900");
                n.Draw(var+":mode/1e5+0.04>>+h","mode%1000==900");
                n.Draw(var+":mode/1e5+0.06>>+h","mode%1000==900");

                h->SetContour(50);
                h->Draw(opt);
                
                for (int j=0;j<lecutmap[var].size();j+=2) 
                {
                        double sqs = lecutmap[var][j], val = lecutmap[var][j+1];
                        printf("  sqs=%4.2f : %s<%f\n", sqs, var.Data(), val);
                        l1.DrawLine(sqs-0.15, val, sqs+0.15, val);
                        ar1.DrawArrow(sqs,val,sqs,val-d*0.05,0.015,"|>");
                }
                
                for (int j=0;j<gecutmap[var].size();j+=2) 
                {
                        double sqs = gecutmap[var][j], val = gecutmap[var][j+1];
                        printf("  sqs=%4.2f : %s>%f\n", sqs, var.Data(), val);
                        l1.DrawLine(sqs-0.15, val, sqs+0.15, val);
                        ar1.DrawArrow(sqs,val,sqs,val+d*0.05,0.015,"|>");
                }
                
                /*      
                TH2F *h=new TH2F("h",var,70,2.3,5.8,200,min,max);
                h->SetStats(0);
                
                n.Draw(var+":mode/1e8+0.05>>h","xmct",opt);
                n.Draw(var+":mode/1e5-0.05>>+h","mode%1000==900",opt);

                h->SetContour(50);
                
                */
                c1->Update();
                c1->SaveAs(TString::Format("fig/mode%d_%s.gif",mode,var.Data()));
                
                
        }
        cout <<endl;
        
        
  return 0;
}