#include "toonz/tcamera.h"
#include "toonz/stage.h"
#include "tstream.h"
#include "texception.h"

//=============================================================================
// TCamera

TCamera::TCamera()
    //: m_size(12, 9), m_res(768, 576), m_xPrevalence(true)
    : m_size(30, 16.875),
      m_res(1920, 1080),
      m_xPrevalence(true) {}

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

void TCamera::setSize(const TDimensionD &size, bool preserveDpi,
                      bool preserveAR) {
  double currAR   = getAspectRatio();
  TPointD currDpi = getDpi();
  m_size.lx       = size.lx;
  if (preserveAR)
    m_size.ly = m_size.lx / currAR;  // WARNING! if also preserveDpi==true, the
                                     // AR could lose precision...
  else
    m_size.ly = size.ly;

  if (!preserveDpi) return;

  m_res.lx  = troundp(currDpi.x * m_size.lx);
  m_res.ly  = troundp(currDpi.y * m_size.ly);
  m_size.lx = m_res.lx / currDpi.x;
  if (preserveAR)  //&& currDpi.x==currDpi.y)
  {
    double aux = m_size.lx / currAR - m_res.ly / currDpi.y;
    m_size.ly  = m_size.lx / currAR;
  } else
    m_size.ly = m_res.ly / currDpi.y;
}

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

double TCamera::getAspectRatio() const { return m_size.lx / m_size.ly; }

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

void TCamera::setRes(const TDimension &res) {
  assert(res.lx > 0);
  assert(res.ly > 0);

  if (m_res != res) {
    m_res          = res;
    m_interestRect = TRect();
  }
}

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

TPointD TCamera::getDpi() const {
  TPointD dpi;
  if (m_size.lx > 0 && m_size.ly > 0) {
    dpi.x = m_res.lx / m_size.lx;
    dpi.y = m_res.ly / m_size.ly;
  }
  return dpi;
}

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

bool TCamera::isPixelSquared() const {
  return areAlmostEqual(m_res.lx * m_size.ly, m_res.ly * m_size.lx);
}

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

TAffine TCamera::getStageToCameraRef() const {
  return TAffine(m_res.lx / (Stage::inch * m_size.lx), 0, 0.5 * m_res.lx, 0,
                 m_res.ly / (Stage::inch * m_size.ly), 0.5 * m_res.ly);
}

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

TAffine TCamera::getCameraToStageRef() const {
  const double factor = Stage::inch;

  TDimensionD cameraSize = getSize();
  cameraSize.lx *= factor;
  cameraSize.ly *= factor;
  TPointD center(0.5 * cameraSize.lx, 0.5 * cameraSize.ly);

  return TAffine(factor * m_size.lx / (double)m_res.lx, 0, -center.x, 0,
                 factor * m_size.ly / (double)m_res.ly, -center.y);
}

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

TRectD TCamera::getStageRect() const {
  TDimensionD cameraSize = getSize();
  const double factor    = Stage::inch;
  cameraSize.lx *= factor;
  cameraSize.ly *= factor;
  TRectD rect(cameraSize);
  rect -= 0.5 * (rect.getP00() + rect.getP11());
  return rect;
}

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

void TCamera::setInterestRect(const TRect &rect) {
  // Not using the TRect's common intersection. Unfortunately, in case
  // the rect's coordinates have lx or ly < 0, the intersection returns
  // the default (empty) rect. We want to maintain the coordinates instead.
  // m_interestRect = rect * TRect(m_res);

  m_interestRect.x0 = std::max(rect.x0, 0);
  m_interestRect.y0 = std::max(rect.y0, 0);
  m_interestRect.x1 = std::min(rect.x1, m_res.lx - 1);
  m_interestRect.y1 = std::min(rect.y1, m_res.ly - 1);
}

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

TRectD TCamera::getInterestStageRect() const {
  return getCameraToStageRef() * TRectD(m_interestRect.x0, m_interestRect.y0,
                                        m_interestRect.x1 + 1,
                                        m_interestRect.y1 + 1);
}

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

void TCamera::setInterestStageRect(const TRectD &rect) {
  TRectD cameraInterestRectD(getStageToCameraRef() * rect);

  setInterestRect(TRect(
      tfloor(cameraInterestRectD.x0), tfloor(cameraInterestRectD.y0),
      tceil(cameraInterestRectD.x1) - 1, tceil(cameraInterestRectD.y1) - 1));
}

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

void TCamera::saveData(TOStream &os) const {
  os.child("cameraSize") << m_size.lx << m_size.ly;
  os.child("cameraRes") << m_res.lx << m_res.ly;
  os.child("cameraXPrevalence") << (int)m_xPrevalence;
  os.child("interestRect") << m_interestRect.x0 << m_interestRect.y0
                           << m_interestRect.x1 << m_interestRect.y1;
}

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

void TCamera::loadData(TIStream &is) {
  std::string tagName;
  while (is.matchTag(tagName)) {
    if (tagName == "cameraSize" || tagName == "size")
      is >> m_size.lx >> m_size.ly;
    else if (tagName == "cameraRes" || tagName == "res")
      is >> m_res.lx >> m_res.ly;
    else if (tagName == "cameraXPrevalence") {
      int xPrevalence;
      is >> xPrevalence;
      m_xPrevalence = (bool)xPrevalence;
    } else if (tagName == "interestRect") {
      is >> m_interestRect.x0 >> m_interestRect.y0 >> m_interestRect.x1 >>
          m_interestRect.y1;
    } else
      throw TException("TCamera::loadData. unexpected tag: " + tagName);
    is.matchEndTag();
  }
}