PamelaLevel2/inc/PamLevel2.h



/**
 * \file PamLevel2.h
 * \author Elena Vannuccini
 */
#ifndef pamlevel2_h
#define pamlevel2_h

#include <TRefArray.h>

#include <TrkStruct.h>
#include <TrkParams.h>
#include <TrkLevel0.h>
#include <TrkLevel1.h>
#include <TrkLevel2.h>
#include <TrkHough.h>
#include <CaloLevel0.h>
#include <CaloLevel1.h>
#include <CaloLevel2.h>
#include <ToFLevel2.h>
#include <TrigLevel2.h>
#include <S4Level2.h>
#include <NDLevel2.h>
#include <AcLevel2.h>
#include <OrbitalInfo.h>
#include <InclinationInfo.h> // EMILIANO, now this library is included in libDarthVader.so
                         // and the header installed in $PAM_INC. If you get the
                         // error "Cannot find InclinationInfo.h" update your DV installation
#include <OrientationInfo.h> // EMILIANO, now this library is included in libDarthVader.so
                         // and the header installed in $PAM_INC. If you get the
                         // error "Cannot find InclinationInfo.h" update your DV installation
#include <RunInfo.h>
#include <GLTables.h>


#include <TTree.h>
#include <TFriendElement.h>
#include <TChain.h>
#include <TChainElement.h>
#include <TFile.h>
#include <TList.h>
#include <TKey.h>
#include <TSystemFile.h>
#include <TSystemDirectory.h>
#include <TSQLServer.h>

#include <stdlib.h>
#include <iostream>
#include <iomanip>
using namespace std;

#define NCLONES 10

using namespace std;


/////////////////////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////////////////////
/**
 * \brief Class to describe GP output (to be moved to PamelaDigitizer)
 *
 */

class GPamela : public TObject {

 private:

 public:

    // Declaration of leave types
    Int_t           Irun;
    Int_t           Ievnt;
    Int_t           Ipa;
    Float_t         X0;
    Float_t         Y0;
    Float_t         Z0;
    Float_t         Theta;
    Float_t         Phi;
    Float_t         P0;
    Int_t           Nthtof;
    UChar_t         Ipltof[100];   //[Nthtof]
    UChar_t         Ipaddle[100];   //[Nthtof]
    UChar_t         Ipartof[100];   //[Nthtof]
    Float_t         Xintof[100];   //[Nthtof]
    Float_t         Yintof[100];   //[Nthtof]
    Float_t         Zintof[100];   //[Nthtof]
    Float_t         Xouttof[100];   //[Nthtof]
    Float_t         Youttof[100];   //[Nthtof]
    Float_t         Zouttof[100];   //[Nthtof]
    Float_t         Ereltof[100];   //[Nthtof]
    Float_t         Timetof[100];   //[Nthtof]
    Float_t         Pathtof[100];   //[Nthtof]
    Float_t         P0tof[100];   //[Nthtof]
    Int_t           Nthcat;
    UChar_t         Iparcat[50];   //[Nthcat]
    UChar_t         Icat[50];   //[Nthcat]
    Float_t         Xincat[50];   //[Nthcat]
    Float_t         Yincat[50];   //[Nthcat]
    Float_t         Zincat[50];   //[Nthcat]
    Float_t         Xoutcat[50];   //[Nthcat]
    Float_t         Youtcat[50];   //[Nthcat]
    Float_t         Zoutcat[50];   //[Nthcat]
    Float_t         Erelcat[50];   //[Nthcat]
    Float_t         Timecat[50];   //[Nthcat]
    Float_t         Pathcat[50];   //[Nthcat]
    Float_t         P0cat[50];   //[Nthcat]
    Int_t           Nthcas;
    UChar_t         Iparcas[50];   //[Nthcas]
    UChar_t         Icas[50];   //[Nthcas]
    Float_t         Xincas[50];   //[Nthcas]
    Float_t         Yincas[50];   //[Nthcas]
    Float_t         Zincas[50];   //[Nthcas]
    Float_t         Xoutcas[50];   //[Nthcas]
    Float_t         Youtcas[50];   //[Nthcas]
    Float_t         Zoutcas[50];   //[Nthcas]
    Float_t         Erelcas[50];   //[Nthcas]
    Float_t         Timecas[50];   //[Nthcas]
    Float_t         Pathcas[50];   //[Nthcas]
    Float_t         P0cas[50];   //[Nthcas]
    Int_t           Nthspe;
    UChar_t         Iparspe[100];   //[Nthspe]
    UChar_t         Itrpb[100];   //[Nthspe]
    UChar_t         Itrsl[100];   //[Nthspe]
    UChar_t         Itspa[100];   //[Nthspe]
    Float_t         Xinspe[100];   //[Nthspe]
    Float_t         Yinspe[100];   //[Nthspe]
    Float_t         Zinspe[100];   //[Nthspe]
    Float_t         Xoutspe[100];   //[Nthspe]
    Float_t         Youtspe[100];   //[Nthspe]
    Float_t         Zoutspe[100];   //[Nthspe]
    Float_t         Xavspe[100];   //[Nthspe]
    Float_t         Yavspe[100];   //[Nthspe]
    Float_t         Zavspe[100];   //[Nthspe]
    Float_t         Erelspe[100];   //[Nthspe]
    Float_t         Pathspe[100];   //[Nthspe]
    Float_t         P0spe[100];   //[Nthspe]
    UChar_t         Nxmult[100];   //[Nthspe]
    UChar_t         Nymult[100];   //[Nthspe]
    Int_t           Nstrpx;
    UChar_t         Npstripx[1000];   //[Nstrpx]
    UChar_t         Ntstripx[1000];   //[Nstrpx]
    UShort_t        Istripx[1000];   //[Nstrpx]
    //UInt_t        Istripx[1000];   //[Nstrpx]
    Float_t         Qstripx[1000];   //[Nstrpx]
    Float_t         Xstripx[1000];   //[Nstrpx]
    Int_t           Nstrpy;
    UChar_t         Npstripy[1000];   //[Nstrpy]
    UChar_t         Ntstripy[1000];   //[Nstrpy]
    UShort_t        Istripy[1000];   //[Nstrpy]
    //UInt_t        Istripy[1000];   //[Nstrpy]
    Float_t         Qstripy[1000];   //[Nstrpy]
    Float_t         Ystripy[1000];   //[Nstrpy]
    Int_t           Nthcali;
    UChar_t         Icaplane[4224];   //[Nthcali]
    UChar_t         Icastrip[4224];   //[Nthcali]
    Int_t           Icamod[4224];   //[Nthcali]
    Float_t         Enestrip[4224];   //[Nthcali]
    Int_t           Nthcal;
    UChar_t         Icapl[1000];   //[Nthcal]
    UChar_t         Icasi[1000];   //[Nthcal]
    UChar_t         Icast[1000];   //[Nthcal]
    Float_t         Xincal[1000];   //[Nthcal]
    Float_t         Yincal[1000];   //[Nthcal]
    Float_t         Zincal[1000];   //[Nthcal]
    Float_t         Erelcal[1000];   //[Nthcal]
    Int_t           Nthnd;
    UChar_t         Itubend[200];   //[Nthnd]
    UChar_t         Iparnd[200];   //[Nthnd]
    Float_t         Xinnd[200];   //[Nthnd]
    Float_t         Yinnd[200];   //[Nthnd]
    Float_t         Zinnd[200];   //[Nthnd]
    Float_t         Xoutnd[200];   //[Nthnd]
    Float_t         Youtnd[200];   //[Nthnd]
    Float_t         Zoutnd[200];   //[Nthnd]
    Float_t         Erelnd[200];   //[Nthnd]
    Float_t         Timend[200];   //[Nthnd]
    Float_t         Pathnd[200];   //[Nthnd]
    Float_t         P0nd[200];   //[Nthnd]
    Int_t           Nthcard;
    UChar_t         Iparcard[100];   //[Nthcard]
    UChar_t         Icard[100];   //[Nthcard]
    Float_t         Xincard[100];   //[Nthcard]
    Float_t         Yincard[100];   //[Nthcard]
    Float_t         Zincard[100];   //[Nthcard]
    Float_t         Xoutcard[100];   //[Nthcard]
    Float_t         Youtcard[100];   //[Nthcard]
    Float_t         Zoutcard[100];   //[Nthcard]
    Float_t         Erelcard[100];   //[Nthcard]
    Float_t         Timecard[100];   //[Nthcard]
    Float_t         Pathcard[100];   //[Nthcard]
    Float_t         P0card[100];   //[Nthcard]

