#define TWO_PI (4*atan2(1,1) * 2)
-void nmap_to_hmap(unsigned char *map, int w, int h, double scale, double offset)
+void nmap_to_hmap(unsigned char *map, const unsigned char *refmap, int w, int h, double scale, double offset)
{
int x, y;
double nx, ny, nz;
fy -= h;
#ifdef C99
if(fx||fy)
- freqspace1[(w*y+x)] = I * (fx * freqspace1[(w*y+x)] + fy * freqspace2[(w*y+x)]) / (fx*fx + fy*fy) / TWO_PI;
+ freqspace1[(w*y+x)] = _Complex_I * (fx * freqspace1[(w*y+x)] + fy * freqspace2[(w*y+x)]) / (fx*fx + fy*fy) / TWO_PI;
else
freqspace1[(w*y+x)] = 0;
#else
fftw_execute(f12i1);
- if(scale == 0)
+ /* renormalize */
+ for(y = 0; y < h; ++y)
+ for(x = 0; x < w; ++x)
+ {
+#ifdef C99
+ imgspace1[(w*y+x)] /= (w*h);
+#else
+ imgspace1[(w*y+x)][0] /= (w*h);
+ imgspace1[(w*y+x)][1] /= (w*h);
+#endif
+ }
+
+ if(refmap)
+ {
+ double f, a;
+ double o, s;
+ double sa, sfa, sffa, sfva, sva;
+ double mi, ma;
+ sa = sfa = sffa = sfva = sva = 0;
+ mi = 1;
+ ma = -1;
+ for(y = 0; y < h; ++y)
+ for(x = 0; x < w; ++x)
+ {
+ a = (int)refmap[(w*y+x)*4+3];
+ v = (refmap[(w*y+x)*4+0]*0.114 + refmap[(w*y+x)*4+1]*0.587 + refmap[(w*y+x)*4+2]*0.299);
+ v = (v - 128.0) / 127.0;
+#ifdef C99
+ f = creal(imgspace1[(w*y+x)]);
+#else
+ f = imgspace1[(w*y+x)][0];
+#endif
+ if(a <= 0)
+ continue;
+ if(v < mi)
+ mi = v;
+ if(v > ma)
+ ma = v;
+ sa += a;
+ sfa += f*a;
+ sffa += f*f*a;
+ sfva += f*v*a;
+ sva += v*a;
+ }
+ if(mi < ma)
+ {
+ /* linear regression ftw */
+ o = (sfa*sfva - sffa*sva) / (sfa*sfa-sa*sffa);
+ s = (sfa*sva - sa*sfva) / (sfa*sfa-sa*sffa);
+ }
+ else /* all values of v are equal, so we cannot get scale; we can still get offset */
+ {
+ o = ((sva - sfa) / sa);
+ s = 1;
+ }
+
+ printf("Ref-computed scale: %f\nRef-computed offset: %f\n", s, o);
+
+ /*
+ * now apply user-given offset and scale to these values
+ * (x * s + o) * scale + offset
+ * x * s * scale + o * scale + offset
+ */
+ offset += o * scale;
+ scale *= s;
+ }
+ else if(scale == 0)
{
#ifdef C99
vmin = vmax = creal(imgspace1[0]);
vmax = v;
}
- vmin /= (w*h);
- vmax /= (w*h);
-
/*
* map vmin to -1
* map vmax to +1
printf("Autocomputed scale: %f\nAutocomputed offset: %f\n", scale, offset);
}
- scale /= (w*h);
-
for(y = 0; y < h; ++y)
for(x = 0; x < w; ++x)
{
freqspace1[(w*y+x)] *= 1 - pow(abs(fx) / (double)(w/2), 1);
freqspace1[(w*y+x)] *= 1 - pow(abs(fy) / (double)(h/2), 1);
- freqspace2[(w*y+x)] = TWO_PI*I * fy * freqspace1[(w*y+x)]; /* y derivative */
- freqspace1[(w*y+x)] = TWO_PI*I * fx * freqspace1[(w*y+x)]; /* x derivative */
+ freqspace2[(w*y+x)] = TWO_PI*_Complex_I * fy * freqspace1[(w*y+x)]; /* y derivative */
+ freqspace1[(w*y+x)] = TWO_PI*_Complex_I * fx * freqspace1[(w*y+x)]; /* x derivative */
#else
/* a lowpass to prevent the worst */
freqspace1[(w*y+x)][0] *= 1 - pow(abs(fx) / (double)(w/2), 1);
int usage(const char *me)
{
- printf("Usage: %s <infile_norm.tga> <outfile_normandheight.tga> [<scale> [<offset>]] (get heightmap from normalmap)\n", me);
- printf("or: %s <infile_height.tga> <outfile_normandheight.tga> -1 [<scale>] (read from R, Diff)\n", me);
+ printf("Usage: %s <infile_norm.tga> <outfile_normandheight.tga> [<scale> [<offset> [<infile_ref.tga>]]] (get heightmap from normalmap)\n", me);
+ printf("or: %s <infile_height.tga> <outfile_normandheight.tga> -1 [<scale>] (read from B, Diff)\n", me);
printf("or: %s <infile_height.tga> <outfile_normandheight.tga> -2 [<scale>] (read from G, Diff)\n", me);
printf("or: %s <infile_height.tga> <outfile_normandheight.tga> -3 [<scale>] (read from R, Diff)\n", me);
printf("or: %s <infile_height.tga> <outfile_normandheight.tga> -4 [<scale>] (read from A, Diff)\n", me);
printf("or: %s <infile_height.tga> <outfile_normandheight.tga> -5 [<scale>] (read from (R+G+B)/3, Diff)\n", me);
printf("or: %s <infile_height.tga> <outfile_normandheight.tga> -6 [<scale>] (read from Y, Diff)\n", me);
- printf("or: %s <infile_height.tga> <outfile_normandheight.tga> -7 [<scale>] (read from R, FFT)\n", me);
+ printf("or: %s <infile_height.tga> <outfile_normandheight.tga> -7 [<scale>] (read from B, FFT)\n", me);
printf("or: %s <infile_height.tga> <outfile_normandheight.tga> -8 [<scale>] (read from G, FFT)\n", me);
printf("or: %s <infile_height.tga> <outfile_normandheight.tga> -9 [<scale>] (read from R, FFT)\n", me);
printf("or: %s <infile_height.tga> <outfile_normandheight.tga> -10 [<scale>] (read from A, FFT)\n", me);
int main(int argc, char **argv)
{
- const char *infile, *outfile;
+ const char *infile, *outfile, *reffile;
double scale, offset;
- int nmaplen;
- unsigned char *nmapdata, *nmap;
+ int nmaplen, w, h;
+ unsigned char *nmapdata, *nmap, *refmap;
if(argc > 1)
infile = argv[1];
if(argc > 4)
offset = atof(argv[4]);
else
- offset = 0;
+ offset = (scale<0) ? 1 : 0;
+
+ if(argc > 5)
+ reffile = argv[5];
+ else
+ reffile = NULL;
nmapdata = FS_LoadFile(infile, &nmaplen);
if(!nmapdata)
printf("LoadTGA_BGRA failed\n");
return 2;
}
+ w = image_width;
+ h = image_height;
+
+ if(reffile)
+ {
+ nmapdata = FS_LoadFile(reffile, &nmaplen);
+ if(!nmapdata)
+ {
+ printf("FS_LoadFile failed\n");
+ return 2;
+ }
+ refmap = LoadTGA_BGRA(nmapdata, nmaplen);
+ free(nmapdata);
+ if(!refmap)
+ {
+ printf("LoadTGA_BGRA failed\n");
+ return 2;
+ }
+ if(image_width != w || image_height != h)
+ {
+ printf("reference map must have same size as input normalmap\n");
+ return 2;
+ }
+ }
+ else
+ refmap = NULL;
+
if(scale < -6)
hmap_to_nmap(nmap, image_width, image_height, -scale-7, offset);
else if(scale < 0)
hmap_to_nmap_local(nmap, image_width, image_height, -scale-1, offset);
else
- nmap_to_hmap(nmap, image_width, image_height, scale, offset);
+ nmap_to_hmap(nmap, refmap, image_width, image_height, scale, offset);
if(!Image_WriteTGABGRA(outfile, image_width, image_height, nmap))
{
printf("Image_WriteTGABGRA failed\n");