gimp/plug-ins/gimpressionist/ppmtool.c

690 lines
16 KiB
C

#ifdef HAVE_CONFIG_H
#include "config.h"
#else
#define HAVE_DIRENT_H
#define HAVE_UNISTD_H
#endif
#include <stdio.h>
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
#include <string.h>
#include <stdlib.h>
#include <math.h>
#include "ppmtool.h"
#include "gimpressionist.h"
int readline(FILE *f, char *buffer, int len)
{
again:
if(!fgets(buffer, len, f))
return -1;
if(*buffer == '#') {
goto again;
}
while(strlen(buffer) && buffer[strlen(buffer)-1] <= ' ')
buffer[strlen(buffer)-1] = '\0';
return 0;
}
void fatal(char *s)
{
fprintf(stderr, "%s\n", s);
exit(1);
}
void killppm(struct ppm *p)
{
free(p->col);
p->col = NULL;
p->height = p->width = 0;
}
void newppm(struct ppm *p, int xs, int ys)
{
int x;
guchar bgcol[3] = {0,0,0};
if(xs < 1)
xs = 1;
if(ys < 1)
ys = 1;
p->width = xs;
p->height = ys;
p->col = g_malloc(xs * 3 * ys);
for(x = 0; x < xs * 3 * ys; x += 3) {
p->col[x+0] = bgcol[0];
p->col[x+1] = bgcol[1];
p->col[x+2] = bgcol[2];
}
}
void getrgb(struct ppm *s, float xo, float yo, guchar *d)
{
float ix, iy;
int x1, x2, y1, y2;
float x1y1, x2y1, x1y2, x2y2;
float r, g, b;
int bail = 0;
int rowstride = s->width * 3;
if(xo < 0.0) bail=1;
else if(xo >= s->width-1) { xo = s->width-1; } /* bail=1; */
if(yo < 0.0) bail=1;
else if(yo >= s->height-1) { yo= s->height-1; } /* bail=1; */
if(bail) {
d[0] = d[1] = d[2] = 0;
return;
}
ix = (int)xo;
iy = (int)yo;
/*
x1 = wrap(ix, s->width);
x2 = wrap(ix+1, s->width);
y1 = wrap(iy, s->height);
y2 = wrap(iy+1, s->height);
*/
x1 = ix; x2 = ix + 1;
y1 = iy; y2 = iy + 1;
/* printf("x1=%d y1=%d x2=%d y2=%d\n",x1,y1,x2,y2); */
x1y1 = (1.0-xo+ix)*(1.0-yo+iy);
x2y1 = (xo-ix)*(1.0-yo+iy);
x1y2 = (1.0-xo+ix)*(yo-iy);
x2y2 = (xo-ix)*(yo-iy);
r = s->col[y1*rowstride + x1*3 + 0] * x1y1;
g = s->col[y1*rowstride + x1*3 + 1] * x1y1;
b = s->col[y1*rowstride + x1*3 + 2] * x1y1;
if(x2y1 > 0.0) r += s->col[y1*rowstride + x2*3 + 0] * x2y1;
if(x2y1 > 0.0) g += s->col[y1*rowstride + x2*3 + 1] * x2y1;
if(x2y1 > 0.0) b += s->col[y1*rowstride + x2*3 + 2] * x2y1;
if(x1y2 > 0.0) r += s->col[y2*rowstride + x1*3 + 0] * x1y2;
if(x1y2 > 0.0) g += s->col[y2*rowstride + x1*3 + 1] * x1y2;
if(x1y2 > 0.0) b += s->col[y2*rowstride + x1*3 + 2] * x1y2;
if(x2y2 > 0.0) r += s->col[y2*rowstride + x2*3 + 0] * x2y2;
if(x2y2 > 0.0) g += s->col[y2*rowstride + x2*3 + 1] * x2y2;
if(x2y2 > 0.0) b += s->col[y2*rowstride + x2*3 + 2] * x2y2;
d[0] = r;
d[1] = g;
d[2] = b;
}
void resize(struct ppm *p, int nx, int ny)
{
int x, y;
float xs = p->width/(float)nx;
float ys = p->height/(float)ny;
struct ppm tmp = {0,0,NULL};
newppm(&tmp, nx, ny);
for(y = 0; y < ny; y++) {
guchar *row = tmp.col + y * tmp.width * 3;
for(x = 0; x < nx; x++) {
getrgb(p, x*xs, y*ys, &row[x*3]);
}
}
killppm(p);
p->width = tmp.width;
p->height = tmp.height;
p->col = tmp.col;
}
void rescale(struct ppm *p, double sc)
{
resize(p, p->width * sc, p->height * sc);
}
void resize_fast(struct ppm *p, int nx, int ny)
{
int x, y;
float xs = p->width/(float)nx;
float ys = p->height/(float)ny;
struct ppm tmp = {0,0,NULL};
newppm(&tmp, nx, ny);
for(y = 0; y < ny; y++) {
for(x = 0; x < nx; x++) {
int rx = x*xs, ry = y*ys;
memcpy(&tmp.col[y*tmp.width*3+x*3], &p->col[ry*p->width*3+rx*3], 3);
}
}
killppm(p);
p->width = tmp.width;
p->height = tmp.height;
p->col = tmp.col;
}
struct _BrushHeader
{
unsigned int header_size; /* header_size = sz_BrushHeader + brush name */
unsigned int version; /* brush file version # */
unsigned int width; /* width of brush */
unsigned int height; /* height of brush */
unsigned int bytes; /* depth of brush in bytes--always 1 */
unsigned int magic_number;/* GIMP brush magic number */
unsigned int spacing; /* brush spacing */
};
void msb2lsb(unsigned int *i)
{
guchar *p = (guchar *)i, c;
c = p[1]; p[1] = p[2]; p[2] = c;
c = p[0]; p[0] = p[3]; p[3] = c;
}
void loadgbr(char *fn, struct ppm *p)
{
FILE *f;
struct _BrushHeader hdr;
guchar *ptr;
int x, y;
f = fopen(fn, "rb");
if(!f) {
ptr = findfile(fn);
f = fopen(ptr, "rb");
}
if(p->col) killppm(p);
if(!f) {
fprintf(stderr, "loadgbr: Unable to open file \"%s\"!\n", fn);
newppm(p, 10,10);
return;
}
fread(&hdr, 1, sizeof(struct _BrushHeader), f);
for(x = 0; x < 7; x++)
msb2lsb(&((unsigned int *)&hdr)[x]);
newppm(p, hdr.width, hdr.height);
ptr = g_malloc(hdr.width);
fseek(f, hdr.header_size, SEEK_SET);
for(y = 0; y < p->height; y++) {
fread(ptr, p->width, 1, f);
for(x = 0; x < p->width; x++) {
int k = y*p->width*3 + x*3;
p->col[k+0] = p->col[k+1] = p->col[k+2] = ptr[x];
}
}
fclose(f);
free(ptr);
}
void loadppm(char *fn, struct ppm *p)
{
char line[200];
int y, pgm = 0;
FILE *f;
if(!strcmp(&fn[strlen(fn)-4], ".gbr")) {
loadgbr(fn, p);
return;
}
f = fopen(fn, "rb");
if(!f) f = fopen(findfile(fn), "rb");
if(p->col) killppm(p);
if(!f) {
fprintf(stderr, "loadppm: Unable to open file \"%s\"!\n", fn);
newppm(p, 10,10);
return;
/* fatal("Aborting!"); */
}
readline(f, line, 200);
if(strcmp(line, "P6")) {
if(strcmp(line, "P5")) {
fclose(f);
printf( "loadppm: File \"%s\" not PPM/PGM? (line=\"%s\")%c\n", fn, line, 7);
newppm(p, 10,10);
return;
/* fatal("Aborting!"); */
}
pgm = 1;
}
readline(f, line, 200);
p->width = atoi(line);
p->height = atoi(strchr(line, ' ')+1);
readline(f, line, 200);
if(strcmp(line, "255")) {
printf ("loadppm: File \"%s\" not valid PPM/PGM? (line=\"%s\")%c\n", fn, line, 7);
newppm(p, 10,10);
return;
/* fatal("Aborting!"); */
}
p->col = g_malloc(p->height * p->width * 3);
if(!pgm) {
fread(p->col, p->height * 3 * p->width, 1, f);
} else {
guchar *tmpcol = g_malloc(p->width * p->height);
fread(tmpcol, p->height * p->width, 1, f);
for(y = 0; y < p->width * p->height * 3; y++) {
p->col[y] = tmpcol[y/3];
}
}
fclose(f);
}
void fill(struct ppm *p, guchar *c)
{
int x, y;
if((c[0] == c[1]) && (c[0] == c[2])) {
guchar col = c[0];
for(y = 0; y < p->height; y++) {
memset(p->col + y*p->width*3, col, p->width*3);
}
} else {
for(y = 0; y < p->height; y++) {
guchar *row = p->col + y * p->width * 3;
for(x = 0; x < p->width; x++) {
int k = x * 3;
row[k+0] = c[0];
row[k+1] = c[1];
row[k+2] = c[2];
}
}
}
}
void copyppm(struct ppm *s, struct ppm *p)
{
if(p->col)
killppm(p);
p->width = s->width;
p->height = s->height;
p->col = g_malloc(p->width * 3 * p->height);
memcpy(p->col, s->col, p->width * 3 * p->height);
}
void freerotate(struct ppm *p, double amount)
{
int x, y;
double nx, ny;
double R, a;
struct ppm tmp = {0,0,NULL};
double f = amount*G_PI*2/360.0;
int rowstride = p->width * 3;
a = p->width/(float)p->height;
R = p->width<p->height?p->width/2:p->height/2;
newppm(&tmp, p->width, p->height);
for(y = 0; y < p->height; y++) {
for(x = 0; x < p->width; x++) {
double r, d;
nx = fabs(x-p->width/2.0);
ny = fabs(y-p->height/2.0);
r = sqrt(nx*nx + ny*ny);
d = atan2((y-p->height/2.0),(x-p->width/2.0));
nx = (p->width/2.0 + cos(d-f) * r);
ny = (p->height/2.0 + sin(d-f) * r);
getrgb(p, nx, ny, tmp.col + y*rowstride+x*3);
}
}
killppm(p);
p->width = tmp.width;
p->height = tmp.height;
p->col = tmp.col;
}
void crop(struct ppm *p, int lx, int ly, int hx, int hy)
{
struct ppm tmp = {0,0,NULL};
int x, y;
int srowstride = p->width * 3;
int drowstride;
newppm(&tmp, hx-lx, hy-ly);
drowstride = tmp.width * 3;
for(y = ly; y < hy; y++)
for(x = lx; x < hx; x++)
memcpy(&tmp.col[(y-ly)*drowstride+(x-lx)*3],
&p->col[y*srowstride+x*3], 3);
killppm(p);
p->col = tmp.col;
p->width = tmp.width;
p->height = tmp.height;
}
void autocrop(struct ppm *p, int room)
{
int lx = 0, hx = p->width, ly = 0, hy = p->height;
int x, y, n = 0;
guchar tc[3];
struct ppm tmp = {0,0,NULL};
int rowstride = p->width * 3;
int drowstride;
/* upper */
memcpy(&tc, p->col, 3);
for(y = 0; y < p->height; y++) {
n = 0;
for(x = 0; x < p->width; x++) {
if(memcmp(&tc, &p->col[y*rowstride+x*3], 3)) { n++; break; }
}
if(n) break;
}
if(n) ly = y;
/* printf("ly = %d\n", ly); */
/* lower */
memcpy(&tc, &p->col[(p->height-1)*rowstride], 3);
for(y = p->height-1; y >= 0; y--) {
n = 0;
for(x = 0; x < p->width; x++) {
if(memcmp(&tc, &p->col[y*rowstride+x*3], 3)) { n++; break; }
}
if(n) break;
}
if(n) hy = y+1;
if(hy >= p->height) hy = p->height - 1;
/* printf("hy = %d\n", hy); */
/* left */
memcpy(&tc, &p->col[ly*rowstride], 3);
for(x = 0; x < p->width; x++) {
n = 0;
for(y = ly; y <= hy && y < p->height; y++) {
if(memcmp(&tc, &p->col[y*rowstride+x*3], 3)) { n++; break; }
}
if(n) break;
}
if(n) lx = x;
/* printf("lx = %d\n", lx); */
/* right */
memcpy(&tc, &p->col[ly*rowstride + (p->width-1)*3], 3);
for(x = p->width-1; x >= 0; x--) {
n = 0;
for(y = ly; y <= hy; y++) {
if(memcmp(&tc, &p->col[y*rowstride+x*3], 3)) { n++; break; }
}
if(n) break;
}
if(n) hx = x+1;
/* printf("hx = %d\n", hx); */
lx -= room; if(lx<0) lx = 0;
ly -= room; if(ly<0) ly = 0;
hx += room; if(hx>=p->width) hx = p->width-1;
hy += room; if(hy>=p->height) hy = p->height-1;
newppm(&tmp, hx-lx, hy-ly);
drowstride = tmp.width * 3;
for(y = ly; y < hy; y++)
for(x = lx; x < hx; x++)
memcpy(&tmp.col[(y-ly)*drowstride+(x-lx)*3],
&p->col[y*rowstride+x*3], 3);
killppm(p);
p->col = tmp.col;
p->width = tmp.width;
p->height = tmp.height;
}
void pad(struct ppm *p, int left,int right, int top, int bottom, guchar *bg)
{
int x, y;
struct ppm tmp = {0,0,NULL};
newppm(&tmp, p->width+left+right, p->height+top+bottom);
for(y = 0; y < tmp.height; y++) {
guchar *row, *srcrow;
row = tmp.col + y * tmp.width * 3;
if((y < top) || (y >= tmp.height-bottom)) {
for(x = 0; x < tmp.width; x++) {
int k = x * 3;
row[k+0] = bg[0];
row[k+1] = bg[1];
row[k+2] = bg[2];
}
continue;
}
srcrow = p->col + (y-top) * p->width * 3;
for(x = 0; x < left; x++) {
int k = x * 3;
row[k+0] = bg[0];
row[k+1] = bg[1];
row[k+2] = bg[2];
}
for(; x < tmp.width-right; x++) {
int k = y * tmp.width * 3 + x * 3;
tmp.col[k+0] = srcrow[(x-left)*3+0];
tmp.col[k+1] = srcrow[(x-left)*3+1];
tmp.col[k+2] = srcrow[(x-left)*3+2];
}
for(; x < tmp.width; x++) {
int k = x * 3;
row[k+0] = bg[0];
row[k+1] = bg[1];
row[k+2] = bg[2];
}
}
killppm(p);
p->width = tmp.width;
p->height = tmp.height;
p->col = tmp.col;
}
void saveppm(struct ppm *p, char *fn)
{
FILE *f = fopen(fn, "wb");
fprintf(f, "P6\n%d %d\n255\n", p->width, p->height);
fwrite(p->col, p->width * 3 * p->height, 1, f);
fclose(f);
}
void edgepad(struct ppm *p, int left,int right, int top, int bottom)
{
int x,y;
struct ppm tmp = {0,0,NULL};
guchar testcol[3] = {0,255,0};
int srowstride, drowstride;
newppm(&tmp, p->width+left+right, p->height+top+bottom);
fill(&tmp, testcol);
srowstride = p->width * 3;
drowstride = tmp.width * 3;
for(y = 0; y < top; y++) {
memcpy(&tmp.col[y*drowstride+left*3], p->col, srowstride);
}
for(; y-top < p->height; y++) {
memcpy(&tmp.col[y*drowstride+left*3], p->col + (y-top)*srowstride, srowstride);
}
for(; y < tmp.height; y++) {
memcpy(&tmp.col[y*drowstride+left*3], p->col + (p->height-1)*srowstride, srowstride);
}
for(y = 0; y < tmp.height; y++) {
guchar *col, *tmprow;
tmprow = tmp.col + y*drowstride;
col = tmp.col + y*drowstride + left*3;
for(x = 0; x < left; x++) {
memcpy(&tmprow[x*3], col, 3);
}
col = tmp.col + y*drowstride + (tmp.width-right-1)*3;
for(x = 0; x < right; x++) {
memcpy(&tmprow[(x+tmp.width-right-1)*3], col, 3);
}
}
killppm(p);
p->width = tmp.width;
p->height = tmp.height;
p->col = tmp.col;
}
void ppmgamma(struct ppm *p, float e, int r, int g, int b)
{
int x, l = p->width * 3 * p->height;
guchar xlat[256], *pix;
if(e > 0.0) for(x = 0; x < 256; x++) {
xlat[x] = pow((x/255.0),(1.0/e))*255.0;
} else if(e < 0.0) for(x = 0; x < 256; x++) {
xlat[255-x] = pow((x/255.0),(-1.0/e))*255.0;
} else for(x = 0; x < 256; x++) { xlat[x] = 0; }
pix = p->col;
if(r) for(x = 0; x < l; x += 3) pix[x] = xlat[pix[x]];
if(g) for(x = 1; x < l; x += 3) pix[x] = xlat[pix[x]];
if(b) for(x = 2; x < l; x += 3) pix[x] = xlat[pix[x]];
}
void ppmbrightness(struct ppm *p, float e, int r, int g, int b)
{
int x, l = p->width * 3 * p->height;
guchar xlat[256], *pix;
for(x = 0; x < 256; x++) {
xlat[x] = x*e;
}
pix = p->col;
if(r) for(x = 0; x < l; x += 3) pix[x] = xlat[pix[x]];
if(g) for(x = 1; x < l; x += 3) pix[x] = xlat[pix[x]];
if(b) for(x = 2; x < l; x += 3) pix[x] = xlat[pix[x]];
}
void blur(struct ppm *p, int xrad, int yrad)
{
int x, y, k;
int tx, ty;
struct ppm tmp = {0,0,NULL};
int r, g, b, n;
int rowstride = p->width * 3;
newppm(&tmp, p->width, p->height);
for(y = 0; y < p->height; y++) {
for(x = 0; x < p->width; x++) {
r = g = b = n = 0;
for(ty = y-yrad; ty <= y+yrad; ty++) {
for(tx = x-xrad; tx <= x+xrad; tx++) {
if(ty<0) continue;
if(ty>=p->height) continue;
if(tx<0) continue;
if(tx>=p->width) continue;
k = ty*rowstride + tx*3;
r += p->col[k+0];
g += p->col[k+1];
b += p->col[k+2];
n++;
}
}
k = y*rowstride + x*3;
tmp.col[k+0] = r / n;
tmp.col[k+1] = g / n;
tmp.col[k+2] = b / n;
}
}
killppm(p);
p->width = tmp.width;
p->height = tmp.height;
p->col = tmp.col;
}
void putrgb_fast(struct ppm *s, float xo, float yo, guchar *d)
{
guchar *tp;
tp = s->col + s->width * 3 * (int)(yo+0.5) + 3 * (int)(xo+0.5);
tp[0] = d[0];
tp[1] = d[1];
tp[2] = d[2];
}
void putrgb(struct ppm *s, float xo, float yo, guchar *d)
{
int x, y;
float aa, ab, ba, bb;
int k, rowstride = s->width * 3;
x = xo;
y = yo;
if((x < 0) || (y < 0) || (x >= s->width-1) || (y >= s->height-1))
return;
xo -= x;
yo -= y;
aa = (1.0-xo)*(1.0-yo);
ab = xo*(1.0-yo);
ba = (1.0-xo)*yo;
bb = xo*yo;
k = y*rowstride + x*3;
s->col[k+0] *= (1.0-aa);
s->col[k+1] *= (1.0-aa);
s->col[k+2] *= (1.0-aa);
s->col[k+3] *= (1.0-ab);
s->col[k+4] *= (1.0-ab);
s->col[k+5] *= (1.0-ab);
s->col[k+rowstride+0] *= (1.0-ba);
s->col[k+rowstride+1] *= (1.0-ba);
s->col[k+rowstride+2] *= (1.0-ba);
s->col[k+rowstride+3] *= (1.0-bb);
s->col[k+rowstride+4] *= (1.0-bb);
s->col[k+rowstride+5] *= (1.0-bb);
s->col[k+0] += aa * d[0];
s->col[k+1] += aa * d[1];
s->col[k+2] += aa * d[2];
s->col[k+3] += ab * d[0];
s->col[k+4] += ab * d[1];
s->col[k+5] += ab * d[2];
s->col[k+rowstride+0] += ba * d[0];
s->col[k+rowstride+1] += ba * d[1];
s->col[k+rowstride+2] += ba * d[2];
s->col[k+rowstride+3] += bb * d[0];
s->col[k+rowstride+4] += bb * d[1];
s->col[k+rowstride+5] += bb * d[2];
}
void drawline(struct ppm *p, float fx, float fy, float tx, float ty, guchar *col)
{
float i;
float d, x, y;
if(fabs(fx-tx) > fabs(fy-ty)) {
if(fx > tx) { i=tx; tx=fx; fx=i; i=ty; ty=fy; fy=i; }
d = (ty-fy)/(tx-fx);
y = fy;
for(x = fx; x <= tx; x+=1.0) {
putrgb(p, x, y, col);
y += d;
}
} else {
if(fy > ty) { i=tx; tx=fx; fx=i; i=ty; ty=fy; fy=i; }
d = (tx-fx)/(ty-fy);
x = fx;
for(y = fy; y <= ty; y+=1.0) {
putrgb(p, x, y, col);
x += d;
}
}
}