MCP 3D Printer Server

by DMontgomery40
Verified
import { AnimationClip, AnimationMixer, Matrix4, Quaternion, QuaternionKeyframeTrack, SkeletonHelper, Vector3, VectorKeyframeTrack } from 'three'; function getBoneName( bone, options ) { if ( options.getBoneName !== undefined ) { return options.getBoneName( bone ); } return options.names[ bone.name ]; } function retarget( target, source, options = {} ) { const quat = new Quaternion(), scale = new Vector3(), relativeMatrix = new Matrix4(), globalMatrix = new Matrix4(); options.preserveBoneMatrix = options.preserveBoneMatrix !== undefined ? options.preserveBoneMatrix : true; options.preserveBonePositions = options.preserveBonePositions !== undefined ? options.preserveBonePositions : true; options.useTargetMatrix = options.useTargetMatrix !== undefined ? options.useTargetMatrix : false; options.hip = options.hip !== undefined ? options.hip : 'hip'; options.hipInfluence = options.hipInfluence !== undefined ? options.hipInfluence : new Vector3( 1, 1, 1 ); options.scale = options.scale !== undefined ? options.scale : 1; options.names = options.names || {}; const sourceBones = source.isObject3D ? source.skeleton.bones : getBones( source ), bones = target.isObject3D ? target.skeleton.bones : getBones( target ); let bone, name, boneTo, bonesPosition; // reset bones if ( target.isObject3D ) { target.skeleton.pose(); } else { options.useTargetMatrix = true; options.preserveBoneMatrix = false; } if ( options.preserveBonePositions ) { bonesPosition = []; for ( let i = 0; i < bones.length; i ++ ) { bonesPosition.push( bones[ i ].position.clone() ); } } if ( options.preserveBoneMatrix ) { // reset matrix target.updateMatrixWorld(); target.matrixWorld.identity(); // reset children matrix for ( let i = 0; i < target.children.length; ++ i ) { target.children[ i ].updateMatrixWorld( true ); } } for ( let i = 0; i < bones.length; ++ i ) { bone = bones[ i ]; name = getBoneName( bone, options ); boneTo = getBoneByName( name, sourceBones ); globalMatrix.copy( bone.matrixWorld ); if ( boneTo ) { boneTo.updateMatrixWorld(); if ( options.useTargetMatrix ) { relativeMatrix.copy( boneTo.matrixWorld ); } else { relativeMatrix.copy( target.matrixWorld ).invert(); relativeMatrix.multiply( boneTo.matrixWorld ); } // ignore scale to extract rotation scale.setFromMatrixScale( relativeMatrix ); relativeMatrix.scale( scale.set( 1 / scale.x, 1 / scale.y, 1 / scale.z ) ); // apply to global matrix globalMatrix.makeRotationFromQuaternion( quat.setFromRotationMatrix( relativeMatrix ) ); if ( target.isObject3D ) { if ( options.localOffsets ) { if ( options.localOffsets[ bone.name ] ) { globalMatrix.multiply( options.localOffsets[ bone.name ] ); } } } globalMatrix.copyPosition( relativeMatrix ); } if ( name === options.hip ) { globalMatrix.elements[ 12 ] *= options.scale * options.hipInfluence.x; globalMatrix.elements[ 13 ] *= options.scale * options.hipInfluence.y; globalMatrix.elements[ 14 ] *= options.scale * options.hipInfluence.z; if ( options.hipPosition !== undefined ) { globalMatrix.elements[ 12 ] += options.hipPosition.x * options.scale; globalMatrix.elements[ 13 ] += options.hipPosition.y * options.scale; globalMatrix.elements[ 14 ] += options.hipPosition.z * options.scale; } } if ( bone.parent ) { bone.matrix.copy( bone.parent.matrixWorld ).invert(); bone.matrix.multiply( globalMatrix ); } else { bone.matrix.copy( globalMatrix ); } bone.matrix.decompose( bone.position, bone.quaternion, bone.scale ); bone.updateMatrixWorld(); } if ( options.preserveBonePositions ) { for ( let i = 0; i < bones.length; ++ i ) { bone = bones[ i ]; name = getBoneName( bone, options ) || bone.name; if ( name !== options.hip ) { bone.position.copy( bonesPosition[ i ] ); } } } if ( options.preserveBoneMatrix ) { // restore matrix target.updateMatrixWorld( true ); } } function retargetClip( target, source, clip, options = {} ) { options.useFirstFramePosition = options.useFirstFramePosition !== undefined ? options.useFirstFramePosition : false; // Calculate the fps from the source clip based on the track with the most frames, unless fps is already provided. options.fps = options.fps !== undefined ? options.fps : ( Math.max( ...clip.tracks.map( track => track.times.length ) ) / clip.duration ); options.names = options.names || []; if ( ! source.isObject3D ) { source = getHelperFromSkeleton( source ); } const numFrames = Math.round( clip.duration * ( options.fps / 1000 ) * 1000 ), delta = clip.duration / ( numFrames - 1 ), convertedTracks = [], mixer = new AnimationMixer( source ), bones = getBones( target.skeleton ), boneDatas = []; let positionOffset, bone, boneTo, boneData, name; mixer.clipAction( clip ).play(); // trim let start = 0, end = numFrames; if ( options.trim !== undefined ) { start = Math.round( options.trim[ 0 ] * options.fps ); end = Math.min( Math.round( options.trim[ 1 ] * options.fps ), numFrames ) - start; mixer.update( options.trim[ 0 ] ); } else { mixer.update( 0 ); } source.updateMatrixWorld(); // for ( let frame = 0; frame < end; ++ frame ) { const time = frame * delta; retarget( target, source, options ); for ( let j = 0; j < bones.length; ++ j ) { bone = bones[ j ]; name = getBoneName( bone, options ) || bone.name; boneTo = getBoneByName( name, source.skeleton ); if ( boneTo ) { boneData = boneDatas[ j ] = boneDatas[ j ] || { bone: bone }; if ( options.hip === name ) { if ( ! boneData.pos ) { boneData.pos = { times: new Float32Array( end ), values: new Float32Array( end * 3 ) }; } if ( options.useFirstFramePosition ) { if ( frame === 0 ) { positionOffset = bone.position.clone(); } bone.position.sub( positionOffset ); } boneData.pos.times[ frame ] = time; bone.position.toArray( boneData.pos.values, frame * 3 ); } if ( ! boneData.quat ) { boneData.quat = { times: new Float32Array( end ), values: new Float32Array( end * 4 ) }; } boneData.quat.times[ frame ] = time; bone.quaternion.toArray( boneData.quat.values, frame * 4 ); } } if ( frame === end - 2 ) { // last mixer update before final loop iteration // make sure we do not go over or equal to clip duration mixer.update( delta - 0.0000001 ); } else { mixer.update( delta ); } source.updateMatrixWorld(); } for ( let i = 0; i < boneDatas.length; ++ i ) { boneData = boneDatas[ i ]; if ( boneData ) { if ( boneData.pos ) { convertedTracks.push( new VectorKeyframeTrack( '.bones[' + boneData.bone.name + '].position', boneData.pos.times, boneData.pos.values ) ); } convertedTracks.push( new QuaternionKeyframeTrack( '.bones[' + boneData.bone.name + '].quaternion', boneData.quat.times, boneData.quat.values ) ); } } mixer.uncacheAction( clip ); return new AnimationClip( clip.name, - 1, convertedTracks ); } function clone( source ) { const sourceLookup = new Map(); const cloneLookup = new Map(); const clone = source.clone(); parallelTraverse( source, clone, function ( sourceNode, clonedNode ) { sourceLookup.set( clonedNode, sourceNode ); cloneLookup.set( sourceNode, clonedNode ); } ); clone.traverse( function ( node ) { if ( ! node.isSkinnedMesh ) return; const clonedMesh = node; const sourceMesh = sourceLookup.get( node ); const sourceBones = sourceMesh.skeleton.bones; clonedMesh.skeleton = sourceMesh.skeleton.clone(); clonedMesh.bindMatrix.copy( sourceMesh.bindMatrix ); clonedMesh.skeleton.bones = sourceBones.map( function ( bone ) { return cloneLookup.get( bone ); } ); clonedMesh.bind( clonedMesh.skeleton, clonedMesh.bindMatrix ); } ); return clone; } // internal helper function getBoneByName( name, skeleton ) { for ( let i = 0, bones = getBones( skeleton ); i < bones.length; i ++ ) { if ( name === bones[ i ].name ) return bones[ i ]; } } function getBones( skeleton ) { return Array.isArray( skeleton ) ? skeleton : skeleton.bones; } function getHelperFromSkeleton( skeleton ) { const source = new SkeletonHelper( skeleton.bones[ 0 ] ); source.skeleton = skeleton; return source; } function parallelTraverse( a, b, callback ) { callback( a, b ); for ( let i = 0; i < a.children.length; i ++ ) { parallelTraverse( a.children[ i ], b.children[ i ], callback ); } } export { retarget, retargetClip, clone, };