ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillerPileupEnergyDensity.cc
Revision: 1.4.2.1
Committed: Fri May 18 18:11:38 2012 UTC (12 years, 11 months ago) by paus
Content type: text/plain
Branch: Mit_025c_branch
CVS Tags: Mit_025c_branch2, Mit_025c_branch1, Mit_025c_branch0
Changes since 1.4: +74 -18 lines
Log Message:
Completing the rho implemention.

File Contents

# User Rev Content
1 paus 1.4.2.1 // $Id: FillerPileupEnergyDensity.cc,v 1.4 2011/06/15 20:04:14 bendavid Exp $
2 mzanetti 1.1
3     #include "MitProd/TreeFiller/interface/FillerPileupEnergyDensity.h"
4     #include "FWCore/MessageLogger/interface/MessageLogger.h"
5     #include "DataFormats/Common/interface/Handle.h"
6     #include "MitAna/DataTree/interface/Names.h"
7     #include "MitAna/DataTree/interface/PileupEnergyDensityCol.h"
8     #include "MitProd/ObjectService/interface/ObjectService.h"
9    
10     using namespace std;
11     using namespace edm;
12     using namespace mithep;
13    
14     //--------------------------------------------------------------------------------------------------
15 paus 1.4.2.1 FillerPileupEnergyDensity::FillerPileupEnergyDensity(const ParameterSet &cfg, const char *name, bool active) :
16     BaseFiller (cfg,name,active),
17     edmName_ (Conf().getUntrackedParameter<edm::InputTag>("edmName")),
18     edmNameLowEta_ (Conf().getUntrackedParameter<edm::InputTag>("edmNameLowEta")),
19     edmNameRandom_ (Conf().getUntrackedParameter<edm::InputTag>("edmNameRandom")),
20     edmNameRandomLowEta_ (Conf().getUntrackedParameter<edm::InputTag>("edmNameRandomLowEta")),
21     edmNameFixedGridAll_ (Conf().getUntrackedParameter<edm::InputTag>("edmNameFixedGridAll")),
22     edmNameFixedGridFastjetAll_ (Conf().getUntrackedParameter<edm::InputTag>("edmNameFixedGridFastjetAll")),
23     edmNameKt6CaloJets_ (Conf().getUntrackedParameter<edm::InputTag>("edmNameKt6CaloJets")),
24     edmNameKt6CaloJetsCentral_ (Conf().getUntrackedParameter<edm::InputTag>("edmNameKt6CaloJetsCentral")),
25     edmNameKt6PFJets_ (Conf().getUntrackedParameter<edm::InputTag>("edmNameKt6PFJets")),
26     edmNameKt6PFJetsCentralChargedPileUp_(Conf().getUntrackedParameter<edm::InputTag>("edmNameKt6PFJetsCentralChargedPileUp")),
27     edmNameKt6PFJetsCentralNeutral_ (Conf().getUntrackedParameter<edm::InputTag>("edmNameKt6PFJetsCentralNeutral")),
28     edmNameKt6PFJetsCentralNeutralTight_ (Conf().getUntrackedParameter<edm::InputTag>("edmNameKt6PFJetsCentralNeutralTight")),
29     edmNameKt6PFJetsForIso31_ (Conf().getUntrackedParameter<edm::InputTag>("edmNameKt6PFJetsForIso31")),
30     edmNameKt6PFJetsForIso25_ (Conf().getUntrackedParameter<edm::InputTag>("edmNameKt6PFJetsForIso25")),
31     mitName_ (Conf().getUntrackedParameter<string>("mitName",Names::gkPileupEnergyDensityBrn)),
32     rhos_ (new mithep::PileupEnergyDensityArr)
33 mzanetti 1.1 {
34     // Constructor.
35     }
36    
37     //--------------------------------------------------------------------------------------------------
38     FillerPileupEnergyDensity::~FillerPileupEnergyDensity()
39     {
40     // Destructor.
41    
42     delete rhos_;
43     }
44    
45     //--------------------------------------------------------------------------------------------------
46     void FillerPileupEnergyDensity::BookDataBlock(TreeWriter &tws)
47     {
48     // Add pileup branch to tree.
49     tws.AddBranch(mitName_,&rhos_);
50     OS()->add<mithep::PileupEnergyDensityArr>(rhos_,mitName_);
51     }
52    
53     //--------------------------------------------------------------------------------------------------
54 paus 1.4.2.1 void FillerPileupEnergyDensity::FillDataBlock(const edm::Event &event,
55     const edm::EventSetup &setup)
56 mzanetti 1.1 {
57     // Fill missing energy from edm collection into our collection.
58    
59     rhos_->Delete();
60    
61     Handle<double> hRho;
62 mzanetti 1.2 event.getByLabel(edmName_,hRho);
63 mzanetti 1.1
64 bendavid 1.4 Handle<double> hRhoLowEta;
65     event.getByLabel(edmNameLowEta_,hRhoLowEta);
66    
67     Handle<double> hRhoRandom;
68     event.getByLabel(edmNameRandom_,hRhoRandom);
69    
70     Handle<double> hRhoRandomLowEta;
71     event.getByLabel(edmNameRandomLowEta_,hRhoRandomLowEta);
72 mzanetti 1.3
73 paus 1.4.2.1 Handle<double> hRhoFixedGridAll;
74     event.getByLabel(edmNameFixedGridAll_,hRhoFixedGridAll);
75    
76     Handle<double> hRhoFixedGridFastjetAll;
77     event.getByLabel(edmNameFixedGridFastjetAll_,hRhoFixedGridFastjetAll);
78    
79     Handle<double> hRhoKt6CaloJets;
80     event.getByLabel(edmNameKt6CaloJets_,hRhoKt6CaloJets);
81    
82     Handle<double> hRhoKt6CaloJetsCentral;
83     event.getByLabel(edmNameKt6CaloJetsCentral_,hRhoKt6CaloJetsCentral);
84    
85     Handle<double> hRhoKt6PFJets;
86     event.getByLabel(edmNameKt6PFJets_,hRhoKt6PFJets);
87    
88     Handle<double> hRhoKt6PFJetsCentralChargedPileUp;
89     event.getByLabel(edmNameKt6PFJetsCentralChargedPileUp_,hRhoKt6PFJetsCentralChargedPileUp);
90    
91     Handle<double> hRhoKt6PFJetsCentralNeutral;
92     event.getByLabel(edmNameKt6PFJetsCentralNeutral_,hRhoKt6PFJetsCentralNeutral);
93    
94     Handle<double> hRhoKt6PFJetsCentralNeutralTight;
95     event.getByLabel(edmNameKt6PFJetsCentralNeutralTight_,hRhoKt6PFJetsCentralNeutralTight);
96    
97     Handle<double> hRhoKt6PFJetsForIso31;
98     event.getByLabel(edmNameKt6PFJetsForIso31_,hRhoKt6PFJetsForIso31);
99    
100     Handle<double> hRhoKt6PFJetsForIso25;
101     event.getByLabel(edmNameKt6PFJetsForIso25_,hRhoKt6PFJetsForIso25);
102    
103 mzanetti 1.1 mithep::PileupEnergyDensity *rho = rhos_->AddNew();
104    
105 paus 1.4.2.1 rho->SetRho (*hRho);
106     rho->SetRhoLowEta (*hRhoLowEta);
107     rho->SetRhoRandom (*hRhoRandom);
108 bendavid 1.4 rho->SetRhoRandomLowEta(*hRhoRandomLowEta);
109 mzanetti 1.1
110 paus 1.4.2.1 if (hRhoFixedGridAll.isValid())
111     rho->SetRhoFixedGridAll(*hRhoFixedGridAll);
112     else
113     rho->SetRhoFixedGridAll(-9999);
114    
115     if (hRhoFixedGridFastjetAll.isValid())
116     rho->SetRhoFixedGridFastjetAll(*hRhoFixedGridFastjetAll);
117     else
118     rho->SetRhoFixedGridFastjetAll(-9999);
119     // this was not there in 4x
120     rho->SetRhoKt6CaloJets (-9999);
121     rho->SetRhoKt6CaloJetsCentral (-9999);
122     rho->SetRhoKt6PFJets (-9999);
123     rho->SetRhoKt6PFJetsCentralChargedPileUp(-9999);
124     rho->SetRhoKt6PFJetsCentralNeutral (-9999);
125     rho->SetRhoKt6PFJetsCentralNeutralTight (-9999);
126     // new
127     rho->SetRhoKt6PFJetsForIso31 (*hRhoKt6PFJetsForIso31);
128     rho->SetRhoKt6PFJetsForIso25 (*hRhoKt6PFJetsForIso25);
129    
130 mzanetti 1.1 rhos_->Trim();
131     }