/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_6_2_7/src/PhysicsTools/TagAndProbe/plugins/ColinsSoperVariablesComputer.cc

Go to the documentation of this file.

//

#include "FWCore/Framework/interface/EDProducer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"

#include "DataFormats/Common/interface/ValueMap.h"
#include "DataFormats/Common/interface/View.h"

#include "DataFormats/Candidate/interface/CandidateFwd.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "PhysicsTools/TagAndProbe/interface/ColinsSoperVariables.h"
#include "TLorentzVector.h"


class ColinsSoperVariablesComputer : public edm::EDProducer {
    public:
        explicit ColinsSoperVariablesComputer(const edm::ParameterSet & iConfig);
        virtual ~ColinsSoperVariablesComputer() ;

        virtual void produce(edm::Event & iEvent, const edm::EventSetup& iSetup) override;

    private:
        edm::InputTag parentBoson_;            
};

ColinsSoperVariablesComputer::ColinsSoperVariablesComputer(const edm::ParameterSet & iConfig) :
    parentBoson_(iConfig.getParameter<edm::InputTag>("parentBoson"))
{
    produces<edm::ValueMap<float> >("costheta");
    produces<edm::ValueMap<float> >("sin2theta");
    produces<edm::ValueMap<float> >("tanphi");
}


ColinsSoperVariablesComputer::~ColinsSoperVariablesComputer()
{
}

void 
ColinsSoperVariablesComputer::produce(edm::Event & iEvent, const edm::EventSetup & iSetup) {
    using namespace edm;

    // read input
    Handle<View<reco::Candidate> > bosons;
    iEvent.getByLabel(parentBoson_,  bosons);


    // prepare vector for output   
    std::vector<float> values;
    std::vector<float> values2;
    std::vector<float> values3;

    // fill: use brute force    
    double costheta = -10.0;
    double sin2theta = -10.0;
    double tanphi = -10.0;

    const reco::Candidate* daughter1=NULL;
    const reco::Candidate* daughter2=NULL;
    TLorentzVector mu (0., 0., 0., 0.);
    TLorentzVector mubar (0., 0., 0., 0.);
    bool isOS  = false;
    int charge1=0, charge2 =0;
    double res[3] = { -10., -10., -10.};

    View<reco::Candidate>::const_iterator boson, endbosons = bosons->end(); 

    for (boson = bosons->begin(); boson != endbosons; ++boson) {

      daughter1 = boson->daughter(0);
      daughter2 = boson->daughter(1);

      if( !(0==daughter1 || 0==daughter2) ) {
        isOS = false;
        charge1 = daughter1->charge();
        charge2 = daughter2->charge();
        isOS = charge1*charge2<0;
        if(isOS && charge1<0) {
          mu.SetPxPyPzE( daughter1->px(), daughter1->py(), daughter1->pz(), daughter1->energy() );
          mubar.SetPxPyPzE( daughter2->px(), daughter2->py(), daughter2->pz(), daughter2->energy() );
        }
        if(isOS && charge1>0) {
          mu.SetPxPyPzE( daughter2->px(), daughter2->py(), daughter2->pz(), daughter2->energy() );
          mubar.SetPxPyPzE( daughter1->px(), daughter1->py(), daughter1->pz(), daughter1->energy() );
        }
      }

      
      calCSVariables(mu, mubar, res, boson->pz()<0.0);

      costheta = res[0];
      sin2theta = res[1];
      tanphi = res[2];

      values.push_back(costheta);
      values2.push_back(sin2theta);
      values3.push_back(tanphi);
    }


    // convert into ValueMap and store
    std::auto_ptr<ValueMap<float> > valMap(new ValueMap<float>());
    ValueMap<float>::Filler filler(*valMap);
    filler.insert(bosons, values.begin(), values.end());
    filler.fill();
    iEvent.put(valMap, "costheta");


    // ---> same for sin2theta   
    std::auto_ptr<ValueMap<float> > valMap2(new ValueMap<float>());
    ValueMap<float>::Filler filler2(*valMap2);
    filler2.insert(bosons, values2.begin(), values2.end());
    filler2.fill();
    iEvent.put(valMap2, "sin2theta");

    // ---> same for tanphi   
    std::auto_ptr<ValueMap<float> > valMap3(new ValueMap<float>());
    ValueMap<float>::Filler filler3(*valMap3);
    filler3.insert(bosons, values3.begin(), values3.end());
    filler3.fill();
    iEvent.put(valMap3, "tanphi");

}


#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(ColinsSoperVariablesComputer);