TreeFiller/interface/AssociationMap.h

//--------------------------------------------------------------------------------------------------
// $Id: AssociationMap.h,v 1.7 2009/03/13 09:35:19 loizides Exp $
//
// Association Map
//
// Wrapper for std::map, used to resolve links during tree filling.
//
// Authors: J.Bendavid, C.Loizides
//--------------------------------------------------------------------------------------------------

#ifndef MITPROD_TREEFILLER_ASSOCIATIONMAP_H
#define MITPROD_TREEFILLER_ASSOCIATIONMAP_H

#include <map>
#include <TObject.h>
#include "FWCore/MessageLogger/interface/MessageLogger.h"

namespace mithep
{
  template <class EdmClass, class MitClass> class AssociationMap : public TObject
  { 
    typedef std::map<EdmClass, MitClass> fwdMapType;
    typedef std::map<MitClass, EdmClass> revMapType;
    
    public:
      AssociationMap() : edmProductId_(0) {}
      ~AssociationMap() {}
      
      void         Add(EdmClass edmObj, MitClass mitObj);
      EdmClass     GetEdm(MitClass mitObj)    const;
      Int_t        GetEdmProductId()          const { return edmProductId_;  }
      Int_t        GetEntries()               const { return fwdMap_.size(); }
      MitClass     GetMit(EdmClass edmObj)    const;
      const char  *GetBrName()                const { return brname_.c_str();           }
      bool         HasMit(EdmClass edmObj)    const;
      void         Reset()                          { fwdMap_.clear(); revMap_.clear(); }
      void         SetEdmProductId(Int_t id)        { edmProductId_ = id;               }
      void         SetBrName(const char *n)         { brname_ = n;                      }

    protected:
      fwdMapType   fwdMap_;       //map between edm ref and mit ptr 
      revMapType   revMap_;       //map between mit ptr and edm ref
      Int_t        edmProductId_; //product id for consistency check
      std::string  brname_;       //branch name of MIT objects 
  };
}

//--------------------------------------------------------------------------------------------------
template <class EdmClass, class MitClass>
inline void mithep::AssociationMap<EdmClass,MitClass>::Add(EdmClass edmObj, MitClass mitObj) 
{
  fwdMap_[edmObj]=mitObj;
  revMap_.insert(std::pair<MitClass, EdmClass>(mitObj,edmObj));
}

//--------------------------------------------------------------------------------------------------
template <class EdmClass, class MitClass>
inline MitClass mithep::AssociationMap<EdmClass,MitClass>::GetMit(EdmClass edmObj) const
{
  typename fwdMapType::const_iterator iter = fwdMap_.find(edmObj);

  if (iter != fwdMap_.end())
    return iter->second;
  else throw edm::Exception(edm::errors::Configuration, "AssociationMap::GetMit()\n")
         << "Error! EDM Object (" << typeid(edmObj).name() 
         << ") not found in AssociationMap (" << typeid(*this).name() << ")." << std::endl;
}
      
//--------------------------------------------------------------------------------------------------
template <class EdmClass, class MitClass>
inline EdmClass mithep::AssociationMap<EdmClass,MitClass>::GetEdm(MitClass mitObj) const
{
  typename revMapType::const_iterator iter = revMap_.find(mitObj);
  if (iter != revMap_.end())
    return iter->second;
  else throw edm::Exception(edm::errors::Configuration, "AssociationMap::GetEdm()\n")
         << "Error! MITHEP Object (" << typeid(mitObj).name() 
         << ") not found in AssociationMap (" << typeid(*this).name() << ")." << std::endl;
}

//--------------------------------------------------------------------------------------------------
template <class EdmClass, class MitClass>
inline bool mithep::AssociationMap<EdmClass,MitClass>::HasMit(EdmClass edmObj) const
{
  typename fwdMapType::const_iterator iter = fwdMap_.find(edmObj);

  if (iter != fwdMap_.end())
    return true;
  else 
    return false;
}
#endif
Revision:	1.8
Committed:	Sat Mar 14 06:04:59 2009 UTC (16 years, 1 month ago) by loizides
Content type:	text/plain
Branch:	MAIN
Changes since 1.7:	+4 -4 lines
Log Message:	Bugfix.
#	User	Rev	Content
1	loizides	1.1	//--------------------------------------------------------------------------------------------------
2	loizides	1.8	// $Id: AssociationMap.h,v 1.7 2009/03/13 09:35:19 loizides Exp $
3	loizides	1.1	//
4			// Association Map
5			//
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.8	void SetBrName(const char *n) { brname_ = n; }
39	loizides	1.1
40			protected:
41	loizides	1.7	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	loizides	1.2	};
46			}
47
48			//--------------------------------------------------------------------------------------------------
49			template <class EdmClass, class MitClass>
50			inline void mithep::AssociationMap<EdmClass,MitClass>::Add(EdmClass edmObj, MitClass mitObj)
51			{
52			fwdMap_[edmObj]=mitObj;
53			revMap_.insert(std::pair<MitClass, EdmClass>(mitObj,edmObj));
54			}
55
56			//--------------------------------------------------------------------------------------------------
57			template <class EdmClass, class MitClass>
58			inline MitClass mithep::AssociationMap<EdmClass,MitClass>::GetMit(EdmClass edmObj) const
59			{
60			typename fwdMapType::const_iterator iter = fwdMap_.find(edmObj);
61
62			if (iter != fwdMap_.end())
63			return iter->second;
64	bendavid	1.3	else throw edm::Exception(edm::errors::Configuration, "AssociationMap::GetMit()\n")
65	loizides	1.4	<< "Error! EDM Object (" << typeid(edmObj).name()
66			<< ") not found in AssociationMap (" << typeid(*this).name() << ")." << std::endl;
67	loizides	1.2	}
68	loizides	1.1
69	loizides	1.2	//--------------------------------------------------------------------------------------------------
70			template <class EdmClass, class MitClass>
71	loizides	1.4	inline EdmClass mithep::AssociationMap<EdmClass,MitClass>::GetEdm(MitClass mitObj) const
72	loizides	1.2	{
73			typename revMapType::const_iterator iter = revMap_.find(mitObj);
74			if (iter != revMap_.end())
75			return iter->second;
76	loizides	1.4	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	loizides	1.1	}
80	sixie	1.5
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	loizides	1.1	#endif