    GPamela(){ Clear(); };
//    ~GPamela(){ Delete(); };
    ~GPamela(){ Clear(); };

    void Clear();
    void Delete();
    void SetBranchAddress( TChain* );
    void SetBranchAddress( TTree* t){ SetBranchAddress( (TChain*)t );};

    GPamela*    GetGPamela(){return this;};

    ClassDef(GPamela,1);

};
/////////////////////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////////////////////
/////////////////////////////////////////////////////////////////////////////


/**
 * \brief Class to describe particle tracks in the apparatus, including tracker and track-related variables from other detectors (calorimeter and TOF).
 *
 * This class inherites from TrkTrack, CaloTrkVar ecc...
 *
 */
//class PamTrack : public TrkTrack, public CaloTrkVar, public ToFTrkVar {
class PamTrack : public TObject {

 private:

    TrkTrack*   trk_track;
    CaloTrkVar* calo_track;
    ToFTrkVar*  tof_track;
    OrbitalInfoTrkVar*  orb_track;
    Int_t pscore;
    Int_t iscore;
    bool candeleteobj;

//    TClonesArray *Cluster;        ///< clusters (if stored)

 public:

    PamTrack();
    PamTrack(TrkTrack* , CaloTrkVar* , ToFTrkVar*, OrbitalInfoTrkVar* );
    PamTrack(const PamTrack&);
    ~PamTrack(){ Delete(); };

    void Clear();
    void Clear(Option_t *option){Clear();};
    void Delete();

    Bool_t IsSolved(){if ((pscore-iscore)>0) return true; return false;};
    Int_t GetScore(){return (pscore-iscore);};
    Int_t GetPScore(){return (pscore);};
    Int_t GetIScore(){return (iscore);};
    void SetPScore(Int_t p){pscore=p;};
    void SetIScore(Int_t i){iscore=i;};

    TrkTrack*   GetTrkTrack(){return trk_track;};
    CaloTrkVar* GetCaloTrack(){return calo_track;};
    ToFTrkVar*  GetToFTrack(){return tof_track;};
    OrbitalInfoTrkVar*  GetOrbitalInfoTrack(){return orb_track;};

    ClassDef(PamTrack,2);
};

/**
 * \brief Class to describe PAMELA LEVEL2 data.
 *
 * This class inherites from TrkLevel2, CaloLevel2 ecc... and it overrides the TrkLevel2:GetTrack(int it) and TrkLevel2:GetTrackImage(int it) methods.
 * This allows to solve the tracker ambiguity (that is, to choose between track images) by using also information from other detectors.
 */

//class PamLevel2 : public CaloLevel2, public CaloLevel1, public ToFLevel2, public TrkLevel2, public TrkLevel1, public TrkHough, public TrigLevel2, public S4Level2, public NDLevel2, public AcLevel2, public OrbitalInfo {
class PamLevel2 : public TObject {

 private:

    // ------------------------------
    // variables to retrieve run info
    // ------------------------------
    Long64_t  irun;          //current run entry, relative to the TChain
    Long64_t  irunt;         //current run entry, relative to the TTree in the TChain
    Long64_t  runfirstentry; //first entry of run, relative to the TChain
    Long64_t  runlastentry;  //last  entry of run, relative to the TChain
    Long64_t  irunentry;     //position of current entry, relative to the run
    Long64_t  totrunentry;
    Long64_t  totrunentrymax;
    Long64_t  totrunentrymin;
    ULong64_t    prevabstime; // abs time of previous event
    UInt_t    prevpktnum; // packet number of previous event
    ULong64_t    abstime; // current abs time 
    UInt_t    pktnum; // current packet number 
    Bool_t isFragment;
    Bool_t isSync; ///< the file is fully sinchronized between L2 and L0/DB data
        //    Bool_t hasL0EE; ///< backward compatibility flag for preselected event list
    Long64_t il0entry; ///< entry of level0 file in selection list
    TChain*   run_tree;
    TTree*    run_tree_clone;
    TChain*   sel_tree;
    TTree*    sel_tree_clone;
    int *irunoffset;
    GL_TIMESYNC *gltsync; // Emiliano
    Bool_t fUpdateRunInfo; // Emiliano
    Bool_t fUseDBinRunInfo; // Emiliano

