encoder.js 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768
  1. /*
  2. Copyright (c) 2008, Adobe Systems Incorporated
  3. All rights reserved.
  4. Redistribution and use in source and binary forms, with or without
  5. modification, are permitted provided that the following conditions are
  6. met:
  7. * Redistributions of source code must retain the above copyright notice,
  8. this list of conditions and the following disclaimer.
  9. * Redistributions in binary form must reproduce the above copyright
  10. notice, this list of conditions and the following disclaimer in the
  11. documentation and/or other materials provided with the distribution.
  12. * Neither the name of Adobe Systems Incorporated nor the names of its
  13. contributors may be used to endorse or promote products derived from
  14. this software without specific prior written permission.
  15. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
  16. IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  17. THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  18. PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
  19. CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  20. EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  21. PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  22. PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  23. LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  24. NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  25. SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26. */
  27. /*
  28. JPEG encoder ported to JavaScript and optimized by Andreas Ritter, www.bytestrom.eu, 11/2009
  29. Basic GUI blocking jpeg encoder
  30. */
  31. var btoa = btoa || function(buf) {
  32. return new Buffer(buf).toString('base64');
  33. };
  34. function JPEGEncoder(quality) {
  35. var self = this;
  36. var fround = Math.round;
  37. var ffloor = Math.floor;
  38. var YTable = new Array(64);
  39. var UVTable = new Array(64);
  40. var fdtbl_Y = new Array(64);
  41. var fdtbl_UV = new Array(64);
  42. var YDC_HT;
  43. var UVDC_HT;
  44. var YAC_HT;
  45. var UVAC_HT;
  46. var bitcode = new Array(65535);
  47. var category = new Array(65535);
  48. var outputfDCTQuant = new Array(64);
  49. var DU = new Array(64);
  50. var byteout = [];
  51. var bytenew = 0;
  52. var bytepos = 7;
  53. var YDU = new Array(64);
  54. var UDU = new Array(64);
  55. var VDU = new Array(64);
  56. var clt = new Array(256);
  57. var RGB_YUV_TABLE = new Array(2048);
  58. var currentQuality;
  59. var ZigZag = [
  60. 0, 1, 5, 6,14,15,27,28,
  61. 2, 4, 7,13,16,26,29,42,
  62. 3, 8,12,17,25,30,41,43,
  63. 9,11,18,24,31,40,44,53,
  64. 10,19,23,32,39,45,52,54,
  65. 20,22,33,38,46,51,55,60,
  66. 21,34,37,47,50,56,59,61,
  67. 35,36,48,49,57,58,62,63
  68. ];
  69. var std_dc_luminance_nrcodes = [0,0,1,5,1,1,1,1,1,1,0,0,0,0,0,0,0];
  70. var std_dc_luminance_values = [0,1,2,3,4,5,6,7,8,9,10,11];
  71. var std_ac_luminance_nrcodes = [0,0,2,1,3,3,2,4,3,5,5,4,4,0,0,1,0x7d];
  72. var std_ac_luminance_values = [
  73. 0x01,0x02,0x03,0x00,0x04,0x11,0x05,0x12,
  74. 0x21,0x31,0x41,0x06,0x13,0x51,0x61,0x07,
  75. 0x22,0x71,0x14,0x32,0x81,0x91,0xa1,0x08,
  76. 0x23,0x42,0xb1,0xc1,0x15,0x52,0xd1,0xf0,
  77. 0x24,0x33,0x62,0x72,0x82,0x09,0x0a,0x16,
  78. 0x17,0x18,0x19,0x1a,0x25,0x26,0x27,0x28,
  79. 0x29,0x2a,0x34,0x35,0x36,0x37,0x38,0x39,
  80. 0x3a,0x43,0x44,0x45,0x46,0x47,0x48,0x49,
  81. 0x4a,0x53,0x54,0x55,0x56,0x57,0x58,0x59,
  82. 0x5a,0x63,0x64,0x65,0x66,0x67,0x68,0x69,
  83. 0x6a,0x73,0x74,0x75,0x76,0x77,0x78,0x79,
  84. 0x7a,0x83,0x84,0x85,0x86,0x87,0x88,0x89,
  85. 0x8a,0x92,0x93,0x94,0x95,0x96,0x97,0x98,
  86. 0x99,0x9a,0xa2,0xa3,0xa4,0xa5,0xa6,0xa7,
  87. 0xa8,0xa9,0xaa,0xb2,0xb3,0xb4,0xb5,0xb6,
  88. 0xb7,0xb8,0xb9,0xba,0xc2,0xc3,0xc4,0xc5,
  89. 0xc6,0xc7,0xc8,0xc9,0xca,0xd2,0xd3,0xd4,
  90. 0xd5,0xd6,0xd7,0xd8,0xd9,0xda,0xe1,0xe2,
  91. 0xe3,0xe4,0xe5,0xe6,0xe7,0xe8,0xe9,0xea,
  92. 0xf1,0xf2,0xf3,0xf4,0xf5,0xf6,0xf7,0xf8,
  93. 0xf9,0xfa
  94. ];
  95. var std_dc_chrominance_nrcodes = [0,0,3,1,1,1,1,1,1,1,1,1,0,0,0,0,0];
  96. var std_dc_chrominance_values = [0,1,2,3,4,5,6,7,8,9,10,11];
  97. var std_ac_chrominance_nrcodes = [0,0,2,1,2,4,4,3,4,7,5,4,4,0,1,2,0x77];
  98. var std_ac_chrominance_values = [
  99. 0x00,0x01,0x02,0x03,0x11,0x04,0x05,0x21,
  100. 0x31,0x06,0x12,0x41,0x51,0x07,0x61,0x71,
  101. 0x13,0x22,0x32,0x81,0x08,0x14,0x42,0x91,
  102. 0xa1,0xb1,0xc1,0x09,0x23,0x33,0x52,0xf0,
  103. 0x15,0x62,0x72,0xd1,0x0a,0x16,0x24,0x34,
  104. 0xe1,0x25,0xf1,0x17,0x18,0x19,0x1a,0x26,
  105. 0x27,0x28,0x29,0x2a,0x35,0x36,0x37,0x38,
  106. 0x39,0x3a,0x43,0x44,0x45,0x46,0x47,0x48,
  107. 0x49,0x4a,0x53,0x54,0x55,0x56,0x57,0x58,
  108. 0x59,0x5a,0x63,0x64,0x65,0x66,0x67,0x68,
  109. 0x69,0x6a,0x73,0x74,0x75,0x76,0x77,0x78,
  110. 0x79,0x7a,0x82,0x83,0x84,0x85,0x86,0x87,
  111. 0x88,0x89,0x8a,0x92,0x93,0x94,0x95,0x96,
  112. 0x97,0x98,0x99,0x9a,0xa2,0xa3,0xa4,0xa5,
  113. 0xa6,0xa7,0xa8,0xa9,0xaa,0xb2,0xb3,0xb4,
  114. 0xb5,0xb6,0xb7,0xb8,0xb9,0xba,0xc2,0xc3,
  115. 0xc4,0xc5,0xc6,0xc7,0xc8,0xc9,0xca,0xd2,
  116. 0xd3,0xd4,0xd5,0xd6,0xd7,0xd8,0xd9,0xda,
  117. 0xe2,0xe3,0xe4,0xe5,0xe6,0xe7,0xe8,0xe9,
  118. 0xea,0xf2,0xf3,0xf4,0xf5,0xf6,0xf7,0xf8,
  119. 0xf9,0xfa
  120. ];
  121. function initQuantTables(sf){
  122. var YQT = [
  123. 16, 11, 10, 16, 24, 40, 51, 61,
  124. 12, 12, 14, 19, 26, 58, 60, 55,
  125. 14, 13, 16, 24, 40, 57, 69, 56,
  126. 14, 17, 22, 29, 51, 87, 80, 62,
  127. 18, 22, 37, 56, 68,109,103, 77,
  128. 24, 35, 55, 64, 81,104,113, 92,
  129. 49, 64, 78, 87,103,121,120,101,
  130. 72, 92, 95, 98,112,100,103, 99
  131. ];
  132. for (var i = 0; i < 64; i++) {
  133. var t = ffloor((YQT[i]*sf+50)/100);
  134. if (t < 1) {
  135. t = 1;
  136. } else if (t > 255) {
  137. t = 255;
  138. }
  139. YTable[ZigZag[i]] = t;
  140. }
  141. var UVQT = [
  142. 17, 18, 24, 47, 99, 99, 99, 99,
  143. 18, 21, 26, 66, 99, 99, 99, 99,
  144. 24, 26, 56, 99, 99, 99, 99, 99,
  145. 47, 66, 99, 99, 99, 99, 99, 99,
  146. 99, 99, 99, 99, 99, 99, 99, 99,
  147. 99, 99, 99, 99, 99, 99, 99, 99,
  148. 99, 99, 99, 99, 99, 99, 99, 99,
  149. 99, 99, 99, 99, 99, 99, 99, 99
  150. ];
  151. for (var j = 0; j < 64; j++) {
  152. var u = ffloor((UVQT[j]*sf+50)/100);
  153. if (u < 1) {
  154. u = 1;
  155. } else if (u > 255) {
  156. u = 255;
  157. }
  158. UVTable[ZigZag[j]] = u;
  159. }
  160. var aasf = [
  161. 1.0, 1.387039845, 1.306562965, 1.175875602,
  162. 1.0, 0.785694958, 0.541196100, 0.275899379
  163. ];
  164. var k = 0;
  165. for (var row = 0; row < 8; row++)
  166. {
  167. for (var col = 0; col < 8; col++)
  168. {
  169. fdtbl_Y[k] = (1.0 / (YTable [ZigZag[k]] * aasf[row] * aasf[col] * 8.0));
  170. fdtbl_UV[k] = (1.0 / (UVTable[ZigZag[k]] * aasf[row] * aasf[col] * 8.0));
  171. k++;
  172. }
  173. }
  174. }
  175. function computeHuffmanTbl(nrcodes, std_table){
  176. var codevalue = 0;
  177. var pos_in_table = 0;
  178. var HT = new Array();
  179. for (var k = 1; k <= 16; k++) {
  180. for (var j = 1; j <= nrcodes[k]; j++) {
  181. HT[std_table[pos_in_table]] = [];
  182. HT[std_table[pos_in_table]][0] = codevalue;
  183. HT[std_table[pos_in_table]][1] = k;
  184. pos_in_table++;
  185. codevalue++;
  186. }
  187. codevalue*=2;
  188. }
  189. return HT;
  190. }
  191. function initHuffmanTbl()
  192. {
  193. YDC_HT = computeHuffmanTbl(std_dc_luminance_nrcodes,std_dc_luminance_values);
  194. UVDC_HT = computeHuffmanTbl(std_dc_chrominance_nrcodes,std_dc_chrominance_values);
  195. YAC_HT = computeHuffmanTbl(std_ac_luminance_nrcodes,std_ac_luminance_values);
  196. UVAC_HT = computeHuffmanTbl(std_ac_chrominance_nrcodes,std_ac_chrominance_values);
  197. }
  198. function initCategoryNumber()
  199. {
  200. var nrlower = 1;
  201. var nrupper = 2;
  202. for (var cat = 1; cat <= 15; cat++) {
  203. //Positive numbers
  204. for (var nr = nrlower; nr<nrupper; nr++) {
  205. category[32767+nr] = cat;
  206. bitcode[32767+nr] = [];
  207. bitcode[32767+nr][1] = cat;
  208. bitcode[32767+nr][0] = nr;
  209. }
  210. //Negative numbers
  211. for (var nrneg =-(nrupper-1); nrneg<=-nrlower; nrneg++) {
  212. category[32767+nrneg] = cat;
  213. bitcode[32767+nrneg] = [];
  214. bitcode[32767+nrneg][1] = cat;
  215. bitcode[32767+nrneg][0] = nrupper-1+nrneg;
  216. }
  217. nrlower <<= 1;
  218. nrupper <<= 1;
  219. }
  220. }
  221. function initRGBYUVTable() {
  222. for(var i = 0; i < 256;i++) {
  223. RGB_YUV_TABLE[i] = 19595 * i;
  224. RGB_YUV_TABLE[(i+ 256)>>0] = 38470 * i;
  225. RGB_YUV_TABLE[(i+ 512)>>0] = 7471 * i + 0x8000;
  226. RGB_YUV_TABLE[(i+ 768)>>0] = -11059 * i;
  227. RGB_YUV_TABLE[(i+1024)>>0] = -21709 * i;
  228. RGB_YUV_TABLE[(i+1280)>>0] = 32768 * i + 0x807FFF;
  229. RGB_YUV_TABLE[(i+1536)>>0] = -27439 * i;
  230. RGB_YUV_TABLE[(i+1792)>>0] = - 5329 * i;
  231. }
  232. }
  233. // IO functions
  234. function writeBits(bs)
  235. {
  236. var value = bs[0];
  237. var posval = bs[1]-1;
  238. while ( posval >= 0 ) {
  239. if (value & (1 << posval) ) {
  240. bytenew |= (1 << bytepos);
  241. }
  242. posval--;
  243. bytepos--;
  244. if (bytepos < 0) {
  245. if (bytenew == 0xFF) {
  246. writeByte(0xFF);
  247. writeByte(0);
  248. }
  249. else {
  250. writeByte(bytenew);
  251. }
  252. bytepos=7;
  253. bytenew=0;
  254. }
  255. }
  256. }
  257. function writeByte(value)
  258. {
  259. //byteout.push(clt[value]); // write char directly instead of converting later
  260. byteout.push(value);
  261. }
  262. function writeWord(value)
  263. {
  264. writeByte((value>>8)&0xFF);
  265. writeByte((value )&0xFF);
  266. }
  267. // DCT & quantization core
  268. function fDCTQuant(data, fdtbl)
  269. {
  270. var d0, d1, d2, d3, d4, d5, d6, d7;
  271. /* Pass 1: process rows. */
  272. var dataOff=0;
  273. var i;
  274. var I8 = 8;
  275. var I64 = 64;
  276. for (i=0; i<I8; ++i)
  277. {
  278. d0 = data[dataOff];
  279. d1 = data[dataOff+1];
  280. d2 = data[dataOff+2];
  281. d3 = data[dataOff+3];
  282. d4 = data[dataOff+4];
  283. d5 = data[dataOff+5];
  284. d6 = data[dataOff+6];
  285. d7 = data[dataOff+7];
  286. var tmp0 = d0 + d7;
  287. var tmp7 = d0 - d7;
  288. var tmp1 = d1 + d6;
  289. var tmp6 = d1 - d6;
  290. var tmp2 = d2 + d5;
  291. var tmp5 = d2 - d5;
  292. var tmp3 = d3 + d4;
  293. var tmp4 = d3 - d4;
  294. /* Even part */
  295. var tmp10 = tmp0 + tmp3; /* phase 2 */
  296. var tmp13 = tmp0 - tmp3;
  297. var tmp11 = tmp1 + tmp2;
  298. var tmp12 = tmp1 - tmp2;
  299. data[dataOff] = tmp10 + tmp11; /* phase 3 */
  300. data[dataOff+4] = tmp10 - tmp11;
  301. var z1 = (tmp12 + tmp13) * 0.707106781; /* c4 */
  302. data[dataOff+2] = tmp13 + z1; /* phase 5 */
  303. data[dataOff+6] = tmp13 - z1;
  304. /* Odd part */
  305. tmp10 = tmp4 + tmp5; /* phase 2 */
  306. tmp11 = tmp5 + tmp6;
  307. tmp12 = tmp6 + tmp7;
  308. /* The rotator is modified from fig 4-8 to avoid extra negations. */
  309. var z5 = (tmp10 - tmp12) * 0.382683433; /* c6 */
  310. var z2 = 0.541196100 * tmp10 + z5; /* c2-c6 */
  311. var z4 = 1.306562965 * tmp12 + z5; /* c2+c6 */
  312. var z3 = tmp11 * 0.707106781; /* c4 */
  313. var z11 = tmp7 + z3; /* phase 5 */
  314. var z13 = tmp7 - z3;
  315. data[dataOff+5] = z13 + z2; /* phase 6 */
  316. data[dataOff+3] = z13 - z2;
  317. data[dataOff+1] = z11 + z4;
  318. data[dataOff+7] = z11 - z4;
  319. dataOff += 8; /* advance pointer to next row */
  320. }
  321. /* Pass 2: process columns. */
  322. dataOff = 0;
  323. for (i=0; i<I8; ++i)
  324. {
  325. d0 = data[dataOff];
  326. d1 = data[dataOff + 8];
  327. d2 = data[dataOff + 16];
  328. d3 = data[dataOff + 24];
  329. d4 = data[dataOff + 32];
  330. d5 = data[dataOff + 40];
  331. d6 = data[dataOff + 48];
  332. d7 = data[dataOff + 56];
  333. var tmp0p2 = d0 + d7;
  334. var tmp7p2 = d0 - d7;
  335. var tmp1p2 = d1 + d6;
  336. var tmp6p2 = d1 - d6;
  337. var tmp2p2 = d2 + d5;
  338. var tmp5p2 = d2 - d5;
  339. var tmp3p2 = d3 + d4;
  340. var tmp4p2 = d3 - d4;
  341. /* Even part */
  342. var tmp10p2 = tmp0p2 + tmp3p2; /* phase 2 */
  343. var tmp13p2 = tmp0p2 - tmp3p2;
  344. var tmp11p2 = tmp1p2 + tmp2p2;
  345. var tmp12p2 = tmp1p2 - tmp2p2;
  346. data[dataOff] = tmp10p2 + tmp11p2; /* phase 3 */
  347. data[dataOff+32] = tmp10p2 - tmp11p2;
  348. var z1p2 = (tmp12p2 + tmp13p2) * 0.707106781; /* c4 */
  349. data[dataOff+16] = tmp13p2 + z1p2; /* phase 5 */
  350. data[dataOff+48] = tmp13p2 - z1p2;
  351. /* Odd part */
  352. tmp10p2 = tmp4p2 + tmp5p2; /* phase 2 */
  353. tmp11p2 = tmp5p2 + tmp6p2;
  354. tmp12p2 = tmp6p2 + tmp7p2;
  355. /* The rotator is modified from fig 4-8 to avoid extra negations. */
  356. var z5p2 = (tmp10p2 - tmp12p2) * 0.382683433; /* c6 */
  357. var z2p2 = 0.541196100 * tmp10p2 + z5p2; /* c2-c6 */
  358. var z4p2 = 1.306562965 * tmp12p2 + z5p2; /* c2+c6 */
  359. var z3p2 = tmp11p2 * 0.707106781; /* c4 */
  360. var z11p2 = tmp7p2 + z3p2; /* phase 5 */
  361. var z13p2 = tmp7p2 - z3p2;
  362. data[dataOff+40] = z13p2 + z2p2; /* phase 6 */
  363. data[dataOff+24] = z13p2 - z2p2;
  364. data[dataOff+ 8] = z11p2 + z4p2;
  365. data[dataOff+56] = z11p2 - z4p2;
  366. dataOff++; /* advance pointer to next column */
  367. }
  368. // Quantize/descale the coefficients
  369. var fDCTQuant;
  370. for (i=0; i<I64; ++i)
  371. {
  372. // Apply the quantization and scaling factor & Round to nearest integer
  373. fDCTQuant = data[i]*fdtbl[i];
  374. outputfDCTQuant[i] = (fDCTQuant > 0.0) ? ((fDCTQuant + 0.5)|0) : ((fDCTQuant - 0.5)|0);
  375. //outputfDCTQuant[i] = fround(fDCTQuant);
  376. }
  377. return outputfDCTQuant;
  378. }
  379. function writeAPP0()
  380. {
  381. writeWord(0xFFE0); // marker
  382. writeWord(16); // length
  383. writeByte(0x4A); // J
  384. writeByte(0x46); // F
  385. writeByte(0x49); // I
  386. writeByte(0x46); // F
  387. writeByte(0); // = "JFIF",'\0'
  388. writeByte(1); // versionhi
  389. writeByte(1); // versionlo
  390. writeByte(0); // xyunits
  391. writeWord(1); // xdensity
  392. writeWord(1); // ydensity
  393. writeByte(0); // thumbnwidth
  394. writeByte(0); // thumbnheight
  395. }
  396. function writeSOF0(width, height)
  397. {
  398. writeWord(0xFFC0); // marker
  399. writeWord(17); // length, truecolor YUV JPG
  400. writeByte(8); // precision
  401. writeWord(height);
  402. writeWord(width);
  403. writeByte(3); // nrofcomponents
  404. writeByte(1); // IdY
  405. writeByte(0x11); // HVY
  406. writeByte(0); // QTY
  407. writeByte(2); // IdU
  408. writeByte(0x11); // HVU
  409. writeByte(1); // QTU
  410. writeByte(3); // IdV
  411. writeByte(0x11); // HVV
  412. writeByte(1); // QTV
  413. }
  414. function writeDQT()
  415. {
  416. writeWord(0xFFDB); // marker
  417. writeWord(132); // length
  418. writeByte(0);
  419. for (var i=0; i<64; i++) {
  420. writeByte(YTable[i]);
  421. }
  422. writeByte(1);
  423. for (var j=0; j<64; j++) {
  424. writeByte(UVTable[j]);
  425. }
  426. }
  427. function writeDHT()
  428. {
  429. writeWord(0xFFC4); // marker
  430. writeWord(0x01A2); // length
  431. writeByte(0); // HTYDCinfo
  432. for (var i=0; i<16; i++) {
  433. writeByte(std_dc_luminance_nrcodes[i+1]);
  434. }
  435. for (var j=0; j<=11; j++) {
  436. writeByte(std_dc_luminance_values[j]);
  437. }
  438. writeByte(0x10); // HTYACinfo
  439. for (var k=0; k<16; k++) {
  440. writeByte(std_ac_luminance_nrcodes[k+1]);
  441. }
  442. for (var l=0; l<=161; l++) {
  443. writeByte(std_ac_luminance_values[l]);
  444. }
  445. writeByte(1); // HTUDCinfo
  446. for (var m=0; m<16; m++) {
  447. writeByte(std_dc_chrominance_nrcodes[m+1]);
  448. }
  449. for (var n=0; n<=11; n++) {
  450. writeByte(std_dc_chrominance_values[n]);
  451. }
  452. writeByte(0x11); // HTUACinfo
  453. for (var o=0; o<16; o++) {
  454. writeByte(std_ac_chrominance_nrcodes[o+1]);
  455. }
  456. for (var p=0; p<=161; p++) {
  457. writeByte(std_ac_chrominance_values[p]);
  458. }
  459. }
  460. function writeSOS()
  461. {
  462. writeWord(0xFFDA); // marker
  463. writeWord(12); // length
  464. writeByte(3); // nrofcomponents
  465. writeByte(1); // IdY
  466. writeByte(0); // HTY
  467. writeByte(2); // IdU
  468. writeByte(0x11); // HTU
  469. writeByte(3); // IdV
  470. writeByte(0x11); // HTV
  471. writeByte(0); // Ss
  472. writeByte(0x3f); // Se
  473. writeByte(0); // Bf
  474. }
  475. function processDU(CDU, fdtbl, DC, HTDC, HTAC){
  476. var EOB = HTAC[0x00];
  477. var M16zeroes = HTAC[0xF0];
  478. var pos;
  479. var I16 = 16;
  480. var I63 = 63;
  481. var I64 = 64;
  482. var DU_DCT = fDCTQuant(CDU, fdtbl);
  483. //ZigZag reorder
  484. for (var j=0;j<I64;++j) {
  485. DU[ZigZag[j]]=DU_DCT[j];
  486. }
  487. var Diff = DU[0] - DC; DC = DU[0];
  488. //Encode DC
  489. if (Diff==0) {
  490. writeBits(HTDC[0]); // Diff might be 0
  491. } else {
  492. pos = 32767+Diff;
  493. writeBits(HTDC[category[pos]]);
  494. writeBits(bitcode[pos]);
  495. }
  496. //Encode ACs
  497. var end0pos = 63; // was const... which is crazy
  498. for (; (end0pos>0)&&(DU[end0pos]==0); end0pos--) {};
  499. //end0pos = first element in reverse order !=0
  500. if ( end0pos == 0) {
  501. writeBits(EOB);
  502. return DC;
  503. }
  504. var i = 1;
  505. var lng;
  506. while ( i <= end0pos ) {
  507. var startpos = i;
  508. for (; (DU[i]==0) && (i<=end0pos); ++i) {}
  509. var nrzeroes = i-startpos;
  510. if ( nrzeroes >= I16 ) {
  511. lng = nrzeroes>>4;
  512. for (var nrmarker=1; nrmarker <= lng; ++nrmarker)
  513. writeBits(M16zeroes);
  514. nrzeroes = nrzeroes&0xF;
  515. }
  516. pos = 32767+DU[i];
  517. writeBits(HTAC[(nrzeroes<<4)+category[pos]]);
  518. writeBits(bitcode[pos]);
  519. i++;
  520. }
  521. if ( end0pos != I63 ) {
  522. writeBits(EOB);
  523. }
  524. return DC;
  525. }
  526. function initCharLookupTable(){
  527. var sfcc = String.fromCharCode;
  528. for(var i=0; i < 256; i++){ ///// ACHTUNG // 255
  529. clt[i] = sfcc(i);
  530. }
  531. }
  532. this.encode = function(image,quality) // image data object
  533. {
  534. var time_start = new Date().getTime();
  535. if(quality) setQuality(quality);
  536. // Initialize bit writer
  537. byteout = new Array();
  538. bytenew=0;
  539. bytepos=7;
  540. // Add JPEG headers
  541. writeWord(0xFFD8); // SOI
  542. writeAPP0();
  543. writeDQT();
  544. writeSOF0(image.width,image.height);
  545. writeDHT();
  546. writeSOS();
  547. // Encode 8x8 macroblocks
  548. var DCY=0;
  549. var DCU=0;
  550. var DCV=0;
  551. bytenew=0;
  552. bytepos=7;
  553. this.encode.displayName = "_encode_";
  554. var imageData = image.data;
  555. var width = image.width;
  556. var height = image.height;
  557. var quadWidth = width*4;
  558. var tripleWidth = width*3;
  559. var x, y = 0;
  560. var r, g, b;
  561. var start,p, col,row,pos;
  562. while(y < height){
  563. x = 0;
  564. while(x < quadWidth){
  565. start = quadWidth * y + x;
  566. p = start;
  567. col = -1;
  568. row = 0;
  569. for(pos=0; pos < 64; pos++){
  570. row = pos >> 3;// /8
  571. col = ( pos & 7 ) * 4; // %8
  572. p = start + ( row * quadWidth ) + col;
  573. if(y+row >= height){ // padding bottom
  574. p-= (quadWidth*(y+1+row-height));
  575. }
  576. if(x+col >= quadWidth){ // padding right
  577. p-= ((x+col) - quadWidth +4)
  578. }
  579. r = imageData[ p++ ];
  580. g = imageData[ p++ ];
  581. b = imageData[ p++ ];
  582. /* // calculate YUV values dynamically
  583. YDU[pos]=((( 0.29900)*r+( 0.58700)*g+( 0.11400)*b))-128; //-0x80
  584. UDU[pos]=(((-0.16874)*r+(-0.33126)*g+( 0.50000)*b));
  585. VDU[pos]=((( 0.50000)*r+(-0.41869)*g+(-0.08131)*b));
  586. */
  587. // use lookup table (slightly faster)
  588. YDU[pos] = ((RGB_YUV_TABLE[r] + RGB_YUV_TABLE[(g + 256)>>0] + RGB_YUV_TABLE[(b + 512)>>0]) >> 16)-128;
  589. UDU[pos] = ((RGB_YUV_TABLE[(r + 768)>>0] + RGB_YUV_TABLE[(g + 1024)>>0] + RGB_YUV_TABLE[(b + 1280)>>0]) >> 16)-128;
  590. VDU[pos] = ((RGB_YUV_TABLE[(r + 1280)>>0] + RGB_YUV_TABLE[(g + 1536)>>0] + RGB_YUV_TABLE[(b + 1792)>>0]) >> 16)-128;
  591. }
  592. DCY = processDU(YDU, fdtbl_Y, DCY, YDC_HT, YAC_HT);
  593. DCU = processDU(UDU, fdtbl_UV, DCU, UVDC_HT, UVAC_HT);
  594. DCV = processDU(VDU, fdtbl_UV, DCV, UVDC_HT, UVAC_HT);
  595. x+=32;
  596. }
  597. y+=8;
  598. }
  599. ////////////////////////////////////////////////////////////////
  600. // Do the bit alignment of the EOI marker
  601. if ( bytepos >= 0 ) {
  602. var fillbits = [];
  603. fillbits[1] = bytepos+1;
  604. fillbits[0] = (1<<(bytepos+1))-1;
  605. writeBits(fillbits);
  606. }
  607. writeWord(0xFFD9); //EOI
  608. //return new Uint8Array(byteout);
  609. return new Buffer(byteout);
  610. var jpegDataUri = 'data:image/jpeg;base64,' + btoa(byteout.join(''));
  611. byteout = [];
  612. // benchmarking
  613. var duration = new Date().getTime() - time_start;
  614. //console.log('Encoding time: '+ duration + 'ms');
  615. //
  616. return jpegDataUri
  617. }
  618. function setQuality(quality){
  619. if (quality <= 0) {
  620. quality = 1;
  621. }
  622. if (quality > 100) {
  623. quality = 100;
  624. }
  625. if(currentQuality == quality) return // don't recalc if unchanged
  626. var sf = 0;
  627. if (quality < 50) {
  628. sf = Math.floor(5000 / quality);
  629. } else {
  630. sf = Math.floor(200 - quality*2);
  631. }
  632. initQuantTables(sf);
  633. currentQuality = quality;
  634. //console.log('Quality set to: '+quality +'%');
  635. }
  636. function init(){
  637. var time_start = new Date().getTime();
  638. if(!quality) quality = 50;
  639. // Create tables
  640. initCharLookupTable()
  641. initHuffmanTbl();
  642. initCategoryNumber();
  643. initRGBYUVTable();
  644. setQuality(quality);
  645. var duration = new Date().getTime() - time_start;
  646. //console.log('Initialization '+ duration + 'ms');
  647. }
  648. init();
  649. };
  650. if (typeof module !== undefined) {
  651. module.exports = encode;
  652. }
  653. function encode(imgData, qu) {
  654. if (typeof qu === 'undefined') qu = 50;
  655. var encoder = new JPEGEncoder(qu);
  656. var data = encoder.encode(imgData, qu);
  657. return {
  658. data: data,
  659. width: imgData.width,
  660. height: imgData.height
  661. };
  662. }
  663. // helper function to get the imageData of an existing image on the current page.
  664. function getImageDataFromImage(idOrElement){
  665. var theImg = (typeof(idOrElement)=='string')? document.getElementById(idOrElement):idOrElement;
  666. var cvs = document.createElement('canvas');
  667. cvs.width = theImg.width;
  668. cvs.height = theImg.height;
  669. var ctx = cvs.getContext("2d");
  670. ctx.drawImage(theImg,0,0);
  671. return (ctx.getImageData(0, 0, cvs.width, cvs.height));
  672. }