ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/src/FillerConversions.cc
Revision: 1.11
Committed: Thu Feb 26 17:04:03 2009 UTC (16 years, 2 months ago) by bendavid
Content type: text/plain
Branch: MAIN
CVS Tags: Mit_008pre1
Changes since 1.10: +2 -2 lines
Log Message: 
Switch from Reset to Delete calls on arrays, since we now use some heap

File Contents

# User Rev Content
1 bendavid 1.11 // $Id: FillerConversions.cc,v 1.10 2008/11/03 18:11:10 bendavid Exp $
2 bendavid 1.1
3 loizides 1.2 #include "MitProd/TreeFiller/interface/FillerConversions.h"
4 bendavid 1.1 #include "FWCore/MessageLogger/interface/MessageLogger.h"
5     #include "DataFormats/Common/interface/Handle.h"
6 bendavid 1.10 #include "DataFormats/Common/interface/RefToPtr.h"
7 bendavid 1.1 #include "MitAna/DataTree/interface/Names.h"
8    
9     using namespace std;
10     using namespace edm;
11     using namespace mithep;
12    
13 loizides 1.2 //--------------------------------------------------------------------------------------------------
14 loizides 1.8 FillerConversions::FillerConversions(const ParameterSet &cfg, bool active) :
15 loizides 1.6 BaseFiller(cfg,"Conversions",active),
16 bendavid 1.1 edmName_(Conf().getUntrackedParameter<string>("edmName","conversions")),
17     mitName_(Conf().getUntrackedParameter<string>("mitName","Conversions")),
18 loizides 1.8 convElectronMapName_(Conf().getUntrackedParameter<string>("convElectronMapName","")),
19     conversionMapName_(Conf().getUntrackedParameter<string>("conversionMapName",
20     Form("%sMapName",mitName_.c_str()))),
21     convElectronMap_(0),
22 loizides 1.6 conversions_(new mithep::ConversionArr(16)),
23 bendavid 1.1 conversionMap_(new mithep::ConversionMap)
24     {
25 loizides 1.2 // Constructor.
26 bendavid 1.1 }
27    
28 loizides 1.2 //--------------------------------------------------------------------------------------------------
29 bendavid 1.1 FillerConversions::~FillerConversions()
30     {
31 loizides 1.2 // Destructor.
32    
33 loizides 1.4 delete conversions_;
34 loizides 1.2 delete conversionMap_;
35 bendavid 1.1 }
36    
37 loizides 1.2 //--------------------------------------------------------------------------------------------------
38 bendavid 1.1 void FillerConversions::BookDataBlock(TreeWriter &tws)
39     {
40 loizides 1.8 // Add conversions to tree. Publish and get our objects.
41 loizides 1.2
42 bendavid 1.1 tws.AddBranch(mitName_.c_str(),&conversions_);
43 loizides 1.8
44     if (!convElectronMapName_.empty())
45     OS()->add(conversionMap_,conversionMapName_.c_str());
46     convElectronMap_ = OS()->get<ConversionElectronMap>(convElectronMapName_.c_str());
47 bendavid 1.1 }
48    
49 loizides 1.2 //--------------------------------------------------------------------------------------------------
50 bendavid 1.1 void FillerConversions::FillDataBlock(const edm::Event &event,
51     const edm::EventSetup &setup)
52     {
53 loizides 1.5 // Fill conversions data structure and maps.
54 bendavid 1.1
55 bendavid 1.11 conversions_->Delete();
56 bendavid 1.1 conversionMap_->Reset();
57 loizides 1.5
58     Handle<reco::ConversionCollection> hConversionProduct;
59     GetProduct(edmName_, hConversionProduct, event);
60    
61     conversionMap_->SetEdmProductId(hConversionProduct.id().id());
62 bendavid 1.1
63 loizides 1.5 const reco::ConversionCollection inConversions = *(hConversionProduct.product());
64 bendavid 1.1
65     for (reco::ConversionCollection::const_iterator inConversion = inConversions.begin();
66     inConversion != inConversions.end(); ++inConversion) {
67    
68 loizides 1.6 mithep::Conversion *outConversion = conversions_->Allocate();
69 loizides 1.5 new (outConversion) mithep::Conversion(inConversion->conversionVertex().x(),
70     inConversion->conversionVertex().y(),
71     inConversion->conversionVertex().z(),
72     inConversion->conversionVertex().xError(),
73     inConversion->conversionVertex().yError(),
74     inConversion->conversionVertex().zError());
75 bendavid 1.3
76 loizides 1.7 outConversion->DecayVertex().SetChi2(inConversion->conversionVertex().chi2());
77 bendavid 1.9 outConversion->DecayVertex().SetNdof((Int_t)inConversion->conversionVertex().ndof());
78 loizides 1.5
79     outConversion->SetDCotTheta(inConversion->pairCotThetaSeparation());
80     outConversion->SetEOverP(inConversion->EoverP());
81     outConversion->SetPairMass(inConversion->pairInvariantMass());
82     outConversion->SetPairMomentum(inConversion->pairMomentum().x(),
83     inConversion->pairMomentum().y(),
84     inConversion->pairMomentum().z());
85 bendavid 1.1
86 loizides 1.8 if (convElectronMap_) {
87 loizides 1.5 std::vector<reco::TrackRef> trackRefs = inConversion->tracks();
88     for (std::vector<reco::TrackRef>::const_iterator trackRef = trackRefs.begin();
89     trackRef != trackRefs.end(); ++trackRef) {
90 bendavid 1.10 outConversion->AddDaughter(convElectronMap_->GetMit(refToPtr(*trackRef)));
91 loizides 1.5 }
92     }
93    
94     reco::ConversionRef theRef(hConversionProduct, inConversion-inConversions.begin());
95     conversionMap_->Add(theRef, outConversion);
96 bendavid 1.1 }
97 loizides 1.2
98 bendavid 1.1 conversions_->Trim();
99     }