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