AFPSiClusterTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef AFP_SICLUSTERTOOLS_AFPSICLUSTERTOOL_H
#define AFP_SICLUSTERTOOLS_AFPSICLUSTERTOOL_H 1

 
 
// FrameWork includes
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ServiceHandle.h"
#include "GaudiKernel/ToolHandle.h" 
 
// Monitoring includes
#include "AthenaMonitoringKernel/Monitored.h"
#include "AthenaMonitoringKernel/GenericMonitoringTool.h"
 
#include "AFP_SiClusterTools/IAFPSiClusterTool.h"
#include "AFP_SiClusterTools/IAFPSiClusterAlgTool.h"
#include "AFP_SiClusterTools/AFPSiClusterLayerBasicObj.h"
#include "AFP_SiClusterTools/IAFPSiRowColToLocalCSTool.h"
#include "AFP_DBTools/ISiLocAlignDBTool.h"
#include "AFP_DBTools/ISiGlobAlignDBTool.h"
#include "AFP_DigiEv/AFP_SiDigiCollection.h"
 
#include "AFP_DBTools/SiLocAlignData.h"
#include "AFP_DBTools/SiGlobAlignData.h"
#include "xAODForward/AFPSiHit.h"
#include "xAODForward/AFPSiHitContainer.h"
#include "xAODForward/AFPSiHitsCluster.h"
#include "xAODForward/AFPSiHitsClusterContainer.h"
#include "xAODForward/AFPSiHitsClusterAuxContainer.h"
 
// STL includes
#include "nlohmann/json.hpp"
#include <string>
#include <vector>

class AFPSiClusterTool
  : public extends<AthAlgTool, IAFPSiClusterTool>
{
public:
  AFPSiClusterTool(const std::string& type, const std::string& name, const IInterface* parent);
 
  virtual ~AFPSiClusterTool() override {}
 
  // Athena algtool's Hooks
  virtual StatusCode initialize() override;

  virtual StatusCode finalize() override;

  StatusCode clusterHits(std::unique_ptr<xAOD::AFPSiHitsClusterContainer>& clustr, const EventContext &ctx) const override;
 
private:
  SG::ReadHandleKey<xAOD::AFPSiHitContainer> m_AFPSiHitContainerKey { this, "AFPSiHitsContainerName", "AFPSiHitContainer", "name of AFPSiHit container" };

  ToolHandle<IAFPSiClusterAlgTool> m_clusterAlgToolHandle{this,"clusterAlgTool", "AFPSiClusterAlgTool","Tool with algorithm to perform actual clustering"};
  
  ToolHandle<IAFPSiRowColToLocalCSTool> m_rowColToLocalCSToolHandle{this,"rowColToLocalCSTool", "AFPSiRowColToLocalCSTool","Tool to rewrite clusters to xAOD and transform rows and columns to station local coordinates"}; 
  
  ToolHandle<GenericMonitoringTool> m_monTool {this, "MonTool", "", "Monitoring tool"};
  
  
  ToolHandle<AFP::ISiLocAlignDBTool> m_siLocAlignDBTool {this, "siLocAlignDBTool", "AFP__SiLocAlignDBTool", "Tool to access DB to get the local alignment constants"};
  
  ToolHandle<AFP::ISiGlobAlignDBTool> m_siGlobAlignDBTool {this, "siGlobAlignDBTool", "AFP__SiGlobAlignDBTool", "Tool to access DB to get the global alignment constants"};
  
  Gaudi::Property<std::vector<int>> m_numberOfLayersInStations {this, "numberOfLayersInStations",{4,4,4,4}, "The size of the array in argument defines number of stations, the value in each entry defines number of layers in the station."};

  StatusCode clearAllLayers(std::vector< std::vector<AFPSiClusterLayerBasicObj> > &my_layers) const;

  StatusCode fillLayersWithHits(std::vector< std::vector<AFPSiClusterLayerBasicObj> > &my_layers, const EventContext &ctx) const;

  StatusCode clusterEachLayer(std::vector< std::vector<AFPSiClusterLayerBasicObj> > &my_layers) const;

  StatusCode saveToXAOD(std::unique_ptr<xAOD::AFPSiHitsClusterContainer>& clustr, std::vector< std::vector<AFPSiClusterLayerBasicObj> > &my_layers, const EventContext &ctx) const;

  std::vector< std::vector<AFPSiClusterLayerBasicObj> > initLayers (const std::vector<int>& layersInStations) const;
};
 
 
#endif // >  !AFP_SICLUSTERTOOLS_AFPSICLUSTERTOOL_H