GU_PackedUSD.h

Go to the documentation of this file.

//
// Copyright 2017 Pixar
//
// Licensed under the Apache License, Version 2.0 (the "Apache License")
// with the following modification; you may not use this file except in
// compliance with the Apache License and the following modification to it:
// Section 6. Trademarks. is deleted and replaced with:
//
// 6. Trademarks. This License does not grant permission to use the trade
//    names, trademarks, service marks, or product names of the Licensor
//    and its affiliates, except as required to comply with Section 4(c) of
//    the License and to reproduce the content of the NOTICE file.
//
// You may obtain a copy of the Apache License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the Apache License with the above modification is
// distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied. See the Apache License for the specific
// language governing permissions and limitations under the Apache License.
//

#ifndef __GUSD_GU_PACKEDIMPL_H__
#define __GUSD_GU_PACKEDIMPL_H__


#include <GU/GU_PackedImpl.h>
#include <GT/GT_Handles.h>
#include <UT/UT_Error.h>

#include <pxr/pxr.h>
#include "pxr/usd/usd/prim.h"
#include "pxr/usd/usdGeom/imageable.h"
#include "gusd/api.h"
#include "gusd/purpose.h"
#include "gusd/stageEdit.h"
#include "gusd/USD_Utils.h"

class GusdPrimDef;
class GU_PrimPacked;
class GT_RefineParms;

PXR_NAMESPACE_OPEN_SCOPE

/// A GU implementation of a packed USD prim. 
///
/// This is a file backed prim that holds a reference to a prim in a usd file
/// at a particular frame. The prim can be a group of prims.
///
/// When a packed prim that references a USD group is unpacked, the result is
/// packed prims that represent the contents of that group. Those packed prims
/// may also be groups. To unpack down to the leafs, you may have to unpack may
/// times.
///
/// When we write USD packed prim to a USD file, we write a reference to the 
/// original file. USD has a limitation that it can only make references to root
/// nodes. 
///
/// When we write a reference into a USD file, we might want to use a different
/// file path than we use in the Houdini session. For example, we might want to
/// use a relative path vs. an absolute one. We might want to use a coalesced 
/// file vs per frame files. We don't want to enshrine this logic in the core so
/// we provide a second alternative file name that can be used for this. If
/// this fileName is left empty, we just us the primary file name.
///

typedef void (*GusdPackedUSDTracker)(const GU_PackedImpl *prim, bool create);

