#include "toonzqt/infoviewer.h"

#include "toonzqt/intfield.h"

#include "tsystem.h"

#include "tlevel.h"

#include "tpalette.h"

#include "tlevel_io.h"

#include "tsound_io.h"

#include "tiio.h"

#include "tstream.h"

#include "ttoonzimage.h"

#include "trasterimage.h"

#include "tvectorimage.h"

#include "toonz/toonzscene.h"

#include "toonzqt/gutil.h"

#include "toonzqt/dvdialog.h"

#include "toutputproperties.h"

#include "toonz/sceneproperties.h"

#include "toonz/tcamera.h"

#include "toonz/levelset.h"

#include "tcontenthistory.h"

#include "tfiletype.h"

#include <qslider></qslider>

#include <qlabel></qlabel>

#include <qtextedit></qtextedit>

#include <qdatetime></qdatetime>

using namespace DVGui;

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

class InfoViewerImp {

public:

  enum {

    eFullpath = 0,

    eFileType,

    eFrames,

    eOwner,

    eSize,

    eCreated,

    eModified,

    eLastAccess,

    eImageSize,

    eSaveBox,

    eBitsSample,

    eSamplePixel,

    eDpi,

    eOrientation,

    eCompression,

    eQuality,

    eSmoothing,

    eCodec,

    eAlphaChannel,

    eByteOrdering,

    eHPos,

    ePalettePages,

    ePaletteStyles,

    eCamera,

    eCameraDpi,

    eFrameCount,

    eLevelCount,

    eOutputPath,

    eEndianess,

    // sound info

    eLength,

    eChannels,

    eSampleRate,

    eSampleSize,

    eSampleType,

    eHowMany

  };

  TFilePath m_path;

  TLevelP m_level;

  std::vector<tframeid> m_fids;</tframeid>

  QStringList m_formats;

  int m_currentIndex;

  int m_frameCount;

  TPalette *m_palette;

  QLabel m_framesLabel;

  IntField m_framesSlider;

  std::vector<std::pair<qlabel *="" *,="" qlabel="">> m_labels;</std::pair<qlabel>

  QLabel m_historyLabel;

  QTextEdit m_history;

  Separator m_separator1, m_separator2;

  void setFileInfo(const TFileStatus &status);

  void setImageInfo();

  void setSoundInfo();

  // void cleanFileInfo();

  void cleanLevelInfo();

  void setToonzSceneInfo();

  void setPaletteInfo();

  void setGeneralFileInfo(const TFilePath &path);

  QString getTypeString();

  void onSliderChanged();

  InfoViewerImp();

  ~InfoViewerImp();

  void clear();

  bool setLabel(TPropertyGroup *pg, int index, std::string type);

  void create(int index, QString str);

  void loadPalette(const TFilePath &path);

  inline void setVal(int index, const QString &str) {

    m_labels[index].second->setText(str);

  }

public slots:

  bool setItem(const TLevelP &level, TPalette *palette, const TFilePath &path);

};

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

InfoViewer::InfoViewer(QWidget *parent)

    : Dialog(parent), m_imp(new InfoViewerImp()) {

  setWindowTitle(tr("File Info"));

  setWindowFlags(windowFlags() | Qt::WindowStaysOnTopHint);

  // setAttribute(Qt::WA_DeleteOnClose);

  int i;

  for (i = 0; i < (int)m_imp->m_labels.size(); i++) {

    addWidgets(m_imp->m_labels[i].first, m_imp->m_labels[i].second);

    if (i == InfoViewerImp::eLastAccess) addWidget(&m_imp->m_separator1);

  }

  addWidget(&m_imp->m_separator2);

  addWidget(&m_imp->m_historyLabel);

  addWidget(&m_imp->m_history);

  addWidgets(&m_imp->m_framesLabel, &m_imp->m_framesSlider);

  connect(&m_imp->m_framesSlider, SIGNAL(valueChanged(bool)), this,

          SLOT(onSliderChanged(bool)));

  hide();

}

InfoViewer::~InfoViewer() {}

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

void InfoViewer::hideEvent(QHideEvent *) { m_imp->m_level = TLevelP(); }

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

void InfoViewer::onSliderChanged(bool) { m_imp->onSliderChanged(); }

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

