opengmpython.hxx

Go to the documentation of this file.

#ifndef OPENGM_PYTHON_EXPORT_TYPEDEFS_HXX
#define OPENGM_PYTHON_EXPORT_TYPEDEFS_HXX

#include <map>

#include <opengm/graphicalmodel/graphicalmodel.hxx>
#include <opengm/functions/potts.hxx>
#include <opengm/utilities/tribool.hxx>

#include "opengm/utilities/functors.hxx"
#include "opengm/functions/explicit_function.hxx"
#include "opengm/functions/absolute_difference.hxx"
#include "opengm/functions/potts.hxx"
#include "opengm/functions/pottsn.hxx"
#include "opengm/functions/pottsg.hxx"
#include "opengm/functions/squared_difference.hxx"
#include "opengm/functions/truncated_absolute_difference.hxx"
#include "opengm/functions/truncated_squared_difference.hxx"
#include "opengm/functions/sparsemarray.hxx"

#include "opengm/functions/learnable/lpotts.hxx"
#include "opengm/functions/learnable/lunary.hxx"

#include <opengm/python/opengmpython.hxx>
#include <opengm/python/converter.hxx>
#include <opengm/python/numpyview.hxx>
#include <opengm/python/pythonfunction.hxx>


#include <algorithm>
#include <vector>
#include <cmath>




namespace opengm{
namespace python{


   template<class V,class I,class O,class F>
   struct GmGen{
      typedef opengm::DiscreteSpace<I,I> SpaceType;
      typedef opengm::GraphicalModel<V,O,F,SpaceType> type;
   };
   template<class V,class I>
   struct ETLGen{
      typedef opengm::ExplicitFunction<V ,I,I> type;
   };

    template<class V,class I>
    struct FTLGen{

        typedef V ValueType;
        typedef I IndexType;
        typedef I LabelType;
        typedef opengm::ExplicitFunction                      <ValueType,IndexType,LabelType> PyExplicitFunction;
        typedef opengm::PottsFunction                         <ValueType,IndexType,LabelType> PyPottsFunction;
        typedef opengm::PottsNFunction                        <ValueType,IndexType,LabelType> PyPottsNFunction;
        typedef opengm::PottsGFunction                        <ValueType,IndexType,LabelType> PyPottsGFunction;
        typedef opengm::TruncatedAbsoluteDifferenceFunction   <ValueType,IndexType,LabelType> PyTruncatedAbsoluteDifferenceFunction;
        typedef opengm::TruncatedSquaredDifferenceFunction    <ValueType,IndexType,LabelType> PyTruncatedSquaredDifferenceFunction;
        typedef opengm::SparseFunction                        <ValueType,IndexType,LabelType> PySparseFunction; 
        typedef opengm::functions::learnable::LPotts          <ValueType,IndexType,LabelType> PyLPottsFunction;
        typedef opengm::functions::learnable::LUnary          <ValueType,IndexType,LabelType> PyLUnaryFunction;


        typedef typename opengm::meta::TypeListGenerator<
            PyExplicitFunction,
            PyPottsFunction,
            PyPottsNFunction,
            PyPottsGFunction,
            PyTruncatedAbsoluteDifferenceFunction,
            PyTruncatedSquaredDifferenceFunction,
            PySparseFunction,
            PyLPottsFunction,
            PyLUnaryFunction
        >::type type;
   };



   typedef double GmValueType;
   typedef opengm::UInt64Type GmIndexType;
   typedef GmIndexType GmLabelType;


   typedef opengm::learning::Weights<GmValueType> PyWeights;


   // different function types
   typedef opengm::ExplicitFunction                      <GmValueType,GmIndexType,GmLabelType> GmExplicitFunction;
   typedef opengm::PottsFunction                         <GmValueType,GmIndexType,GmLabelType> GmPottsFunction;
   typedef opengm::PottsNFunction                        <GmValueType,GmIndexType,GmLabelType> GmPottsNFunction;
   typedef opengm::PottsGFunction                        <GmValueType,GmIndexType,GmLabelType> GmPottsGFunction;
   typedef opengm::AbsoluteDifferenceFunction            <GmValueType,GmIndexType,GmLabelType> GmAbsoluteDifferenceFunction;
   typedef opengm::TruncatedAbsoluteDifferenceFunction   <GmValueType,GmIndexType,GmLabelType> GmTruncatedAbsoluteDifferenceFunction;
   typedef opengm::SquaredDifferenceFunction             <GmValueType,GmIndexType,GmLabelType> GmSquaredDifferenceFunction;
   typedef opengm::TruncatedSquaredDifferenceFunction    <GmValueType,GmIndexType,GmLabelType> GmTruncatedSquaredDifferenceFunction;
   typedef opengm::SparseFunction                        <GmValueType,GmIndexType,GmLabelType> GmSparseFunction; 
   typedef opengm::functions::learnable::LPotts          <GmValueType,GmIndexType,GmLabelType> PyLPottsFunction;
   typedef opengm::functions::learnable::LUnary          <GmValueType,GmIndexType,GmLabelType> PyLUnaryFunction;
   
   typedef std::vector<GmIndexType> IndexVectorType;
   typedef std::vector<IndexVectorType> IndexVectorVectorType;

   typedef GmGen<
      GmValueType,
      GmIndexType,
      opengm::Adder ,
      FTLGen<GmValueType,GmIndexType>::type
   >::type   GmAdder;

   typedef GmAdder::FactorType FactorGmAdder;
   typedef FactorGmAdder GmAdderFactor;

   typedef GmGen<
      GmValueType,
      GmIndexType,
      opengm::Multiplier ,
      FTLGen<GmValueType,GmIndexType>::type
   >::type   GmMultiplier;

   typedef GmMultiplier::FactorType FactorGmMultiplier;

   typedef FactorGmMultiplier GmMultiplierFactor;

   typedef opengm::IndependentFactor<GmValueType,GmIndexType,GmIndexType> GmIndependentFactor;

   namespace pyenums{
      enum AStarHeuristic{
         DEFAULT_HEURISTIC=0,
         FAST_HEURISTIC=1,
         STANDARD_HEURISTIC=2
      };
      enum IcmMoveType{
         SINGLE_VARIABLE=0,
         FACTOR=1
      };
      enum GibbsVariableProposal{
         RANDOM=0,
         CYCLIC=1
      };

      enum TrwsEnergyType{
         VIEW=0, TABLES=1, TL1=2, TL2=3
      };
   }

}
}

#endif   /* OPENGM_PYTHON_EXPORT_TYPEDEFS_HXX */