PREAN/TFndPreanHistos.h

// @(#)fROOT/PREAN:$Name:  $:$Id: TFndPreanHistos.h,v 1.24 2007/10/04 14:10:26 Diego_Faso Exp $
// Author: Diego Faso <mailto:faso@to.infn.it>, 2005/06/24

#ifndef FROOT_TFndPreanHistos
#define FROOT_TFndPreanHistos

#include <Riostream.h>
#include "TROOT.h"
#include "TSystem.h"
#include "TApplication.h"

#include "TString.h"
#include "TThread.h"

#include "TFile.h"
#include "TTree.h"
#include "TCanvas.h"
#include "TStyle.h"
#include "TH2.h"
#include "TH3F.h"

#include "TFndGraphs.h"
#include "TFndPrean.h"

class TFndPreanHistos: public TFndPrean{

 public:
  enum E_FndPreanShowNCanv
    {
      PreanShow_N_BhaCanv = 2,
      PreanShow_N_HypCanv = 3,
    };
  
  Int_t TrkTypes() { return 7; }
  Int_t TrkType(Int_t id,TString &str){
    switch(id){
    case 0: str = "ANY";        return    -1; // ANY
    case 1: str = "FWD-LONG";   return 11110; // FWD-LONG
    case 2: str = "FWD-SHORT";  return  1110; // FWD-SHORT
    case 3: str = "BCK-XLONG";  return 11112; // BCK-EXTRA-LONG
    case 4: str = "BCK-LONG";   return  1112; // BCK-LONG
    case 5: str = "BCK-SHORT";  return   112; // BCK-LONG
    case 6: str = "BCK-XSHORT"; return    12; // BCK-EXTRA-SHORT
    default: return 0;
    }
    return 0;
  }

  // following paths are referred to targets (fwd positive particles)
  UInt_t GetSelMuPath(Int_t pathnum){
    switch(pathnum){
      /*     case 1: return 0x33100; */
      /*     case 2: return 0x22300; */
      /*     case 3: return 0x11400; */
      /*     case 4: return 0x88500; */
      /*     case 5: return 0x77600; */
      /*     case 6: return 0x66800; */
      /*     case 7: return 0x55900; */
      /*     case 8: return 0x44A00; */
    case 1: return 0x44100;
    case 2: return 0x33200;
    case 3: return 0x22300;
    case 4: return 0x11500;
    case 5: return 0x88600;
    case 6: return 0x77700;
    case 7: return 0x66900;
    case 8: return 0x55A00;
    default:
      Error("MuPathName","pathnum \"%d\" not available (possible BUG!)",pathnum);
      TerminateFroot();
    }

  }

  
  TString MuPathNameTitle(Int_t pathnum,Bool_t is_name){
    switch(pathnum){
    case 1: return (is_name) ? "t1_1-4-4" : "t1: 1,4,4";
    case 2: return (is_name) ? "t2_2-3-3" : "t2: 2,3,3";
    case 3: return (is_name) ? "t3_3-2-2" : "t3: 3,2,2";
    case 4: return (is_name) ? "t4_5-1-1" : "t4: 5,1,1";
    case 5: return (is_name) ? "t5_6-8-8" : "t5: 6,8,8";
    case 6: return (is_name) ? "t6_7-7-7" : "t6: 7,7,7";
    case 7: return (is_name) ? "t7_9-6-6" : "t7: 9,6,6";
    case 8: return (is_name) ? "t8_A-5-5" : "t8: 10,5,5";
    default:
      Error("MuPathName","pathnum \"%d\" not available (possible BUG!)",pathnum);
      TerminateFroot();
    }
  }

 protected:
  // bhabha show
  Bool_t fGesHistosInitialized;
  Bool_t fBhabhaHistosInitialized;
  Bool_t fHypeHistosInitialized;
  Bool_t fStatGraphsInitialized;

  Int_t fKstopMarkCol[E_Kaon_Stop_Other+1];
  
  // bhabha show
  TCanvas *fBhabhaCanvas[PreanShow_N_BhaCanv];
  TCanvas *fHypeCanvas[PreanShow_N_HypCanv];
    
  // public:
 protected:
  
  // ges histos
  TH1F *fH_GesTrigBit; // trigger-bit value

  // bhabha histos
  TH2F *fH_BhaMom_2D; // negative vs positive momentum

  TH1F *fH_InvMass_EpluEmin; // invariant mass (hypothesys: bhabha e+e- pair)
  TH1F *fH_InvMass_PIpluPImin; // invariant mass (hypothesys: K_short decay)

  TH3F *fH_BhaVert_XYZ; // 3D reconstructed position (interaction point)
  TH2F *fH_BhaVert_XY; // Y vs X reconstructed position (interaction point)
  TH2F *fH_BhaVert_XZ; // Z vs X reconstructed position (interaction point)

  TH1F *fH_BhaTracksAngle; // angle between the two tracks
  TH1F *fH_BhaTotalMomentum; // total momentum (scalar product) [filled but not drawn]
  
