]> git.xonotic.org Git - xonotic/netradiant.git/blobdiff - tools/quake3/q3map2/image.c
q3map2: dds/ prefix, print log at the right time, when image is found
[xonotic/netradiant.git] / tools / quake3 / q3map2 / image.c
index 2af46a6f7721761f3b2f30f68e95471d46481751..20ad90ff99d59350b5b58cf34982c7bea2ec5220 100644 (file)
@@ -36,7 +36,7 @@
 /* dependencies */
 #include "q3map2.h"
 
-
+#include "webp/decode.h"
 
 /* -------------------------------------------------------------------------------
 
@@ -67,13 +67,13 @@ static void LoadDDSBuffer( byte *buffer, int size, byte **pixels, int *width, in
 
        /* get dds info */
        if ( DDSGetInfo( (ddsBuffer_t*) buffer, &w, &h, &pf ) ) {
-               Sys_Printf( "WARNING: Invalid DDS texture\n" );
+               Sys_FPrintf( SYS_WRN, "WARNING: Invalid DDS texture\n" );
                return;
        }
 
        /* only certain types of dds textures are supported */
        if ( pf != DDS_PF_ARGB8888 && pf != DDS_PF_DXT1 && pf != DDS_PF_DXT3 && pf != DDS_PF_DXT5 ) {
-               Sys_Printf( "WARNING: Only DDS texture formats ARGB8888, DXT1, DXT3, and DXT5 are supported (%d)\n", pf );
+               Sys_FPrintf( SYS_WRN, "WARNING: Only DDS texture formats ARGB8888, DXT1, DXT3, and DXT5 are supported (%d)\n", pf );
                return;
        }
 
@@ -86,6 +86,28 @@ static void LoadDDSBuffer( byte *buffer, int size, byte **pixels, int *width, in
        DDSDecompress( (ddsBuffer_t*) buffer, *pixels );
 }
 
+#ifdef BUILD_CRUNCH
+/*
+    LoadCRNBuffer
+    loads a crn image into a valid rgba image
+*/
+void LoadCRNBuffer( byte *buffer, int size, byte **pixels, int *width, int *height) {
+       /* dummy check */
+       if ( buffer == NULL || size <= 0 || pixels == NULL || width == NULL || height == NULL ) {
+               return;
+       }
+       if ( !GetCRNImageSize( buffer, size, width, height ) ) {
+               Sys_FPrintf( SYS_WRN, "WARNING: Error getting crn imag dimensions.\n");;
+               return;
+       }
+       int outBufSize = *width * *height * 4;
+       *pixels = safe_malloc( outBufSize );
+       if ( !ConvertCRNtoRGBA( buffer, size, outBufSize, *pixels) ) {
+               Sys_FPrintf( SYS_WRN, "WARNING: Error decoding crn image.\n", 0 );
+               return;
+       }
+}
+#endif // BUILD_CRUNCH
 
 
 /*
@@ -97,7 +119,7 @@ static void LoadDDSBuffer( byte *buffer, int size, byte **pixels, int *width, in
 typedef struct pngBuffer_s
 {
        byte    *buffer;
-       int size, offset;
+       png_size_t size, offset;
 } pngBuffer_t;
 
 void PNGReadData( png_struct *png, png_byte *buffer, png_size_t size ){
@@ -122,9 +144,9 @@ void PNGReadData( png_struct *png, png_byte *buffer, png_size_t size ){
 static void LoadPNGBuffer( byte *buffer, int size, byte **pixels, int *width, int *height ){
        png_struct  *png;
        png_info    *info, *end;
-    pngBuffer_t pb;
-       int i, bitDepth, colorType, channels;
-       png_uint_32 w, h;
+       pngBuffer_t pb;
+       int bitDepth, colorType;
+       png_uint_32 w, h, i;
        byte        **rowPointers;
 
        /* dummy check */
