PackedGenParticle.h

Go to the documentation of this file.

#ifndef __AnalysisDataFormats_PackedGenParticle_h__
#define __AnalysisDataFormats_PackedGenParticle_h__

#include "DataFormats/HepMCCandidate/interface/GenParticleFwd.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/Candidate/interface/CandidateFwd.h"
#include "DataFormats/Candidate/interface/iterator_imp_specific.h"
#include "DataFormats/Common/interface/RefVector.h"
#include "DataFormats/Common/interface/Association.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/Math/interface/deltaPhi.h" 
/* #include "DataFormats/Math/interface/PtEtaPhiMass.h" */

namespace pat {
  class PackedGenParticle : public reco::Candidate {
  public:
    typedef reco::CandidateCollection daughters;
    typedef math::XYZTLorentzVector LorentzVector;
    typedef math::PtEtaPhiMLorentzVector PolarLorentzVector;
    typedef math::XYZPoint Point;
    typedef math::XYZVector Vector;

    typedef unsigned int index;

  PackedGenParticle()
    : p4_(0,0,0,0), p4c_(0,0,0,0), vertex_(0,0,0),  pdgId_(0), charge_(0), unpacked_(false) { }
  explicit PackedGenParticle( const reco::GenParticle & c)
    : p4_(c.pt(), c.eta(), c.phi(), c.mass()), p4c_(p4_), vertex_(0,0,0), pdgId_(c.pdgId()), charge_(c.charge()), mother_(c.motherRef(0)), unpacked_(true)  { pack(); }
  explicit PackedGenParticle( const reco::GenParticle & c, const edm::Ref<reco::GenParticleCollection> &  mother)
    : p4_(c.pt(), c.eta(), c.phi(), c.mass()), p4c_(p4_), vertex_(0,0,0), pdgId_(c.pdgId()), charge_(c.charge()), mother_(mother), unpacked_(true)  { pack(); }

    
    virtual ~PackedGenParticle();
    virtual const_iterator begin() const;
    virtual const_iterator end() const;
    virtual iterator begin();
    virtual iterator end();
    virtual size_t numberOfDaughters() const;
    virtual const reco::Candidate * daughter( size_type ) const;
    virtual size_t numberOfMothers() const;
    virtual const reco::Candidate * mother( size_type ) const;
    virtual reco::Candidate * daughter( size_type );
    virtual reco::Candidate * daughter(const std::string& s );
    virtual const reco::Candidate * daughter(const std::string& s ) const;
    virtual size_t numberOfSourceCandidatePtrs() const {return 0;}
    virtual reco::CandidatePtr sourceCandidatePtr( size_type i ) const {
      return reco::CandidatePtr();
    }

    virtual int charge() const {
    return charge_; 
    }
    virtual void setCharge( int charge) {charge_=charge;}
    virtual int threeCharge() const {return charge()*3;}
    virtual void setThreeCharge( int threecharge) {}
    virtual const LorentzVector & p4() const { if (!unpacked_) unpack(); return p4c_; }  
    virtual const PolarLorentzVector & polarP4() const { if (!unpacked_) unpack(); return p4_; }
    virtual Vector momentum() const  { if (!unpacked_) unpack(); return p4c_.Vect(); }
    virtual Vector boostToCM() const { if (!unpacked_) unpack(); return p4c_.BoostToCM(); }
    virtual double p() const { if (!unpacked_) unpack(); return p4c_.P(); }
    virtual double energy() const { if (!unpacked_) unpack(); return p4c_.E(); }
    virtual double et() const { if (!unpacked_) unpack(); return p4_.Et(); }
    virtual float mass() const { if (!unpacked_) unpack(); return p4_.M(); }
    virtual float massSqr() const { if (!unpacked_) unpack(); return p4_.M()*p4_.M(); }

