heretic2-sdk/Toolkit/Programming/Tools/qdata/tmix.c
1998-11-24 00:00:00 +00:00

677 lines
13 KiB
C

#include "qdata.h"
#include "flex.h"
#define MAXFILES 2048
typedef struct
{
int x;
int y;
int w;
int h;
int cw;
int ch;
int rw;
int index;
int depth;
int col;
int baseline;
char name[128];
} Coords;
int filenum;
int valid;
Coords in[MAXFILES];
Coords out;
char outscript[256];
char sourcedir[256];
char outusage[256];
char root[32];
int destsize = 0;
byte *pixels = NULL; // Buffer to load image
long *outpixels = NULL; // Buffer to store combined textures
long *usagemap = NULL; // Buffer of usage map
void *bmptemp = NULL; // Buffer of usage map
byte *map = NULL;
int xcharsize;
int ycharsize;
int dosort = 0;
int missed = 0;
int overlap = 0;
int nobaseline = 0;
int percent;
//////////////////////////////////////////////////
// Setting the char based usage map //
//////////////////////////////////////////////////
byte TryPlace(Coords *coord)
{
int x, y;
byte entry = 0;
byte *mapitem;
mapitem = map + (coord->x / xcharsize) + ((coord->y / ycharsize) * out.cw);
for (y = 0; y < coord->ch; y++, mapitem += out.cw - coord->cw)
{
for (x = 0; x < coord->cw; x++)
{
if (entry |= *mapitem++ & 8)
{
return(entry);
}
}
}
return(entry);
}
void SetMap(Coords *coord)
{
int x, y;
byte *mapitem;
mapitem = map + (coord->x / xcharsize) + ((coord->y / ycharsize) * out.cw);
for (y = 0; y < coord->ch; y++, mapitem += out.cw - coord->cw)
for (x = 0; x < coord->cw; x++)
*mapitem++ |= 8;
}
//////////////////////////////////////////////////
// Setting the pixel based usage map //
//////////////////////////////////////////////////
void CheckOverlap(Coords *coord)
{
int x;
int y;
long *dest;
x = coord->x;
y = coord->y;
dest = (long *)(usagemap + x + (y * out.w));
for (y = 0; y < coord->h; y++, dest += out.w - coord->w)
{
for (x = 0; x < coord->w; x++)
{
if (*dest++)
{
overlap++;
return;
}
}
}
}
void SetUsageMap(Coords *coord)
{
int x;
int y;
long *dest;
x = coord->x;
y = coord->y;
dest = (long *)(usagemap + x + (y * out.w));
for (y = 0; y < coord->h; y++, dest += out.w - coord->w)
{
for (x = 0; x < coord->w; x++)
{
*dest++ = coord->col;
}
}
}
//////////////////////////////////////////////////
// Flips the BMP image to the correct way up //
//////////////////////////////////////////////////
void CopyLine(byte *dest, byte *src, int size)
{
int x;
for (x = 0; x < size; x++)
*dest++ = *src++;
}
/****************************************************/
/* Printing headers etc */
/****************************************************/
void RemoveLeading(char *name)
{
int i;
char temp[128];
for(i = strlen(name) - 1; i > 0; i--)
{
if((name[i] == '\\') || (name[i] == '/'))
{
strcpy(temp, name + i + 1);
strcpy(name, temp);
return;
}
}
}
void RemoveExt(char *name)
{
while ((*name != '.') && *name)
name++;
*name = 0;
}
/****************************************************/
/* Misc calcualtions */
/****************************************************/
int TotalArea()
{
int i;
int total = 0;
for (i = 0; i < (filenum + 2); i++)
total += in[i].w * in[i].h;
return(total);
}
/****************************************************/
/* Setup and checking of all info */
/****************************************************/
void InitVars()
{
filenum = 0;
valid = 0;
dosort = 0;
missed = 0;
overlap = 0;
nobaseline = 0;
memset(outscript, 0, sizeof(outscript));
memset(outscript, 0, sizeof(sourcedir));
memset(outscript, 0, sizeof(outusage));
memset(outscript, 0, sizeof(root));
memset(in, 0, sizeof(in));
memset(&out, 0, sizeof(out));
}
void Cleanup()
{
if (pixels)
free(pixels);
if (usagemap)
free(usagemap);
if (outpixels)
free(outpixels);
if (bmptemp)
free(bmptemp);
if (map)
free(map);
}
typedef struct glxy_s
{
float xl, yt, xr, yb;
int w, h, baseline;
} glxy_t;
int SaveScript(char *name)
{
FILE *fp;
int i, j;
glxy_t buff;
if(fp = fopen(name, "wb"))
{
for (j = 0; j < filenum; j++)
{
for (i = 0; i < filenum; i++)
{
if (in[i].index == j)
{
if (in[i].depth)
{
buff.xl = (float)in[i].x / (float)out.w;
buff.yt = (float)in[i].y / (float)out.h;
buff.xr = ((float)in[i].w + (float)in[i].x) / (float)out.w;
buff.yb = ((float)in[i].h + (float)in[i].y) / (float)out.h;
buff.w = in[i].w;
buff.h = in[i].h;
buff.baseline = in[i].baseline;
}
else
{
memset(&buff, 0, sizeof(glxy_t));
}
fwrite(&buff, 1, sizeof(glxy_t), fp);
i = filenum;
}
}
}
fclose(fp);
return(true);
}
else
return(false);
}
int GetScriptInfo(char *name)
{
FILE *fp;
char buffer[256];
char tempbuff[256];
char delims[] = {" \t,\n"};
printf("Opening script file %s.\n", name);
if (fp = fopen(name, "r"))
{
while(fgets(buffer, 256, fp))
{
if (strncmp(buffer, "//", 2) && strncmp(buffer, "\n", 1))
{
strupr(buffer);
strcpy(tempbuff, buffer);
if (strcmp(strtok(tempbuff, delims), "OUTPUT") == 0)
{
strcpy(out.name, strtok(NULL, delims));
strlwr(out.name);
}
strcpy(tempbuff, buffer);
if (strcmp(strtok(tempbuff, delims), "SOURCEDIR") == 0)
{
strcpy(tempbuff, strtok(NULL, delims));
strcpy(sourcedir, ExpandPathAndArchive(tempbuff));
}
strcpy(tempbuff, buffer);
if (strcmp(strtok(tempbuff, delims), "DOSORT") == 0)
dosort = 1;
strcpy(tempbuff, buffer);
if (strcmp(strtok(tempbuff, delims), "XCHARSIZE") == 0)
xcharsize = strtol(strtok(NULL, delims), NULL, 0);
strcpy(tempbuff, buffer);
if (strcmp(strtok(tempbuff, delims), "YCHARSIZE") == 0)
ycharsize = strtol(strtok(NULL, delims), NULL, 0);
strcpy(tempbuff, buffer);
if (strcmp(strtok(tempbuff, delims), "OUTSCRIPT") == 0)
{
strcpy(outscript, strtok(NULL, delims));
strlwr(outscript);
}
strcpy(tempbuff, buffer);
if (strcmp(strtok(tempbuff, delims), "OUTUSAGE") == 0)
strcpy(outusage, strtok(NULL, delims));
strcpy(tempbuff, buffer);
if (strcmp(strtok(tempbuff, delims), "POS") == 0)
{
out.w = strtol(strtok(NULL, delims), NULL, 0);
out.h = strtol(strtok(NULL, delims), NULL, 0);
}
strcpy(tempbuff, buffer);
if (strcmp(strtok(tempbuff, delims), "FILE") == 0)
{
strcpy(in[filenum].name, strtok(NULL, delims));
in[filenum].x = strtol(strtok(NULL, delims), NULL, 0);
in[filenum].y = strtol(strtok(NULL, delims), NULL, 0);
in[filenum].col = strtol(strtok(NULL, delims), NULL, 0);
filenum++;
}
}
}
fclose(fp);
return(true);
}
else
{
printf("ERROR : Could not open script file.\n");
return(false);
}
}
int CheckVars()
{
int i;
if (out.name[0] == 0)
{
printf("ERROR : No output name specified.\n");
return(false);
}
if ((out.w <= 0) || (out.h <= 0))
{
printf("ERROR : Invalid VRAM coordinates.\n");
return(false);
}
if (filenum == 0)
{
printf("ERROR : No input files specified.\n");
return(false);
}
for (i = 0; i < filenum; i++)
if (in[i].name[0] == 0)
{
printf("ERROR : Input filename invalid.\n");
return(false);
}
return(true);
}
// Makes sure texture is totally within the output area
int CheckCoords(Coords *coord)
{
if ((coord->x + coord->w) > out.w)
return(false);
if ((coord->y + coord->h) > out.h)
return(false);
return(true);
}
// Gets the width, height, palette width and palette height of each BMP file
int GetFileDimensions()
{
int i;
int width, height;
char name[128];
for (i = 0; i < filenum; i++)
{
in[i].index = i;
strcpy(name, sourcedir);
strcat(name, in[i].name);
printf("Getting file dimensions, file : %s \r", in[i].name);
if(FileExists(name))
{
LoadAnyImage(name, NULL, NULL, &width, &height);
in[i].depth = 32;
in[i].rw = width;
in[i].w = width; // makes it width in
in[i].h = height;
in[i].cw = (in[i].w + (xcharsize - 1)) / xcharsize;
in[i].ch = (in[i].h + (ycharsize - 1)) / ycharsize;
if (!CheckCoords(&in[i]) && (in[i].x >= 0))
{
printf("Error : texture %s out of bounds.\n", in[i].name);
return(false);
}
valid++;
}
else
{
in[i].depth = 0;
in[i].x = -1;
in[i].y = -1;
in[i].w = 0;
in[i].h = 0;
}
}
printf("\n\n");
return(true);
}
// Sorts files into order for optimal space finding
// Fixed position ones first, followed by the others in descending size
// The theory being that it is easier to find space for smaller textures.
// size = (width + height)
// For space finding it is easier to place a 32x32 than a 128x2
#define WEIGHT 0x8000
void Swap(Coords *a, Coords *b)
{
Coords c;
c = *a;
*a = *b;
*b = c;
}
void SortInNames()
{
int i, j;
int largest, largcount;
int size;
printf("Sorting filenames by size.\n\n");
for (j = 0; j < filenum; j++)
{
largest = -1;
largcount = -1;
for (i = j; i < filenum; i++)
{
if (in[i].depth)
{
size = in[i].w + in[i].h;
if ((in[i].x < 0) && (size > largest))
{
largcount = i;
largest = size;
}
}
}
if ((largcount >= 0) && (largcount != j))
Swap(&in[j], &in[largcount]);
}
}
int SetVars(char *name)
{
if (!GetScriptInfo(name))
return(false);
if (!CheckVars())
return(false);
destsize = out.w * out.h;
out.cw = out.w / xcharsize;
out.ch = out.h / ycharsize;
if ((usagemap = (long *)SafeMalloc(destsize * 4, "")) == NULL)
return(false);
if ((outpixels = (long *)SafeMalloc(destsize * 4, "")) == NULL)
return(false);
if ((bmptemp = (void *)SafeMalloc(destsize * 4, "")) == NULL)
return(false);
if ((map = (byte *)SafeMalloc(destsize / (xcharsize * ycharsize), "")) == NULL)
return(false);
if (GetFileDimensions() == false)
return(false);
if (dosort)
SortInNames();
return(true);
}
/****************************************************/
/* Actual copying routines */
/****************************************************/
int FindCoords(Coords *coord)
{
int tx, ty;
if (coord->x >= 0)
{
SetMap(coord);
return(true);
}
else
{
for (ty = 0; ty < out.ch; ty++)
{
for (tx = 0; tx < out.cw; tx++)
{
coord->x = (tx * xcharsize);
coord->y = (ty * ycharsize);
if (CheckCoords(coord) && !TryPlace(coord))
{
SetMap(coord);
return(true);
}
}
}
}
coord->x = -1;
coord->y = -1;
return(false);
}
void CheckBaseline(int i)
{
int y;
long *pix;
in[i].baseline = -1;
pix = (long *)pixels;
for(y = 0; y < in[i].h; y++, pix += in[i].w)
{
if((*pix & 0x00ffffff) == 0x00ff00ff)
{
in[i].baseline = y;
break;
}
}
pix = (long *)pixels;
for(y = 0; y < in[i].w * in[i].h; y++, pix++)
{
if((*pix & 0x00ffffff) == 0x00ff00ff)
{
*pix = 0;
}
}
if(in[i].baseline == -1)
{
printf("\nERROR : %s has no baseline\n", in[i].name);
nobaseline++;
}
}
void CopyToMain32(Coords *coord)
{
int x;
int y;
long *source;
long *dest;
x = coord->x;
y = coord->y;
source = (long *)pixels;
dest = (long *)(outpixels + x + (y * out.w));
for (y = 0; y < coord->h; y++, dest += out.w - coord->w)
{
for (x = 0; x < coord->w; x++)
{
*dest++ = *source++;
}
}
}
void CreateMain()
{
int i, count;
int width, height;
char name[128];
for (i = 0, count = 0; i < filenum; i++)
{
if (in[i].depth)
{
printf("\rProcessing %d of %d (%d missed, %d overlapping, %d nobase)\r", count + 1, valid, missed, overlap, nobaseline);
count++;
if (!FindCoords(&in[i]))
missed++;
else
{
strcpy(name, sourcedir);
strcat(name, in[i].name);
LoadAnyImage(name, &pixels, NULL, &width, &height);
CheckBaseline(i);
CheckOverlap(&in[i]);
CopyToMain32(&in[i]);
SetUsageMap(&in[i]);
}
}
}
}
void Cmd_TextureMix()
{
miptex32_t *qtex32;
char filename[1024];
int size;
InitVars();
GetScriptToken (false);
strcpy(root, token);
RemoveExt(root);
RemoveLeading(root);
strcpy(filename, ExpandPathAndArchive(token));
if (SetVars(filename))
{
// Create combined texture
percent = ((TotalArea() * 100) / (out.w * out.h));
printf("Total area consumed : %d%%\n", percent);
printf("Texture resolution : %dx%d pixels.\n", xcharsize, ycharsize);
CreateMain();
// Save image as m32
sprintf (filename, "%spics/misc/%s.m32", gamedir, out.name);
qtex32 = CreateMip32((unsigned long *)outpixels, out.w, out.h, &size, false);
qtex32->contents = 0;
qtex32->value = 0;
qtex32->scale_x = 1.0;
qtex32->scale_y = 1.0;
sprintf (qtex32->name, "misc/%s", out.name);
printf ("\n\nwriting %s\n", filename);
SaveFile (filename, (byte *)qtex32, size);
free (qtex32);
// Save out script file
sprintf (filename, "%spics/misc/%s.fnt", gamedir, outscript);
printf("Writing %s as script file\n", filename);
if (!SaveScript(filename))
{
printf("Unable to save output script.\n");
}
}
printf("Everythings groovy.\n");
Cleanup();
}
// end