MomentumSumVisitor.h

Go to the documentation of this file.

#ifndef ATLFAST_MOMENTUMSUMVISITOR_H
#define ATLFAST_MOMENTUMSUMVISITOR_H


#ifndef ATLFAST_IAOO_H
#include "AtlfastEvent/IAOO.h"
#endif

#ifndef ATLFAST_IAOOVISITOR_H
#include "AtlfastEvent/IAOOvisitor.h"
#endif

#ifndef STD_VECTOR_H
#include <vector>
#define STD_VECTOR_H
#endif


#ifndef CLHEP_LORENTZVECTOR_H
#include "CLHEP/Vector/LorentzVector.h"
#define CLHEP_LORENTZVECTOR_H
#endif

namespace Atlfast{
  class ReconstructedParticle;
  class Cell;
  class IKinematic;
  class TwoCptCell;
  class KtCluster;
  class Cluster;
  class Jet;
  class SimpleKinematic;
  class Track;

  class IAOO;

  using ::HepLorentzVector;
 
  class MomentumSumVisitor: public IAOOvisitor{
  public:
    ~MomentumSumVisitor(){}
    void reset();
    void operator()(const IAOO* ia);

    virtual void process(const Cell*);
    virtual void process(const Cluster*);
    virtual void process(const KtCluster*);
    virtual void process(const Jet*);
    virtual void process(const ReconstructedParticle*);
    virtual void process(const SimpleKinematic*);
    virtual void process(const Track*);
    virtual void process(const TwoCptCell*);

    //
    
    HepLorentzVector   momSum(const Cell&)                   const;
    HepLorentzVector   momSum(const TwoCptCell&)             const;
    HepLorentzVector   momSum(const Cluster&)                const;
    HepLorentzVector   momSum(const KtCluster&)                const;
    HepLorentzVector   momSum(const Jet&)                    const;
    HepLorentzVector   momSum(const SimpleKinematic&)        const;
    HepLorentzVector   momSum(const Track&)                  const;
    HepLorentzVector   momSum(const ReconstructedParticle&)  const;

  private:

    HepLorentzVector m_cellMom;
    HepLorentzVector m_twoCptCellMom;
    HepLorentzVector m_clusterMom;
    HepLorentzVector m_ktClusterMom;
    HepLorentzVector m_jetMom;
    HepLorentzVector m_reconstructedParticleMom;
    HepLorentzVector m_simpleKinematicMom;
    HepLorentzVector m_trackMom;
  };
}//namespace
#endif 
  
 

---