/* * DDS GIMP plugin * * Copyright (C) 2004-2012 Shawn Kirst , * with parts (C) 2003 Arne Reuter where specified. * * This program is free software; you can redistribute it and/or * modify it under the terms of the GNU 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 * General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; see the file COPYING. If not, write to * the Free Software Foundation, 51 Franklin Street, Fifth Floor * Boston, MA 02110-1301, USA. */ /* ** !!! COPYRIGHT NOTICE !!! ** ** The following is based on code (C) 2003 Arne Reuter ** URL: http://www.dr-reuter.de/arne/dds.html ** */ #include "config.h" #include #include #include #include #include #include #include #include #include "ddsplugin.h" #include "dds.h" #include "dxt.h" #include "endian_rw.h" #include "misc.h" #include "imath.h" typedef struct { unsigned char rshift, gshift, bshift, ashift; unsigned char rbits, gbits, bbits, abits; unsigned int rmask, gmask, bmask, amask; unsigned int bpp, gimp_bpp; int tile_height; unsigned char *palette; } dds_load_info_t; static int read_header (dds_header_t *hdr, FILE *fp); static int read_header_dx10 (dds_header_dx10_t *hdr, FILE *fp); static int validate_header (dds_header_t *hdr); static int setup_dxgi_format (dds_header_t *hdr, dds_header_dx10_t *dx10hdr); static int load_layer (FILE *fp, dds_header_t *hdr, dds_load_info_t *d, gint32 image, unsigned int level, char *prefix, unsigned int *l, guchar *pixels, unsigned char *buf); static int load_mipmaps (FILE *fp, dds_header_t *hdr, dds_load_info_t *d, gint32 image, char *prefix, unsigned int *l, guchar *pixels, unsigned char *buf); static int load_face (FILE *fp, dds_header_t *hdr, dds_load_info_t *d, gint32 image, char *prefix, unsigned int *l, guchar *pixels, unsigned char *buf); static unsigned char color_bits (unsigned int mask); static unsigned char color_shift (unsigned int mask); static int load_dialog (void); static gboolean runme = FALSE; GimpPDBStatusType read_dds (gchar *filename, gint32 *imageID, gboolean interactive_dds) { gint32 image = 0; unsigned char *buf; unsigned int l = 0; guchar *pixels; gchar *tmp; FILE *fp; dds_header_t hdr; dds_header_dx10_t dx10hdr; dds_load_info_t d; gint *layers, layer_count; GimpImageBaseType type; int i, j; if (interactive_dds) { if (!load_dialog ()) return GIMP_PDB_CANCEL; } fp = g_fopen (filename, "rb"); if (fp == 0) { g_message ("Error opening file.\n"); return GIMP_PDB_EXECUTION_ERROR; } if (strrchr (filename, '/')) tmp = g_strdup_printf ("Loading %s:", strrchr (filename, '/') + 1); else tmp = g_strdup_printf ("Loading %s:", filename); gimp_progress_init (tmp); g_free (tmp); /* read header */ read_header (&hdr, fp); memset (&dx10hdr, 0, sizeof (dds_header_dx10_t)); /* read DX10 header if necessary */ if (GETL32(hdr.pixelfmt.fourcc) == FOURCC ('D','X','1','0')) { read_header_dx10(&dx10hdr, fp); if (!setup_dxgi_format (&hdr, &dx10hdr)) { fclose (fp); return GIMP_PDB_EXECUTION_ERROR; } } if (!validate_header (&hdr)) { fclose (fp); g_message ("Invalid DDS header!\n"); return GIMP_PDB_EXECUTION_ERROR; } /* a lot of DDS images out there don't have this for some reason -_- */ if (hdr.pitch_or_linsize == 0) { if (hdr.pixelfmt.flags & DDPF_FOURCC) /* assume linear size */ { hdr.pitch_or_linsize = ((hdr.width + 3) >> 2) * ((hdr.height + 3) >> 2); switch (GETL32(hdr.pixelfmt.fourcc)) { case FOURCC ('D','X','T','1'): case FOURCC ('A','T','I','1'): case FOURCC ('B','C','4','U'): case FOURCC ('B','C','4','S'): hdr.pitch_or_linsize *= 8; break; default: hdr.pitch_or_linsize *= 16; break; } } else /* assume pitch */ { hdr.pitch_or_linsize = hdr.height * hdr.width * (hdr.pixelfmt.bpp >> 3); } } if (hdr.pixelfmt.flags & DDPF_FOURCC) { /* fourcc is dXt* or rXgb */ if (hdr.pixelfmt.fourcc[1] == 'X') hdr.pixelfmt.flags |= DDPF_ALPHAPIXELS; } if (hdr.pixelfmt.flags & DDPF_FOURCC) { switch (GETL32(hdr.pixelfmt.fourcc)) { case FOURCC ('A','T','I','1'): case FOURCC ('B','C','4','U'): case FOURCC ('B','C','4','S'): d.bpp = d.gimp_bpp = 1; type = GIMP_GRAY; break; case FOURCC ('A','T','I','2'): case FOURCC ('B','C','5','U'): case FOURCC ('B','C','5','S'): d.bpp = d.gimp_bpp = 3; type = GIMP_RGB; break; default: d.bpp = d.gimp_bpp = 4; type = GIMP_RGB; break; } } else { d.bpp = hdr.pixelfmt.bpp >> 3; if (d.bpp == 2) { if (hdr.pixelfmt.amask == 0xf000) // RGBA4 { d.gimp_bpp = 4; type = GIMP_RGB; } else if (hdr.pixelfmt.amask == 0xff00) //L8A8 { d.gimp_bpp = 2; type = GIMP_GRAY; } else if (hdr.pixelfmt.bmask == 0x1f) //R5G6B5 or RGB5A1 { if (hdr.pixelfmt.amask == 0x8000) // RGB5A1 d.gimp_bpp = 4; else d.gimp_bpp = 3; type = GIMP_RGB; } else //L16 { d.gimp_bpp = 1; type = GIMP_GRAY; } } else { if (hdr.pixelfmt.flags & DDPF_PALETTEINDEXED8) { type = GIMP_INDEXED; d.gimp_bpp = 1; } else if (hdr.pixelfmt.rmask == 0xe0) // R3G3B2 { type = GIMP_RGB; d.gimp_bpp = 3; } else { /* test alpha only image */ if (d.bpp == 1 && (hdr.pixelfmt.flags & DDPF_ALPHA)) { d.gimp_bpp = 2; type = GIMP_GRAY; } else { d.gimp_bpp = d.bpp; type = (d.bpp == 1) ? GIMP_GRAY : GIMP_RGB; } } } } image = gimp_image_new (hdr.width, hdr.height, type); if (image == -1) { g_message ("Can't allocate new image.\n"); fclose (fp); return GIMP_PDB_EXECUTION_ERROR; } gimp_image_set_filename (image, filename); if (hdr.pixelfmt.flags & DDPF_PALETTEINDEXED8) { d.palette = g_malloc (256 * 4); if (fread (d.palette, 1, 1024, fp) != 1024) { g_message ("Error reading palette.\n"); fclose (fp); gimp_image_delete (image); return GIMP_PDB_EXECUTION_ERROR; } for (i = j = 0; i < 768; i += 3, j += 4) { d.palette[i + 0] = d.palette[j + 0]; d.palette[i + 1] = d.palette[j + 1]; d.palette[i + 2] = d.palette[j + 2]; } gimp_image_set_colormap (image, d.palette, 256); } d.tile_height = gimp_tile_height (); pixels = g_new (guchar, d.tile_height * hdr.width * d.gimp_bpp); buf = g_malloc (hdr.pitch_or_linsize); d.rshift = color_shift (hdr.pixelfmt.rmask); d.gshift = color_shift (hdr.pixelfmt.gmask); d.bshift = color_shift (hdr.pixelfmt.bmask); d.ashift = color_shift (hdr.pixelfmt.amask); d.rbits = color_bits (hdr.pixelfmt.rmask); d.gbits = color_bits (hdr.pixelfmt.gmask); d.bbits = color_bits (hdr.pixelfmt.bmask); d.abits = color_bits (hdr.pixelfmt.amask); d.rmask = (hdr.pixelfmt.rmask >> d.rshift) << (8 - d.rbits); d.gmask = (hdr.pixelfmt.gmask >> d.gshift) << (8 - d.gbits); d.bmask = (hdr.pixelfmt.bmask >> d.bshift) << (8 - d.bbits); d.amask = (hdr.pixelfmt.amask >> d.ashift) << (8 - d.abits); if (!(hdr.caps.caps2 & DDSCAPS2_CUBEMAP) && !(hdr.caps.caps2 & DDSCAPS2_VOLUME) && dx10hdr.arraySize == 0) { if (!load_layer (fp, &hdr, &d, image, 0, "", &l, pixels, buf)) { fclose (fp); gimp_image_delete (image); return GIMP_PDB_EXECUTION_ERROR; } if (!load_mipmaps (fp, &hdr, &d, image, "", &l, pixels, buf)) { fclose (fp); gimp_image_delete (image); return GIMP_PDB_EXECUTION_ERROR; } } else if (hdr.caps.caps2 & DDSCAPS2_CUBEMAP) { if ((hdr.caps.caps2 & DDSCAPS2_CUBEMAP_POSITIVEX) && !load_face (fp, &hdr, &d, image, "(positive x)", &l, pixels, buf)) { fclose (fp); gimp_image_delete (image); return GIMP_PDB_EXECUTION_ERROR; } if ((hdr.caps.caps2 & DDSCAPS2_CUBEMAP_NEGATIVEX) && !load_face (fp, &hdr, &d, image, "(negative x)", &l, pixels, buf)) { fclose (fp); gimp_image_delete (image); return GIMP_PDB_EXECUTION_ERROR; } if ((hdr.caps.caps2 & DDSCAPS2_CUBEMAP_POSITIVEY) && !load_face (fp, &hdr, &d, image, "(positive y)", &l, pixels, buf)) { fclose (fp); gimp_image_delete (image); return GIMP_PDB_EXECUTION_ERROR; } if ((hdr.caps.caps2 & DDSCAPS2_CUBEMAP_NEGATIVEY) && !load_face (fp, &hdr, &d, image, "(negative y)", &l, pixels, buf)) { fclose (fp); gimp_image_delete (image); return GIMP_PDB_EXECUTION_ERROR; } if ((hdr.caps.caps2 & DDSCAPS2_CUBEMAP_POSITIVEZ) && !load_face (fp, &hdr, &d, image, "(positive z)", &l, pixels, buf)) { fclose (fp); gimp_image_delete (image); return GIMP_PDB_EXECUTION_ERROR; } if ((hdr.caps.caps2 & DDSCAPS2_CUBEMAP_NEGATIVEZ) && !load_face (fp, &hdr, &d, image, "(negative z)", &l, pixels, buf)) { fclose (fp); gimp_image_delete (image); return GIMP_PDB_EXECUTION_ERROR; } } else if ((hdr.caps.caps2 & DDSCAPS2_VOLUME) && (hdr.flags & DDSD_DEPTH)) { unsigned int i, level; char *plane; for (i = 0; i < hdr.depth; ++i) { plane = g_strdup_printf ("(z = %d)", i); if (!load_layer (fp, &hdr, &d, image, 0, plane, &l, pixels, buf)) { g_free (plane); fclose (fp); gimp_image_delete (image); return GIMP_PDB_EXECUTION_ERROR; } g_free (plane); } if ((hdr.flags & DDSD_MIPMAPCOUNT) && (hdr.caps.caps1 & DDSCAPS_MIPMAP) && (dds_read_vals.mipmaps != 0)) { for (level = 1; level < hdr.num_mipmaps; ++level) { int n = hdr.depth >> level; if (n < 1) n = 1; for (i = 0; i < n; ++i) { plane = g_strdup_printf ("(z = %d)", i); if (!load_layer (fp, &hdr, &d, image, level, plane, &l, pixels, buf)) { g_free (plane); fclose (fp); gimp_image_delete (image); return GIMP_PDB_EXECUTION_ERROR; } g_free (plane); } } } } else if (dx10hdr.arraySize > 0) { unsigned int i; char *elem; for (i = 0; i < dx10hdr.arraySize; ++i) { elem = g_strdup_printf ("(array element %d)", i); if (!load_layer (fp, &hdr, &d, image, 0, elem, &l, pixels, buf)) { fclose (fp); gimp_image_delete (image); return GIMP_PDB_EXECUTION_ERROR; } if (!load_mipmaps (fp, &hdr, &d, image, elem, &l, pixels, buf)) { fclose (fp); gimp_image_delete (image); return GIMP_PDB_EXECUTION_ERROR; } g_free (elem); } } gimp_progress_update (1.0); if (hdr.pixelfmt.flags & DDPF_PALETTEINDEXED8) g_free (d.palette); g_free (buf); g_free (pixels); fclose (fp); layers = gimp_image_get_layers (image, &layer_count); if (layers == NULL || layer_count == 0) { g_message ("Oops! NULL image read! Please report this!"); return GIMP_PDB_EXECUTION_ERROR; } gimp_image_set_active_layer (image, layers[0]); g_free (layers); *imageID = image; return GIMP_PDB_SUCCESS; } static int read_header (dds_header_t *hdr, FILE *fp) { unsigned char buf[DDS_HEADERSIZE]; memset (hdr, 0, sizeof (dds_header_t)); if (fread (buf, 1, DDS_HEADERSIZE, fp) != DDS_HEADERSIZE) return 0; hdr->magic = GETL32(buf); hdr->size = GETL32(buf + 4); hdr->flags = GETL32(buf + 8); hdr->height = GETL32(buf + 12); hdr->width = GETL32(buf + 16); hdr->pitch_or_linsize = GETL32(buf + 20); hdr->depth = GETL32(buf + 24); hdr->num_mipmaps = GETL32(buf + 28); hdr->pixelfmt.size = GETL32(buf + 76); hdr->pixelfmt.flags = GETL32(buf + 80); hdr->pixelfmt.fourcc[0] = buf[84]; hdr->pixelfmt.fourcc[1] = buf[85]; hdr->pixelfmt.fourcc[2] = buf[86]; hdr->pixelfmt.fourcc[3] = buf[87]; hdr->pixelfmt.bpp = GETL32(buf + 88); hdr->pixelfmt.rmask = GETL32(buf + 92); hdr->pixelfmt.gmask = GETL32(buf + 96); hdr->pixelfmt.bmask = GETL32(buf + 100); hdr->pixelfmt.amask = GETL32(buf + 104); hdr->caps.caps1 = GETL32(buf + 108); hdr->caps.caps2 = GETL32(buf + 112); /* GIMP-DDS special info */ if (GETL32(buf + 32) == FOURCC ('G','I','M','P') && GETL32(buf + 36) == FOURCC ('-','D','D','S')) { hdr->reserved.gimp_dds_special.magic1 = GETL32(buf + 32); hdr->reserved.gimp_dds_special.magic2 = GETL32(buf + 36); hdr->reserved.gimp_dds_special.version = GETL32(buf + 40); hdr->reserved.gimp_dds_special.extra_fourcc = GETL32(buf + 44); } return 1; } static int read_header_dx10 (dds_header_dx10_t *hdr, FILE *fp) { char buf[DDS_HEADERSIZE_DX10]; memset (hdr, 0, sizeof (dds_header_dx10_t)); if (fread (buf, 1, DDS_HEADERSIZE_DX10, fp) != DDS_HEADERSIZE_DX10) return 0; hdr->dxgiFormat = GETL32(buf); hdr->resourceDimension = GETL32(buf + 4); hdr->miscFlag = GETL32(buf + 8); hdr->arraySize = GETL32(buf + 12); hdr->reserved = GETL32(buf + 16); return 1; } static int validate_header (dds_header_t *hdr) { unsigned int fourcc; if (hdr->magic != FOURCC ('D','D','S',' ')) { g_message ("Invalid DDS file.\n"); return 0; } if ((hdr->flags & DDSD_PITCH) == (hdr->flags & DDSD_LINEARSIZE)) { //g_message ("Warning: DDSD_PITCH or DDSD_LINEARSIZE is not set.\n"); if (hdr->pixelfmt.flags & DDPF_FOURCC) hdr->flags |= DDSD_LINEARSIZE; else hdr->flags |= DDSD_PITCH; } /* if ((hdr->pixelfmt.flags & DDPF_FOURCC) == (hdr->pixelfmt.flags & DDPF_RGB)) { g_message ("Invalid pixel format.\n"); return 0; } */ fourcc = GETL32(hdr->pixelfmt.fourcc); if ((hdr->pixelfmt.flags & DDPF_FOURCC) && fourcc != FOURCC ('D','X','T','1') && fourcc != FOURCC ('D','X','T','2') && fourcc != FOURCC ('D','X','T','3') && fourcc != FOURCC ('D','X','T','4') && fourcc != FOURCC ('D','X','T','5') && fourcc != FOURCC ('R','X','G','B') && fourcc != FOURCC ('A','T','I','1') && fourcc != FOURCC ('B','C','4','U') && fourcc != FOURCC ('B','C','4','S') && fourcc != FOURCC ('A','T','I','2') && fourcc != FOURCC ('B','C','5','U') && fourcc != FOURCC ('B','C','5','S') && fourcc != FOURCC ('D','X','1','0')) { g_message ("Unsupported format (FOURCC: %c%c%c%c, hex: %08x).\n", hdr->pixelfmt.fourcc[0], hdr->pixelfmt.fourcc[1], hdr->pixelfmt.fourcc[2], hdr->pixelfmt.fourcc[3], GETL32(hdr->pixelfmt.fourcc)); return 0; } if (hdr->pixelfmt.flags & DDPF_RGB) { if ((hdr->pixelfmt.bpp != 8) && (hdr->pixelfmt.bpp != 16) && (hdr->pixelfmt.bpp != 24) && (hdr->pixelfmt.bpp != 32)) { g_message ("Invalid BPP.\n"); return 0; } } else if (hdr->pixelfmt.flags & DDPF_LUMINANCE) { if ((hdr->pixelfmt.bpp != 8) && (hdr->pixelfmt.bpp != 16)) { g_message ("Invalid BPP.\n"); return 0; } hdr->pixelfmt.flags |= DDPF_RGB; } else if (hdr->pixelfmt.flags & DDPF_PALETTEINDEXED8) { hdr->pixelfmt.flags |= DDPF_RGB; } if (!(hdr->pixelfmt.flags & DDPF_RGB) && !(hdr->pixelfmt.flags & DDPF_ALPHA) && !(hdr->pixelfmt.flags & DDPF_FOURCC) && !(hdr->pixelfmt.flags & DDPF_LUMINANCE)) { g_message ("Unknown pixel format! Taking a guess, expect trouble!"); switch (fourcc) { case FOURCC ('D','X','T','1'): case FOURCC ('D','X','T','2'): case FOURCC ('D','X','T','3'): case FOURCC ('D','X','T','4'): case FOURCC ('D','X','T','5'): case FOURCC ('R','X','G','B'): case FOURCC ('A','T','I','1'): case FOURCC ('B','C','4','U'): case FOURCC ('B','C','4','S'): case FOURCC ('A','T','I','2'): case FOURCC ('B','C','5','U'): case FOURCC ('B','C','5','S'): hdr->pixelfmt.flags |= DDPF_FOURCC; break; default: switch (hdr->pixelfmt.bpp) { case 8: if (hdr->pixelfmt.flags & DDPF_ALPHAPIXELS) hdr->pixelfmt.flags |= DDPF_ALPHA; else hdr->pixelfmt.flags |= DDPF_LUMINANCE; break; case 16: case 24: case 32: hdr->pixelfmt.flags |= DDPF_RGB; break; default: g_message ("Invalid pixel format."); return 0; } break; } } return 1; } /* * This function will set the necessary flags and attributes in the standard * dds header using the information found in the DX10 header. */ static int setup_dxgi_format (dds_header_t *hdr, dds_header_dx10_t *dx10hdr) { if ((dx10hdr->resourceDimension == D3D10_RESOURCE_DIMENSION_TEXTURE2D) && (dx10hdr->miscFlag & D3D10_RESOURCE_MISC_TEXTURECUBE)) { hdr->caps.caps2 |= DDSCAPS2_CUBEMAP | DDSCAPS2_CUBEMAP_ALL_FACES; } else if (dx10hdr->resourceDimension == D3D10_RESOURCE_DIMENSION_TEXTURE3D) { hdr->flags |= DDSD_DEPTH; hdr->caps.caps2 |= DDSCAPS2_VOLUME; } if ((dx10hdr->resourceDimension != D3D10_RESOURCE_DIMENSION_TEXTURE1D) && (dx10hdr->resourceDimension != D3D10_RESOURCE_DIMENSION_TEXTURE2D) && (dx10hdr->resourceDimension != D3D10_RESOURCE_DIMENSION_TEXTURE3D)) return 0; // check for a compressed DXGI format if ((dx10hdr->dxgiFormat >= DXGI_FORMAT_BC1_TYPELESS) && (dx10hdr->dxgiFormat <= DXGI_FORMAT_BC5_SNORM)) { // set flag and replace FOURCC hdr->pixelfmt.flags |= DDPF_FOURCC; switch (dx10hdr->dxgiFormat) { case DXGI_FORMAT_BC1_TYPELESS: case DXGI_FORMAT_BC1_UNORM: case DXGI_FORMAT_BC1_UNORM_SRGB: PUTL32(hdr->pixelfmt.fourcc, FOURCC ('D','X','T','1')); break; case DXGI_FORMAT_BC2_TYPELESS: case DXGI_FORMAT_BC2_UNORM: case DXGI_FORMAT_BC2_UNORM_SRGB: PUTL32(hdr->pixelfmt.fourcc, FOURCC ('D','X','T','3')); break; case DXGI_FORMAT_BC3_TYPELESS: case DXGI_FORMAT_BC3_UNORM: case DXGI_FORMAT_BC3_UNORM_SRGB: PUTL32(hdr->pixelfmt.fourcc, FOURCC ('D','X','T','5')); break; case DXGI_FORMAT_BC4_TYPELESS: case DXGI_FORMAT_BC4_UNORM: PUTL32(hdr->pixelfmt.fourcc, FOURCC ('A','T','I','1')); break; case DXGI_FORMAT_BC4_SNORM: PUTL32(hdr->pixelfmt.fourcc, FOURCC ('B','C','4','S')); break; case DXGI_FORMAT_BC5_TYPELESS: case DXGI_FORMAT_BC5_UNORM: PUTL32(hdr->pixelfmt.fourcc, FOURCC ('A','T','I','2')); break; case DXGI_FORMAT_BC5_SNORM: PUTL32(hdr->pixelfmt.fourcc, FOURCC ('B','C','5','S')); break; default: break; } } else { /* unset the FOURCC flag */ hdr->pixelfmt.flags &= ~DDPF_FOURCC; switch (dx10hdr->dxgiFormat) { case DXGI_FORMAT_B8G8R8A8_TYPELESS: case DXGI_FORMAT_B8G8R8A8_UNORM: case DXGI_FORMAT_B8G8R8A8_UNORM_SRGB: hdr->pixelfmt.bpp = 32; hdr->pixelfmt.flags |= DDPF_ALPHAPIXELS; hdr->pixelfmt.rmask = 0x00ff0000; hdr->pixelfmt.gmask = 0x0000ff00; hdr->pixelfmt.bmask = 0x000000ff; hdr->pixelfmt.amask = 0xff000000; break; case DXGI_FORMAT_B8G8R8X8_TYPELESS: case DXGI_FORMAT_B8G8R8X8_UNORM: case DXGI_FORMAT_B8G8R8X8_UNORM_SRGB: hdr->pixelfmt.bpp = 32; hdr->pixelfmt.flags |= DDPF_ALPHAPIXELS; hdr->pixelfmt.rmask = 0x00ff0000; hdr->pixelfmt.gmask = 0x0000ff00; hdr->pixelfmt.bmask = 0x000000ff; hdr->pixelfmt.amask = 0x00000000; break; case DXGI_FORMAT_R8G8B8A8_TYPELESS: case DXGI_FORMAT_R8G8B8A8_UNORM: case DXGI_FORMAT_R8G8B8A8_UNORM_SRGB: case DXGI_FORMAT_R8G8B8A8_UINT: case DXGI_FORMAT_R8G8B8A8_SNORM: case DXGI_FORMAT_R8G8B8A8_SINT: hdr->pixelfmt.bpp = 32; hdr->pixelfmt.flags |= DDPF_ALPHAPIXELS; hdr->pixelfmt.rmask = 0x000000ff; hdr->pixelfmt.gmask = 0x0000ff00; hdr->pixelfmt.bmask = 0x00ff0000; hdr->pixelfmt.amask = 0xff000000; break; case DXGI_FORMAT_B5G6R5_UNORM: hdr->pixelfmt.bpp = 16; hdr->pixelfmt.rmask = 0x0000f800; hdr->pixelfmt.gmask = 0x000007e0; hdr->pixelfmt.bmask = 0x0000001f; hdr->pixelfmt.amask = 0x00000000; break; case DXGI_FORMAT_B5G5R5A1_UNORM: hdr->pixelfmt.bpp = 16; hdr->pixelfmt.rmask = 0x00007c00; hdr->pixelfmt.gmask = 0x000003e0; hdr->pixelfmt.bmask = 0x0000001f; hdr->pixelfmt.amask = 0x00008000; break; case DXGI_FORMAT_R10G10B10A2_TYPELESS: case DXGI_FORMAT_R10G10B10A2_UNORM: case DXGI_FORMAT_R10G10B10A2_UINT: hdr->pixelfmt.bpp = 32; hdr->pixelfmt.flags |= DDPF_ALPHAPIXELS; hdr->pixelfmt.rmask = 0x000003ff; hdr->pixelfmt.gmask = 0x000ffc00; hdr->pixelfmt.bmask = 0x3ff00000; hdr->pixelfmt.amask = 0xc0000000; break; case DXGI_FORMAT_A8_UNORM: hdr->pixelfmt.bpp = 8; hdr->pixelfmt.flags |= DDPF_ALPHA | DDPF_ALPHAPIXELS; hdr->pixelfmt.rmask = hdr->pixelfmt.gmask = hdr->pixelfmt.bmask = 0; hdr->pixelfmt.amask = 0x000000ff; break; case DXGI_FORMAT_R8_TYPELESS: case DXGI_FORMAT_R8_UNORM: case DXGI_FORMAT_R8_UINT: case DXGI_FORMAT_R8_SNORM: case DXGI_FORMAT_R8_SINT: hdr->pixelfmt.bpp = 8; hdr->pixelfmt.rmask = 0x000000ff; hdr->pixelfmt.gmask = hdr->pixelfmt.bmask = hdr->pixelfmt.amask = 0; break; case DXGI_FORMAT_B4G4R4A4_UNORM: hdr->pixelfmt.bpp = 16; hdr->pixelfmt.flags |= DDPF_ALPHAPIXELS; hdr->pixelfmt.rmask = 0x00000f00; hdr->pixelfmt.gmask = 0x000000f0; hdr->pixelfmt.bmask = 0x0000000f; hdr->pixelfmt.amask = 0x0000f000; break; case DXGI_FORMAT_UNKNOWN: g_message ("Unknown DXGI format. Expect problems..."); break; default: /* unsupported DXGI format */ g_message ("Unsupported DXGI format (%d)", dx10hdr->dxgiFormat); return 0; } } return 1; } static const Babl* premultiplied_variant (const Babl* format) { if (format == babl_format ("R'G'B'A u8")) return babl_format ("R'aG'aB'aA u8"); else g_printerr ("Add format %s to premultiplied_variant () %s: %d\n", babl_get_name (format), __FILE__, __LINE__); return format; } static int load_layer (FILE *fp, dds_header_t *hdr, dds_load_info_t *d, gint32 image, unsigned int level, char *prefix, unsigned int *l, guchar *pixels, unsigned char *buf) { GeglBuffer *buffer; const Babl *bablfmt = NULL; GimpImageType type = GIMP_RGBA_IMAGE; gchar *layer_name; gint x, y, z, n; gint32 layer; unsigned int width = hdr->width >> level; unsigned int height = hdr->height >> level; unsigned int size = hdr->pitch_or_linsize >> (2 * level); unsigned int layerw; int format = DDS_COMPRESS_NONE; if (width < 1) width = 1; if (height < 1) height = 1; switch (d->bpp) { case 1: if (hdr->pixelfmt.flags & DDPF_PALETTEINDEXED8) { type = GIMP_INDEXED_IMAGE; bablfmt = babl_format ("R'G'B' u8"); } else if (hdr->pixelfmt.rmask == 0xe0) { type = GIMP_RGB_IMAGE; bablfmt = babl_format ("R'G'B' u8"); } else if (hdr->pixelfmt.flags & DDPF_ALPHA) { type = GIMP_GRAYA_IMAGE; bablfmt = babl_format ("Y'A u8"); } else { type = GIMP_GRAY_IMAGE; bablfmt = babl_format ("Y' u8"); } break; case 2: if (hdr->pixelfmt.amask == 0xf000) /* RGBA4 */ { type = GIMP_RGBA_IMAGE; bablfmt = babl_format ("R'G'B'A u8"); } else if (hdr->pixelfmt.amask == 0xff00) /* L8A8 */ { type = GIMP_GRAYA_IMAGE; bablfmt = babl_format ("Y'A u8"); } else if (hdr->pixelfmt.bmask == 0x1f) /* R5G6B5 or RGB5A1 */ { type = (hdr->pixelfmt.amask == 0x8000) ? GIMP_RGBA_IMAGE : GIMP_RGB_IMAGE; bablfmt = (hdr->pixelfmt.amask == 0x8000) ? babl_format ("R'G'B'A u8") : babl_format ("R'G'B' u8"); } else /* L16 */ { type = GIMP_GRAY_IMAGE; bablfmt = babl_format ("Y' u8"); } break; case 3: type = GIMP_RGB_IMAGE; bablfmt = babl_format ("R'G'B' u8"); break; case 4: type = GIMP_RGBA_IMAGE; bablfmt = babl_format ("R'G'B'A u8"); break; } layer_name = (level) ? g_strdup_printf ("mipmap %d %s", level, prefix) : g_strdup_printf ("main surface %s", prefix); layer = gimp_layer_new (image, layer_name, width, height, type, 100, GIMP_LAYER_MODE_NORMAL); g_free (layer_name); gimp_image_insert_layer (image, layer, 0, *l); if ((*l)++) gimp_item_set_visible (layer, FALSE); buffer = gimp_drawable_get_buffer (layer); layerw = gegl_buffer_get_width (buffer); if (hdr->pixelfmt.flags & DDPF_FOURCC) { unsigned int w = (width + 3) >> 2; unsigned int h = (height + 3) >> 2; switch (GETL32(hdr->pixelfmt.fourcc)) { case FOURCC ('D','X','T','1'): format = DDS_COMPRESS_BC1; break; case FOURCC ('D','X','T','2'): bablfmt = premultiplied_variant (bablfmt); case FOURCC ('D','X','T','3'): format = DDS_COMPRESS_BC2; break; case FOURCC ('D','X','T','4'): bablfmt = premultiplied_variant (bablfmt); case FOURCC ('D','X','T','5'): format = DDS_COMPRESS_BC3; break; case FOURCC ('R','X','G','B'): format = DDS_COMPRESS_BC3; break; case FOURCC ('A','T','I','1'): case FOURCC ('B','C','4','U'): case FOURCC ('B','C','4','S'): format = DDS_COMPRESS_BC4; break; case FOURCC ('A','T','I','2'): case FOURCC ('B','C','5','U'): case FOURCC ('B','C','5','S'): format = DDS_COMPRESS_BC5; break; } size = w * h; if ((format == DDS_COMPRESS_BC1) || (format == DDS_COMPRESS_BC4)) size *= 8; else size *= 16; } if ((hdr->flags & DDSD_LINEARSIZE) && !fread (buf, size, 1, fp)) { g_message ("Unexpected EOF.\n"); return 0; } if ((hdr->pixelfmt.flags & DDPF_RGB) || (hdr->pixelfmt.flags & DDPF_ALPHA)) { z = 0; for (y = 0, n = 0; y < height; ++y, ++n) { if (n >= d->tile_height) { gegl_buffer_set (buffer, GEGL_RECTANGLE (0, y - n, layerw, n), 0, bablfmt, pixels, GEGL_AUTO_ROWSTRIDE); n = 0; gimp_progress_update ((double)y / (double)hdr->height); } if ((hdr->flags & DDSD_PITCH) && !fread (buf, width * d->bpp, 1, fp)) { g_message ("Unexpected EOF.\n"); return 0; } if (!(hdr->flags & DDSD_LINEARSIZE)) z = 0; for (x = 0; x < layerw; ++x) { unsigned int pixel = buf[z]; unsigned int pos = (n * layerw + x) * d->gimp_bpp; if (d->bpp > 1) pixel += ((unsigned int)buf[z + 1] << 8); if (d->bpp > 2) pixel += ((unsigned int)buf[z + 2] << 16); if (d->bpp > 3) pixel += ((unsigned int)buf[z + 3] << 24); if (d->bpp >= 3) { if (hdr->pixelfmt.amask == 0xc0000000) // handle RGB10A2 { pixels[pos + 0] = (pixel >> d->bshift) >> 2; pixels[pos + 1] = (pixel >> d->gshift) >> 2; pixels[pos + 2] = (pixel >> d->rshift) >> 2; if (hdr->pixelfmt.flags & DDPF_ALPHAPIXELS) pixels[pos + 3] = (pixel >> d->ashift << (8 - d->abits) & d->amask) * 255 / d->amask; } else { pixels[pos] = (pixel >> d->rshift << (8 - d->rbits) & d->rmask) * 255 / d->rmask; pixels[pos + 1] = (pixel >> d->gshift << (8 - d->gbits) & d->gmask) * 255 / d->gmask; pixels[pos + 2] = (pixel >> d->bshift << (8 - d->bbits) & d->bmask) * 255 / d->bmask; if (hdr->pixelfmt.flags & DDPF_ALPHAPIXELS) { pixels[pos + 3] = (pixel >> d->ashift << (8 - d->abits) & d->amask) * 255 / d->amask; } } } else if (d->bpp == 2) { if (hdr->pixelfmt.amask == 0xf000) //RGBA4 { pixels[pos] = (pixel >> d->rshift << (8 - d->rbits) & d->rmask) * 255 / d->rmask; pixels[pos + 1] = (pixel >> d->gshift << (8 - d->gbits) & d->gmask) * 255 / d->gmask; pixels[pos + 2] = (pixel >> d->bshift << (8 - d->bbits) & d->bmask) * 255 / d->bmask; pixels[pos + 3] = (pixel >> d->ashift << (8 - d->abits) & d->amask) * 255 / d->amask; } else if (hdr->pixelfmt.amask == 0xff00) //L8A8 { pixels[pos] = (pixel >> d->rshift << (8 - d->rbits) & d->rmask) * 255 / d->rmask; pixels[pos + 1] = (pixel >> d->ashift << (8 - d->abits) & d->amask) * 255 / d->amask; } else if (hdr->pixelfmt.bmask == 0x1f) //R5G6B5 or RGB5A1 { pixels[pos] = (pixel >> d->rshift << (8 - d->rbits) & d->rmask) * 255 / d->rmask; pixels[pos + 1] = (pixel >> d->gshift << (8 - d->gbits) & d->gmask) * 255 / d->gmask; pixels[pos + 2] = (pixel >> d->bshift << (8 - d->bbits) & d->bmask) * 255 / d->bmask; if (hdr->pixelfmt.amask == 0x8000) { pixels[pos + 3] = (pixel >> d->ashift << (8 - d->abits) & d->amask) * 255 / d->amask; } } else //L16 pixels[pos] = (unsigned char)(255 * ((float)(pixel & 0xffff) / 65535.0f)); } else { if (hdr->pixelfmt.flags & DDPF_PALETTEINDEXED8) { pixels[pos] = pixel & 0xff; } else if (hdr->pixelfmt.rmask == 0xe0) // R3G3B2 { pixels[pos] = (pixel >> d->rshift << (8 - d->rbits) & d->rmask) * 255 / d->rmask; pixels[pos + 1] = (pixel >> d->gshift << (8 - d->gbits) & d->gmask) * 255 / d->gmask; pixels[pos + 2] = (pixel >> d->bshift << (8 - d->bbits) & d->bmask) * 255 / d->bmask; } else if (hdr->pixelfmt.flags & DDPF_ALPHA) { pixels[pos + 0] = 255; pixels[pos + 1] = pixel & 0xff; } else // LUMINANCE { pixels[pos] = pixel & 0xff; } } z += d->bpp; } } gegl_buffer_set (buffer, GEGL_RECTANGLE (0, y - n, layerw, n), 0, bablfmt, pixels, GEGL_AUTO_ROWSTRIDE); } else if (hdr->pixelfmt.flags & DDPF_FOURCC) { unsigned char *dst; if (!(hdr->flags & DDSD_LINEARSIZE)) { g_message ("Image marked as compressed, but DDSD_LINEARSIZE is not set.\n"); return 0; } dst = g_malloc (width * height * d->gimp_bpp); memset (dst, 0, width * height * d->gimp_bpp); if (d->gimp_bpp == 4) { for (y = 0; y < height; ++y) for (x = 0; x < width; ++x) dst[y * (width * 4) + (x * 4) + 3] = 255; } dxt_decompress (dst, buf, format, size, width, height, d->gimp_bpp, hdr->pixelfmt.flags & DDPF_NORMAL); z = 0; for (y = 0, n = 0; y < height; ++y, ++n) { if (n >= d->tile_height) { gegl_buffer_set (buffer, GEGL_RECTANGLE (0, y - n, layerw, n), 0, bablfmt, pixels, GEGL_AUTO_ROWSTRIDE); n = 0; gimp_progress_update ((double)y / (double)hdr->height); } memcpy (pixels + n * layerw * d->gimp_bpp, dst + y * layerw * d->gimp_bpp, width * d->gimp_bpp); } gegl_buffer_set (buffer, GEGL_RECTANGLE (0, y - n, layerw, n), 0, bablfmt, pixels, GEGL_AUTO_ROWSTRIDE); g_free (dst); } gegl_buffer_flush (buffer); g_object_unref (buffer); /* gimp dds specific. decode encoded images */ if (dds_read_vals.decode_images && hdr->reserved.gimp_dds_special.magic1 == FOURCC ('G','I','M','P') && hdr->reserved.gimp_dds_special.magic2 == FOURCC ('-','D','D','S')) { switch (hdr->reserved.gimp_dds_special.extra_fourcc) { case FOURCC ('A','E','X','P'): decode_alpha_exp_image (layer, FALSE); break; case FOURCC ('Y','C','G','1'): decode_ycocg_image (layer, FALSE); break; case FOURCC ('Y','C','G','2'): decode_ycocg_scaled_image (layer, FALSE); break; default: break; } } return 1; } static int load_mipmaps (FILE *fp, dds_header_t *hdr, dds_load_info_t *d, gint32 image, char *prefix, unsigned int *l, guchar *pixels, unsigned char *buf) { unsigned int level; if ((hdr->flags & DDSD_MIPMAPCOUNT) && (hdr->caps.caps1 & DDSCAPS_MIPMAP) && (dds_read_vals.mipmaps != 0)) { for (level = 1; level < hdr->num_mipmaps; ++level) { if (!load_layer (fp, hdr, d, image, level, prefix, l, pixels, buf)) return 0; } } return 1; } static int load_face (FILE *fp, dds_header_t *hdr, dds_load_info_t *d, gint32 image, char *prefix, unsigned int *l, guchar *pixels, unsigned char *buf) { if (!load_layer (fp, hdr, d, image, 0, prefix, l, pixels, buf)) return 0; return load_mipmaps (fp, hdr, d, image, prefix, l, pixels, buf); } static unsigned char color_bits (unsigned int mask) { unsigned char i = 0; while (mask) { if (mask & 1) ++i; mask >>= 1; } return i; } static unsigned char color_shift (unsigned int mask) { guchar i = 0; if (! mask) return 0; while (!((mask >> i) & 1)) ++i; return i; } static void load_dialog_response (GtkWidget *widget, gint response_id, gpointer data) { switch (response_id) { case GTK_RESPONSE_OK: runme = TRUE; default: gtk_widget_destroy (widget); break; } } static void toggle_clicked (GtkWidget *widget, gpointer data) { int *flag = (int*) data; (*flag) = !(*flag); } static int load_dialog (void) { GtkWidget *dlg; GtkWidget *vbox; GtkWidget *check; dlg = gimp_dialog_new (_("Load DDS"), "dds", NULL, GTK_WIN_POS_MOUSE, gimp_standard_help_func, LOAD_PROC, _("Cancel"), GTK_RESPONSE_CANCEL, _("OK"), GTK_RESPONSE_OK, NULL); g_signal_connect (dlg, "response", G_CALLBACK (load_dialog_response), 0); g_signal_connect (dlg, "destroy", G_CALLBACK (gtk_main_quit), 0); vbox = gtk_box_new (GTK_ORIENTATION_VERTICAL, 8); gtk_container_set_border_width (GTK_CONTAINER (vbox), 8); gtk_box_pack_start (GTK_BOX (gtk_dialog_get_content_area (GTK_DIALOG (dlg))), vbox, 1, 1, 0); gtk_widget_show (vbox); check = gtk_check_button_new_with_label (_("Load mipmaps")); gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (check), dds_read_vals.mipmaps); g_signal_connect (check, "clicked", G_CALLBACK (toggle_clicked), &dds_read_vals.mipmaps); gtk_box_pack_start (GTK_BOX (vbox), check, 1, 1, 0); gtk_widget_show (check); check = gtk_check_button_new_with_label (_("Automatically decode YCoCg/AExp images when detected")); gtk_toggle_button_set_active (GTK_TOGGLE_BUTTON (check), dds_read_vals.decode_images); g_signal_connect (check, "clicked", G_CALLBACK (toggle_clicked), &dds_read_vals.decode_images); gtk_box_pack_start (GTK_BOX (vbox), check, 1, 1, 0); gtk_widget_show (check); gtk_widget_show (dlg); runme = FALSE; gtk_main (); return runme; }