#ifndef _WORLDMAP_H
#define _WORLDMAP_H

#include <string>
#include <vector>

#include "Common.h"

using namespace std;

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 WorldMap {
public:
   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();

   void createObjectsFromStructures();

   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, ObjectType t, int x, int y);
   bool removeObject(unsigned int id);

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

#endif