jpg.js 35 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258
  1. /**
  2. * @licstart The following is the entire license notice for the
  3. * Javascript code in this page
  4. *
  5. * Copyright 2021 Mozilla Foundation
  6. *
  7. * Licensed under the Apache License, Version 2.0 (the "License");
  8. * you may not use this file except in compliance with the License.
  9. * You may obtain a copy of the License at
  10. *
  11. * http://www.apache.org/licenses/LICENSE-2.0
  12. *
  13. * Unless required by applicable law or agreed to in writing, software
  14. * distributed under the License is distributed on an "AS IS" BASIS,
  15. * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16. * See the License for the specific language governing permissions and
  17. * limitations under the License.
  18. *
  19. * @licend The above is the entire license notice for the
  20. * Javascript code in this page
  21. */
  22. "use strict";
  23. Object.defineProperty(exports, "__esModule", {
  24. value: true
  25. });
  26. exports.JpegImage = void 0;
  27. var _util = require("../shared/util.js");
  28. var _core_utils = require("./core_utils.js");
  29. class JpegError extends _util.BaseException {
  30. constructor(msg) {
  31. super(`JPEG error: ${msg}`);
  32. }
  33. }
  34. class DNLMarkerError extends _util.BaseException {
  35. constructor(message, scanLines) {
  36. super(message);
  37. this.scanLines = scanLines;
  38. }
  39. }
  40. class EOIMarkerError extends _util.BaseException {}
  41. const dctZigZag = new Uint8Array([0, 1, 8, 16, 9, 2, 3, 10, 17, 24, 32, 25, 18, 11, 4, 5, 12, 19, 26, 33, 40, 48, 41, 34, 27, 20, 13, 6, 7, 14, 21, 28, 35, 42, 49, 56, 57, 50, 43, 36, 29, 22, 15, 23, 30, 37, 44, 51, 58, 59, 52, 45, 38, 31, 39, 46, 53, 60, 61, 54, 47, 55, 62, 63]);
  42. const dctCos1 = 4017;
  43. const dctSin1 = 799;
  44. const dctCos3 = 3406;
  45. const dctSin3 = 2276;
  46. const dctCos6 = 1567;
  47. const dctSin6 = 3784;
  48. const dctSqrt2 = 5793;
  49. const dctSqrt1d2 = 2896;
  50. function buildHuffmanTable(codeLengths, values) {
  51. let k = 0,
  52. i,
  53. j,
  54. length = 16;
  55. while (length > 0 && !codeLengths[length - 1]) {
  56. length--;
  57. }
  58. const code = [{
  59. children: [],
  60. index: 0
  61. }];
  62. let p = code[0],
  63. q;
  64. for (i = 0; i < length; i++) {
  65. for (j = 0; j < codeLengths[i]; j++) {
  66. p = code.pop();
  67. p.children[p.index] = values[k];
  68. while (p.index > 0) {
  69. p = code.pop();
  70. }
  71. p.index++;
  72. code.push(p);
  73. while (code.length <= i) {
  74. code.push(q = {
  75. children: [],
  76. index: 0
  77. });
  78. p.children[p.index] = q.children;
  79. p = q;
  80. }
  81. k++;
  82. }
  83. if (i + 1 < length) {
  84. code.push(q = {
  85. children: [],
  86. index: 0
  87. });
  88. p.children[p.index] = q.children;
  89. p = q;
  90. }
  91. }
  92. return code[0].children;
  93. }
  94. function getBlockBufferOffset(component, row, col) {
  95. return 64 * ((component.blocksPerLine + 1) * row + col);
  96. }
  97. function decodeScan(data, offset, frame, components, resetInterval, spectralStart, spectralEnd, successivePrev, successive, parseDNLMarker = false) {
  98. const mcusPerLine = frame.mcusPerLine;
  99. const progressive = frame.progressive;
  100. const startOffset = offset;
  101. let bitsData = 0,
  102. bitsCount = 0;
  103. function readBit() {
  104. if (bitsCount > 0) {
  105. bitsCount--;
  106. return bitsData >> bitsCount & 1;
  107. }
  108. bitsData = data[offset++];
  109. if (bitsData === 0xff) {
  110. const nextByte = data[offset++];
  111. if (nextByte) {
  112. if (nextByte === 0xdc && parseDNLMarker) {
  113. offset += 2;
  114. const scanLines = (0, _core_utils.readUint16)(data, offset);
  115. offset += 2;
  116. if (scanLines > 0 && scanLines !== frame.scanLines) {
  117. throw new DNLMarkerError("Found DNL marker (0xFFDC) while parsing scan data", scanLines);
  118. }
  119. } else if (nextByte === 0xd9) {
  120. if (parseDNLMarker) {
  121. const maybeScanLines = blockRow * (frame.precision === 8 ? 8 : 0);
  122. if (maybeScanLines > 0 && Math.round(frame.scanLines / maybeScanLines) >= 10) {
  123. throw new DNLMarkerError("Found EOI marker (0xFFD9) while parsing scan data, " + "possibly caused by incorrect `scanLines` parameter", maybeScanLines);
  124. }
  125. }
  126. throw new EOIMarkerError("Found EOI marker (0xFFD9) while parsing scan data");
  127. }
  128. throw new JpegError(`unexpected marker ${(bitsData << 8 | nextByte).toString(16)}`);
  129. }
  130. }
  131. bitsCount = 7;
  132. return bitsData >>> 7;
  133. }
  134. function decodeHuffman(tree) {
  135. let node = tree;
  136. while (true) {
  137. node = node[readBit()];
  138. switch (typeof node) {
  139. case "number":
  140. return node;
  141. case "object":
  142. continue;
  143. }
  144. throw new JpegError("invalid huffman sequence");
  145. }
  146. }
  147. function receive(length) {
  148. let n = 0;
  149. while (length > 0) {
  150. n = n << 1 | readBit();
  151. length--;
  152. }
  153. return n;
  154. }
  155. function receiveAndExtend(length) {
  156. if (length === 1) {
  157. return readBit() === 1 ? 1 : -1;
  158. }
  159. const n = receive(length);
  160. if (n >= 1 << length - 1) {
  161. return n;
  162. }
  163. return n + (-1 << length) + 1;
  164. }
  165. function decodeBaseline(component, blockOffset) {
  166. const t = decodeHuffman(component.huffmanTableDC);
  167. const diff = t === 0 ? 0 : receiveAndExtend(t);
  168. component.blockData[blockOffset] = component.pred += diff;
  169. let k = 1;
  170. while (k < 64) {
  171. const rs = decodeHuffman(component.huffmanTableAC);
  172. const s = rs & 15,
  173. r = rs >> 4;
  174. if (s === 0) {
  175. if (r < 15) {
  176. break;
  177. }
  178. k += 16;
  179. continue;
  180. }
  181. k += r;
  182. const z = dctZigZag[k];
  183. component.blockData[blockOffset + z] = receiveAndExtend(s);
  184. k++;
  185. }
  186. }
  187. function decodeDCFirst(component, blockOffset) {
  188. const t = decodeHuffman(component.huffmanTableDC);
  189. const diff = t === 0 ? 0 : receiveAndExtend(t) << successive;
  190. component.blockData[blockOffset] = component.pred += diff;
  191. }
  192. function decodeDCSuccessive(component, blockOffset) {
  193. component.blockData[blockOffset] |= readBit() << successive;
  194. }
  195. let eobrun = 0;
  196. function decodeACFirst(component, blockOffset) {
  197. if (eobrun > 0) {
  198. eobrun--;
  199. return;
  200. }
  201. let k = spectralStart;
  202. const e = spectralEnd;
  203. while (k <= e) {
  204. const rs = decodeHuffman(component.huffmanTableAC);
  205. const s = rs & 15,
  206. r = rs >> 4;
  207. if (s === 0) {
  208. if (r < 15) {
  209. eobrun = receive(r) + (1 << r) - 1;
  210. break;
  211. }
  212. k += 16;
  213. continue;
  214. }
  215. k += r;
  216. const z = dctZigZag[k];
  217. component.blockData[blockOffset + z] = receiveAndExtend(s) * (1 << successive);
  218. k++;
  219. }
  220. }
  221. let successiveACState = 0,
  222. successiveACNextValue;
  223. function decodeACSuccessive(component, blockOffset) {
  224. let k = spectralStart;
  225. const e = spectralEnd;
  226. let r = 0;
  227. let s;
  228. let rs;
  229. while (k <= e) {
  230. const offsetZ = blockOffset + dctZigZag[k];
  231. const sign = component.blockData[offsetZ] < 0 ? -1 : 1;
  232. switch (successiveACState) {
  233. case 0:
  234. rs = decodeHuffman(component.huffmanTableAC);
  235. s = rs & 15;
  236. r = rs >> 4;
  237. if (s === 0) {
  238. if (r < 15) {
  239. eobrun = receive(r) + (1 << r);
  240. successiveACState = 4;
  241. } else {
  242. r = 16;
  243. successiveACState = 1;
  244. }
  245. } else {
  246. if (s !== 1) {
  247. throw new JpegError("invalid ACn encoding");
  248. }
  249. successiveACNextValue = receiveAndExtend(s);
  250. successiveACState = r ? 2 : 3;
  251. }
  252. continue;
  253. case 1:
  254. case 2:
  255. if (component.blockData[offsetZ]) {
  256. component.blockData[offsetZ] += sign * (readBit() << successive);
  257. } else {
  258. r--;
  259. if (r === 0) {
  260. successiveACState = successiveACState === 2 ? 3 : 0;
  261. }
  262. }
  263. break;
  264. case 3:
  265. if (component.blockData[offsetZ]) {
  266. component.blockData[offsetZ] += sign * (readBit() << successive);
  267. } else {
  268. component.blockData[offsetZ] = successiveACNextValue << successive;
  269. successiveACState = 0;
  270. }
  271. break;
  272. case 4:
  273. if (component.blockData[offsetZ]) {
  274. component.blockData[offsetZ] += sign * (readBit() << successive);
  275. }
  276. break;
  277. }
  278. k++;
  279. }
  280. if (successiveACState === 4) {
  281. eobrun--;
  282. if (eobrun === 0) {
  283. successiveACState = 0;
  284. }
  285. }
  286. }
  287. let blockRow = 0;
  288. function decodeMcu(component, decode, mcu, row, col) {
  289. const mcuRow = mcu / mcusPerLine | 0;
  290. const mcuCol = mcu % mcusPerLine;
  291. blockRow = mcuRow * component.v + row;
  292. const blockCol = mcuCol * component.h + col;
  293. const blockOffset = getBlockBufferOffset(component, blockRow, blockCol);
  294. decode(component, blockOffset);
  295. }
  296. function decodeBlock(component, decode, mcu) {
  297. blockRow = mcu / component.blocksPerLine | 0;
  298. const blockCol = mcu % component.blocksPerLine;
  299. const blockOffset = getBlockBufferOffset(component, blockRow, blockCol);
  300. decode(component, blockOffset);
  301. }
  302. const componentsLength = components.length;
  303. let component, i, j, k, n;
  304. let decodeFn;
  305. if (progressive) {
  306. if (spectralStart === 0) {
  307. decodeFn = successivePrev === 0 ? decodeDCFirst : decodeDCSuccessive;
  308. } else {
  309. decodeFn = successivePrev === 0 ? decodeACFirst : decodeACSuccessive;
  310. }
  311. } else {
  312. decodeFn = decodeBaseline;
  313. }
  314. let mcu = 0,
  315. fileMarker;
  316. let mcuExpected;
  317. if (componentsLength === 1) {
  318. mcuExpected = components[0].blocksPerLine * components[0].blocksPerColumn;
  319. } else {
  320. mcuExpected = mcusPerLine * frame.mcusPerColumn;
  321. }
  322. let h, v;
  323. while (mcu <= mcuExpected) {
  324. const mcuToRead = resetInterval ? Math.min(mcuExpected - mcu, resetInterval) : mcuExpected;
  325. if (mcuToRead > 0) {
  326. for (i = 0; i < componentsLength; i++) {
  327. components[i].pred = 0;
  328. }
  329. eobrun = 0;
  330. if (componentsLength === 1) {
  331. component = components[0];
  332. for (n = 0; n < mcuToRead; n++) {
  333. decodeBlock(component, decodeFn, mcu);
  334. mcu++;
  335. }
  336. } else {
  337. for (n = 0; n < mcuToRead; n++) {
  338. for (i = 0; i < componentsLength; i++) {
  339. component = components[i];
  340. h = component.h;
  341. v = component.v;
  342. for (j = 0; j < v; j++) {
  343. for (k = 0; k < h; k++) {
  344. decodeMcu(component, decodeFn, mcu, j, k);
  345. }
  346. }
  347. }
  348. mcu++;
  349. }
  350. }
  351. }
  352. bitsCount = 0;
  353. fileMarker = findNextFileMarker(data, offset);
  354. if (!fileMarker) {
  355. break;
  356. }
  357. if (fileMarker.invalid) {
  358. const partialMsg = mcuToRead > 0 ? "unexpected" : "excessive";
  359. (0, _util.warn)(`decodeScan - ${partialMsg} MCU data, current marker is: ${fileMarker.invalid}`);
  360. offset = fileMarker.offset;
  361. }
  362. if (fileMarker.marker >= 0xffd0 && fileMarker.marker <= 0xffd7) {
  363. offset += 2;
  364. } else {
  365. break;
  366. }
  367. }
  368. return offset - startOffset;
  369. }
  370. function quantizeAndInverse(component, blockBufferOffset, p) {
  371. const qt = component.quantizationTable,
  372. blockData = component.blockData;
  373. let v0, v1, v2, v3, v4, v5, v6, v7;
  374. let p0, p1, p2, p3, p4, p5, p6, p7;
  375. let t;
  376. if (!qt) {
  377. throw new JpegError("missing required Quantization Table.");
  378. }
  379. for (let row = 0; row < 64; row += 8) {
  380. p0 = blockData[blockBufferOffset + row];
  381. p1 = blockData[blockBufferOffset + row + 1];
  382. p2 = blockData[blockBufferOffset + row + 2];
  383. p3 = blockData[blockBufferOffset + row + 3];
  384. p4 = blockData[blockBufferOffset + row + 4];
  385. p5 = blockData[blockBufferOffset + row + 5];
  386. p6 = blockData[blockBufferOffset + row + 6];
  387. p7 = blockData[blockBufferOffset + row + 7];
  388. p0 *= qt[row];
  389. if ((p1 | p2 | p3 | p4 | p5 | p6 | p7) === 0) {
  390. t = dctSqrt2 * p0 + 512 >> 10;
  391. p[row] = t;
  392. p[row + 1] = t;
  393. p[row + 2] = t;
  394. p[row + 3] = t;
  395. p[row + 4] = t;
  396. p[row + 5] = t;
  397. p[row + 6] = t;
  398. p[row + 7] = t;
  399. continue;
  400. }
  401. p1 *= qt[row + 1];
  402. p2 *= qt[row + 2];
  403. p3 *= qt[row + 3];
  404. p4 *= qt[row + 4];
  405. p5 *= qt[row + 5];
  406. p6 *= qt[row + 6];
  407. p7 *= qt[row + 7];
  408. v0 = dctSqrt2 * p0 + 128 >> 8;
  409. v1 = dctSqrt2 * p4 + 128 >> 8;
  410. v2 = p2;
  411. v3 = p6;
  412. v4 = dctSqrt1d2 * (p1 - p7) + 128 >> 8;
  413. v7 = dctSqrt1d2 * (p1 + p7) + 128 >> 8;
  414. v5 = p3 << 4;
  415. v6 = p5 << 4;
  416. v0 = v0 + v1 + 1 >> 1;
  417. v1 = v0 - v1;
  418. t = v2 * dctSin6 + v3 * dctCos6 + 128 >> 8;
  419. v2 = v2 * dctCos6 - v3 * dctSin6 + 128 >> 8;
  420. v3 = t;
  421. v4 = v4 + v6 + 1 >> 1;
  422. v6 = v4 - v6;
  423. v7 = v7 + v5 + 1 >> 1;
  424. v5 = v7 - v5;
  425. v0 = v0 + v3 + 1 >> 1;
  426. v3 = v0 - v3;
  427. v1 = v1 + v2 + 1 >> 1;
  428. v2 = v1 - v2;
  429. t = v4 * dctSin3 + v7 * dctCos3 + 2048 >> 12;
  430. v4 = v4 * dctCos3 - v7 * dctSin3 + 2048 >> 12;
  431. v7 = t;
  432. t = v5 * dctSin1 + v6 * dctCos1 + 2048 >> 12;
  433. v5 = v5 * dctCos1 - v6 * dctSin1 + 2048 >> 12;
  434. v6 = t;
  435. p[row] = v0 + v7;
  436. p[row + 7] = v0 - v7;
  437. p[row + 1] = v1 + v6;
  438. p[row + 6] = v1 - v6;
  439. p[row + 2] = v2 + v5;
  440. p[row + 5] = v2 - v5;
  441. p[row + 3] = v3 + v4;
  442. p[row + 4] = v3 - v4;
  443. }
  444. for (let col = 0; col < 8; ++col) {
  445. p0 = p[col];
  446. p1 = p[col + 8];
  447. p2 = p[col + 16];
  448. p3 = p[col + 24];
  449. p4 = p[col + 32];
  450. p5 = p[col + 40];
  451. p6 = p[col + 48];
  452. p7 = p[col + 56];
  453. if ((p1 | p2 | p3 | p4 | p5 | p6 | p7) === 0) {
  454. t = dctSqrt2 * p0 + 8192 >> 14;
  455. if (t < -2040) {
  456. t = 0;
  457. } else if (t >= 2024) {
  458. t = 255;
  459. } else {
  460. t = t + 2056 >> 4;
  461. }
  462. blockData[blockBufferOffset + col] = t;
  463. blockData[blockBufferOffset + col + 8] = t;
  464. blockData[blockBufferOffset + col + 16] = t;
  465. blockData[blockBufferOffset + col + 24] = t;
  466. blockData[blockBufferOffset + col + 32] = t;
  467. blockData[blockBufferOffset + col + 40] = t;
  468. blockData[blockBufferOffset + col + 48] = t;
  469. blockData[blockBufferOffset + col + 56] = t;
  470. continue;
  471. }
  472. v0 = dctSqrt2 * p0 + 2048 >> 12;
  473. v1 = dctSqrt2 * p4 + 2048 >> 12;
  474. v2 = p2;
  475. v3 = p6;
  476. v4 = dctSqrt1d2 * (p1 - p7) + 2048 >> 12;
  477. v7 = dctSqrt1d2 * (p1 + p7) + 2048 >> 12;
  478. v5 = p3;
  479. v6 = p5;
  480. v0 = (v0 + v1 + 1 >> 1) + 4112;
  481. v1 = v0 - v1;
  482. t = v2 * dctSin6 + v3 * dctCos6 + 2048 >> 12;
  483. v2 = v2 * dctCos6 - v3 * dctSin6 + 2048 >> 12;
  484. v3 = t;
  485. v4 = v4 + v6 + 1 >> 1;
  486. v6 = v4 - v6;
  487. v7 = v7 + v5 + 1 >> 1;
  488. v5 = v7 - v5;
  489. v0 = v0 + v3 + 1 >> 1;
  490. v3 = v0 - v3;
  491. v1 = v1 + v2 + 1 >> 1;
  492. v2 = v1 - v2;
  493. t = v4 * dctSin3 + v7 * dctCos3 + 2048 >> 12;
  494. v4 = v4 * dctCos3 - v7 * dctSin3 + 2048 >> 12;
  495. v7 = t;
  496. t = v5 * dctSin1 + v6 * dctCos1 + 2048 >> 12;
  497. v5 = v5 * dctCos1 - v6 * dctSin1 + 2048 >> 12;
  498. v6 = t;
  499. p0 = v0 + v7;
  500. p7 = v0 - v7;
  501. p1 = v1 + v6;
  502. p6 = v1 - v6;
  503. p2 = v2 + v5;
  504. p5 = v2 - v5;
  505. p3 = v3 + v4;
  506. p4 = v3 - v4;
  507. if (p0 < 16) {
  508. p0 = 0;
  509. } else if (p0 >= 4080) {
  510. p0 = 255;
  511. } else {
  512. p0 >>= 4;
  513. }
  514. if (p1 < 16) {
  515. p1 = 0;
  516. } else if (p1 >= 4080) {
  517. p1 = 255;
  518. } else {
  519. p1 >>= 4;
  520. }
  521. if (p2 < 16) {
  522. p2 = 0;
  523. } else if (p2 >= 4080) {
  524. p2 = 255;
  525. } else {
  526. p2 >>= 4;
  527. }
  528. if (p3 < 16) {
  529. p3 = 0;
  530. } else if (p3 >= 4080) {
  531. p3 = 255;
  532. } else {
  533. p3 >>= 4;
  534. }
  535. if (p4 < 16) {
  536. p4 = 0;
  537. } else if (p4 >= 4080) {
  538. p4 = 255;
  539. } else {
  540. p4 >>= 4;
  541. }
  542. if (p5 < 16) {
  543. p5 = 0;
  544. } else if (p5 >= 4080) {
  545. p5 = 255;
  546. } else {
  547. p5 >>= 4;
  548. }
  549. if (p6 < 16) {
  550. p6 = 0;
  551. } else if (p6 >= 4080) {
  552. p6 = 255;
  553. } else {
  554. p6 >>= 4;
  555. }
  556. if (p7 < 16) {
  557. p7 = 0;
  558. } else if (p7 >= 4080) {
  559. p7 = 255;
  560. } else {
  561. p7 >>= 4;
  562. }
  563. blockData[blockBufferOffset + col] = p0;
  564. blockData[blockBufferOffset + col + 8] = p1;
  565. blockData[blockBufferOffset + col + 16] = p2;
  566. blockData[blockBufferOffset + col + 24] = p3;
  567. blockData[blockBufferOffset + col + 32] = p4;
  568. blockData[blockBufferOffset + col + 40] = p5;
  569. blockData[blockBufferOffset + col + 48] = p6;
  570. blockData[blockBufferOffset + col + 56] = p7;
  571. }
  572. }
  573. function buildComponentData(frame, component) {
  574. const blocksPerLine = component.blocksPerLine;
  575. const blocksPerColumn = component.blocksPerColumn;
  576. const computationBuffer = new Int16Array(64);
  577. for (let blockRow = 0; blockRow < blocksPerColumn; blockRow++) {
  578. for (let blockCol = 0; blockCol < blocksPerLine; blockCol++) {
  579. const offset = getBlockBufferOffset(component, blockRow, blockCol);
  580. quantizeAndInverse(component, offset, computationBuffer);
  581. }
  582. }
  583. return component.blockData;
  584. }
  585. function findNextFileMarker(data, currentPos, startPos = currentPos) {
  586. const maxPos = data.length - 1;
  587. let newPos = startPos < currentPos ? startPos : currentPos;
  588. if (currentPos >= maxPos) {
  589. return null;
  590. }
  591. const currentMarker = (0, _core_utils.readUint16)(data, currentPos);
  592. if (currentMarker >= 0xffc0 && currentMarker <= 0xfffe) {
  593. return {
  594. invalid: null,
  595. marker: currentMarker,
  596. offset: currentPos
  597. };
  598. }
  599. let newMarker = (0, _core_utils.readUint16)(data, newPos);
  600. while (!(newMarker >= 0xffc0 && newMarker <= 0xfffe)) {
  601. if (++newPos >= maxPos) {
  602. return null;
  603. }
  604. newMarker = (0, _core_utils.readUint16)(data, newPos);
  605. }
  606. return {
  607. invalid: currentMarker.toString(16),
  608. marker: newMarker,
  609. offset: newPos
  610. };
  611. }
  612. class JpegImage {
  613. constructor({
  614. decodeTransform = null,
  615. colorTransform = -1
  616. } = {}) {
  617. this._decodeTransform = decodeTransform;
  618. this._colorTransform = colorTransform;
  619. }
  620. parse(data, {
  621. dnlScanLines = null
  622. } = {}) {
  623. function readDataBlock() {
  624. const length = (0, _core_utils.readUint16)(data, offset);
  625. offset += 2;
  626. let endOffset = offset + length - 2;
  627. const fileMarker = findNextFileMarker(data, endOffset, offset);
  628. if (fileMarker && fileMarker.invalid) {
  629. (0, _util.warn)("readDataBlock - incorrect length, current marker is: " + fileMarker.invalid);
  630. endOffset = fileMarker.offset;
  631. }
  632. const array = data.subarray(offset, endOffset);
  633. offset += array.length;
  634. return array;
  635. }
  636. function prepareComponents(frame) {
  637. const mcusPerLine = Math.ceil(frame.samplesPerLine / 8 / frame.maxH);
  638. const mcusPerColumn = Math.ceil(frame.scanLines / 8 / frame.maxV);
  639. for (let i = 0, ii = frame.components.length; i < ii; i++) {
  640. const component = frame.components[i];
  641. const blocksPerLine = Math.ceil(Math.ceil(frame.samplesPerLine / 8) * component.h / frame.maxH);
  642. const blocksPerColumn = Math.ceil(Math.ceil(frame.scanLines / 8) * component.v / frame.maxV);
  643. const blocksPerLineForMcu = mcusPerLine * component.h;
  644. const blocksPerColumnForMcu = mcusPerColumn * component.v;
  645. const blocksBufferSize = 64 * blocksPerColumnForMcu * (blocksPerLineForMcu + 1);
  646. component.blockData = new Int16Array(blocksBufferSize);
  647. component.blocksPerLine = blocksPerLine;
  648. component.blocksPerColumn = blocksPerColumn;
  649. }
  650. frame.mcusPerLine = mcusPerLine;
  651. frame.mcusPerColumn = mcusPerColumn;
  652. }
  653. let offset = 0;
  654. let jfif = null;
  655. let adobe = null;
  656. let frame, resetInterval;
  657. let numSOSMarkers = 0;
  658. const quantizationTables = [];
  659. const huffmanTablesAC = [],
  660. huffmanTablesDC = [];
  661. let fileMarker = (0, _core_utils.readUint16)(data, offset);
  662. offset += 2;
  663. if (fileMarker !== 0xffd8) {
  664. throw new JpegError("SOI not found");
  665. }
  666. fileMarker = (0, _core_utils.readUint16)(data, offset);
  667. offset += 2;
  668. markerLoop: while (fileMarker !== 0xffd9) {
  669. let i, j, l;
  670. switch (fileMarker) {
  671. case 0xffe0:
  672. case 0xffe1:
  673. case 0xffe2:
  674. case 0xffe3:
  675. case 0xffe4:
  676. case 0xffe5:
  677. case 0xffe6:
  678. case 0xffe7:
  679. case 0xffe8:
  680. case 0xffe9:
  681. case 0xffea:
  682. case 0xffeb:
  683. case 0xffec:
  684. case 0xffed:
  685. case 0xffee:
  686. case 0xffef:
  687. case 0xfffe:
  688. const appData = readDataBlock();
  689. if (fileMarker === 0xffe0) {
  690. if (appData[0] === 0x4a && appData[1] === 0x46 && appData[2] === 0x49 && appData[3] === 0x46 && appData[4] === 0) {
  691. jfif = {
  692. version: {
  693. major: appData[5],
  694. minor: appData[6]
  695. },
  696. densityUnits: appData[7],
  697. xDensity: appData[8] << 8 | appData[9],
  698. yDensity: appData[10] << 8 | appData[11],
  699. thumbWidth: appData[12],
  700. thumbHeight: appData[13],
  701. thumbData: appData.subarray(14, 14 + 3 * appData[12] * appData[13])
  702. };
  703. }
  704. }
  705. if (fileMarker === 0xffee) {
  706. if (appData[0] === 0x41 && appData[1] === 0x64 && appData[2] === 0x6f && appData[3] === 0x62 && appData[4] === 0x65) {
  707. adobe = {
  708. version: appData[5] << 8 | appData[6],
  709. flags0: appData[7] << 8 | appData[8],
  710. flags1: appData[9] << 8 | appData[10],
  711. transformCode: appData[11]
  712. };
  713. }
  714. }
  715. break;
  716. case 0xffdb:
  717. const quantizationTablesLength = (0, _core_utils.readUint16)(data, offset);
  718. offset += 2;
  719. const quantizationTablesEnd = quantizationTablesLength + offset - 2;
  720. let z;
  721. while (offset < quantizationTablesEnd) {
  722. const quantizationTableSpec = data[offset++];
  723. const tableData = new Uint16Array(64);
  724. if (quantizationTableSpec >> 4 === 0) {
  725. for (j = 0; j < 64; j++) {
  726. z = dctZigZag[j];
  727. tableData[z] = data[offset++];
  728. }
  729. } else if (quantizationTableSpec >> 4 === 1) {
  730. for (j = 0; j < 64; j++) {
  731. z = dctZigZag[j];
  732. tableData[z] = (0, _core_utils.readUint16)(data, offset);
  733. offset += 2;
  734. }
  735. } else {
  736. throw new JpegError("DQT - invalid table spec");
  737. }
  738. quantizationTables[quantizationTableSpec & 15] = tableData;
  739. }
  740. break;
  741. case 0xffc0:
  742. case 0xffc1:
  743. case 0xffc2:
  744. if (frame) {
  745. throw new JpegError("Only single frame JPEGs supported");
  746. }
  747. offset += 2;
  748. frame = {};
  749. frame.extended = fileMarker === 0xffc1;
  750. frame.progressive = fileMarker === 0xffc2;
  751. frame.precision = data[offset++];
  752. const sofScanLines = (0, _core_utils.readUint16)(data, offset);
  753. offset += 2;
  754. frame.scanLines = dnlScanLines || sofScanLines;
  755. frame.samplesPerLine = (0, _core_utils.readUint16)(data, offset);
  756. offset += 2;
  757. frame.components = [];
  758. frame.componentIds = {};
  759. const componentsCount = data[offset++];
  760. let maxH = 0,
  761. maxV = 0;
  762. for (i = 0; i < componentsCount; i++) {
  763. const componentId = data[offset];
  764. const h = data[offset + 1] >> 4;
  765. const v = data[offset + 1] & 15;
  766. if (maxH < h) {
  767. maxH = h;
  768. }
  769. if (maxV < v) {
  770. maxV = v;
  771. }
  772. const qId = data[offset + 2];
  773. l = frame.components.push({
  774. h,
  775. v,
  776. quantizationId: qId,
  777. quantizationTable: null
  778. });
  779. frame.componentIds[componentId] = l - 1;
  780. offset += 3;
  781. }
  782. frame.maxH = maxH;
  783. frame.maxV = maxV;
  784. prepareComponents(frame);
  785. break;
  786. case 0xffc4:
  787. const huffmanLength = (0, _core_utils.readUint16)(data, offset);
  788. offset += 2;
  789. for (i = 2; i < huffmanLength;) {
  790. const huffmanTableSpec = data[offset++];
  791. const codeLengths = new Uint8Array(16);
  792. let codeLengthSum = 0;
  793. for (j = 0; j < 16; j++, offset++) {
  794. codeLengthSum += codeLengths[j] = data[offset];
  795. }
  796. const huffmanValues = new Uint8Array(codeLengthSum);
  797. for (j = 0; j < codeLengthSum; j++, offset++) {
  798. huffmanValues[j] = data[offset];
  799. }
  800. i += 17 + codeLengthSum;
  801. (huffmanTableSpec >> 4 === 0 ? huffmanTablesDC : huffmanTablesAC)[huffmanTableSpec & 15] = buildHuffmanTable(codeLengths, huffmanValues);
  802. }
  803. break;
  804. case 0xffdd:
  805. offset += 2;
  806. resetInterval = (0, _core_utils.readUint16)(data, offset);
  807. offset += 2;
  808. break;
  809. case 0xffda:
  810. const parseDNLMarker = ++numSOSMarkers === 1 && !dnlScanLines;
  811. offset += 2;
  812. const selectorsCount = data[offset++],
  813. components = [];
  814. for (i = 0; i < selectorsCount; i++) {
  815. const index = data[offset++];
  816. const componentIndex = frame.componentIds[index];
  817. const component = frame.components[componentIndex];
  818. component.index = index;
  819. const tableSpec = data[offset++];
  820. component.huffmanTableDC = huffmanTablesDC[tableSpec >> 4];
  821. component.huffmanTableAC = huffmanTablesAC[tableSpec & 15];
  822. components.push(component);
  823. }
  824. const spectralStart = data[offset++],
  825. spectralEnd = data[offset++],
  826. successiveApproximation = data[offset++];
  827. try {
  828. const processed = decodeScan(data, offset, frame, components, resetInterval, spectralStart, spectralEnd, successiveApproximation >> 4, successiveApproximation & 15, parseDNLMarker);
  829. offset += processed;
  830. } catch (ex) {
  831. if (ex instanceof DNLMarkerError) {
  832. (0, _util.warn)(`${ex.message} -- attempting to re-parse the JPEG image.`);
  833. return this.parse(data, {
  834. dnlScanLines: ex.scanLines
  835. });
  836. } else if (ex instanceof EOIMarkerError) {
  837. (0, _util.warn)(`${ex.message} -- ignoring the rest of the image data.`);
  838. break markerLoop;
  839. }
  840. throw ex;
  841. }
  842. break;
  843. case 0xffdc:
  844. offset += 4;
  845. break;
  846. case 0xffff:
  847. if (data[offset] !== 0xff) {
  848. offset--;
  849. }
  850. break;
  851. default:
  852. const nextFileMarker = findNextFileMarker(data, offset - 2, offset - 3);
  853. if (nextFileMarker && nextFileMarker.invalid) {
  854. (0, _util.warn)("JpegImage.parse - unexpected data, current marker is: " + nextFileMarker.invalid);
  855. offset = nextFileMarker.offset;
  856. break;
  857. }
  858. if (!nextFileMarker || offset >= data.length - 1) {
  859. (0, _util.warn)("JpegImage.parse - reached the end of the image data " + "without finding an EOI marker (0xFFD9).");
  860. break markerLoop;
  861. }
  862. throw new JpegError("JpegImage.parse - unknown marker: " + fileMarker.toString(16));
  863. }
  864. fileMarker = (0, _core_utils.readUint16)(data, offset);
  865. offset += 2;
  866. }
  867. this.width = frame.samplesPerLine;
  868. this.height = frame.scanLines;
  869. this.jfif = jfif;
  870. this.adobe = adobe;
  871. this.components = [];
  872. for (let i = 0, ii = frame.components.length; i < ii; i++) {
  873. const component = frame.components[i];
  874. const quantizationTable = quantizationTables[component.quantizationId];
  875. if (quantizationTable) {
  876. component.quantizationTable = quantizationTable;
  877. }
  878. this.components.push({
  879. index: component.index,
  880. output: buildComponentData(frame, component),
  881. scaleX: component.h / frame.maxH,
  882. scaleY: component.v / frame.maxV,
  883. blocksPerLine: component.blocksPerLine,
  884. blocksPerColumn: component.blocksPerColumn
  885. });
  886. }
  887. this.numComponents = this.components.length;
  888. return undefined;
  889. }
  890. _getLinearizedBlockData(width, height, isSourcePDF = false) {
  891. const scaleX = this.width / width,
  892. scaleY = this.height / height;
  893. let component, componentScaleX, componentScaleY, blocksPerScanline;
  894. let x, y, i, j, k;
  895. let index;
  896. let offset = 0;
  897. let output;
  898. const numComponents = this.components.length;
  899. const dataLength = width * height * numComponents;
  900. const data = new Uint8ClampedArray(dataLength);
  901. const xScaleBlockOffset = new Uint32Array(width);
  902. const mask3LSB = 0xfffffff8;
  903. let lastComponentScaleX;
  904. for (i = 0; i < numComponents; i++) {
  905. component = this.components[i];
  906. componentScaleX = component.scaleX * scaleX;
  907. componentScaleY = component.scaleY * scaleY;
  908. offset = i;
  909. output = component.output;
  910. blocksPerScanline = component.blocksPerLine + 1 << 3;
  911. if (componentScaleX !== lastComponentScaleX) {
  912. for (x = 0; x < width; x++) {
  913. j = 0 | x * componentScaleX;
  914. xScaleBlockOffset[x] = (j & mask3LSB) << 3 | j & 7;
  915. }
  916. lastComponentScaleX = componentScaleX;
  917. }
  918. for (y = 0; y < height; y++) {
  919. j = 0 | y * componentScaleY;
  920. index = blocksPerScanline * (j & mask3LSB) | (j & 7) << 3;
  921. for (x = 0; x < width; x++) {
  922. data[offset] = output[index + xScaleBlockOffset[x]];
  923. offset += numComponents;
  924. }
  925. }
  926. }
  927. let transform = this._decodeTransform;
  928. if (!isSourcePDF && numComponents === 4 && !transform) {
  929. transform = new Int32Array([-256, 255, -256, 255, -256, 255, -256, 255]);
  930. }
  931. if (transform) {
  932. for (i = 0; i < dataLength;) {
  933. for (j = 0, k = 0; j < numComponents; j++, i++, k += 2) {
  934. data[i] = (data[i] * transform[k] >> 8) + transform[k + 1];
  935. }
  936. }
  937. }
  938. return data;
  939. }
  940. get _isColorConversionNeeded() {
  941. if (this.adobe) {
  942. return !!this.adobe.transformCode;
  943. }
  944. if (this.numComponents === 3) {
  945. if (this._colorTransform === 0) {
  946. return false;
  947. } else if (this.components[0].index === 0x52 && this.components[1].index === 0x47 && this.components[2].index === 0x42) {
  948. return false;
  949. }
  950. return true;
  951. }
  952. if (this._colorTransform === 1) {
  953. return true;
  954. }
  955. return false;
  956. }
  957. _convertYccToRgb(data) {
  958. let Y, Cb, Cr;
  959. for (let i = 0, length = data.length; i < length; i += 3) {
  960. Y = data[i];
  961. Cb = data[i + 1];
  962. Cr = data[i + 2];
  963. data[i] = Y - 179.456 + 1.402 * Cr;
  964. data[i + 1] = Y + 135.459 - 0.344 * Cb - 0.714 * Cr;
  965. data[i + 2] = Y - 226.816 + 1.772 * Cb;
  966. }
  967. return data;
  968. }
  969. _convertYcckToRgb(data) {
  970. let Y, Cb, Cr, k;
  971. let offset = 0;
  972. for (let i = 0, length = data.length; i < length; i += 4) {
  973. Y = data[i];
  974. Cb = data[i + 1];
  975. Cr = data[i + 2];
  976. k = data[i + 3];
  977. data[offset++] = -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);
  978. data[offset++] = 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);
  979. data[offset++] = -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);
  980. }
  981. return data.subarray(0, offset);
  982. }
  983. _convertYcckToCmyk(data) {
  984. let Y, Cb, Cr;
  985. for (let i = 0, length = data.length; i < length; i += 4) {
  986. Y = data[i];
  987. Cb = data[i + 1];
  988. Cr = data[i + 2];
  989. data[i] = 434.456 - Y - 1.402 * Cr;
  990. data[i + 1] = 119.541 - Y + 0.344 * Cb + 0.714 * Cr;
  991. data[i + 2] = 481.816 - Y - 1.772 * Cb;
  992. }
  993. return data;
  994. }
  995. _convertCmykToRgb(data) {
  996. let c, m, y, k;
  997. let offset = 0;
  998. for (let i = 0, length = data.length; i < length; i += 4) {
  999. c = data[i];
  1000. m = data[i + 1];
  1001. y = data[i + 2];
  1002. k = data[i + 3];
  1003. data[offset++] = 255 + c * (-0.00006747147073602441 * c + 0.0008379262121013727 * m + 0.0002894718188643294 * y + 0.003264231057537806 * k - 1.1185611867203937) + m * (0.000026374107616089405 * m - 0.00008626949158638572 * y - 0.0002748769067499491 * k - 0.02155688794978967) + y * (-0.00003878099212869363 * y - 0.0003267808279485286 * k + 0.0686742238595345) - k * (0.0003361971776183937 * k + 0.7430659151342254);
  1004. data[offset++] = 255 + c * (0.00013596372813588848 * c + 0.000924537132573585 * m + 0.00010567359618683593 * y + 0.0004791864687436512 * k - 0.3109689587515875) + m * (-0.00023545346108370344 * m + 0.0002702845253534714 * y + 0.0020200308977307156 * k - 0.7488052167015494) + y * (0.00006834815998235662 * y + 0.00015168452363460973 * k - 0.09751927774728933) - k * (0.00031891311758832814 * k + 0.7364883807733168);
  1005. data[offset++] = 255 + c * (0.000013598650411385307 * c + 0.00012423956175490851 * m + 0.0004751985097583589 * y - 0.0000036729317476630422 * k - 0.05562186980264034) + m * (0.00016141380598724676 * m + 0.0009692239130725186 * y + 0.0007782692450036253 * k - 0.44015232367526463) + y * (5.068882914068769e-7 * y + 0.0017778369011375071 * k - 0.7591454649749609) - k * (0.0003435319965105553 * k + 0.7063770186160144);
  1006. }
  1007. return data.subarray(0, offset);
  1008. }
  1009. getData({
  1010. width,
  1011. height,
  1012. forceRGB = false,
  1013. isSourcePDF = false
  1014. }) {
  1015. if (this.numComponents > 4) {
  1016. throw new JpegError("Unsupported color mode");
  1017. }
  1018. const data = this._getLinearizedBlockData(width, height, isSourcePDF);
  1019. if (this.numComponents === 1 && forceRGB) {
  1020. const dataLength = data.length;
  1021. const rgbData = new Uint8ClampedArray(dataLength * 3);
  1022. let offset = 0;
  1023. for (let i = 0; i < dataLength; i++) {
  1024. const grayColor = data[i];
  1025. rgbData[offset++] = grayColor;
  1026. rgbData[offset++] = grayColor;
  1027. rgbData[offset++] = grayColor;
  1028. }
  1029. return rgbData;
  1030. } else if (this.numComponents === 3 && this._isColorConversionNeeded) {
  1031. return this._convertYccToRgb(data);
  1032. } else if (this.numComponents === 4) {
  1033. if (this._isColorConversionNeeded) {
  1034. if (forceRGB) {
  1035. return this._convertYcckToRgb(data);
  1036. }
  1037. return this._convertYcckToCmyk(data);
  1038. } else if (forceRGB) {
  1039. return this._convertCmykToRgb(data);
  1040. }
  1041. }
  1042. return data;
  1043. }
  1044. }
  1045. exports.JpegImage = JpegImage;