#include <fbx_wrap.h>
#include <stdio.h>
#include <iostream>
#include <db_ops.h>
#include <tools.h>

using namespace std;

int load_triangles(fbxsdk::FbxNodeAttribute *attr, std::vector<unsigned int> indices, std::vector<V3> &vertices)
{
    if (!attr || attr->GetAttributeType() != FbxNodeAttribute::eMesh)
        return false;
    FbxMesh *mesh = static_cast<FbxMesh *>(attr);


    unsigned int index_p = 0;
    FbxVector4* control_points = mesh->GetControlPoints();
    for (int p = 0; p < mesh->GetPolygonCount(); p++)
    {
        if (mesh->GetPolygonSize(p) != 3) {
            // 不是三角形
            return index_p;
        }
        for (int p_vertex = 0; p_vertex < mesh->GetPolygonSize(p); p_vertex++) 
        {
            int vertex_indexed = mesh->GetPolygonVertex(p, p_vertex);
            FbxVector4 control_point = control_points[vertex_indexed];
            V3 v = V3(control_point.mData[0], control_point.mData[1], control_point.mData[2]);
            vertices.push_back(v);
            indices.push_back(index_p++);
        }
    }
    
    return index_p;
}

FbxNode *get_fbx_root_node(const char *fbx_path)
{
    printf("loading fbx file: %s\n", fbx_path);
    FbxManager *lSdkManager = FbxManager::Create();
    if (lSdkManager)
    {
        printf("FbxManager created successfully.\n");
    }
    else
    {
        printf("FbxManager creation failed.\n");
    }
    FbxScene *lScene = FbxScene::Create(lSdkManager, "myScene");
    if (lScene)
    {
        printf("FbxScene created successfully.\n");
    }
    else
    {
        printf("FbxScene creation failed.\n");
    }
    FbxImporter *lImporter = FbxImporter::Create(lSdkManager, "");
    bool s = lImporter->Initialize(fbx_path, -1, lSdkManager->GetIOSettings());
    if (s)
    {
        printf("FbxImporter initialized successfully.\n");
    }
    else
    {
        printf("FbxImporter initialization failed.\n");
    }
    lImporter->Import(lScene);
    lImporter->Destroy();

    return lScene->GetRootNode();
}

void each_node(FttContext *ctx, FbxNode *parent_node, int level)
{
    // info
    print_node_info(parent_node, level);
    // data
    for (int i = 0; i < parent_node->GetNodeAttributeCount(); i++)
    {
        FbxNodeAttribute *attr = parent_node->GetNodeAttributeByIndex(i);
        if (attr && attr->GetAttributeType() == FbxNodeAttribute::eMesh)
        {
            std::vector<unsigned int> indices;
            std::vector<V3> vertices;

            int count = load_triangles(attr, indices, vertices);

            uint id = attr->GetUniqueID();
            uint pid = parent_node->GetUniqueID();

            printf("[%d]vertices: %ld\n", count);
            // save data to database
            sqlite3 *db = ctx->db;
            // todo save mesh into db

            save_mesh_to_table(ctx, ctx->db,
                               id,
                               pid,
                               attr->GetName(),
                               vertices.size(), indices.size(),
                               parent_node->GetNameOnly().Buffer());

            save_mesh_vertices(ctx, id, vertices);
            save_mesh_indices(ctx, id, indices);
        }
    }

    // children
    if (parent_node->GetChildCount())
    {
        for (int i = 0; i < parent_node->GetChildCount(); i++)
        {
            FbxNode *child = parent_node->GetChild(i);
            each_node(ctx, child, level + 1);
        }
    }
}

void print_node_info(FbxNode *node, int level)
{
    // todo 打印一下节点信息
    // 1. 平移、旋转、缩放等矩阵
    // 2. 名称，id, pid等
    // 3. 节点其他properties


    // 打印一下节点信息
    printf("|");
    for (int i = 0; i < level; i++)
    {
        printf("-");
    }
    printf("> ");
    FbxNode* parent = node->GetParent();
    uint parent_id = 0;
    if (parent != NULL) {
        parent_id = parent->GetUniqueID();
    }
    printf("[%lld/%lld]: %s\n", node->GetUniqueID(), parent_id, node->GetNameOnly().Buffer());
}