CommandGuiHelper.h

#pragma once
#include "AL/maya/Common.h"

#include "maya/MString.h"
#include "maya/MPxCommand.h"

#include <sstream>
#include <string>
#include <vector>

namespace AL {
namespace maya {

typedef MStringArray (*GenerateListFn)(const MString& context);

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
class CommandGuiListGen
  : public MPxCommand
{
public:
  AL_MAYA_DECLARE_COMMAND();

  static int32_t registerListFunc(GenerateListFn generateListFunc, const MString& menuName);

private:
  bool isUndoable() const override { return false; }
  MStatus doIt(const MArgList& args) override;
  static std::vector<std::pair<MString, GenerateListFn>> m_funcs;
};

//----------------------------------------------------------------------------------------------------------------------
//----------------------------------------------------------------------------------------------------------------------
class CommandGuiHelper
{
public:

  enum FileMode
  {
    kSave = 0,  
    kLoad = 1,  
    kDirectoryWithFiles = 2, 
    kDirectory = 3, 
    kMultiLoad = 4  
  };

  enum StringPolicy
  {
    kStringOptional, 
    kStringMustHaveValue 
  };

  CommandGuiHelper(const char* commandName, const char* windowTitle, const char* doitLabel, const char* menuItemPath, bool hasOptionBox = true);

  CommandGuiHelper(const char* commandName, const char* menuItemPath, bool checkBoxValue = false);

  ~CommandGuiHelper();

  void addFlagOption(const char* commandFlag, const char* label, bool defaultVal, bool persist = true);

  void addBoolOption(const char* commandFlag, const char* label, bool defaultVal, bool persist = true);

  void addIntOption(const char* commandFlag, const char* label, int32_t defaultVal, bool persist = true);

  void addIntSliderOption(const char* commandFlag, const char* label, int32_t minVal, int32_t maxVal, int32_t defaultVal, bool persist = true);

  void addInt2Option(const char* commandFlag, const char* label, const int32_t defaultVal[2], bool persist = true);

  void addInt2Option(const char* commandFlag, const char* label, const int32_t a, const int32_t b, bool persist = true)
    {
      const int32_t temp[] = {a, b};
      addInt2Option(commandFlag, label, temp, persist);
    }

  void addInt3Option(const char* commandFlag, const char* label, const int32_t defaultVal[3], bool persist = true);

  void addInt3Option(const char* commandFlag, const char* label, const int32_t a, const int32_t b, const int32_t c, bool persist = true)
    {
      const int32_t temp[] = {a, b, c};
      addInt3Option(commandFlag, label, temp, persist);
    }

  void addInt4Option(const char* commandFlag, const char* label, const int32_t defaultVal[4], bool persist = true);

  void addInt4Option(const char* commandFlag, const char* label, const int32_t a, const int32_t b, const int32_t c, const int32_t d, bool persist = true)
    {
      const int32_t temp[] = {a, b, c, d};
      addInt4Option(commandFlag, label, temp, persist);
    }

  void addListOption(const char* commandFlag, const char* label, GenerateListFn generateList);

  void addEnumOption(
      const char* commandFlag,
      const char* label,
      int defaultIndex,
      const char* const enumNames[],
      const int32_t enumValues[],
      bool persist = true,
      bool passAsString = false);

  void addRadioButtonGroupOption(
      const char* commandFlag,
      const char* label,
      int defaultIndex,
      const char* const enumNames[],
      const int32_t enumValues[],
      bool persist = true,
      bool passAsString = false);

  void addDoubleOption(const char* commandFlag, const char* label, double defaultVal, bool persist = true);

  void addDoubleSliderOption(const char* commandFlag, const char* label, double minVal, double maxVal, double defaultVal, bool persist = true);

  void addVec2Option(const char* commandFlag, const char* label, const double defaultVal[2], bool persist = true);

  void addVec2Option(const char* commandFlag, const char* label, const double a, const double b, bool persist = true)
    {
      const double temp[] = {a, b};
      addVec2Option(commandFlag, label, temp, persist);
    }

  void addVec3Option(const char* commandFlag, const char* label, const double defaultVal[3], bool persist = true);

  void addVec3Option(const char* commandFlag, const char* label, const double a, const double b, const double c, bool persist = true)
    {
      const double temp[] = {a, b, c};
      addVec3Option(commandFlag, label, temp, persist);
    }

  void addVec4Option(const char* commandFlag, const char* label, const double defaultVal[4], bool persist = true);

  void addVec4Option(const char* commandFlag, const char* label, const double a, const double b, const double c, const double d, bool persist = true)
    {
      const double temp[] = {a, b, c, d};
      addVec4Option(commandFlag, label, temp, persist);
    }

  void addColourOption(const char* commandFlag, const char* label, const double defaultVal[3], bool persist = true);

  void addColourOption(const char* commandFlag, const char* label, const double a, const double b, const double c, bool persist = true)
    {
      const double temp[] = {a, b, c};
      addColourOption(commandFlag, label, temp, persist);
    }

  void addStringOption(const char* commandFlag, const char* label, MString defaultVal, bool persist = true, StringPolicy policy = kStringOptional);

  void addFilePathOption(const char* commandFlag, const char* label, FileMode fileMode, const char* filter = "All files (*) (*)", StringPolicy policy = kStringOptional);

private:
  std::ostringstream m_global;
  std::ostringstream m_init;
  std::ostringstream m_save;
  std::ostringstream m_load;
  std::ostringstream m_reset;
  std::ostringstream m_execute;
  std::ostringstream m_labels;
  std::ostringstream m_controls;
  std::string m_commandName;
  bool m_hasFilePath;
  bool m_checkBoxCommand;
};

//----------------------------------------------------------------------------------------------------------------------
} // usdmaya
} // al
//----------------------------------------------------------------------------------------------------------------------