From 2ea77c86d67e79bec966c400530c7120962bc803 Mon Sep 17 00:00:00 2001
From: Mark Hodgkinson <m.hodgkinson@sheffield.ac.uk>
Date: Fri, 21 Jul 2017 15:29:07 +0100
Subject: [PATCH] Add cluster selector algorithm and relevane lines in
eflowRec_entries
Former-commit-id: 669e7918536dfc2890ce79abff2459fe4e3bdeeb
---
.../eflowRec/eflowRec/PFClusterSelector.h | 41 +++++++
.../eflowRec/src/PFClusterSelector.cxx | 109 ++++++++++++++++++
2 files changed, 150 insertions(+)
create mode 100644 Reconstruction/eflowRec/eflowRec/PFClusterSelector.h
create mode 100644 Reconstruction/eflowRec/src/PFClusterSelector.cxx
diff --git a/Reconstruction/eflowRec/eflowRec/PFClusterSelector.h b/Reconstruction/eflowRec/eflowRec/PFClusterSelector.h
new file mode 100644
index 00000000000..529e36637f4
--- /dev/null
+++ b/Reconstruction/eflowRec/eflowRec/PFClusterSelector.h
@@ -0,0 +1,41 @@
+#ifndef PFCLUSTERSELECTOR_H
+#define PFCLUSTERSELECTOR_H
+
+#include "AthenaBaseComps/AthAlgorithm.h"
+#include "GaudiKernel/ToolHandle.h"
+#include "xAODCaloEvent/CaloClusterContainer.h"
+
+#include <map>
+
+class eflowRecClusterContainer;
+
+class PFClusterSelector : public AthAlgorithm {
+
+public:
+ /** Default constructor */
+ PFClusterSelector(const std::string& name, ISvcLocator* pSvcLocator);
+ /** Default destructor */
+ ~PFClusterSelector() {};
+
+ /** Gaudi AthAlgorithm hooks */
+ StatusCode initialize();
+ StatusCode execute();
+ StatusCode finalize();
+
+private:
+ /** for EM mode, LC weight for cells are retrieved before doing any subtraction; they will be used after subtraction */
+ void retrieveLCCalCellWeight(const double& energy, const unsigned& index, std::map<IdentifierHash,double>& cellsWeight);
+
+ /** ReadHandle for the CaloClusterContainer to be used as input */
+ SG::ReadHandle<xAOD::CaloClusterContainer> m_caloClusterReadHandle;
+
+ /** ReadHandle for the CaloClusterContainer, at LC scale, to be used as input */
+ SG::ReadHandle<xAOD::CaloClusterContainer> m_caloCalClusterReadHandle;
+
+ /** WriteHandle for the eflowRecClusterContainer to write out: */
+ SG::WriteHandle<eflowRecClusterContainer> m_eflowRecClusterContainerWriteHandle;
+
+
+
+};
+#endif
diff --git a/Reconstruction/eflowRec/src/PFClusterSelector.cxx b/Reconstruction/eflowRec/src/PFClusterSelector.cxx
new file mode 100644
index 00000000000..8e203479f08
--- /dev/null
+++ b/Reconstruction/eflowRec/src/PFClusterSelector.cxx
@@ -0,0 +1,109 @@
+#include "CaloDetDescr/CaloDetDescrManager.h"
+#include "CaloIdentifier/CaloCell_ID.h"
+#include "eflowRec/eflowRecCluster.h"
+#include "eflowRec/PFClusterSelector.h"
+#include "xAODCaloEvent/CaloCluster.h"
+
+PFClusterSelector::PFClusterSelector(const std::string& name, ISvcLocator* pSvcLocator):
+ AthAlgorithm(name, pSvcLocator),
+ m_caloClusterReadHandle("CaloTopoCluster"),
+ m_caloCalClusterReadHandle("CaloCalTopoCluster"),
+ m_eflowRecClusterContainerWriteHandle("eflowRecClusters")
+{
+ declareProperty("clustersName",m_caloClusterReadHandle );
+ declareProperty("calClustersName",m_caloCalClusterReadHandle );
+ declareProperty("eflowRecClustersOutputName", m_eflowRecClusterContainerWriteHandle);
+}
+
+StatusCode PFClusterSelector::initialize(){
+
+ ATH_CHECK(m_caloClusterReadHandle.initialize());
+ ATH_CHECK(m_caloCalClusterReadHandle.initialize());
+
+ ATH_CHECK(m_eflowRecClusterContainerWriteHandle.initialize());
+
+ return StatusCode::SUCCESS;
+}
+
+StatusCode PFClusterSelector::execute(){
+
+ /* Verify the read handle has a valid pointer, and if not return */
+ if (!m_caloClusterReadHandle.isValid()){
+ msg(MSG::WARNING) << " Invalid ReadHandle for xAOD::CaloCluster with key: " << m_caloClusterReadHandle.key() << endmsg;
+ return StatusCode::SUCCESS;
+ }
+ /* Record the eflowRecCluster output container */
+ ATH_CHECK(m_eflowRecClusterContainerWriteHandle.record(std::make_unique<eflowRecClusterContainer>()));
+
+ /* Fill the vector of eflowRecClusters */
+ unsigned int nClusters = m_caloClusterReadHandle->size();
+ for (unsigned int iCluster = 0; iCluster < nClusters; ++iCluster) {
+ /* Create the eflowRecCluster and put it in the container */
+ std::unique_ptr<eflowRecCluster> thisEFRecCluster = std::make_unique<eflowRecCluster>(ElementLink<xAOD::CaloClusterContainer>(*m_caloClusterReadHandle, iCluster));
+
+ if (m_caloCalClusterReadHandle.isValid()){
+ std::map<IdentifierHash,double> cellsWeightMap;
+ retrieveLCCalCellWeight(m_caloClusterReadHandle->at(iCluster)->e(), iCluster, cellsWeightMap);
+
+ if (msgLvl(MSG::DEBUG)) {
+ //zhangr
+ std::map<IdentifierHash, double>::iterator it = cellsWeightMap.begin();
+ for (; it != cellsWeightMap.end(); ++it) {
+ msg(MSG::DEBUG) << "zhangrui eflowPreparation " << iCluster << "/" << nClusters << ": e="
+ << m_caloClusterReadHandle->at(iCluster)->e() << " (" << it->first << " "
+ << it->second << ")" << endmsg;
+ }
+ }
+
+ thisEFRecCluster->setCellsWeight(cellsWeightMap);
+ }
+
+ thisEFRecCluster->setClusterId(iCluster);
+ m_eflowRecClusterContainerWriteHandle->push_back(std::move(thisEFRecCluster));
+
+ if (msgLvl(MSG::DEBUG)) {
+ const xAOD::CaloCluster* thisCluster = m_caloClusterReadHandle->at(iCluster);
+ msg(MSG::DEBUG) << "eflowPreparation clus = " << thisCluster->eta() << " "
+ << thisCluster->phi() << " " << thisCluster->e()/cosh(thisCluster->eta()) << " " << endmsg;
+ }
+ }
+
+ return StatusCode::SUCCESS;
+}
+
+StatusCode PFClusterSelector::finalize(){
+ return StatusCode::SUCCESS;
+}
+
+void PFClusterSelector::retrieveLCCalCellWeight(const double& energy, const unsigned& index, std::map<IdentifierHash,double>& cellsWeight) {
+
+ /* match CaloCluster with CaloCalCluster to obtain cell weight */
+ /* first try the position at 'index'. If we are lucky, the loop can be avoided. */
+ /* Note the read handle has been tested to be valid prior to the call of this function */
+ const xAOD::CaloCluster* matchedCalCluster = m_caloCalClusterReadHandle->at(index);
+ if (!(fabs(energy - matchedCalCluster->rawE()) < 0.001)) {
+ matchedCalCluster = nullptr;
+ for (unsigned iCalCalCluster = 0; iCalCalCluster < m_caloCalClusterReadHandle->size();
+ ++iCalCalCluster) {
+ matchedCalCluster = m_caloCalClusterReadHandle->at(iCalCalCluster);
+ if (fabs(energy - matchedCalCluster->rawE()) < 0.001) {
+ break;
+ }
+ }
+ }
+ assert (matchedCalCluster);
+
+ /* obtain cell index and cell weight */
+ const CaloDetDescrManager* calo_dd_man = CaloDetDescrManager::instance();
+ const CaloCell_ID* calo_id = calo_dd_man->getCaloCell_ID();
+ xAOD::CaloCluster::const_cell_iterator itCell = matchedCalCluster->cell_begin();
+ xAOD::CaloCluster::const_cell_iterator endCell = matchedCalCluster->cell_end();
+ for (; itCell != endCell; ++itCell) {
+ const CaloCell* pCell = *itCell;
+ Identifier myId = pCell->ID();
+ IdentifierHash myHashId = calo_id->calo_cell_hash(myId);
+ cellsWeight[myHashId] = itCell.weight();
+ }
+
+ return ;
+}
--
GitLab