WorldState.hh

#ifndef GRAIL_WORLD_STATE_H
#define GRAIL_WORLD_STATE_H
 
#include "WorldObject.hh"
#include "../Domain.hh"
#include <vector>
#include <sstream>
 
namespace grail
{
 namespace planning
 {
 class WorldState : public ParametrizedObject
 {
 public:
 WorldState(MemoryPool& memory);
 WorldState(const std::shared_ptr<Domain>& domain, const std::vector<WorldObject>& worldObjects, MemoryPool& memory);
 WorldState(const WorldState& other);
 WorldState(WorldState&& other) = default;
 virtual ~WorldState() = default;
 
 WorldState& operator=(const WorldState& other);
 WorldState& operator=(WorldState&& other) = default;
 
 bool operator==(const WorldState& other) const;
 
 std::vector<class Action> GetLegalActions() const;
 std::vector<std::vector<unsigned int>> GenerateParameterLists(const std::vector<std::vector<unsigned int>>& precedingParameterLists, unsigned int searchedTypeIndex) const;
 
 const WorldObject* GetObjectByIndex(unsigned int index) const;
 WorldObject* GetObjectByIndex(unsigned int index);
 const std::map<unsigned int, WorldObject>& GetObjectMap() const;
 std::shared_ptr<Domain> GetDomain() const;
 
 WorldObject& AddObject(const std::string& typeName);
 
 void RemoveObject(unsigned int objectIndex);
 void AddQueuedObjects();
 void RemoveQueuedObjects();
 
 std::vector<const WorldObject*> GetObjectsOfType(const std::string& typeName) const;
 
 std::vector<WorldObject*> GetObjectsOfType(const std::string& typeName);
 
 int CountObjectsOfType(const std::string& typeName) const;
 
 private:
 std::shared_ptr<Domain> domain = nullptr;
 std::map<unsigned int, WorldObject> objects{};
 std::vector<WorldObject> objectsToAdd{};
 std::vector<unsigned int> indicesToRemove{};
 unsigned int nextIndex = 0;
 };
 }
}
#endif //GRAIL_WORLD_STATE_H