    // -------------------------------------------
    // variables to store total live and dead time
    // -------------------------------------------
    ULong64_t  totdltime[3];// 0=live 1=dead 2=n.events

    // ---------------------------------
    // variables to retrieve level0 info
    // ---------------------------------
    TFile*      l0_file;
    TTree*      l0_tree;
    Int_t       iroot;
    Long64_t       prevshift; // shift needed to sync level2 data with the DB (nested+DV_skipped packets) - global position in level2 file EM
    Long64_t       yprevshift; // shift needed to sync level2 data with level0 data (nested+DV_skipped packets) - run wise EM
    Int_t       maxshift; // EMILIANO maximum discrepancy between L2 and L0 data (the bigger the better concering synch but the processing speed drops) default = 10
    // --------------------------------------------
    // variables to store Level0 info (if required)
    // --------------------------------------------
    EventHeader* h0_obj;
    TrkLevel0*   trk0_obj;
    CaloLevel0*  calo0_obj;

    // ---------------------------
    // variables to connect the DB
    // ---------------------------
    TSQLServer* dbc;
    TString host;
    TString user;
    TString psw;

    // ---------------------------------------------
    // variables to retrieve and clone level1-2 info
    // ---------------------------------------------
    TChain* pam_tree;
    TTree*  pam_tree_clone[NCLONES];

    // --------------------------------------------
    // variables to store Level1-2 info
    // --------------------------------------------
    TrkLevel1*   trk1_obj;
    TrkLevel2*   trk2_obj;
    TrkHough*    trkh_obj;
    CaloLevel1*  calo1_obj;
    CaloLevel2*  calo2_obj;
    ToFLevel2*   tof_obj;
    TrigLevel2*  trig_obj;
    S4Level2*    s4_obj;
    NDLevel2*    nd_obj;
    AcLevel2*    ac_obj;
    OrbitalInfo* orb_obj;
    GL_RUN*      run_obj;
    SoftInfo*    soft_obj; // Emiliano

    // --------------------------------------------
    // variables to handle GP
    // --------------------------------------------
    GPamela*     gp_obj;

    // -----------------------------------
    // variables to sort the track
    // -----------------------------------
    TClonesArray *tsorted;
    TClonesArray *timage;
    PamTrack    *track; //store the last required
    TString howtosort;
    Float_t sortthr;

    Bool_t issorted;     // true if the current event has already been sorted with the current sort criteria
    Long64_t lastsorted; // the last event whose tracks has been sorted
    // --------------------------------------
    // flags to load level0-1-2 trees/braches
    // --------------------------------------
    Bool_t CAL2; //level2 branch
    Bool_t CAL1; //level1 branch
    Bool_t CAL0; //yoda

    Bool_t TRK2; //level2 branch
    Bool_t TRK1; //level1 branch
    Bool_t TRKh; //hough branch
    Bool_t TRK0; //yoda

    Bool_t TOF;
    Bool_t TOF0;

    Bool_t TRG;
    Bool_t S4;
    Bool_t ND;
    Bool_t AC;
    Bool_t ORB;

    Bool_t GP; //GP simulation tree

    // -----------
    // other flags
    // -----------
    Bool_t RUN;   // RunInfo tree
    Bool_t ISGP;  // GP simulation (...sort of patch, to skip OBT check, since OrbitalInfo is sometime missing)
    Int_t  SELLI; // tree of selected events (there is a selection list)

    Bool_t DBG;

    void Initialize();


 public:

    PamLevel2(TString ddir,TList *list,TString detlist);
    PamLevel2(TString ddir,TString list,TString detlist);
    PamLevel2(TString ddir,TString list);
    PamLevel2();
    ~PamLevel2(){Delete();};

    void Clear();
    void Reset();
    void Delete();


//  =============================================
//  methods needed to read/write Level0-1-2 trees
//  =============================================
    void    SetWhichTrees(TString);
    void    GetWhichTrees(TFile*);

    TTree*  GetPamTree(TFile*,TString);
    TTree*  GetPamTree(TFile* f){return GetPamTree(f,"");};

    TList*  GetListOfLevel2Files(TString, TString);
    Bool_t  CheckLevel2File(TString);
    TChain* GetPamTree(TList*,TString);
    TChain* GetPamTree(TString ddir,TString list,TString detlist){ return GetPamTree(GetListOfLevel2Files(ddir,list),detlist); };
    TChain* GetPamTree(TString ddir,TString list){ return GetPamTree(GetListOfLevel2Files(ddir,list),""); };
    TChain* AddFriend(TString);

    TTree*  GetRunTree(TFile*);
    TChain* GetRunTree(TList*);
    TChain* GetRunTree(TString ddir,TString list){ return GetRunTree(GetListOfLevel2Files(ddir,list)); };
    Bool_t  UpdateRunInfo(Long64_t);///< new implementation to solve Mori's bug
    Bool_t  UpdateRunInfo(TTree *,Long64_t);
    Bool_t  UpdateRunInfo(TChain *,Long64_t);

    void    SkipRunInfoUpdate(); // Emiliano
    void    DoRunInfoUpdate(){fUpdateRunInfo = true;}; // [default] Emiliano
    void    UseDBinRunInfo(){fUseDBinRunInfo = true;}; // [default] Emiliano
    void    NoDBinRunInfo(){fUseDBinRunInfo = false;}; // EM
    void    NoDBconnections(){SkipRunInfoUpdate();}; // Emiliano

    void    CreateCloneTrees0(TChain *fChain, TFile* file);
    void    CreateCloneTrees(TFile* file);
    void    CreateCloneTrees(){CreateCloneTrees(NULL);};
    void    FillCloneTrees();
    TTree  *GetCloneTree(TString);
    void    WriteCloneTrees();

    void    SetBranchAddress(TTree *);
    void    SetBranchAddress(TChain *);

//    Int_t   GetEntry(Int_t);
    Int_t   GetEntry(Long64_t);
    Long64_t GetReadEntry();
    Int_t   GetYodaEntry();

