PIKApp/app/operations/pikaoperationcagecoefcalc.c

299 lines
11 KiB
C

/* PIKA - Photo and Image Kooker Application
* a rebranding of The GNU Image Manipulation Program (created with heckimp)
* A derived work which may be trivial. However, any changes may be (C)2023 by Aldercone Studio
*
* Original copyright, applying to most contents (license remains unchanged):
*
* pikaoperationcagecoefcalc.c
* Copyright (C) 2010 Michael Muré <batolettre@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 <https://www.gnu.org/licenses/>.
*/
#include "config.h"
#include <gdk-pixbuf/gdk-pixbuf.h>
#include <gegl.h>
#include "libpikamath/pikamath.h"
#include "operations-types.h"
#include "pikaoperationcagecoefcalc.h"
#include "pikacageconfig.h"
#include "pika-intl.h"
static void pika_operation_cage_coef_calc_finalize (GObject *object);
static void pika_operation_cage_coef_calc_get_property (GObject *object,
guint property_id,
GValue *value,
GParamSpec *pspec);
static void pika_operation_cage_coef_calc_set_property (GObject *object,
guint property_id,
const GValue *value,
GParamSpec *pspec);
static void pika_operation_cage_coef_calc_prepare (GeglOperation *operation);
static GeglRectangle pika_operation_cage_coef_calc_get_bounding_box (GeglOperation *operation);
static gboolean pika_operation_cage_coef_calc_process (GeglOperation *operation,
GeglBuffer *output,
const GeglRectangle *roi,
gint level);
G_DEFINE_TYPE (PikaOperationCageCoefCalc, pika_operation_cage_coef_calc,
GEGL_TYPE_OPERATION_SOURCE)
#define parent_class pika_operation_cage_coef_calc_parent_class
static void
pika_operation_cage_coef_calc_class_init (PikaOperationCageCoefCalcClass *klass)
{
GObjectClass *object_class = G_OBJECT_CLASS (klass);
GeglOperationSourceClass *source_class = GEGL_OPERATION_SOURCE_CLASS (klass);
GeglOperationClass *operation_class = GEGL_OPERATION_CLASS (klass);
gegl_operation_class_set_keys (operation_class,
"name", "pika:cage-coef-calc",
"categories", "transform",
"description", _("Compute a set of coefficient buffer for the PIKA cage tool"),
NULL);
operation_class->prepare = pika_operation_cage_coef_calc_prepare;
operation_class->get_bounding_box = pika_operation_cage_coef_calc_get_bounding_box;
operation_class->cache_policy = GEGL_CACHE_POLICY_ALWAYS;
operation_class->get_cached_region = NULL;
source_class->process = pika_operation_cage_coef_calc_process;
object_class->get_property = pika_operation_cage_coef_calc_get_property;
object_class->set_property = pika_operation_cage_coef_calc_set_property;
object_class->finalize = pika_operation_cage_coef_calc_finalize;
g_object_class_install_property (object_class,
PIKA_OPERATION_CAGE_COEF_CALC_PROP_CONFIG,
g_param_spec_object ("config",
"Config",
"A PikaCageConfig object, that define the transformation",
PIKA_TYPE_CAGE_CONFIG,
G_PARAM_READWRITE |
G_PARAM_CONSTRUCT));
}
static void
pika_operation_cage_coef_calc_init (PikaOperationCageCoefCalc *self)
{
}
static void
pika_operation_cage_coef_calc_finalize (GObject *object)
{
PikaOperationCageCoefCalc *self = PIKA_OPERATION_CAGE_COEF_CALC (object);
g_clear_object (&self->config);
G_OBJECT_CLASS (parent_class)->finalize (object);
}
static void
pika_operation_cage_coef_calc_get_property (GObject *object,
guint property_id,
GValue *value,
GParamSpec *pspec)
{
PikaOperationCageCoefCalc *self = PIKA_OPERATION_CAGE_COEF_CALC (object);
switch (property_id)
{
case PIKA_OPERATION_CAGE_COEF_CALC_PROP_CONFIG:
g_value_set_object (value, self->config);
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, property_id, pspec);
break;
}
}
static void
pika_operation_cage_coef_calc_set_property (GObject *object,
guint property_id,
const GValue *value,
GParamSpec *pspec)
{
PikaOperationCageCoefCalc *self = PIKA_OPERATION_CAGE_COEF_CALC (object);
switch (property_id)
{
case PIKA_OPERATION_CAGE_COEF_CALC_PROP_CONFIG:
if (self->config)
g_object_unref (self->config);
self->config = g_value_dup_object (value);
break;
default:
G_OBJECT_WARN_INVALID_PROPERTY_ID (object, property_id, pspec);
break;
}
}
static gboolean
pika_operation_cage_coef_calc_is_on_straight (PikaVector2 *d1,
PikaVector2 *d2,
PikaVector2 *p)
{
PikaVector2 v1, v2;
gfloat deter;
v1.x = p->x - d1->x;
v1.y = p->y - d1->y;
v2.x = d2->x - d1->x;
v2.y = d2->y - d1->y;
pika_vector2_normalize (&v1);
pika_vector2_normalize (&v2);
deter = v1.x * v2.y - v2.x * v1.y;
return (deter < 0.000000001) && (deter > -0.000000001);
}
static void
pika_operation_cage_coef_calc_prepare (GeglOperation *operation)
{
PikaOperationCageCoefCalc *occc = PIKA_OPERATION_CAGE_COEF_CALC (operation);
PikaCageConfig *config = PIKA_CAGE_CONFIG (occc->config);
gegl_operation_set_format (operation,
"output",
babl_format_n (babl_type ("float"),
2 * pika_cage_config_get_n_points (config)));
}
static GeglRectangle
pika_operation_cage_coef_calc_get_bounding_box (GeglOperation *operation)
{
PikaOperationCageCoefCalc *occc = PIKA_OPERATION_CAGE_COEF_CALC (operation);
PikaCageConfig *config = PIKA_CAGE_CONFIG (occc->config);
return pika_cage_config_get_bounding_box (config);
}
static gboolean
pika_operation_cage_coef_calc_process (GeglOperation *operation,
GeglBuffer *output,
const GeglRectangle *roi,
gint level)
{
PikaOperationCageCoefCalc *occc = PIKA_OPERATION_CAGE_COEF_CALC (operation);
PikaCageConfig *config = PIKA_CAGE_CONFIG (occc->config);
const Babl *format;
GeglBufferIterator *it;
guint n_cage_vertices;
PikaCagePoint *current, *last;
if (! config)
return FALSE;
format = babl_format_n (babl_type ("float"), 2 * pika_cage_config_get_n_points (config));
n_cage_vertices = pika_cage_config_get_n_points (config);
it = gegl_buffer_iterator_new (output, roi, 0, format,
GEGL_ACCESS_WRITE, GEGL_ABYSS_NONE, 1);
while (gegl_buffer_iterator_next (it))
{
/* iterate inside the roi */
gfloat *coef = it->items[0].data;
gint n_pixels = it->length;
gint x = it->items[0].roi.x; /* initial x */
gint y = it->items[0].roi.y; /* and y coordinates */
gint j;
memset (coef, 0, sizeof * coef * n_pixels * 2 * n_cage_vertices);
while(n_pixels--)
{
if (pika_cage_config_point_inside(config, x, y))
{
last = &(g_array_index (config->cage_points, PikaCagePoint, 0));
for( j = 0; j < n_cage_vertices; j++)
{
PikaVector2 v1,v2,a,b,p;
gdouble BA,SRT,L0,L1,A0,A1,A10,L10, Q,S,R, absa;
current = &(g_array_index (config->cage_points, PikaCagePoint, (j+1) % n_cage_vertices));
v1 = last->src_point;
v2 = current->src_point;
p.x = x;
p.y = y;
a.x = v2.x - v1.x;
a.y = v2.y - v1.y;
absa = pika_vector2_length (&a);
b.x = v1.x - x;
b.y = v1.y - y;
Q = a.x * a.x + a.y * a.y;
S = b.x * b.x + b.y * b.y;
R = 2.0 * (a.x * b.x + a.y * b.y);
BA = b.x * a.y - b.y * a.x;
SRT = sqrt(4.0 * S * Q - R * R);
L0 = log(S);
L1 = log(S + Q + R);
A0 = atan2(R, SRT) / SRT;
A1 = atan2(2.0 * Q + R, SRT) / SRT;
A10 = A1 - A0;
L10 = L1 - L0;
/* edge coef */
coef[j + n_cage_vertices] = (-absa / (4.0 * G_PI)) * ((4.0*S-(R*R)/Q) * A10 + (R / (2.0 * Q)) * L10 + L1 - 2.0);
if (isnan(coef[j + n_cage_vertices]))
{
coef[j + n_cage_vertices] = 0.0;
}
/* vertice coef */
if (!pika_operation_cage_coef_calc_is_on_straight (&v1, &v2, &p))
{
coef[j] += (BA / (2.0 * G_PI)) * (L10 /(2.0*Q) - A10 * (2.0 + R / Q));
coef[(j+1)%n_cage_vertices] -= (BA / (2.0 * G_PI)) * (L10 / (2.0 * Q) - A10 * (R / Q));
}
last = current;
}
}
coef += 2 * n_cage_vertices;
/* update x and y coordinates */
x++;
if (x >= (it->items[0].roi.x + it->items[0].roi.width))
{
x = it->items[0].roi.x;
y++;
}
}
}
return TRUE;
}