SCT_ClusterValidationNtupleWriter.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2018 CERN for the benefit of the ATLAS collaboration
*/

// SCT_ClusterValidationNtupleWriter.h
 
#ifndef INDET_SCT_CLUSTERVALIDATIONNTUPLEWRITER_H
#define INDET_SCT_CLUSTERVALIDATIONNTUPLEWRITER_H
 
// Gaudi includes
#include "AthenaBaseComps/AthAlgorithm.h"
 
#include "InDetPrepRawData/SCT_ClusterContainer.h"  // typedef
#include "InDetRawData/SCT_RDO_Container.h"
#include "InDetReadoutGeometry/SiDetectorElementCollection.h"
#include "SCT_Cabling/ISCT_CablingTool.h"
#include "SCT_ConditionsTools/ISCT_ByteStreamErrorsTool.h"
#include "StoreGate/ReadCondHandleKey.h"
#include "StoreGate/ReadHandleKey.h"
#include "TrkSpacePoint/SpacePointContainer.h"
#include "TrkTrack/TrackCollection.h"
#include "xAODEventInfo/EventInfo.h"
 
#include <set>
#include <string>
#include <vector>
 
class SCT_ID;
class TTree;
 
namespace InDet {

 
  class SCT_ClusterValidationNtupleWriter : public AthAlgorithm {
  public:
    SCT_ClusterValidationNtupleWriter(const std::string& name, ISvcLocator* pSvcLocator);
    ~SCT_ClusterValidationNtupleWriter() = default;

    virtual StatusCode initialize() override;
    virtual StatusCode execute() override;
    virtual StatusCode finalize() override;
 
  private:
 
    const SCT_ID* m_sctid; 
    const InDet::SCT_ClusterContainer* m_riocontainer; 
    SG::ReadHandleKey<xAOD::EventInfo> m_eventInfoKey{this, "EventInfoKey", "EventInfo", "key for xAOD::EventInfo"};
    SG::ReadHandleKey<SCT_ClusterContainer> m_jo_riocontainername{this, "SCT_ClusterContainer", "SCT_Clusters", "jobOption: name of container with RIOs"};
    SG::ReadHandleKey<SCT_RDO_Container> m_dataObjectName{this, "SCT_RDOContainer", "SCT_RDOs", "Data object name: for the SCT this is SCT_RDOs"};
    SG::ReadHandleKey<SpacePointContainer> m_spacePointContainerName{this, "SCT_SpacePointContainer", "SCT_SpacePoints", "SpacePoint container name: for the SCT this is SCT_SpacePoints"};
    SG::ReadHandleKey<TrackCollection> m_inputTrackCollection{this, "SCT_InputTrackCollection", "CombinedInDetTracks", "TrackCollection name, needed for hits-on-tracks, default is CombinedInDetTracks"};
 
    // For P->T converter of SCT_Clusters
    SG::ReadCondHandleKey<InDetDD::SiDetectorElementCollection> m_SCTDetEleCollKey{this, "SCTDetEleCollKey", "SCT_DetectorElementCollection", "Key of SiDetectorElementCollection for SCT"};
 
    ToolHandle<ISCT_ByteStreamErrorsTool> m_byteStreamErrTool{this, "ByteStreamErrTool", "SCT_ByteStreamErrorsTool", "Tool to retrieve SCT ByteStream Errors"};
    ToolHandle<ISCT_CablingTool> m_cabling{this, "SCT_CablingTool", "SCT_CablingTool", "Tool to retrieve SCT Cabling"};
    StringProperty m_ntupleFileName{this, "NtupleFileName", "/NTUPLES/FILE1"};     
    StringProperty m_ntupleDirName{this, "NtupleDirectoryName", "FitterValidation"};      
    StringProperty m_ntupleTreeName{this, "NtupleTreeName", "RIOs"};     
    BooleanProperty m_fillCluster{this, "FillCluster", false};        
    BooleanProperty m_fillRDO{this, "FillRDO", true};            
    BooleanProperty m_fillSpacePoint{this, "FillSpacePoint", false};     
    BooleanProperty m_fillBSErrs{this, "FillBSErrs", true};         
    BooleanProperty m_doHitsOnTracks{this, "DoHitsOnTracks", false};     

    TTree* m_nt;
 
    // --- ntuple items. Pls describe here to allow copying to wiki+doxygen ---
    int         m_runNumber;       
    int         m_eventNumber;     
    int         m_lumiBlock;       
    int         m_timeStamp;       
    int         m_bunchCrossing;   
 
    // ntuple items for SCT clusters
    int         m_nRIOs;              
    std::vector<float>* m_rioLoc1;    
    std::vector<float>* m_rioSurfaceX; 
    std::vector<float>* m_rioSurfaceY; 
    std::vector<float>* m_rioSurfaceZ; 
    std::vector<int>*   m_SctBarrelEndcap; 
    std::vector<int>*   m_SctLayerDisk; 
    std::vector<int>*   m_SctEtaModule; 
    std::vector<int>*   m_SctPhiModule; 
    std::vector<int>*   m_SctSide;      
    std::vector<float>* m_SctDeltaPhi;  
    std::vector<float>* m_SctHitErr;    
 
    //SpacePoints
    int m_nSP;
    std::vector<int>* m_sp_bec;          
    std::vector<int>* m_sp_layer;        
    std::vector<int>* m_sp_eta;          
    std::vector<int>* m_sp_phi;          
    std::vector<float>* m_sp_x;          
    std::vector<float>* m_sp_y;          
    std::vector<float>* m_sp_z;          
    
    //  RDO
    int  m_nRDOs;                         
    std::vector<int>* m_sct_rdoGroupSize; 
    std::vector<int>* m_sct_rdoIsOnTrack; 
    std::vector<int>* m_sct_layer;        
    std::vector<int>* m_sct_eta;          
    std::vector<int>* m_sct_phi;          
    std::vector<int>* m_sct_side;         
    std::vector<int>* m_sct_barrelec;     
    std::vector<int>* m_sct_firstStrip;   
    std::vector<int>* m_sct_numHitsInWafer; 
    std::vector<int>* m_sct_waferHash;    
    std::vector<int>* m_sct_tbin;         
    std::vector<int>* m_sct_rodid;        
    std::vector<int>* m_sct_channel;      
 
    /* ByteStream errors: these variables represent a bit set in the header or trailer for a
       link (i.e. SCT module side) in the Bytestream, indicating that the ROD found
       something wrong for that event for that link. In general, the ROD expects to
       get from each link for each event: [link header] [zero or more hits] [link trailer].
       These errors can either be a bit in the header or the trailer, or they can take the place
       of a hit (for the ABCD or Raw) errors. */
    int               m_totalNumErrors;   
    std::vector<int>* m_scterr_bec;       
    std::vector<int>* m_scterr_layer;     
    std::vector<int>* m_scterr_eta;       
    std::vector<int>* m_scterr_phi;       
    std::vector<int>* m_scterr_side;      
    std::vector<int>* m_scterr_rodid;     
    std::vector<int>* m_scterr_channel;   
    std::vector<int>* m_scterr_type;      
 
  };
 
} // close of namespace
 
#endif // INDET_SCT_CLUSTERVALIDATIONNTUPLEWRITER_H