#ifndef _LGB_H
#define _LGB_H

#include <cstring>
#include <memory>
#include <cstdint>
#include <iostream>
#include <vector>
#include <map>
#include <string>

#include "matrix4.h"
#include "vec3.h"
#include "sgb.h"

// all credit to
// https://github.com/ufx/SaintCoinach/blob/master/SaintCoinach/Graphics/Lgb/
// this is simply their work ported to c++ since we dont c#
struct LGB_FILE;
struct LGB_FILE_HEADER;
struct LGB_GROUP;
struct LGB_GROUP_HEADER;

enum class LgbEntryType : uint32_t
{
   BgParts = 1,
   Light = 3,
   Vfx = 4,
   PositionMarker = 5,
   Gimmick = 6,
   SharedGroup6 = 6,// secondary variable is set to 2
   Sound = 7,
   EventNpc = 8,
   BattleNpc = 9,
   Aetheryte = 12,
   EnvSpace = 13,
   Gathering = 14,
   SharedGroup15 = 15,// secondary variable is set to 13
   Treasure = 16,
   Weapon = 39,
   PopRange = 40,
   ExitRange = 41,
   MapRange = 43,
   NaviMeshRange = 44,
   EventObject = 45,
   EnvLocation = 47,
   EventRange = 49,
   QuestMarker = 51,
   CollisionBox = 57,
   DoorRange = 58,
   LineVfx = 59,
   ClientPath = 65,
   ServerPath = 66,
   GimmickRange = 67,
   TargetMarker = 68,
   ChairMarker = 69,
   ClickableRange = 70,
   PrefetchRange = 71,
   FateRange = 72,
   SphereCastRange = 75,
};

struct LGB_ENTRY_HEADER
{
   LgbEntryType type;
   uint32_t unknown;
   uint32_t nameOffset;
   vec3 translation;
   vec3 rotation;
   vec3 scale;
};

class LGB_MODEL_ENTRY
{
public:
   char* m_buf;
   uint32_t m_offset;

   LGB_MODEL_ENTRY()
   {
      m_buf = nullptr;
      m_offset = 0;
   };
   LGB_MODEL_ENTRY( char* buf, uint32_t offset )
   {
      m_buf = buf;
      m_offset = offset;
   };
   virtual ~LGB_MODEL_ENTRY() {};
};


struct LGB_BGPARTS_HEADER : public LGB_ENTRY_HEADER
{
   uint32_t modelFileOffset;
   uint32_t collisionFileOffset;
   uint32_t unknown4;
   uint32_t unknown5;
   uint32_t unknown6;
   uint32_t unknown7;
   uint32_t unknown8;
   uint32_t unknown9;
};

class LGB_BGPARTS_ENTRY : public LGB_MODEL_ENTRY
{
public:
   LGB_BGPARTS_HEADER header;
   std::string name;
   std::string modelFileName;
   std::string collisionFileName;
   LGB_BGPARTS_ENTRY() {};
   LGB_BGPARTS_ENTRY( char* buf, uint32_t offset )
   {
      header = *reinterpret_cast<LGB_BGPARTS_HEADER*>( buf + offset );
      name = std::string( buf + offset + header.nameOffset );
      modelFileName = std::string( buf + offset + header.modelFileOffset );
      collisionFileName = std::string( buf + offset + header.collisionFileOffset );
   };
};

struct LGB_GIMMICK_HEADER : public LGB_ENTRY_HEADER
{
   uint32_t gimmickFileOffset;
   char unknownBytes[100];
};

class LGB_GIMMICK_ENTRY : public LGB_MODEL_ENTRY
{
public:
   LGB_GIMMICK_HEADER header;
   std::string name;
   std::string gimmickFileName;

   LGB_GIMMICK_ENTRY( char* buf, uint32_t offset )
   {
      header = *reinterpret_cast<LGB_GIMMICK_HEADER*>( buf + offset );
      name = std::string( buf + offset + header.nameOffset );
      gimmickFileName = std::string( buf + offset + header.gimmickFileOffset );
      //std::cout << "\t " << gimmickFileName << " unknown: " << header.unknown << "\n";
   };
};

struct LGB_ENPC_HEADER : public LGB_ENTRY_HEADER
{
   uint32_t enpcId;
   uint8_t unknown1[0x24];
};

class LGB_ENPC_ENTRY : public LGB_MODEL_ENTRY
{
public:
   LGB_ENPC_HEADER header;
   std::string name;

   LGB_ENPC_ENTRY( char* buf, uint32_t offset )
   {
      header = *reinterpret_cast< LGB_ENPC_HEADER* >( buf + offset );
      name = std::string( buf + offset + header.nameOffset );
      //std::cout << "\t ENpc " << header.enpcId << " " << name << "\n";
   };
};

struct LGB_EOBJ_HEADER : public LGB_ENTRY_HEADER
{
   uint32_t eobjId;
   uint8_t unknown1[0x10];
};

class LGB_EOBJ_ENTRY : public LGB_MODEL_ENTRY
{
public:
   LGB_EOBJ_HEADER header;
   std::string name;

