CMSSW_4_4_3_patch1/src/RecoParticleFlow/PFClusterTools/interface/ParticleDeposit.h

Go to the documentation of this file.

#ifndef PARTICLEDEPOSIT_HH_
#define PARTICLEDEPOSIT_HH_

#include <vector>

#include "RecoParticleFlow/PFClusterTools/interface/Deposition.h"
#include "RecoParticleFlow/PFClusterTools/interface/DetectorElement.h"


#include <iostream>

namespace pftools {

class ParticleDeposit {
public:

        ParticleDeposit(double truthEnergy = -1.0, double eta = 0, double phi = 0);
        virtual ~ParticleDeposit();

        virtual void addRecDeposition(Deposition rec);
        virtual void addTruthDeposition(Deposition truth);

        virtual const std::vector<Deposition>& getRecDepositions() const;
        virtual std::vector<Deposition> getTruthDepositions() const;

        /*
         * Returns the overall MC particle energy.
         */
        virtual double getTruthEnergy() const {
                return myTruthEnergy;
        }

        /* 
         * Returns the detected energy from this detector element, including calibration.
         */
        virtual double getRecEnergy(const DetectorElementPtr de) const;

        virtual double getRecEnergy() const;
        
        virtual void setRecEnergy(const DetectorElementPtr de, double energy);

        virtual double getEnergyResolution() const;

        /*
         * Returns the raw MC energy input into this detector element.
         */
        virtual double getTruthEnergy(const DetectorElementPtr de) const;

        virtual unsigned getId() const {
                return myId;
        }

        inline double getEta() const {
                return myEta;
        }

        virtual double getPhi() const {
                return myPhi;
        }
        
        void setTruthEnergy(const double truth) {
                myTruthEnergy = truth;
        }
        
        void setPhi(const double phi) {
                myPhi = phi;
        }
        
        void setEta(const double eta) {
                myEta = eta;
        }

        double getTargetFunctionContrib() const;

        friend std::ostream& operator<<(std::ostream& s, const ParticleDeposit& p);

private:
        static unsigned count;
        //ParticleDeposit(const ParticleDeposit& pd);
        std::vector<Deposition> myRecDepositions;
        std::vector<Deposition> myTruthDepositions;
        unsigned myId;

        double myTruthEnergy;
        double myEta;
        double myPhi;

        /*
         * For general ROOT dictionary building happiness!
         */
        std::vector<ParticleDeposit*> pdps_;
        //      std::vector<ParticleDeposit> pds_;
};


typedef boost::shared_ptr<ParticleDeposit> ParticleDepositPtr;

}

#endif /*PARTICLEDEPOSIT_HH_*/