123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351 |
- function splitOnSpaceHandleQuotesWithEscapes( str, splits = ' \t\n\r' ) {
- const strings = [];
- let quoteType;
- let escape;
- let s = [];
- for ( let i = 0; i < str.length; ++ i ) {
- const c = str[ i ];
- if ( escape ) {
- escape = false;
- s.push( c );
- } else {
- if ( quoteType ) { // we're inside quotes
- if ( c === quoteType ) {
- quoteType = undefined;
- strings.push( s.join( '' ) );
- s = [];
- } else if ( c === '\\' ) {
- escape = true;
- } else {
- s.push( c );
- }
- } else { // we're not in quotes
- if ( splits.indexOf( c ) >= 0 ) {
- if ( s.length ) {
- strings.push( s.join( '' ) );
- s = [];
- }
- } else if ( c === '"' || c === '\'' ) {
- if ( s.length ) { // its in th middle of a word
- s.push( c );
- } else {
- quoteType = c;
- }
- } else {
- s.push( c );
- }
- }
- }
- }
- if ( s.length || strings.length === 0 ) {
- strings.push( s.join( '' ) );
- }
- return strings;
- }
- const startWhitespaceRE = /^\s/;
- const intRE = /^\d+$/;
- const isNum = s => intRE.test( s );
- const quotesRE = /^".*"$/;
- function trimQuotes( s ) {
- return quotesRE.test( s ) ? s.slice( 1, - 1 ) : s;
- }
- const splitToNumbers = s => s.split( ' ' ).map( parseFloat );
- export function parseCSP( str ) {
- const data = [];
- const lut = {
- name: 'unknown',
- type: '1D',
- size: 0,
- data,
- min: [ 0, 0, 0 ],
- max: [ 1, 1, 1 ],
- };
- const lines = str.split( '\n' ).map( s => s.trim() ).filter( s => s.length > 0 && ! startWhitespaceRE.test( s ) );
- // check header
- lut.type = lines[ 1 ];
- if ( lines[ 0 ] !== 'CSPLUTV100' ||
- ( lut.type !== '1D' && lut.type !== '3D' ) ) {
- throw new Error( 'not CSP' );
- }
- // skip meta (read to first number)
- let lineNdx = 2;
- for ( ; lineNdx < lines.length; ++ lineNdx ) {
- const line = lines[ lineNdx ];
- if ( isNum( line ) ) {
- break;
- }
- if ( line.startsWith( 'TITLE ' ) ) {
- lut.name = trimQuotes( line.slice( 6 ).trim() );
- }
- }
- // read ranges
- for ( let i = 0; i < 3; ++ i ) {
- ++ lineNdx;
- const input = splitToNumbers( lines[ lineNdx ++ ] );
- const output = splitToNumbers( lines[ lineNdx ++ ] );
- if ( input.length !== 2 || output.length !== 2 ||
- input[ 0 ] !== 0 || input[ 1 ] !== 1 ||
- output[ 0 ] !== 0 || output[ 1 ] !== 1 ) {
- throw new Error( 'mapped ranges not support' );
- }
- }
- // read sizes
- const sizes = splitToNumbers( lines[ lineNdx ++ ] );
- if ( sizes[ 0 ] !== sizes[ 1 ] || sizes[ 0 ] !== sizes[ 2 ] ) {
- throw new Error( 'only cubic sizes supported' );
- }
- lut.size = sizes[ 0 ];
- // read data
- for ( ; lineNdx < lines.length; ++ lineNdx ) {
- const parts = splitToNumbers( lines[ lineNdx ] );
- if ( parts.length !== 3 ) {
- throw new Error( 'malformed file' );
- }
- data.push( ...parts );
- }
- return lut;
- }
- export function parseCUBE( str ) {
- const data = [];
- const lut = {
- name: 'unknown',
- type: '1D',
- size: 0,
- data,
- min: [ 0, 0, 0 ],
- max: [ 1, 1, 1 ],
- };
- const lines = str.split( '\n' );
- for ( const origLine of lines ) {
- const hashNdx = origLine.indexOf( '#' );
- const line = hashNdx >= 0 ? origLine.substring( 0, hashNdx ) : origLine;
- const parts = splitOnSpaceHandleQuotesWithEscapes( line );
- switch ( parts[ 0 ].toUpperCase() ) {
- case 'TITLE':
- lut.name = parts[ 1 ];
- break;
- case 'LUT_1D_SIZE':
- lut.size = parseInt( parts[ 1 ] );
- lut.type = '1D';
- break;
- case 'LUT_3D_SIZE':
- lut.size = parseInt( parts[ 1 ] );
- lut.type = '3D';
- break;
- case 'DOMAIN_MIN':
- lut.min = parts.slice( 1 ).map( parseFloat );
- break;
- case 'DOMAIN_MAX':
- lut.max = parts.slice( 1 ).map( parseFloat );
- break;
- default:
- if ( parts.length === 3 ) {
- data.push( ...parts.map( parseFloat ) );
- }
- break;
- }
- }
- if ( ! lut.size ) {
- lut.size = lut.type === '1D'
- ? ( data.length / 3 )
- : Math.cbrt( data.length / 3 );
- }
- return lut;
- }
- function lerp( a, b, t ) {
- return a + ( b - a ) * t;
- }
- function lut1Dto3D( lut ) {
- let src = lut.data;
- if ( src.length / 3 !== lut.size ) {
- src = [];
- for ( let i = 0; i < lut.size; ++ i ) {
- const u = i / lut.size * lut.data.length;
- const i0 = ( u | 0 ) * 3;
- const i1 = i0 + 3;
- const t = u % 1;
- src.push(
- lerp( lut.data[ i0 + 0 ], lut.data[ i1 + 0 ], t ),
- lerp( lut.data[ i0 + 0 ], lut.data[ i1 + 1 ], t ),
- lerp( lut.data[ i0 + 0 ], lut.data[ i1 + 2 ], t ),
- );
- }
- }
- const data = [];
- for ( let i = 0; i < lut.size * lut.size; ++ i ) {
- data.push( ...src );
- }
- return { ...lut, data };
- }
- const parsers = {
- 'cube': parseCUBE,
- 'csp': parseCSP,
- };
- // for backward compatibility
- export function parse( str, format = 'cube' ) {
- const parser = parsers[ format.toLowerCase() ];
- if ( ! parser ) {
- throw new Error( `no parser for format: ${format}` );
- }
- return parser( str );
- }
- export function lutTo2D3Drgba8( lut ) {
- if ( lut.type === '1D' ) {
- lut = lut1Dto3D( lut );
- }
- const { min, max, size } = lut;
- const range = min.map( ( min, ndx ) => {
- return max[ ndx ] - min;
- } );
- const src = lut.data;
- const data = new Uint8Array( size * size * size * 4 );
- const srcOffset = ( offX, offY, offZ ) => {
- return ( offX + offY * size + offZ * size * size ) * 3;
- };
- const dOffset = ( offX, offY, offZ ) => {
- return ( offX + offY * size + offZ * size * size ) * 4;
- };
- for ( let dz = 0; dz < size; ++ dz ) {
- for ( let dy = 0; dy < size; ++ dy ) {
- for ( let dx = 0; dx < size; ++ dx ) {
- const sx = dx;
- const sy = dz;
- const sz = dy;
- const sOff = srcOffset( sx, sy, sz );
- const dOff = dOffset( dx, dy, dz );
- data[ dOff + 0 ] = ( src[ sOff + 0 ] - min[ 0 ] ) / range[ 0 ] * 255;
- data[ dOff + 1 ] = ( src[ sOff + 1 ] - min[ 1 ] ) / range[ 1 ] * 255;
- data[ dOff + 2 ] = ( src[ sOff + 2 ] - min[ 2 ] ) / range[ 2 ] * 255;
- data[ dOff + 3 ] = 255;
- }
- }
- }
- return { ...lut, data };
- }
|