ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillerPileupEnergyDensity.cc
Revision: 1.6
Committed: Sun May 6 12:45:05 2012 UTC (13 years ago) by paus
Content type: text/plain
Branch: MAIN
CVS Tags: Mit_032, Mit_031, Mit_030, Mit_029c, Mit_029b, Mit_030_pre1, Mit_029a, Mit_029, Mit_029_pre1, Mit_028a, Mit_028, Mit_027a, Mit_027, HEAD
Changes since 1.5: +3 -3 lines
Log Message:
Small fixes.

File Contents

# User Rev Content
1 paus 1.6 // $Id: FillerPileupEnergyDensity.cc,v 1.5 2012/05/05 16:49:59 paus 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.5 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 paus 1.6 edmNameFixedGridAll_ (Conf().getUntrackedParameter<edm::InputTag>("edmNameFixedGridAll")),
22     edmNameFixedGridFastjetAll_ (Conf().getUntrackedParameter<edm::InputTag>("edmNameFixedGridFastjetAll")),
23 paus 1.5 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     mitName_ (Conf().getUntrackedParameter<string>("mitName",Names::gkPileupEnergyDensityBrn)),
30     rhos_ (new mithep::PileupEnergyDensityArr)
31 mzanetti 1.1 {
32     // Constructor.
33     }
34    
35     //--------------------------------------------------------------------------------------------------
36     FillerPileupEnergyDensity::~FillerPileupEnergyDensity()
37     {
38     // Destructor.
39    
40     delete rhos_;
41     }
42    
43     //--------------------------------------------------------------------------------------------------
44     void FillerPileupEnergyDensity::BookDataBlock(TreeWriter &tws)
45     {
46     // Add pileup branch to tree.
47     tws.AddBranch(mitName_,&rhos_);
48     OS()->add<mithep::PileupEnergyDensityArr>(rhos_,mitName_);
49     }
50    
51     //--------------------------------------------------------------------------------------------------
52 paus 1.5 void FillerPileupEnergyDensity::FillDataBlock(const edm::Event &event,
53     const edm::EventSetup &setup)
54 mzanetti 1.1 {
55     // Fill missing energy from edm collection into our collection.
56    
57     rhos_->Delete();
58    
59     Handle<double> hRho;
60 mzanetti 1.2 event.getByLabel(edmName_,hRho);
61 mzanetti 1.1
62 bendavid 1.4 Handle<double> hRhoLowEta;
63     event.getByLabel(edmNameLowEta_,hRhoLowEta);
64    
65     Handle<double> hRhoRandom;
66     event.getByLabel(edmNameRandom_,hRhoRandom);
67    
68     Handle<double> hRhoRandomLowEta;
69     event.getByLabel(edmNameRandomLowEta_,hRhoRandomLowEta);
70 mzanetti 1.3
71 paus 1.5 Handle<double> hRhoFixedGridAll;
72     event.getByLabel(edmNameFixedGridAll_,hRhoFixedGridAll);
73    
74     Handle<double> hRhoFixedGridFastjetAll;
75     event.getByLabel(edmNameFixedGridFastjetAll_,hRhoFixedGridFastjetAll);
76    
77     Handle<double> hRhoKt6CaloJets;
78     event.getByLabel(edmNameKt6CaloJets_,hRhoKt6CaloJets);
79    
80     Handle<double> hRhoKt6CaloJetsCentral;
81     event.getByLabel(edmNameKt6CaloJetsCentral_,hRhoKt6CaloJetsCentral);
82    
83     Handle<double> hRhoKt6PFJets;
84     event.getByLabel(edmNameKt6PFJets_,hRhoKt6PFJets);
85    
86     Handle<double> hRhoKt6PFJetsCentralChargedPileUp;
87     event.getByLabel(edmNameKt6PFJetsCentralChargedPileUp_,hRhoKt6PFJetsCentralChargedPileUp);
88    
89     Handle<double> hRhoKt6PFJetsCentralNeutral;
90     event.getByLabel(edmNameKt6PFJetsCentralNeutral_,hRhoKt6PFJetsCentralNeutral);
91    
92     Handle<double> hRhoKt6PFJetsCentralNeutralTight;
93     event.getByLabel(edmNameKt6PFJetsCentralNeutralTight_,hRhoKt6PFJetsCentralNeutralTight);
94    
95 mzanetti 1.1 mithep::PileupEnergyDensity *rho = rhos_->AddNew();
96    
97     rho->SetRho(*hRho);
98 bendavid 1.4 rho->SetRhoLowEta(*hRhoLowEta);
99     rho->SetRhoRandom(*hRhoRandom);
100     rho->SetRhoRandomLowEta(*hRhoRandomLowEta);
101 mzanetti 1.1
102 paus 1.5 if (hRhoFixedGridAll.isValid())
103     rho->SetRhoFixedGridAll(*hRhoFixedGridAll);
104     else
105     rho->SetRhoFixedGridAll(-9999);
106    
107     if (hRhoFixedGridFastjetAll.isValid())
108     rho->SetRhoFixedGridFastjetAll(*hRhoFixedGridFastjetAll);
109     else
110     rho->SetRhoFixedGridFastjetAll(-9999);
111    
112     rho->SetRhoKt6CaloJets (*hRhoKt6CaloJets);
113     rho->SetRhoKt6CaloJetsCentral (*hRhoKt6CaloJetsCentral);
114     rho->SetRhoKt6PFJets (*hRhoKt6PFJets);
115     rho->SetRhoKt6PFJetsCentralChargedPileUp(*hRhoKt6PFJetsCentralChargedPileUp);
116     rho->SetRhoKt6PFJetsCentralNeutral (*hRhoKt6PFJetsCentralNeutral);
117     rho->SetRhoKt6PFJetsCentralNeutralTight (*hRhoKt6PFJetsCentralNeutralTight);
118    
119 mzanetti 1.1 rhos_->Trim();
120     }