#include "../util/typedef.h"
#include "../util/string.h"
#include "../util/vector.h"
#include "../util/file.h"

#include <assert.h>

struct MODEL {

};

static inline char* skip_space( char* s ) {
  while( *s == ' ' || *s == '\t' ) ++s;
  return s;
}

static inline VEC3 read_vec3( char* s ) {
  VEC3 temp{};
  s = skip_space( s );
  temp.x = strtof( s, &s );
  s = skip_space( s );
  temp.y = strtof( s, &s );
  s = skip_space( s );
  temp.z = strtof( s, &s );
  return temp;
}

namespace MTL {
  enum LINE_TYPES : U16 {
    comment = ( '#' | ( ' ' << 8 ) ),
    newmtl  = ( 'n' | ( 'e' << 8 ) ),
    Ns      = ( 'N' | ( 's' << 8 ) ),
    Ka      = ( 'K' | ( 'a' << 8 ) ),
    Kd      = ( 'K' | ( 'd' << 8 ) ),
    Ks      = ( 'K' | ( 's' << 8 ) ),
    Ke      = ( 'K' | ( 'e' << 8 ) ),
    Ni      = ( 'N' | ( 'i' << 8 ) ),
    d       = ( 'd' | ( ' ' << 8 ) ),
    illum   = ( 'i' | ( 'l' << 8 ) ),
    map_Kd  = ( 'm' | ( 'a' << 8 ) )
  };

  struct MTLDATA {
    STR name;
    F32 Ns;
    VEC3 Ka;
    VEC3 Kd;
    VEC3 Ks;
    VEC3 Ke;
    F32 Ni;
    F32 d;
    I32 illum;
    STR map_Kd;
  };

  struct MTLLIB {
    STR name;
    LIST<MTLDATA> data;
  };

  inline MTLLIB mtl_from_file( const char* path, const char* name ) {
    U64 mtl_size = 0;
    STR mtl_path( path );
    mtl_path += name;
    char* f_mtl = (char*)file_read( mtl_path.data, &mtl_size );
    if( !f_mtl ) {
      dlog( "mtl_from_file() : failed to read mtl file %s\n", mtl_path.data );
      return {};
    }

    MTLLIB mtl;
    mtl.name = name;
    for( U64 i = 0; i < mtl_size - 1 && f_mtl[i]; ++i ) {
      if( f_mtl[i] == '\n' )
        continue;

      I32 c0 = f_mtl[i];
      U16 c1 = f_mtl[i + 1];
      I32 first_two = c0 | c1 << 8;

      switch( first_two ) {
      case LINE_TYPES::newmtl: {
        char* start = f_mtl + i + 7;
        char* end = start;
        while( *end && *end != '\n' ) ++end;
        U32 len = end - start;
        STR temp;
        temp.reserve( len + 1 );
        memcpy( temp, start, len );
        temp.data[len] = 0;
        mtl.data.resize( mtl.data.size + 1 );
        mtl.data[mtl.data.size - 1].name = temp.data;
      } break;
      case LINE_TYPES::d:
      case LINE_TYPES::Ni:
      case LINE_TYPES::Ns: {
        char* s = f_mtl + i + 2;
        s = skip_space( s );
        F32 temp = strtof( s, &s );
        switch( first_two ) {
        case LINE_TYPES::d:
          mtl.data[mtl.data.size - 1].d  = temp;
          break;
        case LINE_TYPES::Ni:
          mtl.data[mtl.data.size - 1].Ni = temp;
          break;
        case LINE_TYPES::Ns:
          mtl.data[mtl.data.size - 1].Ns = temp;
          break;
        }
      } break;
      case LINE_TYPES::illum: {
        char* s = f_mtl + i + 6;
        mtl.data[mtl.data.size - 1].illum = (I32)strtol( s, &s, 10 );
      } break;
      case LINE_TYPES::Ka:
      case LINE_TYPES::Kd:
      case LINE_TYPES::Ke:
      case LINE_TYPES::Ks: {
        char* s = f_mtl + i + 3;
        switch( first_two ) {
        case LINE_TYPES::Ka:
          mtl.data[mtl.data.size - 1].Ka = read_vec3( s );
          break;
        case LINE_TYPES::Kd:
          mtl.data[mtl.data.size - 1].Kd = read_vec3( s );
          break;
        case LINE_TYPES::Ke:
          mtl.data[mtl.data.size - 1].Ke = read_vec3( s );
          break;
        case LINE_TYPES::Ks:
          mtl.data[mtl.data.size - 1].Ks = read_vec3( s );
          break;
        }
      } break;
      case LINE_TYPES::map_Kd: {
        char* start = f_mtl + i + 7;
        char* end = start;
        while( *end && *end != '\n' ) ++end;
        U32 len = end - start;
        STR temp;
        temp.reserve( len + 1 );
        memcpy( temp, start, len );
        temp.data[len] = 0;
        mtl.data[mtl.data.size - 1].map_Kd = temp.data;
      } break;
      case LINE_TYPES::comment: break;
      default: dlog( "mtl_from_file() : unhandled char combo %c%c\n", c0, c1 ); break;
      }

      while( f_mtl[i] && f_mtl[i] != '\n' ) ++i;
    }

    free( (void*)f_mtl );
    return mtl;
  }
};

