CaloLCDeadMaterialTool.h

Go to the documentation of this file.

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

 
 
#include "CaloInterface/IClusterCellWeightTool.h"
#include "AthenaBaseComps/AthAlgTool.h"
#include "CaloConditions/CaloLocalHadCoeff.h"
#include "GaudiKernel/ToolHandle.h" 
#include "CaloGeoHelpers/CaloSampling.h"
#include "StoreGate/ReadCondHandleKey.h"
#include "GaudiKernel/EventContext.h"
#include "CaloEvent/CaloRecoStatus.h"
#include "CLHEP/Units/SystemOfUnits.h"
 
#include <map>
#include <vector>
#include <string>
 
 
class CaloCell_ID;
 
 
//class CaloLCDeadMaterialTool : public AthAlgTool, virtual public IClusterCellWeightTool
class CaloLCDeadMaterialTool : public extends<AthAlgTool, IClusterCellWeightTool>
{
  public:
    using base_class::base_class;
  
    enum dm_area_keys
    {
      sDM_EMB0_EMB1,   // 0 before PreSamplerB, between PreSamplerB and EMB1
      sDM_EMB3_TILE0,  // 1 between barrel and tile
      sDM_SCN,         // 2 before TileGap3 (scintillator)
      sDM_EME0_EME12,  // 3 before PreSamplerE, between PreSamplerE and EME1
      sDM_EME3_HEC0,   // 4 between EME3 and HEC0
      sDM_FCAL,        // 5 between HEC and FCAL, before FCAL
      sDM_LEAKAGE,     // 6 leakage behind calorimeter
      sDM_UNCLASS,     // 7 unclassified DM enegry
      sDM
    };
 
    virtual ~CaloLCDeadMaterialTool() override;
 
    virtual StatusCode weight(xAOD::CaloCluster* theCluster, const EventContext& ctx) const override;
    virtual StatusCode initialize() override;
 
  private:
    // Hold information about one cell.
    struct Cell
    {
      int dm;
      float energy;
      float weight;
    };
 
    // Hold information about one dead material area.
    struct Area
    {
      Area() : eprep(0), erec(0), sub_ener_old(0), corr(0), edm_wrong(0) {}
      float eprep;
      float erec;
      float sub_ener_old;
      float corr;
      float edm_wrong;
    };
 
    StatusCode prepare_for_cluster(const xAOD::CaloCluster* theCluster,
                                   std::vector<Area>& areas,
                                   std::vector<Cell>& cells,
                                   float* smp_energy,
                                   float& cls_unweighted_energy, const CaloLocalHadCoeff* data) const;

    SG::ReadCondHandleKey<CaloLocalHadCoeff>  m_key{this, "HadDMCoeffKey", "HadDMCoeff2"};

    Gaudi::Property<int> m_recoStatus{this, "ClusterRecoStatus", CaloRecoStatus::UNKNOWNSTATUS};

    Gaudi::Property<int> m_weightModeDM{this, "WeightModeDM", 1};

    Gaudi::Property<float> m_MinClusterEnergyToDeal{this, "MinClusterEnergyToDeal", 200.0*CLHEP::MeV};

    Gaudi::Property<int> m_MinLookupBinNentry{this, "MinLookupBinNentry", 40};

    Gaudi::Property<float> m_MinCellEnergyToDeal{this, "MinCellEnergyToDeal", 0.0};

    Gaudi::Property<float> m_MaxChangeInCellWeight{this, "MaxChangeInCellWeight", 30.0};

    Gaudi::Property<bool> m_useHadProbability{this, "UseHadProbability", false};

    Gaudi::Property<bool> m_interpolate{this, "Interpolate", false};

    Gaudi::Property<bool> m_updateSamplingVars{this, "UpdateSamplingVars", false};

    Gaudi::Property<std::map<std::string, std::vector<std::string>>>  m_interpolateDimensionNames {this
     , "InterpolateDimensionNames"
     , {
        {"AREA_DMFIT" , {"DIMD_ETA", "DIMD_ENER"}}
    ,  {"AREA_DMLOOKUP" , {"DIMD_ETA", "DIMD_ENER", "DIMD_LAMBDA"}}
    , {"AREA_DMSMPW" , {"DIMD_ETA", "DIMD_LAMBDA"}}
      }
    };
  
   std::vector<int > m_interpolateDimensionsFit;
   std::vector<int > m_interpolateDimensionsLookup;
   std::vector<int > m_interpolateDimensionsSampling;
   
   Gaudi::Property<bool> m_absOpt{this, "WeightingOfNegClusters", false};
 
};
 
 
#endif