ProceduralMesh转StaticMesh

在ProceduralMeshComponent的细节面板上，有一个按键convert to static mesh，可以把你当前的PMC转换为静态模型。这个function的位置在ProceduralMeshComponentDetails.cpp，使用的是FStaticMeshSourceModel来构建。但是这个函数无法在运行时使用，因为有FStaticMeshSourceModel& AddSourceModel();这个函数有WITH_EDITORONLY_DATA的宏限制。

如果想在运行时把程序化模型转换为静态模型该怎么办呢？这就要使用FMeshDescription了。下面贴一段代码。

UStaticMesh* AProceduralMeshActor::ConvertProceduralMeshToStaticMesh()
{
    // 1. 验证输入参数
    if (!ProceduralMeshComponent)
    {
        return nullptr;
    }

    const int32 NumSections = ProceduralMeshComponent->GetNumSections();
    if (NumSections == 0)
    {
        return nullptr;
    }

    // 2. 创建一个临时的、只存在于内存中的UStaticMesh对象
    UStaticMesh* StaticMesh = NewObject<UStaticMesh>(
        GetTransientPackage(), NAME_None, RF_Public | RF_Standalone);

    if (!StaticMesh)
    {
        return nullptr;
    }

    // 3. 创建并设置 FMeshDescription
    FMeshDescription MeshDescription;
    FStaticMeshAttributes Attributes(MeshDescription);
    Attributes.Register();  // 这会注册包括材质槽名称在内的所有必要属性

    // 获取对多边形组材质槽名称属性的引用，以便后续设置
    // 这是将几何体与材质关联的关键
    TPolygonGroupAttributesRef<FName> PolygonGroupMaterialSlotNames =
        Attributes.GetPolygonGroupMaterialSlotNames();

    FMeshDescriptionBuilder MeshDescBuilder;
    MeshDescBuilder.SetMeshDescription(&MeshDescription);
    MeshDescBuilder.EnablePolyGroups();
    MeshDescBuilder.SetNumUVLayers(1);

    // 4. 从PMC数据填充 FMeshDescription
    TMap<FVector, FVertexID> VertexMap;

    for (int32 SectionIdx = 0; SectionIdx < NumSections; ++SectionIdx)
    {
        FProcMeshSection* SectionData =
            ProceduralMeshComponent->GetProcMeshSection(SectionIdx);
        if (!SectionData || SectionData->ProcVertexBuffer.Num() < 3 ||
            SectionData->ProcIndexBuffer.Num() < 3)
        {
            continue;
        }

        // --- 开始材质设置修正 ---
        // a. 为每个材质段创建一个唯一的槽名称
        const FName MaterialSlotName =
            FName(*FString::Printf(TEXT("MaterialSlot_%d"), SectionIdx));

        // b. 将材质和槽名称添加到StaticMesh的材质列表中
        // 这会在最终的UStaticMesh上创建对应的材质槽
        FStaticMaterial NewStaticMaterial(nullptr, MaterialSlotName);

        // 初始化UV通道数据，避免ensure失败
        NewStaticMaterial.UVChannelData.bInitialized = true;
        NewStaticMaterial.UVChannelData.bOverrideDensities = false;

        // 计算UV密度（基于实际的UV数据）
        // 分析当前section的UV数据来计算合适的密度值
        float UVDensity = 1.0f; // 默认密度
        if (SectionData->ProcVertexBuffer.Num() > 0)
        {
            // 计算UV坐标的范围
            FVector2D MinUV(FLT_MAX, FLT_MAX);
            FVector2D MaxUV(-FLT_MAX, -FLT_MAX);

            for (const FProcMeshVertex& Vertex : SectionData->ProcVertexBuffer)
            {
                MinUV.X = FMath::Min(MinUV.X, Vertex.UV0.X);
                MinUV.Y = FMath::Min(MinUV.Y, Vertex.UV0.Y);
                MaxUV.X = FMath::Max(MaxUV.X, Vertex.UV0.X);
                MaxUV.Y = FMath::Max(MaxUV.Y, Vertex.UV0.Y);
            }

            // 计算UV范围
            FVector2D UVRange = MaxUV - MinUV;
            if (UVRange.X > 0.0f && UVRange.Y > 0.0f)
            {
                // 基于UV范围计算密度，范围越大密度越小
                UVDensity = FMath::Clamp(1.0f / FMath::Max(UVRange.X, UVRange.Y), 0.1f, 10.0f);
            }
        }

        // 设置UV通道密度
        for (int32 UVIndex = 0; UVIndex < 4; ++UVIndex)
        {
            NewStaticMaterial.UVChannelData.LocalUVDensities[UVIndex] = UVDensity;
        }

        StaticMesh->StaticMaterials.Add(NewStaticMaterial);

        // c. 在MeshDescription中为这个材质段创建一个新的多边形组
        const FPolygonGroupID PolygonGroup = MeshDescBuilder.AppendPolygonGroup();

        // d. 将多边形组与我们刚刚创建的材质槽名称关联起来
        // 这是最关键的一步！
        PolygonGroupMaterialSlotNames.Set(PolygonGroup, MaterialSlotName);
        // --- 结束材质设置修正 ---

        TArray<FVertexInstanceID> VertexInstanceIDs;

        for (const FProcMeshVertex& ProcVertex : SectionData->ProcVertexBuffer)
        {
            FVertexID VertexID;

            if (FVertexID* FoundID = VertexMap.Find(ProcVertex.Position))
            {
                VertexID = *FoundID;
            }
            else
            {
                VertexID = MeshDescBuilder.AppendVertex(ProcVertex.Position);
                VertexMap.Add(ProcVertex.Position, VertexID);
            }

            FVertexInstanceID InstanceID = MeshDescBuilder.AppendInstance(VertexID);
            MeshDescBuilder.SetInstanceNormal(InstanceID, ProcVertex.Normal);
            MeshDescBuilder.SetInstanceUV(InstanceID, ProcVertex.UV0, 0);
            MeshDescBuilder.SetInstanceColor(InstanceID, FVector4(ProcVertex.Color));
            VertexInstanceIDs.Add(InstanceID);
        }

        for (int32 i = 0; i < SectionData->ProcIndexBuffer.Num(); i += 3) {
            if (i + 2 >= SectionData->ProcIndexBuffer.Num()) {
                break;
            }

            int32 Index1 = SectionData->ProcIndexBuffer[i + 0];
            int32 Index2 = SectionData->ProcIndexBuffer[i + 1];
            int32 Index3 = SectionData->ProcIndexBuffer[i + 2];

            if (Index1 >= VertexInstanceIDs.Num() ||
                Index2 >= VertexInstanceIDs.Num() ||
                Index3 >= VertexInstanceIDs.Num()) {
                continue;
            }

            FVertexInstanceID V1 = VertexInstanceIDs[Index1];
            FVertexInstanceID V2 = VertexInstanceIDs[Index2];
            FVertexInstanceID V3 = VertexInstanceIDs[Index3];

            // 将三角形添加到指定了材质信息的PolygonGroup中
            MeshDescBuilder.AppendTriangle(V1, V2, V3, PolygonGroup);
        }
    }

    if (MeshDescription.Vertices().Num() == 0) {
        return nullptr;
    }

    // 7. 使用 FMeshDescription 构建 Static Mesh
    TArray<const FMeshDescription*> MeshDescPtrs;
    MeshDescPtrs.Emplace(&MeshDescription);
    UStaticMesh::FBuildMeshDescriptionsParams BuildParams;
    BuildParams.bBuildSimpleCollision = true;  // StaticMesh默认开启碰撞

    StaticMesh->BuildFromMeshDescriptions(MeshDescPtrs, BuildParams);

    // 处理碰撞 - StaticMesh默认开启碰撞
    {
        StaticMesh->CreateBodySetup();
        UBodySetup* NewBodySetup = StaticMesh->BodySetup;
        if (NewBodySetup)
        {
            // 如果PMC有BodySetup，复制其碰撞数据
            if (ProceduralMeshComponent->ProcMeshBodySetup)
            {
                NewBodySetup->AggGeom.ConvexElems =
                    ProceduralMeshComponent->ProcMeshBodySetup->AggGeom.ConvexElems;
            }

            // 设置碰撞参数
            NewBodySetup->bGenerateMirroredCollision = false;
            NewBodySetup->bDoubleSidedGeometry = true;
            NewBodySetup->CollisionTraceFlag = CTF_UseDefault;

            // 创建物理网格
            NewBodySetup->CreatePhysicsMeshes();
        }
    }

    return StaticMesh;
}