LCRelationNavigator.h

// -*- C++ -*-
#ifndef EVENT_LCRELATIONNAVIGATORIMPL_H
#define EVENT_LCRELATIONNAVIGATORIMPL_H 1

#include <vector>
#include <map>
#include "EVENT/LCObject.h"
#include "EVENT/LCCollection.h"
#include "LCIOSTLTypes.h"


namespace UTIL {

  class LCRelationNavigator {
    
    typedef std::map< EVENT::LCObject* , std::pair< EVENT::LCObjectVec , EVENT::FloatVec > > RelMap ; 

  public: 

                                          
    LCRelationNavigator(const std::string &fromType, const std::string &toType) :
      _from( fromType ),
      _to( toType )  { /* nop */; }
    
    LCRelationNavigator( const EVENT::LCCollection* col ) ;
    
    virtual ~LCRelationNavigator() { /* nop */; }
    
    virtual const std::string & getFromType() const ;
    
    virtual const std::string & getToType() const ;
    
    virtual const EVENT::LCObjectVec & getRelatedToObjects(EVENT::LCObject * from) const ;

    virtual const EVENT::LCObjectVec & getRelatedFromObjects(EVENT::LCObject * to) const ;

    virtual const EVENT::FloatVec & getRelatedToWeights(EVENT::LCObject * from) const ;

    virtual const EVENT::FloatVec & getRelatedFromWeights(EVENT::LCObject * to) const ;

    virtual void addRelation(EVENT::LCObject * from, EVENT::LCObject * to, float weight = 1.0) ;
    
    virtual void removeRelation(EVENT::LCObject * from, EVENT::LCObject * to) ;

    virtual EVENT::LCCollection * createLCCollection() ;

  protected:

    LCRelationNavigator() ;

    virtual void initialize( const EVENT::LCCollection* col ) ;

    void removeRelation(EVENT::LCObject * from, EVENT::LCObject * to, RelMap& map ) ;
    void addRelation(EVENT::LCObject * from, EVENT::LCObject * to, float weight, RelMap& map) ;

    mutable RelMap _map ;
    mutable RelMap _rMap ;
    std::string _from ;
    std::string _to ;
    

}; // class
} // namespace EVENT
#endif /* ifndef EVENT_LCRELATIONNAVIGATOR_H */