singl_stack.h

/*
 *  Copyright (C) 2004-2024 Edward F. Valeev
 *
 *  This file is part of Libint compiler.
 *
 *  Libint compiler 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 compiler 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 compiler.  If not, see <http://www.gnu.org/licenses/>.
 *
 */
 
#ifndef _libint2_src_bin_libint_singlstack_h_
#define _libint2_src_bin_libint_singlstack_h_
 
#include <hashable.h>
#include <key.h>
#include <purgeable.h>
#include <smart_ptr.h>
 
#include <iostream>
#include <map>
 
namespace libint2 {
 
class RecurrenceRelation;
 
template <class T, class KeyType>
class SingletonStack : public PurgeableStack<T> {
 public:
  typedef KeyType key_type;
  typedef std::shared_ptr<T> data_type;
  typedef KeyTypes::InstanceID InstanceID;
  typedef std::pair<InstanceID, std::shared_ptr<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 std::shared_ptr<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), std::shared_ptr<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), std::shared_ptr<T>()));
    for (auto& i : map_) {
      if (i.second.first == hashed_key) return i.second;
    }
    return null_value;
  }
 
  void remove(const std::shared_ptr<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_;
};
 
};  // namespace libint2
 
#endif