ClusterSortingAlg.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
//  ClusterSortingAlg.cxx
//  TopoCore
//  Created by V Sorin, Joerg Stelzer .
//  algorithm to make list of cluster, pt order applied
//
#include "L1TopoAlgorithms/ClusterSortingAlg.h"
#include "L1TopoEvent/TOBArray.h"
#include "L1TopoEvent/ClusterTOBArray.h"
#include "L1TopoEvent/GenericTOB.h"
#include <algorithm>
 
REGISTER_ALG_TCS(ClusterSortingAlg)
 
bool SortByEtLargestFirst(TCS::GenericTOB* tob1, TCS::GenericTOB* tob2)
{
   return tob1->Et() > tob2->Et();
}
 
bool SortByEtSmallestFirst(TCS::GenericTOB* tob1, TCS::GenericTOB* tob2)
{
   return tob1->Et() < tob2->Et();
}
 
// constructor
TCS::ClusterSortingAlg::ClusterSortingAlg(const std::string & name) : SortingAlg(name) {
   defineParameter( "NumberOfClusters", 0 );
   defineParameter( "ReverseOrder", 0 );
   defineParameter( "MinIsolation", 0);
   defineParameter( "MinEta", 0 );
   defineParameter( "MaxEta", 49);
}
 
 
// destructor
TCS::ClusterSortingAlg::~ClusterSortingAlg() {}
 
TCS::StatusCode
TCS::ClusterSortingAlg::sort(const InputTOBArray & input, TOBArray & output) {
 
   const ClusterTOBArray & clusters = dynamic_cast<const ClusterTOBArray&>(input);
 
   // fill output array with GenericTOB buildt from clusters
   for(ClusterTOBArray::const_iterator cl = clusters.begin(); cl!= clusters.end(); ++cl ) {
      const GenericTOB gtob(**cl);
 
      if (parType_t(std::abs((*cl)-> eta())) < parameter("MinEta").value()) continue; 
      if (parType_t(std::abs((*cl)-> eta())) > parameter("MaxEta").value()) continue;
 
      // isolation cut
      if(parType_t((*cl)->isolation()) < parameter("MinIsolation").value()) continue;
      
      output.push_back( gtob );
   }
 
   // sort, possibly in reversed order
   int reverseOrder = parameter("ReverseOrder").value();
   if(reverseOrder==1) {
      output.sort(SortByEtSmallestFirst);
   } else {
      output.sort(SortByEtLargestFirst);
   }
 
   // keep only max number of clusters
   int par = parameter("NumberOfClusters").value();
   unsigned int maxNumberOfClusters = std::clamp(par, 0, std::abs(par));
   if(maxNumberOfClusters>0) {
      while( output.size()> maxNumberOfClusters ) {
         output.pop_back();
      }
   }
   return TCS::StatusCode::SUCCESS;
}