TaskModel.h

Go to the documentation of this file.

#ifndef SPATIAL_TEMPORAL_LEARNING_WORLDLIB_MODEL_TASK_MODEL_H_
#define SPATIAL_TEMPORAL_LEARNING_WORLDLIB_MODEL_TASK_MODEL_H_

// worldlib
#include "PlacementModel.h"

// C++ Standard Library
#include <stdint.h>
#include <string>
#include <vector>

namespace rail
{
namespace spatial_temporal_learning
{
namespace worldlib
{
namespace model
{

class TaskModel
{
public:
  TaskModel(const uint32_t task_id = 0, const std::string &name = "");

  uint32_t getTaskID();

  void setTaskID(const uint32_t task_id);

  const std::string &getName() const;

  void setName(const std::string &name);

  const std::vector<PlacementModel> &getPlacementModels() const;

  std::vector<PlacementModel> &getPlacementModels();

  size_t getNumPlacementModels() const;

  const PlacementModel &getPlacementModel(const size_t index) const;

  PlacementModel &getPlacementModel(const size_t index);

  void addPlacementModel(const PlacementModel &placement_model);

  void removePlacementModel(const size_t index);

  void getUniqueItems(std::vector<world::Item> &items) const;

  void getUniqueSurfaces(std::vector<world::Surface> &surfaces) const;

private:
  uint32_t task_id_;
  std::string name_;
  /* The associated placement models */
  std::vector<PlacementModel> placement_models_;
};

}
}
}
}

#endif