--- UserCode/MitProd/TreeFiller/interface/AssociationMap.h	2008/07/01 14:39:28	1.1
+++ UserCode/MitProd/TreeFiller/interface/AssociationMap.h	2012/05/05 16:49:58	1.10
@@ -1,70 +1,95 @@
 //--------------------------------------------------------------------------------------------------
-// $Id: AssociationMap.h,v 1.1 2008/07/01 14:39:28 loizides Exp $
+// $Id: AssociationMap.h,v 1.10 2012/05/05 16:49:58 paus Exp $
 //
-// Association Map
+// AssociationMap
 //
-// wrapper for std::map, used to resolve links during tree filling
+// Wrapper for std::map, used to resolve links during tree filling.
 //
-// Authors: J.Bendavid
+// Authors: C.Paus, J.Bendavid, C.Loizides
 //--------------------------------------------------------------------------------------------------
 
-#ifndef TREEFILLER_ASSOCIATIONMAP_H
-#define TREEFILLER_ASSOCIATIONMAP_H
+#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
   { 
+  public:
     typedef std::map<EdmClass, MitClass> fwdMapType;
     typedef std::map<MitClass, EdmClass> revMapType;
     
-    public:
-      AssociationMap() : edmProductId_(0) {}
-      ~AssociationMap() {}
-      
-      void Add(EdmClass edmObj, MitClass mitObj) {
-        fwdMap[edmObj]=mitObj;
-	//revMap[mitObj]=edmObj;
-	revMap.insert(std::pair<MitClass, EdmClass>(mitObj,edmObj));
-      }
-      
-      MitClass GetMit(EdmClass edmObj) const {
-	//MitClass iter = fwdMap.find(edmObj);
-	//fwdMapType::iterator iter;
-	//fwdIter = fwdMap.find(edmObj);
-	//if ( iter != fwdMap.end() )
-	//	return iter->second;
-	//else return 0;
-	return fwdMap.find(edmObj)->second;
-      }
-      
-       EdmClass GetEdmRef(MitClass mitObj) const {
-//         //EdmClass iter = revMap.find(mitObj);
-// // 	revMapType::iterator iter = revMap.find(mitObj);
-// // 	if ( iter != revMap.end() )
-// // 		return iter->second;
-// // 	else return 0;
- 	return revMap.find(mitObj)->second;
-       }
-      
-      Int_t GetEntries() { return fwdMap.size(); }
-      
-      void Reset() {
-      	fwdMap.clear();
-	revMap.clear();
-      }
-      
-	Int_t GetEdmProductId() const { return edmProductId_; }
-	void  SetEdmProductId(Int_t id) { edmProductId_ = id; }
+    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 std::string &n)  { brname_ = n;                      }
+    void              SetBrName(const char *n)         { brname_ = n;                      }
+    const fwdMapType &FwdMap()                   const { return fwdMap_;                   }
+    const revMapType &RevMap()                   const { return revMap_;                   }
+    
+  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);
 
-    protected:
-	fwdMapType fwdMap;
-	revMapType revMap;
-	Int_t edmProductId_;
+  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