ProtoJet.cc

Go to the documentation of this file.

//  ProtJet2.cc
//  Revision History:  R. Harris 10/19/05  Modified to work with real CaloTowers from Jeremy Mans
//
// #include <stdio.h>
// #include "stl_algobase.h"
// #include "stl_algo.h"

#include "DataFormats/Candidate/interface/Candidate.h"
#include "RecoParticleFlow/PFRootEvent/interface/ProtoJet.h"
#include "RecoParticleFlow/PFRootEvent/interface/JetAlgoHelper.h"

#include <vector>
#include <algorithm>               // include STL algorithm implementations
#include <numeric>         // For the use of numeric


namespace {
  class ERecombination {
    public: inline double weight (const reco::Candidate& c) {return c.energy();}
  };
  class EtRecombination {
    public: inline double weight (const reco::Candidate& c) {return c.et();}
  };
  class PRecombination {
    public: inline double weight (const reco::Candidate& c) {return c.p();}
  };
  class PtRecombination {
    public: inline double weight (const reco::Candidate& c) {return c.pt();}
  };

  template <class T>
   inline ProtoJet::LorentzVector calculateLorentzVectorRecombination(const ProtoJet::Constituents& fConstituents) {
    T weightMaker;
    ProtoJet::LorentzVector result (0,0,0,0);
    double weights = 0;
    for(ProtoJet::Constituents::const_iterator i = fConstituents.begin(); i !=  fConstituents.end(); ++i) {
      const ProtoJet::Constituent c = *i;
      double weight = weightMaker.weight (*c);
      result += c->p4() * weight;
      weights += weight;
    } //end of loop over the jet constituents
    result = result / weights;
    return result;
  }

}

ProtoJet::ProtoJet()
  : mOrdered (false), 
    mJetArea (0), 
    mPileupEnergy (0), 
    mPassNumber (0) 
{}

ProtoJet::ProtoJet(const Constituents& fConstituents) 
  : mConstituents (fConstituents),
    mOrdered (false), 
    mJetArea (0), 
    mPileupEnergy (0), 
    mPassNumber (0) 
{
  calculateLorentzVector(); 
}

ProtoJet::ProtoJet(const LorentzVector& fP4, const Constituents& fConstituents) 
  : mP4 (fP4), 
    mConstituents (fConstituents),
    mOrdered (false), 
    mJetArea (0), 
    mPileupEnergy (0), 
    mPassNumber (0) 

{}

void ProtoJet::setJetArea (float fArea) {mJetArea = fArea;}
float ProtoJet::jetArea () const {return mJetArea;}
void ProtoJet::setPileup (float fEnergy) {mPileupEnergy = fEnergy;}
float ProtoJet::pileup () const {return mPileupEnergy;}
void ProtoJet::setNPasses (int fPasses) {mPassNumber = fPasses;}
int ProtoJet::nPasses () const {return mPassNumber;}

const ProtoJet::Constituents& ProtoJet::getTowerList() {
  reorderTowers ();
  return mConstituents;
}
  
ProtoJet::Constituents ProtoJet::getTowerList() const {
  if (mOrdered) return mConstituents;
  ProtoJet::Constituents result (mConstituents);
  sortByEtRef (&result);
  return result;
}

const ProtoJet::Constituents& ProtoJet::getPresortedTowerList() const {
  if (!mOrdered) std::cerr << "ProtoJet::getPresortedTowerList-> ERROR: constituents are not sorted." << std::endl;
  return mConstituents;
}

void ProtoJet::putTowers(const Constituents& towers) {
  mConstituents = towers; 
  mOrdered = false;
  calculateLorentzVector();
}

void ProtoJet::reorderTowers () {
  if (!mOrdered) {
    sortByEtRef (&mConstituents);
    mOrdered = true;
  }
}

void ProtoJet::calculateLorentzVectorERecombination() {
  mP4 = LorentzVector (0,0,0,0);
  for(Constituents::const_iterator i = mConstituents.begin(); i !=  mConstituents.end(); ++i) {
    const Constituent c = *i;
    mP4 += c->p4();
  } //end of loop over the jet constituents
}

void ProtoJet::calculateLorentzVectorEtRecombination() {
  mP4 = calculateLorentzVectorRecombination <EtRecombination> (mConstituents);
}