PREAN/TFndPrean.h

// @(#)fROOT/PREAN:$Name:  $:$Id: TFndPrean.h,v 1.59 2007/09/24 07:32:41 Diego_Faso Exp $
// Author: Diego Faso <mailto:faso@to.infn.it>, 2005/06/24

#ifndef FROOT_TFndPrean
#define FROOT_TFndPrean

#include <Riostream.h>
#include <TObject.h>
#include <TString.h>
#include "TDatime.h"
#include "TTree.h"
#include "TFile.h"
#include <TThread.h>

#include "TClonesArray.h"

#include "TFndTrack.h" // this includes FROOT.h and FIN_PHYS.h
#include "TFndProcessRec.h" // this includes FROOT.h

//#define FndPrean_CheckForDaqRunning

using namespace FROOT;
using namespace FIN_PHYS;

class TFndPrean: public TObject
{

 public:
  
  enum E_Scaler_Inhibit_Status {
    E_Scal_Inhib_No  = 0,
    E_Scal_Inhib_Inj = 1,
    E_Scal_Inhib_All = 2,
    E_Scal_Inhib_End = 3 // used for arrays (forced while creating branches)
  };
  
  TString GetScalInhibitStatusName(Int_t which,Bool_t isname)
    {
      E_Scaler_Inhibit_Status wh = (E_Scaler_Inhibit_Status) which;
      switch(wh){
      case E_Scal_Inhib_No:  return (isname)? "no_inhib"  : "no inhib.";
      case E_Scal_Inhib_Inj: return (isname)? "inj_inhib" : "inj inhib.";
      case E_Scal_Inhib_All: return (isname)? "all_inhib" : "all inhib.";
      default:  
        Warning("GetScalInhibitStatusName","Status not available ()",which);
        return "";
      }
    }
  
  enum E_Kaon_StopCode {
    E_Kaon_Stop_Target = 0,
    E_Kaon_Stop_Isim = 1,
    E_Kaon_Stop_Glass = 2,
    E_Kaon_Stop_Upilex = 3,
    E_Kaon_Stop_Osim = 4,
    E_Kaon_Stop_Other = 5,
  };
  
  enum E_Count_id{ // id used by counters (#trig / #analyzed / #patt.rec / #rec /#cut)
    E_count_trigger = 0,
    E_count_analyze = 1,
    E_count_pat_rec = 2,
    E_count_reconst = 3,    
    E_count_rec_sel = 4, // in case of new entries remember to correct data-members
  };

protected:


  // --- folowing enums could be moved into FROOT.h
  enum E_Fnd_Scal_01_Descr{ // index of the "single-scaler" array (mod 0-1)
    E_FndScal_01_Desc_M27_HYP      = 0,
    E_FndScal_01_Desc_M27_BHA      = 1,
    E_FndScal_01_Desc_BTB_11       = 2,
    E_FndScal_01_Desc_BTB_EL       = 3,
    E_FndScal_01_Desc_PC           = 4, // prompt-coincidence
    E_FndScal_01_Desc_Presc_BTB_MB = 5,
    E_FndScal_01_Desc_OR_MB        = 6,
    E_FndScal_01_Desc_OR_EL        = 7,
    E_FndScal_01_Desc_HYP          = 8,
    E_FndScal_01_Desc_BHA          = 9,
    E_FndScal_01_Desc_Presc_BHA    = 10,
    E_FndScal_01_Desc_Presc_BTB_EL = 11,
    E_FndScal_01_Desc_OR_TOFO      = 12,
    E_FndScal_01_Desc_Presc_OR_MB  = 13,
    E_FndScal_01_Desc_TotTrig      = 14,
    E_FndScal_01_Desc_Time         = 15, // (100 Hz)   
    E_FndScal_01_Desc_End          = 16
  };
  
