usdPmcEncoder.h

Go to the documentation of this file.

//
// Copyright 2026 Apple
//
// Licensed under the terms set forth in the LICENSE.txt file available at
// https://openusd.org/license.
//
 
 
#ifndef USD_PMC_ENCODER_H
#define USD_PMC_ENCODER_H
 
#include "api.h"
 
#include "pxr/pxr.h"
#include "pxr/base/vt/dictionary.h"
#include "pxr/usd/usdGeom/mesh.h"
 
#include <filesystem>
#include <list>
#include <set>
#include <string>
 
PXR_NAMESPACE_OPEN_SCOPE
 
class UsdPmcMeshEncoder {
public:
    USDPMC_API UsdPmcMeshEncoder();
    USDPMC_API ~UsdPmcMeshEncoder();
 
    USDPMC_API
    bool CanEncode(const UsdGeomMesh& mesh);
 
    USDPMC_API
    std::vector<uint8_t> Encode(UsdGeomMesh* mesh,
                                const VtDictionary& options,
                                std::set<std::string>* processedAttributes,
                                std::set<std::string>* processedSubSets,
                                VtDictionary* resultInfo);
 
    USDPMC_API
    bool EncodeStage(std::filesystem::path inFile,
                                std::filesystem::path outFile,
                                const VtDictionary& options);
 
private:
    std::filesystem::path   _inFile;
    std::filesystem::path   _outFile;
    std::filesystem::path   _outRootFile;
    std::list<std::string>  _references;
    std::filesystem::path   _tempDir;
    std::string             _entryFileName;
 
    bool _ExtractUSDZFiles();
 
    bool _ProcessMesh(UsdGeomMesh* mesh, const VtDictionary& options,
                      const uint32_t meshCounter,
                      std::set<std::string>* processedAttributes,
                      std::set<std::string>* processedSubSets,
                      VtDictionary* resultInfo);
 
    bool _RemoveAttributes(UsdGeomMesh* mesh,
                           std::set<std::string>* processedAttributes,
                           std::set<std::string>* processedSubSets);
 
    bool _PackUSDZ();
 
    bool _WriteNonUsdzOutput();
 
    bool _RemoveFlattenedReferences();
 
    bool _CreateFlattenOutput (std::filesystem::path input,
                               std::filesystem::path* output);
};
 
PXR_NAMESPACE_CLOSE_SCOPE
 
#endif // USD_PMC_ENCODER_H