HUSD_Save.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_Save_h__
#define __HUSD_Save_h__

#include "HUSD_API.h"
#include "HUSD_DataHandle.h"
#include "HUSD_OutputProcessor.h"
#include "HUSD_TimeCode.h"
#include <UT/UT_Interval.h>
#include <UT/UT_Options.h>
#include <UT/UT_PathPattern.h>
#include <UT/UT_StringHolder.h>
#include <UT/UT_StringMap.h>
#include <UT/UT_UniquePtr.h>
#include <SYS/SYS_Types.h>

class OP_Node;

enum HUSD_SaveStyle {
    HUSD_SAVE_FLATTENED_IMPLICIT_LAYERS,
    HUSD_SAVE_FLATTENED_ALL_LAYERS,
    HUSD_SAVE_FLATTENED_STAGE,
    HUSD_SAVE_SEPARATE_LAYERS,
};

struct HUSD_OutputProcessorAndOverrides
{
    HUSD_OutputProcessorAndOverrides(HUSD_OutputProcessorPtr processor,
                                     const UT_Options &overrides=UT_Options())
            : myProcessor(processor), myOverrides(overrides) {}
    HUSD_OutputProcessorPtr myProcessor;
    UT_Options myOverrides;
};
using HUSD_OutputProcessorAndOverridesArray =
        UT_Array<HUSD_OutputProcessorAndOverrides>;

// Some simple structs that just handle lumping bits of save configuration
// data together.
class husd_SaveTimeData
{
public:
                         husd_SaveTimeData()
                             : myStartFrame(-SYS_FP64_MAX),
                               myEndFrame(SYS_FP64_MAX),
                               myTimeCodesPerSecond(SYS_FP64_MAX),
                               myFramesPerSecond(SYS_FP64_MAX)
                         { }

    fpreal64             myStartFrame;
    fpreal64             myEndFrame;
    fpreal64             myTimeCodesPerSecond;
    fpreal64             myFramesPerSecond;
};

class husd_SaveProcessorData
{
public:
                         husd_SaveProcessorData()
                             : myConfigNode(nullptr),
                               myLopNode(nullptr),
                               myConfigTime(0.0),
                               myReplaceHfs(true)
                         { }

    HUSD_OutputProcessorAndOverridesArray myProcessors;
    OP_Node             *myConfigNode;
    OP_Node             *myLopNode;
    fpreal               myConfigTime;
    bool                 myReplaceHfs;
};

class husd_SaveDefaultPrimData
{
public:
                         husd_SaveDefaultPrimData()
                             : myRequireDefaultPrim(false)
                         { }

    UT_StringHolder      myDefaultPrim;
    bool                 myRequireDefaultPrim;
};

class husd_SaveConfigFlags
{
public:
                         husd_SaveConfigFlags()
                             : myClearHoudiniCustomData(false),
                               myFlattenFileLayers(false),
                               myFlattenSopLayers(false),
                               myErrorSavingImplicitPaths(false),
                               myIgnoreSavingImplicitPaths(false),
                               mySaveFilesFromDisk(false),
                               myEnsureMetricsSet(false),
                               myTrackPrimExistence(false),
                               myMuteLayersBeforeSave(false),
                               myStripLayersAboveLayerBreaks(false),
                               myTimeSamplesRange(SYS_FP64_MAX, -SYS_FP64_MAX), // initially "invalid"
                               myTimeSamplesRangePadding(0)
                         { }

    bool                 myClearHoudiniCustomData;
    bool                 myFlattenFileLayers;
    bool                 myFlattenSopLayers;
    bool                 myErrorSavingImplicitPaths;
    bool                 myIgnoreSavingImplicitPaths;
    bool                 mySaveFilesFromDisk;
    bool                 myEnsureMetricsSet;
    bool                 myTrackPrimExistence;
    bool                 myMuteLayersBeforeSave;
    bool                 myStripLayersAboveLayerBreaks;
    UT_IntervalD         myTimeSamplesRange;
    fpreal64             myTimeSamplesRangePadding;  
};

class HUSD_API HUSD_Save
{
public:
                         HUSD_Save();
                        ~HUSD_Save();

    bool                 addCombinedTimeSample(const HUSD_AutoReadLock &lock,
                                const HUSD_TimeCode &timecode);
    bool                 saveCombined(const UT_StringRef &filepath,
                                bool filepath_is_time_dependent,
                                UT_StringMap<UT_StringHolder> &saved_paths);
    void                 clearSaveHistory();
    bool                 save(const HUSD_AutoReadLock &lock,
                                const HUSD_TimeCode &timecode,
                                const UT_StringRef &filepath,
                                bool filepath_is_time_dependent,
                                UT_StringMap<UT_StringHolder> &saved_paths);

    HUSD_SaveStyle       saveStyle() const
                         { return mySaveStyle; }
    void                 setSaveStyle(HUSD_SaveStyle save_style)
                         { mySaveStyle = save_style; }

    bool                 requireDefaultPrim() const
                         { return myDefaultPrimData.myRequireDefaultPrim; }
    void                 setRequireDefaultPrim(bool require_default_prim)
                         { myDefaultPrimData.
                             myRequireDefaultPrim = require_default_prim; }

    const UT_StringHolder &defaultPrim() const
                         { return myDefaultPrimData.myDefaultPrim; }
    void                 setDefaultPrim(const UT_StringHolder &defaultprim)
                         { myDefaultPrimData.myDefaultPrim = defaultprim; }

    bool                 clearHoudiniCustomData() const
                         { return myFlags.myClearHoudiniCustomData; }
    void                 setClearHoudiniCustomData(bool clear_data)
                         { myFlags.myClearHoudiniCustomData = clear_data; }

