提交 5e176f8f 编写于 作者: G Garrett Johnson

Add LUT Loaders

上级 9ab55519
// http://download.autodesk.com/us/systemdocs/help/2011/lustre/index.html?url=./files/WSc4e151a45a3b785a24c3d9a411df9298473-7ffd.htm,topicNumber=d0e9492
import {
Loader,
FileLoader,
Vector3,
DataTexture,
DataTexture3D,
RGBFormat,
UnsignedByteType,
ClampToEdgeWrapping,
LinearFilter,
} from '//unpkg.com/three@0.120.1/build/three.module.js';
export class LUT3dlLoader extends Loader {
load( url, onLoad, onProgress, onError ) {
const loader = new FileLoader( this.manager );
loader.setPath( this.path );
loader.setResponseType( 'text' );
loader.load( url, text => {
try {
onLoad( this.parse( text ) );
} catch ( e ) {
if ( onError ) {
onError( e );
} else {
console.error( e );
}
this.manager.itemError( url );
}
}, onProgress, onError );
}
parse( str ) {
// remove empty lines and comment lints
str = str
.replace( /^#.*?(\n|\r)/gm, '' )
.replace( /^\s*?(\n|\r)/gm, '' )
.trim();
const lines = str.split( /[\n\r]+/g );
// first line is the positions on the grid that are provided by the LUT
const gridLines = lines[ 0 ].trim().split( /\s+/g ).map( e => parseFloat( e ) );
const gridStep = gridLines[ 1 ] - gridLines[ 0 ];
const size = gridLines.length;
for ( let i = 1, l = gridLines.length; i < l; i ++ ) {
if ( gridStep !== ( gridLines[ i ] - gridLines[ i - 1 ] ) ) {
throw new Error( 'LUT3dlLoader: Inconsistent grid size not supported.' );
}
}
const dataArray = new Array( size * size * size * 3 );
let index = 0;
let maxOutputValue = 0.0;
for ( let i = 1, l = lines.length; i < l; i ++ ) {
const line = lines[ i ].trim();
const split = line.split( /\s/g );
const r = parseFloat( split[ 0 ] );
const g = parseFloat( split[ 1 ] );
const b = parseFloat( split[ 2 ] );
maxOutputValue = Math.max( maxOutputValue, r, g, b );
const bLayer = index % size;
const gLayer = Math.floor( index / size ) % size;
const rLayer = Math.floor( index / ( size * size ) ) % size;
// b grows first, then g, then r
const pixelIndex = bLayer * size * size + gLayer * size + rLayer;
dataArray[ 3 * pixelIndex + 0 ] = r;
dataArray[ 3 * pixelIndex + 1 ] = g;
dataArray[ 3 * pixelIndex + 2 ] = b;
index += 1;
}
// Find the apparent bit depth of the stored RGB values and scale the
// values to [ 0, 255 ].
const bits = Math.ceil( Math.log2( maxOutputValue ) );
const maxBitValue = Math.pow( 2.0, bits );
for ( let i = 0, l = dataArray.length; i < l; i ++ ) {
const val = dataArray[ i ];
dataArray[ i ] = 255 * val / maxBitValue;
}
const data = new Uint8Array( dataArray );
const texture = new DataTexture();
texture.image.data = data;
texture.image.width = size;
texture.image.height = size * size;
texture.format = RGBFormat;
texture.type = UnsignedByteType;
texture.magFilter = LinearFilter;
texture.wrapS = ClampToEdgeWrapping;
texture.wrapT = ClampToEdgeWrapping;
texture.generateMipmaps = false;
const texture3D = new DataTexture3D();
texture3D.image.data = data;
texture3D.image.width = size;
texture3D.image.height = size;
texture3D.image.depth = size;
texture3D.format = RGBFormat;
texture3D.type = UnsignedByteType;
texture3D.magFilter = LinearFilter;
texture3D.wrapS = ClampToEdgeWrapping;
texture3D.wrapT = ClampToEdgeWrapping;
texture3D.wrapR = ClampToEdgeWrapping;
texture3D.generateMipmaps = false;
return {
size,
texture,
texture3D,
};
}
}
// https://wwwimages2.adobe.com/content/dam/acom/en/products/speedgrade/cc/pdfs/cube-lut-specification-1.0.pdf
import {
Loader,
FileLoader,
Vector3,
DataTexture,
DataTexture3D,
RGBFormat,
UnsignedByteType,
ClampToEdgeWrapping,
LinearFilter,
} from '//unpkg.com/three@0.120.1/build/three.module.js';
export class LUTCubeLoader extends Loader {
load( url, onLoad, onProgress, onError ) {
const loader = new FileLoader( this.manager );
loader.setPath( this.path );
loader.setResponseType( 'text' );
loader.load( url, text => {
try {
onLoad( this.parse( text ) );
} catch ( e ) {
if ( onError ) {
onError( e );
} else {
console.error( e );
}
this.manager.itemError( url );
}
}, onProgress, onError );
}
parse( str ) {
// Remove empty lines and comments
str = str
.replace( /^#.*?(\n|\r)/gm, '' )
.replace( /^\s*?(\n|\r)/gm, '' )
.trim();
let title = null;
let size = null;
const domainMin = new Vector3( 0, 0, 0 );
const domainMax = new Vector3( 1, 1, 1 );
const lines = str.split( /[\n\r]+/g );
let data = null;
let currIndex = 0;
for ( let i = 0, l = lines.length; i < l; i ++ ) {
const line = lines[ i ].trim();
const split = line.split( /\s/g );
switch ( split[ 0 ] ) {
case 'TITLE':
title = line.substring( 7, line.length - 1 );
break;
case 'LUT_3D_SIZE':
// TODO: A .CUBE LUT file specifies floating point values and could be represented with
// more precision than can be captured with Uint8Array.
const sizeToken = split[ 1 ];
size = parseFloat( sizeToken );
data = new Uint8Array( size * size * size * 3 );
break;
case 'DOMAIN_MIN':
domainMin.x = parseFloat( split[ 1 ] );
domainMin.y = parseFloat( split[ 2 ] );
domainMin.z = parseFloat( split[ 3 ] );
break;
case 'DOMAIN_MAX':
domainMax.x = parseFloat( split[ 1 ] );
domainMax.y = parseFloat( split[ 2 ] );
domainMax.z = parseFloat( split[ 3 ] );
break;
default:
const r = parseFloat( split[ 0 ] );
const g = parseFloat( split[ 1 ] );
const b = parseFloat( split[ 2 ] );
if (
r > 1.0 || r < 0.0 ||
g > 1.0 || g < 0.0 ||
b > 1.0 || b < 0.0
) {
throw new Error( 'LUTCubeLoader : Non normalized values not supported.' );
}
data[ currIndex + 0 ] = r * 255;
data[ currIndex + 1 ] = g * 255;
data[ currIndex + 2 ] = b * 255;
currIndex += 3;
}
}
const texture = new DataTexture();
texture.image.data = data;
texture.image.width = size;
texture.image.height = size * size;
texture.format = RGBFormat;
texture.type = UnsignedByteType;
texture.magFilter = LinearFilter;
texture.wrapS = ClampToEdgeWrapping;
texture.wrapT = ClampToEdgeWrapping;
texture.generateMipmaps = false;
const texture3D = new DataTexture3D();
texture3D.image.data = data;
texture3D.image.width = size;
texture3D.image.height = size;
texture3D.image.depth = size;
texture3D.format = RGBFormat;
texture3D.type = UnsignedByteType;
texture3D.magFilter = LinearFilter;
texture3D.wrapS = ClampToEdgeWrapping;
texture3D.wrapT = ClampToEdgeWrapping;
texture3D.wrapR = ClampToEdgeWrapping;
texture3D.generateMipmaps = false;
return {
title,
size,
domainMin,
domainMax,
texture,
texture3D,
};
}
}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册