void InfoViewerImp::onSliderChanged() {

  if (m_framesSlider.getValue() - 1 == m_currentIndex) return;

  m_currentIndex = m_framesSlider.getValue() - 1;

  setImageInfo();

}

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

namespace {

void setLabelStyle(QLabel *l) { l->setObjectName("TitleTxtLabel"); }

}  // namespace

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

void InfoViewerImp::create(int index, QString str) {

  m_labels[index] =

      std::pair<qlabel *="" *,="" qlabel="">(new QLabel(str), new QLabel(""));</qlabel>

  setLabelStyle(m_labels[index].first);

}

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

InfoViewerImp::InfoViewerImp()

    : m_palette(0)

    , m_framesLabel(QObject::tr("Current Frame: "))

    , m_framesSlider()

    , m_history()

    , m_historyLabel(QObject::tr("File History")) {

  setLabelStyle(&m_framesLabel);

  TLevelReader::getSupportedFormats(m_formats);

  TSoundTrackReader::getSupportedFormats(m_formats);

  m_labels.resize(eHowMany);

  create(eFullpath, QObject::tr("Fullpath:     "));

  create(eFileType, QObject::tr("File Type:    "));

  create(eFrames, QObject::tr("Frames:       "));

  create(eOwner, QObject::tr("Owner:        "));

  create(eSize, QObject::tr("Size:         "));

  create(eCreated, QObject::tr("Created:      "));

  create(eModified, QObject::tr("Modified:     "));

  create(eLastAccess, QObject::tr("Last Access:  "));

  // level info

  create(eImageSize, QObject::tr("Image Size:   "));

  create(eSaveBox, QObject::tr("SaveBox:      "));

  create(eBitsSample, QObject::tr("Bits/Sample:  "));

  create(eSamplePixel, QObject::tr("Sample/Pixel: "));

  create(eDpi, QObject::tr("Dpi:          "));

  create(eOrientation, QObject::tr("Orientation:  "));

  create(eCompression, QObject::tr("Compression:  "));

  create(eQuality, QObject::tr("Quality:      "));

  create(eSmoothing, QObject::tr("Smoothing:    "));

  create(eCodec, QObject::tr("Codec:        "));

  create(eAlphaChannel, QObject::tr("Alpha Channel:"));

  create(eByteOrdering, QObject::tr("Byte Ordering:"));

  create(eHPos, QObject::tr("H Pos:"));

  create(ePalettePages, QObject::tr("Palette Pages:"));

  create(ePaletteStyles, QObject::tr("Palette Styles:"));

  create(eCamera, QObject::tr("Camera Size:      "));

  create(eCameraDpi, QObject::tr("Camera Dpi:       "));

  create(eFrameCount, QObject::tr("Number of Frames: "));

  create(eLevelCount, QObject::tr("Number of Levels: "));

  create(eOutputPath, QObject::tr("Output Path:      "));

  create(eEndianess, QObject::tr("Endianess:      "));

  // sound info

  create(eLength, QObject::tr("Length:       "));

  create(eChannels, QObject::tr("Channels: "));

  create(eSampleRate, QObject::tr("Sample Rate: "));

  create(eSampleSize, QObject::tr("Sample Size:      "));

  create(eSampleType, QObject::tr("Sample Type: "));

  m_historyLabel.setStyleSheet("color: rgb(0, 0, 200);");

  m_history.setStyleSheet("font-size: 12px; font-family: \"courier\";");

  // m_history.setStyleSheet("font-family: \"courier\";");

  m_history.setReadOnly(true);

  m_history.setFixedWidth(490);

}

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

void InfoViewerImp::clear() {

  int i;

  for (i = 0; i < (int)m_labels.size(); i++) setVal(i, "");

  m_history.clear();

}

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

InfoViewerImp::~InfoViewerImp() {

  int i;

  for (i = 0; i < (int)m_labels.size(); i++) {

    delete m_labels[i].first;

    delete m_labels[i].second;

  }

  m_labels.clear();

}

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

void InfoViewerImp::setFileInfo(const TFileStatus &status) {

  // m_fPath.setText(status.

}

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