  // --- BHABHA TOF time added after version 9.00/01
  TH2F *fH_BhaParticleTOF_2D; // negative vs positive TOF (calib. TOFONE - calib. TOFINO)
  TH1F *fH_BhaTofi_TimeDiff; // Time difference between e+ and e- at TOFINO
  TH2F *fH_BhaTofi_Time_2D;  // Time of e- vs time of e+
  TH1F *fH_BhaTofo_TimeDiff; // Time difference between e+ and e- at TOFONE

  // ---

  // hype histos
  TH1F *fH_Hyp_PattRecErr;

  TH2F *fH_HypKmin_stop_XY[E_Kaon_Stop_Other+1];
  TH2F *fH_HypKplu_stop_XY[E_Kaon_Stop_Other+1];

  TH1F *fH_Hyp_Kmin_StopCode; // volume in which kaon stopped
  TH1F *fH_Hyp_Kplu_StopCode; // volume in which kaon stopped

  TH1F *fH_Hyp_Kmin_StopTgt; // K- stop target number if stopped inside target
  TH1F *fH_Hyp_Kplu_StopTgt; // K+ stop target number if stopped inside target
  TH2F *fH_Hyp_K_StopTgt_2D; // (K- vs K+) if stopped inside target
  
  TH1F *fH_Hyp_Kmin_StopVol; // K- stop volume number if stopped inside target
  TH1F *fH_Hyp_Kplu_StopVol; // K+ stop volume number if stopped inside target
  TH2F *fH_Hyp_K_StopVol_2D; // (K- vs K+) if stopped outside target

  TH3F *fH_Hyp_PhiDecPos_XYZ; // 3D reconstructed position (phi decay)
  TH2F *fH_Hyp_PhiDecPos_XY; // Y vs X reconstructed position (phi decay)
  TH2F *fH_Hyp_PhiDecPos_XZ; // Z vs X reconstructed position (phi decay)
  TH1F *fH_PhiDecPos_X;      // X reconstructed position (phi decay)
  TH1F *fH_PhiDecPos_Y;      // Y reconstructed position (phi decay)
  TH1F *fH_PhiDecPos_Z;      // Z reconstructed position (phi decay)
  
  TH1F *fH_HypTrk_KpluPosAll; // pos tracks from K+ overall targets (no path selection)
  TH1F *fH_HypTrk_KpluPosPthAll; // pos tracks from K+ overall targets (path selection)

  TH1F *fH_HypTrk_KpluPosTgt[8]; // one histo for each target (no path selection)
  TH1F *fH_HypTrk_KpluPosPth[8]; // one histo for each target (path selection)

  // --- K+/K-/mu+ TOF time added at version 9.01/02
  TH1F *fH_Hyp_KTofi_TimeDiff; // Time difference between K- and K+ at TOFINO
  TH2F *fH_Hyp_KTofi_Time_2D;  // Time of K- vs time of K+ at TOFINO
  TH1F *fH_Hyp_MuTof; // Muon+ time of flight (raw)
  TH1F *fH_Hyp_MuTofCorr; // Muon+ time of flight (corrected by track length/momentum)
  // ---
  Int_t fCount_EventTotal;
  // --- run-dependences (variables as a function of the run number)
  // scalers variables (resetted at every run)
  Int_t fCount_EventPerRun;
  Int_t fCount_KminPerRun;
  Int_t fCount_KpluPerRun;
  Int_t fCount_Stat_Time[E_Scal_Inhib_End];
  Int_t fCount_Stat_BhaNum[E_Scal_Inhib_End];
  Int_t fCount_Stat_HypNum[E_Scal_Inhib_End];
  //

  // --- evaluated variables (on the ovaerall statistics)
  // --- HYPE from K+
  Int_t fEval_NumOfStoppedKplu[8]; // K+ stopped per target
  Int_t fEval_NumOfStoppedKplu_All; // K+ stopped overall targets
  
  Int_t fEval_NumOf_Mu_plu[8]; // mu+ stopped per target
  Int_t fEval_NumOf_Mu_plu_All; // mu+ stopped overall targets

  Int_t fEval_NumOf_Mu_plu_Pth[8]; // mu+ stopped per target (forward)
  Int_t fEval_NumOf_Mu_plu_Pth_All; // mu+ stopped overall targets (forward)

  Float_t fEval_RecEff[8]; // reconstruction efficiency per target (eval on all mu+)
  Float_t fEval_RecEff_All; // reconstruction efficiency overall targets (eval on all mu+)
  Float_t fEval_RecEff_Pth[8]; // reconstruction efficiency per target (eval on fwd mu+)
  Float_t fEval_RecEff_Pth_All; // reconstruction efficiency overall targets (eval on fwd mu+)

  // --- HYPE from K-
  Int_t fEval_NumOfStoppedKmin[8]; // K+ stopped per target
  Int_t fEval_NumOfStoppedKmin_All; // K+ stopped overall targets


  // stat histos (scalers)
  TFndGraph *fGR_Stat_Evs;

