HUSD_PrimHandle.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_PrimHandle_h__
#define __HUSD_PrimHandle_h__

#include "HUSD_API.h"
#include "HUSD_ObjectHandle.h"
#include "HUSD_Utils.h"
#include <UT/UT_StringHolder.h>
#include <UT/UT_ArrayStringSet.h>
#include <UT/UT_Array.h>
#include <UT/UT_Options.h>
#include <SYS/SYS_Inline.h>

enum HUSD_PrimAttribState {
    HUSD_FALSE,
    HUSD_TRUE,
    HUSD_ANIMATED_FALSE,
    HUSD_ANIMATED_TRUE,
    HUSD_OVERRIDDEN_FALSE,
    HUSD_OVERRIDDEN_TRUE,
    HUSD_NOTPOSSIBLE,
    HUSD_NOTAPPLICABLE
};

enum HUSD_SoloState {
    HUSD_SOLO_NOSOLO,
    HUSD_SOLO_FALSE,
    HUSD_SOLO_TRUE,
    HUSD_SOLO_NOTAPPLICABLE
};

enum HUSD_PrimStatus {
    HUSD_PRIM_HASARCS,
    HUSD_PRIM_HASPAYLOAD,
    HUSD_PRIM_INSTANCE,
    HUSD_PRIM_ISINSTANCEPROXY,
    HUSD_PRIM_ISINACTIVE,
    HUSD_PRIM_NORMAL,
    HUSD_PRIM_ROOT,
    HUSD_PRIM_UNKNOWN,
    HUSD_PRIM_INSTANCEPAYLOAD,
    HUSD_PRIM_INSIDEPAYLOAD,
    HUSD_PRIM_ISIMPLICITPROTOTYPE
};

class HUSD_MaterialInfo
{
public:
    UT_StringHolder  myMaterialPath;
    UT_StringHolder  myMaterialName;
    bool             myBindingIsInherited;
};
SYS_FORCE_INLINE bool
HUSDstateAsBool(HUSD_PrimAttribState state)
{
    return (state == HUSD_TRUE ||
            state == HUSD_OVERRIDDEN_TRUE ||
            state == HUSD_ANIMATED_TRUE);
}

SYS_FORCE_INLINE bool
HUSDstateCanChange(HUSD_PrimAttribState state)
{
    return (state != HUSD_NOTPOSSIBLE &&
            state != HUSD_NOTAPPLICABLE);
}

class HUSD_TimeCode;
class HUSD_PropertyHandle;

// This class is a standalone wrapper around a specific primitice in a USD
// stage wrapped in an HUSD_DataHandle. It's purpose is to serve as the data
// accessor for tree nodes in the Scene Graph Tree. It should not be used for
// any other purpose, as it is extremely inefficient. Each function call locks
// the HUSD_DataHandle, queries its information, then unlocks it again. This
// is a matter of convenience for the calling pattern of the scene graph tree.
// Because it is inefficient the scene graph tree caches any information that
// comes out of this object.
//
// Anyone else tempted to use this object should use HUSD_Info instead.
class HUSD_API HUSD_PrimHandle : public HUSD_ObjectHandle
{
public:
                         HUSD_PrimHandle();
                         HUSD_PrimHandle(
                                const HUSD_DataHandle &data_handle,
                                const HUSD_Path &prim_path =
                                    HUSD_Path::theRootPrimPath);
                         HUSD_PrimHandle(
                                const HUSD_DataHandle &data_handle,
                                const HUSD_ConstOverridesPtr &overrides,
                                const HUSD_ConstPostLayersPtr &postlayers,
                                OverridesHandling overrides_handling,
                                const HUSD_Path &prim_path =
                                    HUSD_Path::theRootPrimPath);
                        ~HUSD_PrimHandle() override;

    const HUSD_DataHandle               &dataHandle() const override;
    const HUSD_ConstOverridesPtr        &overrides() const override;
    const HUSD_ConstPostLayersPtr       &postLayers() const override;

    HUSD_PrimStatus      getStatus() const;
    UT_StringHolder      getPrimType() const;
    UT_StringHolder      getVariantInfo() const;
    UT_StringHolder      getKind(bool *kind_is_valid) const;
    UT_StringHolder      getDrawMode(bool *has_override = nullptr) const;
    UT_StringHolder      getPurpose() const;
    HUSD_MaterialInfo    getMaterialInfo() const;
    UT_StringHolder      getProxyPath() const;
    UT_StringHolder      getAncestorPayloadPath() const;
    UT_StringHolder      getSpecifier() const;
    UT_StringHolder      getIcon() const;
    HUSD_PrimAttribState getActive() const;
    HUSD_PrimAttribState getVisible(const HUSD_TimeCode &timecode) const;
    HUSD_PrimAttribState getSelectable() const;
    HUSD_SoloState       getSoloState() const;
    int64                getChildCount(
                            HUSD_PrimTraversalDemands demands) const;
    int64                getDescendantCount(
                            HUSD_PrimTraversalDemands demands) const;
    bool                 hasAnyOverrides() const;
    bool                 hasAnyPostLayers() const;
    bool                 hasPayload() const;
    bool                 isDefined() const;
    bool                 isHiddenInUi() const;
    bool                 isEditable() const;

    bool                 hasChildren(HUSD_PrimTraversalDemands demands) const;
//    void               getParent(HUSD_PrimHandle &parent) const;
    void                 getChildren(UT_Array<HUSD_PrimHandle> &children,
                                HUSD_PrimTraversalDemands demands) const;
    void                 getProperties(
                                UT_Array<HUSD_PropertyHandle> &props,
                                bool include_attributes,
                                bool include_relationships,
                                bool include_shader_inputs) const;

    // Debugging only... Do not use in production code.
    void                 getAttributeNames(
                                UT_ArrayStringSet &attrib_names) const;
    void                 extractAttributes(
                                const UT_ArrayStringSet &which_attribs,
                                const HUSD_TimeCode &tc,
                                UT_Options &values);

private:
    HUSD_DataHandle              myDataHandle;
    HUSD_ConstOverridesPtr       myOverrides;
    HUSD_ConstPostLayersPtr      myPostLayers;
};

#endif