From 8c79e05df0e8f704dca101b78d22cddd6eed82fd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Branimir=20Karad=C5=BEi=C4=87?= <branimirkaradzic@gmail.com>
Date: Fri, 21 Apr 2017 18:20:28 -0700
Subject: [PATCH] Added support for palettized and 1-bit .png files.

---
 src/image_decode.cpp        | 66 +++++++++++++++++++++++++++++++------
 tools/texturec/texturec.cpp |  7 ++--
 2 files changed, 60 insertions(+), 13 deletions(-)

diff --git a/src/image_decode.cpp b/src/image_decode.cpp
index a231567..5103580 100644
--- a/src/image_decode.cpp
+++ b/src/image_decode.cpp
@@ -91,28 +91,44 @@ namespace bimg
 
 		if (0 == error)
 		{
+			bool palette   = false;
+			bool supported = false;
+
 			switch (state.info_raw.bitdepth)
 			{
+				case 1:
+					format    = bimg::TextureFormat::R1;
+					palette   = true;
+					supported = true;
+					break;
+
 				case 8:
 					switch (state.info_raw.colortype)
 					{
 						case LCT_GREY:
 							format = bimg::TextureFormat::R8;
+							supported = true;
 							break;
 
 						case LCT_GREY_ALPHA:
 							format = bimg::TextureFormat::RG8;
+							supported = true;
 							break;
 
 						case LCT_RGB:
 							format = bimg::TextureFormat::RGB8;
+							supported = true;
 							break;
 
 						case LCT_RGBA:
 							format = bimg::TextureFormat::RGBA8;
+							supported = true;
 							break;
 
 						case LCT_PALETTE:
+							format  = bimg::TextureFormat::RGBA8;
+							palette = true;
+							supported = true;
 							break;
 					}
 					break;
@@ -127,6 +143,7 @@ namespace bimg
 								rgba[0] = bx::toHostEndian(rgba[0], false);
 							}
 							format = bimg::TextureFormat::R16;
+							supported = true;
 							break;
 
 						case LCT_GREY_ALPHA:
@@ -137,6 +154,7 @@ namespace bimg
 								rgba[1] = bx::toHostEndian(rgba[1], false);
 							}
 							format = bimg::TextureFormat::RG16;
+							supported = true;
 							break;
 
 						case LCT_RGBA:
@@ -149,6 +167,7 @@ namespace bimg
 								rgba[3] = bx::toHostEndian(rgba[3], false);
 							}
 							format = bimg::TextureFormat::RGBA16;
+							supported = true;
 							break;
 
 						case LCT_RGB:
@@ -161,16 +180,43 @@ namespace bimg
 					break;
 			}
 
-			output = imageAlloc(_allocator
-				, format
-				, uint16_t(width)
-				, uint16_t(height)
-				, 0
-				, 1
-				, false
-				, false
-				, data
-				);
+			if (supported)
+			{
+				output = imageAlloc(_allocator
+					, bimg::TextureFormat::R1 == format ? bimg::TextureFormat::R8 : format
+					, uint16_t(width)
+					, uint16_t(height)
+					, 0
+					, 1
+					, false
+					, false
+					, palette ? NULL : data
+					);
+
+				if (bimg::TextureFormat::R1 == format)
+				{
+					for (uint32_t ii = 0, num = width*height/8; ii < num; ++ii)
+					{
+						uint8_t value = data[ii];
+						uint8_t* dst = (uint8_t*)output->m_data + ii * 8;
+						dst[0] = value & 0x01 ? 255 : 0;
+						dst[1] = value & 0x02 ? 255 : 0;
+						dst[2] = value & 0x04 ? 255 : 0;
+						dst[3] = value & 0x08 ? 255 : 0;
+						dst[4] = value & 0x10 ? 255 : 0;
+						dst[5] = value & 0x20 ? 255 : 0;
+						dst[6] = value & 0x40 ? 255 : 0;
+						dst[7] = value & 0x80 ? 255 : 0;
+					}
+				}
+				else if (palette)
+				{
+					for (uint32_t ii = 0, num = width*height; ii < num; ++ii)
+					{
+						bx::memCopy( (uint8_t*)output->m_data + ii*4, state.info_raw.palette + data[ii]*4, 4);
+					}
+				}
+			}
 		}
 
 		lodepng_state_cleanup(&state);
diff --git a/tools/texturec/texturec.cpp b/tools/texturec/texturec.cpp
index 1898c9a..70bcfb6 100644
--- a/tools/texturec/texturec.cpp
+++ b/tools/texturec/texturec.cpp
@@ -176,7 +176,8 @@ bimg::ImageContainer* convert(bx::AllocatorI* _allocator, const void* _inputData
 
 				BX_FREE(_allocator, rgbaDst);
 			}
-			else if (8 != bimg::getBlockInfo(input->m_format).rBits)
+			else if (!bimg::isCompressed(input->m_format)
+				 &&  8 != bimg::getBlockInfo(input->m_format).rBits)
 			{
 				output = bimg::imageAlloc(_allocator, format, uint16_t(mip.m_width), uint16_t(mip.m_height), 0, 1, false, _options.mips);
 
@@ -215,7 +216,7 @@ bimg::ImageContainer* convert(bx::AllocatorI* _allocator, const void* _inputData
 					, mip.m_data
 					, mip.m_width
 					, mip.m_height
-					, mip.m_width*mip.m_bpp/8
+					, mip.m_width*16
 					, mip.m_format
 					);
 				bimg::imageEncodeFromRgba32f(_allocator, output->m_data, rgba, dstMip.m_width, dstMip.m_height, format);
@@ -283,7 +284,7 @@ bimg::ImageContainer* convert(bx::AllocatorI* _allocator, const void* _inputData
 					, mip.m_data
 					, mip.m_width
 					, mip.m_height
-					, mip.m_width*mip.m_bpp/8
+					, mip.m_width*4
 					, mip.m_format
 					);