237 lines
9.6 KiB
C++
237 lines
9.6 KiB
C++
#include "Document/Dither.h"
|
||
#include "Document/BrushStroke.h"
|
||
#include "IO/ImageExporter.h"
|
||
#include <QPainter>
|
||
#include <QPainterPath>
|
||
#include <QTextBoundaryFinder>
|
||
#include <algorithm>
|
||
#include <cmath>
|
||
#include <set>
|
||
extern "C" {
|
||
#include "DitherPixels.h"
|
||
}
|
||
|
||
// Swift's Dither.swift: the settings, the chunky pixels, the glyphs.
|
||
namespace {
|
||
// Swift's Characters: the text's graphemes.
|
||
QStringList graphemes(const QString &text)
|
||
{
|
||
QStringList found;
|
||
QTextBoundaryFinder finder(QTextBoundaryFinder::Grapheme, text);
|
||
qsizetype start = 0;
|
||
while (finder.toNextBoundary() >= 0) {
|
||
found << text.mid(start, finder.position() - start);
|
||
start = finder.position();
|
||
}
|
||
return found;
|
||
}
|
||
|
||
// Swift's Character.isNewline.
|
||
bool isNewline(const QString &grapheme)
|
||
{
|
||
static const QString breaks = QStringLiteral("\n\r\u000B\u000C\u0085
");
|
||
return grapheme == QStringLiteral("\r\n") || (grapheme.size() == 1 && breaks.contains(grapheme.front()));
|
||
}
|
||
|
||
std::array<uint8_t, 3> bytes(const AdjustmentColor &color)
|
||
{
|
||
return {uint8_t(std::lround(color.red * 255)), uint8_t(std::lround(color.green * 255)), uint8_t(std::lround(color.blue * 255))};
|
||
}
|
||
}
|
||
|
||
QString rawValue(DitherStyle style)
|
||
{
|
||
switch (style) {
|
||
case DitherStyle::atkinson: return QStringLiteral("Atkinson (Classic Mac)");
|
||
case DitherStyle::floydSteinberg: return QStringLiteral("Floyd–Steinberg");
|
||
case DitherStyle::bayer2: return QStringLiteral("Bayer 2 × 2");
|
||
case DitherStyle::bayer4: return QStringLiteral("Bayer 4 × 4");
|
||
case DitherStyle::bayer8: return QStringLiteral("Bayer 8 × 8");
|
||
case DitherStyle::dots: return QStringLiteral("Halftone Dots");
|
||
case DitherStyle::lines: return QStringLiteral("Halftone Lines");
|
||
case DitherStyle::diamonds: return QStringLiteral("Halftone Diamonds");
|
||
case DitherStyle::patterns: return QStringLiteral("Mac Patterns");
|
||
case DitherStyle::ascii: return QStringLiteral("ASCII");
|
||
}
|
||
throw std::logic_error("unknown dither style");
|
||
}
|
||
|
||
int ditherGroup(DitherStyle style)
|
||
{
|
||
const int index = int(style);
|
||
return index < 2 ? 0 : index < 5 ? 1 : index < 8 ? 2 : 3;
|
||
}
|
||
|
||
bool diffuses(DitherStyle style)
|
||
{
|
||
return ditherGroup(style) == 0;
|
||
}
|
||
|
||
bool hasTones(DitherStyle style)
|
||
{
|
||
return ditherGroup(style) < 2;
|
||
}
|
||
|
||
bool isHalftone(DitherStyle style)
|
||
{
|
||
return ditherGroup(style) == 2;
|
||
}
|
||
|
||
bool drawsMarks(DitherStyle style)
|
||
{
|
||
return !hasTones(style);
|
||
}
|
||
|
||
QString rawValue(DitherPixelShape shape)
|
||
{
|
||
return shape == DitherPixelShape::square ? QStringLiteral("Square") : QStringLiteral("Dot");
|
||
}
|
||
|
||
QString rawValue(DitherColors colors)
|
||
{
|
||
switch (colors) {
|
||
case DitherColors::blackWhite: return QStringLiteral("Black & White");
|
||
case DitherColors::twoColors: return QStringLiteral("Two Colors");
|
||
case DitherColors::original: return QStringLiteral("Original");
|
||
}
|
||
throw std::logic_error("unknown dither colours");
|
||
}
|
||
|
||
DitherSettings DitherSettings::normalized() const
|
||
{
|
||
using ImageAdjustmentPixels::clamp;
|
||
DitherSettings result = *this;
|
||
result.pixelSize = std::round(clamp(pixelSize, pixelSizeLow, pixelSizeHigh, 2));
|
||
result.cellSize = std::round(clamp(cellSize, cellSizeLow, cellSizeHigh, 8));
|
||
result.angle = clamp(angle, -90, 90, 45);
|
||
result.levels = std::round(clamp(levels, levelsLow, levelsHigh, 2));
|
||
result.diffusion = clamp(diffusion, 0, 100, 100);
|
||
result.density = clamp(density, -100, 100, 0);
|
||
result.contrast = clamp(contrast, -100, 100, 0);
|
||
result.dark = dark.clamped();
|
||
result.light = light.clamped();
|
||
// Line breaks go; 64 characters at most.
|
||
QStringList kept;
|
||
for (const QString &grapheme : graphemes(characters)) {
|
||
if (!isNewline(grapheme) && kept.size() < 64)
|
||
kept << grapheme;
|
||
}
|
||
result.characters = kept.join(QString());
|
||
return result;
|
||
}
|
||
|
||
QImage DitherSettings::apply(const QImage &image) const
|
||
{
|
||
const DitherSettings settings = normalized();
|
||
const int block = int(settings.pixelSize);
|
||
const QImage source = image.convertToFormat(QImage::Format_RGBA8888_Premultiplied);
|
||
if (source.isNull())
|
||
throw ExportError(ExportError::Kind::render);
|
||
const int width = source.width(), height = source.height();
|
||
QImage working = source;
|
||
if (block > 1) {
|
||
// Chunky pixels: each block's area averaged, then dithered small.
|
||
working = BrushRaster::context((width + block - 1) / block, (height + block - 1) / block, false);
|
||
const int area = block * block;
|
||
for (int y = 0; y < working.height(); ++y) {
|
||
uchar *out = working.scanLine(y);
|
||
for (int x = 0; x < working.width(); ++x) {
|
||
std::array<int, 4> sum{};
|
||
for (int sy = y * block; sy < std::min(height, (y + 1) * block); ++sy) {
|
||
const uchar *row = source.constScanLine(sy);
|
||
for (int sx = x * block; sx < std::min(width, (x + 1) * block); ++sx) {
|
||
for (int channel = 0; channel < 4; ++channel)
|
||
sum[size_t(channel)] += row[sx * 4 + channel];
|
||
}
|
||
}
|
||
for (int channel = 0; channel < 4; ++channel)
|
||
out[x * 4 + channel] = uchar((sum[size_t(channel)] + area / 2) / area);
|
||
}
|
||
}
|
||
}
|
||
const QImage dithered = settings.dither(working);
|
||
if (block <= 1)
|
||
return dithered;
|
||
// Blown back up without smoothing.
|
||
QImage full = BrushRaster::context(width, height, false);
|
||
for (int y = 0; y < height; ++y) {
|
||
const uchar *row = dithered.constScanLine(y / block);
|
||
uchar *out = full.scanLine(y);
|
||
for (int x = 0; x < width; ++x)
|
||
std::copy_n(row + (x / block) * 4, 4, out + x * 4);
|
||
}
|
||
if (settings.pixelShape == DitherPixelShape::dot) {
|
||
// The gaps are the dark colour: black, or picked.
|
||
const std::array<uint8_t, 3> gap = settings.colors == DitherColors::twoColors ? bytes(settings.dark) : std::array<uint8_t, 3>{0, 0, 0};
|
||
dither_dots(full.bits(), size_t(width), size_t(height), size_t(full.bytesPerLine()), block, gap.data());
|
||
}
|
||
return full;
|
||
}
|
||
|
||
QImage DitherSettings::dither(const QImage &image) const
|
||
{
|
||
const int cell = int(cellSize);
|
||
const Glyphs glyphs = style == DitherStyle::ascii ? DitherSettings::glyphs(characters.isEmpty() ? defaultCharacters() : characters, cell) : Glyphs();
|
||
const bool two = colors == DitherColors::twoColors;
|
||
const std::array<uint8_t, 3> darkColor = two ? bytes(dark) : std::array<uint8_t, 3>{0, 0, 0};
|
||
const std::array<uint8_t, 3> lightColor = two ? bytes(light) : std::array<uint8_t, 3>{255, 255, 255};
|
||
bool failed = false;
|
||
const QImage result = ImageAdjustmentPixels::run(image, [&](uchar *pixels, int width, int height, qsizetype stride) {
|
||
DitherParams params{.style = int(style), .levels = int(levels), .diffusion = float(diffusion / 100), .density = float(density / 100),
|
||
.contrast = float(contrast / 100), .cell = cell, .angle = float(angle * M_PI / 180), .lightOnDark = lightOnDark ? 1 : 0,
|
||
.originalColors = colors == DitherColors::original ? 1 : 0, .dark = {darkColor[0], darkColor[1], darkColor[2]},
|
||
.light = {lightColor[0], lightColor[1], lightColor[2]}, .glyphs = glyphs.maps.data(),
|
||
.glyphCoverage = glyphs.coverage.data(), .glyphCount = int(glyphs.coverage.size())};
|
||
failed = dither_apply(pixels, size_t(width), size_t(height), size_t(stride), ¶ms) == 0;
|
||
});
|
||
if (failed)
|
||
throw ExportError(ExportError::Kind::render);
|
||
return result;
|
||
}
|
||
|
||
DitherSettings::Glyphs DitherSettings::glyphs(const QString &characters, int cell)
|
||
{
|
||
// Swift's bold system monospace, at a whole pixel size.
|
||
QFont font(QStringLiteral("monospace"));
|
||
font.setWeight(QFont::Bold);
|
||
font.setPixelSize(std::max(1, int(std::lround(cell * 1.15))));
|
||
std::vector<std::pair<std::vector<uint8_t>, float>> drawn;
|
||
// Swift's Set<Character>: canonically equivalent graphemes are one.
|
||
std::set<QString> seen;
|
||
for (const QString &character : graphemes(characters)) {
|
||
if (!seen.insert(character.normalized(QString::NormalizationForm_C)).second)
|
||
continue;
|
||
QImage map(cell, cell, QImage::Format_Grayscale8);
|
||
if (map.isNull())
|
||
throw ExportError(ExportError::Kind::render);
|
||
map.fill(0);
|
||
QPainterPath path;
|
||
path.addText(0, 0, font, character);
|
||
// Overlapping outlines stay inked, as text rendering fills them.
|
||
path.setFillRule(Qt::WindingFill);
|
||
const QRectF bounds = path.boundingRect();
|
||
{
|
||
// Centred on its ink, as CTLineGetImageBounds centres it.
|
||
QPainter painter(&map);
|
||
painter.setRenderHint(QPainter::Antialiasing);
|
||
painter.translate((cell - bounds.width()) / 2 - bounds.left(), (cell - bounds.height()) / 2 - bounds.top());
|
||
painter.fillPath(path, Qt::white);
|
||
}
|
||
std::vector<uint8_t> bytes(size_t(cell * cell));
|
||
long sum = 0;
|
||
for (int y = 0; y < cell; ++y) {
|
||
std::copy_n(map.constScanLine(y), cell, bytes.begin() + y * cell);
|
||
for (int x = 0; x < cell; ++x)
|
||
sum += map.constScanLine(y)[x];
|
||
}
|
||
drawn.emplace_back(std::move(bytes), float(sum) / float(255 * cell * cell));
|
||
}
|
||
std::stable_sort(drawn.begin(), drawn.end(), [](const auto &a, const auto &b) { return a.second < b.second; });
|
||
Glyphs result;
|
||
for (const auto &[map, coverage] : drawn) {
|
||
result.maps.insert(result.maps.end(), map.begin(), map.end());
|
||
result.coverage.push_back(coverage);
|
||
}
|
||
return result;
|
||
}
|