  enum E_Fnd_Scal_23_Descr{ // index of the "single-scaler" array (mod 2-3)
    E_FndScal_23_Desc_HYP          = 0,
    E_FndScal_23_Desc_BHA          = 1,
    E_FndScal_23_Desc_Presc_OR_EL  = 2,
    E_FndScal_23_Desc_BTB_EL       = 3,
    E_FndScal_23_Desc_OR_TOFO      = 4,
    E_FndScal_23_Desc_LASER        = 5,
    E_FndScal_23_Desc_OR_TOFI      = 6,
    E_FndScal_23_Desc_SC           = 7, // slow coincidence
    E_FndScal_23_Desc_BTB_MB       = 8,
    E_FndScal_23_Desc_ORtofiORtofo = 9, 
    E_FndScal_23_Desc_ORelORtofo   = 10,
    E_FndScal_23_Desc_void1        = 11,// slow coincidence
    E_FndScal_23_Desc_void2        = 12,// slow coincidence
    E_FndScal_23_Desc_RF           = 13, // RF over 4
    E_FndScal_23_Desc_Trig_Busy    = 14,
    E_FndScal_23_Desc_Time         = 15, // (100 Hz)  
    E_FndScal_23_Desc_End          = 16,
  };

  enum E_Fnd_Scal_4_Descr{ // index of the "single-scaler" array (mod 4)
    E_FndScal_4_Desc_TOFI_01       = 0,
    E_FndScal_4_Desc_TOFI_02       = 1,
    E_FndScal_4_Desc_TOFI_03       = 2,
    E_FndScal_4_Desc_TOFI_04       = 3,
    E_FndScal_4_Desc_TOFI_05       = 4,
    E_FndScal_4_Desc_TOFI_06       = 5,
    E_FndScal_4_Desc_TOFI_07       = 6,
    E_FndScal_4_Desc_TOFI_08       = 7,
    E_FndScal_4_Desc_TOFI_09       = 8,
    E_FndScal_4_Desc_TOFI_10       = 9,
    E_FndScal_4_Desc_TOFI_11       = 10,
    E_FndScal_4_Desc_TOFI_12       = 11,
    E_FndScal_4_Desc_End           = 12,
  };
  
  TString GetScal_01_DescName(Int_t bit)
    {
      // bit: 0,...,15
      E_Fnd_Scal_01_Descr desc = (E_Fnd_Scal_01_Descr)(bit);
      switch(desc){
      case E_FndScal_01_Desc_M27_HYP:      return "Mul 2-7 (HYP).";
      case E_FndScal_01_Desc_M27_BHA:      return "Mul 2-7 (BHA).";
      case E_FndScal_01_Desc_BTB_11:       return "BTB 1-1.......";
      case E_FndScal_01_Desc_BTB_EL:       return "BTB EL........";
      case E_FndScal_01_Desc_PC:           return "Prompt Coinc. ";
      case E_FndScal_01_Desc_Presc_BTB_MB: return "Presc. BTB MB.";
      case E_FndScal_01_Desc_OR_MB:        return "OR TOFINO (MB)";
      case E_FndScal_01_Desc_OR_EL:        return "OR TOFINO (EL)";
      case E_FndScal_01_Desc_HYP:          return "HYP...........";
      case E_FndScal_01_Desc_BHA:          return "BHA...........";
      case E_FndScal_01_Desc_Presc_BHA:    return "Presc BHA.....";
      case E_FndScal_01_Desc_Presc_BTB_EL: return "Presc BTB EL..";
      case E_FndScal_01_Desc_OR_TOFO:      return "OR TOFONE.....";
      case E_FndScal_01_Desc_Presc_OR_MB:  return "Presc OR MB...";
      case E_FndScal_01_Desc_TotTrig:      return "Total Triggers";
      case E_FndScal_01_Desc_Time:         return "Time..........";
      default :
        Error("GetScal_01_DescName","Scalers bit %d not available",bit);
        return "";
      }
    }

