earthquake_3d_viewer_front/three/manual/examples/resources/lut-reader.js

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 };

}