EXROperation_WriteLayers.h

#ifndef EXROperation_WriteLayers_h
#define EXROperation_WriteLayers_h

#include <memory>
#include <string>
using namespace std;

#include "EXROperation.h"
#include "DeepImage.h"
#include "SimpleImage.h"

class EXROperation_WriteLayers: public EXROperation
{
public:
    EXROperation_WriteLayers(const SharedConfig &sharedConfig, string opt, vector<pair<string,string>> arguments);

    void AddChannels(shared_ptr<DeepImage> image, Imf::DeepFrameBuffer &frameBuffer) const;
    void Run(shared_ptr<EXROperationState> state) const;

private:
    const SharedConfig &sharedConfig;
    string outputPattern = "<inputname> <ordername> <layer>.exr";

    // This represents a single output file.
    struct OutputImage
    {
        string filename;
        string layerName;
        string layerType;
        int order = 0;
        vector<SimpleImage::EXRLayersToWrite> layers;

        OutputImage()
        {
        }
    };

    struct MaskDesc
    {
        void ParseOptionsString(string optionsString);

        enum MaskType
        {
            // The mask value will be output on the RGB channels.
            MaskType_Greyscale,

            // The mask value will be output on the alpha channel.
            MaskType_Alpha,

            // The mask will be composited with the color channel and output as a pre-masked
            // RGBA image.
            MaskType_CompositedRGB,

            // The mask will be output as a luminance channel in the output EXR file.
            MaskType_EXRLayer,
        };
        MaskType maskType = MaskType_Greyscale;
        string maskChannel;
        string maskName;
    };

    struct LayerDesc
    {
        string layerName;
        int objectId;
        vector<MaskDesc> masks;
    };
    vector<LayerDesc> layerDescs;

    // A list of (dst, src) pairs to combine layers before writing them.
    vector<pair<int,int>> combines;

    string MakeOutputFilename(const OutputImage &layer) const;
    string GetFrameNumberFromFilename(string s) const;
};

#endif