libvips/libsrc/mosaicing/im_lrmerge.c
2007-08-29 16:23:50 +00:00

1129 lines
27 KiB
C

/* Merge two images left-right. dx, dy is the offset needed to get from sec
* (secondary image) to ref (reference image).
*
* Usage:
*
* int
* im_lrmerge( ref, sec, out, dx, dy )
* IMAGE *ref, *sec, *out;
* int dx, dy;
*
* Returns 0 on success and -1 on error
*
* Copyright: 1990, 1991 N. Dessipris
* Author: N. Dessipris
* Written on: 20/09/1990
* Updated on: 17/04/1991
* 1/6/92: JC
* - check for difference bug fixed
* - geometry calculations improved and simplified
* - small speedups
Kirk Martinez for Sys5 29/4/93
* 7/8/93 JC
* - ANSIfied
* - memory leaks fixed, ready for partial v2
* - now does IM_CODING_LABQ too
* 8/11/93 JC
* - now propogates both input histories
* - adds magic lines for global mosaic optimisation
*
*
*
May/1994 Ahmed Abbood
*
* - Modified to use partials on all IO
*
June/1995 Ahmed Abbood
*
* - Modified to work with different types of images.
*
* 16/6/95 JC
* - tidied up a little
* - added to VIPS!
* 7/9/95 JC
* - split into two parts: im_lrmerge() and im__lrmerge()
* - latter called by im_lrmosaic()
* - just the same as public im_lrmerge(), but adds no history
* - necessary for im_global_balance()
* - small bugs fixed
* 10/10/95 JC
* - better checks that parameters are sensible
* 11/10/95 JC
* - Kirk spotted what a load of rubbish Ahmed's code is
* - rewritten - many, many bugs fixed
* 24/1/97 JC
* - now outputs bounding area of input images, rather than clipping
* - ignores 0 pixels in blend
* - small tidies
* 7/2/97 JC
* - new blend, caching
* 25/2/97 JC
* - old blend back, much simpler
* - speed this up at some point if you think of an easy way to do it
* 29/7/97 JC
* - IM_CODING_LABQ blend now works, was bug in im_wrapone()
* - small tidies
* 10/1/98 JC
* - merge LUTs now shared between all running mergers
* - frees memory explicitly in im__stop_merge, for much better memory
* use in large mosaics, huge improvement!
* 18/2/98 JC
* - im_demand_hint() call added
* 19/2/98 JC
* - now works for any dx/dy by calling im_insert() for bizarre cases
* 26/9/99 JC
* - ooops, blend lut was wrong! wonder how long that's been broken,
* since feb97 I guess
* 2/2/01 JC
* - added tunable max blend width
* 8/3/01 JC
* - switched to integer arithmetic for integer blends
* 7/11/01 JC
* - more sophisticated transparency handling
* - tiny blend speed up
* 19/3/02 JC
* - move fl cache to main state for better sharing
* 15/8/02 JC
* - records Xoffset/Yoffset
* 20/6/05
* - now requires all bands == 0 for transparency (used to just check
* band 0)
*/
/*
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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 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 <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <limits.h>
/*
#define DEBUG
*/
#include <vips/vips.h>
#include <vips/thread.h>
#include "merge.h"
#ifdef WITH_DMALLOC
#include <dmalloc.h>
#endif /*WITH_DMALLOC*/
/* Blend luts. Shared between all lr and tb blends.
*/
double *im__coef1 = NULL;
double *im__coef2 = NULL;
int *im__icoef1 = NULL;
int *im__icoef2 = NULL;
/* Create a lut for the merging area. Always BLEND_SIZE entries, we
* scale later when we index it.
*/
int
im__make_blend_luts()
{
int x;
/* Already done?
*/
if( im__coef1 && im__coef2 )
return( 0 );
/* Allocate and fill.
*/
im__coef1 = IM_ARRAY( NULL, BLEND_SIZE, double );
im__coef2 = IM_ARRAY( NULL, BLEND_SIZE, double );
im__icoef1 = IM_ARRAY( NULL, BLEND_SIZE, int );
im__icoef2 = IM_ARRAY( NULL, BLEND_SIZE, int );
if( !im__coef1 || !im__coef2 || !im__icoef1 || !im__icoef2 )
return( -1 );
for( x = 0; x < BLEND_SIZE; x++ ) {
double a = IM_PI * x / (BLEND_SIZE - 1.0);
im__coef1[x] = (cos( a ) + 1.0) / 2.0;
im__coef2[x] = 1.0 - im__coef1[x];
im__icoef1[x] = im__coef1[x] * BLEND_SCALE;
im__icoef2[x] = im__coef2[x] * BLEND_SCALE;
}
return( 0 );
}
/* Return the position of the first non-zero pel from the left.
*/
static int
find_first( REGION *ir, int *pos, int x, int y, int w )
{
PEL *pr = (PEL *) IM_REGION_ADDR( ir, x, y );
IMAGE *im = ir->im;
int ne = w * im->Bands;
int i;
/* Double the number of bands in a complex.
*/
if( im_iscomplex( im ) )
ne *= 2;
/* Search for the first non-zero band element from the left edge of the image.
*/
#define lsearch( TYPE ) { \
TYPE *p = (TYPE *) pr; \
\
for( i = 0; i < ne; i++ ) \
if( p[i] )\
break;\
}
switch( im->BandFmt ) {
case IM_BANDFMT_UCHAR: lsearch( unsigned char ); break;
case IM_BANDFMT_CHAR: lsearch( signed char ); break;
case IM_BANDFMT_USHORT: lsearch( unsigned short ); break;
case IM_BANDFMT_SHORT: lsearch( signed short ); break;
case IM_BANDFMT_UINT: lsearch( unsigned int ); break;
case IM_BANDFMT_INT: lsearch( signed int ); break;
case IM_BANDFMT_FLOAT: lsearch( float ); break;
case IM_BANDFMT_DOUBLE: lsearch( double ); break;
case IM_BANDFMT_COMPLEX:lsearch( float ); break;
case IM_BANDFMT_DPCOMPLEX:lsearch( double ); break;
default:
im_errormsg( "im_lrmerge: internal error" );
return( -1 );
}
/* i is first non-zero band element, we want first non-zero pixel.
*/
*pos = x + i / im->Bands;
return( 0 );
}
/* Return the position of the first non-zero pel from the right.
*/
static int
find_last( REGION *ir, int *pos, int x, int y, int w )
{
PEL *pr = (PEL *) IM_REGION_ADDR( ir, x, y );
IMAGE *im = ir->im;
int ne = w * im->Bands;
int i;
/* Double the number of bands in a complex.
*/
if( im_iscomplex( im ) )
ne *= 2;
/* Search for the first non-zero band element from the right.
*/
#define rsearch( TYPE ) { \
TYPE *p = (TYPE *) pr; \
\
for( i = ne - 1; i >= 0; i-- )\
if( p[i] )\
break;\
}
switch( im->BandFmt ) {
case IM_BANDFMT_UCHAR: rsearch( unsigned char ); break;
case IM_BANDFMT_CHAR: rsearch( signed char ); break;
case IM_BANDFMT_USHORT: rsearch( unsigned short ); break;
case IM_BANDFMT_SHORT: rsearch( signed short ); break;
case IM_BANDFMT_UINT: rsearch( unsigned int ); break;
case IM_BANDFMT_INT: rsearch( signed int ); break;
case IM_BANDFMT_FLOAT: rsearch( float ); break;
case IM_BANDFMT_DOUBLE: rsearch( double ); break;
case IM_BANDFMT_COMPLEX:rsearch( float ); break;
case IM_BANDFMT_DPCOMPLEX:rsearch( double ); break;
default:
im_errormsg( "im_lrmerge: internal error" );
return( -1 );
}
/* i is first non-zero band element, we want first non-zero pixel.
*/
*pos = x + i / im->Bands;
return( 0 );
}
/* Make sure we have first/last for this area.
*/
static int
make_firstlast( MergeInfo *inf, Overlapping *ovlap, Rect *oreg )
{
REGION *rir = inf->rir;
REGION *sir = inf->sir;
Rect rr, sr;
int y, yr, ys;
int missing;
/* We're going to build first/last ... lock it from other generate
* threads. In fact it's harmless if we do get two writers, but we may
* avoid duplicating work.
*/
g_mutex_lock( ovlap->fl_lock );
/* Do we already have first/last for this area? Bail out if we do.
*/
missing = 0;
for( y = oreg->top; y < IM_RECT_BOTTOM( oreg ); y++ ) {
const int j = y - ovlap->overlap.top;
const int first = ovlap->first[j];
if( first < 0 ) {
missing = 1;
break;
}
}
if( !missing ) {
/* No work to do!
*/
g_mutex_unlock( ovlap->fl_lock );
return( 0 );
}
/* Entire width of overlap in ref for scan-lines we want.
*/
rr.left = ovlap->overlap.left;
rr.top = oreg->top;
rr.width = ovlap->overlap.width;
rr.height = oreg->height;
rr.left -= ovlap->rarea.left;
rr.top -= ovlap->rarea.top;
/* Entire width of overlap in sec for scan-lines we want.
*/
sr.left = ovlap->overlap.left;
sr.top = oreg->top;
sr.width = ovlap->overlap.width;
sr.height = oreg->height;
sr.left -= ovlap->sarea.left;
sr.top -= ovlap->sarea.top;
#ifdef DEBUG
printf( "im__lrmerge: making first/last for areas:\n" );
printf( "ref: left = %d, top = %d, width = %d, height = %d\n",
rr.left, rr.top, rr.width, rr.height );
printf( "sec: left = %d, top = %d, width = %d, height = %d\n",
sr.left, sr.top, sr.width, sr.height );
#endif
/* Make pixels.
*/
if( im_prepare( rir, &rr ) || im_prepare( sir, &sr ) ) {
g_mutex_unlock( ovlap->fl_lock );
return( -1 );
}
/* Make first/last cache.
*/
for( y = oreg->top, yr = rr.top, ys = sr.top;
y < IM_RECT_BOTTOM( oreg ); y++, yr++, ys++ ) {
const int j = y - ovlap->overlap.top;
int *first = &ovlap->first[j];
int *last = &ovlap->last[j];
/* Done this line already?
*/
if( *first < 0 ) {
/* Search for start/end of overlap on this scan-line.
*/
if( find_first( sir, first,
sr.left, ys, sr.width ) ||
find_last( rir, last,
rr.left, yr, rr.width ) ) {
g_mutex_unlock( ovlap->fl_lock );
return( -1 );
}
/* Translate to output space.
*/
*first += ovlap->sarea.left;
*last += ovlap->rarea.left;
/* Clip to maximum blend width, if necessary.
*/
if( ovlap->mwidth >= 0 &&
*last - *first > ovlap->mwidth ) {
int shrinkby = (*last - *first) - ovlap->mwidth;
*first += shrinkby / 2;
*last -= shrinkby / 2;
}
}
}
g_mutex_unlock( ovlap->fl_lock );
return( 0 );
}
/* Test pixel == 0.
*/
#define TEST_ZERO( TYPE, T, RESULT ) { \
TYPE *tt = (T); \
int ii; \
\
for( ii = 0; ii < cb; ii++ ) \
if( tt[i] ) \
break; \
if( ii == cb ) \
(RESULT) = 1; \
}
/* Blend two integer images.
*/
#define iblend( TYPE, B, IN1, IN2, OUT ) { \
TYPE *tr = (TYPE *) (IN1); \
TYPE *ts = (TYPE *) (IN2); \
TYPE *tq = (TYPE *) (OUT); \
const int cb = (B); \
const int left = IM_CLIP( 0, first - oreg->left, oreg->width ); \
const int right = IM_CLIP( left, last - oreg->left, oreg->width ); \
int ref_zero; \
int sec_zero; \
int x, b; \
int i; \
\
/* Left of the blend area. \
*/ \
for( i = 0, x = 0; x < left; x++ ) { \
ref_zero = 0; \
TEST_ZERO( TYPE, tr, ref_zero ); \
if( !ref_zero ) \
for( b = 0; b < cb; b++, i++ ) \
tq[i] = tr[i]; \
else \
for( b = 0; b < cb; b++, i++ ) \
tq[i] = ts[i]; \
} \
\
/* In blend area. \
*/ \
for( x = left; x < right; x++ ) { \
ref_zero = 0; \
sec_zero = 0; \
TEST_ZERO( TYPE, tr, ref_zero ); \
TEST_ZERO( TYPE, ts, sec_zero ); \
\
if( !ref_zero && !sec_zero ) { \
int inx = ((x + oreg->left - first) << \
BLEND_SHIFT) / bwidth; \
int c1 = im__icoef1[inx]; \
int c2 = im__icoef2[inx]; \
\
for( b = 0; b < cb; b++, i++ ) \
tq[i] = c1 * tr[i] / BLEND_SCALE + \
c2 * ts[i] / BLEND_SCALE; \
} \
else if( !ref_zero ) \
for( b = 0; b < cb; b++, i++ ) \
tq[i] = tr[i]; \
else \
for( b = 0; b < cb; b++, i++ ) \
tq[i] = ts[i]; \
} \
\
/* Right of blend.
*/ \
for( x = right; x < oreg->width; x++ ) { \
sec_zero = 0; \
TEST_ZERO( TYPE, ts, sec_zero ); \
if( !sec_zero ) \
for( b = 0; b < cb; b++, i++ ) \
tq[i] = ts[i]; \
else \
for( b = 0; b < cb; b++, i++ ) \
tq[i] = tr[i]; \
} \
}
/* Blend two float images.
*/
#define fblend( TYPE, B, IN1, IN2, OUT ) { \
TYPE *tr = (TYPE *) (IN1); \
TYPE *ts = (TYPE *) (IN2); \
TYPE *tq = (TYPE *) (OUT); \
const int cb = (B); \
const int left = IM_CLIP( 0, first - oreg->left, oreg->width ); \
const int right = IM_CLIP( left, last - oreg->left, oreg->width ); \
int ref_zero; \
int sec_zero; \
int x, b; \
int i; \
\
/* Left of the blend area. \
*/ \
for( i = 0, x = 0; x < left; x++ ) { \
ref_zero = 0; \
TEST_ZERO( TYPE, tr, ref_zero ); \
if( !ref_zero ) \
for( b = 0; b < cb; b++, i++ ) \
tq[i] = tr[i]; \
else \
for( b = 0; b < cb; b++, i++ ) \
tq[i] = ts[i]; \
} \
\
/* In blend area. \
*/ \
for( x = left; x < right; x++ ) { \
ref_zero = 0; \
sec_zero = 0; \
TEST_ZERO( TYPE, tr, ref_zero ); \
TEST_ZERO( TYPE, ts, sec_zero ); \
\
if( !ref_zero && !sec_zero ) { \
int inx = ((x + oreg->left - first) << \
BLEND_SHIFT) / bwidth; \
double c1 = im__coef1[inx]; \
double c2 = im__coef2[inx]; \
\
for( b = 0; b < cb; b++, i++ ) \
tq[i] = c1 * tr[i] + c2 * ts[i]; \
} \
else if( !ref_zero ) \
for( b = 0; b < cb; b++, i++ ) \
tq[i] = tr[i]; \
else \
for( b = 0; b < cb; b++, i++ ) \
tq[i] = ts[i]; \
} \
\
/* Right of blend.
*/ \
for( x = right; x < oreg->width; x++ ) { \
sec_zero = 0; \
TEST_ZERO( TYPE, ts, sec_zero ); \
if( !sec_zero ) \
for( b = 0; b < cb; b++, i++ ) \
tq[i] = ts[i]; \
else \
for( b = 0; b < cb; b++, i++ ) \
tq[i] = tr[i]; \
} \
}
/* Left-right blend function for non-labpack images.
*/
static int
lr_blend( REGION *or, MergeInfo *inf, Overlapping *ovlap, Rect *oreg )
{
REGION *rir = inf->rir;
REGION *sir = inf->sir;
IMAGE *im = or->im;
Rect prr, psr;
int y, yr, ys;
/* Make sure we have a complete first/last set for this area.
*/
if( make_firstlast( inf, ovlap, oreg ) )
return( -1 );
/* Part of rr which we will output.
*/
prr = *oreg;
prr.left -= ovlap->rarea.left;
prr.top -= ovlap->rarea.top;
/* Part of sr which we will output.
*/
psr = *oreg;
psr.left -= ovlap->sarea.left;
psr.top -= ovlap->sarea.top;
/* Make pixels.
*/
if( im_prepare( rir, &prr ) )
return( -1 );
if( im_prepare( sir, &psr ) )
return( -1 );
/* Loop down overlap area.
*/
for( y = oreg->top, yr = prr.top, ys = psr.top;
y < IM_RECT_BOTTOM( oreg ); y++, yr++, ys++ ) {
PEL *pr = (PEL *) IM_REGION_ADDR( rir, prr.left, yr );
PEL *ps = (PEL *) IM_REGION_ADDR( sir, psr.left, ys );
PEL *q = (PEL *) IM_REGION_ADDR( or, oreg->left, y );
const int j = y - ovlap->overlap.top;
const int first = ovlap->first[j];
const int last = ovlap->last[j];
const int bwidth = last - first;
switch( im->BandFmt ) {
case IM_BANDFMT_UCHAR:
iblend( unsigned char, im->Bands, pr, ps, q ); break;
case IM_BANDFMT_CHAR:
iblend( signed char, im->Bands, pr, ps, q ); break;
case IM_BANDFMT_USHORT:
iblend( unsigned short, im->Bands, pr, ps, q ); break;
case IM_BANDFMT_SHORT:
iblend( signed short, im->Bands, pr, ps, q ); break;
case IM_BANDFMT_UINT:
iblend( unsigned int, im->Bands, pr, ps, q ); break;
case IM_BANDFMT_INT:
iblend( signed int, im->Bands, pr, ps, q ); break;
case IM_BANDFMT_FLOAT:
fblend( float, im->Bands, pr, ps, q ); break;
case IM_BANDFMT_DOUBLE:
fblend( double, im->Bands, pr, ps, q ); break;
case IM_BANDFMT_COMPLEX:
fblend( float, im->Bands*2, pr, ps, q ); break;
case IM_BANDFMT_DPCOMPLEX:
fblend( double, im->Bands*2, pr, ps, q ); break;
default:
im_errormsg( "im_lrmerge: internal error" );
return( -1 );
}
}
return( 0 );
}
/* Left-right blend function for IM_CODING_LABQ images.
*/
static int
lr_blend_labpack( REGION *or, MergeInfo *inf, Overlapping *ovlap, Rect *oreg )
{
REGION *rir = inf->rir;
REGION *sir = inf->sir;
Rect prr, psr;
int y, yr, ys;
/* Make sure we have a complete first/last set for this area. This
* will just look at the top 8 bits of L, not all 10, but should be OK.
*/
if( make_firstlast( inf, ovlap, oreg ) )
return( -1 );
/* Part of rr which we will output.
*/
prr = *oreg;
prr.left -= ovlap->rarea.left;
prr.top -= ovlap->rarea.top;
/* Part of sr which we will output.
*/
psr = *oreg;
psr.left -= ovlap->sarea.left;
psr.top -= ovlap->sarea.top;
/* Make pixels.
*/
if( im_prepare( rir, &prr ) )
return( -1 );
if( im_prepare( sir, &psr ) )
return( -1 );
/* Loop down overlap area.
*/
for( y = oreg->top, yr = prr.top, ys = psr.top;
y < IM_RECT_BOTTOM( oreg ); y++, yr++, ys++ ) {
PEL *pr = (PEL *) IM_REGION_ADDR( rir, prr.left, yr );
PEL *ps = (PEL *) IM_REGION_ADDR( sir, psr.left, ys );
PEL *q = (PEL *) IM_REGION_ADDR( or, oreg->left, y );
const int j = y - ovlap->overlap.top;
const int first = ovlap->first[j];
const int last = ovlap->last[j];
const int bwidth = last - first;
float *fq = inf->merge;
float *r = inf->from1;
float *s = inf->from2;
/* Unpack two bits we want.
*/
imb_LabQ2Lab( pr, r, oreg->width );
imb_LabQ2Lab( ps, s, oreg->width );
/* Blend as floats.
*/
fblend( float, 3, r, s, fq );
/* Re-pack to output buffer.
*/
imb_Lab2LabQ( inf->merge, q, oreg->width );
}
return( 0 );
}
static void *
lock_free( GMutex *lock )
{
g_mutex_free( lock );
return( NULL );
}
/* Build basic per-call state and do some geometry calculations. Shared with
* im_tbmerge, so not static.
*/
Overlapping *
im__build_mergestate( IMAGE *ref, IMAGE *sec, IMAGE *out,
int dx, int dy, int mwidth )
{
Overlapping *ovlap = IM_NEW( out, Overlapping );
int x;
if( !ovlap )
return( NULL );
if( mwidth < -1 ) {
im_errormsg( "im_lr/tbmerge: mwidth must be -1 or >= 0" );
return( NULL );
}
ovlap->ref = ref;
ovlap->sec = sec;
ovlap->out = out;
ovlap->dx = dx;
ovlap->dy = dy;
ovlap->mwidth = mwidth;
/* Area occupied by ref image. Place at (0,0) to start with.
*/
ovlap->rarea.left = 0;
ovlap->rarea.top = 0;
ovlap->rarea.width = ref->Xsize;
ovlap->rarea.height = ref->Ysize;
/* Area occupied by sec image.
*/
ovlap->sarea.left = -dx;
ovlap->sarea.top = -dy;
ovlap->sarea.width = sec->Xsize;
ovlap->sarea.height = sec->Ysize;
/* Compute overlap.
*/
im_rect_intersectrect( &ovlap->rarea, &ovlap->sarea, &ovlap->overlap );
if( im_rect_isempty( &ovlap->overlap ) ) {
im_errormsg( "im_lr/tbmerge: no overlap" );
return( NULL );
}
/* Find position and size of output image.
*/
im_rect_unionrect( &ovlap->rarea, &ovlap->sarea, &ovlap->oarea );
/* Now: translate everything, so that the output image, not the left
* image, is at (0,0).
*/
ovlap->rarea.left -= ovlap->oarea.left;
ovlap->rarea.top -= ovlap->oarea.top;
ovlap->sarea.left -= ovlap->oarea.left;
ovlap->sarea.top -= ovlap->oarea.top;
ovlap->overlap.left -= ovlap->oarea.left;
ovlap->overlap.top -= ovlap->oarea.top;
ovlap->oarea.left = 0;
ovlap->oarea.top = 0;
/* Make sure blend luts are built.
*/
im__make_blend_luts();
/* Size of first/last cache. Could be either of these ... just pick
* the larger.
*/
ovlap->flsize = IM_MAX( ovlap->overlap.width, ovlap->overlap.height );
/* Build first/last cache.
*/
ovlap->first = IM_ARRAY( out, ovlap->flsize, int );
ovlap->last = IM_ARRAY( out, ovlap->flsize, int );
if( !ovlap->first || !ovlap->last )
return( NULL );
for( x = 0; x < ovlap->flsize; x++ )
ovlap->first[x] = -1;
ovlap->fl_lock = g_mutex_new();
if( im_add_close_callback( out,
(im_callback_fn) lock_free, ovlap->fl_lock, NULL ) ) {
g_mutex_free( ovlap->fl_lock );
return( NULL );
}
return( ovlap );
}
/* Build per-call state.
*/
static Overlapping *
build_lrstate( IMAGE *ref, IMAGE *sec, IMAGE *out, int dx, int dy, int mwidth )
{
Overlapping *ovlap;
if( !(ovlap = im__build_mergestate( ref, sec, out, dx, dy, mwidth )) )
return( NULL );
/* Select blender.
*/
switch( ref->Coding ) {
case IM_CODING_LABQ:
ovlap->blend = lr_blend_labpack;
break;
case IM_CODING_NONE:
ovlap->blend = lr_blend;
break;
default:
im_errormsg( "im_lrmerge: unknown coding type" );
return( NULL );
}
/* Find the parts of output which come just from ref and just from sec.
*/
ovlap->rpart = ovlap->rarea;
ovlap->spart = ovlap->sarea;
ovlap->rpart.width -= ovlap->overlap.width;
ovlap->spart.left += ovlap->overlap.width;
ovlap->spart.width -= ovlap->overlap.width;
/* Is there too much overlap? ie. right edge of ref image is greater
* than right edge of sec image, or left > left.
*/
if( IM_RECT_RIGHT( &ovlap->rarea ) > IM_RECT_RIGHT( &ovlap->sarea ) ||
ovlap->rarea.left > ovlap->sarea.left ) {
im_errormsg( "im_lrmerge: too much overlap" );
return( NULL );
}
/* Max number of pixels we may have to blend over.
*/
ovlap->blsize = ovlap->overlap.width;
return( ovlap );
}
/* The area being demanded can be filled using only pels from either the ref
* or the sec images. Attach output to the appropriate part of the input image.
* area is the position that ir->im occupies in the output image.
*
* Shared with im_tbmerge(), so not static.
*/
int
im__attach_input( REGION *or, REGION *ir, Rect *area )
{
Rect r = or->valid;
/* Translate to source coordinate space.
*/
r.left -= area->left;
r.top -= area->top;
/* Demand input.
*/
if( im_prepare( ir, &r ) )
return( -1 );
/* Attach or to ir.
*/
if( im_region_region( or, ir, &or->valid, r.left, r.top ) )
return( -1 );
return( 0 );
}
/* The area being demanded requires pixels from the ref and sec images. As
* above, but just do a sub-area of the output, and make sure we copy rather
* than just pointer-fiddling. reg is the sub-area of or->valid we should do.
*
* Shared with im_tbmerge(), so not static.
*/
int
im__copy_input( REGION *or, REGION *ir, Rect *area, Rect *reg )
{
Rect r = *reg;
/* Translate to source coordinate space.
*/
r.left -= area->left;
r.top -= area->top;
/* Paint this area of ir into or.
*/
if( im_prepare_to( ir, or, &r, reg->left, reg->top ) )
return( -1 );
return( 0 );
}
/* Black out a region.
*/
void
im__black_region( REGION *reg )
{
PEL *q = (PEL *) IM_REGION_ADDR( reg, reg->valid.left, reg->valid.top );
int wd = IM_REGION_SIZEOF_LINE( reg );
int ls = IM_REGION_LSKIP( reg );
int y;
for( y = 0; y < reg->valid.height; y++, q += ls )
memset( (char *) q, 0, wd );
}
/* Generate function for merge. This is shared between im_lrmerge() and
* im_tbmerge().
*/
int
im__merge_gen( REGION *or, MergeInfo *inf, Overlapping *ovlap )
{
Rect *r = &or->valid;
Rect rreg, sreg, oreg;
/* Find intersection with overlap, ref and sec parts.
*/
im_rect_intersectrect( r, &ovlap->rpart, &rreg );
im_rect_intersectrect( r, &ovlap->spart, &sreg );
/* Do easy cases first: can we satisfy this demand with pixels just
* from ref, or just from sec.
*/
if( im_rect_equalsrect( r, &rreg ) ) {
if( im__attach_input( or, inf->rir, &ovlap->rarea ) )
return( -1 );
}
else if( im_rect_equalsrect( r, &sreg ) ) {
if( im__attach_input( or, inf->sir, &ovlap->sarea ) )
return( -1 );
}
else {
/* Difficult case - do in three stages: black out whole area,
* copy in parts of ref and sec we touch, write blend area.
* This could be sped up somewhat ... we will usually black
* out far too much, and write to the blend area three times.
* Upgrade in the future!
*/
/* Need intersections with whole of left & right, and overlap
* too.
*/
im_rect_intersectrect( r, &ovlap->rarea, &rreg );
im_rect_intersectrect( r, &ovlap->sarea, &sreg );
im_rect_intersectrect( r, &ovlap->overlap, &oreg );
im__black_region( or );
if( !im_rect_isempty( &rreg ) )
if( im__copy_input( or,
inf->rir, &ovlap->rarea, &rreg ) )
return( -1 );
if( !im_rect_isempty( &sreg ) )
if( im__copy_input( or,
inf->sir, &ovlap->sarea, &sreg ) )
return( -1 );
/* Nasty: inf->rir and inf->sir now point to the same bit of
* memory (part of or), and we've written twice. We need to
* make sure we get fresh pixels for the blend, so we must
* invalidate them both. Should maybe add a call to the API
* for this.
*/
inf->rir->valid.width = inf->sir->valid.width = 0;
/* Now blat in the blended area.
*/
if( !im_rect_isempty( &oreg ) )
if( ovlap->blend( or, inf, ovlap, &oreg ) )
return( -1 );
}
return( 0 );
}
/* Stop function. Shared with im_tbmerge(). Free explicitly to reduce mem
* requirements quickly for large mosaics.
*/
int
im__stop_merge( MergeInfo *inf )
{
if( inf->rir ) {
im_region_free( inf->rir );
inf->rir = NULL;
}
if( inf->sir ) {
im_region_free( inf->sir );
inf->sir = NULL;
}
if( inf->from1 ) {
im_free( inf->from1 );
inf->from1 = NULL;
}
if( inf->from2 ) {
im_free( inf->from2 );
inf->from2 = NULL;
}
if( inf->merge ) {
im_free( inf->merge );
inf->merge = NULL;
}
im_free( inf );
return( 0 );
}
/* Start function. Shared with im_tbmerge().
*/
void *
im__start_merge( IMAGE *out, Overlapping *ovlap )
{
MergeInfo *inf = IM_NEW( NULL, MergeInfo );
if( !inf )
return( NULL );
/* Clear all ptrs.
*/
inf->rir = NULL;
inf->sir = NULL;
inf->from1 = NULL;
inf->from2 = NULL;
inf->merge = NULL;
/* If this is going to be a IM_CODING_LABQ, we need IM_CODING_LABQ blend buffers.
*/
if( out->Coding == IM_CODING_LABQ ) {
inf->from1 = IM_ARRAY( NULL, ovlap->blsize * 3, float );
inf->from2 = IM_ARRAY( NULL, ovlap->blsize * 3, float );
inf->merge = IM_ARRAY( NULL, ovlap->blsize * 3, float );
if( !inf->from1 || !inf->from2 || !inf->merge ) {
im__stop_merge( inf );
return( NULL );
}
}
/* Make input regions.
*/
inf->rir = im_region_create( ovlap->ref );
inf->sir = im_region_create( ovlap->sec );
if( !inf->rir || !inf->sir ) {
im__stop_merge( inf );
return( NULL );
}
return( inf );
}
int
im__lrmerge( IMAGE *ref, IMAGE *sec, IMAGE *out, int dx, int dy, int mwidth )
{
Overlapping *ovlap;
#ifdef DEBUG
printf( "im__lrmerge %s %s %s %d %d %d\n",
ref->filename, sec->filename, out->filename,
dx, dy, mwidth );
printf( "ref is %d x %d pixels\n", ref->Xsize, ref->Ysize );
printf( "sec is %d x %d pixels\n", sec->Xsize, sec->Ysize );
#endif
/* Check IMAGEs parameters
*/
if( ref->Bands != sec->Bands || ref->Bbits != sec->Bbits ||
ref->BandFmt != sec->BandFmt ||
ref->Coding != sec->Coding ) {
im_errormsg( "im_lrmerge: input images incompatible" );
return( -1 );
}
if( ref->Coding != IM_CODING_NONE && ref->Coding != IM_CODING_LABQ ) {
im_errormsg( "im_lrmerge: inputs not uncoded or IM_CODING_LABQ" );
return( -1 );
}
if( dx > 0 || dx < 1 - ref->Xsize ) {
#ifdef DEBUG
printf( "im__lrmerge: no overlap, using insert\n" );
#endif
/* No overlap, use insert instead.
*/
if( im_insert( ref, sec, out, -dx, -dy ) )
return( -1 );
out->Xoffset = -dx;
out->Yoffset = -dy;
return( 0 );
}
if( im_piocheck( ref, out ) || im_piocheck( sec, out ) )
return( -1 );
/* Build state for this join.
*/
if( !(ovlap = build_lrstate( ref, sec, out, dx, dy, mwidth )) )
return( -1 );
/* Prepare the output IMAGE.
*/
if( im_cp_descv( out, ref, sec, NULL ) )
return( -1 );
out->Xsize = ovlap->oarea.width;
out->Ysize = ovlap->oarea.height;
out->Xoffset = ovlap->sarea.left;
out->Yoffset = ovlap->sarea.top;
/* Set demand hints.
*/
if( im_demand_hint( out, IM_THINSTRIP, ref, sec, NULL ) )
return( -1 );
/* Generate!
*/
if( im_generate( out,
im__start_merge, im__merge_gen, im__stop_merge, ovlap, NULL ) )
return( -1 );
return ( 0 );
}
int
im_lrmerge( IMAGE *ref, IMAGE *sec, IMAGE *out, int dx, int dy, int mwidth )
{
if( im__lrmerge( ref, sec, out, dx, dy, mwidth ) )
return( -1 );
if( im_histlin( out, "#LRJOIN <%s> <%s> <%s> <%d> <%d> <%d>",
ref->filename, sec->filename, out->filename,
-dx, -dy, mwidth ) )
return( -1 );
return( 0 );
}