XUSD_Data.h

Go to the documentation of this file.

/*
 * Copyright 2019 Side Effects Software Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 */

#ifndef __HUSD_Data_h__
#define __HUSD_Data_h__

#include "HUSD_API.h"
#include "HUSD_DataHandle.h"
#include "HUSD_PostLayers.h"
#include "HUSD_Overrides.h"
#include "XUSD_DataLock.h"
#include "XUSD_LockedGeo.h"
#include "XUSD_PathSet.h"
#include "XUSD_RootLayerData.h"
#include <UT/UT_Color.h>
#include <UT/UT_StringArray.h>
#include <UT/UT_StringHolder.h>
#include <UT/UT_StringSet.h>
#include <UT/UT_IntrusivePtr.h>
#include <UT/UT_SharedPtr.h>
#include <UT/UT_UniquePtr.h>
#include <pxr/usd/usd/stage.h>
#include <pxr/usd/usd/stageLoadRules.h>
#include <pxr/usd/usd/stagePopulationMask.h>
#include <pxr/usd/sdf/path.h>

PXR_NAMESPACE_OPEN_SCOPE

// Control the addLayers method. Most of these values only affect the version
// of addLayer which takes a vector of file paths (instead of layers pointers).
enum XUSD_AddLayerOp
{
    // Simply add the layers "as-is" to the root layer's list of sublayers,
    // using the layer's identifier or the provided file paths. The data
    // handle's active layer is set to be created after all the added layers.
    XUSD_ADD_LAYERS_ALL_LOCKED,
    // Each layer is copied into an anonymous layer, and these anonymous
    // layers get added to the sublayer list. This means these layers will
    // we saved as new USD files during a save process. The last layer becomes
    // the data handle's active layer for following edits.
    XUSD_ADD_LAYERS_ALL_EDITABLE,
    // All layers are added "as-is", except the last layer in the list, which
    // is copied to an anonymous layer. This last anonymous layer becomes the
    // data handle's active layer for following edits.
    XUSD_ADD_LAYERS_LAST_EDITABLE,

    // Used only by HUSD_Stitch, layers authored by LOP nodes all get copied
    // to new anonymous layers (so they can be safely modified when stitching
    // in the next time sample of data, if there is one.). If the last layer
    // is a LOP layer, it becomes the active layer modified by following LOP
    // nodes.
    XUSD_ADD_LAYERS_ALL_ANONYMOUS_EDITABLE,
    // Used only by HUSD_Stitch, if the last layer was authored by LOP nodes,
    // it gets copied to a new anonymous layer (so it can be modified by
    // following LOP nodes as the active layer). Preceding layers are not
    // copied, which is fine if this is the last time sample that will be
    // stitched into this data handle.
    XUSD_ADD_LAYERS_LAST_ANONYMOUS_EDITABLE,
};

class XUSD_LayerAtPath
{
public:
                         XUSD_LayerAtPath();
    explicit             XUSD_LayerAtPath(const SdfLayerRefPtr &layer,
                                const SdfLayerOffset &offset = SdfLayerOffset(),
                                int layer_badge_index = 0);
    explicit             XUSD_LayerAtPath(const SdfLayerRefPtr &layer,
                                const std::string &filepath,
                                const SdfLayerOffset &offset = SdfLayerOffset(),
                                int layer_badge_index = 0);

    bool                 hasLayerColorIndex(int &clridx) const;
    bool                 isLopLayer() const;

    SdfLayerRefPtr       myLayer;
    std::string          myIdentifier;
    SdfLayerOffset       myOffset;
    int                  myLayerColorIndex;
    bool                 myRemoveWithLayerBreak;
    bool                 myLayerIsMissingFile;
};

typedef UT_Array<XUSD_LayerAtPath>               XUSD_LayerAtPathArray;

class XUSD_OverridesInfo
{
public:
                         XUSD_OverridesInfo(const UsdStageRefPtr &stage);
                        ~XUSD_OverridesInfo();

    bool                 isEmpty() const
                         { return !myReadOverrides && !myWriteOverrides; }

