singl_stack.h

/*
 *  Copyright (C) 2004-2021 Edward F. Valeev
 *
 *  This file is part of Libint.
 *
 *  Libint is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  Libint is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with Libint.  If not, see <http://www.gnu.org/licenses/>.
 *
 */
 
#ifndef _libint2_src_bin_libint_singlstack_h_
#define _libint2_src_bin_libint_singlstack_h_
 
#include <map>
#include <iostream>
#include <smart_ptr.h>
#include <key.h>
#include <hashable.h>
#include <purgeable.h>
 
namespace libint2 {
 
  class RecurrenceRelation;
 
  template <class T, class KeyType>
  class SingletonStack : public PurgeableStack<T>
  {
  public:
    typedef KeyType key_type;
    typedef SafePtr<T> data_type;
    typedef KeyTypes::InstanceID InstanceID;
    typedef std::pair<InstanceID,SafePtr<T> > value_type;
    typedef std::map<key_type,value_type> map_type;
    typedef typename map_type::iterator iter_type;
    typedef typename map_type::const_iterator citer_type;
    typedef typename KeyTraits<key_type>::ReturnType key_return_type;
    typedef key_return_type (T::* HashingFunction)() const;
    typedef typename PurgeableStack<T>::PurgingPolicy PurgingPolicy;
 
 
    SingletonStack(HashingFunction callback) :
      map_(), callback_(callback), next_instance_(0)
    {
      if (PurgingPolicy::purgeable()) { // if this stack contains objects that can be purged, add to the registry
        PurgeableStacks::Instance()->register_stack(this);
      }
    }
 
    virtual ~SingletonStack() {}
 
    const value_type& find(const SafePtr<T>& obj) {
      key_type key = ((obj.get())->*callback_)();
 
      typedef typename map_type::iterator miter;
      miter pos = map_.find(key);
      if (pos != map_.end()) {
#if DEBUG || LOCAL_DEBUG
        std::cout << "SingletonStack::find -- " << obj->label() << " already found" << std::endl;
#endif
        return (*pos).second;
      }
      else {
        value_type result(next_instance_++,obj);
        map_[key] = result;
#if DEBUG || LOCAL_DEBUG
        std::cout << "SingletonStack::find -- " << obj->label() << " is new (instid_ = " << next_instance_-1 << ")" << std::endl;
#endif
        return map_[key];
      }
    }
 
    const value_type& find(const key_type& key) {
      static value_type null_value(make_pair(InstanceID(0),SafePtr<T>()));
      typedef typename map_type::iterator miter;
      miter pos = map_.find(key);
      if (pos != map_.end()) {
        return (*pos).second;
      }
      else {
        return null_value;
      }
    }
 
    const value_type& find_hashed(const InstanceID& hashed_key) const {
      static value_type null_value(make_pair(InstanceID(0),SafePtr<T>()));
      for(auto& i: map_) {
        if (i.second.first == hashed_key)
          return i.second;
      }
      return null_value;
    }
 
    void remove(const SafePtr<T>& obj) {
      key_type key = ((obj.get())->*callback_)();
 
      typedef typename map_type::iterator miter;
      miter pos = map_.find(key);
      if (pos != map_.end()) {
        map_.erase(pos);
#if DEBUG || LOCAL_DEBUG
        std::cout << "Removed from stack " << obj->label() << std::endl;
#endif
      }
    }
 
    citer_type begin() const { return map_.begin(); }
    citer_type end() const { return map_.end(); }
 
    // Implementation of PurgeableStack::purge()
    void purge() override {
      for(iter_type i = map_.begin(); i!=map_.end();) {
        const T* v = i->second.second.get();
        if (PurgingPolicy::purge(v))
          // map::erase invalidates the iterator, increment it beforehand
          map_.erase(i++);
        else
          ++i;
      }
    }
 
  private:
    map_type map_;
    HashingFunction callback_;
    InstanceID next_instance_;
  };
 
};
 
#endif