set_property_action.h

#ifndef SET_PROPERTY_ACTION_ 
#define SET_PROPERTY_ACTION_

#include <string>

#include "graph_filter.h"
#include "graph_action.h"
#include "graph_properties.h"

template<typename GraphType>
class SetPropertyAction : public GraphAction<GraphType> {
  public:
    typedef typename GraphType::node_value_type V_type;
    typedef typename GraphType::edge_value_type E_type;
    typedef typename GraphType::index_type_t index_type_t;
    typedef std::map<std::string, std::string> PropertyMap;
    typedef PropertyMap::iterator iterator;
    typedef PropertyMap::const_iterator const_iterator;

    SetPropertyAction(GraphProperties<GraphType>& properties,
                      GraphFilter<GraphType>* property_filter) :
                           GraphAction<GraphType>(property_filter), 
                           properties_(properties)
                           {
    }

    SetPropertyAction(SetPropertyAction& copy) : 
          GraphAction<GraphType>(copy), properties_(copy.properties_), property_values_(copy.property_values_) {
    }

   virtual bool operator() (E_type edge) {
     if (this->filter_->enabled() && this->filter_->edges()) {
       if (this->run_filter(edge)) {
         for (iterator it = begin(); it != end(); ++it) {
           properties_.set_property(edge, it->first, it->second);
         }
       }
     }

     return true;
   }

    virtual bool operator() (V_type m) {
      if(this->filter_->enabled() && this->filter_->nodes()) 
        if(this->run_filter(m)) 
          for(iterator it = begin(); it != end(); ++it)
            properties_.set_property(m, it->first, it->second);

      return true;
    }

    template<typename P1, typename P2>
    bool operator() (std::pair<P1, P2>& m) {
      return  operator()(m.second);
    }

    iterator begin() { return property_values_.begin(); }

    iterator end() { return property_values_.end(); }
    
    void set_property(std::string name, std::string value) {
      property_values_[name] = value;
    } 

    std::string get_property(std::string name) {
      return property_values_[name];
    }

  private:
    GraphProperties<GraphType>& properties_;
    PropertyMap property_values_;
};

#endif