2
0

jpg.js 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939
  1. /* Copyright 2017 Mozilla Foundation
  2. *
  3. * Licensed under the Apache License, Version 2.0 (the "License");
  4. * you may not use this file except in compliance with the License.
  5. * You may obtain a copy of the License at
  6. *
  7. * http://www.apache.org/licenses/LICENSE-2.0
  8. *
  9. * Unless required by applicable law or agreed to in writing, software
  10. * distributed under the License is distributed on an "AS IS" BASIS,
  11. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. * See the License for the specific language governing permissions and
  13. * limitations under the License.
  14. */
  15. 'use strict';
  16. var sharedUtil = require('../shared/util.js');
  17. var warn = sharedUtil.warn;
  18. var error = sharedUtil.error;
  19. var JpegImage = function JpegImageClosure() {
  20. var dctZigZag = new Uint8Array([
  21. 0,
  22. 1,
  23. 8,
  24. 16,
  25. 9,
  26. 2,
  27. 3,
  28. 10,
  29. 17,
  30. 24,
  31. 32,
  32. 25,
  33. 18,
  34. 11,
  35. 4,
  36. 5,
  37. 12,
  38. 19,
  39. 26,
  40. 33,
  41. 40,
  42. 48,
  43. 41,
  44. 34,
  45. 27,
  46. 20,
  47. 13,
  48. 6,
  49. 7,
  50. 14,
  51. 21,
  52. 28,
  53. 35,
  54. 42,
  55. 49,
  56. 56,
  57. 57,
  58. 50,
  59. 43,
  60. 36,
  61. 29,
  62. 22,
  63. 15,
  64. 23,
  65. 30,
  66. 37,
  67. 44,
  68. 51,
  69. 58,
  70. 59,
  71. 52,
  72. 45,
  73. 38,
  74. 31,
  75. 39,
  76. 46,
  77. 53,
  78. 60,
  79. 61,
  80. 54,
  81. 47,
  82. 55,
  83. 62,
  84. 63
  85. ]);
  86. var dctCos1 = 4017;
  87. var dctSin1 = 799;
  88. var dctCos3 = 3406;
  89. var dctSin3 = 2276;
  90. var dctCos6 = 1567;
  91. var dctSin6 = 3784;
  92. var dctSqrt2 = 5793;
  93. var dctSqrt1d2 = 2896;
  94. function JpegImage() {
  95. this.decodeTransform = null;
  96. this.colorTransform = -1;
  97. }
  98. function buildHuffmanTable(codeLengths, values) {
  99. var k = 0, code = [], i, j, length = 16;
  100. while (length > 0 && !codeLengths[length - 1]) {
  101. length--;
  102. }
  103. code.push({
  104. children: [],
  105. index: 0
  106. });
  107. var p = code[0], q;
  108. for (i = 0; i < length; i++) {
  109. for (j = 0; j < codeLengths[i]; j++) {
  110. p = code.pop();
  111. p.children[p.index] = values[k];
  112. while (p.index > 0) {
  113. p = code.pop();
  114. }
  115. p.index++;
  116. code.push(p);
  117. while (code.length <= i) {
  118. code.push(q = {
  119. children: [],
  120. index: 0
  121. });
  122. p.children[p.index] = q.children;
  123. p = q;
  124. }
  125. k++;
  126. }
  127. if (i + 1 < length) {
  128. code.push(q = {
  129. children: [],
  130. index: 0
  131. });
  132. p.children[p.index] = q.children;
  133. p = q;
  134. }
  135. }
  136. return code[0].children;
  137. }
  138. function getBlockBufferOffset(component, row, col) {
  139. return 64 * ((component.blocksPerLine + 1) * row + col);
  140. }
  141. function decodeScan(data, offset, frame, components, resetInterval, spectralStart, spectralEnd, successivePrev, successive) {
  142. var mcusPerLine = frame.mcusPerLine;
  143. var progressive = frame.progressive;
  144. var startOffset = offset, bitsData = 0, bitsCount = 0;
  145. function readBit() {
  146. if (bitsCount > 0) {
  147. bitsCount--;
  148. return bitsData >> bitsCount & 1;
  149. }
  150. bitsData = data[offset++];
  151. if (bitsData === 0xFF) {
  152. var nextByte = data[offset++];
  153. if (nextByte) {
  154. error('JPEG error: unexpected marker ' + (bitsData << 8 | nextByte).toString(16));
  155. }
  156. }
  157. bitsCount = 7;
  158. return bitsData >>> 7;
  159. }
  160. function decodeHuffman(tree) {
  161. var node = tree;
  162. while (true) {
  163. node = node[readBit()];
  164. if (typeof node === 'number') {
  165. return node;
  166. }
  167. if (typeof node !== 'object') {
  168. error('JPEG error: invalid huffman sequence');
  169. }
  170. }
  171. }
  172. function receive(length) {
  173. var n = 0;
  174. while (length > 0) {
  175. n = n << 1 | readBit();
  176. length--;
  177. }
  178. return n;
  179. }
  180. function receiveAndExtend(length) {
  181. if (length === 1) {
  182. return readBit() === 1 ? 1 : -1;
  183. }
  184. var n = receive(length);
  185. if (n >= 1 << length - 1) {
  186. return n;
  187. }
  188. return n + (-1 << length) + 1;
  189. }
  190. function decodeBaseline(component, offset) {
  191. var t = decodeHuffman(component.huffmanTableDC);
  192. var diff = t === 0 ? 0 : receiveAndExtend(t);
  193. component.blockData[offset] = component.pred += diff;
  194. var k = 1;
  195. while (k < 64) {
  196. var rs = decodeHuffman(component.huffmanTableAC);
  197. var s = rs & 15, r = rs >> 4;
  198. if (s === 0) {
  199. if (r < 15) {
  200. break;
  201. }
  202. k += 16;
  203. continue;
  204. }
  205. k += r;
  206. var z = dctZigZag[k];
  207. component.blockData[offset + z] = receiveAndExtend(s);
  208. k++;
  209. }
  210. }
  211. function decodeDCFirst(component, offset) {
  212. var t = decodeHuffman(component.huffmanTableDC);
  213. var diff = t === 0 ? 0 : receiveAndExtend(t) << successive;
  214. component.blockData[offset] = component.pred += diff;
  215. }
  216. function decodeDCSuccessive(component, offset) {
  217. component.blockData[offset] |= readBit() << successive;
  218. }
  219. var eobrun = 0;
  220. function decodeACFirst(component, offset) {
  221. if (eobrun > 0) {
  222. eobrun--;
  223. return;
  224. }
  225. var k = spectralStart, e = spectralEnd;
  226. while (k <= e) {
  227. var rs = decodeHuffman(component.huffmanTableAC);
  228. var s = rs & 15, r = rs >> 4;
  229. if (s === 0) {
  230. if (r < 15) {
  231. eobrun = receive(r) + (1 << r) - 1;
  232. break;
  233. }
  234. k += 16;
  235. continue;
  236. }
  237. k += r;
  238. var z = dctZigZag[k];
  239. component.blockData[offset + z] = receiveAndExtend(s) * (1 << successive);
  240. k++;
  241. }
  242. }
  243. var successiveACState = 0, successiveACNextValue;
  244. function decodeACSuccessive(component, offset) {
  245. var k = spectralStart;
  246. var e = spectralEnd;
  247. var r = 0;
  248. var s;
  249. var rs;
  250. while (k <= e) {
  251. var z = dctZigZag[k];
  252. switch (successiveACState) {
  253. case 0:
  254. rs = decodeHuffman(component.huffmanTableAC);
  255. s = rs & 15;
  256. r = rs >> 4;
  257. if (s === 0) {
  258. if (r < 15) {
  259. eobrun = receive(r) + (1 << r);
  260. successiveACState = 4;
  261. } else {
  262. r = 16;
  263. successiveACState = 1;
  264. }
  265. } else {
  266. if (s !== 1) {
  267. error('JPEG error: invalid ACn encoding');
  268. }
  269. successiveACNextValue = receiveAndExtend(s);
  270. successiveACState = r ? 2 : 3;
  271. }
  272. continue;
  273. case 1:
  274. case 2:
  275. if (component.blockData[offset + z]) {
  276. component.blockData[offset + z] += readBit() << successive;
  277. } else {
  278. r--;
  279. if (r === 0) {
  280. successiveACState = successiveACState === 2 ? 3 : 0;
  281. }
  282. }
  283. break;
  284. case 3:
  285. if (component.blockData[offset + z]) {
  286. component.blockData[offset + z] += readBit() << successive;
  287. } else {
  288. component.blockData[offset + z] = successiveACNextValue << successive;
  289. successiveACState = 0;
  290. }
  291. break;
  292. case 4:
  293. if (component.blockData[offset + z]) {
  294. component.blockData[offset + z] += readBit() << successive;
  295. }
  296. break;
  297. }
  298. k++;
  299. }
  300. if (successiveACState === 4) {
  301. eobrun--;
  302. if (eobrun === 0) {
  303. successiveACState = 0;
  304. }
  305. }
  306. }
  307. function decodeMcu(component, decode, mcu, row, col) {
  308. var mcuRow = mcu / mcusPerLine | 0;
  309. var mcuCol = mcu % mcusPerLine;
  310. var blockRow = mcuRow * component.v + row;
  311. var blockCol = mcuCol * component.h + col;
  312. var offset = getBlockBufferOffset(component, blockRow, blockCol);
  313. decode(component, offset);
  314. }
  315. function decodeBlock(component, decode, mcu) {
  316. var blockRow = mcu / component.blocksPerLine | 0;
  317. var blockCol = mcu % component.blocksPerLine;
  318. var offset = getBlockBufferOffset(component, blockRow, blockCol);
  319. decode(component, offset);
  320. }
  321. var componentsLength = components.length;
  322. var component, i, j, k, n;
  323. var decodeFn;
  324. if (progressive) {
  325. if (spectralStart === 0) {
  326. decodeFn = successivePrev === 0 ? decodeDCFirst : decodeDCSuccessive;
  327. } else {
  328. decodeFn = successivePrev === 0 ? decodeACFirst : decodeACSuccessive;
  329. }
  330. } else {
  331. decodeFn = decodeBaseline;
  332. }
  333. var mcu = 0, marker;
  334. var mcuExpected;
  335. if (componentsLength === 1) {
  336. mcuExpected = components[0].blocksPerLine * components[0].blocksPerColumn;
  337. } else {
  338. mcuExpected = mcusPerLine * frame.mcusPerColumn;
  339. }
  340. var h, v;
  341. while (mcu < mcuExpected) {
  342. var mcuToRead = resetInterval ? Math.min(mcuExpected - mcu, resetInterval) : mcuExpected;
  343. for (i = 0; i < componentsLength; i++) {
  344. components[i].pred = 0;
  345. }
  346. eobrun = 0;
  347. if (componentsLength === 1) {
  348. component = components[0];
  349. for (n = 0; n < mcuToRead; n++) {
  350. decodeBlock(component, decodeFn, mcu);
  351. mcu++;
  352. }
  353. } else {
  354. for (n = 0; n < mcuToRead; n++) {
  355. for (i = 0; i < componentsLength; i++) {
  356. component = components[i];
  357. h = component.h;
  358. v = component.v;
  359. for (j = 0; j < v; j++) {
  360. for (k = 0; k < h; k++) {
  361. decodeMcu(component, decodeFn, mcu, j, k);
  362. }
  363. }
  364. }
  365. mcu++;
  366. }
  367. }
  368. bitsCount = 0;
  369. marker = data[offset] << 8 | data[offset + 1];
  370. while (data[offset] === 0x00 && offset < data.length - 1) {
  371. offset++;
  372. marker = data[offset] << 8 | data[offset + 1];
  373. }
  374. if (marker <= 0xFF00) {
  375. error('JPEG error: marker was not found');
  376. }
  377. if (marker >= 0xFFD0 && marker <= 0xFFD7) {
  378. offset += 2;
  379. } else {
  380. break;
  381. }
  382. }
  383. return offset - startOffset;
  384. }
  385. function quantizeAndInverse(component, blockBufferOffset, p) {
  386. var qt = component.quantizationTable, blockData = component.blockData;
  387. var v0, v1, v2, v3, v4, v5, v6, v7;
  388. var p0, p1, p2, p3, p4, p5, p6, p7;
  389. var t;
  390. if (!qt) {
  391. error('JPEG error: missing required Quantization Table.');
  392. }
  393. for (var row = 0; row < 64; row += 8) {
  394. p0 = blockData[blockBufferOffset + row];
  395. p1 = blockData[blockBufferOffset + row + 1];
  396. p2 = blockData[blockBufferOffset + row + 2];
  397. p3 = blockData[blockBufferOffset + row + 3];
  398. p4 = blockData[blockBufferOffset + row + 4];
  399. p5 = blockData[blockBufferOffset + row + 5];
  400. p6 = blockData[blockBufferOffset + row + 6];
  401. p7 = blockData[blockBufferOffset + row + 7];
  402. p0 *= qt[row];
  403. if ((p1 | p2 | p3 | p4 | p5 | p6 | p7) === 0) {
  404. t = dctSqrt2 * p0 + 512 >> 10;
  405. p[row] = t;
  406. p[row + 1] = t;
  407. p[row + 2] = t;
  408. p[row + 3] = t;
  409. p[row + 4] = t;
  410. p[row + 5] = t;
  411. p[row + 6] = t;
  412. p[row + 7] = t;
  413. continue;
  414. }
  415. p1 *= qt[row + 1];
  416. p2 *= qt[row + 2];
  417. p3 *= qt[row + 3];
  418. p4 *= qt[row + 4];
  419. p5 *= qt[row + 5];
  420. p6 *= qt[row + 6];
  421. p7 *= qt[row + 7];
  422. v0 = dctSqrt2 * p0 + 128 >> 8;
  423. v1 = dctSqrt2 * p4 + 128 >> 8;
  424. v2 = p2;
  425. v3 = p6;
  426. v4 = dctSqrt1d2 * (p1 - p7) + 128 >> 8;
  427. v7 = dctSqrt1d2 * (p1 + p7) + 128 >> 8;
  428. v5 = p3 << 4;
  429. v6 = p5 << 4;
  430. v0 = v0 + v1 + 1 >> 1;
  431. v1 = v0 - v1;
  432. t = v2 * dctSin6 + v3 * dctCos6 + 128 >> 8;
  433. v2 = v2 * dctCos6 - v3 * dctSin6 + 128 >> 8;
  434. v3 = t;
  435. v4 = v4 + v6 + 1 >> 1;
  436. v6 = v4 - v6;
  437. v7 = v7 + v5 + 1 >> 1;
  438. v5 = v7 - v5;
  439. v0 = v0 + v3 + 1 >> 1;
  440. v3 = v0 - v3;
  441. v1 = v1 + v2 + 1 >> 1;
  442. v2 = v1 - v2;
  443. t = v4 * dctSin3 + v7 * dctCos3 + 2048 >> 12;
  444. v4 = v4 * dctCos3 - v7 * dctSin3 + 2048 >> 12;
  445. v7 = t;
  446. t = v5 * dctSin1 + v6 * dctCos1 + 2048 >> 12;
  447. v5 = v5 * dctCos1 - v6 * dctSin1 + 2048 >> 12;
  448. v6 = t;
  449. p[row] = v0 + v7;
  450. p[row + 7] = v0 - v7;
  451. p[row + 1] = v1 + v6;
  452. p[row + 6] = v1 - v6;
  453. p[row + 2] = v2 + v5;
  454. p[row + 5] = v2 - v5;
  455. p[row + 3] = v3 + v4;
  456. p[row + 4] = v3 - v4;
  457. }
  458. for (var col = 0; col < 8; ++col) {
  459. p0 = p[col];
  460. p1 = p[col + 8];
  461. p2 = p[col + 16];
  462. p3 = p[col + 24];
  463. p4 = p[col + 32];
  464. p5 = p[col + 40];
  465. p6 = p[col + 48];
  466. p7 = p[col + 56];
  467. if ((p1 | p2 | p3 | p4 | p5 | p6 | p7) === 0) {
  468. t = dctSqrt2 * p0 + 8192 >> 14;
  469. t = t < -2040 ? 0 : t >= 2024 ? 255 : t + 2056 >> 4;
  470. blockData[blockBufferOffset + col] = t;
  471. blockData[blockBufferOffset + col + 8] = t;
  472. blockData[blockBufferOffset + col + 16] = t;
  473. blockData[blockBufferOffset + col + 24] = t;
  474. blockData[blockBufferOffset + col + 32] = t;
  475. blockData[blockBufferOffset + col + 40] = t;
  476. blockData[blockBufferOffset + col + 48] = t;
  477. blockData[blockBufferOffset + col + 56] = t;
  478. continue;
  479. }
  480. v0 = dctSqrt2 * p0 + 2048 >> 12;
  481. v1 = dctSqrt2 * p4 + 2048 >> 12;
  482. v2 = p2;
  483. v3 = p6;
  484. v4 = dctSqrt1d2 * (p1 - p7) + 2048 >> 12;
  485. v7 = dctSqrt1d2 * (p1 + p7) + 2048 >> 12;
  486. v5 = p3;
  487. v6 = p5;
  488. v0 = (v0 + v1 + 1 >> 1) + 4112;
  489. v1 = v0 - v1;
  490. t = v2 * dctSin6 + v3 * dctCos6 + 2048 >> 12;
  491. v2 = v2 * dctCos6 - v3 * dctSin6 + 2048 >> 12;
  492. v3 = t;
  493. v4 = v4 + v6 + 1 >> 1;
  494. v6 = v4 - v6;
  495. v7 = v7 + v5 + 1 >> 1;
  496. v5 = v7 - v5;
  497. v0 = v0 + v3 + 1 >> 1;
  498. v3 = v0 - v3;
  499. v1 = v1 + v2 + 1 >> 1;
  500. v2 = v1 - v2;
  501. t = v4 * dctSin3 + v7 * dctCos3 + 2048 >> 12;
  502. v4 = v4 * dctCos3 - v7 * dctSin3 + 2048 >> 12;
  503. v7 = t;
  504. t = v5 * dctSin1 + v6 * dctCos1 + 2048 >> 12;
  505. v5 = v5 * dctCos1 - v6 * dctSin1 + 2048 >> 12;
  506. v6 = t;
  507. p0 = v0 + v7;
  508. p7 = v0 - v7;
  509. p1 = v1 + v6;
  510. p6 = v1 - v6;
  511. p2 = v2 + v5;
  512. p5 = v2 - v5;
  513. p3 = v3 + v4;
  514. p4 = v3 - v4;
  515. p0 = p0 < 16 ? 0 : p0 >= 4080 ? 255 : p0 >> 4;
  516. p1 = p1 < 16 ? 0 : p1 >= 4080 ? 255 : p1 >> 4;
  517. p2 = p2 < 16 ? 0 : p2 >= 4080 ? 255 : p2 >> 4;
  518. p3 = p3 < 16 ? 0 : p3 >= 4080 ? 255 : p3 >> 4;
  519. p4 = p4 < 16 ? 0 : p4 >= 4080 ? 255 : p4 >> 4;
  520. p5 = p5 < 16 ? 0 : p5 >= 4080 ? 255 : p5 >> 4;
  521. p6 = p6 < 16 ? 0 : p6 >= 4080 ? 255 : p6 >> 4;
  522. p7 = p7 < 16 ? 0 : p7 >= 4080 ? 255 : p7 >> 4;
  523. blockData[blockBufferOffset + col] = p0;
  524. blockData[blockBufferOffset + col + 8] = p1;
  525. blockData[blockBufferOffset + col + 16] = p2;
  526. blockData[blockBufferOffset + col + 24] = p3;
  527. blockData[blockBufferOffset + col + 32] = p4;
  528. blockData[blockBufferOffset + col + 40] = p5;
  529. blockData[blockBufferOffset + col + 48] = p6;
  530. blockData[blockBufferOffset + col + 56] = p7;
  531. }
  532. }
  533. function buildComponentData(frame, component) {
  534. var blocksPerLine = component.blocksPerLine;
  535. var blocksPerColumn = component.blocksPerColumn;
  536. var computationBuffer = new Int16Array(64);
  537. for (var blockRow = 0; blockRow < blocksPerColumn; blockRow++) {
  538. for (var blockCol = 0; blockCol < blocksPerLine; blockCol++) {
  539. var offset = getBlockBufferOffset(component, blockRow, blockCol);
  540. quantizeAndInverse(component, offset, computationBuffer);
  541. }
  542. }
  543. return component.blockData;
  544. }
  545. function clamp0to255(a) {
  546. return a <= 0 ? 0 : a >= 255 ? 255 : a;
  547. }
  548. JpegImage.prototype = {
  549. parse: function parse(data) {
  550. function readUint16() {
  551. var value = data[offset] << 8 | data[offset + 1];
  552. offset += 2;
  553. return value;
  554. }
  555. function readDataBlock() {
  556. function isValidMarkerAt(pos) {
  557. if (pos < data.length - 1) {
  558. return data[pos] === 0xFF && data[pos + 1] >= 0xC0 && data[pos + 1] <= 0xFE;
  559. }
  560. return true;
  561. }
  562. var length = readUint16();
  563. var endOffset = offset + length - 2;
  564. if (!isValidMarkerAt(endOffset)) {
  565. warn('readDataBlock - incorrect length, next marker is: ' + (data[endOffset] << 8 | data[endOffset + 1]).toString('16'));
  566. var pos = offset;
  567. while (!isValidMarkerAt(pos)) {
  568. pos++;
  569. }
  570. endOffset = pos;
  571. }
  572. var array = data.subarray(offset, endOffset);
  573. offset += array.length;
  574. return array;
  575. }
  576. function prepareComponents(frame) {
  577. var mcusPerLine = Math.ceil(frame.samplesPerLine / 8 / frame.maxH);
  578. var mcusPerColumn = Math.ceil(frame.scanLines / 8 / frame.maxV);
  579. for (var i = 0; i < frame.components.length; i++) {
  580. component = frame.components[i];
  581. var blocksPerLine = Math.ceil(Math.ceil(frame.samplesPerLine / 8) * component.h / frame.maxH);
  582. var blocksPerColumn = Math.ceil(Math.ceil(frame.scanLines / 8) * component.v / frame.maxV);
  583. var blocksPerLineForMcu = mcusPerLine * component.h;
  584. var blocksPerColumnForMcu = mcusPerColumn * component.v;
  585. var blocksBufferSize = 64 * blocksPerColumnForMcu * (blocksPerLineForMcu + 1);
  586. component.blockData = new Int16Array(blocksBufferSize);
  587. component.blocksPerLine = blocksPerLine;
  588. component.blocksPerColumn = blocksPerColumn;
  589. }
  590. frame.mcusPerLine = mcusPerLine;
  591. frame.mcusPerColumn = mcusPerColumn;
  592. }
  593. var offset = 0;
  594. var jfif = null;
  595. var adobe = null;
  596. var frame, resetInterval;
  597. var quantizationTables = [];
  598. var huffmanTablesAC = [], huffmanTablesDC = [];
  599. var fileMarker = readUint16();
  600. if (fileMarker !== 0xFFD8) {
  601. error('JPEG error: SOI not found');
  602. }
  603. fileMarker = readUint16();
  604. while (fileMarker !== 0xFFD9) {
  605. var i, j, l;
  606. switch (fileMarker) {
  607. case 0xFFE0:
  608. case 0xFFE1:
  609. case 0xFFE2:
  610. case 0xFFE3:
  611. case 0xFFE4:
  612. case 0xFFE5:
  613. case 0xFFE6:
  614. case 0xFFE7:
  615. case 0xFFE8:
  616. case 0xFFE9:
  617. case 0xFFEA:
  618. case 0xFFEB:
  619. case 0xFFEC:
  620. case 0xFFED:
  621. case 0xFFEE:
  622. case 0xFFEF:
  623. case 0xFFFE:
  624. var appData = readDataBlock();
  625. if (fileMarker === 0xFFE0) {
  626. if (appData[0] === 0x4A && appData[1] === 0x46 && appData[2] === 0x49 && appData[3] === 0x46 && appData[4] === 0) {
  627. jfif = {
  628. version: {
  629. major: appData[5],
  630. minor: appData[6]
  631. },
  632. densityUnits: appData[7],
  633. xDensity: appData[8] << 8 | appData[9],
  634. yDensity: appData[10] << 8 | appData[11],
  635. thumbWidth: appData[12],
  636. thumbHeight: appData[13],
  637. thumbData: appData.subarray(14, 14 + 3 * appData[12] * appData[13])
  638. };
  639. }
  640. }
  641. if (fileMarker === 0xFFEE) {
  642. if (appData[0] === 0x41 && appData[1] === 0x64 && appData[2] === 0x6F && appData[3] === 0x62 && appData[4] === 0x65) {
  643. adobe = {
  644. version: appData[5] << 8 | appData[6],
  645. flags0: appData[7] << 8 | appData[8],
  646. flags1: appData[9] << 8 | appData[10],
  647. transformCode: appData[11]
  648. };
  649. }
  650. }
  651. break;
  652. case 0xFFDB:
  653. var quantizationTablesLength = readUint16();
  654. var quantizationTablesEnd = quantizationTablesLength + offset - 2;
  655. var z;
  656. while (offset < quantizationTablesEnd) {
  657. var quantizationTableSpec = data[offset++];
  658. var tableData = new Uint16Array(64);
  659. if (quantizationTableSpec >> 4 === 0) {
  660. for (j = 0; j < 64; j++) {
  661. z = dctZigZag[j];
  662. tableData[z] = data[offset++];
  663. }
  664. } else if (quantizationTableSpec >> 4 === 1) {
  665. for (j = 0; j < 64; j++) {
  666. z = dctZigZag[j];
  667. tableData[z] = readUint16();
  668. }
  669. } else {
  670. error('JPEG error: DQT - invalid table spec');
  671. }
  672. quantizationTables[quantizationTableSpec & 15] = tableData;
  673. }
  674. break;
  675. case 0xFFC0:
  676. case 0xFFC1:
  677. case 0xFFC2:
  678. if (frame) {
  679. error('JPEG error: Only single frame JPEGs supported');
  680. }
  681. readUint16();
  682. frame = {};
  683. frame.extended = fileMarker === 0xFFC1;
  684. frame.progressive = fileMarker === 0xFFC2;
  685. frame.precision = data[offset++];
  686. frame.scanLines = readUint16();
  687. frame.samplesPerLine = readUint16();
  688. frame.components = [];
  689. frame.componentIds = {};
  690. var componentsCount = data[offset++], componentId;
  691. var maxH = 0, maxV = 0;
  692. for (i = 0; i < componentsCount; i++) {
  693. componentId = data[offset];
  694. var h = data[offset + 1] >> 4;
  695. var v = data[offset + 1] & 15;
  696. if (maxH < h) {
  697. maxH = h;
  698. }
  699. if (maxV < v) {
  700. maxV = v;
  701. }
  702. var qId = data[offset + 2];
  703. l = frame.components.push({
  704. h: h,
  705. v: v,
  706. quantizationId: qId,
  707. quantizationTable: null
  708. });
  709. frame.componentIds[componentId] = l - 1;
  710. offset += 3;
  711. }
  712. frame.maxH = maxH;
  713. frame.maxV = maxV;
  714. prepareComponents(frame);
  715. break;
  716. case 0xFFC4:
  717. var huffmanLength = readUint16();
  718. for (i = 2; i < huffmanLength;) {
  719. var huffmanTableSpec = data[offset++];
  720. var codeLengths = new Uint8Array(16);
  721. var codeLengthSum = 0;
  722. for (j = 0; j < 16; j++, offset++) {
  723. codeLengthSum += codeLengths[j] = data[offset];
  724. }
  725. var huffmanValues = new Uint8Array(codeLengthSum);
  726. for (j = 0; j < codeLengthSum; j++, offset++) {
  727. huffmanValues[j] = data[offset];
  728. }
  729. i += 17 + codeLengthSum;
  730. (huffmanTableSpec >> 4 === 0 ? huffmanTablesDC : huffmanTablesAC)[huffmanTableSpec & 15] = buildHuffmanTable(codeLengths, huffmanValues);
  731. }
  732. break;
  733. case 0xFFDD:
  734. readUint16();
  735. resetInterval = readUint16();
  736. break;
  737. case 0xFFDA:
  738. readUint16();
  739. var selectorsCount = data[offset++];
  740. var components = [], component;
  741. for (i = 0; i < selectorsCount; i++) {
  742. var componentIndex = frame.componentIds[data[offset++]];
  743. component = frame.components[componentIndex];
  744. var tableSpec = data[offset++];
  745. component.huffmanTableDC = huffmanTablesDC[tableSpec >> 4];
  746. component.huffmanTableAC = huffmanTablesAC[tableSpec & 15];
  747. components.push(component);
  748. }
  749. var spectralStart = data[offset++];
  750. var spectralEnd = data[offset++];
  751. var successiveApproximation = data[offset++];
  752. var processed = decodeScan(data, offset, frame, components, resetInterval, spectralStart, spectralEnd, successiveApproximation >> 4, successiveApproximation & 15);
  753. offset += processed;
  754. break;
  755. case 0xFFFF:
  756. if (data[offset] !== 0xFF) {
  757. offset--;
  758. }
  759. break;
  760. default:
  761. if (data[offset - 3] === 0xFF && data[offset - 2] >= 0xC0 && data[offset - 2] <= 0xFE) {
  762. offset -= 3;
  763. break;
  764. }
  765. error('JPEG error: unknown marker ' + fileMarker.toString(16));
  766. }
  767. fileMarker = readUint16();
  768. }
  769. this.width = frame.samplesPerLine;
  770. this.height = frame.scanLines;
  771. this.jfif = jfif;
  772. this.adobe = adobe;
  773. this.components = [];
  774. for (i = 0; i < frame.components.length; i++) {
  775. component = frame.components[i];
  776. var quantizationTable = quantizationTables[component.quantizationId];
  777. if (quantizationTable) {
  778. component.quantizationTable = quantizationTable;
  779. }
  780. this.components.push({
  781. output: buildComponentData(frame, component),
  782. scaleX: component.h / frame.maxH,
  783. scaleY: component.v / frame.maxV,
  784. blocksPerLine: component.blocksPerLine,
  785. blocksPerColumn: component.blocksPerColumn
  786. });
  787. }
  788. this.numComponents = this.components.length;
  789. },
  790. _getLinearizedBlockData: function getLinearizedBlockData(width, height) {
  791. var scaleX = this.width / width, scaleY = this.height / height;
  792. var component, componentScaleX, componentScaleY, blocksPerScanline;
  793. var x, y, i, j, k;
  794. var index;
  795. var offset = 0;
  796. var output;
  797. var numComponents = this.components.length;
  798. var dataLength = width * height * numComponents;
  799. var data = new Uint8Array(dataLength);
  800. var xScaleBlockOffset = new Uint32Array(width);
  801. var mask3LSB = 0xfffffff8;
  802. for (i = 0; i < numComponents; i++) {
  803. component = this.components[i];
  804. componentScaleX = component.scaleX * scaleX;
  805. componentScaleY = component.scaleY * scaleY;
  806. offset = i;
  807. output = component.output;
  808. blocksPerScanline = component.blocksPerLine + 1 << 3;
  809. for (x = 0; x < width; x++) {
  810. j = 0 | x * componentScaleX;
  811. xScaleBlockOffset[x] = (j & mask3LSB) << 3 | j & 7;
  812. }
  813. for (y = 0; y < height; y++) {
  814. j = 0 | y * componentScaleY;
  815. index = blocksPerScanline * (j & mask3LSB) | (j & 7) << 3;
  816. for (x = 0; x < width; x++) {
  817. data[offset] = output[index + xScaleBlockOffset[x]];
  818. offset += numComponents;
  819. }
  820. }
  821. }
  822. var transform = this.decodeTransform;
  823. if (transform) {
  824. for (i = 0; i < dataLength;) {
  825. for (j = 0, k = 0; j < numComponents; j++, i++, k += 2) {
  826. data[i] = (data[i] * transform[k] >> 8) + transform[k + 1];
  827. }
  828. }
  829. }
  830. return data;
  831. },
  832. _isColorConversionNeeded: function isColorConversionNeeded() {
  833. if (this.adobe && this.adobe.transformCode) {
  834. return true;
  835. } else if (this.numComponents === 3) {
  836. if (!this.adobe && this.colorTransform === 0) {
  837. return false;
  838. }
  839. return true;
  840. }
  841. if (!this.adobe && this.colorTransform === 1) {
  842. return true;
  843. }
  844. return false;
  845. },
  846. _convertYccToRgb: function convertYccToRgb(data) {
  847. var Y, Cb, Cr;
  848. for (var i = 0, length = data.length; i < length; i += 3) {
  849. Y = data[i];
  850. Cb = data[i + 1];
  851. Cr = data[i + 2];
  852. data[i] = clamp0to255(Y - 179.456 + 1.402 * Cr);
  853. data[i + 1] = clamp0to255(Y + 135.459 - 0.344 * Cb - 0.714 * Cr);
  854. data[i + 2] = clamp0to255(Y - 226.816 + 1.772 * Cb);
  855. }
  856. return data;
  857. },
  858. _convertYcckToRgb: function convertYcckToRgb(data) {
  859. var Y, Cb, Cr, k;
  860. var offset = 0;
  861. for (var i = 0, length = data.length; i < length; i += 4) {
  862. Y = data[i];
  863. Cb = data[i + 1];
  864. Cr = data[i + 2];
  865. k = data[i + 3];
  866. var r = -122.67195406894 + Cb * (-6.60635669420364e-5 * Cb + 0.000437130475926232 * Cr - 5.4080610064599e-5 * Y + 0.00048449797120281 * k - 0.154362151871126) + Cr * (-0.000957964378445773 * Cr + 0.000817076911346625 * Y - 0.00477271405408747 * k + 1.53380253221734) + Y * (0.000961250184130688 * Y - 0.00266257332283933 * k + 0.48357088451265) + k * (-0.000336197177618394 * k + 0.484791561490776);
  867. var g = 107.268039397724 + Cb * (2.19927104525741e-5 * Cb - 0.000640992018297945 * Cr + 0.000659397001245577 * Y + 0.000426105652938837 * k - 0.176491792462875) + Cr * (-0.000778269941513683 * Cr + 0.00130872261408275 * Y + 0.000770482631801132 * k - 0.151051492775562) + Y * (0.00126935368114843 * Y - 0.00265090189010898 * k + 0.25802910206845) + k * (-0.000318913117588328 * k - 0.213742400323665);
  868. var b = -20.810012546947 + Cb * (-0.000570115196973677 * Cb - 2.63409051004589e-5 * Cr + 0.0020741088115012 * Y - 0.00288260236853442 * k + 0.814272968359295) + Cr * (-1.53496057440975e-5 * Cr - 0.000132689043961446 * Y + 0.000560833691242812 * k - 0.195152027534049) + Y * (0.00174418132927582 * Y - 0.00255243321439347 * k + 0.116935020465145) + k * (-0.000343531996510555 * k + 0.24165260232407);
  869. data[offset++] = clamp0to255(r);
  870. data[offset++] = clamp0to255(g);
  871. data[offset++] = clamp0to255(b);
  872. }
  873. return data;
  874. },
  875. _convertYcckToCmyk: function convertYcckToCmyk(data) {
  876. var Y, Cb, Cr;
  877. for (var i = 0, length = data.length; i < length; i += 4) {
  878. Y = data[i];
  879. Cb = data[i + 1];
  880. Cr = data[i + 2];
  881. data[i] = clamp0to255(434.456 - Y - 1.402 * Cr);
  882. data[i + 1] = clamp0to255(119.541 - Y + 0.344 * Cb + 0.714 * Cr);
  883. data[i + 2] = clamp0to255(481.816 - Y - 1.772 * Cb);
  884. }
  885. return data;
  886. },
  887. _convertCmykToRgb: function convertCmykToRgb(data) {
  888. var c, m, y, k;
  889. var offset = 0;
  890. var min = -255 * 255 * 255;
  891. var scale = 1 / 255 / 255;
  892. for (var i = 0, length = data.length; i < length; i += 4) {
  893. c = data[i];
  894. m = data[i + 1];
  895. y = data[i + 2];
  896. k = data[i + 3];
  897. var r = c * (-4.387332384609988 * c + 54.48615194189176 * m + 18.82290502165302 * y + 212.25662451639585 * k - 72734.4411664936) + m * (1.7149763477362134 * m - 5.6096736904047315 * y - 17.873870861415444 * k - 1401.7366389350734) + y * (-2.5217340131683033 * y - 21.248923337353073 * k + 4465.541406466231) - k * (21.86122147463605 * k + 48317.86113160301);
  898. var g = c * (8.841041422036149 * c + 60.118027045597366 * m + 6.871425592049007 * y + 31.159100130055922 * k - 20220.756542821975) + m * (-15.310361306967817 * m + 17.575251261109482 * y + 131.35250912493976 * k - 48691.05921601825) + y * (4.444339102852739 * y + 9.8632861493405 * k - 6341.191035517494) - k * (20.737325471181034 * k + 47890.15695978492);
  899. var b = c * (0.8842522430003296 * c + 8.078677503112928 * m + 30.89978309703729 * y - 0.23883238689178934 * k - 3616.812083916688) + m * (10.49593273432072 * m + 63.02378494754052 * y + 50.606957656360734 * k - 28620.90484698408) + y * (0.03296041114873217 * y + 115.60384449646641 * k - 49363.43385999684) - k * (22.33816807309886 * k + 45932.16563550634);
  900. data[offset++] = r >= 0 ? 255 : r <= min ? 0 : 255 + r * scale | 0;
  901. data[offset++] = g >= 0 ? 255 : g <= min ? 0 : 255 + g * scale | 0;
  902. data[offset++] = b >= 0 ? 255 : b <= min ? 0 : 255 + b * scale | 0;
  903. }
  904. return data;
  905. },
  906. getData: function getData(width, height, forceRGBoutput) {
  907. if (this.numComponents > 4) {
  908. error('JPEG error: Unsupported color mode');
  909. }
  910. var data = this._getLinearizedBlockData(width, height);
  911. if (this.numComponents === 1 && forceRGBoutput) {
  912. var dataLength = data.length;
  913. var rgbData = new Uint8Array(dataLength * 3);
  914. var offset = 0;
  915. for (var i = 0; i < dataLength; i++) {
  916. var grayColor = data[i];
  917. rgbData[offset++] = grayColor;
  918. rgbData[offset++] = grayColor;
  919. rgbData[offset++] = grayColor;
  920. }
  921. return rgbData;
  922. } else if (this.numComponents === 3 && this._isColorConversionNeeded()) {
  923. return this._convertYccToRgb(data);
  924. } else if (this.numComponents === 4) {
  925. if (this._isColorConversionNeeded()) {
  926. if (forceRGBoutput) {
  927. return this._convertYcckToRgb(data);
  928. }
  929. return this._convertYcckToCmyk(data);
  930. } else if (forceRGBoutput) {
  931. return this._convertCmykToRgb(data);
  932. }
  933. }
  934. return data;
  935. }
  936. };
  937. return JpegImage;
  938. }();
  939. exports.JpegImage = JpegImage;