/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/JetRec/JetRec/PseudoJetGetter.h

// this file is -*- C++ -*-
// PseudoJetGetter.h

#ifndef PseudoJetGetter_H
#define PseudoJetGetter_H

// David Adams
// January 2014


#include "AsgTools/AsgTool.h"
#include "JetInterface/IPseudoJetGetter.h"
#include "fastjet/PseudoJet.hh"
#include "xAODCaloEvent/CaloClusterContainer.h"
#include "xAODJet/JetContainer.h"
#include "JetEDM/PseudoJetVector.h"
#include "JetEDM/IndexedConstituentUserInfo.h"
#include "JetEDM/LabelIndex.h"

class PseudoJetGetter
:  public asg::AsgTool,
  virtual public IPseudoJetGetter {
  ASG_TOOL_CLASS(PseudoJetGetter, IPseudoJetGetter)

public:

  typedef jet::LabelIndex LabelIndex;
  typedef jet::PseudoJetVector PseudoJetVector;

  PseudoJetGetter(const std::string& myname);

  virtual StatusCode initialize();

  const PseudoJetVector* get() const;

  virtual int appendTo(PseudoJetVector& psjs, const LabelIndex* pli) const;

  template<typename TList>
  int append(const TList& inputs, PseudoJetVector& psjs, const LabelIndex* pli) const;

  std::string label() const;

  void print() const;

  int inputContainerNames(std::vector<std::string>& connames);

  int outputContainerNames(std::vector<std::string>& connames);

protected:  // member functions

  virtual jet::IConstituentUserInfo*
  buildCUI(const xAOD::IParticle* ppar, jet::IConstituentUserInfo::Index idx,
           const LabelIndex* pli) const;

protected:  // data

  // Job options.
  std::string m_incoll;             
  std::string m_outcoll;            
  std::string m_label;              
  bool m_skipNegativeEnergy;        
  double m_ghostscale;              
  bool m_negEnergyAsGhosts;         


  // Cached flags.
  bool m_emtopo;                    

};


#ifdef USE_BOOST_AUTO
#include <boost/typeof/typeof.hpp>
#endif

template<typename TList>
int PseudoJetGetter::
append(const TList& inputs, PseudoJetVector& psjs, const LabelIndex* pli) const {
  int nskip = 0;
  jet::IConstituentUserInfo::Index labidx = 0;
  if ( pli != 0 ) labidx = pli->index(m_label);
  else ATH_MSG_WARNING("Index-to-label map is not supplied.");
  //  if ( m_ghostscale ) labidx = -labidx;
  if(!m_negEnergyAsGhosts && m_ghostscale) labidx = -labidx;
  ATH_MSG_DEBUG( "Ghost scale =  " << m_ghostscale << "; idx = " << labidx );
  
#ifdef USE_BOOST_AUTO
  for ( BOOST_AUTO(iinp, inputs.begin()); iinp!=inputs.end(); ++iinp ) {
    BOOST_AUTO(ppar, *iinp); // IParticle pointer
#else
  for ( auto iinp=inputs.begin(); iinp!=inputs.end(); ++iinp ) {
    auto ppar = *iinp; // IParticle pointer
#endif
    if ( ppar == 0 || (m_skipNegativeEnergy && ppar->e() <= 0.0) ) {
      if ( ppar == 0 ) ATH_MSG_DEBUG("NUll object!");
      else ATH_MSG_VERBOSE("Skipping cluster with E = " << ppar->e());
      ++nskip;
      continue;
    }
    // Take momentum from IParticle.
    fastjet::PseudoJet psj(ppar->p4());
    // Form EM scale, use uncalibrated four-vector.1
    if ( m_emtopo ) {
      const xAOD::CaloCluster* pclu = dynamic_cast<const xAOD::CaloCluster*>(ppar);
      if ( pclu == 0 ) {
        ATH_MSG_WARNING("EM particle is not type CaloCluster");
        continue;
      }
      psj.reset(pclu->p4(xAOD::CaloCluster::UNCALIBRATED));
    }

    if( m_negEnergyAsGhosts) {
      if (ppar->e() <= 0.0)  { // Important use the IParticle::e(), because IParticle::p4().e() might always be >0.
        psj.reset_momentum(psj.px(), psj.py(), psj.pz(), std::abs( psj.e() )); // make sure e>0 in fastjet
        psj *= m_ghostscale;
      }
    } else if ( m_ghostscale ) psj *= m_ghostscale;

    bool show = psjs.size() < 20;
    if ( show ) ATH_MSG_VERBOSE("index/label: " << labidx << "/" << m_label);
    if ( show ) ATH_MSG_VERBOSE("old/p4/new"
                  << " pt: "   << ppar->pt()  << "/" << ppar->p4().Pt()  << "/" << psj.pt()
                  << ", pz: /"                       << ppar->p4().Pz()  << "/" << psj.pz()
                  << ", p: /"                        << ppar->p4().P()   << "/"
                  << ", E: "   << ppar->e()   << "/" << ppar->p4().E()   << "/" << psj.e()
                  << ", m: "   << ppar->m()   << "/" << ppar->p4().M()   << "/" << psj.m()
                  << ", eta: " << ppar->eta() << "/" << ppar->p4().Eta() << "/" << psj.eta()
                );
    if ( pli != 0 ) {
      jet::IConstituentUserInfo* pcui = this->buildCUI(ppar, labidx, pli);
      psj.set_user_info(pcui);
      if ( show ) ATH_MSG_VERBOSE("User info particle: " << pcui->particle());
    }
    psjs.push_back(psj);
  } // end loop over input 

  if ( nskip ) ATH_MSG_DEBUG("Skipped constituent count: " << nskip);
  ATH_MSG_DEBUG("After append, PseudoJet count is " << psjs.size());
  return 0;
}

#endif

---