  TString GetScal_23_DescName(Int_t bit)
    {
      // bit: 0,...,15
      E_Fnd_Scal_23_Descr desc = (E_Fnd_Scal_23_Descr)(bit);
      switch(desc){
      case E_FndScal_23_Desc_HYP:          return "HYP..............";
      case E_FndScal_23_Desc_BHA:          return "BHA..............";
      case E_FndScal_23_Desc_Presc_OR_EL:  return "Presc OR-EL......";
      case E_FndScal_23_Desc_BTB_EL:       return "BTB-EL...........";
      case E_FndScal_23_Desc_OR_TOFO:      return "OR-TOFONE........";
      case E_FndScal_23_Desc_LASER:        return "LASER............";
      case E_FndScal_23_Desc_OR_TOFI:      return "OR-TOFINO.(MB)..."; // low-threshold
      case E_FndScal_23_Desc_SC:           return "Slow Coinc. .....";
      case E_FndScal_23_Desc_BTB_MB:       return "BTB-MB...........";
      case E_FndScal_23_Desc_ORtofiORtofo: return "ORtofino*ORtofone";
      case E_FndScal_23_Desc_ORelORtofo:   return "OR-EL*ORtofone...";
      case E_FndScal_23_Desc_void1:        return "(void)...........";
      case E_FndScal_23_Desc_void2:        return "(void)...........";
      case E_FndScal_23_Desc_RF:           return "RF/4.............";
      case E_FndScal_23_Desc_Trig_Busy:    return "Trig*busy........";
      case E_FndScal_23_Desc_Time:         return "Time.............";
      default :
        Error("GetScal_23_DescName","Scalers bit %d not available",bit);
        return "";
      }
    }

  TString GetScal_4_DescName(Int_t bit)
    {
      // bit: 0,...,11
      E_Fnd_Scal_4_Descr desc = (E_Fnd_Scal_4_Descr)(bit);
      switch(desc){
      case E_FndScal_4_Desc_TOFI_01:  return "TOFINO-01";
      case E_FndScal_4_Desc_TOFI_02:  return "TOFINO-02";
      case E_FndScal_4_Desc_TOFI_03:  return "TOFINO-03";
      case E_FndScal_4_Desc_TOFI_04:  return "TOFINO-04";
      case E_FndScal_4_Desc_TOFI_05:  return "TOFINO-05";
      case E_FndScal_4_Desc_TOFI_06:  return "TOFINO-06";
      case E_FndScal_4_Desc_TOFI_07:  return "TOFINO-07";
      case E_FndScal_4_Desc_TOFI_08:  return "TOFINO-08";
      case E_FndScal_4_Desc_TOFI_09:  return "TOFINO-09";
      case E_FndScal_4_Desc_TOFI_10:  return "TOFINO-10";
      case E_FndScal_4_Desc_TOFI_11:  return "TOFINO-11";
      case E_FndScal_4_Desc_TOFI_12:  return "TOFINO-12";
      default :
        Error("GetScal_4_DescName","Scalers bit %d not available",bit);
        return "";
      }
    }
  // ---
  
  Int_t  fDebugLev;  // can be 0-1-2 (default is 'zero')
  Bool_t fIsOnline;  // True in online mode

  ofstream *fCurLogFile; 

  Int_t fTrkStream; // store streamer version for loaded TFndTrack

  // -----
  TTree *fScalTree;  // tree filled with the content of scalers (raw-data)
  TTree *fGesTree;   // tree filled from the content of JFGES zebra bank (before event-reconstruction)
  TTree *fBabaTree;  // tree filled by baba reconstruction process (single helix reconstruction)
  TTree *fBhabhaTree; // tree filled by bhabha reconstruction process (double helix reconstruction)
  TTree *fHypeTree;   // tree filled by hype reconstruction process (full reconstruction)

  // ----- scalers reading (used to evaluate rates) -----
  //  temporary buffers (containing the whole information)
  UInt_t fScal0[E_FndScal_01_Desc_End]; // scaler number 0 (injection inhibit)
  UInt_t fScal1[E_FndScal_01_Desc_End]; // scaler number 1 (inhibit)
  UInt_t fScal2[E_FndScal_23_Desc_End]; // scaler number 2 (no inhibit)
  UInt_t fScal3[E_FndScal_23_Desc_End]; // scaler number 3 (inhibit)
  UInt_t fScal4[E_FndScal_4_Desc_End];  // scaler number 4 (TOFINO rates)

