1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155
|
/*
* Copyright (C) 2007 - 2008 Valentin Ziegler, ExactCODE GmbH Germany.
*
* 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 2. A copy of the GNU General
* Public License can be found in the file LICENSE.
*
* This program is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANT-
* ABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
* Public License for more details.
*
* Alternatively, commercial licensing options are available from the
* copyright holder ExactCODE GmbH Germany.
*/
#include <math.h>
#include <iostream>
#include <iomanip>
#include "ArgumentList.hh"
#include "DistanceMatrix.hh"
#include "Codecs.hh"
#include "Colorspace.hh"
#include "Matrix.hh"
#include "optimize2bw.hh"
using namespace Utility;
int main (int argc, char* argv[])
{
ArgumentList arglist;
// setup the argument list
Argument<bool> arg_help ("", "help",
"display this help text and exit");
Argument<std::string> arg_input ("i", "input", "input file",
1, 1);
Argument<std::string> arg_output ("o", "output", "output file",
1, 1);
Argument<int> arg_low ("l", "low",
"low normalization value", 0, 0, 1);
Argument<int> arg_high ("h", "high",
"high normalization value", 0, 0, 1);
Argument<int> arg_threshold ("t", "threshold",
"bi-level threshold value", 0, 0, 1);
Argument<int> arg_radius ("r", "radius",
"\"unsharp mask\" radius", 0, 0, 1);
Argument<double> arg_sd ("sd", "standard-deviation",
"standard deviation for Gaussian distribution", 0.0, 0, 1);
arglist.Add (&arg_help);
arglist.Add (&arg_input);
arglist.Add (&arg_output);
arglist.Add (&arg_low);
arglist.Add (&arg_high);
arglist.Add (&arg_threshold);
arglist.Add (&arg_radius);
arglist.Add (&arg_sd);
// parse the specified argument list - and maybe output the Usage
if (!arglist.Read (argc, argv) || arg_help.Get() == true)
{
std::cerr << "Based on Color / Gray image to Bi-level optimizer"
<< " - Copyright 2005, 2006 by René Rebe" << std::endl
<< "Usage:" << std::endl;
arglist.Usage (std::cerr);
return 1;
}
Image o_image;
if (!ImageCodec::Read (arg_input.Get(), o_image)) {
std::cerr << "Error reading input file." << std::endl;
return 1;
}
Image image=o_image;
int low = 0;
int high = 0;
int sloppy_threshold = 0;
int radius = 3;
double sd = 2.1;
if (arg_low.Get() != 0) {
low = arg_low.Get();
std::cerr << "Low value overwritten: " << low << std::endl;
}
if (arg_high.Get() != 0) {
high = arg_high.Get();
std::cerr << "High value overwritten: " << high << std::endl;
}
if (arg_radius.Get() != 0) {
radius = arg_radius.Get();
std::cerr << "Radius: " << radius << std::endl;
}
if (arg_sd.Get() != 0) {
sd = arg_sd.Get();
std::cerr << "SD overwritten: " << sd << std::endl;
}
// convert to 1-bit (threshold)
int threshold = 0;
if (arg_threshold.Get() != 0) {
threshold = arg_threshold.Get();
std::cerr << "Threshold: " << threshold << std::endl;
}
optimize2bw (image, low, high, threshold, sloppy_threshold, radius, sd);
if (arg_threshold.Get() == 0)
threshold = 200;
//FGMatrix m(image, threshold);
//DistanceMatrix dm(m);
DistanceMatrix dm(image, threshold);
unsigned int line=0;
unsigned int row=0;
Image::iterator i=o_image.begin();
Image::iterator end=o_image.end();
Image::iterator color=o_image.begin();
for (; i!=end ; ++i) {
int value=255-(int)dm(row,line);
uint16_t r = value < 0 ? 0 : value, g = value==255 ? 255 : 0, b = 0;
color.setRGB(r, g, b);
i.set(color);
if (++row == (unsigned int)image.w) {
line++;
row=0;
}
}
if (!ImageCodec::Write(arg_output.Get(), o_image)) {
std::cerr << "Error writing output file." << std::endl;
return 1;
}
return 0;
}
|