    void SetMaxShift(Int_t sh); // EMILIANO method to change the maxshift variable

    Long64_t GetEntries(){return pam_tree->GetEntries();};
    TChain* GetPamTree(){return pam_tree;};
    TChain* GetRunTree(){return run_tree;};
    TTree*  GetYodaTree();

    TFile*  GetYodaFile(){return l0_file;}; // can be useful to know the name of level0 file we are using or similar things

//  =============
//  DB connection
//  =============
    void SetDBHost(TString str){host=str;};
    void SetDBUser(TString str){user=str;};
    void SetDBPsw(TString str){psw=str;};
    TSQLServer* GetSQLServer(){return dbc;};
    Bool_t SetDBConnection();

//  ========================================================
//  methods needed to retrieve subdetector general variables
//  ========================================================
    void*       GetPointerTo(const char*);
    EventHeader  *GetEventHeader()  { return h0_obj; };

/*     TrkLevel0    *GetTrkLevel0()   { return trk0_obj; }; */
/*     CaloLevel0   *GetCaloLevel0()  { return calo0_obj; }; */
    TrkLevel0    *GetTrkLevel0();
    CaloLevel0   *GetCaloLevel0();


    TrkLevel1    *GetTrkLevel1()  { return trk1_obj; };
    TrkLevel2    *GetTrkLevel2()  { return trk2_obj; };
    TrkHough     *GetTrkHough()   { return trkh_obj; };
    CaloLevel1   *GetCaloLevel1() { return calo1_obj;};
    CaloLevel2   *GetCaloLevel2() { return calo2_obj;};
    ToFLevel2    *GetToFLevel2()  { return tof_obj;  };
    TrigLevel2   *GetTrigLevel2() { return trig_obj; };
    S4Level2     *GetS4Level2()   { return s4_obj;   };
    NDLevel2     *GetNDLevel2()   { return nd_obj;   };
    AcLevel2     *GetAcLevel2()   { return ac_obj;   };
    OrbitalInfo  *GetOrbitalInfo(){ return orb_obj;  };
    GL_RUN       *GetRunInfo()    { return run_obj;  };
    SoftInfo     *GetSoftInfo()   { return soft_obj; }; // Emiliano
    GPamela      *GetGPamela()    { return gp_obj;   };

//  ==============================================
//  methods to retrieve subdetectors stored tracks
//  ==============================================
    TrkTrack*   GetTrkStoredTrack(int seqno){return trk2_obj->TrkLevel2::GetStoredTrack(seqno); };
    CaloTrkVar* GetCaloStoredTrack(int);
    ToFTrkVar*  GetToFStoredTrack(int);
    OrbitalInfoTrkVar*  GetOrbitalInfoStoredTrack(int);
    //
    PamTrack*   GetPamTrackAlong(TrkTrack*);
//      PamTrack*   GetPamTrackAlong(CaloTrkVar*);      //se serve
//      PamTrack*   GetPamTrackAlong(ToFTrkVar*);      //se serve

//  =================================
//  methods to retrieve pamela tracks
//  =================================
    PamTrack*     GetStoredTrack(Int_t);
    //
    void          SortTracks(TString);          //assign value to variable sorted_tracks
    void          SortTracks();          //assign value to variable sorted_tracks
//    TRefArray*    GetTracks();                  //return sorted_tracks
    TClonesArray* GetTracks();                  //return sorted_tracks
    PamTrack*     GetTrack(int);
    PamTrack*     GetTrackImage(int);
    void          SetSortingMethod(TString how);
    TString       GetSortingMethod(){return howtosort;};
    void          SetNucleiSortingThreshold(Float_t nucleithr){ sortthr = nucleithr; return;};
    Float_t       GetNucleiSortingThreshold(){ return sortthr;};

//  =================================
//  other methods
//  =================================
//    Int_t ntrk(){ return trk2_obj->TrkLevel2::ntrk(); };


    Long64_t  GetRunID(){ return irun;};
    Long64_t  GetRunEntryID(){ return irunentry; };

    // -----
    // flags methods
    // -----
    Bool_t IsCAL2(){return CAL2;}; //level2 branch
    Bool_t IsCAL1(){return CAL1;}; //level1 branch
    Bool_t IsCAL0(){return CAL0;}; //yoda

    Bool_t IsTRK2(){return TRK2;}; //level2 branch
    Bool_t IsTRK1(){return TRK1;}; //level1 branch
    Bool_t IsTRKh(){return TRKh;}; //hough branch
    Bool_t IsTRK0(){return TRK0;}; //yoda

    Bool_t IsTOF(){return TOF;};
    Bool_t IsTOF0(){return TOF0;};

    Bool_t IsTRG(){return TRG;};
    Bool_t IsS4(){return S4;};
    Bool_t IsND(){return ND;};
    Bool_t IsAC(){return AC;};
    Bool_t IsORB(){return ORB;};

    Bool_t IsRUN(){return RUN;};

    Bool_t IsGood(Bool_t strict=true);

    Int_t  IsSELLI(){return SELLI;}; //trees of selected events (there is a selection list)
    void  SetSELLI(Int_t selli){SELLI=selli;};

    Bool_t IsGP(){return ISGP;};

    void   SetGP(Bool_t tt){ISGP=tt;};

    void SetDebug(Bool_t set){ DBG=set; TrkParams::SetVerboseMode(); };


    ClassDef(PamLevel2,2);
};

//===============================================================================================
//
//
//
//
//===============================================================================================


#endif