   LGB_EOBJ_ENTRY( char* buf, uint32_t offset )
   {
      header = *reinterpret_cast< LGB_EOBJ_HEADER* >( buf + offset );
      //std::cout << "\t " << header.eobjId << " " << name << " unknown: " << header.unknown << "\n";
      name = std::string( buf + offset + header.nameOffset );
   };
};

struct LGB_GROUP_HEADER
{
   uint32_t unknown;
   int32_t groupNameOffset;
   int32_t entriesOffset;
   int32_t entryCount;
   uint32_t unknown2;
   uint32_t unknown3;
   uint32_t unknown4;
   uint32_t unknown5;
   uint32_t unknown6;
   uint32_t unknown7;
   uint32_t unknown8;
   uint32_t unknown9;
   uint32_t unknown10;
};

struct LGB_GROUP
{
   LGB_FILE* parent;
   LGB_GROUP_HEADER header;
   std::string name;
   std::vector< std::shared_ptr< LGB_MODEL_ENTRY > > entries;

   LGB_GROUP( char* buf, LGB_FILE* parentStruct, uint32_t offset )
   {
      parent = parentStruct;
      header = *reinterpret_cast< LGB_GROUP_HEADER* >( buf + offset );
      name = std::string( buf + offset + header.groupNameOffset );
      //entries.resize( header.entryCount );
      //std::cout << name << "\n\t unknown: " << header.unknown << "\n";
      const auto entriesOffset = offset + header.entriesOffset;
      for( auto i = 0; i < header.entryCount; ++i )
      {
         const auto entryOffset = entriesOffset + *reinterpret_cast< int32_t* >( buf + ( entriesOffset + i * 4 ) );

         try
         {
            const auto type = *reinterpret_cast<LgbEntryType*>( buf + entryOffset );
            if( type == LgbEntryType::BgParts )
            {
               entries.push_back( std::make_shared< LGB_BGPARTS_ENTRY >( buf, entryOffset ) );
            }
            else if( type == LgbEntryType::Gimmick )
            {
               entries.push_back( std::make_shared< LGB_GIMMICK_ENTRY >( buf, entryOffset ) );
            }
            else if( type == LgbEntryType::EventNpc )
            {
               entries.push_back( std::make_shared< LGB_ENPC_ENTRY >( buf, entryOffset ) );
            }
            else if( type == LgbEntryType::EventObject )
            {
               entries.push_back( std::make_shared< LGB_EOBJ_ENTRY >( buf, entryOffset ) );
            }
            /*
            else
            {
               //entries[i] = nullptr;
            }
            */
         }
         catch( std::exception& e )
         {
            std::cout << name << " " << e.what() << std::endl;
         }
      }
   };
};

struct LGB_FILE_HEADER
{
   char magic[4]; // LGB 1
   uint32_t fileSize;
   uint32_t unknown;
   char magic2[4]; // LGP1
   uint32_t unknown2;
   uint32_t unknown3;
   uint32_t unknown4;
   uint32_t unknown5;
   int32_t groupCount;
};

struct LGB_FILE
{
   LGB_FILE_HEADER header;
   std::vector< LGB_GROUP > groups;

   LGB_FILE( char* buf )
   {
      header = *reinterpret_cast< LGB_FILE_HEADER* >( buf );
      if( strncmp( &header.magic[0], "LGB1", 4 ) != 0 || strncmp( &header.magic2[0], "LGP1", 4 ) != 0 )
         throw std::runtime_error( "Invalid LGB file!" );

      //groups.resize(header.groupCount);

      constexpr auto baseOffset = sizeof( header );
      for( auto i = 0; i < header.groupCount; ++i )
      {
         const auto groupOffset = baseOffset + *reinterpret_cast< int32_t* >( buf + ( baseOffset + i * 4 ) );
         const auto group = LGB_GROUP( buf, this, groupOffset );
         groups.push_back( group );
      }
   };
};

/*
#if __cplusplus >= 201703L
#include <experimental/filesystem>
std::map<std::string, LGB_FILE> getLgbFiles( const std::string& dir )
{
   namespace fs = std::experimental::filesystem;
   std::map<std::string, LGB_FILE> fileMap;
   for( const auto& path : fs::recursive_directory_iterator( dir ) )
   {
      if( path.path().extension() == ".lgb" )
      {
         const auto& strPath = path.path().string();
         auto f = fopen( strPath.c_str(), "rb" );
         fseek( f, 0, SEEK_END );
         const auto size = ftell( f );
         std::vector<char> bytes( size );
         rewind( f );
         fread( bytes.data(), 1, size, f );
         fclose( f );
         try
         {
            LGB_FILE lgbFile( bytes.data() );
            fileMap.insert( std::make_pair( strPath, lgbFile ) );
         }
         catch( std::exception& e )
         {
            std::cout << "Unable to load " << strPath << std::endl;
         }
      }
   }
   return fileMap;
}
#endif
*/
#endif