#include"AssimpLoader.h" #include #include #include #include #include #include using namespace hgl; using namespace hgl::graph; #define LOG_BR LOG_INFO(OS_TEXT("------------------------------------------------------------")); #define aisgl_min(x,y) (xx?y:x) void set_float4(float f[4], float a, float b, float c, float d) { f[0] = a; f[1] = b; f[2] = c; f[3] = d; } void color4_to_float4(const aiColor4D *c, float f[4]) { f[0] = c->r; f[1] = c->g; f[2] = c->b; f[3] = c->a; } void AssimpLoader::get_bounding_box_for_node (const aiNode* nd, aiVector3D* min, aiVector3D* max, aiMatrix4x4* trafo) { aiMatrix4x4 prev; unsigned int n = 0, t; prev = *trafo; aiMultiplyMatrix4(trafo,&nd->mTransformation); for (; n < nd->mNumMeshes; ++n) { const aiMesh* mesh = scene->mMeshes[nd->mMeshes[n]]; for (t = 0; t < mesh->mNumVertices; ++t) { aiVector3D tmp = mesh->mVertices[t]; aiTransformVecByMatrix4(&tmp,trafo); min->x = aisgl_min(min->x,tmp.x); min->y = aisgl_min(min->y,tmp.y); min->z = aisgl_min(min->z,tmp.z); max->x = aisgl_max(max->x,tmp.x); max->y = aisgl_max(max->y,tmp.y); max->z = aisgl_max(max->z,tmp.z); } } for (n = 0; n < nd->mNumChildren; ++n) get_bounding_box_for_node(nd->mChildren[n],min,max,trafo); *trafo = prev; } void AssimpLoader::get_bounding_box (const aiNode *node,aiVector3D* min, aiVector3D* max) { aiMatrix4x4 trafo; aiIdentityMatrix4(&trafo); min->x = min->y = min->z = 1e10f; max->x = max->y = max->z = -1e10f; get_bounding_box_for_node(node,min,max,&trafo); } AssimpLoader::AssimpLoader() { scene=0; material_list=nullptr; } AssimpLoader::~AssimpLoader() { delete[] material_list; if(scene) aiReleaseImport(scene); } void AssimpLoader::SaveFile(const void *data,const uint &size,const OSString &ext_name) { if(!data||size<=0)return; OSString filename=main_filename+OS_TEXT(".")+ext_name; if(filesystem::SaveMemoryToFile(filename,data,size)==size) { LOG_INFO(OS_TEXT("SaveFile: ")+filename+OS_TEXT(" , size: ")+OSString(size)); total_file_bytes+=size; } } void AssimpLoader::SaveFile(void **data,const int64 *size,const int &count,const OSString &ext_name) { if(!data||size<=0)return; OSString filename=main_filename+OS_TEXT(".")+ext_name; int64 result=filesystem::SaveMemoryToFile(filename,data,size,count); if(result>0) { LOG_INFO(OS_TEXT("SaveFile: ")+filename+OS_TEXT(" , size: ")+OSString(result)); total_file_bytes+=result; } } void OutFloat3(const OSString &front,const Color4f &c) { LOG_INFO(front+OS_TEXT(": ")+OSString(c.r) +OS_TEXT("," )+OSString(c.g) +OS_TEXT("," )+OSString(c.b)); } constexpr VkSamplerAddressMode vk_wrap[4]= //对应aiTextureMapMode,注意最后一个其实不对 { VK_SAMPLER_ADDRESS_MODE_REPEAT, VK_SAMPLER_ADDRESS_MODE_CLAMP_TO_EDGE, VK_SAMPLER_ADDRESS_MODE_MIRRORED_REPEAT, VK_SAMPLER_ADDRESS_MODE_MIRROR_CLAMP_TO_EDGE //这个是declas }; void OutMaterialTexture(const MaterialTextureData &mt,io::DataOutputStream *dos) { dos->WriteUint8((uint8)mt.type); dos->WriteInt32(mt.tex_id); dos->WriteUint8(mt.uvindex); dos->WriteFloat(mt.blend); dos->WriteUint8(mt.op); dos->WriteUint8(mt.wrap_mode,2); LOG_INFO(OS_TEXT("\tTexture Type: ")+OSString((uint)mt.type)); LOG_INFO(OS_TEXT("\tTexture ID: ")+OSString(mt.tex_id)); LOG_INFO(OS_TEXT("\tuvindex: ")+OSString(mt.uvindex)); LOG_INFO(OS_TEXT("\tblend: ")+OSString(mt.blend)); LOG_INFO(OS_TEXT("\top: ")+OSString(mt.op)); LOG_INFO(OS_TEXT("\twrap_mode: ")+OSString(mt.wrap_mode[0])+OS_TEXT(",")+OSString(mt.wrap_mode[1])); } void OutMaterial(const MaterialData *ms,const OSString &filename) { io::FileOutputStream fos; io::LEDataOutputStream dos(&fos); fos.CreateTrunc(filename); dos.WriteFully("Material\x1A\x1",10); dos.WriteFloat(ms->diffuse,3); dos.WriteFloat(ms->specular,3); dos.WriteFloat(ms->ambient,3); dos.WriteFloat(ms->emission,3); dos.WriteFloat(ms->shininess); dos.WriteBool(ms->wireframe); dos.WriteBool(ms->two_sided); dos.WriteUint8(ms->tex_count); OutFloat3(OSString(OS_TEXT("diffuse")),ms->diffuse); OutFloat3(OSString(OS_TEXT("specular")),ms->specular); OutFloat3(OSString(OS_TEXT("ambient")),ms->ambient); OutFloat3(OSString(OS_TEXT("emission")),ms->emission); LOG_INFO(OS_TEXT("shininess: ")+OSString(ms->shininess)); LOG_INFO(OS_TEXT("wireframe: ")+OSString(ms->wireframe?OS_TEXT("true"):OS_TEXT("false"))); LOG_INFO(OS_TEXT("two_sided: ")+OSString(ms->two_sided?OS_TEXT("true"):OS_TEXT("false"))); LOG_INFO(OS_TEXT("Material Texture Count ")+OSString(ms->tex_count)); for(int i=0;itex_count;i++) OutMaterialTexture(ms->tex_list[i],&dos); fos.Close(); } bool GetMaterialBool(aiMaterial *mtl,const char *key,unsigned int type,unsigned int index,const bool default_value) { int value; if(aiGetMaterialInteger(mtl, AI_MATKEY_TWOSIDED, &value)==AI_SUCCESS) return value; else return default_value; } uint GetMaterialInteger(aiMaterial *mtl,const char *key,unsigned int type,unsigned int index,const uint default_value) { int value; if(aiGetMaterialInteger(mtl, AI_MATKEY_TWOSIDED, &value)==AI_SUCCESS) return value; else return default_value; } float GetMaterialFloat(aiMaterial *mtl,const char *key,unsigned int type,unsigned int index,const float default_value) { float value; if(aiGetMaterialFloat(mtl, AI_MATKEY_TWOSIDED, &value)==AI_SUCCESS) return value; else return default_value; } Color4f GetMaterialColor(aiMaterial *mtl,const char *key,unsigned int type,unsigned int index,const Color4f &default_value) { unsigned int max = 1; aiColor4D color; if(aiGetMaterialColor(mtl, AI_MATKEY_TWOSIDED, &color)==AI_SUCCESS) return Color4f(color.r,color.g,color.b,color.a); else return default_value; } void AssimpLoader::LoadMaterial() { if(!scene->HasMaterials())return; int ret1, ret2; float c[4]; aiColor4D diffuse; aiColor4D specular; aiColor4D ambient; aiColor4D emission; float shininess; float strength; float bump_scaling; float opacity; float refracti; aiColor4D transparent; aiColor4D reflective; int two_sided; int wireframe; unsigned int max; // changed: to unsigned LOG_BR; LOG_INFO(OS_TEXT("Material Count ")+OSString(scene->mNumMaterials)); LOG_BR; aiString filename; aiTextureMapping tm; uint uvindex=0; float blend; aiTextureOp op; aiTextureMapMode wrap_mode[2]; material_count=scene->mNumMaterials; material_list=new MaterialData[material_count]; for(unsigned int m=0;mmNumMaterials;m++) { int tex_index=0; int tex_count=0; int tt_count=0; int32 tex_id; aiReturn found=AI_SUCCESS; aiMaterial *mtl=scene->mMaterials[m]; for(uint tt=aiTextureType_DIFFUSE;ttGetTextureCount((aiTextureType)tt); } MaterialData *ms=&(material_list[m]); ms->Init(tex_count); for(uint tt=aiTextureType_DIFFUSE;ttGetTextureCount((aiTextureType)tt); for(int t=0;tGetTexture((aiTextureType)tt,t,&filename,&tm,&uvindex,&blend,&op,wrap_mode); ms->tex_list[tex_index].type=(TextureType)tt; ms->uv_use.Add(uvindex); { char *fn=strrchr(filename.data,'\\'); char *ext; if(!fn) fn=strrchr(filename.data,'/'); else ++fn; if(fn) ++fn; else fn=filename.data; ext=strrchr(fn,'.'); if(ext)*ext=0; #if HGL_OS == HGL_OS_Window tex_id=tex_list.CaseFind(fn); #else tex_id=tex_list.Find(fn); #endif// if(tex_id==-1) tex_id=tex_list.Add(fn); ms->tex_list[tex_index].tex_id=tex_id; } ms->tex_list[tex_index].uvindex=uvindex; ms->tex_list[tex_index].blend=blend; ms->tex_list[tex_index].op=op; ms->tex_list[tex_index].wrap_mode[0]=vk_wrap[wrap_mode[0]]; ms->tex_list[tex_index].wrap_mode[1]=vk_wrap[wrap_mode[1]]; ++tex_index; } } ms->two_sided =GetMaterialBool (mtl,AI_MATKEY_TWOSIDED, true); ms->shading_model =GetMaterialInteger (mtl,AI_MATKEY_SHADING_MODEL, aiShadingMode_Phong); ms->wireframe =GetMaterialBool (mtl,AI_MATKEY_ENABLE_WIREFRAME,false); // ms->blend_func = ms->opacity =GetMaterialFloat(mtl,AI_MATKEY_OPACITY, 0); // ms->transparency_factor = ms->bump_scaling =GetMaterialFloat(mtl,AI_MATKEY_BUMPSCALING, 0); ms->shininess =GetMaterialFloat(mtl,AI_MATKEY_SHININESS, 0); ms->shininess_strength =GetMaterialFloat(mtl,AI_MATKEY_SHININESS_STRENGTH, 0); ms->reflectivity =GetMaterialFloat(mtl,AI_MATKEY_REFLECTIVITY, 0); ms->refracti =GetMaterialFloat(mtl,AI_MATKEY_REFRACTI, 0); ms->diffuse =GetMaterialColor(mtl,AI_MATKEY_COLOR_DIFFUSE, pure_white_color); ms->ambient =GetMaterialColor(mtl,AI_MATKEY_COLOR_AMBIENT, Color4f(0.2,0.2,0.2,1.0)); ms->specular =GetMaterialColor(mtl,AI_MATKEY_COLOR_SPECULAR, pure_black_color); ms->emission =GetMaterialColor(mtl,AI_MATKEY_COLOR_EMISSIVE, pure_black_color); ms->transparent =GetMaterialColor(mtl,AI_MATKEY_COLOR_TRANSPARENT, pure_black_color); ms->reflective =GetMaterialColor(mtl,AI_MATKEY_COLOR_REFLECTIVE, pure_black_color); OutMaterial(ms,main_filename+OS_TEXT(".")+OSString(m)+OS_TEXT(".material")); LOG_BR; } } void AssimpLoader::SaveTextures() { const int tex_count=tex_list.GetCount(); LOG_INFO(OS_TEXT("TOTAL Texture Count: ")+OSString(tex_count)); if(tex_count<=0) return; for(int i=0;i void AssimpLoader::SaveFaces(io::FileOutputStream *fos,const aiFace *face,const T count) { T *data=new T[count*3]; const aiFace *sp=face; T *tp=data; for(T i=0;imIndices[0];++tp; *tp=sp->mIndices[1];++tp; *tp=sp->mIndices[2];++tp; ++sp; } fos->WriteFully(data,sizeof(T)*3*count); delete[] data; } void AssimpLoader::SaveTexCoord(float *tp,const aiVector3D *texcoord,const uint comp,const uint count) { if(comp<=0||comp>3) return; const uint size=1+sizeof(float)*comp*count; const aiVector3D *sp=texcoord; if(comp==1) { for(uint i=0;ix;++tp; ++sp; } } else if(comp==2) { for(uint i=0;ix;++tp; *tp=sp->y;++tp; ++sp; } } else if(comp==3) { for(uint i=0;ix;++tp; *tp=sp->y;++tp; *tp=sp->z;++tp; ++sp; } } } void c4f_to_4b(uint8 *tp,const aiColor4D *c4,const int count) { for(int i=0;ir*255;++tp; *tp=c4->g*255;++tp; *tp=c4->b*255;++tp; *tp=c4->a*255;++tp; } } void AxisY2Z(aiVector3D *v,const uint32 count) { ai_real t; for(uint32 i=0;iy; v->y=-(v->z); v->z=t; ++v; } } void AssimpLoader::LoadMesh() { if(!scene->HasMeshes())return; UTF8String mesh_name; total_file_bytes=0; LOG_BR; LOG_INFO(OS_TEXT("Mesh Count ")+OSString(scene->mNumMeshes)); uint vertex_total=0; uint tri_total=0; uint bone_total=0; for(unsigned int i=0;imNumMeshes;i++) { const aiMesh *mesh=scene->mMeshes[i]; const uint v3fdata_size=mesh->mNumVertices*3*sizeof(float); const uint v4fdata_size=mesh->mNumVertices*4*sizeof(float); if(mesh->mName.length) mesh_name=U8_TEXT("Mesh [")+UTF8String(i)+U8_TEXT(":")+UTF8String(mesh->mName.C_Str())+U8_TEXT("]"); else mesh_name=U8_TEXT("Mesh [")+UTF8String(i)+U8_TEXT("]"); int pn=mesh->mFaces[0].mNumIndices; LOG_BR; LOG_INFO(mesh_name+U8_TEXT(" PrimitiveTypes is ")+UTF8String(mesh->mPrimitiveTypes)); LOG_INFO(mesh_name+U8_TEXT(" vertex color count is ")+UTF8String(mesh->GetNumColorChannels())); LOG_INFO(mesh_name+U8_TEXT(" texcoord count is ")+UTF8String(mesh->GetNumUVChannels())); if(mesh->GetNumUVChannels()>0) LOG_INFO(mesh_name+U8_TEXT(" Material Index is ")+UTF8String(mesh->mMaterialIndex)); if(mesh->HasPositions())LOG_INFO(mesh_name+U8_TEXT(" have Position, vertices count ")+UTF8String(mesh->mNumVertices)); if(mesh->HasFaces())LOG_INFO(mesh_name+U8_TEXT(" have Faces,faces count ")+UTF8String(mesh->mNumFaces)); if(mesh->HasNormals())LOG_INFO(mesh_name+U8_TEXT(" have Normals")); if(mesh->HasTangentsAndBitangents())LOG_INFO(mesh_name+U8_TEXT(" have Tangent & Bitangents")); if(mesh->HasBones())LOG_INFO(mesh_name+U8_TEXT(" have Bones, Bones Count ")+UTF8String(mesh->mNumBones)); vertex_total+=mesh->mNumVertices; tri_total+=mesh->mNumFaces; bone_total+=mesh->mNumBones; LOG_INFO(mesh_name+U8_TEXT(" PN is ")+UTF8String(pn)); if(pn!=3) continue; MaterialData *mtl=&(material_list[mesh->mMaterialIndex]); const uint uv_channels=mtl->uv_use.GetCount(); LOG_INFO(mesh_name+U8_TEXT(" use UV Channels is ")+UTF8String(uv_channels)); io::FileOutputStream fos; OSString mesh_filename=main_filename+OS_TEXT(".")+OSString(i)+OS_TEXT(".mesh"); if(!fos.CreateTrunc(mesh_filename)) { LOG_INFO(OS_TEXT("Create Mesh file Failed,filename: ")+mesh_filename); continue; } { MeshFileHeader ms; ms.primitive_type =PRIM_TRIANGLES; ms.vertices_number =mesh->mNumVertices; ms.faces_number =mesh->mNumFaces; ms.color_channels =mesh->GetNumColorChannels(); ms.texcoord_channels=uv_channels; ms.material_index =mesh->mMaterialIndex; if(mesh->HasNormals()) { if(mesh->HasTangentsAndBitangents()) ms.ntb=NTB_BIT_ALL; else ms.ntb=NTB_BIT_NORMAL; } ms.bones_number =mesh->mNumBones; fos.WriteFully(&ms,sizeof(MeshFileHeader)); } if(mesh->HasPositions()) { AxisY2Z(mesh->mVertices,mesh->mNumVertices); fos.WriteFully(mesh->mVertices,v3fdata_size); } if(mesh->HasNormals()) { AxisY2Z(mesh->mNormals,mesh->mNumVertices); fos.WriteFully(mesh->mNormals,v3fdata_size); if(mesh->HasTangentsAndBitangents()) { AxisY2Z(mesh->mTangents,mesh->mNumVertices); AxisY2Z(mesh->mBitangents,mesh->mNumVertices); fos.WriteFully(mesh->mTangents,v3fdata_size); fos.WriteFully(mesh->mBitangents,v3fdata_size); } } if(mesh->GetNumColorChannels()>0) { const int64 vertex_color_size=mesh->mNumVertices*4*mesh->GetNumColorChannels(); uint8 *vertex_color=new uint8[vertex_color_size]; uint8 *tp=vertex_color; for(int c=0;cGetNumColorChannels();c++) { c4f_to_4b(tp,mesh->mColors[c],mesh->mNumVertices); tp+=mesh->mNumVertices*4; } fos.WriteFully(vertex_color,vertex_color_size); delete[] vertex_color; } if(mtl->uv_use.GetCount()>0 &&mesh->GetNumUVChannels()) { int tc=0; int comp_total=0; uint *uv_use=mtl->uv_use.GetData(); //这里要重新审视数据来源,并不是每一个纹理通道都有数据,并且和材质对应。 //、、材质中的uv index具体对应啥 还不是很清楚 for(int c=0;c=mesh->GetNumUVChannels()) uv_use[c]=0; comp_total+=mesh->mNumUVComponents[uv_use[c]]; } comp_total*=mesh->mNumVertices; uint64 tc_size=mesh->GetNumUVChannels()+comp_total*sizeof(float); uint8 *tc_data=new uint8[tc_size]; float *tp=(float *)(tc_data+mesh->GetNumUVChannels()); for(int c=0;cmNumUVComponents[uv_use[c]]; SaveTexCoord(tp,mesh->mTextureCoords[c],mesh->mNumUVComponents[c],mesh->mNumVertices); tp+=mesh->mNumUVComponents[c]*mesh->mNumVertices; } fos.WriteFully(tc_data,tc_size); delete[] tc_data; } if(mesh->HasFaces()) { if(mesh->mNumVertices>0xFFFF) SaveFaces(&fos,mesh->mFaces,mesh->mNumFaces); else SaveFaces(&fos,mesh->mFaces,mesh->mNumFaces); } fos.Close(); } LOG_BR; LOG_INFO(OS_TEXT("TOTAL Vertices Count ")+OSString(vertex_total)); LOG_INFO(OS_TEXT("TOTAL Faces Count ")+OSString(tri_total)); LOG_INFO(OS_TEXT("TOTAL Bones Count ")+OSString(bone_total)); LOG_INFO(OS_TEXT("TOTAL File Size ")+OSString(total_file_bytes)); LOG_BR; } void AssimpLoader::LoadScene(const UTF8String &front,io::DataOutputStream *dos,const aiNode *nd) { aiVector3D bb_min,bb_max; float box[6]; get_bounding_box(nd,&bb_min,&bb_max); box[0]=bb_min.x;box[1]=bb_min.y;box[2]=bb_min.z; box[3]=bb_max.x;box[4]=bb_max.y;box[5]=bb_max.z; dos->WriteUTF8String(nd->mName.C_Str()); dos->WriteFloat(box,6); dos->WriteFloat((float *)&(nd->mTransformation),16); dos->WriteUint32(nd->mNumMeshes); LOG_INFO(front+U8_TEXT("Node[")+UTF8String(nd->mName.C_Str())+U8_TEXT("][Mesh:")+UTF8String(nd->mNumMeshes)+U8_TEXT("][SubNode:")+UTF8String(nd->mNumChildren)+U8_TEXT("]")) const UTF8String new_front=U8_TEXT(" ")+front; if(nd->mNumMeshes>0) { dos->WriteUint32(nd->mMeshes,nd->mNumMeshes); for(unsigned int i=0;imNumMeshes;i++) { const aiMesh *mesh=scene->mMeshes[nd->mMeshes[i]]; LOG_INFO(new_front+U8_TEXT("Mesh[")+UTF8String(i)+U8_TEXT("] ")+UTF8String(mesh->mName.C_Str())); } } dos->WriteUint32(nd->mNumChildren); for(unsigned int n=0;nmNumChildren;++n) LoadScene(new_front,dos,nd->mChildren[n]); } bool AssimpLoader::LoadFile(const OSString &filename) { uint filesize; char *filedata; filesize=filesystem::LoadFileToMemory(filename,(void **)&filedata); if(!filedata) return(false); #ifdef _WIN32 const UTF8String ext_name=ToUTF8String(filesystem::ClipFileExtName(filename)); #else const UTF8String ext_name=filesystem::ClipFileExtName(filename); #endif// scene=aiImportFileFromMemory(filedata,filesize,aiProcessPreset_TargetRealtime_MaxQuality|aiProcess_FlipUVs,ext_name.c_str()); delete[] filedata; if(!scene) return(false); filename.SubString(main_filename,0,filename.FindRightChar(OS_TEXT('.'))); get_bounding_box(scene->mRootNode,&scene_min,&scene_max); scene_center.x = (scene_min.x + scene_max.x) / 2.0f; scene_center.y = (scene_min.y + scene_max.y) / 2.0f; scene_center.z = (scene_min.z + scene_max.z) / 2.0f; //这个顺序是假设了所有的材质和模型都有被使用 //aiProcessPreset_TargetRealtime_Quality已经指定了会删除多余材质,所以这里不用处理。 //但多余模型并不确定是否存在 io::FileOutputStream fos; io::LEDataOutputStream dos(&fos); OSString scene_filename=main_filename+OS_TEXT(".scene"); if(!fos.CreateTrunc(scene_filename)) { LOG_INFO(OS_TEXT("Create Scene file Failed,filename: ")+scene_filename); return(false); } dos.WriteFully("SCENE\x1a\x01",7); dos.WriteUint32(scene->mNumMaterials); dos.WriteUint32(scene->mNumMeshes); LoadMaterial(); //载入所有材质 dos.WriteUint32(tex_list.GetCount()); LoadMesh(); //载入所有mesh LoadScene(UTF8String(""),&dos,scene->mRootNode); //载入场景节点 fos.Close(); SaveTextures(); return(true); }