@@ -139,40 +161,40 @@ static void LoadPNGBuffer( byte *buffer, int size, byte **pixels, int *width, in
 
        /* determine if this is a png file */
        if ( png_sig_cmp( buffer, 0, 8 ) != 0 ) {
-               Sys_Printf( "WARNING: Invalid PNG file\n" );
+               Sys_FPrintf( SYS_WRN, "WARNING: Invalid PNG file\n" );
                return;
        }
 
        /* create png structs */
        png = png_create_read_struct( PNG_LIBPNG_VER_STRING, NULL, NULL, NULL );
        if ( png == NULL ) {
-               Sys_Printf( "WARNING: Unable to create PNG read struct\n" );
+               Sys_FPrintf( SYS_WRN, "WARNING: Unable to create PNG read struct\n" );
                return;
        }
 
        info = png_create_info_struct( png );
        if ( info == NULL ) {
-               Sys_Printf( "WARNING: Unable to create PNG info struct\n" );
+               Sys_FPrintf( SYS_WRN, "WARNING: Unable to create PNG info struct\n" );
                png_destroy_read_struct( &png, NULL, NULL );
                return;
        }
 
        end = png_create_info_struct( png );
        if ( end == NULL ) {
-               Sys_Printf( "WARNING: Unable to create PNG end info struct\n" );
+               Sys_FPrintf( SYS_WRN, "WARNING: Unable to create PNG end info struct\n" );
                png_destroy_read_struct( &png, &info, NULL );
                return;
        }
 
        /* set read callback */
-    pb.buffer = buffer;
-    pb.size = size;
-    pb.offset = 0;
-       png_set_read_fn( png, (void*)&pb, PNGReadData );
+       pb.buffer = buffer;
+       pb.size = size;
+       pb.offset = 0;
+       png_set_read_fn( png, &pb, PNGReadData );
 
        /* set error longjmp */
        if ( setjmp( png_jmpbuf(png) ) ) {
-               Sys_Printf( "WARNING: An error occurred reading PNG image\n" );
+               Sys_FPrintf( SYS_WRN, "WARNING: An error occurred reading PNG image\n" );
                png_destroy_read_struct( &png, &info, &end );
                return;
        }
@@ -186,9 +208,6 @@ static void LoadPNGBuffer( byte *buffer, int size, byte **pixels, int *width, in
        png_get_IHDR( png, info,
                                  &w, &h, &bitDepth, &colorType, NULL, NULL, NULL );
 
-       /* read number of channels */
-       channels = png_get_channels( png, info );
-
        /* the following will probably bork on certain types of png images, but hey... */
 
        /* force indexed/gray/trans chunk to rgb */
@@ -211,7 +230,8 @@ static void LoadPNGBuffer( byte *buffer, int size, byte **pixels, int *width, in
        /* create image pixel buffer */
        *width = w;
        *height = h;
-       *pixels = safe_malloc( w * h * 4 );
+       /* initialize with zeros, this memory area may not be entirely rewritten */
+       *pixels = safe_malloc0( w * h * 4 );
 
        /* create row pointers */
        rowPointers = safe_malloc( h * sizeof( byte* ) );
@@ -224,7 +244,34 @@ static void LoadPNGBuffer( byte *buffer, int size, byte **pixels, int *width, in
        /* clean up */
        free( rowPointers );
        png_destroy_read_struct( &png, &info, &end );
+}
+
 
+
+static void LoadWEBPBuffer( byte *buffer, int size, byte **pixels, int *width, int *height ){
+
+       int image_width;
+       int image_height;
+       
+       if ( !WebPGetInfo( buffer, ( size_t) size, &image_width, &image_height ) )
+       {
+               Sys_FPrintf( SYS_WRN, "WARNING: An error occurred reading WEBP image info\n" );
+               return;
+       }
+
+       /* create image pixel buffer */
+       *pixels = safe_malloc( image_width * image_height * 4 );
+       *width = image_width;
+       *height = image_height;
+
+       int out_stride = image_width  * 4;
+       int out_size =  image_height * out_stride;
+
+               if ( !WebPDecodeRGBAInto( buffer, (size_t) size, *pixels, out_size, out_stride ) )
+               {
+               Sys_FPrintf( SYS_WRN, "WARNING: An error occurred reading WEBP image\n" );
+                       return;
+               }
 }
 
 
@@ -338,6 +385,7 @@ image_t *ImageLoad( const char *filename ){
        char name[ 1024 ];
        int size;
        byte        *buffer = NULL;
+       qboolean alphaHack = qfalse;
 
 
        /* init */
@@ -384,54 +432,81 @@ image_t *ImageLoad( const char *filename ){
        size = vfsLoadFile( (const char*) name, (void**) &buffer, 0 );
        if ( size > 0 ) {
                LoadTGABuffer( buffer, buffer + size, &image->pixels, &image->width, &image->height );
+               goto image_load_success;
        }
-       else
-       {
-               /* attempt to load png */
+
+       /* attempt to load png */
+       StripExtension( name );
+       strcat( name, ".png" );
+       size = vfsLoadFile( (const char*) name, (void**) &buffer, 0 );
+       if ( size > 0 ) {
+               LoadPNGBuffer( buffer, size, &image->pixels, &image->width, &image->height );
+               goto image_load_success;
+       }
+
+       /* attempt to load jpg */
+       StripExtension( name );
+       strcat( name, ".jpg" );
+       size = vfsLoadFile( (const char*) name, (void**) &buffer, 0 );
+       if ( size > 0 ) {
+               if ( LoadJPGBuff( buffer, size, &image->pixels, &image->width, &image->height ) == -1 && image->pixels != NULL ) {
+                       // On error, LoadJPGBuff might store a pointer to the error message in image->pixels
+                       Sys_FPrintf( SYS_WRN, "WARNING: LoadJPGBuff: %s\n", (unsigned char*) image->pixels );
+               }
+               alphaHack = qtrue;
+               goto image_load_success;
+       }
+
+       /* attempt to load dds */
+       StripExtension( name );
+       strcat( name, ".dds" );
+       size = vfsLoadFile( (const char*) name, (void**) &buffer, 0 );
+
+       /* also look for .dds image in dds/ prefix like Doom3 or DarkPlaces */
+       if ( size <= 0 ) {
+               strcpy( name, "dds/" );
+               strcat( name, image->name );
                StripExtension( name );
-               strcat( name, ".png" );
+               strcat( name, ".dds" );
                size = vfsLoadFile( (const char*) name, (void**) &buffer, 0 );
-               if ( size > 0 ) {
-                       LoadPNGBuffer( buffer, size, &image->pixels, &image->width, &image->height );
-               }
-               else
-               {
-                       /* attempt to load jpg */
-                       StripExtension( name );
-                       strcat( name, ".jpg" );
-                       size = vfsLoadFile( (const char*) name, (void**) &buffer, 0 );
-                       if ( size > 0 ) {
-                               if ( LoadJPGBuff( buffer, size, &image->pixels, &image->width, &image->height ) == -1 && image->pixels != NULL ) {
-                                       Sys_Printf( "WARNING: LoadJPGBuff: %s\n", (unsigned char*) image->pixels );
-                               }
-                       }
-                       else
-                       {
-                               /* attempt to load dds */
-                               StripExtension( name );
-                               strcat( name, ".dds" );
-                               size = vfsLoadFile( (const char*) name, (void**) &buffer, 0 );
-                               if ( size > 0 ) {
-                                       LoadDDSBuffer( buffer, size, &image->pixels, &image->width, &image->height );
-
-                                       /* debug code */
-                                       #if 1
-                                       {
-                                               ddsPF_t pf;
-                                               DDSGetInfo( (ddsBuffer_t*) buffer, NULL, NULL, &pf );
-                                               Sys_Printf( "pf = %d\n", pf );
-                                               if ( image->width > 0 ) {
-                                                       StripExtension( name );
-                                                       strcat( name, "_converted.tga" );
-                                                       WriteTGA( "C:\\games\\quake3\\baseq3\\textures\\rad\\dds_converted.tga", image->pixels, image->width, image->height );
-                                               }
-                                       }
-                                       #endif
-                               }
-                       }
-               }
        }
 
+       if ( size > 0 ) {
+               LoadDDSBuffer( buffer, size, &image->pixels, &image->width, &image->height );
+               goto image_load_success;
+       }
+
+       /* attempt to load ktx */
+       StripExtension( name );
+       strcat( name, ".ktx" );
+       size = vfsLoadFile( (const char*) name, (void**) &buffer, 0 );
+       if ( size > 0 ) {
+               LoadKTXBufferFirstImage( buffer, size, &image->pixels, &image->width, &image->height );
+               goto image_load_success;
+       }
+
+       #ifdef BUILD_CRUNCH
+       /* attempt to load crn */
+       StripExtension( name );
+       strcat( name, ".crn" );
+       size = vfsLoadFile( ( const char* ) name, ( void** ) &buffer, 0 );
+       if ( size > 0 ) {
+               LoadCRNBuffer( buffer, size, &image->pixels, &image->width, &image->height );
+               goto image_load_success;
+       }
+       #endif // BUILD_CRUNCH
+
+       /* attempt to load webp */
+       StripExtension( name );
+       strcat( name, ".webp" );
+       size = vfsLoadFile( (const char*) name, (void**) &buffer, 0 );
+       if ( size > 0 ) {
+               LoadWEBPBuffer( buffer, size, &image->pixels, &image->width, &image->height );
+               goto image_load_success;
+       }
+
+       image_load_success:
+
        /* free file buffer */
        free( buffer );
 
@@ -444,6 +519,9 @@ image_t *ImageLoad( const char *filename ){
                return NULL;
        }
 
+       /* tell user which image file is found for the given texture path */
+       Sys_FPrintf( SYS_VRB, "Loaded image: \"%s\"\n", name );
+
        /* set filename */
        image->filename = safe_malloc( strlen( name ) + 1 );
        strcpy( image->filename, name );
@@ -452,6 +530,30 @@ image_t *ImageLoad( const char *filename ){
        image->refCount = 1;
        numImages++;
 
+       if ( alphaHack ) {
+               StripExtension( name );
+               strcat( name, "_alpha.jpg" );
+               size = vfsLoadFile( (const char*) name, (void**) &buffer, 0 );
+               if ( size > 0 ) {
+                       unsigned char *pixels;
+                       int width, height;
+                       if ( LoadJPGBuff( buffer, size, &pixels, &width, &height ) == -1 ) {
+                               if (pixels) {
+                                       // On error, LoadJPGBuff might store a pointer to the error message in pixels
+                                       Sys_FPrintf( SYS_WRN, "WARNING: LoadJPGBuff %s %s\n", name, (unsigned char*) pixels );
+                               }
+                       } else {
+                               if ( width == image->width && height == image->height ) {
+                                       int i;
+                                       for ( i = 0; i < width * height; ++i )
+                                               image->pixels[4 * i + 3] = pixels[4 * i + 2];  // copy alpha from blue channel
+                               }
+                               free( pixels );
+                       }
+                       free( buffer );
+               }
+       }
+
        /* return the image */
        return image;
 }