Spaces:
Running
Running
| import { | |
| DataTextureLoader, | |
| LinearMipmapLinearFilter | |
| } from 'three'; | |
| class TGALoader extends DataTextureLoader { | |
| constructor( manager ) { | |
| super( manager ); | |
| } | |
| parse( buffer ) { | |
| // reference from vthibault, https://github.com/vthibault/roBrowser/blob/master/src/Loaders/Targa.js | |
| function tgaCheckHeader( header ) { | |
| switch ( header.image_type ) { | |
| // check indexed type | |
| case TGA_TYPE_INDEXED: | |
| case TGA_TYPE_RLE_INDEXED: | |
| if ( header.colormap_length > 256 || header.colormap_size !== 24 || header.colormap_type !== 1 ) { | |
| throw new Error( 'THREE.TGALoader: Invalid type colormap data for indexed type.' ); | |
| } | |
| break; | |
| // check colormap type | |
| case TGA_TYPE_RGB: | |
| case TGA_TYPE_GREY: | |
| case TGA_TYPE_RLE_RGB: | |
| case TGA_TYPE_RLE_GREY: | |
| if ( header.colormap_type ) { | |
| throw new Error( 'THREE.TGALoader: Invalid type colormap data for colormap type.' ); | |
| } | |
| break; | |
| // What the need of a file without data ? | |
| case TGA_TYPE_NO_DATA: | |
| throw new Error( 'THREE.TGALoader: No data.' ); | |
| // Invalid type ? | |
| default: | |
| throw new Error( 'THREE.TGALoader: Invalid type ' + header.image_type ); | |
| } | |
| // check image width and height | |
| if ( header.width <= 0 || header.height <= 0 ) { | |
| throw new Error( 'THREE.TGALoader: Invalid image size.' ); | |
| } | |
| // check image pixel size | |
| if ( header.pixel_size !== 8 && header.pixel_size !== 16 && | |
| header.pixel_size !== 24 && header.pixel_size !== 32 ) { | |
| throw new Error( 'THREE.TGALoader: Invalid pixel size ' + header.pixel_size ); | |
| } | |
| } | |
| // parse tga image buffer | |
| function tgaParse( use_rle, use_pal, header, offset, data ) { | |
| let pixel_data, | |
| palettes; | |
| const pixel_size = header.pixel_size >> 3; | |
| const pixel_total = header.width * header.height * pixel_size; | |
| // read palettes | |
| if ( use_pal ) { | |
| palettes = data.subarray( offset, offset += header.colormap_length * ( header.colormap_size >> 3 ) ); | |
| } | |
| // read RLE | |
| if ( use_rle ) { | |
| pixel_data = new Uint8Array( pixel_total ); | |
| let c, count, i; | |
| let shift = 0; | |
| const pixels = new Uint8Array( pixel_size ); | |
| while ( shift < pixel_total ) { | |
| c = data[ offset ++ ]; | |
| count = ( c & 0x7f ) + 1; | |
| // RLE pixels | |
| if ( c & 0x80 ) { | |
| // bind pixel tmp array | |
| for ( i = 0; i < pixel_size; ++ i ) { | |
| pixels[ i ] = data[ offset ++ ]; | |
| } | |
| // copy pixel array | |
| for ( i = 0; i < count; ++ i ) { | |
| pixel_data.set( pixels, shift + i * pixel_size ); | |
| } | |
| shift += pixel_size * count; | |
| } else { | |
| // raw pixels | |
| count *= pixel_size; | |
| for ( i = 0; i < count; ++ i ) { | |
| pixel_data[ shift + i ] = data[ offset ++ ]; | |
| } | |
| shift += count; | |
| } | |
| } | |
| } else { | |
| // raw pixels | |
| pixel_data = data.subarray( | |
| offset, offset += ( use_pal ? header.width * header.height : pixel_total ) | |
| ); | |
| } | |
| return { | |
| pixel_data: pixel_data, | |
| palettes: palettes | |
| }; | |
| } | |
| function tgaGetImageData8bits( imageData, y_start, y_step, y_end, x_start, x_step, x_end, image, palettes ) { | |
| const colormap = palettes; | |
| let color, i = 0, x, y; | |
| const width = header.width; | |
| for ( y = y_start; y !== y_end; y += y_step ) { | |
| for ( x = x_start; x !== x_end; x += x_step, i ++ ) { | |
| color = image[ i ]; | |
| imageData[ ( x + width * y ) * 4 + 3 ] = 255; | |
| imageData[ ( x + width * y ) * 4 + 2 ] = colormap[ ( color * 3 ) + 0 ]; | |
| imageData[ ( x + width * y ) * 4 + 1 ] = colormap[ ( color * 3 ) + 1 ]; | |
| imageData[ ( x + width * y ) * 4 + 0 ] = colormap[ ( color * 3 ) + 2 ]; | |
| } | |
| } | |
| return imageData; | |
| } | |
| function tgaGetImageData16bits( imageData, y_start, y_step, y_end, x_start, x_step, x_end, image ) { | |
| let color, i = 0, x, y; | |
| const width = header.width; | |
| for ( y = y_start; y !== y_end; y += y_step ) { | |
| for ( x = x_start; x !== x_end; x += x_step, i += 2 ) { | |
| color = image[ i + 0 ] + ( image[ i + 1 ] << 8 ); | |
| imageData[ ( x + width * y ) * 4 + 0 ] = ( color & 0x7C00 ) >> 7; | |
| imageData[ ( x + width * y ) * 4 + 1 ] = ( color & 0x03E0 ) >> 2; | |
| imageData[ ( x + width * y ) * 4 + 2 ] = ( color & 0x001F ) << 3; | |
| imageData[ ( x + width * y ) * 4 + 3 ] = ( color & 0x8000 ) ? 0 : 255; | |
| } | |
| } | |
| return imageData; | |
| } | |
| function tgaGetImageData24bits( imageData, y_start, y_step, y_end, x_start, x_step, x_end, image ) { | |
| let i = 0, x, y; | |
| const width = header.width; | |
| for ( y = y_start; y !== y_end; y += y_step ) { | |
| for ( x = x_start; x !== x_end; x += x_step, i += 3 ) { | |
| imageData[ ( x + width * y ) * 4 + 3 ] = 255; | |
| imageData[ ( x + width * y ) * 4 + 2 ] = image[ i + 0 ]; | |
| imageData[ ( x + width * y ) * 4 + 1 ] = image[ i + 1 ]; | |
| imageData[ ( x + width * y ) * 4 + 0 ] = image[ i + 2 ]; | |
| } | |
| } | |
| return imageData; | |
| } | |
| function tgaGetImageData32bits( imageData, y_start, y_step, y_end, x_start, x_step, x_end, image ) { | |
| let i = 0, x, y; | |
| const width = header.width; | |
| for ( y = y_start; y !== y_end; y += y_step ) { | |
| for ( x = x_start; x !== x_end; x += x_step, i += 4 ) { | |
| imageData[ ( x + width * y ) * 4 + 2 ] = image[ i + 0 ]; | |
| imageData[ ( x + width * y ) * 4 + 1 ] = image[ i + 1 ]; | |
| imageData[ ( x + width * y ) * 4 + 0 ] = image[ i + 2 ]; | |
| imageData[ ( x + width * y ) * 4 + 3 ] = image[ i + 3 ]; | |
| } | |
| } | |
| return imageData; | |
| } | |
| function tgaGetImageDataGrey8bits( imageData, y_start, y_step, y_end, x_start, x_step, x_end, image ) { | |
| let color, i = 0, x, y; | |
| const width = header.width; | |
| for ( y = y_start; y !== y_end; y += y_step ) { | |
| for ( x = x_start; x !== x_end; x += x_step, i ++ ) { | |
| color = image[ i ]; | |
| imageData[ ( x + width * y ) * 4 + 0 ] = color; | |
| imageData[ ( x + width * y ) * 4 + 1 ] = color; | |
| imageData[ ( x + width * y ) * 4 + 2 ] = color; | |
| imageData[ ( x + width * y ) * 4 + 3 ] = 255; | |
| } | |
| } | |
| return imageData; | |
| } | |
| function tgaGetImageDataGrey16bits( imageData, y_start, y_step, y_end, x_start, x_step, x_end, image ) { | |
| let i = 0, x, y; | |
| const width = header.width; | |
| for ( y = y_start; y !== y_end; y += y_step ) { | |
| for ( x = x_start; x !== x_end; x += x_step, i += 2 ) { | |
| imageData[ ( x + width * y ) * 4 + 0 ] = image[ i + 0 ]; | |
| imageData[ ( x + width * y ) * 4 + 1 ] = image[ i + 0 ]; | |
| imageData[ ( x + width * y ) * 4 + 2 ] = image[ i + 0 ]; | |
| imageData[ ( x + width * y ) * 4 + 3 ] = image[ i + 1 ]; | |
| } | |
| } | |
| return imageData; | |
| } | |
| function getTgaRGBA( data, width, height, image, palette ) { | |
| let x_start, | |
| y_start, | |
| x_step, | |
| y_step, | |
| x_end, | |
| y_end; | |
| switch ( ( header.flags & TGA_ORIGIN_MASK ) >> TGA_ORIGIN_SHIFT ) { | |
| default: | |
| case TGA_ORIGIN_UL: | |
| x_start = 0; | |
| x_step = 1; | |
| x_end = width; | |
| y_start = 0; | |
| y_step = 1; | |
| y_end = height; | |
| break; | |
| case TGA_ORIGIN_BL: | |
| x_start = 0; | |
| x_step = 1; | |
| x_end = width; | |
| y_start = height - 1; | |
| y_step = - 1; | |
| y_end = - 1; | |
| break; | |
| case TGA_ORIGIN_UR: | |
| x_start = width - 1; | |
| x_step = - 1; | |
| x_end = - 1; | |
| y_start = 0; | |
| y_step = 1; | |
| y_end = height; | |
| break; | |
| case TGA_ORIGIN_BR: | |
| x_start = width - 1; | |
| x_step = - 1; | |
| x_end = - 1; | |
| y_start = height - 1; | |
| y_step = - 1; | |
| y_end = - 1; | |
| break; | |
| } | |
| if ( use_grey ) { | |
| switch ( header.pixel_size ) { | |
| case 8: | |
| tgaGetImageDataGrey8bits( data, y_start, y_step, y_end, x_start, x_step, x_end, image ); | |
| break; | |
| case 16: | |
| tgaGetImageDataGrey16bits( data, y_start, y_step, y_end, x_start, x_step, x_end, image ); | |
| break; | |
| default: | |
| throw new Error( 'THREE.TGALoader: Format not supported.' ); | |
| break; | |
| } | |
| } else { | |
| switch ( header.pixel_size ) { | |
| case 8: | |
| tgaGetImageData8bits( data, y_start, y_step, y_end, x_start, x_step, x_end, image, palette ); | |
| break; | |
| case 16: | |
| tgaGetImageData16bits( data, y_start, y_step, y_end, x_start, x_step, x_end, image ); | |
| break; | |
| case 24: | |
| tgaGetImageData24bits( data, y_start, y_step, y_end, x_start, x_step, x_end, image ); | |
| break; | |
| case 32: | |
| tgaGetImageData32bits( data, y_start, y_step, y_end, x_start, x_step, x_end, image ); | |
| break; | |
| default: | |
| throw new Error( 'THREE.TGALoader: Format not supported.' ); | |
| break; | |
| } | |
| } | |
| // Load image data according to specific method | |
| // let func = 'tgaGetImageData' + (use_grey ? 'Grey' : '') + (header.pixel_size) + 'bits'; | |
| // func(data, y_start, y_step, y_end, x_start, x_step, x_end, width, image, palette ); | |
| return data; | |
| } | |
| // TGA constants | |
| const TGA_TYPE_NO_DATA = 0, | |
| TGA_TYPE_INDEXED = 1, | |
| TGA_TYPE_RGB = 2, | |
| TGA_TYPE_GREY = 3, | |
| TGA_TYPE_RLE_INDEXED = 9, | |
| TGA_TYPE_RLE_RGB = 10, | |
| TGA_TYPE_RLE_GREY = 11, | |
| TGA_ORIGIN_MASK = 0x30, | |
| TGA_ORIGIN_SHIFT = 0x04, | |
| TGA_ORIGIN_BL = 0x00, | |
| TGA_ORIGIN_BR = 0x01, | |
| TGA_ORIGIN_UL = 0x02, | |
| TGA_ORIGIN_UR = 0x03; | |
| if ( buffer.length < 19 ) throw new Error( 'THREE.TGALoader: Not enough data to contain header.' ); | |
| let offset = 0; | |
| const content = new Uint8Array( buffer ), | |
| header = { | |
| id_length: content[ offset ++ ], | |
| colormap_type: content[ offset ++ ], | |
| image_type: content[ offset ++ ], | |
| colormap_index: content[ offset ++ ] | content[ offset ++ ] << 8, | |
| colormap_length: content[ offset ++ ] | content[ offset ++ ] << 8, | |
| colormap_size: content[ offset ++ ], | |
| origin: [ | |
| content[ offset ++ ] | content[ offset ++ ] << 8, | |
| content[ offset ++ ] | content[ offset ++ ] << 8 | |
| ], | |
| width: content[ offset ++ ] | content[ offset ++ ] << 8, | |
| height: content[ offset ++ ] | content[ offset ++ ] << 8, | |
| pixel_size: content[ offset ++ ], | |
| flags: content[ offset ++ ] | |
| }; | |
| // check tga if it is valid format | |
| tgaCheckHeader( header ); | |
| if ( header.id_length + offset > buffer.length ) { | |
| throw new Error( 'THREE.TGALoader: No data.' ); | |
| } | |
| // skip the needn't data | |
| offset += header.id_length; | |
| // get targa information about RLE compression and palette | |
| let use_rle = false, | |
| use_pal = false, | |
| use_grey = false; | |
| switch ( header.image_type ) { | |
| case TGA_TYPE_RLE_INDEXED: | |
| use_rle = true; | |
| use_pal = true; | |
| break; | |
| case TGA_TYPE_INDEXED: | |
| use_pal = true; | |
| break; | |
| case TGA_TYPE_RLE_RGB: | |
| use_rle = true; | |
| break; | |
| case TGA_TYPE_RGB: | |
| break; | |
| case TGA_TYPE_RLE_GREY: | |
| use_rle = true; | |
| use_grey = true; | |
| break; | |
| case TGA_TYPE_GREY: | |
| use_grey = true; | |
| break; | |
| } | |
| // | |
| const imageData = new Uint8Array( header.width * header.height * 4 ); | |
| const result = tgaParse( use_rle, use_pal, header, offset, content ); | |
| getTgaRGBA( imageData, header.width, header.height, result.pixel_data, result.palettes ); | |
| return { | |
| data: imageData, | |
| width: header.width, | |
| height: header.height, | |
| flipY: true, | |
| generateMipmaps: true, | |
| minFilter: LinearMipmapLinearFilter, | |
| }; | |
| } | |
| } | |
| export { TGALoader }; | |