QString InfoViewerImp::getTypeString() {

  QString ext = QString::fromStdString(m_path.getType());

  if (ext == "tlv" || ext == "tzp" || ext == "tzu")

    return "Toonz Cmapped Raster Level";

  else if (ext == "pli" || ext == "svg")

    return "Toonz Vector Level";

  else if (ext == "mov" || ext == "avi" || ext == "3gp")

    return "Movie File";

  else if (ext == "tnz")

    return "Toonz Scene";

  else if (ext == "tab")

    return "Tab Scene";

  else if (ext == "plt")

    return "Toonz Palette";

  else if (ext == "wav" || ext == "aiff" || ext == "aif" || ext == "raw" ||

           ext == "mp3" || ext == "ogg" || ext == "flac")

    return "Audio File";

  else if (ext == "mesh")

    return "Toonz Mesh Level";

  else if (ext == "pic")

    return "Pic File";

  else if (Tiio::makeReader(ext.toStdString()))

    return (m_fids.size() == 1) ? "Single Raster Image" : "Raster Image Level";

  else if (ext == "psd")

    return "Photoshop Image";

  else

    return "Unmanaged File Type";

}

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

void InfoViewerImp::setGeneralFileInfo(const TFilePath &path) {

  QFileInfo fi = toQString(path);

  assert(fi.exists());

  setVal(eFullpath, fi.absoluteFilePath());

  setVal(eFileType, getTypeString());

  if (fi.owner() != "") setVal(eOwner, fi.owner());

  setVal(eSize, fileSizeString(fi.size()));

  setVal(eCreated, fi.birthTime().toString());

  setVal(eModified, fi.lastModified().toString());

  setVal(eLastAccess, fi.lastRead().toString());

  m_separator1.show();

}

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

bool InfoViewerImp::setLabel(TPropertyGroup *pg, int index, std::string type) {

  TProperty *p = pg->getProperty(type);

  if (!p) return false;

  QString str = QString::fromStdString(p->getValueAsString());

  if (dynamic_cast<tboolproperty *="">(p))</tboolproperty>

    setVal(index, str == "0" ? "No" : "Yes");

  else

    setVal(index, str);

  return true;

}

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