  TFndGraph *fGR_Stat_BhaNum[E_Scal_Inhib_End];  
  TFndGraph *fGR_Stat_HypNum[E_Scal_Inhib_End];  
  TFndGraph *fGR_Stat_AvgLum[E_Scal_Inhib_End];  
  TFndGraph *fGR_Stat_IntLum[E_Scal_Inhib_End];  
  TFndGraph *fGR_Stat_IntLum_Tot[E_Scal_Inhib_End];  

  TFndGraph *fGR_Stat_KminStop_Num; // total number of stopped K- / inv.nb (all targets)
  TFndGraph *fGR_Stat_KpluStop_Num; // total number of stopped K+ / inv.nb (all targets)

  TFndGraph *fGR_Stat_MuPluAll_Num; // total number of mu-plu/ inv.nb /run (all targets)
  TFndGraph *fGR_Stat_MuPluAll_Mom; // mu-plu momentum/run (all targets)

  TFndGraph *fGR_Stat_BhaRec_Num; // total number of Bhabha rec / inv.nb (from "e+e-" invariant-mass)
  TFndGraph *fGR_Stat_BhaRec_ECMee; // center of mass energy (from "e+e-" invariant-mass)
  TFndGraph *fGR_Stat_BhaRec_Kshort_Num; // total number of K_short rec / inv.nb (from "K_short")
  TFndGraph *fGR_Stat_BhaRec_Kshort_InvMass; // invariant mass pi+pi- (from "K_short" )
  TFndGraph *fGR_Stat_BhaRec_Ratio; // K_short/Bhabha ratio

  // temporary storage histograms
  TH1F *fHTmp_MuPluAll;
  TH1F *fHTmp_BhaInvMass;
  TH1F *fHTmp_PionsInvMass;

  // stat histos (hype)
  // --- mu+ vs run_num

  // --- evaluation data-members
  //  some of them are available from histograms
  //  (not really calculated)
  //  but they are reported as single data-member
  //  in order to be easier to handle

  // --- SELECTIONS (quality cuts)
  Bool_t fCut_Fitted;
  Int_t  fCut_TrkType; // -1 means any
  
 private:
  void InitStatGraphs();
  void DeleteStatGraphs();

  void InitGesHistos();
  void InitBhabhaHistos();
  void InitHypeHistos();
  
  void ResetGesHistos();
  void ResetBhabhaHistos();
  void ResetHypeHistos();

  void WriteGesHistos();
  void WriteBhabhaHistos();
  void WriteHypeHistos();
  void WriteStatGraphs();

  void LoadGesHistos(TFile *fin);
  void LoadBhabhaHistos(TFile *fin);
  void LoadHypeHistos(TFile *fin);
  void LoadStatHistos(TFile *fin);

  void FillGesHistos();
  void FillBhabhaHistos();
  void FillHypeHistos();
  void FillScalVariables();
  
  void EvalGesHistos();
  void EvalBhabhaHistos();
  void EvalHypeHistos();
  void EvalStatGraphs(const Int_t &cur_run);
  
  void AddSingleHistoFromFile(TH1 *hori,TFile *fin){
    if(!hori || !fin){
      Error("AddSingleHistoFromFile","At least one NULL argument!");
      return;
    }
    hori->Add( (TH1*) (fin->Get( hori->GetName() ) ) );
  }
  
  void LoadSingleGraphFromFile(TFndGraph *pori,TFile *fin){
    if(!pori || !fin){
      Error("LoadSingleGraphFromFile","At least one NULL argument!");
      return;
    }
    
    TFndGraph *grfil =  (TFndGraph *) ( fin->Get( pori->GetName() ) );
    Double_t px = 0;
    Double_t py = 0;
    Double_t ex = 0;
    Double_t ey = 0;
    for(Int_t p=0;p<grfil->GetN();p++){
      grfil->GetPoint(p,px,py);
      ex = grfil->GetErrorX(p);
      ey = grfil->GetErrorY(p);
      pori->AddPoint(px,py,ex,ey);
    }
    grfil = 0; // deleted lately while closing source file
  }

 public:
  TFndPreanHistos();
  ~TFndPreanHistos();
  
  void InitHistos()
    {
    // need to use the "hist" draw option
    //  in order to remove error-bars
    InitGesHistos();
    InitBhabhaHistos();
    InitHypeHistos();
    InitStatGraphs();
  }

  void ResetHistos()
  {
    ResetGesHistos();
    ResetBhabhaHistos();
    ResetHypeHistos();
  }

  void ResetStatGraphs(const Bool_t &recreate);
  //  void ResetStatHistos(const Int_t &first_run,const Int_t &last_run);

  void FillHistos()
  {
    FillGesHistos();
    FillBhabhaHistos();
    FillHypeHistos();
    FillScalVariables();
  }
  
  // --- Calculations on histograms content
  void EvalHistos()
    {
      EvalGesHistos();
      EvalBhabhaHistos();
      EvalHypeHistos();
    }
  
  void WriteHistosToFile(const TString &fnam);
  void LoadHistosFromFile(const TString &fnam);

  // ---
  friend class TFndPreanGUI;
    
  ClassDef(TFndPreanHistos,1)
};


#endif // FROOT_TFndPreanHistos

---