352 lines
8.4 KiB
C++
352 lines
8.4 KiB
C++
/* Pass VIPS images through gmic
|
|
*
|
|
* AF, 6/10/14
|
|
*/
|
|
|
|
/*
|
|
|
|
This file is part of VIPS.
|
|
|
|
VIPS is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU Lesser General Public License as published by
|
|
the Free Software Foundation; either version 2 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 Lesser General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Lesser General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
|
|
02110-1301 USA
|
|
|
|
*/
|
|
|
|
/*
|
|
|
|
These files are distributed with VIPS - http://www.vips.ecs.soton.ac.uk
|
|
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include <config.h>
|
|
#endif /*HAVE_CONFIG_H*/
|
|
#include <vips/intl.h>
|
|
|
|
#include <vips/vips.h>
|
|
#include <vips/dispatch.h>
|
|
|
|
#include <limits.h>
|
|
|
|
#include <iostream>
|
|
|
|
#include "CImg.h"
|
|
#include "gmic.h"
|
|
|
|
using namespace cimg_library;
|
|
|
|
typedef struct _VipsGMic {
|
|
VipsOperation parent_instance;
|
|
|
|
VipsArrayImage *in;
|
|
VipsImage *out;
|
|
char *command;
|
|
int padding;
|
|
double x_scale;
|
|
double y_scale;
|
|
} VipsGMic;
|
|
|
|
typedef VipsOperationClass VipsGMicClass;
|
|
|
|
#define VIPS_TYPE_GMIC (vips_gmic_get_type())
|
|
#define VIPS_GMIC( obj ) \
|
|
(G_TYPE_CHECK_INSTANCE_CAST( (obj), VIPS_TYPE_GMIC, VipsGMic ))
|
|
#define VIPS_GMIC_CLASS( klass ) \
|
|
(G_TYPE_CHECK_CLASS_CAST( (klass), VIPS_TYPE_GMIC, VipsGMicClass))
|
|
#define VIPS_IS_GMIC( obj ) \
|
|
(G_TYPE_CHECK_INSTANCE_TYPE( (obj), VIPS_TYPE_GMIC ))
|
|
#define VIPS_IS_GMIC_CLASS( klass ) \
|
|
(G_TYPE_CHECK_CLASS_TYPE( (klass), VIPS_TYPE_GMIC ))
|
|
#define VIPS_GMIC_GET_CLASS( obj ) \
|
|
(G_TYPE_INSTANCE_GET_CLASS( (obj), VIPS_TYPE_GMIC, VipsGMicClass ))
|
|
|
|
extern "C" {
|
|
G_DEFINE_TYPE( VipsGMic, vips_gmic, VIPS_TYPE_OPERATION );
|
|
}
|
|
|
|
static int
|
|
vips_gmic_get_tile_border( VipsGMic *vipsgmic )
|
|
{
|
|
return( vipsgmic->padding );
|
|
}
|
|
|
|
#define INDEX( IMG, X, Y, Z ) \
|
|
((*(IMG))( (guint) (X), (guint) (Y), (guint) (Z), 0 ))
|
|
|
|
// copy part of a vips region into a cimg
|
|
template<typename T> static void
|
|
vips_to_gmic( VipsRegion *in, VipsRect *area, CImg<T> *img )
|
|
{
|
|
VipsImage *im = in->im;
|
|
|
|
for( int y = 0; y < area->height; y++ ) {
|
|
T *p = (T *) VIPS_REGION_ADDR( in, area->left, area->top + y );
|
|
|
|
for( int x = 0; x < area->width; x++ ) {
|
|
for( int z = 0; z < im->Bands; z++ )
|
|
INDEX( img, x, y, z ) = p[z];
|
|
|
|
p += im->Bands;
|
|
}
|
|
}
|
|
}
|
|
|
|
// write a CImg to a vips region
|
|
// fill out->valid, img has pixels in img_rect
|
|
template<typename T> static void
|
|
vips_from_gmic( gmic_image<T> *img, VipsRect *img_rect, VipsRegion *out )
|
|
{
|
|
VipsImage *im = out->im;
|
|
VipsRect *valid = &out->valid;
|
|
|
|
g_assert( vips_rect_includesrect( img_rect, valid ) );
|
|
|
|
int x_off = valid->left - img_rect->left;
|
|
int y_off = valid->top - img_rect->top;
|
|
|
|
for( int y = 0; y < valid->height; y++ ) {
|
|
T *p = (T *) \
|
|
VIPS_REGION_ADDR( out, valid->left, valid->top + y );
|
|
|
|
for( int x = 0; x < valid->width; x++ ) {
|
|
for( int z = 0; z < im->Bands; z++ )
|
|
p[z] = INDEX( img, x + x_off, y + y_off, z );
|
|
|
|
p += im->Bands;
|
|
}
|
|
}
|
|
}
|
|
|
|
template<typename T> static int
|
|
vips_gmic_gen_template( VipsRegion *oreg,
|
|
void *seq, void *a, void *b, gboolean *stop )
|
|
{
|
|
VipsRegion **ir = (VipsRegion **) seq;
|
|
VipsGMic *vipsgmic = (VipsGMic *) b;
|
|
int ninput = VIPS_AREA( vipsgmic->in )->n;
|
|
const int tile_border = vips_gmic_get_tile_border( vipsgmic );
|
|
const VipsRect *r = &oreg->valid;
|
|
|
|
VipsRect need;
|
|
VipsRect image;
|
|
|
|
need = *r;
|
|
vips_rect_marginadjust( &need, tile_border );
|
|
image.left = 0;
|
|
image.top = 0;
|
|
image.width = ir[0]->im->Xsize;
|
|
image.height = ir[0]->im->Ysize;
|
|
vips_rect_intersectrect( &need, &image, &need );
|
|
|
|
for( int i = 0; ir[i]; i++ )
|
|
if( vips_region_prepare( ir[i], &need ) )
|
|
return( -1 );
|
|
|
|
gmic gmic_instance;
|
|
gmic_list<T> images;
|
|
gmic_list<char> images_names;
|
|
|
|
try {
|
|
images.assign( (guint) ninput );
|
|
|
|
for( int i = 0; ir[i]; i++ ) {
|
|
gmic_image<T> &img = images._data[i];
|
|
img.assign( need.width, need.height,
|
|
1, ir[i]->im->Bands );
|
|
vips_to_gmic<T>( ir[0], &need, &img );
|
|
}
|
|
|
|
gmic_instance.run( vipsgmic->command, images, images_names );
|
|
vips_from_gmic<T>( &images._data[0], &need, oreg );
|
|
}
|
|
catch( gmic_exception e ) {
|
|
images.assign( (guint) 0 );
|
|
|
|
vips_error( "VipsGMic", "%s", e.what() );
|
|
|
|
return( -1 );
|
|
}
|
|
images.assign( (guint) 0 );
|
|
|
|
return( 0 );
|
|
}
|
|
|
|
static int
|
|
vips_gmic_gen( VipsRegion *oreg, void *seq, void *a, void *b, gboolean *stop )
|
|
{
|
|
VipsRegion **ir = (VipsRegion **) seq;
|
|
|
|
switch( ir[0]->im->BandFmt ) {
|
|
case VIPS_FORMAT_UCHAR:
|
|
return( vips_gmic_gen_template<unsigned char>( oreg,
|
|
seq, a, b, stop ) );
|
|
break;
|
|
|
|
case VIPS_FORMAT_USHORT:
|
|
return( vips_gmic_gen_template<unsigned short int>( oreg,
|
|
seq, a, b, stop ) );
|
|
break;
|
|
|
|
case VIPS_FORMAT_FLOAT:
|
|
return( vips_gmic_gen_template<float>( oreg,
|
|
seq, a, b, stop ) );
|
|
break;
|
|
|
|
default:
|
|
g_assert( 0 );
|
|
break;
|
|
}
|
|
|
|
return( 0 );
|
|
}
|
|
|
|
/* Save a bit of typing.
|
|
*/
|
|
#define UC VIPS_FORMAT_UCHAR
|
|
#define C VIPS_FORMAT_CHAR
|
|
#define US VIPS_FORMAT_USHORT
|
|
#define S VIPS_FORMAT_SHORT
|
|
#define UI VIPS_FORMAT_UINT
|
|
#define I VIPS_FORMAT_INT
|
|
#define F VIPS_FORMAT_FLOAT
|
|
#define X VIPS_FORMAT_COMPLEX
|
|
#define D VIPS_FORMAT_DOUBLE
|
|
#define DX VIPS_FORMAT_DPCOMPLEX
|
|
|
|
/* Type promotion.
|
|
*/
|
|
static const VipsBandFormat vips_gmic_format_table[10] = {
|
|
/* UC C US S UI I F X D DX */
|
|
UC, F, US, F, F, F, F, F, F, F
|
|
};
|
|
|
|
static int
|
|
vips_gmic_build( VipsObject *object )
|
|
{
|
|
VipsObjectClass *klass = VIPS_OBJECT_GET_CLASS( object );
|
|
VipsGMic *vipsgmic = (VipsGMic *) object;
|
|
|
|
VipsImage **in;
|
|
VipsImage **t;
|
|
int ninput;
|
|
VipsBandFormat format;
|
|
|
|
if( VIPS_OBJECT_CLASS( vips_gmic_parent_class )->build( object ) )
|
|
return( -1 );
|
|
|
|
in = vips_array_image_get( vipsgmic->in, &ninput );
|
|
|
|
for( int i = 0; i < ninput; i++ )
|
|
if( vips_image_pio_input( in[i] ) ||
|
|
vips_check_coding_known( klass->nickname, in[i] ) )
|
|
return( -1 );
|
|
|
|
/* Cast all inputs up to the largest common supported format.
|
|
*/
|
|
format = VIPS_FORMAT_UCHAR;
|
|
for( int i = 0; i < ninput; i++ )
|
|
format = VIPS_MAX( format, in[i]->BandFmt );
|
|
format = vips_gmic_format_table[format];
|
|
t = (VipsImage **) vips_object_local_array( object, ninput );
|
|
for( int i = 0; i < ninput; i++ )
|
|
if( vips_cast( in[i], &t[i], format, NULL ) )
|
|
return( -1 );
|
|
in = t;
|
|
|
|
g_object_set( vipsgmic, "out", vips_image_new(), NULL );
|
|
|
|
if( vips_image_pipeline_array( vipsgmic->out,
|
|
VIPS_DEMAND_STYLE_ANY, in ) )
|
|
return( -1 );
|
|
|
|
if( ninput > 0 ) {
|
|
if( vips_image_generate( vipsgmic->out,
|
|
vips_start_many, vips_gmic_gen, vips_stop_many,
|
|
in, vipsgmic ) )
|
|
return( -1 );
|
|
}
|
|
else {
|
|
if( vips_image_generate( vipsgmic->out,
|
|
NULL, vips_gmic_gen, NULL,
|
|
NULL, vipsgmic ) )
|
|
return( -1 );
|
|
}
|
|
|
|
return( 0 );
|
|
}
|
|
|
|
static void
|
|
vips_gmic_class_init( VipsGMicClass *klass )
|
|
{
|
|
GObjectClass *gobject_class = G_OBJECT_CLASS( klass );
|
|
VipsObjectClass *vobject_class = VIPS_OBJECT_CLASS( klass );
|
|
VipsOperationClass *operation_class = VIPS_OPERATION_CLASS( klass );
|
|
|
|
gobject_class->set_property = vips_object_set_property;
|
|
gobject_class->get_property = vips_object_get_property;
|
|
|
|
vobject_class->nickname = "gmic";
|
|
vobject_class->description = _( "Vips G'MIC" );
|
|
vobject_class->build = vips_gmic_build;
|
|
|
|
operation_class->flags = VIPS_OPERATION_SEQUENTIAL_UNBUFFERED;
|
|
|
|
VIPS_ARG_BOXED( klass, "in", 0,
|
|
_( "Input" ),
|
|
_( "Array of input images" ),
|
|
VIPS_ARGUMENT_REQUIRED_INPUT,
|
|
G_STRUCT_OFFSET( VipsGMic, in ),
|
|
VIPS_TYPE_ARRAY_IMAGE );
|
|
|
|
VIPS_ARG_IMAGE( klass, "out", 1,
|
|
_( "Output" ),
|
|
_( "Output image" ),
|
|
VIPS_ARGUMENT_REQUIRED_OUTPUT,
|
|
G_STRUCT_OFFSET( VipsGMic, out ) );
|
|
|
|
VIPS_ARG_INT( klass, "padding", 3,
|
|
_( "padding" ),
|
|
_( "Tile overlap" ),
|
|
VIPS_ARGUMENT_REQUIRED_INPUT,
|
|
G_STRUCT_OFFSET( VipsGMic, padding ),
|
|
0, INT_MAX, 0);
|
|
|
|
VIPS_ARG_DOUBLE( klass, "x_scale", 4,
|
|
_( "x_scale" ),
|
|
_( "X Scale" ),
|
|
VIPS_ARGUMENT_REQUIRED_INPUT,
|
|
G_STRUCT_OFFSET( VipsGMic, x_scale ),
|
|
0, 100000000, 1);
|
|
|
|
VIPS_ARG_DOUBLE( klass, "y_scale", 5,
|
|
_( "y_scale" ),
|
|
_( "Y Scale" ),
|
|
VIPS_ARGUMENT_REQUIRED_INPUT,
|
|
G_STRUCT_OFFSET( VipsGMic, y_scale ),
|
|
0, 100000000, 1);
|
|
|
|
VIPS_ARG_STRING( klass, "command", 10,
|
|
_( "command" ),
|
|
_( "G'MIC command string" ),
|
|
VIPS_ARGUMENT_REQUIRED_INPUT,
|
|
G_STRUCT_OFFSET( VipsGMic, command ),
|
|
NULL );
|
|
}
|
|
|
|
static void
|
|
vips_gmic_init( VipsGMic *vipsgmic )
|
|
{
|
|
}
|