void InfoViewerImp::setImageInfo() {

  if (m_path != TFilePath() && !m_fids.empty())

    setGeneralFileInfo(m_path.getType() == "tlv" || !m_path.isLevelName()

                           ? m_path

                           : m_path.withFrame(m_fids[m_currentIndex]));

  assert(m_level);

  setVal(eFrames, QString::number(m_level->getFrameCount()));

  TLevelReaderP lr(m_path);

  const TImageInfo *ii;

  try {

    ii = lr->getImageInfo(m_fids[m_currentIndex]);

  } catch (...) {

    return;

  }

  if (!m_fids.empty() && lr && ii) {

    setVal(eImageSize,

           QString::number(ii->m_lx) + " X " + QString::number(ii->m_ly));

    if (ii->m_x0 <= ii->m_x1)

      setVal(eSaveBox, "(" + QString::number(ii->m_x0) + ", " +

                           QString::number(ii->m_y0) + ", " +

                           QString::number(ii->m_x1) + ", " +

                           QString::number(ii->m_y1) + ")");

    if (ii->m_bitsPerSample > 0)

      setVal(eBitsSample, QString::number(ii->m_bitsPerSample));

    if (ii->m_samplePerPixel > 0)

      setVal(eSamplePixel, QString::number(ii->m_samplePerPixel));

    if (ii->m_dpix > 0 || ii->m_dpiy > 0)

      setVal(eDpi, "(" + QString::number(ii->m_dpix) + ", " +

                       QString::number(ii->m_dpiy) + ")");

    TPropertyGroup *pg = ii->m_properties;

    if (pg) {

      setLabel(pg, eOrientation, "Orientation");

      if (!setLabel(pg, eCompression, "Compression") &&

          !setLabel(pg, eCompression, "Compression Type") &&

          !setLabel(pg, eCompression, "RLE-Compressed"))

        setLabel(pg, eCompression, "File Compression");

      setLabel(pg, eQuality, "Quality");

      setLabel(pg, eSmoothing, "Smoothing");

      setLabel(pg, eCodec, "Codec");

      setLabel(pg, eAlphaChannel, "Alpha Channel");

      setLabel(pg, eByteOrdering, "Byte Ordering");

      setLabel(pg, eEndianess, "Endianess");

    }

  } else

    m_separator1.hide();

  const TContentHistory *ch = 0;

  if (lr) ch = lr->getContentHistory();

  if (ch) {

    QString str = ch->serialize();

    str         = str.remove('\n');

    str         = str.remove(QChar(0));

    str         = str.replace("||", "\n");

    str         = str.remove('|');

    m_history.setPlainText(str);

  }

  TImageP img        = m_level->frame(m_fids[m_currentIndex]);

  TToonzImageP timg  = (TToonzImageP)img;

  TRasterImageP rimg = (TRasterImageP)img;

  TVectorImageP vimg = (TVectorImageP)img;

  if (img) {

    TRect r = convert(timg->getBBox());

    if (r.x0 <= r.x1)

      setVal(eSaveBox,

             "(" + QString::number(r.x0) + ", " + QString::number(r.y0) + ", " +

                 QString::number(r.x1) + ", " + QString::number(r.y1) + ")");

  }

  double dpix, dpiy;

  if (timg) {

    // setVal(eHPos, QString::number(timg->gethPos()));

    timg->getDpi(dpix, dpiy);

    setVal(eDpi,

           "(" + QString::number(dpix) + ", " + QString::number(dpiy) + ")");

    TDimension dim = timg->getRaster()->getSize();

    setVal(eImageSize,

           QString::number(dim.lx) + " X " + QString::number(dim.ly));

    m_palette = timg->getPalette();

  } else if (rimg) {

    rimg->getDpi(dpix, dpiy);

    setVal(eDpi,

           "(" + QString::number(dpix) + ", " + QString::number(dpiy) + ")");

    TDimension dim = rimg->getRaster()->getSize();

    setVal(eImageSize,

           QString::number(dim.lx) + " X " + QString::number(dim.ly));

  } else if (vimg)

    m_palette = vimg->getPalette();

  // TImageP img = m_level->frame(m_fids[m_currentIndex]);

}

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

void InfoViewerImp::setSoundInfo() {

  if (m_path != TFilePath()) setGeneralFileInfo(m_path);

  TSoundTrackP sndTrack;

  try {

    TSoundTrackReaderP sr(m_path);

    if (sr) sndTrack = sr->load();

  } catch (...) {

    return;

  }

  if (!sndTrack) return;

  int seconds = sndTrack->getDuration();

  int minutes = seconds / 60;

  seconds     = seconds % 60;

  QString label;

  if (minutes > 0) label += QString::number(minutes) + " min ";

  label += QString::number(seconds) + " sec";

  setVal(eLength, label);

  label = QString::number(sndTrack->getChannelCount());

  setVal(eChannels, label);

  TUINT32 sampleRate = sndTrack->getSampleRate();

  label              = QString::number(sampleRate / 1000) + " KHz";

  setVal(eSampleRate, label);

  label = QString::number(sndTrack->getBitPerSample()) + " bit";

  setVal(eSampleSize, label);

  switch (sndTrack->getSampleType()) {

  case TSound::INT:

    label = "Signed integer";

    break;

  case TSound::UINT:

    label = "Unsigned integer";

    break;

  case TSound::FLOAT:

    label = "Floating-point";

    break;

  default:

    label = "Unknown";

    break;

  }

  setVal(eSampleType, label);

}

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

void InfoViewerImp::cleanLevelInfo() {}

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

void InfoViewer::setItem(const TLevelP &level, TPalette *palette,

                         const TFilePath &path) {

  if (m_imp->setItem(level, palette, path))

    show();

  else

    hide();

}

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