1
2
3	/**
4	* \file PamLevel2.h
5	* \author Elena Vannuccini
6	*/
7	#ifndef pamlevel2_h
8	#define pamlevel2_h
9
10	#include <TRefArray.h>
11
12	#include <TrkStruct.h>
13	#include <TrkParams.h>
14	#include <TrkLevel0.h>
15	#include <TrkLevel1.h>
16	#include <TrkLevel2.h>
17	#include <TrkHough.h>
18	#include <CaloLevel0.h>
19	#include <CaloLevel1.h>
20	#include <CaloLevel2.h>
21	#include <ToFLevel2.h>
22	#include <TrigLevel2.h>
23	#include <S4Level2.h>
24	#include <NDLevel2.h>
25	#include <AcLevel2.h>
26	#include <OrbitalInfo.h>
27	#include <InclinationInfo.h> // EMILIANO, now this library is included in libDarthVader.so
28	// and the header installed in $PAM_INC. If you get the
29	// error "Cannot find InclinationInfo.h" update your DV installation
30	#include <OrientationInfo.h> // EMILIANO, now this library is included in libDarthVader.so
31	// and the header installed in $PAM_INC. If you get the
32	// error "Cannot find InclinationInfo.h" update your DV installation
33	#include <RunInfo.h>
34	#include <GLTables.h>
35
36
37	#include <TTree.h>
38	#include <TFriendElement.h>
39	#include <TChain.h>
40	#include <TChainElement.h>
41	#include <TFile.h>
42	#include <TList.h>
43	#include <TKey.h>
44	#include <TSystemFile.h>
45	#include <TSystemDirectory.h>
46	#include <TSQLServer.h>
47
48	#include <stdlib.h>
49	#include <iostream>
50	#include <iomanip>
51	using namespace std;
52
53	#define NCLONES 10
54
55	using namespace std;
56
57
58	/////////////////////////////////////////////////////////////////////////////
59	/////////////////////////////////////////////////////////////////////////////
60	/////////////////////////////////////////////////////////////////////////////
61	/////////////////////////////////////////////////////////////////////////////
62	/**
63	* \brief Class to describe GP output (to be moved to PamelaDigitizer)
64	*
65	*/
66
67	class GPamela : public TObject {
68
69	private:
70
71	public:
72
73	// Declaration of leave types
74	Int_t Irun;
75	Int_t Ievnt;
76	Int_t Ipa;
77	Float_t X0;
78	Float_t Y0;
79	Float_t Z0;
80	Float_t Theta;
81	Float_t Phi;
82	Float_t P0;
83	Int_t Nthtof;
84	UChar_t Ipltof[100]; //[Nthtof]
85	UChar_t Ipaddle[100]; //[Nthtof]
86	UChar_t Ipartof[100]; //[Nthtof]
87	Float_t Xintof[100]; //[Nthtof]
88	Float_t Yintof[100]; //[Nthtof]
89	Float_t Zintof[100]; //[Nthtof]
90	Float_t Xouttof[100]; //[Nthtof]
91	Float_t Youttof[100]; //[Nthtof]
92	Float_t Zouttof[100]; //[Nthtof]
93	Float_t Ereltof[100]; //[Nthtof]
94	Float_t Timetof[100]; //[Nthtof]
95	Float_t Pathtof[100]; //[Nthtof]
96	Float_t P0tof[100]; //[Nthtof]
97	Int_t Nthcat;
98	UChar_t Iparcat[50]; //[Nthcat]
99	UChar_t Icat[50]; //[Nthcat]
100	Float_t Xincat[50]; //[Nthcat]
101	Float_t Yincat[50]; //[Nthcat]
102	Float_t Zincat[50]; //[Nthcat]
103	Float_t Xoutcat[50]; //[Nthcat]
104	Float_t Youtcat[50]; //[Nthcat]
105	Float_t Zoutcat[50]; //[Nthcat]
106	Float_t Erelcat[50]; //[Nthcat]
107	Float_t Timecat[50]; //[Nthcat]
108	Float_t Pathcat[50]; //[Nthcat]
109	Float_t P0cat[50]; //[Nthcat]
110	Int_t Nthcas;
111	UChar_t Iparcas[50]; //[Nthcas]
112	UChar_t Icas[50]; //[Nthcas]
113	Float_t Xincas[50]; //[Nthcas]
114	Float_t Yincas[50]; //[Nthcas]
115	Float_t Zincas[50]; //[Nthcas]
116	Float_t Xoutcas[50]; //[Nthcas]
117	Float_t Youtcas[50]; //[Nthcas]
118	Float_t Zoutcas[50]; //[Nthcas]
119	Float_t Erelcas[50]; //[Nthcas]
120	Float_t Timecas[50]; //[Nthcas]
121	Float_t Pathcas[50]; //[Nthcas]
122	Float_t P0cas[50]; //[Nthcas]
123	Int_t Nthspe;
124	UChar_t Iparspe[100]; //[Nthspe]
125	UChar_t Itrpb[100]; //[Nthspe]
126	UChar_t Itrsl[100]; //[Nthspe]
127	UChar_t Itspa[100]; //[Nthspe]
128	Float_t Xinspe[100]; //[Nthspe]
129	Float_t Yinspe[100]; //[Nthspe]
130	Float_t Zinspe[100]; //[Nthspe]
131	Float_t Xoutspe[100]; //[Nthspe]
132	Float_t Youtspe[100]; //[Nthspe]
133	Float_t Zoutspe[100]; //[Nthspe]
134	Float_t Xavspe[100]; //[Nthspe]
135	Float_t Yavspe[100]; //[Nthspe]
136	Float_t Zavspe[100]; //[Nthspe]
137	Float_t Erelspe[100]; //[Nthspe]
138	Float_t Pathspe[100]; //[Nthspe]
139	Float_t P0spe[100]; //[Nthspe]
140	UChar_t Nxmult[100]; //[Nthspe]
141	UChar_t Nymult[100]; //[Nthspe]
142	Int_t Nstrpx;
143	UChar_t Npstripx[1000]; //[Nstrpx]
144	UChar_t Ntstripx[1000]; //[Nstrpx]
145	UShort_t Istripx[1000]; //[Nstrpx]
146	//UInt_t Istripx[1000]; //[Nstrpx]
147	Float_t Qstripx[1000]; //[Nstrpx]
148	Float_t Xstripx[1000]; //[Nstrpx]
149	Int_t Nstrpy;
150	UChar_t Npstripy[1000]; //[Nstrpy]
151	UChar_t Ntstripy[1000]; //[Nstrpy]
152	UShort_t Istripy[1000]; //[Nstrpy]
153	//UInt_t Istripy[1000]; //[Nstrpy]
154	Float_t Qstripy[1000]; //[Nstrpy]
155	Float_t Ystripy[1000]; //[Nstrpy]
156	Int_t Nthcali;
157	UChar_t Icaplane[4224]; //[Nthcali]
158	UChar_t Icastrip[4224]; //[Nthcali]
159	Int_t Icamod[4224]; //[Nthcali]
160	Float_t Enestrip[4224]; //[Nthcali]
161	Int_t Nthcal;
162	UChar_t Icapl[1000]; //[Nthcal]
163	UChar_t Icasi[1000]; //[Nthcal]
164	UChar_t Icast[1000]; //[Nthcal]
165	Float_t Xincal[1000]; //[Nthcal]
166	Float_t Yincal[1000]; //[Nthcal]
167	Float_t Zincal[1000]; //[Nthcal]
168	Float_t Erelcal[1000]; //[Nthcal]
169	Int_t Nthnd;
170	UChar_t Itubend[200]; //[Nthnd]
171	UChar_t Iparnd[200]; //[Nthnd]
172	Float_t Xinnd[200]; //[Nthnd]
173	Float_t Yinnd[200]; //[Nthnd]
174	Float_t Zinnd[200]; //[Nthnd]
175	Float_t Xoutnd[200]; //[Nthnd]
176	Float_t Youtnd[200]; //[Nthnd]
177	Float_t Zoutnd[200]; //[Nthnd]
178	Float_t Erelnd[200]; //[Nthnd]
179	Float_t Timend[200]; //[Nthnd]
180	Float_t Pathnd[200]; //[Nthnd]
181	Float_t P0nd[200]; //[Nthnd]
182	Int_t Nthcard;
183	UChar_t Iparcard[100]; //[Nthcard]
184	UChar_t Icard[100]; //[Nthcard]
185	Float_t Xincard[100]; //[Nthcard]
186	Float_t Yincard[100]; //[Nthcard]
187	Float_t Zincard[100]; //[Nthcard]
188	Float_t Xoutcard[100]; //[Nthcard]
189	Float_t Youtcard[100]; //[Nthcard]
190	Float_t Zoutcard[100]; //[Nthcard]
191	Float_t Erelcard[100]; //[Nthcard]
192	Float_t Timecard[100]; //[Nthcard]
193	Float_t Pathcard[100]; //[Nthcard]
194	Float_t P0card[100]; //[Nthcard]
195
196	GPamela(){ Clear(); };
197	// ~GPamela(){ Delete(); };
198	~GPamela(){ Clear(); };
199
200	void Clear();
201	void Delete();
202	void SetBranchAddress( TChain* );
203	void SetBranchAddress( TTree* t){ SetBranchAddress( (TChain*)t );};
204
205	GPamela* GetGPamela(){return this;};
206
207	ClassDef(GPamela,1);
208
209	};
210	/////////////////////////////////////////////////////////////////////////////
211	/////////////////////////////////////////////////////////////////////////////
212	/////////////////////////////////////////////////////////////////////////////
213	/////////////////////////////////////////////////////////////////////////////
214
215
216
217	/**
218	* \brief Class to describe particle tracks in the apparatus, including tracker and track-related variables from other detectors (calorimeter and TOF).
219	*
220	* This class inherites from TrkTrack, CaloTrkVar ecc...
221	*
222	*/
223	//class PamTrack : public TrkTrack, public CaloTrkVar, public ToFTrkVar {
224	class PamTrack : public TObject {
225
226	private:
227
228	TrkTrack* trk_track;
229	CaloTrkVar* calo_track;
230	ToFTrkVar* tof_track;
231	OrbitalInfoTrkVar* orb_track;
232	Int_t pscore;
233	Int_t iscore;
234	bool candeleteobj;
235
236	// TClonesArray *Cluster; ///< clusters (if stored)
237
238	public:
239
240	PamTrack();
241	PamTrack(TrkTrack* , CaloTrkVar* , ToFTrkVar, OrbitalInfoTrkVar );
242	PamTrack(const PamTrack&);
243	~PamTrack(){ Delete(); };
244
245	void Clear();
246	void Clear(Option_t *option){Clear();};
247	void Delete();
248
249	Bool_t IsSolved(){if ((pscore-iscore)>0) return true; return false;};
250	Int_t GetScore(){return (pscore-iscore);};
251	Int_t GetPScore(){return (pscore);};
252	Int_t GetIScore(){return (iscore);};
253	void SetPScore(Int_t p){pscore=p;};
254	void SetIScore(Int_t i){iscore=i;};
255
256	TrkTrack* GetTrkTrack(){return trk_track;};
257	CaloTrkVar* GetCaloTrack(){return calo_track;};
258	ToFTrkVar* GetToFTrack(){return tof_track;};
259	OrbitalInfoTrkVar* GetOrbitalInfoTrack(){return orb_track;};
260
261	ClassDef(PamTrack,2);
262	};
263
264	/**
265	* \brief Class to describe PAMELA LEVEL2 data.
266	*
267	* This class inherites from TrkLevel2, CaloLevel2 ecc... and it overrides the TrkLevel2:GetTrack(int it) and TrkLevel2:GetTrackImage(int it) methods.
268	* This allows to solve the tracker ambiguity (that is, to choose between track images) by using also information from other detectors.
269	*/
270
271	//class PamLevel2 : public CaloLevel2, public CaloLevel1, public ToFLevel2, public TrkLevel2, public TrkLevel1, public TrkHough, public TrigLevel2, public S4Level2, public NDLevel2, public AcLevel2, public OrbitalInfo {
272	class PamLevel2 : public TObject {
273
274	private:
275
276	// ------------------------------
277	// variables to retrieve run info
278	// ------------------------------
279	Long64_t irun; //current run entry, relative to the TChain
280	Long64_t irunt; //current run entry, relative to the TTree in the TChain
281	Long64_t runfirstentry; //first entry of run, relative to the TChain
282	Long64_t runlastentry; //last entry of run, relative to the TChain
283	Long64_t irunentry; //position of current entry, relative to the run
284	Long64_t totrunentry;
285	Long64_t totrunentrymax;
286	Long64_t totrunentrymin;
287	ULong64_t prevabstime; // abs time of previous event
288	UInt_t prevpktnum; // packet number of previous event
289	ULong64_t abstime; // current abs time
290	UInt_t pktnum; // current packet number
291	Bool_t isFragment;
292	Bool_t isSync; ///< the file is fully sinchronized between L2 and L0/DB data
293	// Bool_t hasL0EE; ///< backward compatibility flag for preselected event list
294	Long64_t il0entry; ///< entry of level0 file in selection list
295	TChain* run_tree;
296	TTree* run_tree_clone;
297	TChain* sel_tree;
298	TTree* sel_tree_clone;
299	int *irunoffset;
300	GL_TIMESYNC *gltsync; // Emiliano
301	Bool_t fUpdateRunInfo; // Emiliano
302	Bool_t fUseDBinRunInfo; // Emiliano
303
304	// -------------------------------------------
305	// variables to store total live and dead time
306	// -------------------------------------------
307	ULong64_t totdltime[3];// 0=live 1=dead 2=n.events
308
309	// ---------------------------------
310	// variables to retrieve level0 info
311	// ---------------------------------
312	TFile* l0_file;
313	TTree* l0_tree;
314	Int_t iroot;
315	Long64_t prevshift; // shift needed to sync level2 data with the DB (nested+DV_skipped packets) - global position in level2 file EM
316	Long64_t yprevshift; // shift needed to sync level2 data with level0 data (nested+DV_skipped packets) - run wise EM
317	Int_t maxshift; // EMILIANO maximum discrepancy between L2 and L0 data (the bigger the better concering synch but the processing speed drops) default = 10
318	// --------------------------------------------
319	// variables to store Level0 info (if required)
320	// --------------------------------------------
321	EventHeader* h0_obj;
322	TrkLevel0* trk0_obj;
323	CaloLevel0* calo0_obj;
324
325	// ---------------------------
326	// variables to connect the DB
327	// ---------------------------
328	TSQLServer* dbc;
329	TString host;
330	TString user;
331	TString psw;
332
333	// ---------------------------------------------
334	// variables to retrieve and clone level1-2 info
335	// ---------------------------------------------
336	TChain* pam_tree;
337	TTree* pam_tree_clone[NCLONES];
338
339	// --------------------------------------------
340	// variables to store Level1-2 info
341	// --------------------------------------------
342	TrkLevel1* trk1_obj;
343	TrkLevel2* trk2_obj;
344	TrkHough* trkh_obj;
345	CaloLevel1* calo1_obj;
346	CaloLevel2* calo2_obj;
347	ToFLevel2* tof_obj;
348	TrigLevel2* trig_obj;
349	S4Level2* s4_obj;
350	NDLevel2* nd_obj;
351	AcLevel2* ac_obj;
352	OrbitalInfo* orb_obj;
353	GL_RUN* run_obj;
354	SoftInfo* soft_obj; // Emiliano
355
356	// --------------------------------------------
357	// variables to handle GP
358	// --------------------------------------------
359	GPamela* gp_obj;
360
361	// -----------------------------------
362	// variables to sort the track
363	// -----------------------------------
364	TClonesArray *tsorted;
365	TClonesArray *timage;
366	PamTrack *track; //store the last required
367	TString howtosort;
368	Float_t sortthr;
369
370	Bool_t issorted; // true if the current event has already been sorted with the current sort criteria
371	Long64_t lastsorted; // the last event whose tracks has been sorted
372	// --------------------------------------
373	// flags to load level0-1-2 trees/braches
374	// --------------------------------------
375	Bool_t CAL2; //level2 branch
376	Bool_t CAL1; //level1 branch
377	Bool_t CAL0; //yoda
378
379	Bool_t TRK2; //level2 branch
380	Bool_t TRK1; //level1 branch
381	Bool_t TRKh; //hough branch
382	Bool_t TRK0; //yoda
383
384	Bool_t TOF;
385	Bool_t TOF0;
386
387	Bool_t TRG;
388	Bool_t S4;
389	Bool_t ND;
390	Bool_t AC;
391	Bool_t ORB;
392
393	Bool_t GP; //GP simulation tree
394
395	// -----------
396	// other flags
397	// -----------
398	Bool_t RUN; // RunInfo tree
399	Bool_t ISGP; // GP simulation (...sort of patch, to skip OBT check, since OrbitalInfo is sometime missing)
400	Int_t SELLI; // tree of selected events (there is a selection list)
401
402	Bool_t DBG;
403
404	void Initialize();
405
406
407	public:
408
409	PamLevel2(TString ddir,TList *list,TString detlist);
410	PamLevel2(TString ddir,TString list,TString detlist);
411	PamLevel2(TString ddir,TString list);
412	PamLevel2();
413	~PamLevel2(){Delete();};
414
415	void Clear();
416	void Reset();
417	void Delete();
418
419
420	// =============================================
421	// methods needed to read/write Level0-1-2 trees
422	// =============================================
423	void SetWhichTrees(TString);
424	void GetWhichTrees(TFile*);
425
426	TTree* GetPamTree(TFile*,TString);
427	TTree* GetPamTree(TFile* f){return GetPamTree(f,"");};
428
429	TList* GetListOfLevel2Files(TString, TString);
430	Bool_t CheckLevel2File(TString);
431	TChain* GetPamTree(TList*,TString);
432	TChain* GetPamTree(TString ddir,TString list,TString detlist){ return GetPamTree(GetListOfLevel2Files(ddir,list),detlist); };
433	TChain* GetPamTree(TString ddir,TString list){ return GetPamTree(GetListOfLevel2Files(ddir,list),""); };
434	TChain* AddFriend(TString);
435
436	TTree* GetRunTree(TFile*);
437	TChain* GetRunTree(TList*);
438	TChain* GetRunTree(TString ddir,TString list){ return GetRunTree(GetListOfLevel2Files(ddir,list)); };
439	Bool_t UpdateRunInfo(Long64_t);///< new implementation to solve Mori's bug
440	Bool_t UpdateRunInfo(TTree *,Long64_t);
441	Bool_t UpdateRunInfo(TChain *,Long64_t);
442
443	void SkipRunInfoUpdate(); // Emiliano
444	void DoRunInfoUpdate(){fUpdateRunInfo = true;}; // [default] Emiliano
445	void UseDBinRunInfo(){fUseDBinRunInfo = true;}; // [default] Emiliano
446	void NoDBinRunInfo(){fUseDBinRunInfo = false;}; // EM
447	void NoDBconnections(){SkipRunInfoUpdate();}; // Emiliano
448
449	void CreateCloneTrees0(TChain fChain, TFile file);
450	void CreateCloneTrees(TFile* file);
451	void CreateCloneTrees(){CreateCloneTrees(NULL);};
452	void FillCloneTrees();
453	TTree *GetCloneTree(TString);
454	void WriteCloneTrees();
455
456	void SetBranchAddress(TTree *);
457	void SetBranchAddress(TChain *);
458
459	// Int_t GetEntry(Int_t);
460	Int_t GetEntry(Long64_t);
461	Long64_t GetReadEntry();
462	Int_t GetYodaEntry();
463
464	void SetMaxShift(Int_t sh); // EMILIANO method to change the maxshift variable
465
466	Long64_t GetEntries(){return pam_tree->GetEntries();};
467	TChain* GetPamTree(){return pam_tree;};
468	TChain* GetRunTree(){return run_tree;};
469	TTree* GetYodaTree();
470
471	TFile* GetYodaFile(){return l0_file;}; // can be useful to know the name of level0 file we are using or similar things
472
473	// =============
474	// DB connection
475	// =============
476	void SetDBHost(TString str){host=str;};
477	void SetDBUser(TString str){user=str;};
478	void SetDBPsw(TString str){psw=str;};
479	TSQLServer* GetSQLServer(){return dbc;};
480	Bool_t SetDBConnection();
481
482	// ========================================================
483	// methods needed to retrieve subdetector general variables
484	// ========================================================
485	void* GetPointerTo(const char*);
486	EventHeader *GetEventHeader() { return h0_obj; };
487
488	/* TrkLevel0 GetTrkLevel0() { return trk0_obj; }; /
489	/* CaloLevel0 GetCaloLevel0() { return calo0_obj; }; /
490	TrkLevel0 *GetTrkLevel0();
491	CaloLevel0 *GetCaloLevel0();
492
493
494	TrkLevel1 *GetTrkLevel1() { return trk1_obj; };
495	TrkLevel2 *GetTrkLevel2() { return trk2_obj; };
496	TrkHough *GetTrkHough() { return trkh_obj; };
497	CaloLevel1 *GetCaloLevel1() { return calo1_obj;};
498	CaloLevel2 *GetCaloLevel2() { return calo2_obj;};
499	ToFLevel2 *GetToFLevel2() { return tof_obj; };
500	TrigLevel2 *GetTrigLevel2() { return trig_obj; };
501	S4Level2 *GetS4Level2() { return s4_obj; };
502	NDLevel2 *GetNDLevel2() { return nd_obj; };
503	AcLevel2 *GetAcLevel2() { return ac_obj; };
504	OrbitalInfo *GetOrbitalInfo(){ return orb_obj; };
505	GL_RUN *GetRunInfo() { return run_obj; };
506	SoftInfo *GetSoftInfo() { return soft_obj; }; // Emiliano
507	GPamela *GetGPamela() { return gp_obj; };
508
509	// ==============================================
510	// methods to retrieve subdetectors stored tracks
511	// ==============================================
512	TrkTrack* GetTrkStoredTrack(int seqno){return trk2_obj->TrkLevel2::GetStoredTrack(seqno); };
513	CaloTrkVar* GetCaloStoredTrack(int);
514	ToFTrkVar* GetToFStoredTrack(int);
515	OrbitalInfoTrkVar* GetOrbitalInfoStoredTrack(int);
516	//
517	PamTrack* GetPamTrackAlong(TrkTrack*);
518	// PamTrack* GetPamTrackAlong(CaloTrkVar*); //se serve
519	// PamTrack* GetPamTrackAlong(ToFTrkVar*); //se serve
520
521	// =================================
522	// methods to retrieve pamela tracks
523	// =================================
524	PamTrack* GetStoredTrack(Int_t);
525	//
526	void SortTracks(TString); //assign value to variable sorted_tracks
527	void SortTracks(); //assign value to variable sorted_tracks
528	// TRefArray* GetTracks(); //return sorted_tracks
529	TClonesArray* GetTracks(); //return sorted_tracks
530	PamTrack* GetTrack(int);
531	PamTrack* GetTrackImage(int);
532	void SetSortingMethod(TString how);
533	TString GetSortingMethod(){return howtosort;};
534	void SetNucleiSortingThreshold(Float_t nucleithr){ sortthr = nucleithr; return;};
535	Float_t GetNucleiSortingThreshold(){ return sortthr;};
536
537	// =================================
538	// other methods
539	// =================================
540	// Int_t ntrk(){ return trk2_obj->TrkLevel2::ntrk(); };
541
542
543	Long64_t GetRunID(){ return irun;};
544	Long64_t GetRunEntryID(){ return irunentry; };
545
546	// -----
547	// flags methods
548	// -----
549	Bool_t IsCAL2(){return CAL2;}; //level2 branch
550	Bool_t IsCAL1(){return CAL1;}; //level1 branch
551	Bool_t IsCAL0(){return CAL0;}; //yoda
552
553	Bool_t IsTRK2(){return TRK2;}; //level2 branch
554	Bool_t IsTRK1(){return TRK1;}; //level1 branch
555	Bool_t IsTRKh(){return TRKh;}; //hough branch
556	Bool_t IsTRK0(){return TRK0;}; //yoda
557
558	Bool_t IsTOF(){return TOF;};
559	Bool_t IsTOF0(){return TOF0;};
560
561	Bool_t IsTRG(){return TRG;};
562	Bool_t IsS4(){return S4;};
563	Bool_t IsND(){return ND;};
564	Bool_t IsAC(){return AC;};
565	Bool_t IsORB(){return ORB;};
566
567	Bool_t IsRUN(){return RUN;};
568
569	Bool_t IsGood(Bool_t strict=true);
570
571	Int_t IsSELLI(){return SELLI;}; //trees of selected events (there is a selection list)
572	void SetSELLI(Int_t selli){SELLI=selli;};
573
574	Bool_t IsGP(){return ISGP;};
575
576	void SetGP(Bool_t tt){ISGP=tt;};
577
578	void SetDebug(Bool_t set){ DBG=set; TrkParams::SetVerboseMode(); };
579
580
581	ClassDef(PamLevel2,2);
582	};
583
584	//===============================================================================================
585	//
586	//
587	//
588	//
589	//===============================================================================================
590
591
592	#endif
593