dqn
/
fbx_to_3dtiles
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

pid

This commit is contained in:
dqn 2025-09-15 10:33:59 +08:00
parent cfe1fa46e0
commit a0ca1bffc3
5 changed files with 84 additions and 64 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

76
cpp_src/db_ops.cpp
View File

@ -31,6 +31,7 @@ void create_tables(sqlite3 *db)
const char *sql_meshes = R"(
CREATE TABLE IF NOT EXISTS meshes (
id INTEGER PRIMARY KEY AUTOINCREMENT,
pid INTEGER,
name TEXT NOT NULL,
vertex_count INTEGER NOT NULL,
index_count INTEGER NOT NULL,
@ -56,11 +57,6 @@ void create_tables(sqlite3 *db)
position_x REAL NOT NULL,
position_y REAL NOT NULL,
position_z REAL NOT NULL,
normal_x REAL,
normal_y REAL,
normal_z REAL,
uv_u REAL,
uv_v REAL,
FOREIGN KEY(mesh_id) REFERENCES meshes(id) ON DELETE CASCADE
);
)";
@ -125,6 +121,7 @@ void create_tables(sqlite3 *db)
int save_mesh_to_table(FttContext *ctx,
sqlite3 *db,
uint id,
uint pid,
const std::string &name,
int vertex_count, int index_count,
const std::string &fbx_id)
@ -134,7 +131,7 @@ int save_mesh_to_table(FttContext *ctx,
return -1;
sqlite3_stmt *stmt = nullptr;
const char *sql = "INSERT OR IGNORE INTO meshes (id, name, vertex_count, index_count, fbx_id) VALUES (?, ?, ?, ?, ?);";
const char *sql = "INSERT OR IGNORE INTO meshes (id, pid, name, vertex_count, index_count, fbx_id) VALUES (?, ?, ?, ?, ?, ?);";
int rc = sqlite3_prepare_v2(db, sql, -1, &stmt, nullptr);
if (rc != SQLITE_OK)
{
@ -143,10 +140,11 @@ int save_mesh_to_table(FttContext *ctx,
}
sqlite3_bind_int64(stmt, 1, id);
sqlite3_bind_text(stmt, 2, name.c_str(), -1, SQLITE_STATIC);
sqlite3_bind_int(stmt, 3, vertex_count);
sqlite3_bind_int(stmt, 4, index_count);
sqlite3_bind_text(stmt, 5, fbx_id.c_str(), -1, SQLITE_STATIC);
sqlite3_bind_int64(stmt, 2, pid);
sqlite3_bind_text(stmt, 3, name.c_str(), -1, SQLITE_STATIC);
sqlite3_bind_int(stmt, 4, vertex_count);
sqlite3_bind_int(stmt, 5, index_count);
sqlite3_bind_text(stmt, 6, fbx_id.c_str(), -1, SQLITE_STATIC);
rc = sqlite3_step(stmt);
if (rc != SQLITE_DONE && rc != SQLITE_CONSTRAINT)
@ -193,37 +191,43 @@ int save_mesh_vertices(FttContext *ctx, uint mesh_id, std::vector<V3> vertices)
return 0;
}
int save_mesh_indices(FttContext* ctx, uint mesh_id, std::vector<V3> indices) {
int save_mesh_indices(FttContext* ctx, uint mesh_id, std::vector<unsigned int> indices) {
if (!ctx->db) return -1;
sqlite3_stmt *stmt = nullptr;
const char *sql = "INSERT OR IGNORE INTO indices (mesh_id, index_id, position_x, position_y, position_z) VALUES (?, ?, ?, ?, ?);";
int rc = sqlite3_prepare_v2(ctx->db, sql, -1, &stmt, nullptr);
if (rc != SQLITE_OK) {
fprintf(stderr, "SQL prepare failed: %s\n", sqlite3_errmsg(ctx->db));
return rc;
printf("save indices: %d\n", indices.size());
for (int i = 0; i < indices.size(); i++)
{
printf("%d ", indices[i]);
}
sqlite3_exec(ctx->db, "BEGIN TRANSACTION;", nullptr, nullptr, nullptr); // 开启事务
for (int i = 0; i < indices.size(); i++) {
sqlite3_reset(stmt); // 重置语句
sqlite3_bind_int64(stmt, 1, mesh_id);
sqlite3_bind_int64(stmt, 2, i); // vert_id使用索引
sqlite3_bind_double(stmt, 3, std::get<0>(indices[i])); // 修正为double类型绑定
sqlite3_bind_double(stmt, 4, std::get<1>(indices[i]));
sqlite3_bind_double(stmt, 5, std::get<2>(indices[i]));
// sqlite3_stmt *stmt = nullptr;
// const char *sql = "INSERT OR IGNORE INTO indices (mesh_id, index_id, position_x, position_y, position_z) VALUES (?, ?, ?, ?, ?);";
// int rc = sqlite3_prepare_v2(ctx->db, sql, -1, &stmt, nullptr);
// if (rc != SQLITE_OK) {
// fprintf(stderr, "SQL prepare failed: %s\n", sqlite3_errmsg(ctx->db));
// return rc;
// }
rc = sqlite3_step(stmt);
if (rc != SQLITE_DONE && rc != SQLITE_CONSTRAINT) {
fprintf(stderr, "SQL step failed at vertex %d: %s\n", i, sqlite3_errmsg(ctx->db));
sqlite3_exec(ctx->db, "ROLLBACK;", nullptr, nullptr, nullptr);
sqlite3_finalize(stmt);
return rc;
}
}
// sqlite3_exec(ctx->db, "BEGIN TRANSACTION;", nullptr, nullptr, nullptr); // 开启事务
sqlite3_exec(ctx->db, "COMMIT;", nullptr, nullptr, nullptr); // 提交事务
sqlite3_finalize(stmt);
// for (int i = 0; i < indices.size(); i++) {
// sqlite3_reset(stmt); // 重置语句
// sqlite3_bind_int64(stmt, 1, mesh_id);
// sqlite3_bind_int64(stmt, 2, i); // vert_id使用索引
// sqlite3_bind_double(stmt, 3, std::get<0>(indices[i])); // 修正为double类型绑定
// sqlite3_bind_double(stmt, 4, std::get<1>(indices[i]));
// sqlite3_bind_double(stmt, 5, std::get<2>(indices[i]));
// rc = sqlite3_step(stmt);
// if (rc != SQLITE_DONE && rc != SQLITE_CONSTRAINT) {
// fprintf(stderr, "SQL step failed at vertex %d: %s\n", i, sqlite3_errmsg(ctx->db));
// sqlite3_exec(ctx->db, "ROLLBACK;", nullptr, nullptr, nullptr);
// sqlite3_finalize(stmt);
// return rc;
// }
// }
// sqlite3_exec(ctx->db, "COMMIT;", nullptr, nullptr, nullptr); // 提交事务
// sqlite3_finalize(stmt);
return 0;
}

62
cpp_src/fbx_wrap.cpp
View File

@ -6,32 +6,32 @@
using namespace std;
int load_triangles(fbxsdk::FbxNodeAttribute *attr, std::vector<V3> &indices, std::vector<V3> &triangles)
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);
int polyCount = mesh->GetPolygonCount();
for (int i = 0; i < polyCount; ++i)
unsigned int index_p = 0;
FbxVector4* control_points = mesh->GetControlPoints();
for (int p = 0; p < mesh->GetPolygonCount(); p++)
{
int vertexCount = mesh->GetPolygonSize(i);
std::vector<int> polyIndices;
for (int j = 0; j < vertexCount; ++j)
{
polyIndices.push_back(mesh->GetPolygonVertex(i, j));
if (mesh->GetPolygonSize(p) != 3) {
// 不是三角形
return index_p;
}
for (int j = 2; j < vertexCount; ++j)
for (int p_vertex = 0; p_vertex < mesh->GetPolygonSize(p); p_vertex++)
{
indices.push_back({polyIndices[0], polyIndices[j - 1], polyIndices[j]});
for (int k : {0, j - 1, j})
{
FbxVector4 pos = mesh->GetControlPointAt(polyIndices[k]);
triangles.push_back({(float)pos[0], (float)pos[1], (float)pos[2]});
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 0;
return index_p;
}
FbxNode *get_fbx_root_node(const char *fbx_path)
@ -81,23 +81,28 @@ void each_node(FttContext *ctx, FbxNode *parent_node, int level)
FbxNodeAttribute *attr = parent_node->GetNodeAttributeByIndex(i);
if (attr && attr->GetAttributeType() == FbxNodeAttribute::eMesh)
{
std::vector<unsigned int> indices;
std::vector<V3> vertices;
std::vector<V3> triangles;
int count = load_triangles(attr, vertices, triangles);
int count = load_triangles(attr, indices, vertices);
uint id = attr->GetUniqueID();
printf("[%d]triangles: %ld\n", id, triangles.size());
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(), triangles.size(),
vertices.size(), indices.size(),
parent_node->GetNameOnly().Buffer());
save_mesh_vertices(ctx, id, triangles);
save_mesh_indices(ctx, id, vertices);
save_mesh_vertices(ctx, id, vertices);
save_mesh_indices(ctx, id, indices);
}
}
@ -114,6 +119,12 @@ void each_node(FttContext *ctx, FbxNode *parent_node, int level)
void print_node_info(FbxNode *node, int level)
{
// todo 打印一下节点信息
// 1. 平移、旋转、缩放等矩阵
// 2. 名称，id, pid等
// 3. 节点其他properties
// 打印一下节点信息
printf("|");
for (int i = 0; i < level; i++)
@ -121,5 +132,10 @@ void print_node_info(FbxNode *node, int level)
printf("-");
}
printf("> ");
printf("[%lld]: %s\n", node->GetUniqueID(), node->GetNameOnly().Buffer());
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());
}

4
cpp_src/include/db_ops.h
View File

@ -32,7 +32,7 @@ void create_tables(sqlite3* db);
* @note 1. 参数需确保有效：ctx/db非NULL，name/fbx_id非空
* 2. 内部自动处理事务，重复name会触发唯一键冲突
*/
int save_mesh_to_table(FttContext* ctx, sqlite3 *db, uint id, const std::string &name, int vertex_count, int index_count, const std::string &fbx_id);
int save_mesh_to_table(FttContext* ctx, sqlite3 *db, uint id, uint pid, const std::string &name, int vertex_count, int index_count, const std::string &fbx_id);
int save_mesh_vertices(FttContext* ctx, uint mesh_id, std::vector<V3> vertices);
int save_mesh_indices(FttContext* ctx, uint mesh_id, std::vector<V3> indices);
int save_mesh_indices(FttContext* ctx, uint mesh_id, std::vector<unsigned int> indices);

4
cpp_src/include/fbx_wrap.h
View File

@ -32,9 +32,9 @@ FbxNode* get_fbx_root_node(const char* fbx_path);
/**
* @brief 从FBX网格属性加载顶点数据和三角化索引
* @param[in] attr FBX节点属性，需为FbxMesh类型
* @param[out] indices 输出三角形索引数组（每个Triangle存储3个顶点索引）
* @param[out] vertices 输出顶点坐标数组（每个Triangle存储3个FbxVector4顶点）
* @param[out] triangles 输出三角形索引数组（每个Triangle存储3个顶点索引）
* @return bool 成功返回true，失败返回false
* @note 内部会强制三角化网格，非三角形多边形将被忽略
*/
int load_triangles(fbxsdk::FbxNodeAttribute* attr, std::vector<V3>& indices, std::vector<V3>& triangles);
int load_triangles(fbxsdk::FbxNodeAttribute* attr, std::vector<unsigned int> indices, std::vector<V3> &vertices);

2
cpp_src/include/tools.h
View File

@ -17,7 +17,7 @@ typedef struct {
sqlite3* db;
} FttContext;
using V3 = std::tuple<int, int, int>;
using V3 = std::tuple<double, double, double>;
FttContext ctx_from_const(const FttContext* ctx);