1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000 1.2 +++ b/image/test/unit/test_encoder_png.js Wed Dec 31 06:09:35 2014 +0100 1.3 @@ -0,0 +1,256 @@ 1.4 +/* 1.5 + * Test for PNG encoding in libpr0n 1.6 + * 1.7 + */ 1.8 + 1.9 +const Ci = Components.interfaces; 1.10 +const Cc = Components.classes; 1.11 + 1.12 +var png1A = { 1.13 + // A 3x3 image, rows are red, green, blue. 1.14 + // RGB format, transparency defaults. 1.15 + 1.16 + transparency : null, 1.17 + 1.18 + frames : [ 1.19 + { 1.20 + width : 3, height : 3, 1.21 + 1.22 + format : Ci.imgIEncoder.INPUT_FORMAT_RGB, stride : 9, 1.23 + 1.24 + pixels : [ 1.25 + 255,0,0, 255,0,0, 255,0,0, 1.26 + 0,255,0, 0,255,0, 0,255,0, 1.27 + 0,0,255, 0,0,255, 0,0,255, 1.28 + ] 1.29 + } 1.30 + 1.31 + ], 1.32 + expected : "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAMAAAADCAIAAADZSiLoAAAAEUlEQVQImWP4z8AAQTAamQkAhpcI+DeMzFcAAAAASUVORK5CYII=" 1.33 +}; 1.34 + 1.35 +var png1B = { 1.36 + // A 3x3 image, rows are red, green, blue. 1.37 + // RGB format, transparency=none. 1.38 + 1.39 + transparency : "none", 1.40 + 1.41 + frames : [ 1.42 + { 1.43 + width : 3, height : 3, 1.44 + 1.45 + format : Ci.imgIEncoder.INPUT_FORMAT_RGB, stride : 9, 1.46 + 1.47 + pixels : [ 1.48 + 255,0,0, 255,0,0, 255,0,0, 1.49 + 0,255,0, 0,255,0, 0,255,0, 1.50 + 0,0,255, 0,0,255, 0,0,255, 1.51 + ] 1.52 + } 1.53 + 1.54 + ], 1.55 + expected : "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAMAAAADCAIAAADZSiLoAAAAEUlEQVQImWP4z8AAQTAamQkAhpcI+DeMzFcAAAAASUVORK5CYII=" 1.56 +}; 1.57 + 1.58 +var png2A = { 1.59 + // A 3x3 image, rows are: red, green, blue. Columns are: 0%, 33%, 66% transparent. 1.60 + 1.61 + transparency : null, 1.62 + 1.63 + frames : [ 1.64 + { 1.65 + width : 3, height : 3, 1.66 + 1.67 + format : Ci.imgIEncoder.INPUT_FORMAT_RGBA, stride : 12, 1.68 + 1.69 + pixels : [ 1.70 + 255,0,0,255, 255,0,0,170, 255,0,0,85, 1.71 + 0,255,0,255, 0,255,0,170, 0,255,0,85, 1.72 + 0,0,255,255, 0,0,255,170, 0,0,255,85 1.73 + ] 1.74 + } 1.75 + 1.76 + ], 1.77 + expected : "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAMAAAADCAYAAABWKLW/AAAAIElEQVQImSXJMQEAMAwEIUy+yZi8DmVFFBcjycn86GgPJw4O8v9DkkEAAAAASUVORK5CYII=" 1.78 +}; 1.79 + 1.80 +var png2B = { 1.81 + // A 3x3 image, rows are: red, green, blue. Columns are: 0%, 33%, 66% transparent, 1.82 + // but transparency will be ignored. 1.83 + 1.84 + transparency : "none", 1.85 + 1.86 + frames : [ 1.87 + { 1.88 + width : 3, height : 3, 1.89 + 1.90 + format : Ci.imgIEncoder.INPUT_FORMAT_RGBA, stride : 12, 1.91 + 1.92 + pixels : [ 1.93 + 255,0,0,255, 255,0,0,170, 255,0,0,85, 1.94 + 0,255,0,255, 0,255,0,170, 0,255,0,85, 1.95 + 0,0,255,255, 0,0,255,170, 0,0,255,85 1.96 + ] 1.97 + } 1.98 + 1.99 + ], 1.100 + expected : "data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAAAMAAAADCAIAAADZSiLoAAAAEUlEQVQImWP4z8AAQTAamQkAhpcI+DeMzFcAAAAASUVORK5CYII=" 1.101 +}; 1.102 + 1.103 +// Main test entry point. 1.104 +function run_test() { 1.105 + dump("Checking png1A...\n") 1.106 + run_test_for(png1A); 1.107 + dump("Checking png1B...\n") 1.108 + run_test_for(png1B); 1.109 + dump("Checking png2A...\n") 1.110 + run_test_for(png2A); 1.111 + dump("Checking png2B...\n") 1.112 + run_test_for(png2B); 1.113 +}; 1.114 + 1.115 + 1.116 +function run_test_for(input) { 1.117 + var encoder, dataURL; 1.118 + 1.119 + encoder = encodeImage(input); 1.120 + dataURL = makeDataURL(encoder, "image/png"); 1.121 + do_check_eq(dataURL, input.expected); 1.122 + 1.123 + encoder = encodeImageAsync(input); 1.124 + dataURL = makeDataURLFromAsync(encoder, "image/png", input.expected); 1.125 +}; 1.126 + 1.127 + 1.128 +function encodeImage(input) { 1.129 + var encoder = Cc["@mozilla.org/image/encoder;2?type=image/png"].createInstance(); 1.130 + encoder.QueryInterface(Ci.imgIEncoder); 1.131 + 1.132 + var options = ""; 1.133 + if (input.transparency) { 1.134 + options += "transparency=" + input.transparency; 1.135 + } 1.136 + 1.137 + var frame = input.frames[0]; 1.138 + encoder.initFromData(frame.pixels, frame.pixels.length, 1.139 + frame.width, frame.height, frame.stride, 1.140 + frame.format, options); 1.141 + return encoder; 1.142 +} 1.143 + 1.144 +function _encodeImageAsyncFactory(frame, options, encoder) 1.145 +{ 1.146 + function finishEncode() { 1.147 + encoder.addImageFrame(frame.pixels, frame.pixels.length, 1.148 + frame.width, frame.height, frame.stride, 1.149 + frame.format, options); 1.150 + encoder.endImageEncode(); 1.151 + } 1.152 + return finishEncode; 1.153 +} 1.154 + 1.155 +function encodeImageAsync(input) 1.156 +{ 1.157 + var encoder = Cc["@mozilla.org/image/encoder;2?type=image/png"].createInstance(); 1.158 + encoder.QueryInterface(Ci.imgIEncoder); 1.159 + 1.160 + var options = ""; 1.161 + if (input.transparency) { 1.162 + options += "transparency=" + input.transparency; 1.163 + } 1.164 + 1.165 + var frame = input.frames[0]; 1.166 + encoder.startImageEncode(frame.width, frame.height, 1.167 + frame.format, options); 1.168 + 1.169 + do_timeout(50, _encodeImageAsyncFactory(frame, options, encoder)); 1.170 + return encoder; 1.171 +} 1.172 + 1.173 + 1.174 +function makeDataURL(encoder, mimetype) { 1.175 + var rawStream = encoder.QueryInterface(Ci.nsIInputStream); 1.176 + 1.177 + var stream = Cc["@mozilla.org/binaryinputstream;1"].createInstance(); 1.178 + stream.QueryInterface(Ci.nsIBinaryInputStream); 1.179 + 1.180 + stream.setInputStream(rawStream); 1.181 + 1.182 + var bytes = stream.readByteArray(stream.available()); // returns int[] 1.183 + 1.184 + var base64String = toBase64(bytes); 1.185 + 1.186 + return "data:" + mimetype + ";base64," + base64String; 1.187 +} 1.188 + 1.189 +function makeDataURLFromAsync(encoder, mimetype, expected) { 1.190 + do_test_pending(); 1.191 + var rawStream = encoder.QueryInterface(Ci.nsIAsyncInputStream); 1.192 + 1.193 + var currentThread = Cc["@mozilla.org/thread-manager;1"].getService().currentThread; 1.194 + 1.195 + var bytes = []; 1.196 + 1.197 + var binarystream = Cc["@mozilla.org/binaryinputstream;1"].createInstance(); 1.198 + binarystream.QueryInterface(Ci.nsIBinaryInputStream); 1.199 + 1.200 + var asyncReader = 1.201 + { 1.202 + onInputStreamReady: function(stream) 1.203 + { 1.204 + binarystream.setInputStream(stream); 1.205 + var available = 0; 1.206 + try { 1.207 + available = stream.available(); 1.208 + } catch(e) { } 1.209 + 1.210 + if (available > 0) 1.211 + { 1.212 + bytes = bytes.concat(binarystream.readByteArray(available)); 1.213 + stream.asyncWait(this, 0, 0, currentThread); 1.214 + } else { 1.215 + var base64String = toBase64(bytes); 1.216 + var dataURL = "data:" + mimetype + ";base64," + base64String; 1.217 + do_check_eq(dataURL, expected); 1.218 + do_test_finished(); 1.219 + } 1.220 + 1.221 + } 1.222 + }; 1.223 + rawStream.asyncWait(asyncReader, 0, 0, currentThread); 1.224 +} 1.225 + 1.226 +/* toBase64 copied from extensions/xml-rpc/src/nsXmlRpcClient.js */ 1.227 + 1.228 +/* Convert data (an array of integers) to a Base64 string. */ 1.229 +const toBase64Table = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz' + 1.230 + '0123456789+/'; 1.231 +const base64Pad = '='; 1.232 +function toBase64(data) { 1.233 + var result = ''; 1.234 + var length = data.length; 1.235 + var i; 1.236 + // Convert every three bytes to 4 ascii characters. 1.237 + for (i = 0; i < (length - 2); i += 3) { 1.238 + result += toBase64Table[data[i] >> 2]; 1.239 + result += toBase64Table[((data[i] & 0x03) << 4) + (data[i+1] >> 4)]; 1.240 + result += toBase64Table[((data[i+1] & 0x0f) << 2) + (data[i+2] >> 6)]; 1.241 + result += toBase64Table[data[i+2] & 0x3f]; 1.242 + } 1.243 + 1.244 + // Convert the remaining 1 or 2 bytes, pad out to 4 characters. 1.245 + if (length%3) { 1.246 + i = length - (length%3); 1.247 + result += toBase64Table[data[i] >> 2]; 1.248 + if ((length%3) == 2) { 1.249 + result += toBase64Table[((data[i] & 0x03) << 4) + (data[i+1] >> 4)]; 1.250 + result += toBase64Table[(data[i+1] & 0x0f) << 2]; 1.251 + result += base64Pad; 1.252 + } else { 1.253 + result += toBase64Table[(data[i] & 0x03) << 4]; 1.254 + result += base64Pad + base64Pad; 1.255 + } 1.256 + } 1.257 + 1.258 + return result; 1.259 +}