    bool                 flattenFileLayers() const
                         { return myFlags.myFlattenFileLayers; }
    void                 setFlattenFileLayers(bool flatten_file_layers)
                         { myFlags.myFlattenFileLayers = flatten_file_layers; }

    bool                 flattenSopLayers() const
                         { return myFlags.myFlattenSopLayers; }
    void                 setFlattenSopLayers(bool flatten_sop_layers)
                         { myFlags.myFlattenSopLayers = flatten_sop_layers; }

    bool                 errorSavingImplicitPaths() const
                         { return myFlags.myErrorSavingImplicitPaths; }
    void                 setErrorSavingImplicitPaths(bool error)
                         { myFlags.myErrorSavingImplicitPaths = error; }

    bool                 ignoreSavingImplicitPaths() const
                         { return myFlags.myIgnoreSavingImplicitPaths; }
    void                 setIgnoreSavingImplicitPaths(bool ignore)
                         { myFlags.myIgnoreSavingImplicitPaths = ignore; }

    bool                 saveFilesFromDisk() const
                         { return myFlags.mySaveFilesFromDisk; }
    void                 setSaveFilesFromDisk(bool save)
                         { myFlags.mySaveFilesFromDisk = save; }

    bool                 ensureMetricsSet() const
                         { return myFlags.myEnsureMetricsSet; }
    void                 setEnsureMetricsSet(bool set)
                         { myFlags.myEnsureMetricsSet = set; }

    bool                 trackPrimExistence() const
                         { return myFlags.myTrackPrimExistence; }
    void                 setTrackPrimExistence(bool track_existence)
                         { myFlags.myTrackPrimExistence = track_existence; }

    bool                 muteLayersBeforeSave() const
                         { return myFlags.myMuteLayersBeforeSave; }
    void                 setMuteLayersBeforeSave(bool mute_layers)
                         { myFlags.myMuteLayersBeforeSave = mute_layers; }
                         
    bool                 stripLayersAboveLayerBreaks() const
                         { return myFlags.myStripLayersAboveLayerBreaks; }
    void                 setStripLayersAboveLayerBreaks(bool strip_layers)
                         { myFlags.myStripLayersAboveLayerBreaks = strip_layers; }

    const UT_IntervalD   &timeSamplesRange() const
                         { return myFlags.myTimeSamplesRange; }
    void                 setTimeSamplesRange(const UT_IntervalD &range)
                         { myFlags.myTimeSamplesRange = range; }
    fpreal64             timeSamplesRangePadding() const
                         { return myFlags.myTimeSamplesRangePadding; }
    void                 setTimeSamplesRangePadding(fpreal64 padding)
                         { myFlags.myTimeSamplesRangePadding = padding; }

    const UT_PathPattern *saveFilesPattern() const
                         { return mySaveFilesPattern.get(); }
    void                 setSaveFilesPattern(const UT_StringHolder &pattern)
                         {
                            if (pattern.isstring())
                                mySaveFilesPattern.reset(
                                    new UT_PathPattern(pattern));
                            else
                                mySaveFilesPattern.reset();
                         }

    fpreal64             startFrame() const
                         { return myTimeData.myStartFrame; }
    void                 setStartFrame(fpreal64 start_time = -SYS_FP64_MAX)
                         { myTimeData.myStartFrame = start_time; }

    fpreal64             endFrame() const
                         { return myTimeData.myEndFrame; }
    void                 setEndFrame(fpreal64 end_time = SYS_FP64_MAX)
                         { myTimeData.myEndFrame = end_time; }

    fpreal64             timeCodesPerSecond() const
                         { return myTimeData.myTimeCodesPerSecond; }
    void                 setTimeCodesPerSecond(fpreal64 tps = SYS_FP64_MAX)
                         { myTimeData.myTimeCodesPerSecond = tps; }

    fpreal64             framesPerSecond() const
                         { return myTimeData.myFramesPerSecond; }
    void                 setFramesPerSecond(fpreal64 fps = SYS_FP64_MAX)
                         { myTimeData.myFramesPerSecond = fps; }

    const HUSD_OutputProcessorAndOverridesArray &outputProcessors() const
                         { return myProcessorData.myProcessors; }
    void                 setOutputProcessors(const HUSD_OutputProcessorArray &aps)
                         {
                             myProcessorData.myProcessors.clear();
                             for (auto &&processor : aps)
                             {
                                 myProcessorData.myProcessors.append({processor});
                             }
                         }
    void                 setOutputProcessors(
                                const HUSD_OutputProcessorAndOverridesArray &aps)
                         { myProcessorData.myProcessors = aps; }

    OP_Node             *outputProcessorsConfigNode() const
                         { return myProcessorData.myConfigNode; }
    void                 setOutputProcessorsConfigNode(OP_Node *config_node)
                         { myProcessorData.myConfigNode = config_node; }

    fpreal               outputProcessorsTime() const
                         { return myProcessorData.myConfigTime; }
    void                 setOutputProcessorsTime(fpreal t)
                         { myProcessorData.myConfigTime = t; }

    bool                 outputProcessorsReplaceHfs() const
                         { return myProcessorData.myReplaceHfs; }
    void                 setOutputProcessorsReplaceHfs(bool replacehfs)
                         { myProcessorData.myReplaceHfs = replacehfs; }

private:
    class                husd_SavePrivate;

    UT_UniquePtr<husd_SavePrivate>       myPrivate;
    UT_UniquePtr<UT_PathPattern>         mySaveFilesPattern;
    HUSD_SaveStyle                       mySaveStyle;
    husd_SaveProcessorData               myProcessorData;
    husd_SaveDefaultPrimData             myDefaultPrimData;
    husd_SaveTimeData                    myTimeData;
    husd_SaveConfigFlags                 myFlags;
};

#endif