Doodle3D-Slicer/three.js-master/examples/js/math/Lut.js
2017-06-22 13:21:07 +02:00

472 lines
13 KiB
JavaScript
Executable File

/**
* @author daron1337 / http://daron1337.github.io/
*/
THREE.Lut = function ( colormap, numberofcolors ) {
this.lut = [];
this.map = THREE.ColorMapKeywords[ colormap ];
this.n = numberofcolors;
this.mapname = colormap;
var step = 1.0 / this.n;
for ( var i = 0; i <= 1; i += step ) {
for ( var j = 0; j < this.map.length - 1; j ++ ) {
if ( i >= this.map[ j ][ 0 ] && i < this.map[ j + 1 ][ 0 ] ) {
var min = this.map[ j ][ 0 ];
var max = this.map[ j + 1 ][ 0 ];
var color = new THREE.Color( 0xffffff );
var minColor = new THREE.Color( 0xffffff ).setHex( this.map[ j ][ 1 ] );
var maxColor = new THREE.Color( 0xffffff ).setHex( this.map[ j + 1 ][ 1 ] );
color = minColor.lerp( maxColor, ( i - min ) / ( max - min ) );
this.lut.push(color);
}
}
}
return this.set( this );
};
THREE.Lut.prototype = {
constructor: THREE.Lut,
lut: [], map: [], mapname: 'rainbow', n: 256, minV: 0, maxV: 1, legend: null,
set: function ( value ) {
if ( value instanceof THREE.Lut ) {
this.copy( value );
}
return this;
},
setMin: function ( min ) {
this.minV = min;
return this;
},
setMax: function ( max ) {
this.maxV = max;
return this;
},
changeNumberOfColors: function ( numberofcolors ) {
this.n = numberofcolors;
return new THREE.Lut( this.mapname, this.n );
},
changeColorMap: function ( colormap ) {
this.mapname = colormap;
return new THREE.Lut( this.mapname, this.n );
},
copy: function ( lut ) {
this.lut = lut.lut;
this.mapname = lut.mapname;
this.map = lut.map;
this.n = lut.n;
this.minV = lut.minV;
this.maxV = lut.maxV;
return this;
},
getColor: function ( alpha ) {
if ( alpha <= this.minV ) {
alpha = this.minV;
} else if ( alpha >= this.maxV ) {
alpha = this.maxV;
}
alpha = ( alpha - this.minV ) / ( this.maxV - this.minV );
var colorPosition = Math.round ( alpha * this.n );
colorPosition == this.n ? colorPosition -= 1 : colorPosition;
return this.lut[ colorPosition ];
},
addColorMap: function ( colormapName, arrayOfColors ) {
THREE.ColorMapKeywords[ colormapName ] = arrayOfColors;
},
setLegendOn: function ( parameters ) {
if ( parameters === undefined ) { parameters = {}; }
this.legend = {};
this.legend.layout = parameters.hasOwnProperty( 'layout' ) ? parameters[ 'layout' ] : 'vertical';
this.legend.position = parameters.hasOwnProperty( 'position' ) ? parameters[ 'position' ] : { 'x': 21.5, 'y': 8, 'z': 5 };
this.legend.dimensions = parameters.hasOwnProperty( 'dimensions' ) ? parameters[ 'dimensions' ] : { 'width': 0.5, 'height': 3 };
this.legend.canvas = document.createElement( 'canvas' );
this.legend.canvas.setAttribute( 'id', 'legend' );
this.legend.canvas.setAttribute( 'hidden', true );
document.body.appendChild( this.legend.canvas );
this.legend.ctx = this.legend.canvas.getContext( '2d' );
this.legend.canvas.setAttribute( 'width', 1 );
this.legend.canvas.setAttribute( 'height', this.n );
this.legend.texture = new THREE.Texture( this.legend.canvas );
imageData = this.legend.ctx.getImageData( 0, 0, 1, this.n );
data = imageData.data;
len = data.length;
this.map = THREE.ColorMapKeywords[ this.mapname ];
var k = 0;
var step = 1.0 / this.n;
for ( var i = 1; i >= 0; i-=step ) {
for ( var j = this.map.length - 1; j >= 0; j -- ) {
if ( i < this.map[ j ][ 0 ] && i >= this.map[ j - 1 ][ 0 ] ) {
var min = this.map[ j - 1 ][ 0 ];
var max = this.map[ j ][ 0 ];
var color = new THREE.Color( 0xffffff );
var minColor = new THREE.Color( 0xffffff ).setHex( this.map[ j - 1][ 1 ] );
var maxColor = new THREE.Color( 0xffffff ).setHex( this.map[ j ][ 1 ] );
color = minColor.lerp( maxColor, ( i - min ) / ( max - min ) );
data[ k * 4 ] = Math.round( color.r * 255 );
data[ k * 4 + 1 ] = Math.round( color.g * 255 );
data[ k * 4 + 2 ] = Math.round( color.b * 255 );
data[ k * 4 + 3 ] = 255;
k+=1;
}
}
}
this.legend.ctx.putImageData( imageData, 0, 0 );
this.legend.texture.needsUpdate = true;
this.legend.legendGeometry = new THREE.PlaneBufferGeometry( this.legend.dimensions.width, this.legend.dimensions.height );
this.legend.legendMaterial = new THREE.MeshBasicMaterial( { map : this.legend.texture, side : THREE.DoubleSide } );
this.legend.mesh = new THREE.Mesh( this.legend.legendGeometry, this.legend.legendMaterial );
if ( this.legend.layout == 'horizontal') {
this.legend.mesh.rotation.z = - 90 * ( Math.PI / 180 );
}
this.legend.mesh.position.copy( this.legend.position );
return this.legend.mesh;
},
setLegendOff: function () {
this.legend = null;
return this.legend;
},
setLegendLayout: function ( layout ) {
if ( ! this.legend ) { return false; }
if ( this.legend.layout == layout ) { return false; }
if ( layout != 'horizontal' && layout != 'vertical' ) { return false; }
this.layout = layout;
if ( layout == 'horizontal' ) {
this.legend.mesh.rotation.z = 90 * ( Math.PI / 180 );
}
if ( layout == 'vertical' ) {
this.legend.mesh.rotation.z = -90 * ( Math.PI / 180 );
}
return this.legend.mesh;
},
setLegendPosition: function ( position ) {
this.legend.position = new THREE.Vector3( position.x, position.y, position.z );
return this.legend;
},
setLegendLabels: function ( parameters, callback ) {
if ( ! this.legend ) { return false; }
if ( typeof parameters === 'function') { callback = parameters; }
if ( parameters === undefined ) { parameters = {}; }
this.legend.labels = {};
this.legend.labels.fontsize = parameters.hasOwnProperty( 'fontsize' ) ? parameters[ 'fontsize' ] : 24;
this.legend.labels.fontface = parameters.hasOwnProperty( 'fontface' ) ? parameters[ 'fontface' ] : 'Arial';
this.legend.labels.title = parameters.hasOwnProperty( 'title' ) ? parameters[ 'title' ] : '';
this.legend.labels.um = parameters.hasOwnProperty( 'um' ) ? ' [ ' + parameters[ 'um' ] + ' ]' : '';
this.legend.labels.ticks = parameters.hasOwnProperty( 'ticks' ) ? parameters[ 'ticks' ] : 0;
this.legend.labels.decimal = parameters.hasOwnProperty( 'decimal' ) ? parameters[ 'decimal' ] : 2;
this.legend.labels.notation = parameters.hasOwnProperty( 'notation' ) ? parameters[ 'notation' ] : 'standard';
var backgroundColor = { r: 255, g: 100, b: 100, a: 0.8 };
var borderColor = { r: 255, g: 0, b: 0, a: 1.0 };
var borderThickness = 4;
var canvasTitle = document.createElement( 'canvas' );
var contextTitle = canvasTitle.getContext( '2d' );
contextTitle.font = 'Normal ' + this.legend.labels.fontsize * 1.2 + 'px ' + this.legend.labels.fontface;
var metrics = contextTitle.measureText( this.legend.labels.title.toString() + this.legend.labels.um.toString() );
var textWidth = metrics.width;
contextTitle.fillStyle = 'rgba(' + backgroundColor.r + ',' + backgroundColor.g + ',' + backgroundColor.b + ',' + backgroundColor.a + ')';
contextTitle.strokeStyle = 'rgba(' + borderColor.r + ',' + borderColor.g + ',' + borderColor.b + ',' + borderColor.a + ')';
contextTitle.lineWidth = borderThickness;
contextTitle.fillStyle = 'rgba( 0, 0, 0, 1.0 )';
contextTitle.fillText( this.legend.labels.title.toString() + this.legend.labels.um.toString(), borderThickness, this.legend.labels.fontsize + borderThickness );
var txtTitle = new THREE.Texture( canvasTitle );
txtTitle.minFilter = THREE.LinearFilter;
txtTitle.needsUpdate = true;
var spriteMaterialTitle = new THREE.SpriteMaterial( { map: txtTitle, useScreenCoordinates: false } );
var spriteTitle = new THREE.Sprite( spriteMaterialTitle );
spriteTitle.scale.set( 2, 1, 1.0 );
if ( this.legend.layout == 'vertical' ) {
spriteTitle.position.set( this.legend.position.x + this.legend.dimensions.width, this.legend.position.y + ( this.legend.dimensions.height * 0.45 ), this.legend.position.z );
}
if ( this.legend.layout == 'horizontal' ) {
spriteTitle.position.set( this.legend.position.x * 1.015, this.legend.position.y + ( this.legend.dimensions.height * 0.03 ), this.legend.position.z );
}
if ( this.legend.labels.ticks > 0 ) {
var ticks = {};
var lines = {};
if ( this.legend.layout == 'vertical' ) {
var topPositionY = this.legend.position.y + ( this.legend.dimensions.height * 0.36 );
var bottomPositionY = this.legend.position.y - ( this.legend.dimensions.height * 0.61 );
}
if ( this.legend.layout == 'horizontal' ) {
var topPositionX = this.legend.position.x + ( this.legend.dimensions.height * 0.75 );
var bottomPositionX = this.legend.position.x - ( this.legend.dimensions.width * 1.2 ) ;
}
for ( var i = 0; i < this.legend.labels.ticks; i ++ ) {
var value = ( this.maxV - this.minV ) / ( this.legend.labels.ticks - 1 ) * i ;
if ( callback ) {
value = callback ( value );
}
else {
if ( this.legend.labels.notation == 'scientific' ) {
value = value.toExponential( this.legend.labels.decimal );
}
else {
value = value.toFixed( this.legend.labels.decimal );
}
}
var canvasTick = document.createElement( 'canvas' );
var contextTick = canvasTick.getContext( '2d' );
contextTick.font = 'Normal ' + this.legend.labels.fontsize + 'px ' + this.legend.labels.fontface;
var metrics = contextTick.measureText( value.toString() );
var textWidth = metrics.width;
contextTick.fillStyle = 'rgba(' + backgroundColor.r + ',' + backgroundColor.g + ',' + backgroundColor.b + ',' + backgroundColor.a + ')';
contextTick.strokeStyle = 'rgba(' + borderColor.r + ',' + borderColor.g + ',' + borderColor.b + ',' + borderColor.a + ')';
contextTick.lineWidth = borderThickness;
contextTick.fillStyle = 'rgba( 0, 0, 0, 1.0 )';
contextTick.fillText( value.toString(), borderThickness, this.legend.labels.fontsize + borderThickness );
var txtTick = new THREE.Texture( canvasTick );
txtTick.minFilter = THREE.LinearFilter;
txtTick.needsUpdate = true;
var spriteMaterialTick = new THREE.SpriteMaterial( { map: txtTick, useScreenCoordinates: false } );
var spriteTick = new THREE.Sprite( spriteMaterialTick );
spriteTick.scale.set( 2, 1, 1.0 );
if ( this.legend.layout == 'vertical' ) {
var position = bottomPositionY + ( topPositionY - bottomPositionY ) * ( value / ( this.maxV - this.minV ) );
spriteTick.position.set( this.legend.position.x + ( this.legend.dimensions.width * 2.7 ), position, this.legend.position.z );
}
if ( this.legend.layout == 'horizontal' ) {
var position = bottomPositionX + ( topPositionX - bottomPositionX ) * ( value / ( this.maxV - this.minV ) );
if ( this.legend.labels.ticks > 5 ) {
if ( i % 2 === 0 ) { var offset = 1.7; }
else { var offset = 2.1; }
}
else { var offset = 1.7; }
spriteTick.position.set( position, this.legend.position.y - this.legend.dimensions.width * offset, this.legend.position.z );
}
var material = new THREE.LineBasicMaterial( { color: 0x000000, linewidth: 2 } );
var geometry = new THREE.Geometry();
if ( this.legend.layout == 'vertical' ) {
var linePosition = ( this.legend.position.y - ( this.legend.dimensions.height * 0.5 ) + 0.01 ) + ( this.legend.dimensions.height ) * ( value / ( this.maxV - this.minV ) * 0.99 );
geometry.vertices.push( new THREE.Vector3( this.legend.position.x + this.legend.dimensions.width * 0.55, linePosition, this.legend.position.z ) );
geometry.vertices.push( new THREE.Vector3( this.legend.position.x + this.legend.dimensions.width * 0.7, linePosition, this.legend.position.z ) );
}
if ( this.legend.layout == 'horizontal' ) {
var linePosition = ( this.legend.position.x - ( this.legend.dimensions.height * 0.5 ) + 0.01 ) + ( this.legend.dimensions.height ) * ( value / ( this.maxV - this.minV ) * 0.99 );
geometry.vertices.push( new THREE.Vector3( linePosition, this.legend.position.y - this.legend.dimensions.width * 0.55, this.legend.position.z ) );
geometry.vertices.push( new THREE.Vector3( linePosition, this.legend.position.y - this.legend.dimensions.width * 0.7, this.legend.position.z ) );
}
var line = new THREE.Line( geometry, material );
lines[ i ] = line;
ticks[ i ] = spriteTick;
}
}
return { 'title': spriteTitle, 'ticks': ticks, 'lines': lines };
}
};
THREE.ColorMapKeywords = {
"rainbow": [ [ 0.0, '0x0000FF' ], [ 0.2, '0x00FFFF' ], [ 0.5, '0x00FF00' ], [ 0.8, '0xFFFF00' ], [ 1.0, '0xFF0000' ] ],
"cooltowarm": [ [ 0.0, '0x3C4EC2' ], [ 0.2, '0x9BBCFF' ], [ 0.5, '0xDCDCDC' ], [ 0.8, '0xF6A385' ], [ 1.0, '0xB40426' ] ],
"blackbody" : [ [ 0.0, '0x000000' ], [ 0.2, '0x780000' ], [ 0.5, '0xE63200' ], [ 0.8, '0xFFFF00' ], [ 1.0, '0xFFFFFF' ] ],
"grayscale" : [ [ 0.0, '0x000000' ], [ 0.2, '0x404040' ], [ 0.5, '0x7F7F80' ], [ 0.8, '0xBFBFBF' ], [ 1.0, '0xFFFFFF' ] ]
};