ViewVC Help
View File | Revision Log | Show Annotations | Root Listing
root/cvsroot/UserCode/MitProd/TreeFiller/interface/AssociationMap.h
Revision: 1.9
Committed: Sun Mar 15 11:20:40 2009 UTC (16 years, 1 month ago) by loizides
Content type: text/plain
Branch: MAIN
CVS Tags: Mit_026, Mit_025e, Mit_025d, Mit_025c, Mit_025b, Mit_025a, Mit_025, Mit_025pre2, Mit_024b, Mit_025pre1, Mit_024a, Mit_024, Mit_023, Mit_022a, Mit_022, Mit_020d, TMit_020d, Mit_020c, Mit_021, Mit_021pre2, Mit_021pre1, Mit_020b, Mit_020a, Mit_020, Mit_020pre1, Mit_018, Mit_017, Mit_017pre3, Mit_017pre2, Mit_017pre1, Mit_016, Mit_015b, Mit_015a, Mit_015, Mit_014e, Mit_014d, Mit_014c, Mit_014b, Mit_014a, Mit_014, Mit_014pre3, Mit_014pre2, Mit_014pre1, Mit_013d, Mit_013c, Mit_013b, Mit_013a, Mit_013, Mit_013pre1, Mit_012i, Mit_012h, Mit_012g, Mit_012f, Mit_012e, Mit_012d, Mit_012c, Mit_012b, Mit_012a, Mit_012, Mit_011a, Mit_011, Mit_010a, Mit_010, Mit_009c, Mit_009b, Mit_009a, Mit_009, Mit_008, Mit_008pre2
Branch point for: Mit_025c_branch
Changes since 1.8: +3 -2 lines
Log Message: 
Introduced BranchTable plus general cleanup.

File Contents

# User Rev Content
1 loizides 1.1 //--------------------------------------------------------------------------------------------------
2 loizides 1.9 // $Id: AssociationMap.h,v 1.8 2009/03/14 06:04:59 loizides Exp $
3 loizides 1.1 //
4 loizides 1.9 // AssociationMap
5 loizides 1.1 //
6 loizides 1.2 // Wrapper for std::map, used to resolve links during tree filling.
7 loizides 1.1 //
8 loizides 1.2 // Authors: J.Bendavid, C.Loizides
9 loizides 1.1 //--------------------------------------------------------------------------------------------------
10    
11 loizides 1.6 #ifndef MITPROD_TREEFILLER_ASSOCIATIONMAP_H
12     #define MITPROD_TREEFILLER_ASSOCIATIONMAP_H
13 loizides 1.1
14     #include <map>
15     #include <TObject.h>
16 bendavid 1.3 #include "FWCore/MessageLogger/interface/MessageLogger.h"
17 loizides 1.1
18     namespace mithep
19     {
20     template <class EdmClass, class MitClass> class AssociationMap : public TObject
21     {
22     typedef std::map<EdmClass, MitClass> fwdMapType;
23     typedef std::map<MitClass, EdmClass> revMapType;
24    
25     public:
26 loizides 1.8 AssociationMap() : edmProductId_(0) {}
27 loizides 1.1 ~AssociationMap() {}
28    
29 loizides 1.7 void Add(EdmClass edmObj, MitClass mitObj);
30     EdmClass GetEdm(MitClass mitObj) const;
31     Int_t GetEdmProductId() const { return edmProductId_; }
32     Int_t GetEntries() const { return fwdMap_.size(); }
33     MitClass GetMit(EdmClass edmObj) const;
34 loizides 1.8 const char *GetBrName() const { return brname_.c_str(); }
35 loizides 1.7 bool HasMit(EdmClass edmObj) const;
36     void Reset() { fwdMap_.clear(); revMap_.clear(); }
37     void SetEdmProductId(Int_t id) { edmProductId_ = id; }
38 loizides 1.9 void SetBrName(const std::string &n) { brname_ = n; }
39 loizides 1.8 void SetBrName(const char *n) { brname_ = n; }
40 loizides 1.1
41     protected:
42 loizides 1.7 fwdMapType fwdMap_; //map between edm ref and mit ptr
43     revMapType revMap_; //map between mit ptr and edm ref
44     Int_t edmProductId_; //product id for consistency check
45     std::string brname_; //branch name of MIT objects
46 loizides 1.2 };
47     }
48    
49     //--------------------------------------------------------------------------------------------------
50     template <class EdmClass, class MitClass>
51     inline void mithep::AssociationMap<EdmClass,MitClass>::Add(EdmClass edmObj, MitClass mitObj)
52     {
53     fwdMap_[edmObj]=mitObj;
54     revMap_.insert(std::pair<MitClass, EdmClass>(mitObj,edmObj));
55     }
56    
57     //--------------------------------------------------------------------------------------------------
58     template <class EdmClass, class MitClass>
59     inline MitClass mithep::AssociationMap<EdmClass,MitClass>::GetMit(EdmClass edmObj) const
60     {
61     typename fwdMapType::const_iterator iter = fwdMap_.find(edmObj);
62    
63     if (iter != fwdMap_.end())
64     return iter->second;
65 bendavid 1.3 else throw edm::Exception(edm::errors::Configuration, "AssociationMap::GetMit()\n")
66 loizides 1.4 << "Error! EDM Object (" << typeid(edmObj).name()
67     << ") not found in AssociationMap (" << typeid(*this).name() << ")." << std::endl;
68 loizides 1.2 }
69 loizides 1.1
70 loizides 1.2 //--------------------------------------------------------------------------------------------------
71     template <class EdmClass, class MitClass>
72 loizides 1.4 inline EdmClass mithep::AssociationMap<EdmClass,MitClass>::GetEdm(MitClass mitObj) const
73 loizides 1.2 {
74     typename revMapType::const_iterator iter = revMap_.find(mitObj);
75     if (iter != revMap_.end())
76     return iter->second;
77 loizides 1.4 else throw edm::Exception(edm::errors::Configuration, "AssociationMap::GetEdm()\n")
78     << "Error! MITHEP Object (" << typeid(mitObj).name()
79     << ") not found in AssociationMap (" << typeid(*this).name() << ")." << std::endl;
80 loizides 1.1 }
81 sixie 1.5
82     //--------------------------------------------------------------------------------------------------
83     template <class EdmClass, class MitClass>
84     inline bool mithep::AssociationMap<EdmClass,MitClass>::HasMit(EdmClass edmObj) const
85     {
86     typename fwdMapType::const_iterator iter = fwdMap_.find(edmObj);
87    
88     if (iter != fwdMap_.end())
89     return true;
90     else
91     return false;
92     }
93 loizides 1.1 #endif