DebugImages.cpp

/*
    Scan Tailor - Interactive post-processing tool for scanned pages.
    Copyright (C)  Joseph Artsimovich <joseph.artsimovich@gmail.com>

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "DebugImages.h"
#include <QDir>
#include <QImage>
#include <QImageWriter>
#include <QTemporaryFile>
#include "imageproc/BinaryImage.h"

void DebugImages::add(const QImage& image,
                      const QString& label,
                      const boost::function<QWidget*(const QImage&)>& image_view_factory) {
  QTemporaryFile file(QDir::tempPath() + "/scantailor-dbg-XXXXXX.png");
  if (!file.open()) {
    return;
  }

  AutoRemovingFile arem_file(file.fileName());
  file.setAutoRemove(false);

  QImageWriter writer(&file, "png");
  writer.setCompression(2);  // Trade space for speed.
  if (!writer.write(image)) {
    return;
  }

  m_sequence.push_back(make_intrusive<Item>(arem_file, label, image_view_factory));
}

void DebugImages::add(const imageproc::BinaryImage& image,
                      const QString& label,
                      const boost::function<QWidget*(const QImage&)>& image_view_factory) {
  add(image.toQImage(), label, image_view_factory);
}

AutoRemovingFile DebugImages::retrieveNext(QString* label,
                                           boost::function<QWidget*(const QImage&)>* image_view_factory) {
  if (m_sequence.empty()) {
    return AutoRemovingFile();
  }

  AutoRemovingFile file(m_sequence.front()->file);
  if (label) {
    *label = m_sequence.front()->label;
  }
  if (image_view_factory) {
    *image_view_factory = m_sequence.front()->imageViewFactory;
  }

  m_sequence.pop_front();

  return file;
}