src/QRCommand.h
author František Kučera <franta-hg@frantovo.cz>
Wed, 14 Apr 2021 22:02:23 +0200
branchv_0
changeset 2 6cc693048318
parent 1 79d699ed88ab
child 4 500ce0b934e7
permissions -rw-r--r--
simplify arbitrary polygon to a rectangular box

/**
 * Relational pipes
 * Copyright © 2021 František Kučera (Frantovo.cz, GlobalCode.info)
 *
 * 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, version 3 of the License.
 *
 * 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/>.
 */
#pragma once

#include <codecvt>
#include <memory>
#include <algorithm>
#include <iostream>
#include <stdexcept>

#include <Magick++.h>
#include <zbar.h>

#include "Configuration.h"

namespace relpipe {
namespace in {
namespace qr {

class QRCommand {
private:
	std::wstring_convert<codecvt_utf8<wchar_t>> convertor; // TODO: use platform encoding as default / check zbar encoding


public:

	void process(Configuration& configuration, std::shared_ptr<writer::RelationalWriter> writer, std::function<void() > relationalWriterFlush) {

		Magick::Image magick("/dev/stdin");
		int width = magick.columns();
		int height = magick.rows();
		Magick::Blob blob;
		magick.modifyImage();
		magick.write(&blob, "GRAY", 8);
		const void *raw = blob.data();

		zbar::Image image(width, height, "Y800", raw, width * height);
		zbar::ImageScanner scanner;

		writer->startRelation(L"qr",{
			{L"type", relpipe::writer::TypeId::STRING},
			// {L"addon_name", relpipe::writer::TypeId::STRING},
			{L"value", relpipe::writer::TypeId::STRING},
			// {L"xml", relpipe::writer::TypeId::STRING},
			{L"x", relpipe::writer::TypeId::INTEGER},
			{L"y", relpipe::writer::TypeId::INTEGER},
			{L"width", relpipe::writer::TypeId::INTEGER},
			{L"height", relpipe::writer::TypeId::INTEGER},
		}, true);

		int n = scanner.scan(image);

		for (zbar::Image::SymbolIterator symbol = image.symbol_begin(); symbol != image.symbol_end(); ++symbol) {
			writer->writeAttribute(convertor.from_bytes(symbol->get_type_name()));
			// writer->writeAttribute(convertor.from_bytes(symbol->get_addon_name()));
			writer->writeAttribute(convertor.from_bytes(symbol->get_data()));
			// writer->writeAttribute(convertor.from_bytes(symbol->xml()));

			relpipe::common::type::Integer minX = 0;
			relpipe::common::type::Integer minY = 0;
			relpipe::common::type::Integer maxX = 0;
			relpipe::common::type::Integer maxY = 0;

			// TODO: return original polygon in XML
			for (int i = 0, locationSize = symbol->get_location_size(); i < locationSize; i++) {
				relpipe::common::type::Integer x = symbol->get_location_x(i);
				relpipe::common::type::Integer y = symbol->get_location_y(i);
				minX = minX ? std::min(minX, x) : x;
				minY = minY ? std::min(minY, y) : y;
				maxX = std::max(maxX, x);
				maxY = std::max(maxY, y);
			}

			relpipe::common::type::Integer width = maxX - minX;
			relpipe::common::type::Integer height = maxY - minY;

			writer->writeAttribute(&minX, typeid (minX));
			writer->writeAttribute(&minY, typeid (minY));
			writer->writeAttribute(&width, typeid (width));
			writer->writeAttribute(&height, typeid (height));
		}

		image.set_data(nullptr, 0);
	}
};

}
}
}