  // data-members for scaler tree
  Int_t fScalEvNum; // event number used by the scaler tree
  Int_t fScal_Time[E_Scal_Inhib_End]; // real-time (must be used for rates evaluation)
  Int_t fScal_HYP[E_Scal_Inhib_End];  // number of HYPE triggers
  Int_t fScal_BHA[E_Scal_Inhib_End];  // number of BHABHA triggers
  Int_t fScal_ORtofino_MB[E_Scal_Inhib_End]; // number of low-th OR-tofino triggers
  Int_t fScal_ORtofone[E_Scal_Inhib_End]; // number of OR-tofone triggers

  // --- GES content
  Int_t fGesEvNum;     // data-part [2]
  Int_t fGesTrigType;  // data-part [3]
  UInt_t fGesEvTime; // Unix time of the run [4] + Time of the event from the run start (s) [5]
  Int_t fGesNentryBhabha; // current number of entry in Bhabha tree
  Int_t fGesNentryHype; // current number of entry in Hype tree

  // --- Reconstructed information handling
  Int_t fForeseenSign[2]; // foreseen sign for reconstructed tracks

  // Bhabha tree variables (see the "EEVERT1" routine for variables meaning)
  Int_t   fBhaEvNum; // software number of event
  Bool_t  fBhaVertImpossible; // true if one track only is present (vertex determination impossible)
  Bool_t  fBhaVertDoubtful; // true if the distance between e-,e+ tracks is too large (vertex doubtful)
  Float_t fBhaVertPos[3]; // bhabha vertex position (x,y,z) 
  Float_t fBhaMom[2];     // see FIN_PHYS::E_FinPhys_BhabhaPart_ID
  Float_t fBhaPipe_X[2];  // X position at outer beam-pipe  (after backtracking)
  Float_t fBhaPipe_Y[2];  // Y position at outer beam-pipe  (after backtracking)
  Float_t fBhaPipe_Z[2];  // Z position at outer beam-pipe  (after backtracking)
  Float_t fBhaDirCos_X[2];     // X director cosines (after backtracking)
  Float_t fBhaDirCos_Y[2];     // X director cosines (after backtracking)
  Float_t fBhaDirCos_Z[2];     // X director cosines (after backtracking)
  Bool_t  fBhaIsLong[2];   // false/true : short-track/long-track
  Float_t fBhaPipeAngle[2];   // angle between the track and the normal to the pipe (deg)
  Float_t fBhaTofiTime[2]; // Calibrated time at TOFINO (added after version 9.00/01)
  Float_t fBhaTofoTime[2]; // Calibrated time at TOFONE (added after version 9.00/01)
  Float_t fBhaTrackLength[2]; // track length is needed for displaying purposes
  // --- both tracks will be required for filling following variables:
  Float_t fBhaInvMass_EpluEmin; // invariant mass (hypothesys: bhabha e+e- pair)
  Float_t fBhaInvMass_PIpluPImin; // invariant mass (hypothesys: K_short decay)
  Float_t fBhaTotalMomentum; // scalar product of e+e- tracks
  Float_t fBhaAngle; // angle between the two tracks (deg)

  // --- work in progress: adding geometric info for displaying purposes
  Float_t fBhaGeoPRradius[2];    // radius (1/curvature) from P.R. zebra: 4
  Float_t fBhaGeoFitPoint_X[2];    // zebra: 13
  Float_t fBhaGeoFitPoint_Y[2];    // zebra: 14
  Float_t fBhaGeoFitPoint_Z[2];    // zebra: 15
  Float_t fBhaGeoFitDirCos_X[2];   // zebra: 16
  Float_t fBhaGeoFitDirCos_Y[2];   // zebra: 17
  Float_t fBhaGeoFitDirCos_Z[2];   // zebra: 18

  // Hype tree variables
  Int_t   fHypEvNum; // software number of event
  Int_t fHyp_PattRec_Err;      // K+/K- pattern recognition error-code (see zebra doc [blankdek])
  Float_t fHyp_PhiDecayPos[3]; // x,y,z position of the PHI dacay position (MRS)

