LUTCubeLoader.js 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167
  1. // https://wwwimages2.adobe.com/content/dam/acom/en/products/speedgrade/cc/pdfs/cube-lut-specification-1.0.pdf
  2. import {
  3. ClampToEdgeWrapping,
  4. DataTexture,
  5. Data3DTexture,
  6. FileLoader,
  7. FloatType,
  8. LinearFilter,
  9. Loader,
  10. UnsignedByteType,
  11. Vector3,
  12. } from 'three';
  13. export class LUTCubeLoader extends Loader {
  14. constructor( manager ) {
  15. super( manager );
  16. this.type = UnsignedByteType;
  17. }
  18. setType( type ) {
  19. if ( type !== UnsignedByteType && type !== FloatType ) {
  20. throw new Error( 'LUTCubeLoader: Unsupported type' );
  21. }
  22. this.type = type;
  23. return this;
  24. }
  25. load( url, onLoad, onProgress, onError ) {
  26. const loader = new FileLoader( this.manager );
  27. loader.setPath( this.path );
  28. loader.setResponseType( 'text' );
  29. loader.load( url, text => {
  30. try {
  31. onLoad( this.parse( text ) );
  32. } catch ( e ) {
  33. if ( onError ) {
  34. onError( e );
  35. } else {
  36. console.error( e );
  37. }
  38. this.manager.itemError( url );
  39. }
  40. }, onProgress, onError );
  41. }
  42. parse( input ) {
  43. const regExpTitle = /TITLE +"([^"]*)"/;
  44. const regExpSize = /LUT_3D_SIZE +(\d+)/;
  45. const regExpDomainMin = /DOMAIN_MIN +([\d.]+) +([\d.]+) +([\d.]+)/;
  46. const regExpDomainMax = /DOMAIN_MAX +([\d.]+) +([\d.]+) +([\d.]+)/;
  47. const regExpDataPoints = /^([\d.e+-]+) +([\d.e+-]+) +([\d.e+-]+) *$/gm;
  48. let result = regExpTitle.exec( input );
  49. const title = ( result !== null ) ? result[ 1 ] : null;
  50. result = regExpSize.exec( input );
  51. if ( result === null ) {
  52. throw new Error( 'LUTCubeLoader: Missing LUT_3D_SIZE information' );
  53. }
  54. const size = Number( result[ 1 ] );
  55. const length = size ** 3 * 4;
  56. const data = this.type === UnsignedByteType ? new Uint8Array( length ) : new Float32Array( length );
  57. const domainMin = new Vector3( 0, 0, 0 );
  58. const domainMax = new Vector3( 1, 1, 1 );
  59. result = regExpDomainMin.exec( input );
  60. if ( result !== null ) {
  61. domainMin.set( Number( result[ 1 ] ), Number( result[ 2 ] ), Number( result[ 3 ] ) );
  62. }
  63. result = regExpDomainMax.exec( input );
  64. if ( result !== null ) {
  65. domainMax.set( Number( result[ 1 ] ), Number( result[ 2 ] ), Number( result[ 3 ] ) );
  66. }
  67. if ( domainMin.x > domainMax.x || domainMin.y > domainMax.y || domainMin.z > domainMax.z ) {
  68. throw new Error( 'LUTCubeLoader: Invalid input domain' );
  69. }
  70. const scale = this.type === UnsignedByteType ? 255 : 1;
  71. let i = 0;
  72. while ( ( result = regExpDataPoints.exec( input ) ) !== null ) {
  73. data[ i ++ ] = Number( result[ 1 ] ) * scale;
  74. data[ i ++ ] = Number( result[ 2 ] ) * scale;
  75. data[ i ++ ] = Number( result[ 3 ] ) * scale;
  76. data[ i ++ ] = scale;
  77. }
  78. const texture = new DataTexture();
  79. texture.image.data = data;
  80. texture.image.width = size;
  81. texture.image.height = size * size;
  82. texture.type = this.type;
  83. texture.magFilter = LinearFilter;
  84. texture.minFilter = LinearFilter;
  85. texture.wrapS = ClampToEdgeWrapping;
  86. texture.wrapT = ClampToEdgeWrapping;
  87. texture.generateMipmaps = false;
  88. texture.needsUpdate = true;
  89. const texture3D = new Data3DTexture();
  90. texture3D.image.data = data;
  91. texture3D.image.width = size;
  92. texture3D.image.height = size;
  93. texture3D.image.depth = size;
  94. texture3D.type = this.type;
  95. texture3D.magFilter = LinearFilter;
  96. texture3D.minFilter = LinearFilter;
  97. texture3D.wrapS = ClampToEdgeWrapping;
  98. texture3D.wrapT = ClampToEdgeWrapping;
  99. texture3D.wrapR = ClampToEdgeWrapping;
  100. texture3D.generateMipmaps = false;
  101. texture3D.needsUpdate = true;
  102. return {
  103. title,
  104. size,
  105. domainMin,
  106. domainMax,
  107. texture,
  108. texture3D,
  109. };
  110. }
  111. }