libopengm-dev 2.3.6-2 / usr / include / opengm / unittests / blackboxtests / blackboxtestgrid.hxx

#pragma once
#ifndef OPENGM_TEST_INFERENCE_BLACKBOXTEST_GRID_HXX
#define OPENGM_TEST_INFERENCE_BLACKBOXTEST_GRID_HXX

#include <vector>
#include <iostream>
#include <sstream>
#include <string>

#include <opengm/opengm.hxx>
#include <opengm/unittests/test.hxx>
#include <opengm/graphicalmodel/modelgenerators/syntheticmodelgenerator.hxx>
#include <opengm/unittests/blackboxtests/blackboxtestbase.hxx>

/// \cond HIDDEN_SYMBOLS

namespace opengm {

   template<class GM> class BlackBoxTestBase;

   template<class GM>
   class BlackBoxTestGrid : public BlackBoxTestBase<GM>
   {
   public:
      enum BlackBoxFunction
      {
         RANDOM, POTTS, IPOTTS, L1, RPOTTS
      };
      const size_t height_;
      const size_t width_;
      const size_t numVar_;
      const size_t numStates_;
      const bool varStates_;
      const bool withUnary_;
      const BlackBoxFunction function_;
      const BlackBoxBehaviour behaviour_;
      const size_t numTests_;
      const size_t modelIdOffset_;

      BlackBoxTestGrid(const size_t height, const size_t width, const size_t numStates, const bool varStates, const bool withUnary, const BlackBoxFunction f, const BlackBoxBehaviour b, const size_t numTests, const size_t modelIdOffset=0);
      virtual std::string infoText();
      virtual GM getModel(size_t);
      virtual size_t numberOfTests();
      virtual BlackBoxBehaviour behaviour();

   private:
      std::string functionString(BlackBoxFunction f);
   };

   template<class GM>
   size_t BlackBoxTestGrid<GM>::numberOfTests()
   {
      return numTests_;
   }

   template<class GM>
   BlackBoxBehaviour BlackBoxTestGrid<GM>::behaviour()
   {
      return behaviour_;
   }

   template<class GM>
   BlackBoxTestGrid<GM>::BlackBoxTestGrid(
      const size_t height, const size_t width, const size_t numStates,
      const bool varStates, const bool withUnary,
      const BlackBoxFunction f, const BlackBoxBehaviour b,
      const size_t numTests, const size_t modelIdOffset)
      : height_(height), width_(width), numVar_(height * width), numStates_(numStates),
        varStates_(varStates), withUnary_(withUnary),
        function_(f), behaviour_(b),
        numTests_(numTests), modelIdOffset_(modelIdOffset)
   {
   }

   template<class GM>
   std::string BlackBoxTestGrid<GM>::infoText()
   {
      std::string s = varStates_ ? "<" : "";
      std::string u = withUnary_ ? " with unary " : " without unary ";
      std::stringstream oss;
      oss << "    - 2nd order grid model with " << numVar_ << " variables and " << s << numStates_ << " states ";
      oss << "(functionType = " << functionString(function_) << u << ")";
      return oss.str();
   }

   template<class GM>
   GM BlackBoxTestGrid<GM>::getModel(size_t id)
   {
      typedef opengm::SyntheticModelGenerator2<GM> ModelGeneratorType;

      ModelGeneratorType modelGenerator;
      typename ModelGeneratorType::Parameter modelGeneratorPara;

      if(withUnary_) {
         modelGeneratorPara.functionTypes_[0] = ModelGeneratorType::URANDOM;
         modelGeneratorPara.functionParameters_[1].resize(2);
         modelGeneratorPara.functionParameters_[1][0] = 1;
         modelGeneratorPara.functionParameters_[1][1] = 2;
      }else{
         modelGeneratorPara.functionTypes_[0] = ModelGeneratorType::EMPTY;
      }

      switch (function_) {
      case RANDOM:
         modelGeneratorPara.functionParameters_[1][0] = 1;
         modelGeneratorPara.functionTypes_[1] = ModelGeneratorType::URANDOM;
         modelGeneratorPara.functionParameters_[1].resize(2);
         modelGeneratorPara.functionParameters_[1][0] = 1;
         modelGeneratorPara.functionParameters_[1][1] = 2;
         break;
      case POTTS:
         modelGeneratorPara.functionTypes_[1] = ModelGeneratorType::GPOTTS;
         modelGeneratorPara.functionParameters_[1].resize(1);
         modelGeneratorPara.functionParameters_[1][0] = 3;
         modelGeneratorPara.functionParameters_[1][0] = 3;
         break;
      case IPOTTS:
         modelGeneratorPara.functionTypes_[1] = ModelGeneratorType::GPOTTS;
         modelGeneratorPara.functionParameters_[1].resize(1);
         modelGeneratorPara.functionParameters_[1][0] = -3;
         modelGeneratorPara.functionParameters_[1][0] = -3;
         break;
      case L1:
         modelGeneratorPara.functionTypes_[1] = ModelGeneratorType::L1;
         modelGeneratorPara.functionParameters_[1].resize(1);
         modelGeneratorPara.functionParameters_[1][0] = 1;
         modelGeneratorPara.functionParameters_[1][0] = 1;
         break; 
      case RPOTTS:
         modelGeneratorPara.functionTypes_[1] = ModelGeneratorType::RPOTTS;
         modelGeneratorPara.functionParameters_[1].resize(1);
         modelGeneratorPara.functionParameters_[1][0] = 3;
         break;
      }
      modelGeneratorPara.randomNumberOfStates_ = varStates_;
      return modelGenerator.buildGrid(id+modelIdOffset_, width_, height_, numStates_, modelGeneratorPara);
   }

   template<class GM>
   std::string BlackBoxTestGrid<GM>::functionString(BlackBoxFunction f)
   {
      switch (f) {
      case RANDOM:
         return "random";
      case POTTS:
         return "potts";
      case IPOTTS:
         return "negative potts"; 
      case L1:
         return "L1"; 
      case RPOTTS:
         return "RPOTTS";
      default:
         return "";
      };
      return "";
   }
}

/// \endcond

#endif // #ifndef OPENGM_TEST_INFERENCE_BLACKBOXTEST_GRID_HXX