  Int_t fHypStopID[2]; // encoded identification of K-/K+ stopping volume
  Int_t fHypStopTgtNum[2];  // number of the target the kaon stopped into (-1 if out of targets)
  Int_t fHypStopVolNum[2];  // number of the volume the kaon stopped into (-1 if inside targets)

  Float_t fHyp_K_StopPoint_X[2]; // x position of the kaon stopping point (MRS)
  Float_t fHyp_K_StopPoint_Y[2]; // y position of the kaon stopping point (MRS)
  Float_t fHyp_K_StopPoint_Z[2]; // z position of the kaon stopping point (MRS)

  Float_t fHyp_K_DeDx_Isi[2]; // dE/dX in ISIM

  // --- work in progress: adding geometric info for displaying purposes
  //       Check if using double-helix method
  Float_t fHyp_K_Vert_Center_X[2]; //zebra: GRAV+1  
  Float_t fHyp_K_Vert_Center_Y[2]; //zebra: GRAV+2
  Float_t fHyp_K_Vert_Radius[2];   //zebra: GRAV+3

  // --- support for reconstructed tracks
  //  tracks are used for physics analysis
  //  bhabha tracks are handled apart (single data-members),
  //  since the reconstructed bhabha structure is fixed:
  //  handling every kind of track inside the spectrometer
  //  is more complex, then a dedicated class is used
  TClonesArray *fTracks_Kmin; // tracks starting from the K- vertex
  TClonesArray *fTracks_Kplu; // tracks starting from the K+ vertex
  TClonesArray *fVertices; // Vertices different from K+/K- stop

  TFndTrack *fCurTrack; // used while filling tracks from zebra
  TFndVertex *fCurVertex; // used while filling vertices from zebra

 protected:
  TFile *fCurPreanFile; // current root-file (contains pre-analysis trees)
  void AddTrack(Int_t part_id);  // will change fCurTrack
  void AddVertex();  // will change fCurVertex

 private:
  void InitTrees();

 public:
  explicit TFndPrean(Int_t debug_lev = 0);
  ~TFndPrean();
  
  void SetDeamonBehaviour(); // Set class attributes for using it as a deamon
  void Init(Bool_t is_online);
  void Finish();
  Int_t SaveTrees(TFile *file); // save all trees to the given root-file
  
  void ResetDataVariables(); // reset all variables used by trees
  void Reset(); // empty all trees, reset rates evaluation variables
  
  
  // --- General preanalysis methods
  Int_t GoToEvent(const Int_t &ev_num,Int_t &mode); // find event "ev_num" (numbering starts from 1) in every tree

  void ReadScalers(UInt_t *RawEvent);
  void InspectGES(TFndProcessRec *prec);
  void AnalyzeBhabha(TFndProcessRec *prec);
  void AnalyzeHype(TFndProcessRec *prec);
  void UpdateGESTree(); // some information are referred to other trees

  TTree *GetScalTree()   { return fScalTree; }
  TTree *GetGesTree()    { return fGesTree; }
  TTree *GetBabaTree()   { return fBabaTree; }
  TTree *GetBhabhaTree() { return fBhabhaTree; }
  TTree *GetHypeTree()   { return fHypeTree; }

  void PrintScaler(Int_t scal_id); // scal_id = 0,1,2,3,4

  // --- support for reading saved preanalysis-trees
  Int_t LoadTrees(const TString &fNam);
  Int_t CloseCurrentSourceFile();

  void SetLogFile(ofstream *logfile) 
    { //  logfile must be deleted out of here.
      fCurLogFile = logfile;
    }

  static TClonesArray *fgKminTracks;
  static TClonesArray *fgKpluTracks;
  static TClonesArray *fgVertices;

  //--- check methods ---
  Int_t CheckTofinoCFDs(TFndProcessRec *prec); // find CFD glitches

  friend class TFndEvd;

  ClassDef(TFndPrean,0) // Pre-analysis class: heart of the preanalysis program
};

#endif  // FROOT_TFndPrean

---