lut-reader.js 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242
  1. function splitOnSpaceHandleQuotesWithEscapes(str, splits = ' \t\n\r') {
  2. const strings = [];
  3. let quoteType;
  4. let escape;
  5. let s = [];
  6. for (let i = 0; i < str.length; ++i) {
  7. const c = str[i];
  8. if (escape) {
  9. escape = false;
  10. s.push(c);
  11. } else {
  12. if (quoteType) { // we're inside quotes
  13. if (c === quoteType) {
  14. quoteType = undefined;
  15. strings.push(s.join(''));
  16. s = [];
  17. } else if (c === '\\') {
  18. escape = true;
  19. } else {
  20. s.push(c);
  21. }
  22. } else { // we're not in quotes
  23. if (splits.indexOf(c) >= 0) {
  24. if (s.length) {
  25. strings.push(s.join(''));
  26. s = [];
  27. }
  28. } else if (c === '"' || c === '\'') {
  29. if (s.length) { // its in th middle of a word
  30. s.push(c);
  31. } else {
  32. quoteType = c;
  33. }
  34. } else {
  35. s.push(c);
  36. }
  37. }
  38. }
  39. }
  40. if (s.length || strings.length === 0) {
  41. strings.push(s.join(''));
  42. }
  43. return strings;
  44. }
  45. const startWhitespaceRE = /^\s/;
  46. const intRE = /^\d+$/;
  47. const isNum = s => intRE.test(s);
  48. const quotesRE = /^".*"$/;
  49. function trimQuotes(s) {
  50. return quotesRE.test(s) ? s.substr(s, s.length - 2) : s;
  51. }
  52. const splitToNumbers = s => s.split(' ').map(parseFloat);
  53. export function parseCSP(str) {
  54. const data = [];
  55. const lut = {
  56. name: 'unknown',
  57. type: '1D',
  58. size: 0,
  59. data,
  60. min: [0, 0, 0],
  61. max: [1, 1, 1],
  62. };
  63. const lines = str.split('\n').map(s => s.trim()).filter(s => s.length > 0 && !startWhitespaceRE.test(s));
  64. // check header
  65. lut.type = lines[1];
  66. if (lines[0] !== 'CSPLUTV100' ||
  67. (lut.type !== '1D' && lut.type !== '3D')) {
  68. throw new Error('not CSP');
  69. }
  70. // skip meta (read to first number)
  71. let lineNdx = 2;
  72. for (; lineNdx < lines.length; ++lineNdx) {
  73. const line = lines[lineNdx];
  74. if (isNum(line)) {
  75. break;
  76. }
  77. if (line.startsWith('TITLE ')) {
  78. lut.name = trimQuotes(line.substr(6).trim());
  79. }
  80. }
  81. // read ranges
  82. const ranges = [];
  83. for (let i = 0; i < 3; ++i) {
  84. ++lineNdx;
  85. const input = splitToNumbers(lines[lineNdx++]);
  86. const output = splitToNumbers(lines[lineNdx++]);
  87. ranges.push({input, output});
  88. if (input.length !== 2 || output.length !== 2 ||
  89. input[0] !== 0 || input[1] !== 1 ||
  90. output[0] !== 0 || output[1] !== 1) {
  91. throw new Error('mapped ranges not support');
  92. }
  93. }
  94. // read sizes
  95. const sizes = splitToNumbers(lines[lineNdx++]);
  96. if (sizes[0] !== sizes[1] || sizes[0] !== sizes[2]) {
  97. throw new Error('only cubic sizes supported');
  98. }
  99. lut.size = sizes[0];
  100. // read data
  101. for (; lineNdx < lines.length; ++lineNdx) {
  102. const parts = splitToNumbers(lines[lineNdx]);
  103. if (parts.length !== 3) {
  104. throw new Error('malformed file');
  105. }
  106. data.push(...parts);
  107. }
  108. return lut;
  109. }
  110. export function parseCUBE(str) {
  111. const data = [];
  112. const lut = {
  113. name: 'unknown',
  114. type: '1D',
  115. size: 0,
  116. data,
  117. min: [0, 0, 0],
  118. max: [1, 1, 1],
  119. };
  120. const lines = str.split('\n');
  121. for (const origLine of lines) {
  122. const hashNdx = origLine.indexOf('#');
  123. const line = hashNdx >= 0 ? origLine.substring(0, hashNdx) : origLine;
  124. const parts = splitOnSpaceHandleQuotesWithEscapes(line);
  125. switch (parts[0].toUpperCase()) {
  126. case 'TITLE':
  127. lut.name = parts[1];
  128. break;
  129. case 'LUT_1D_SIZE':
  130. lut.size = parseInt(parts[1]);
  131. lut.type = '1D';
  132. break;
  133. case 'LUT_3D_SIZE':
  134. lut.size = parseInt(parts[1]);
  135. lut.type = '3D';
  136. break;
  137. case 'DOMAIN_MIN':
  138. lut.min = parts.slice(1).map(parseFloat);
  139. break;
  140. case 'DOMAIN_MAX':
  141. lut.max = parts.slice(1).map(parseFloat);
  142. break;
  143. default:
  144. if (parts.length === 3) {
  145. data.push(...parts.map(parseFloat));
  146. }
  147. break;
  148. }
  149. }
  150. if (!lut.size) {
  151. lut.size = lut.type === '1D'
  152. ? (data.length / 3)
  153. : Math.cbrt(data.length / 3);
  154. }
  155. return lut;
  156. }
  157. function lerp(a, b, t) {
  158. return a + (b - a) * t;
  159. }
  160. function lut1Dto3D(lut) {
  161. let src = lut.data;
  162. if (src.length / 3 !== lut.size) {
  163. src = [];
  164. for (let i = 0; i < lut.size; ++i) {
  165. const u = i / lut.size * lut.data.length;
  166. const i0 = (u | 0) * 3;
  167. const i1 = i0 + 3;
  168. const t = u % 1;
  169. src.push(
  170. lerp(lut.data[i0 + 0], lut.data[i1 + 0], t),
  171. lerp(lut.data[i0 + 0], lut.data[i1 + 1], t),
  172. lerp(lut.data[i0 + 0], lut.data[i1 + 2], t),
  173. );
  174. }
  175. }
  176. const data = [];
  177. for (let i = 0; i < lut.size * lut.size; ++i) {
  178. data.push(...src);
  179. }
  180. return {...lut, data};
  181. }
  182. const parsers = {
  183. 'cube': parseCUBE,
  184. 'csp': parseCSP,
  185. };
  186. // for backward compatibility
  187. export function parse(str, format = 'cube') {
  188. const parser = parsers[format.toLowerCase()];
  189. if (!parser) {
  190. throw new Error(`no parser for format: ${format}`);
  191. }
  192. return parser(str);
  193. }
  194. export function lutTo2D3Drgb8(lut) {
  195. if (lut.type === '1D') {
  196. lut = lut1Dto3D(lut);
  197. }
  198. const {min, max, size} = lut;
  199. const range = min.map((min, ndx) => {
  200. return max[ndx] - min;
  201. });
  202. const src = lut.data;
  203. const data = new Uint8Array(src.length);
  204. const offset = (offX, offY, offZ) => {
  205. return (offX + offY * size + offZ * size * size) * 3;
  206. };
  207. for (let dz = 0; dz < size; ++dz) {
  208. for (let dy = 0; dy < size; ++dy) {
  209. for (let dx = 0; dx < size; ++dx) {
  210. const sx = dx;
  211. const sy = dz;
  212. const sz = dy;
  213. const sOff = offset(sx, sy, sz);
  214. const dOff = offset(dx, dy, dz);
  215. data[dOff + 0] = (src[sOff + 0] - min[0]) / range[0] * 255;
  216. data[dOff + 1] = (src[sOff + 1] - min[1]) / range[1] * 255;
  217. data[dOff + 2] = (src[sOff + 2] - min[2]) / range[2] * 255;
  218. }
  219. }
  220. }
  221. return {...lut, data};
  222. }