    HUSD_ConstOverridesPtr       myReadOverrides;
    HUSD_OverridesPtr            myWriteOverrides;
    SdfLayerRefPtr               mySessionLayers[HUSD_OVERRIDES_NUM_LAYERS];
    exint                        myVersionId;
};

class XUSD_PostLayersInfo
{
public:
                         XUSD_PostLayersInfo(const UsdStageRefPtr &stage);
                        ~XUSD_PostLayersInfo();

    bool                 isEmpty() const
                         { return !myPostLayers; }

    HUSD_ConstPostLayersPtr      myPostLayers;
    SdfLayerRefPtrVector         mySessionLayers;
    exint                        myVersionId;
};

class HUSD_API XUSD_Data : public UT_IntrusiveRefCounter<XUSD_Data>,
                           public UT_NonCopyable
{
public:
                                 XUSD_Data(HUSD_MirroringType mirroring);
    explicit                     XUSD_Data(const UsdStageRefPtr &stage);
                                ~XUSD_Data();

    // Return true if our stage value is set and has a valid root prim.
    bool                         isStageValid() const;
    // Returns our stage. Should only be accessed when this data is locked.
    UsdStageRefPtr               stage() const;
    // Returns the active layer on our composed stage.
    SdfLayerRefPtr               activeLayer() const;
    // Return true if the active layer of this data will be returned as the
    // active layer when a downstream copy of this data is locked for writing.
    bool                         activeLayerIsReusable() const;
    // Return the on-stage identifiers of any layers that are marked in the
    // source layers array to be removed due to a layer break.
    std::set<std::string>        getStageLayersToRemoveFromLayerBreak() const;
    // Creates a layer by flattening all our source layers together. Also
    // strips out any layers tagged by a Layer Break LOP.
    SdfLayerRefPtr               createFlattenedLayer(
                                        HUSD_StripLayerResponse response) const;
    // Creates a layer by flattening a stage consisting of our source layers
    // after stripping out any layers tagged by a Layer Break LOP.
    SdfLayerRefPtr               createFlattenedStage(
                                        HUSD_StripLayerResponse response) const;

    // Return the array of source layers that are combined to make our stage.
    const XUSD_LayerAtPathArray &sourceLayers() const;
    // Return the current session layer overrides set on our stage.
    const HUSD_ConstOverridesPtr&overrides() const;
    // Return the current session layer post layers set on our stage.
    const HUSD_ConstPostLayersPtr&postLayers() const;
    // Return a specific session layer object on our stage.
    const SdfLayerRefPtr        &sessionLayer(HUSD_OverridesLayerId id) const;
    // Return the current load masks set on our stage.
    const HUSD_LoadMasksPtr     &loadMasks() const;
    // Return the identifier of our stage's root layer. This can be used as
    // a quick check as to whether we have created a brand new stage.
    const std::string           &rootLayerIdentifier() const;

    // Add a layer using a file path, layer pointer, or an empty layer.
    // Files and layer pointers can be inserted at any position in the
    // sublayer list (0 is strongest, -1 is weakest). The layer op value
    // indicates whether the layer should be editable, which implies that
    // it must be added to the strongest position (the only place where a
    // layer can be edited).
    bool                         addLayer(const std::string &filepath,
                                        const SdfLayerOffset &offset,
                                        int position,
                                        XUSD_AddLayerOp add_layer_op,
                                        bool copy_root_prim_metadata);
    bool                         addLayer(const XUSD_LayerAtPath &layer,
                                        int position,
                                        XUSD_AddLayerOp add_layer_op,
                                        bool copy_root_prim_metadata);
    // Methods for adding a bunch of layers to our stage in one call. These
    // methods only unlock and re-lock the data once (meaning only a single
    // recomposition is required). The resulting behavior of this call is the
    // same as calling addLayer on each individual layer in order.
    bool                         addLayers(
                                        const std::vector<std::string> &paths,
                                        const std::vector<bool> &above_breaks,
                                        const SdfLayerOffsetVector &offsets,
                                        int position,
                                        XUSD_AddLayerOp add_layer_op,
                                        bool copy_root_prim_metadata);
    bool                         addLayers(
                                        const std::vector<std::string> &paths,
                                        const SdfLayerOffsetVector &offsets,
                                        int position,
                                        XUSD_AddLayerOp add_layer_op,
                                        bool copy_root_prim_metadata);
    bool                         addLayers(const XUSD_LayerAtPathArray &layers,
                                        int position,
                                        XUSD_AddLayerOp add_layer_op,
                                        bool copy_root_prim_metadata);
    // Add a single new empty layer.
    bool                         addLayer();

    // Remove one or more of our source layers.
    bool                         removeLayers(
                                        const std::set<std::string> &filepaths);

    // Used to do something roughyl equivalent to a mirror operation, but on
    // a data handle not meant for mirroring.
    bool                         replaceAllSourceLayers(
                                        const XUSD_LayerAtPathArray &layers,
                                        const XUSD_LockedGeoSet &locked_geos,
                                        const XUSD_LayerSet &held_layers,
                                        const XUSD_LayerSet &replacement_layers,
                                        const HUSD_LockedStageSet &locked_stages,
                                        const UT_SharedPtr<XUSD_RootLayerData>
                                            &root_layer_data,
                                        bool last_sublayer_is_editable);

    // Apply a layer break, which tags all existing layers, and adds a new
    // empty layer for holding future modification.
    bool                         applyLayerBreak();

    // Set a piece of metadata on the root prim of the root layer of the
    // stage. Also record this change to myRootLayerData so that we can
    // restore this metadata value if we lock this stage to another state
    // then return to this state.
    void                         setStageRootPrimMetadata(const TfToken &field,
                                        const VtValue &value);
    void                         setStageRootLayerData(
                                        const UT_SharedPtr<XUSD_RootLayerData>
                                            &rootlayerdata);
    void                         setStageRootLayerData(
                                        const SdfLayerRefPtr &layer);

    // Store a lockedgeo in with this data to keep alive cooked sop data in the
    // XUSD_LockedGeoRegistry as long as it might be referenced by our stage.
    void                         addLockedGeo(
                                        const XUSD_LockedGeoPtr &lockedgeo);
    void                         addLockedGeos(
                                        const XUSD_LockedGeoSet &lockedgeos);
    const XUSD_LockedGeoSet     &lockedGeos() const;

    // Store a reference to a layer with this data, to keep it alive.
    // Anonymous layers can be dropped by the stage in certain circumstances
    // if there are no external references to them. 
    void                         addHeldLayer(
                                        const SdfLayerRefPtr &layer);
    void                         addHeldLayers(
                                        const XUSD_LayerSet &layers);
    const XUSD_LayerSet         &heldLayers() const;

    // Store pointers to arrays that were created automatically as part of a
    // process of replacing a layer on disk with an anonymous layer.
    void                         addReplacements(
                                        const XUSD_LayerSet &replacements);
    const XUSD_LayerSet         &replacements() const;

    // Store a locked stage with this data to keep alive cooked lop data in the
    // HUSD_LockedStageRegistry as long as it might be referenced by our stage.
    void                         addLockedStage(
                                        const HUSD_LockedStagePtr &stage);
    void                         addLockedStages(
                                        const HUSD_LockedStageSet &stages);
    const HUSD_LockedStageSet   &lockedStages() const;

    // Clear out all mirrored stages completely. They must be rebuilt from
    // scratch the next time they are asked to mirror a stage.
    static void                  clearAllMirroredData();

private:
    void                 reset();
    void                 createNewData(const HUSD_LoadMasksPtr &load_masks,
                                int resolver_context_nodeid,
                                const UsdStageWeakPtr &resolver_context_stage,
                                const ArResolverContext *resolver_context);
    void                 createHardCopy(const XUSD_Data &src);
    void                 createSoftCopy(const XUSD_Data &src,
                                const HUSD_LoadMasksPtr &load_masks,
                                bool make_new_implicit_layer);
    void                 createCopyWithReplacement(
                                const XUSD_Data &src,
                                const UT_StringRef &frompath,
                                const UT_StringRef &topath,
                                int nodeid,
                                HUSD_MakeNewPathFunc make_new_path,
                                UT_StringSet &replaced_layers);

    // Return the resolver context for our stage. Note that this method does
    // not require that the stage be locked, because it is unchanging data set
    // on the stage when it was created.
    ArResolverContext    resolverContext() const;

    // Return a layer created by flattening all source layers up to the
    // strongest layer excluded by a Layer Break node.
    void                 flattenLayers(const XUSD_Data &src,
                                int creator_node_id);
    // Return a layer created by flattening the stage composed of all source
    // layers up to the strongest layer excluded by a Layer Break node.
    void                 flattenStage(const XUSD_Data &src,
                                int creator_node_id);
    // Utility method used by the two flatten methods above to separate all
    // layers preceding a layer break point. Optionally creates a warning if
    // a layer break is found.
    UsdStageRefPtr       getOrCreateStageForFlattening(
                                HUSD_StripLayerResponse response,
                                UsdStage::InitialLoadSet loadset) const;

    void                 mirror(const XUSD_Data &src,
                                const HUSD_LoadMasks &load_masks);
    bool                 mirrorUpdateRootLayer(
                                const HUSD_MirrorRootLayer &rootlayer);

    void                 afterLock(bool for_write,
                                const HUSD_ConstOverridesPtr
                                    &read_overrides = HUSD_ConstOverridesPtr(),
                                const HUSD_OverridesPtr
                                    &write_overrides = HUSD_OverridesPtr(),
                                const HUSD_ConstPostLayersPtr
                                    &postlayers = HUSD_ConstPostLayersPtr(),
                                bool remove_layer_breaks = false);
    XUSD_LayerPtr        editActiveSourceLayer(bool create_change_block);
    void                 createInitialPlaceholderSublayers();
    void                 applyRootLayerDataToStage();
    void                 afterRelease();

    static void          exitCallback(void *);

    UsdStageRefPtr                       myStage;
    UT_SharedPtr<UT_StringArray>         myStageLayerAssignments;
    UT_SharedPtr<XUSD_LayerArray>        myStageLayers;
    UT_SharedPtr<int>                    myStageLayerCount;
    UT_SharedPtr<XUSD_OverridesInfo>     myOverridesInfo;
    UT_SharedPtr<XUSD_PostLayersInfo>    myPostLayersInfo;
    UT_SharedPtr<XUSD_RootLayerData>     myRootLayerData;
    XUSD_LayerAtPathArray                mySourceLayers;
    HUSD_LoadMasksPtr                    myLoadMasks;
    XUSD_DataLockPtr                     myDataLock;
    XUSD_LockedGeoSet                    myLockedGeos;
    XUSD_LayerSet                        myReplacementLayers;
    HUSD_LockedStageSet                  myLockedStages;
    XUSD_LayerSet                        myHeldLayers;
    HUSD_MirroringType                   myMirroring;
    UsdStageLoadRules                    myMirrorLoadRules;
    UT_StringMap<UT_StringArray>         myMirrorVariantSelectionFallbacks;
    bool                                 myMirrorLoadRulesChanged;
    int                                  myActiveLayerIndex;
    bool                                 myOwnsActiveLayer;

    friend class ::HUSD_DataHandle;
};

class HUSD_API XUSD_Layer : public UT_IntrusiveRefCounter<XUSD_Layer>
{
public:
                                 XUSD_Layer(const SdfLayerRefPtr &layer,
                                         bool create_change_block)
                                     : myLayer(layer),
                                       myChangeBlock(nullptr)
                                 {
                                     if (create_change_block)
                                         myChangeBlock.reset(
                                             new SdfChangeBlock());
                                 }
                                ~XUSD_Layer()
                                 { }

    const SdfLayerRefPtr        &layer() const
                                 { return myLayer; }

private:
    SdfLayerRefPtr               myLayer;
    UT_UniquePtr<SdfChangeBlock> myChangeBlock;
};

PXR_NAMESPACE_CLOSE_SCOPE

#endif