lut-reader.js 5.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351
  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.slice( 1, - 1 ) : 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.slice( 6 ).trim() );
  79. }
  80. }
  81. // read ranges
  82. for ( let i = 0; i < 3; ++ i ) {
  83. ++ lineNdx;
  84. const input = splitToNumbers( lines[ lineNdx ++ ] );
  85. const output = splitToNumbers( lines[ lineNdx ++ ] );
  86. if ( input.length !== 2 || output.length !== 2 ||
  87. input[ 0 ] !== 0 || input[ 1 ] !== 1 ||
  88. output[ 0 ] !== 0 || output[ 1 ] !== 1 ) {
  89. throw new Error( 'mapped ranges not support' );
  90. }
  91. }
  92. // read sizes
  93. const sizes = splitToNumbers( lines[ lineNdx ++ ] );
  94. if ( sizes[ 0 ] !== sizes[ 1 ] || sizes[ 0 ] !== sizes[ 2 ] ) {
  95. throw new Error( 'only cubic sizes supported' );
  96. }
  97. lut.size = sizes[ 0 ];
  98. // read data
  99. for ( ; lineNdx < lines.length; ++ lineNdx ) {
  100. const parts = splitToNumbers( lines[ lineNdx ] );
  101. if ( parts.length !== 3 ) {
  102. throw new Error( 'malformed file' );
  103. }
  104. data.push( ...parts );
  105. }
  106. return lut;
  107. }
  108. export function parseCUBE( str ) {
  109. const data = [];
  110. const lut = {
  111. name: 'unknown',
  112. type: '1D',
  113. size: 0,
  114. data,
  115. min: [ 0, 0, 0 ],
  116. max: [ 1, 1, 1 ],
  117. };
  118. const lines = str.split( '\n' );
  119. for ( const origLine of lines ) {
  120. const hashNdx = origLine.indexOf( '#' );
  121. const line = hashNdx >= 0 ? origLine.substring( 0, hashNdx ) : origLine;
  122. const parts = splitOnSpaceHandleQuotesWithEscapes( line );
  123. switch ( parts[ 0 ].toUpperCase() ) {
  124. case 'TITLE':
  125. lut.name = parts[ 1 ];
  126. break;
  127. case 'LUT_1D_SIZE':
  128. lut.size = parseInt( parts[ 1 ] );
  129. lut.type = '1D';
  130. break;
  131. case 'LUT_3D_SIZE':
  132. lut.size = parseInt( parts[ 1 ] );
  133. lut.type = '3D';
  134. break;
  135. case 'DOMAIN_MIN':
  136. lut.min = parts.slice( 1 ).map( parseFloat );
  137. break;
  138. case 'DOMAIN_MAX':
  139. lut.max = parts.slice( 1 ).map( parseFloat );
  140. break;
  141. default:
  142. if ( parts.length === 3 ) {
  143. data.push( ...parts.map( parseFloat ) );
  144. }
  145. break;
  146. }
  147. }
  148. if ( ! lut.size ) {
  149. lut.size = lut.type === '1D'
  150. ? ( data.length / 3 )
  151. : Math.cbrt( data.length / 3 );
  152. }
  153. return lut;
  154. }
  155. function lerp( a, b, t ) {
  156. return a + ( b - a ) * t;
  157. }
  158. function lut1Dto3D( lut ) {
  159. let src = lut.data;
  160. if ( src.length / 3 !== lut.size ) {
  161. src = [];
  162. for ( let i = 0; i < lut.size; ++ i ) {
  163. const u = i / lut.size * lut.data.length;
  164. const i0 = ( u | 0 ) * 3;
  165. const i1 = i0 + 3;
  166. const t = u % 1;
  167. src.push(
  168. lerp( lut.data[ i0 + 0 ], lut.data[ i1 + 0 ], t ),
  169. lerp( lut.data[ i0 + 0 ], lut.data[ i1 + 1 ], t ),
  170. lerp( lut.data[ i0 + 0 ], lut.data[ i1 + 2 ], t ),
  171. );
  172. }
  173. }
  174. const data = [];
  175. for ( let i = 0; i < lut.size * lut.size; ++ i ) {
  176. data.push( ...src );
  177. }
  178. return { ...lut, data };
  179. }
  180. const parsers = {
  181. 'cube': parseCUBE,
  182. 'csp': parseCSP,
  183. };
  184. // for backward compatibility
  185. export function parse( str, format = 'cube' ) {
  186. const parser = parsers[ format.toLowerCase() ];
  187. if ( ! parser ) {
  188. throw new Error( `no parser for format: ${format}` );
  189. }
  190. return parser( str );
  191. }
  192. export function lutTo2D3Drgba8( lut ) {
  193. if ( lut.type === '1D' ) {
  194. lut = lut1Dto3D( lut );
  195. }
  196. const { min, max, size } = lut;
  197. const range = min.map( ( min, ndx ) => {
  198. return max[ ndx ] - min;
  199. } );
  200. const src = lut.data;
  201. const data = new Uint8Array( size * size * size * 4 );
  202. const srcOffset = ( offX, offY, offZ ) => {
  203. return ( offX + offY * size + offZ * size * size ) * 3;
  204. };
  205. const dOffset = ( offX, offY, offZ ) => {
  206. return ( offX + offY * size + offZ * size * size ) * 4;
  207. };
  208. for ( let dz = 0; dz < size; ++ dz ) {
  209. for ( let dy = 0; dy < size; ++ dy ) {
  210. for ( let dx = 0; dx < size; ++ dx ) {
  211. const sx = dx;
  212. const sy = dz;
  213. const sz = dy;
  214. const sOff = srcOffset( sx, sy, sz );
  215. const dOff = dOffset( dx, dy, dz );
  216. data[ dOff + 0 ] = ( src[ sOff + 0 ] - min[ 0 ] ) / range[ 0 ] * 255;
  217. data[ dOff + 1 ] = ( src[ sOff + 1 ] - min[ 1 ] ) / range[ 1 ] * 255;
  218. data[ dOff + 2 ] = ( src[ sOff + 2 ] - min[ 2 ] ) / range[ 2 ] * 255;
  219. data[ dOff + 3 ] = 255;
  220. }
  221. }
  222. }
  223. return { ...lut, data };
  224. }