ParticleSortingTool.h

Go to the documentation of this file.

 
/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
// ParticleSortingTool.h
// Header file for class ParticleSortingTool
// Author: Karsten Koeneke <karsten.koeneke@cern.ch>
#ifndef EVENTUTILS_PARTICLESORTINGTOOL_H
#define EVENTUTILS_PARTICLESORTINGTOOL_H 1
 
// FrameWork includes
#include "AthenaBaseComps/AthAlgTool.h"
#include "DerivationFrameworkInterfaces/IAugmentationTool.h"
#include "CxxUtils/fpcompare.h"
 
// EDM inlcudes
#include "xAODBase/IParticle.h"
#include "xAODBase/IParticleContainer.h"
#include "AthContainers/ConstDataVector.h"
 
// STL includes
#include <vector>
#include <string>
#include <cmath>
 
class ParticleSortingTool
  : virtual public ::DerivationFramework::IAugmentationTool,
            public ::AthAlgTool
{
 
  // Public methods:
public:
 
  // Copy constructor:
 
  ParticleSortingTool( const std::string& type,
                       const std::string& name,
                       const IInterface* parent );
 
  virtual ~ParticleSortingTool();
 
  virtual StatusCode  initialize() override;
 
  virtual StatusCode  finalize() override;
 
 
  virtual StatusCode addBranches() const final override;
 
 
 
// Private methods
private:
 
  StatusCode doSort( xAOD::IParticleContainer* cont ) const;
 
  template<class CONTAINERTYPE>
  StatusCode doSortConst( ConstDataVector<CONTAINERTYPE>* cont ) const;
 
  bool comparePt( const xAOD::IParticle* partA, const xAOD::IParticle* partB ) const;
 
  bool compareEta( const xAOD::IParticle* partA, const xAOD::IParticle* partB ) const;
 
  bool comparePhi( const xAOD::IParticle* partA, const xAOD::IParticle* partB ) const;
 
  bool compareMass( const xAOD::IParticle* partA, const xAOD::IParticle* partB ) const;
 
  bool compareEnergy( const xAOD::IParticle* partA, const xAOD::IParticle* partB ) const;
 
  bool compareRapidity( const xAOD::IParticle* partA, const xAOD::IParticle* partB ) const;
 
  bool compareAuxData( const xAOD::IParticle* partA, const xAOD::IParticle* partB ) const;
 
  inline bool compareDouble( double a, double b ) const;
 
 
  // Private data:
private:
 
  StringProperty m_inCollKey;
 
  StringProperty m_outCollKey;
 
  StringProperty m_sortVar;
 
  BooleanProperty m_sortDescending;
 
 
  int m_sortID;
 
  mutable std::atomic<unsigned long> m_nEventsProcessed;
 
};
 
 
inline bool ParticleSortingTool::compareDouble( double a, double b ) const
{
  if ( m_sortID < 0 ) { return CxxUtils::fpcompare::greater(a,b); }
  else { return CxxUtils::fpcompare::less(a,b); }
}
 
 
template<class CONTAINERTYPE>
StatusCode ParticleSortingTool::doSortConst( ConstDataVector<CONTAINERTYPE>* cont ) const
{
  if ( !cont ) {
    ATH_MSG_ERROR("No ConstDataVector to be sorted");
    return StatusCode::FAILURE;
  }
  // Actually do the sorting, using a C++11 lambda function construct
  // to be able to use the member function here
  if ( std::abs(m_sortID) == 1 ) {
    cont->sort( [this](const xAOD::IParticle* a, const xAOD::IParticle* b) {
                  return this->comparePt(a,b);
                } );
  }
  else if ( std::abs(m_sortID) == 2 ) {
    cont->sort( [this](const xAOD::IParticle* a, const xAOD::IParticle* b) {
                  return this->compareEta(a,b);
                } );
  }
  else if ( std::abs(m_sortID) == 3 ) {
    cont->sort( [this](const xAOD::IParticle* a, const xAOD::IParticle* b) {
                  return this->comparePhi(a,b);
                } );
  }
  else if ( std::abs(m_sortID) == 4 ) {
    cont->sort( [this](const xAOD::IParticle* a, const xAOD::IParticle* b) {
                  return this->compareMass(a,b);
                } );
  }
  else if ( std::abs(m_sortID) == 5 ) {
    cont->sort( [this](const xAOD::IParticle* a, const xAOD::IParticle* b) {
                  return this->compareEnergy(a,b);
                } );
  }
  else if ( std::abs(m_sortID) == 6 ) {
    cont->sort( [this](const xAOD::IParticle* a, const xAOD::IParticle* b) {
                  return this->compareRapidity(a,b);
                } );
  }
  else if ( std::abs(m_sortID) == 7 ) {
    cont->sort( [this](const xAOD::IParticle* a, const xAOD::IParticle* b) {
                  return this->compareAuxData(a,b);
                } );
  }
 
  return StatusCode::SUCCESS;
}
 
 
 
#endif //> !EVENTUTILS_PARTICLESORTINGTOOL_H