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 "imageproc/BinaryImage.h"
#include <QImage>
#include <QImageWriter>
#include <QTemporaryFile>
#include <QDir>

void
DebugImages::add(
	QImage const& image, QString const& label,
	boost::function<QWidget* (QImage const&)> const& 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(IntrusivePtr<Item>(new Item(arem_file, label, image_view_factory)));
}

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

AutoRemovingFile
DebugImages::retrieveNext(QString* label, boost::function<QWidget* (QImage const&)>* 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;
}