projectionRegister4MultiDim.h

Go to the documentation of this file.

#ifndef GUM_PROJECTION_REGISTER_4_MULTI_DIM_H
#define GUM_PROJECTION_REGISTER_4_MULTI_DIM_H

#include <agrum/core/hashTable.h>
#include <agrum/core/set.h>
#include <agrum/variables/discreteVariable.h>
#include <iostream>
#include <string>
#include <utility>

namespace gum {

  // the base object used by the projections
  template < typename GUM_SCALAR >
  class MultiDimImplementation;

  // ===========================================================================
  // ===                  GUM_MULTI_DIM_PROJECTION_REGISTER                  ===
  // ===========================================================================
  // clang-format off
  // clang-format on
  template < typename GUM_SCALAR >
  class ProjectionRegister4MultiDim {
    public:
    typedef MultiDimImplementation< GUM_SCALAR >* (*ProjectionPtr)(
       const MultiDimImplementation< GUM_SCALAR >*,
       const Set< const DiscreteVariable* >&);

    // ========================================================================
    // ========================================================================

    void insert(const std::string& projection_name,
                const std::string& type_multidim,
                ProjectionPtr      newFunction);

    void erase(const std::string& projection_name,
               const std::string& type_multidim);

    bool exists(const std::string& projection_name,
                const std::string& type_multidim) const;

    ProjectionPtr get(const std::string& projection_name,
                      const std::string& type_multidim) const;

    // ========================================================================
    // ========================================================================

    static ProjectionRegister4MultiDim& Register();


    private:
    // ========================================================================
    // ========================================================================

    ProjectionRegister4MultiDim();

    ProjectionRegister4MultiDim(const ProjectionRegister4MultiDim&);

    ~ProjectionRegister4MultiDim();


    typedef HashTable< std::string, ProjectionPtr > ProjectionSet;

    HashTable< std::string, ProjectionSet* > __set;
  };

  template < typename GUM_SCALAR >
  void registerProjection(
     const std::string& projection_name,
     const std::string& type_multidim,
     typename ProjectionRegister4MultiDim< GUM_SCALAR >::ProjectionPtr function);

  // a display operator for ProjectionPtrs

} /* namespace gum */

// always include the implementations
#include <agrum/multidim/utils/operators/projectionRegister4MultiDim_tpl.h>

#endif /* GUM_PROJECTION_REGISTER_MULTI_DIM_H */