tahoma2d/toonz/sources/toonzlib/captureparameters.cpp

#include "toonz/captureparameters.h"
#include "tiio.h"
#include "tstream.h"
#include "tproperty.h"

//-----------------------------------------------------------------------------
// CaptureParameters

CaptureParameters::CaptureParameters()
    : m_deviceName(L"")
    , m_resolution()
    , m_brightness(0)
    , m_contranst(0)
    , m_useWhiteImage(false)
    , m_upsideDown(false)
    , m_filePath(TFilePath("+inputs"))
    , m_increment(1)
    , m_step(1)
    , m_format("tif") {}

//-----------------------------------------------------------------------------

TPropertyGroup *CaptureParameters::getFileFormatProperties(std::string ext) {
  std::map<std::string, TPropertyGroup *>::const_iterator it;
  it = m_formatProperties.find(ext);
  if (it == m_formatProperties.end()) {
    TPropertyGroup *ret     = Tiio::makeWriterProperties(ext);
    m_formatProperties[ext] = ret;
    return ret;
  } else
    return it->second;
}

//-----------------------------------------------------------------------------

void CaptureParameters::assign(const CaptureParameters *params) {
  m_deviceName    = params->getDeviceName();
  m_resolution    = params->getResolution();
  m_brightness    = params->getBrightness();
  m_contranst     = params->getContranst();
  m_useWhiteImage = params->isUseWhiteImage();
  m_upsideDown    = params->isUpsideDown();
  m_filePath      = params->getFilePath();
  m_increment     = params->getIncrement();
  m_step          = params->getStep();
  m_format        = params->getFileFormat();

  std::map<std::string, TPropertyGroup *>::const_iterator it =
      params->m_formatProperties.begin();
  while (it != params->m_formatProperties.end()) {
    m_formatProperties[it->first] = it->second;
    ++it;
  }
}

//-----------------------------------------------------------------------------

void CaptureParameters::saveData(TOStream &os) {
  os.child("deviceName") << m_deviceName;
  os.child("reslution") << m_resolution.lx << m_resolution.ly;
  os.child("brightness") << m_brightness;
  os.child("contranst") << m_contranst;
  os.child("useWhiteImage") << (int)m_useWhiteImage;
  os.child("upsideDown") << (int)m_upsideDown;
  os.child("filePath") << m_filePath;
  // os.child("increment") << m_increment;
  // os.child("step") << m_step;
  os.child("format") << m_format;

  os.openChild("formatsProperties");
  std::vector<std::string> fileExtensions;
  getFileFormatPropertiesExtensions(fileExtensions);
  for (int i = 0; i < (int)fileExtensions.size(); i++) {
    std::string ext    = fileExtensions[i];
    TPropertyGroup *pg = getFileFormatProperties(ext);
    assert(pg);
    std::map<std::string, std::string> attr;
    attr["ext"] = ext;
    os.openChild("formatProperties", attr);
    pg->saveData(os);
    os.closeChild();
  }
  os.closeChild();
}

//-----------------------------------------------------------------------------

void CaptureParameters::loadData(TIStream &is) {
  std::string tagName;
  while (is.matchTag(tagName)) {
    if (tagName == "deviceName")
      is >> m_deviceName;
    else if (tagName == "reslution")
      is >> m_resolution.lx >> m_resolution.ly;
    else if (tagName == "brightness")
      is >> m_brightness;
    else if (tagName == "contranst")
      is >> m_contranst;
    else if (tagName == "useWhiteImage") {
      int value;
      is >> value;
      m_useWhiteImage = value != 0;
    } else if (tagName == "upsideDown") {
      int value;
      is >> value;
      m_upsideDown = value != 0;
    } else if (tagName == "filePath") {
      std::wstring str;
      is >> str;
      m_filePath = TFilePath(str);
    }
    // else if(tagName == "increment")
    //  is >> m_increment;
    // else if(tagName == "step")
    //  is >> m_step;
    else if (tagName == "format")
      is >> m_format;
    else if (tagName == "formatsProperties") {
      while (is.matchTag(tagName)) {
        if (tagName == "formatProperties") {
          std::string ext    = is.getTagAttribute("ext");
          TPropertyGroup *pg = getFileFormatProperties(ext);
          if (ext == "avi") {
            TPropertyGroup appProperties;
            appProperties.loadData(is);
            assert(appProperties.getPropertyCount() == 1);
            if (pg->getPropertyCount() != 1) {
              is.closeChild();
              continue;
            }
            TEnumProperty *enumProp =
                dynamic_cast<TEnumProperty *>(pg->getProperty(0));
            TEnumProperty *enumAppProp =
                dynamic_cast<TEnumProperty *>(appProperties.getProperty(0));
            assert(enumAppProp && enumProp);
            if (enumAppProp && enumProp) {
              try {
                enumProp->setValue(enumAppProp->getValue());
              } catch (TProperty::RangeError &) {
              }
            } else
              throw TException();
          } else
            pg->loadData(is);
          is.closeChild();
        } else
          throw TException("unexpected tag: " + tagName);
      }  // end while
    } else
      throw TException("unexpected property tag: " + tagName);
    is.closeChild();
  }
}

//-------------------------------------------------------------------

void CaptureParameters::getFileFormatPropertiesExtensions(
    std::vector<std::string> &v) const {
  v.reserve(m_formatProperties.size());
  std::map<std::string, TPropertyGroup *>::const_iterator it;
  for (it = m_formatProperties.begin(); it != m_formatProperties.end(); ++it)
    v.push_back(it->first);
}