sceneAssembly/adskSceneMetadataCmd.h

#ifndef _adskSceneMetadata_h_
#define _adskSceneMetadata_h_

#include <maya/MPxCommand.h>
#include <maya/MArgDatabase.h>

#include <cassert>
#include <memory>

// Forward declarations
namespace adsk
{
    namespace Data
    {
        class Structure;
        class Accessor;
        class Associations;
    }
}

/*==============================================================================
 * CLASS AdskSceneMetadataCmd
 *============================================================================*/

class AdskSceneMetadataCmd : public MPxCommand {

public: 

   /*----- static member functions -----*/

    static MSyntax cmdSyntax();

    static const char* name();
    static void* creator();

   /*----- member functions -----*/
      
    AdskSceneMetadataCmd();
    ~AdskSceneMetadataCmd() override;

    MStatus doIt( const MArgList& ) override;

private:

    /*----- types and enumerations ----*/

    enum Mode {      
        kEdit   = 1 << 0,
        kQuery  = 1 << 1
    };

    /*----- classes -----*/

    // Helper class for holding command-line flags with one argument.
    template <class T, Mode ValidModes>
    class OptFlag
    {
    public:
        OptFlag() : fIsSet(false), fArg() {}

        void parse(const MArgDatabase& argDb, const char* name)
        {
            MStatus status;
            fIsSet = argDb.isFlagSet(name, &status);
            assert(status == MS::kSuccess);
            
            status = argDb.getFlagArgument(name, 0, fArg);
            fIsArgValid = status == MS::kSuccess;
        }

        bool isModeValid(const Mode currentMode)
        {
            return !fIsSet || ((currentMode & ValidModes) != 0);
        }
        
        bool isSet()        const { return fIsSet; }
        bool isArgValid()   const { return fIsArgValid; }
        const T& arg()      const { return fArg; }

        const T& arg(const T& defValue) const {
            if (isSet()) {
                assert(isArgValid());
                return fArg;
            }
            else {
                return defValue;
            }
        }
        
    private:
        bool        fIsSet;
        bool        fIsArgValid;
        T           fArg;
    };

    /*----- member functions -----*/
  
    MStatus doEdit ( MString fileName );
    MStatus doQuery( MString fileName );

    MStatus getMetadata( MString fileName );
    MStatus setMetadata( MString fileName );

    std::unique_ptr< adsk::Data::Accessor > getAccessorForScene( MString scenePath );
    MStatus getSceneAssociations( adsk::Data::Accessor &accessor, adsk::Data::Associations &out_associations );

    /*----- data members -----*/

    // Command line arguments
    Mode                                    fMode;          // The command mode (edit or query)
    OptFlag<MString, Mode(kEdit|kQuery)>    fChannelName;   // The channel name in which the metadata is written
    OptFlag<MString, Mode(kEdit)>           fData;          // The metadata to write (edit mode only)
};


#endif // _adskSceneMetadata_h_

//-
//*****************************************************************************
// Copyright 2015 Autodesk, Inc.  All rights reserved.
// 
// Use of this software is subject to the terms of the Autodesk license
// agreement provided at the time of installation or download, or which
// otherwise accompanies this software in either electronic or hard copy
// form.
//*****************************************************************************
//+