/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_5_3_14/src/RecoParticleFlow/PFClusterTools/src/ParticleDeposit.cc

Go to the documentation of this file.

#include "RecoParticleFlow/PFClusterTools/interface/ParticleDeposit.h"
#include <cassert>
#include <iostream>
#include <cmath>
#include "RecoParticleFlow/PFClusterTools/interface/DetectorElementType.h"
using namespace pftools;

unsigned ParticleDeposit::count = 0;

ParticleDeposit::ParticleDeposit(double truthEnergy, double eta, double phi) :
        myId(count), myTruthEnergy(truthEnergy), myEta(eta), myPhi(phi) {
        ++count;
}

ParticleDeposit::~ParticleDeposit() {
}

void ParticleDeposit::addRecDeposition(Deposition rec) {
        myRecDepositions.push_back(rec);
}

void ParticleDeposit::addTruthDeposition(Deposition truth) {
        myTruthDepositions.push_back(truth);
}

std::vector<Deposition> ParticleDeposit::getTruthDepositions() const {
        return myTruthDepositions;
}

const std::vector<Deposition>& ParticleDeposit::getRecDepositions() const {
        return myRecDepositions;
}

double ParticleDeposit::getRecEnergy(const DetectorElementPtr de) const {
        double energy(0);
        for (std::vector<Deposition>::const_iterator cit = myRecDepositions.begin(); cit
                        != myRecDepositions.end(); ++cit) {
                Deposition d = *cit;

                if (d.getDetectorElement()->getType() == de->getType()) {
                        energy += de->getCalib(d.getEta(), d.getPhi()) * d.getEnergy();
                }

        }
        return energy;
}

void ParticleDeposit::setRecEnergy(const DetectorElementPtr de, double energy) {
        for (std::vector<Deposition>::const_iterator cit = myRecDepositions.begin(); cit
                        != myRecDepositions.end(); ++cit) {
                Deposition d = *cit;

                if (d.getDetectorElement()->getType() == de->getType()) {
                        d.setEnergy(energy);
                }

        }
}

double ParticleDeposit::getTruthEnergy(const DetectorElementPtr de) const {
        double energy(0);
        for (std::vector<Deposition>::const_iterator
                        cit = myTruthDepositions.begin(); cit!= myTruthDepositions.end(); ++cit) {
                Deposition d = *cit;
                if (d.getDetectorElement() == de) {
                        energy += d.getEnergy();
                }
        }
        assert(!(energy > 0));
        return energy;

}

double ParticleDeposit::getRecEnergy() const {
        double energy(0);
        for (std::vector<Deposition>::const_iterator cit = myRecDepositions.begin(); cit
                        != myRecDepositions.end(); ++cit) {
                Deposition d = *cit;
//              if (d.getDetectorElement()->getType() == OFFSET && d.getDetectorElement()->getCalib() == 1.0) {
//                      //don't add a tiny amount!
//              } else {
                        energy += d.getDetectorElement()->getCalib(d.getEta(), d.getPhi()) * d.getEnergy();
//              }
        }

        //assert(!(energy < 0));
        return energy;

}

double ParticleDeposit::getEnergyResolution() const {
        //assert(!(getRecEnergy() / myTruthEnergy < 0.0));
        return fabs((getRecEnergy() - myTruthEnergy) / sqrt(myTruthEnergy));
}

double ParticleDeposit::getTargetFunctionContrib() const {
        //assert(!(getRecEnergy() / myTruthEnergy < 0.0));
        return pow((getRecEnergy() - myTruthEnergy), 2);
}

std::ostream& pftools::operator<<(std::ostream& s, const pftools::ParticleDeposit& p) {
        s << "Particle id:\t" << p.getId() << ", \t trueEnergy: " << p.getTruthEnergy() << "\n";
        s.width(3);
        s << "\tEta:\t" << p.getEta() << ",\tphi:\t" << p.getPhi() << "\n";
        for (std::vector<Deposition>::const_iterator cit = p.getRecDepositions().begin(); cit
                        != p.getRecDepositions().end(); ++cit) {
                Deposition d = *cit;
                DetectorElementPtr de(d.getDetectorElement());
                s << "\t" << *de << ": \t=> E_contrib = ";
                s << de->getCalib(d.getEta(), d.getPhi()) * d.getEnergy() << "\n";
        }
        s << "\tTotalRecEnergy: " << p.getRecEnergy() << ",\t res: " << p.getEnergyResolution() * 100 << "%\n";
        return s;
}