| 4 |
|
// Association Map |
| 5 |
|
// |
| 6 |
|
// Wrapper for std::map, used to resolve links during tree filling. |
| 7 |
– |
// This class needs work! CL. |
| 7 |
|
// |
| 8 |
|
// Authors: J.Bendavid, C.Loizides |
| 9 |
|
//-------------------------------------------------------------------------------------------------- |
| 10 |
|
|
| 11 |
< |
#ifndef TREEFILLER_ASSOCIATIONMAP_H |
| 12 |
< |
#define TREEFILLER_ASSOCIATIONMAP_H |
| 11 |
> |
#ifndef MITPROD_TREEFILLER_ASSOCIATIONMAP_H |
| 12 |
> |
#define MITPROD_TREEFILLER_ASSOCIATIONMAP_H |
| 13 |
|
|
| 14 |
|
#include <map> |
| 15 |
|
#include <TObject.h> |
| 26 |
|
AssociationMap() : edmProductId_(0) {} |
| 27 |
|
~AssociationMap() {} |
| 28 |
|
|
| 29 |
< |
void Add(EdmClass edmObj, MitClass mitObj); |
| 30 |
< |
EdmClass GetEdmRef(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 |
< |
void Reset() { fwdMap_.clear(); revMap_.clear(); } |
| 35 |
< |
void SetEdmProductId(Int_t id) { edmProductId_ = id; } |
| 29 |
> |
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 |
> |
const char *GetBrName() const { return brname_.c_str(); } |
| 35 |
> |
bool HasMit(EdmClass edmObj) const; |
| 36 |
> |
void Reset() { fwdMap_.clear(); revMap_.clear(); } |
| 37 |
> |
void SetEdmProductId(Int_t id) { edmProductId_ = id; } |
| 38 |
> |
void SetBrName(const char *n) { brname_ = n; } |
| 39 |
|
|
| 40 |
|
protected: |
| 41 |
< |
fwdMapType fwdMap_; //map between edm ref and mit ptr |
| 42 |
< |
revMapType revMap_; //map between mit ptr and edm ref |
| 43 |
< |
Int_t edmProductId_; //product id for consistency check |
| 41 |
> |
fwdMapType fwdMap_; //map between edm ref and mit ptr |
| 42 |
> |
revMapType revMap_; //map between mit ptr and edm ref |
| 43 |
> |
Int_t edmProductId_; //product id for consistency check |
| 44 |
> |
std::string brname_; //branch name of MIT objects |
| 45 |
|
}; |
| 46 |
|
} |
| 47 |
|
|
| 62 |
|
if (iter != fwdMap_.end()) |
| 63 |
|
return iter->second; |
| 64 |
|
else throw edm::Exception(edm::errors::Configuration, "AssociationMap::GetMit()\n") |
| 65 |
< |
<< "Error! Edm Object not found in AssociationMap." << std::endl; |
| 65 |
> |
<< "Error! EDM Object (" << typeid(edmObj).name() |
| 66 |
> |
<< ") not found in AssociationMap (" << typeid(*this).name() << ")." << std::endl; |
| 67 |
|
} |
| 68 |
|
|
| 69 |
|
//-------------------------------------------------------------------------------------------------- |
| 70 |
|
template <class EdmClass, class MitClass> |
| 71 |
< |
inline EdmClass mithep::AssociationMap<EdmClass,MitClass>::GetEdmRef(MitClass mitObj) const |
| 71 |
> |
inline EdmClass mithep::AssociationMap<EdmClass,MitClass>::GetEdm(MitClass mitObj) const |
| 72 |
|
{ |
| 73 |
|
typename revMapType::const_iterator iter = revMap_.find(mitObj); |
| 74 |
|
if (iter != revMap_.end()) |
| 75 |
|
return iter->second; |
| 76 |
< |
else throw edm::Exception(edm::errors::Configuration, "AssociationMap::GetEdmRef()\n") |
| 77 |
< |
<< "Error! mithep Object not found in AssociationMap." << std::endl; |
| 76 |
> |
else throw edm::Exception(edm::errors::Configuration, "AssociationMap::GetEdm()\n") |
| 77 |
> |
<< "Error! MITHEP Object (" << typeid(mitObj).name() |
| 78 |
> |
<< ") not found in AssociationMap (" << typeid(*this).name() << ")." << std::endl; |
| 79 |
> |
} |
| 80 |
> |
|
| 81 |
> |
//-------------------------------------------------------------------------------------------------- |
| 82 |
> |
template <class EdmClass, class MitClass> |
| 83 |
> |
inline bool mithep::AssociationMap<EdmClass,MitClass>::HasMit(EdmClass edmObj) const |
| 84 |
> |
{ |
| 85 |
> |
typename fwdMapType::const_iterator iter = fwdMap_.find(edmObj); |
| 86 |
> |
|
| 87 |
> |
if (iter != fwdMap_.end()) |
| 88 |
> |
return true; |
| 89 |
> |
else |
| 90 |
> |
return false; |
| 91 |
|
} |
| 92 |
|
#endif |
