#pragma once

#include "../../Util/HashExtension.h"
#include "../../Util/JavaIntHash.h"

class Level;
class Packet;
class CompoundTag;

typedef TileEntity* (*tileEntityCreateFn)();

class TileEntity : public std::enable_shared_from_this<TileEntity> {
public:
    static void staticCtor();
    virtual eINSTANCEOF GetType() { return eTYPE_TILEENTITY; }

private:
    typedef std::unordered_map<std::wstring, tileEntityCreateFn>
        idToCreateMapType;
    typedef std::unordered_map<eINSTANCEOF, std::wstring, eINSTANCEOFKeyHash,
                               eINSTANCEOFKeyEq>
        classToIdMapType;
    static idToCreateMapType idCreateMap;
    static classToIdMapType classIdMap;
    static void setId(tileEntityCreateFn createFn, eINSTANCEOF clas,
                      std::wstring id);
    bool remove;
    unsigned char renderRemoveStage;  // 4J added

public:
    Level* level;
    int x, y, z;

    // 4J added
    enum RenderRemoveStage {
        e_RenderRemoveStageKeep,
        e_RenderRemoveStageFlaggedAtChunk,
        e_RenderRemoveStageRemove
    };

    int data;
    Tile* tile;

public:
    // 4J Java does not have a ctor, but we need one to do some initialisation
    // of the member variables
    TileEntity();
    virtual ~TileEntity() {}

    void setRenderRemoveStage(unsigned char stage);  // 4J added
    void upgradeRenderRemoveStage();                 // 4J added
    bool shouldRemoveForRender();                    // 4J added

    virtual Level* getLevel();
    virtual void setLevel(Level* level);
    virtual bool hasLevel();
    virtual void load(CompoundTag* tag);
    virtual void save(CompoundTag* tag);
    virtual void tick();
    static std::shared_ptr<TileEntity> loadStatic(CompoundTag* tag);
    virtual int getData();
    virtual void setData(int data, int updateFlags);
    virtual void setChanged();
    virtual double distanceToSqr(double xPlayer, double yPlayer,
                                 double zPlayer);
    virtual double getViewDistance();
    virtual Tile* getTile();
    virtual std::shared_ptr<Packet> getUpdatePacket();
    virtual bool isRemoved();
    virtual void setRemoved();
    virtual void clearRemoved();
    virtual bool triggerEvent(int b0, int b1);
    virtual void clearCache();

    // 4J Added
    virtual std::shared_ptr<TileEntity> clone() = 0;

protected:
    void clone(std::shared_ptr<TileEntity> tileEntity);
};