policy.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_policy_h_
#define _libint2_src_bin_libint_policy_h_
 
#include <vector>
#include <smart_ptr.h>
#include <traits.h>
 
#include <boost/type_traits/is_same.hpp>
 
 
namespace libint2 {
 
  template <class T, bool exists>
    struct ExistsDefaultSubobjAllocator;
 
  template <class T>
    struct ExistsDefaultSubobjAllocator<T,true>{
      typedef T obj_type;
      typedef typename TypeTraits<obj_type>::StorageType obj_stype;
      typedef typename TypeTraits<obj_type>::StorageType subobj_stype;
      typedef typename obj_type::iter_type subobj_type;
 
      static void default_init_subobj(const obj_stype& obj, std::vector<subobj_stype>& subobj)
      {
        subobj.push_back(obj);
      }
      static void default_dealloc_subobj(std::vector<subobj_stype>& subobj)
      {
      }
    };
 
 
  template < class T>
    struct StdLibintTDPolicy {
      typedef T obj_type;
      typedef typename obj_type::iter_type subobj_type;
      typedef typename TypeTraits<obj_type>::StorageType obj_stype;
      typedef typename TypeTraits<subobj_type>::StorageType subobj_stype;
 
      static void init_subobj(const obj_stype& obj, std::vector<subobj_stype>& subobj)
      {
        // If types are not the same then this function should not be used -- user must provide a specialization
        ExistsDefaultSubobjAllocator< T, boost::is_same<obj_type,subobj_type>::value >::default_init_subobj(obj,subobj);
      }
      static void dealloc_subobj(std::vector<subobj_stype>& subobj)
      {
        // If types are not the same then this function should not be used -- user must provide a specialization
        ExistsDefaultSubobjAllocator< T, boost::is_same<obj_type,subobj_type>::value >::default_dealloc_subobj(subobj);
      }
    };
 
 
  class StdLibintTIPolicy {
 
    static unsigned int max_set_size_to_unroll_;
 
  public:
 
    StdLibintTIPolicy() {}
 
    virtual void set_max_set_size_to_unroll(unsigned int i)
    {
      max_set_size_to_unroll_ = i;
    }
 
    virtual unsigned int max_set_size_to_unroll() const
    {
      return max_set_size_to_unroll_;
    }
 
  };
 
 
#if CXX_ALLOWS_DEFPARAMTEMPLATE_AS_TEMPTEMPPARAM
  template <class T, class TIPol = StdLibintTIPolicy, template <class> class TDPol = StdLibintTDPolicy>
    class Policy : public TDPol<T>, public TIPol
    {
#else
#define TDPol StdLibintTDPolicy
#define TIPol StdLibintTIPolicy
  template <class T>
    class Policy : public TDPol<T>, public TIPol
    {
#endif
    public:
      typedef typename TDPol<T>::obj_stype obj_stype;
      typedef typename TDPol<T>::subobj_stype subobj_stype;
 
    /*
      typedef typename TDPol<T>::obj_type obj_type;
      typedef typename obj_type::iter_type subobj_type;
 
      static void init_subobj(const SafePtr<obj_type>& obj, const vector< SafePtr<subobj_type> >& subobj)
      {
        TDPol<T>::init_subobj(obj,subobj);
      }
 
      static void dealloc_subobj(const vector< SafePtr<subobj_type> >& subobj)
      {
        TDPol<T>::dealloc_subobj(subobj);
      }
     */
 
     private:
     bool can_unroll_intset(const SafePtr<T>& iset)
     {
       return iset->set_size() <= TIPol::max_set_size_to_unroll();
     }
  };
 
 
};
 
 
#endif