Sacado_TemplateManager.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//                           Sacado Package
//
// Copyright 2006 NTESS and the Sacado contributors.
// SPDX-License-Identifier: LGPL-2.1-or-later
// *****************************************************************************
// @HEADER

#ifndef SACADO_TEMPLATEMANAGER_HPP
#define SACADO_TEMPLATEMANAGER_HPP

#include <vector>
#include <typeinfo>

#include "Teuchos_RCP.hpp"

#include "Sacado_mpl_size.hpp"
#include "Sacado_mpl_find.hpp"
#include "Sacado_mpl_for_each.hpp"
#include "Sacado_mpl_apply.hpp"

#include "Sacado_TemplateIterator.hpp"

namespace Sacado {


  template <typename TypeSeq, typename BaseT, typename ObjectT>
  class TemplateManager {

    struct type_info_less {
      bool operator() (const std::type_info* a, const std::type_info* b) {
        return a->before(*b);
      }
    };

    template <typename BuilderOpT>
    struct BuildObject {
      std::vector< Teuchos::RCP<BaseT> >* objects;
      const BuilderOpT& builder;
      BuildObject(std::vector< Teuchos::RCP<BaseT> >& objects_,
                  const BuilderOpT& builder_) :
        objects(&objects_), builder(builder_) {}
      template <typename T> void operator()(T) const {
        int idx = mpl::find<TypeSeq,T>::value;
        (*objects)[idx] = builder.template build<T>();
      }
    };

  public:

    typedef TemplateIterator<BaseT> iterator;

    typedef ConstTemplateIterator<BaseT> const_iterator;

    struct DefaultBuilderOp {

      template<class ScalarT>
      Teuchos::RCP<BaseT> build() const {
        typedef typename Sacado::mpl::apply<ObjectT,ScalarT>::type type;
        return Teuchos::rcp( dynamic_cast<BaseT*>( new type ) );
      }

    };

    TemplateManager();

    ~TemplateManager();

    template <typename BuilderOpT>
    void buildObjects(const BuilderOpT& builder);

    void buildObjects();

    template<typename ScalarT>
    Teuchos::RCP<BaseT> getAsBase();

    template<typename ScalarT>
    Teuchos::RCP<const BaseT> getAsBase() const;

    template<typename ScalarT>
    Teuchos::RCP< typename Sacado::mpl::apply<ObjectT,ScalarT>::type > getAsObject();

    template<typename ScalarT>
    Teuchos::RCP< const typename Sacado::mpl::apply<ObjectT,ScalarT>::type > getAsObject() const;

    typename Sacado::TemplateManager<TypeSeq,BaseT,ObjectT>::iterator begin();

    typename Sacado::TemplateManager<TypeSeq,BaseT,ObjectT>::const_iterator 
    begin() const;

    typename Sacado::TemplateManager<TypeSeq,BaseT,ObjectT>::iterator end();

    typename Sacado::TemplateManager<TypeSeq,BaseT,ObjectT>::const_iterator 
    end() const;

  private:

    std::vector< Teuchos::RCP<BaseT> > objects;

  };

}

// Include template definitions
#include "Sacado_TemplateManagerImp.hpp"

#endif