namespace OBJ {
  enum LINE_TYPES : U16 {
    comment = ( '#' | ( ' ' << 8 ) ),
    f       = ( 'f' | ( ' ' << 8 ) ),
    mtllib  = ( 'm' | ( 't' << 8 ) ),
    o       = ( 'o' | ( ' ' << 8 ) ),
    s       = ( 's' | ( ' ' << 8 ) ),
    v       = ( 'v' | ( ' ' << 8 ) ),
    vn      = ( 'v' | ( 'n' << 8 ) ),
    vt      = ( 'v' | ( 't' << 8 ) ),
    usemtl  = ( 'u' | ( 's' << 8 ) )
  };

  enum FACE_TYPES : U8 {
    V     = 0,
    VVT   = 1,
    VVN   = 2,
    VVTVN = 3
  };

  static inline U32 face_stride( FACE_TYPES t ) {
    switch( t ) {
    case FACE_TYPES::V:     return 1;
    case FACE_TYPES::VVT:   return 2;
    case FACE_TYPES::VVN:   return 2;
    case FACE_TYPES::VVTVN: return 3;
    }
    return 0;
  }

  constexpr U32 STR_LEN = 0x40;

  struct OBJDATA {
    LIST<VEC2> uvs;
    LIST<VEC3> normals;
    LIST<VEC3> vertices;

    LIST<MTL::MTLLIB> mtllib;

    struct UseMtl {
      LIST<U32> indices;
      LIST<STR> data;
    } usemtl;

    struct Indices {
      LIST<U32> data;
      LIST<U32> counts;
      LIST<FACE_TYPES> types;
    } indices;
  };

  inline OBJDATA obj_from_file( const char* path, const char* name ) {
    char* f_obj = nullptr;
    STR obj_path( path );
    U64 obj_size = 0;
    obj_path += name;
    obj_path += ".obj";

    f_obj = (char*)file_read( obj_path.data, &obj_size );
    if( !f_obj ) {
      dlog( "obj_from_file() : failed to read obj file %s\n", obj_path.data );
      return {};
    }

    OBJDATA obj;
    for( U64 i = 0; i < obj_size - 1 && f_obj[i]; ++i ) {
      if( f_obj[i] == (I32)'#' ) {
        while( f_obj[i] && f_obj[i] != '\n' ) ++i;
        continue;
      }

      I32 c0 = f_obj[i];
      U16 c1 = f_obj[i + 1];
      I32 first_two = c0 | c1 << 8;

      switch( first_two ) {
      case LINE_TYPES::f: {
        FACE_TYPES type = FACE_TYPES::V;
        char* s = f_obj + i + 2;
        bool type_set = 0;
        U32 count = 0;

        s = skip_space( s );

        while( *s && *s != '\n' ) {
          bool has_vt = 0, has_vn = 0;
          U32 v = strtoul( s, &s, 10 );
          obj.indices.data.push( --v );

          if( *s == '/' ) {
            ++s;

            if( *s != '/' ) {
              U32 vt = strtoul( s, &s, 10 );
              obj.indices.data.push( --vt );
              has_vt = 1;
            }

            if( *s == '/' ) {
              ++s;
              U32 vn = strtoul( s, &s, 10 );
              obj.indices.data.push( --vn );
              has_vn = 1;
            }
          }

          if( !type_set ) {
            if( has_vt && has_vn )
              type = FACE_TYPES::VVTVN;
            else if( has_vt )
              type = FACE_TYPES::VVT;
            else if( has_vn )
              type = FACE_TYPES::VVN;
            else
              type = FACE_TYPES::V;
            type_set = 1;
          }

          ++count;

          while( *s && *s != ' ' && *s != '\n' )
            ++s;
          s = skip_space( s );
        }
        obj.indices.counts.push( count );
        obj.indices.types.push( type );
      } break;
      case LINE_TYPES::v:
      case LINE_TYPES::vn: {
        bool is_v = first_two == LINE_TYPES::v;
        U32 start = i + ( is_v ? 2 : 3 );
        char* s = f_obj + start;
        if( is_v )
          obj.vertices.push( read_vec3( s ) );
        else
          obj.normals.push( read_vec3( s ) );
      } break;
      case LINE_TYPES::vt: {
        char* s = f_obj + i + 3;
        VEC2 temp{};
        s = skip_space( s );
        temp.x = strtof( s, &s );
        s = skip_space( s );
        temp.y = strtof( s, &s );
        obj.uvs.push( temp );
      } break;
      case LINE_TYPES::mtllib:
      case LINE_TYPES::usemtl: {
        bool is_mtllib = first_two == LINE_TYPES::mtllib;
        char* start = f_obj + i + 7;
        char* end = start;
        while( *end && *end != '\n' ) ++end;
        U32 len = end - start;
        STR temp;
        temp.reserve( len + 1 );
        memcpy( temp, start, len );
        temp.data[len] = 0;
        if( !is_mtllib ) {
          obj.usemtl.data.push( temp );
          obj.usemtl.indices.push( obj.indices.data.size );
        } else
          obj.mtllib.push( MTL::mtl_from_file( path, temp.data ) );
      } break;
      case LINE_TYPES::comment:
      case LINE_TYPES::o:
      case LINE_TYPES::s: break;
      default: dlog( "obj_from_file() : unhandled char combo %c%c\n", c0, c1 ); break;
      }

      while( f_obj[i] && f_obj[i] != '\n' ) ++i;
    }
    free( (void*)f_obj );
    return obj;
  }

  inline MODEL obj_to_model( OBJDATA obj ) {
    MODEL model;

    return model;
  }
};

inline MODEL model_from_obj( const I8* path, const I8* name ) {
  OBJ::OBJDATA obj = OBJ::obj_from_file( path, name );
  return OBJ::obj_to_model( obj );
}