#ifndef _WORLDMAP_H
#define _WORLDMAP_H

#include <string>
#include <vector>

#include "Common.h"

using namespace std;

class WorldMap {
public:
   enum TerrainType {
      TERRAIN_NONE,
      TERRAIN_GRASS,
      TERRAIN_OCEAN,
      TERRAIN_ROCK
   };

   enum StructureType {
      STRUCTURE_NONE,
      STRUCTURE_BLUE_FLAG,
      STRUCTURE_RED_FLAG
   };

   enum ObjectType {
      OBJECT_NONE,
      OBJECT_BLUE_FLAG,
      OBJECT_RED_FLAG
   };

   class Object {
   public:
      unsigned int id;
      ObjectType type;
      POSITION pos;

      Object(unsigned int id, ObjectType type, int x, int y);
      Object(unsigned int id, ObjectType type, POSITION pos);

      ~Object();

      void serialize(char* buffer);
      void deserialize(char* buffer);
   };

   int width, height;
   vector<vector<TerrainType>*>* vctMap;
   vector<vector<StructureType>*>* vctStructures;
   vector<Object>* vctObjects; 

   WorldMap(int width, int height);

   ~WorldMap();

   TerrainType getElement(int x, int y);
   void setElement(int x, int y, TerrainType type);

   StructureType getStructure(int x, int y);
   void setStructure(int x, int y, StructureType type);
   POSITION getStructureLocation(StructureType type);

   vector<Object>* getObjects();
   vector<Object> getObjects(int x, int y);

   void addObject(ObjectType type, int x, int y);
   void updateObject(unsigned int id, WorldMap::ObjectType t, int x, int y);
   bool removeObject(unsigned int id);

   static WorldMap* createDefaultMap();
   static WorldMap* loadMapFromFile(string filename);
};

#endif