#ifndef UNIT_H
#define UNIT_H

#include <vector>
#include <util.h>
#include <quaternion.h>
#include <lineRenderer.h>

#include <solUtil.h>

#include "gameObject.h"
#include "buildableUnit.h"
#include "fxManager.h"

namespace sf{
	class SoundBuffer;
	class Sound;
}

namespace vb01{
	class Mesh;
	class Quad;
	class Node;
	class Light;
	class Camera;
}

namespace battleship{
    class Player;
	class Weapon;
	class Unit;
	class Vehicle;
	class Structure;
	class Factory;
	class Cruiser;
	class Engineer;
	class PointDefense;
	class Projectile;
	class GameObject;
    
    struct Order {
        enum class TYPE {ATTACK, BUILD, MOVE, GARRISON, EJECT, PATROL, LAUNCH, SUPPLY, LOAD, UNLOAD, HACK};
			struct Target{
				GameObject *unit = nullptr;
				vb01::Vector3 pos;

				Target(){}
				Target(GameObject *u, vb01::Vector3 p = vb01::Vector3::VEC_ZERO) : unit(u), pos(p){}
			};

        TYPE type;
		int lineId = -1;
		bool playerAssigned = true;
		vb01::Vector3 direction;
        std::vector<Target> targets;

		Order(){}
		Order(TYPE t, std::vector<Target> targ, vb01::Vector3 dir = vb01::Vector3::VEC_ZERO, int lid = -1, bool pa = true) :
		   	type(t), 
			playerAssigned(pa), 
			lineId(lid), 
			targets(targ), 
			direction(dir){}
    };

    enum class MoveDir {LEFT, UP, FORW};
    enum class Corner {FRONT_LEFT, FRONT_RIGHT, REAR_LEFT, REAR_RIGHT};
    enum class UnitType {UNDERWATER, SEA_LEVEL, HOVER, LAND, AIR, NONE = -1};
    enum class UnitClass {
		WAR_MECH,
	   	TANK,
	   	ARTILLERY,
	   	ENGINEER,
	   	TRANSPORT,
	   	RESOURCE_ROVER,
	   	CRUISER,
	   	CARRIER,
	   	SUBMARINE,
		ICEBREAKER,
		FREEZER,
		EMP_BOAT,
	   	LAND_FACTORY,
	   	NAVAL_FACTORY,
	   	TRADE_CENTER,
	   	LAB,
	   	POINT_DEFENSE,
	   	EXTRACTOR,
	   	REFINERY,
		FORT,
		ICE_SHEET
	};
    
    class Unit : public GameObject{
    public:
		struct GarrisonSlot{
			Vehicle *vehicle = nullptr;
			int category;
			vb01::Node *background, *foreground;
			vb01::Vector2 offset;

			GarrisonSlot(vb01::Node *bg, vb01::Node *fg, vb01::Vector2 off, int cat, Vehicle *v = nullptr) : background(bg), foreground(fg), offset(off), category(cat), vehicle(v){}
		};

		enum class State {CHASE, STAND_GROUND, HOLD_FIRE};
		enum class Condition{ABLE, FROZEN, EM_JAMMED};

        Unit(Player*, int, vb01::Vector3, vb01::Quaternion, State = State::STAND_GROUND);
        virtual ~Unit();
        virtual void update();
        virtual void halt();
		void updateGarrison(Vehicle*, bool);
        virtual void select();
        void setOrder(Order);
        std::vector<Projectile*> getProjectiles();
        virtual void receiveOrder(Order, bool);
		bool canGarrison(Vehicle*);
		void initLosLight();
		void destroyLosLight();
		inline void setState(State s){state = s;}
		inline State getState(){return state;}
		inline Engineer* toEngineer(){return (Engineer*)this;}
		inline Structure* toStructure(){return (Structure*)this;}
		inline Factory* toFactory(){return (Factory*)this;}
		inline Cruiser* toCruiser(){return (Cruiser*)this;}
		inline PointDefense* toPointDefense(){return (PointDefense*)this;}
		inline GameObject* toGameObject(){return (GameObject*)this;}
		inline int getNumGarrisonSlots(){return garrisonSlots.size();}
		inline const std::vector<GarrisonSlot>& getGarrisonSlots(){return garrisonSlots;}
        inline float getLineOfSight() {return lineOfSight;}
        inline UnitType getType() {return type;}
        inline UnitClass getUnitClass() {return unitClass;}
        inline int getPlayerId() {return playerId;}
		inline bool isVehicle(){return vehicle;}
		inline bool isTargetToTheRight(vb01::Vector3 dir, vb01::Vector3 lv){return lv.getAngleBetween(dir) > vb01::PI / 2;}
		inline Order getOrder(int i){return orders[i];}
		inline int getNumOrders(){return orders.size();}
		inline std::string getGuiScreen(){return guiScreen;}
		inline std::vector<BuildableUnit> getBuildableUnits(){return buildableUnits;}
		inline int getNumBuildableUnits(){return buildableUnits.size();}
		inline BuildableUnit getBuildableUnit(int i){return buildableUnits[i];}
		inline vb01::Node* getLosLightNode(){return losLightNode;}
		inline Condition getCondition(){return condition;}
		inline void setCondition(Condition cond){
			this->condition = cond;

			if(cond == Condition::EM_JAMMED) lastJamTime = vb01::getTime();
		}
    private:
		void renderOrderLine(bool);
		void initWeapons();
		void destroyWeapons();
		bool validateOrder(Order);
        inline bool canDisplayOrderLine(){return vb01::getTime() - orderLineDispTime < orderVecDispLength;}

        const int orderVecDispLength = 2000;
        sf::SoundBuffer *selectionSfxBuffer;
        sf::Sound *selectionSfx = nullptr;
		vb01::Node *losLightNode = nullptr;
		bool vehicle, currOrderStarted = false, alignToSurface = false;
		Condition condition = Condition::ABLE;
    protected:
        UnitType type;
        UnitClass unitClass;
        std::vector<Order> orders;
		std::string guiScreen = "";
        int playerId, restartTime, lenHpBar = 200;
		vb01::s64 orderLineDispTime = 0, lastFireTime = 0, lastJamTime = 0;
        float lineOfSight;
		std::vector<Weapon*> weapons;
		std::vector<GarrisonSlot> garrisonSlots;
		std::vector<BuildableUnit> buildableUnits;
		State state = State::STAND_GROUND;
		vb01::Node *hpBackgroundNode = nullptr, *hpForegroundNode = nullptr;

		void placeAt(vb01::Vector3);
		std::vector<Player*> getSelectingPlayers();
        void removeOrder(int);
		virtual void startCurrentOrder(){currOrderStarted = true;}
		virtual bool validateLaunchOrder(){return !getWeaponsByOrder(Order::TYPE::LAUNCH).empty();}
		virtual bool validateGarrisonOrder(Order){return false;}
		virtual void autoAttackTargets();
		virtual void reinit();
		virtual void initProperties();
		virtual void destroySound();
		virtual void initSound();
        virtual void executeOrders();
        virtual void eject(Order);
        virtual void attack(Order);
        virtual void garrison(Order){}
        virtual void build(Order){}
        virtual void move(Order){}
        virtual void patrol(Order){}
        virtual void launch(Order);
		virtual void handleResources(Order){}
		virtual void hack(Order){}
		virtual void freeze(Order){}
		float calculateRotation(vb01::Vector3, float, float);
		std::vector<Weapon*> getWeaponsByOrder(Order::TYPE);
		std::vector<Weapon*> getWeaponsByType(int);
    };
}

#endif