void InfoViewerImp::setToonzSceneInfo() {

  ToonzScene scene;

  try {

    scene.loadNoResources(m_path);

  } catch (...) {

    return;

  }

  TCamera *cam = scene.getCurrentCamera();

  if (!cam) return;

  TContentHistory *ch = scene.getContentHistory();

  if (ch) {

    QString str = ch->serialize();

    str         = str.remove('\n');

    str         = str.remove(QChar(0));

    str         = str.replace("||", "\n");

    str         = str.remove('|');

    m_history.setPlainText(str);

  }

  TLevelSet *set           = scene.getLevelSet();

  TSceneProperties *prop   = scene.getProperties();

  TOutputProperties *oprop = prop->getOutputProperties();

  setVal(eCamera, QString::number(cam->getRes().lx) + " X " +

                      QString::number(cam->getRes().ly));

  setVal(eCameraDpi, QString::number(cam->getDpi().x) + ", " +

                         QString::number(cam->getDpi().y));

  setVal(eFrameCount, QString::number(scene.getFrameCount()));

  if (set) setVal(eLevelCount, QString::number(set->getLevelCount()));

  if (oprop) setVal(eOutputPath, toQString(oprop->getPath()));

}

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

void InfoViewerImp::setPaletteInfo() {

  if (!m_palette) return;

  setVal(ePalettePages, QString::number(m_palette->getPageCount()));

  setVal(ePaletteStyles, QString::number(m_palette->getStyleCount()));

}

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

void InfoViewerImp::loadPalette(const TFilePath &path) {

  TIStream is(path);

  if (is) {

    TPersist *p = 0;

    is >> p;

    m_palette = dynamic_cast<tpalette *="">(p);</tpalette>

  }

}

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

bool InfoViewerImp::setItem(const TLevelP &level, TPalette *palette,

                            const TFilePath &path) {

  int i;

  clear();

  m_path  = path;

  m_level = level;

  m_fids.clear();

  m_currentIndex = 0;

  m_palette      = palette;

  m_framesLabel.hide();

  m_framesSlider.hide();

  m_separator1.hide();

  m_separator2.hide();

  QString ext = QString::fromStdString(m_path.getType());

  if (m_path != TFilePath() && !m_formats.contains(ext) &&

      !Tiio::makeReader(m_path.getType())) {

    // e' un file non  di immagine (plt, tnz, ...)

    assert(!m_level);

    if (!TSystem::doesExistFileOrLevel(m_path)) {

      DVGui::warning(QObject::tr("The file %1 does not exist.")

                         .arg(QString::fromStdWString(path.getWideString())));

      return false;

    }

    setGeneralFileInfo(m_path);

    if (ext == "plt") {

      assert(!m_level && !m_palette);

      loadPalette(m_path);

    } else if (ext == "tnz")

      setToonzSceneInfo();

  } else if (TFileType::getInfo(m_path) == TFileType::AUDIO_LEVEL) {

    setSoundInfo();

  } else {

    if (ext == "tlv") loadPalette(m_path.withNoFrame().withType("tpl"));

    if (!m_level) {

      assert(m_path != TFilePath());

      TLevelReaderP lr;

      try {

        lr = TLevelReaderP(m_path);

      } catch (...) {

        return false;

      }

      if (lr) {

        try {

          m_level = lr->loadInfo();

        } catch (...) {

          return false;

        }

      }

    }

    if (m_level) {

      // Image or level of images case

      // TLVs are not intended as movie file here (why?). Neither are those

      bool isMovieFile =

          (ext != "tlv" && m_formats.contains(ext) && !m_path.isLevelName());

      m_frameCount = m_level->getFrameCount();

      assert(m_frameCount);

      m_fids.resize(m_frameCount);

      TLevel::Iterator it = m_level->begin();

      for (i = 0; it != m_level->end(); ++it, ++i) m_fids[i] = it->first;

      if (m_frameCount > 1 && !isMovieFile) {

        m_framesSlider.setRange(1, m_frameCount);

        m_framesSlider.setValue(0);

        m_framesSlider.show();

        m_framesLabel.show();

      }

      setImageInfo();

    } else

      return false;

  }

  if (m_palette) setPaletteInfo();

  for (i = 0; i < (int)m_labels.size(); i++)

    if (m_labels[i].second->text() == "")

      m_labels[i].first->hide(), m_labels[i].second->hide();

    else

      m_labels[i].first->show(), m_labels[i].second->show();

  if (m_history.toPlainText() == "") {

    m_separator2.hide();

    m_historyLabel.hide();

    m_history.hide();

  } else {

    m_separator2.show();

    m_historyLabel.show();

    m_history.show();

  }

  return true;

}