eflowCellSubtractionFacilitator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef EFLOWCELLSUBTRACTIONFACILITATOR_H
#define EFLOWCELLSUBTRACTIONFACILITATOR_H
 
/********************************************************************
 
NAME:     eflowCellSubtractionFacilitator.h
PACKAGE:  offline/Reconstruction/eflowRec
 
AUTHORS:  M.Hodgkinson
CREATED:  3rd February 2009
 
********************************************************************/
 
#include "AsgMessaging/AsgMessaging.h"
#include "eflowRec/eflowCellList.h"
#include "eflowRec/eflowRecTrack.h"
#include "eflowRec/eflowRingSubtractionManager.h"
#include "xAODCaloEvent/CaloCluster.h"
 
#include <vector>

class eflowCellSubtractionFacilitator : public asg::AsgMessaging
{
 
public:
  eflowCellSubtractionFacilitator();
 
  double subtractCells(eflowRingSubtractionManager& ringSubtractionManager,
                       eflowRecTrack& theTrack,
                       xAOD::CaloCluster* tracksClus,
                       eflowCellList& orderedCells,
                       bool& annFlag, const bool& addCPData) const;
  double subtractCells(
    eflowRingSubtractionManager& ringSubtractionManager,
    eflowRecTrack& theTrack,
    std::vector<std::pair<xAOD::CaloCluster*, bool>>& tracksClus,
    eflowCellList& orderedCells,
    bool& annFlag, const bool& addCPData) const;
 
private:
  static CaloClusterCellLink::iterator getCellIterator(
    xAOD::CaloCluster* thisCluster,
    const CaloCell* thisCell);
 
  static void updateClusterKinematics(
    std::vector<std::pair<xAOD::CaloCluster*, bool>>& tracksClusters);
  static void updateClusterKinematics(xAOD::CaloCluster*);
 
  static double getTotalEnergy(
    const std::vector<std::pair<xAOD::CaloCluster*, bool>>& tracksClusters);
 
  static double getRingsEnergy(
    std::vector<std::pair<xAOD::CaloCluster*, bool>>& tracksClusters,
    CellIt beginRing,
    CellIt endRing);
 
  void subtractPartialRings(
    std::vector<std::pair<xAOD::CaloCluster*, bool>>& tracksClusters,
    CellIt beginRing,
    CellIt endRing,
    double targetRingEnergy,
    double eRing, eflowRecTrack& theTrack,
    const bool& addCPData) const;
 
  void subtractFullRings(
    std::vector<std::pair<xAOD::CaloCluster*, bool>>& tracksClusters,
    CellIt beginRing,
    CellIt endRing, eflowRecTrack& theTrack,
    const bool& addCPData) const;
 
  bool subtractRings(
    eflowRingSubtractionManager& ringSubtractionManager,
    const std::pair<eflowCaloENUM, short>& ring,
    double& eSubtracted,
    const double eExpect,
    eflowCellList& orderedCells,
    std::vector<std::pair<xAOD::CaloCluster*, bool>>& tracksClusters,
    bool& annFlag, eflowRecTrack& theTrack,
    const bool& addCPData) const;
 
  static bool subtractCaloCell(double& eSubtracted,
                               const double eExpect,
                               xAOD::CaloCluster* cluster,
                               const CaloCell* cell, 
                               eflowRecTrack& theTrack,
                               const bool& addCPData);
 
  static void annihilateClusters(
    std::vector<std::pair<xAOD::CaloCluster*, bool>>& tracksClusters, bool& annFlag,
    eflowRecTrack& theTrack, const bool& addCPData) ;
 
  static bool subtractReorderedCells(
    std::vector<std::pair<xAOD::CaloCluster*, bool>>& tracksClusters,
    double eSubtracted,
    const double eExpect,
    eflowCellList& orderedCells,
    eflowRecTrack& theTrack,
    const bool& addCPData);
};
#endif