    virtual double mt() const { if (!unpacked_) unpack(); return p4_.Mt(); }
    virtual double mtSqr() const { if (!unpacked_) unpack(); return p4_.Mt2(); }
    virtual double px() const { if (!unpacked_) unpack(); return p4c_.Px(); }
    virtual double py() const { if (!unpacked_) unpack(); return p4c_.Py(); }
    virtual double pz() const { if (!unpacked_) unpack(); return p4c_.Pz(); }
    virtual float pt() const { if (!unpacked_) unpack(); return p4_.Pt();}
    virtual float phi() const { if (!unpacked_) unpack(); return p4_.Phi(); }
    virtual double theta() const { if (!unpacked_) unpack(); return p4_.Theta(); }
    virtual float eta() const { if (!unpacked_) unpack(); return p4_.Eta(); }
    virtual double rapidity() const { if (!unpacked_) unpack(); return p4_.Rapidity(); }
    virtual double y() const { if (!unpacked_) unpack(); return p4_.Rapidity(); }
    virtual void setP4( const LorentzVector & p4 ) { 
        unpack(); // changing px,py,pz changes also mapping between dxy,dz and x,y,z
        p4_ = PolarLorentzVector(p4.Pt(), p4.Eta(), p4.Phi(), p4.M());
        pack();
    }
    virtual void setP4( const PolarLorentzVector & p4 ) { 
        unpack(); // changing px,py,pz changes also mapping between dxy,dz and x,y,z
        p4_ = p4; 
        pack();
    }
    virtual void setMass( double m ) {
      if (!unpacked_) unpack(); 
      p4_ = PolarLorentzVector(p4_.Pt(), p4_.Eta(), p4_.Phi(), m); 
      pack();
    }
    virtual void setPz( double pz ) {
      unpack(); // changing px,py,pz changes also mapping between dxy,dz and x,y,z
      p4c_ = LorentzVector(p4c_.Px(), p4c_.Py(), pz, p4c_.E());
      p4_  = PolarLorentzVector(p4c_.Pt(), p4c_.Eta(), p4c_.Phi(), p4c_.M());
      pack();
    }
    virtual const Point & vertex() const { return vertex_; }//{ if (fromPV_) return Point(0,0,0); else return Point(0,0,100); }
    virtual double vx() const  {  return vertex_.X(); }//{ return 0; }
    virtual double vy() const  {  return vertex_.Y(); }//{ return 0; }
    virtual double vz() const  {  return vertex_.Z(); }//{ if (fromPV_) return 0; else return 100; }
    virtual void setVertex( const Point & vertex ) { vertex_ = vertex;  }

    enum PVAssoc { NoPV=0, PVLoose=1, PVTight=2, PVUsedInFit=3 } ;


    virtual float dxy() const { unpack(); return dxy_; }
    virtual float dz()  const { unpack(); return dz_; }
    virtual float dxy(const Point &p) const ;
    virtual float dz(const Point &p)  const ;


    virtual int pdgId() const   { return pdgId_; }
    // set PDG identifier                                                                 
    virtual void setPdgId( int pdgId )   { pdgId_ = pdgId; }
    virtual int status() const   { return 1; } /*FIXME*/
    virtual void setStatus( int status ) {} /*FIXME*/
    static const unsigned int longLivedTag = 0; /*FIXME*/
    virtual void setLongLived() {} /*FIXME*/
    virtual bool longLived() const;
    static const unsigned int massConstraintTag = 0; /*FIXME*/ 
    virtual void setMassConstraint() {} /*FIXME*/
    virtual bool massConstraint() const;

    virtual PackedGenParticle * clone() const  {
      return new PackedGenParticle( *this );
    }

    virtual double vertexChi2() const;
    virtual double vertexNdof() const;
    virtual double vertexNormalizedChi2() const;
    virtual double vertexCovariance(int i, int j) const;
    CovarianceMatrix vertexCovariance() const   { CovarianceMatrix m; fillVertexCovariance(m); return m; }
    virtual void fillVertexCovariance(CovarianceMatrix & v) const;
    virtual bool hasMasterClone() const;
    virtual const reco::CandidateBaseRef & masterClone() const;
    virtual bool hasMasterClonePtr() const;