#if defined(WIN32)
class GusdGU_PackedUSD : public GU_PackedImpl
#else
class GUSD_API GusdGU_PackedUSD : public GU_PackedImpl
#endif
{
public:
    enum class PivotLocation : int
    {
        Origin = 0,
        Centroid
    };

    static GU_PrimPacked* Build( 
                            GU_Detail&              detail,
                            const UT_StringHolder&  fileName, 
                            const SdfPath&          primPath, 
                            UsdTimeCode             frame, 
                            const char*             lod = nullptr,
                            GusdPurposeSet          purposes = GUSD_PURPOSE_PROXY,
                            const UsdPrim&          prim = UsdPrim(),
                            const UT_Matrix4D*      xform = nullptr,
                            PivotLocation           pivotloc = PivotLocation::Origin);

    static GU_PrimPacked* Build( 
                            GU_Detail&              detail,
                            const UT_StringHolder&  fileName, 
                            const SdfPath&          primPath, 
                            const SdfPath&          srcPrimPath, 
                            int                     index,
                            UsdTimeCode             frame, 
                            const char*             lod = nullptr,
                            GusdPurposeSet          purposes = GUSD_PURPOSE_PROXY,
                            const UsdPrim&          prim = UsdPrim(),
                            const UT_Matrix4D*      xform = nullptr,
                            PivotLocation           pivotloc = PivotLocation::Origin);

    /// Convenience method for building a packed USD prim for \p prim.
    static GU_PrimPacked* Build(
                            GU_Detail&              detail,
                            const UsdPrim&          prim,
                            UsdTimeCode             frame,
                            const char*             lod = nullptr,
                            GusdPurposeSet          purpose = GUSD_PURPOSE_PROXY,
                            const UT_Matrix4D*      xform = nullptr,
                            PivotLocation           pivotloc = PivotLocation::Origin);

    GusdGU_PackedUSD();
    GusdGU_PackedUSD(const GusdGU_PackedUSD &src );
    ~GusdGU_PackedUSD() override;

    static void install(GA_PrimitiveFactory &factory);

    /// Get the type ID for the packed USD primitive type.
#if defined(WIN32)
    GUSD_API
#endif
    static GA_PrimitiveTypeId typeId() { return theTypeId; }

    /// Sets a static callback that is used for tracking all packed USD prims.
    /// This callback method lives in HUSD_LockedStageRegistry. It holds on to
    /// an HUSD_LockedStagePtr for each GU_PackedUSD that exists. When the
    /// packed prims are all destroyed, the locked stage can be released.
#if defined(WIN32)
    GUSD_API
#endif
    static void setPackedUSDTracker(GusdPackedUSDTracker tracker);

    const UT_StringHolder& fileName() const { return m_fileName; }
    UT_StringHolder intrinsicFileName() const { return m_fileName; }
    UT_StringHolder intrinsicFileName(const GU_PrimPacked *prim) const
    { return intrinsicFileName(); }
    void setFileName(GU_PrimPacked *prim, const UT_StringHolder& fileName);

    const UT_StringHolder& altFileName() const { return m_altFileName; }
    UT_StringHolder intrinsicAltFileName() const { return m_altFileName; }
    void setAltFileName( const UT_StringHolder& fileName );
    UT_StringHolder intrinsicAltFileName(const GU_PrimPacked *prim) const
    { return intrinsicAltFileName(); }
    void setAltFileName(GU_PrimPacked *prim, const UT_StringHolder& fileName)
    { setAltFileName(fileName); }

    const SdfPath& primPath() const { return m_primPath; }
    UT_StringHolder intrinsicPrimPath() const { return m_primPath.GetText(); }
    UT_StringHolder intrinsicPrimPath(const GU_PrimPacked *prim) const
    { return intrinsicPrimPath(); }
    void setPrimPath(GU_PrimPacked *prim, const UT_StringHolder& p);
    void setPrimPath(GU_PrimPacked *prim, const SdfPath& p);

    // If this prim was unpacked from a point instancer, srcPrimPath is the path
    // to the instancer.
    const SdfPath& srcPrimPath() const { return m_srcPrimPath; }
    UT_StringHolder intrinsicSrcPrimPath() const { return m_srcPrimPath.GetText(); }
    void setSrcPrimPath( const UT_StringHolder& p );
    void setSrcPrimPath( const SdfPath& primPath  );
    UT_StringHolder intrinsicSrcPrimPath(const GU_PrimPacked *prim) const
    { return intrinsicSrcPrimPath(); }
    void setSrcPrimPath(GU_PrimPacked *prim, const UT_StringHolder& p)
    { setSrcPrimPath(p); }

    // If this prim was unpacked from a point instancer, index is the array 
    // index in the source point instancer.
    exint index() const { return m_index; }
    void setIndex( exint i );
    exint index(const GU_PrimPacked *prim) const
    { return index(); }
    void setIndex(GU_PrimPacked *prim, exint i)
    { setIndex(i); }

    // Return true if this is a prim that has been unpacked from a point instancer.
    bool isPointInstance() const { return m_index >= 0; }
    
    // return the USD prim type
    UT_StringHolder intrinsicType() const;
    UT_StringHolder intrinsicType(const GU_PrimPacked *prim) const
    { return intrinsicType(); }

    GA_Size usdLocalToWorldTransformSize() const { return 16; }
    void usdLocalToWorldTransform(fpreal64* val, exint size) const;
    GA_Size usdLocalToWorldTransformSize(const GU_PrimPacked *prim) const
    { return 16; }
    void usdLocalToWorldTransform(const GU_PrimPacked *prim,
            fpreal64* val, exint size) const
    { usdLocalToWorldTransform(val, size); }

    UsdTimeCode frame() const { return m_frame; }
    fpreal intrinsicFrame() const { return GusdUSD_Utils::GetNumericTime(m_frame); }
    fpreal intrinsicFrame(const GU_PrimPacked *prim) const
    { return intrinsicFrame(); }
    void setFrame(GU_PrimPacked *prim, fpreal frame);
#if defined(WIN32)
    GUSD_API
#endif
    void setFrame(GU_PrimPacked *prim, UsdTimeCode frame);

    GusdPurposeSet getPurposes() const { return m_purposes; }
    void setPurposes( GU_PrimPacked *prim, GusdPurposeSet purposes );

    /// Returns the purpose of the USD prim.
    UT_StringHolder intrinsicPurpose(const GU_PrimPacked *prim) const;

    /// @{
    /// Returns the viewport purpose(s) which are used for traversals when
    /// displaying or unpacking the prim.
    exint getNumViewportPurposes() const;
    void getIntrinsicViewportPurposes( UT_StringArray& purposes ) const;
    exint getNumViewportPurposes(const GU_PrimPacked *prim) const
    { return getNumViewportPurposes(); }
    void getIntrinsicViewportPurposes(const GU_PrimPacked *prim,
            UT_StringArray& purposes ) const
    { getIntrinsicViewportPurposes(purposes); }
    void setIntrinsicViewportPurposes(GU_PrimPacked *prim,
        const UT_StringArray& purposes );
    /// @}

    GU_PackedFactory *getFactory() const override;
    GU_PackedImpl *copy() const override;
    void clearData() override;

    bool isValid() const override;
    bool save(UT_Options &options, const GA_SaveMap &map) const override;
    bool load(
        GU_PrimPacked *prim,
        const UT_Options &options,
        const GA_LoadMap &map) override;
    void update(
        GU_PrimPacked *prim,
        const UT_Options &options) override;

    bool getBounds(UT_BoundingBox &box) const override;
    bool getRenderingBounds(UT_BoundingBox &box) const override;
    void getVelocityRange(UT_Vector3 &min, UT_Vector3 &max) const override;
    void getWidthRange(fpreal &min, fpreal &max) const override;

    bool getLocalTransform(UT_Matrix4D &m) const override;

    bool unpack(GU_Detail &destgdp,
                const UT_Matrix4D *transform) const override;
    bool unpackUsingPolygons(GU_Detail &destgdp,
                             const GU_PrimPacked *prim) const override;

protected:
    /// This signature is just for the questionable purpose of copying
    /// primitive group membership from prim, so it might be removed
    /// in the future.
    bool unpackWithPrim(
        GU_Detail& destgdp,
        const UT_Matrix4D* transform,
        const GU_PrimPacked* prim) const override;

public:
    bool visibleGT() const;   
    GT_PrimitiveHandle fullGT() const;

    // Return a structure that can be hashed to sort instances by prototype.
    bool getInstanceKey(UT_Options& key) const;
    
    /// Report memory usage (includes all shared memory)
    int64 getMemoryUsage(bool inclusive) const override;

    /// Count memory usage using a UT_MemoryCounter in order to count
    /// shared memory correctly.
    void countMemory(UT_MemoryCounter &counter, bool inclusive) const override;

    /// Get the underlying UsdPrim for this packed prim.
    /// This may involve on-demand loading of a UsdStage to access the prim.
    /// Any errors that occur while loading the stage and accessing the prim
    /// will be reported on the currently scoped error manager with a severity
    /// of \p sev.
    UsdPrim getUsdPrim(UT_ErrorSeverity sev=UT_ERROR_ABORT) const;

    bool unpackGeometry(
        GU_Detail &destgdp,
        const GU_Detail* srcgdp,
        const GA_Offset srcprimoff,
        const UT_StringRef &primvarPattern,
        bool importInheritedPrimvars,
        const UT_StringRef &attributePattern,
        bool translateSTtoUV,
        const UT_StringRef& nonTransformingPrimvarPattern,
        const UT_Matrix4D* transform,
        const GT_RefineParms *parms = nullptr) const;

    /// Convert the USD geometry into one or more GU_Detail's. Use
    /// mergeGeometry() to merge the geometry into a destination detail.
    /// This signature can be used when performing the geometry conversion in
    /// parallel for many prims.
    bool unpackGeometry(UT_Array<GU_DetailHandle> &details,
                        const GU_Detail *srcgdp,
                        const GA_Offset srcprimoff,
                        const UT_StringRef &primvarPattern,
                        bool importInheritedPrimvars,
                        const UT_StringRef &attributePattern,
                        bool translateSTtoUV,
                        const UT_StringRef &nonTransformingPrimvarPattern,
                        const UT_Matrix4D *transform,
                        const UT_StringHolder &filePathAttrib,
                        const UT_StringHolder &primPathAttrib,
                        const GT_RefineParms *refineParms = nullptr) const;

    /// Merges the details together, and also updates the
    /// usdconfigconstantattribs detail attribute.
    static void mergeGeometry(GU_Detail &destgdp,
                              UT_Array<GU_DetailHandle> &details);

    const UT_Matrix4D& getUsdTransform() const;
    
private:
    bool unpackPrim(
            UT_Array<GU_DetailHandle> &details,
            const GU_Detail*        srcgdp,
            const GA_Offset         srcprimoff,
            UsdGeomImageable        prim,
            const SdfPath&          primPath,
            const UT_Matrix4D*      xform,
            const GT_RefineParms&   rparms ) const;

    void resetCaches();
    void updateTransform( GU_PrimPacked* prim );
    void initializePivot(GU_PrimPacked *prim, PivotLocation pivotloc);

    /// Set the overall world transform. This will set the 'transform'
    /// intrinsic and P so that this transform is produced when combined with
    /// getUsdTransform() ('packedlocaltransform' intrinsic).
    void setTransform( GU_PrimPacked* prim, const UT_Matrix4D& mx );

    // intrinsics
    UT_StringHolder m_fileName;
    UT_StringHolder m_altFileName;
    SdfPath         m_srcPrimPath;
    int             m_index;
    SdfPath         m_primPath;
    UsdTimeCode     m_frame;
    GusdPurposeSet  m_purposes;

    // caches    
    mutable UsdPrim             m_usdPrim;
    mutable bool                m_transformCacheValid;
    mutable UT_Matrix4D         m_transformCache;
    mutable GT_PrimitiveHandle  m_gtPrimCache;
    mutable bool                m_prototypePathCacheValid;
    mutable std::string         m_prototypePathCache;

    // static
    static GusdPackedUSDTracker thePackedUSDTracker;
    static GA_PrimitiveTypeId theTypeId;
};

PXR_NAMESPACE_CLOSE_SCOPE

#endif // __GUSD_GU_PACKEDIMPL_H__