    virtual const reco::CandidatePtr & masterClonePtr() const;

    template<typename Ref>
      Ref masterRef() const { return masterClone().template castTo<Ref>(); }

    /* template<typename T> T get() const { */
    /*   if ( hasMasterClone() ) return masterClone()->get<T>(); */
    /*   else return reco::get<T>( * this ); */
    /* } */
    /* /// get a component                                                                                                 */
    /* template<typename T, typename Tag> T get() const { */
    /*   if ( hasMasterClone() ) return masterClone()->get<T, Tag>(); */
    /*   else return reco::get<T, Tag>( * this ); */
    /* } */
    /* /// get a component                                                                                                 */
    /* template<typename T> T get( size_type i ) const { */
    /*   if ( hasMasterClone() ) return masterClone()->get<T>( i ); */
    /*   else return reco::get<T>( * this, i ); */
    /* } */
    /* /// get a component                                                                                                 */
    /* template<typename T, typename Tag> T get( size_type i ) const { */
    /*   if ( hasMasterClone() ) return masterClone()->get<T, Tag>( i ); */
    /*   else return reco::get<T, Tag>( * this, i ); */
    /* } */
    /* /// number of components                                                                                            */
    /* template<typename T> size_type numberOf() const { */
    /*   if ( hasMasterClone() ) return masterClone()->numberOf<T>(); */
    /*   else return reco::numberOf<T>( * this ); */
    /* } */
    /* /// number of components                                                                                            */
    /* template<typename T, typename Tag> size_type numberOf() const { */
    /*   if ( hasMasterClone() ) return masterClone()->numberOf<T, Tag>(); */
    /*   else return reco::numberOf<T, Tag>( * this ); */
    /* } */

    /* template<typename S> */
    /*   struct daughter_iterator   { */
    /*     typedef boost::filter_iterator<S, const_iterator> type; */
    /*   }; */

    /* template<typename S> */
    /*   typename daughter_iterator<S>::type beginFilter( const S & s ) const { */
    /*   return boost::make_filter_iterator(s, begin(), end()); */
    /* } */
    /* template<typename S> */
    /*   typename daughter_iterator<S>::type endFilter( const S & s ) const { */
    /*   return boost::make_filter_iterator(s, end(), end()); */
    /* } */


    virtual bool isElectron() const;
    virtual bool isMuon() const;
    virtual bool isStandAloneMuon() const;
    virtual bool isGlobalMuon() const;
    virtual bool isTrackerMuon() const;
    virtual bool isCaloMuon() const;
    virtual bool isPhoton() const;
    virtual bool isConvertedPhoton() const;
    virtual bool isJet() const;

  protected:
    uint16_t packedPt_, packedEta_, packedPhi_, packedM_;
    void pack(bool unpackAfterwards=true) ;
    void unpack() const ;
 
    mutable PolarLorentzVector p4_;
    mutable LorentzVector p4c_;
    mutable Point vertex_;
    mutable float dxy_, dz_, dphi_;
    int pdgId_;
    int8_t charge_;
    reco::GenParticleRef mother_;
    // is the momentum p4 unpacked
    mutable bool unpacked_;

    virtual bool overlap( const reco::Candidate & ) const;
    template<typename, typename, typename> friend struct component;
    friend class ::OverlapChecker;
    friend class ShallowCloneCandidate;
    friend class ShallowClonePtrCandidate;

  private:
    // const iterator implementation
    typedef reco::candidate::const_iterator_imp_specific<daughters> const_iterator_imp_specific;
    // iterator implementation
    typedef reco::candidate::iterator_imp_specific<daughters> iterator_imp_specific;
  };

  typedef std::vector<pat::PackedGenParticle> PackedGenParticleCollection;
  typedef edm::Ref<pat::PackedGenParticleCollection> PackedGenParticleRef;
  typedef edm::RefVector<pat::PackedGenParticleCollection> PackedGenParticleRefVector;
}

#endif