50 lines
1019 B
JavaScript
50 lines
1019 B
JavaScript
|
import { Return, VariableDeclaration, Accessor } from './AST.js';
|
||
|
|
import GLSLDecoder from './GLSLDecoder.js';
|
||
|
|
|
||
|
|
class ShaderToyDecoder extends GLSLDecoder {
|
||
|
|
|
||
|
|
constructor() {
|
||
|
|
|
||
|
|
super();
|
||
|
|
|
||
|
|
this.addPolyfill( 'iTime', 'float iTime = timerGlobal();' );
|
||
|
|
this.addPolyfill( 'iResolution', 'vec2 iResolution = viewportResolution;' );
|
||
|
|
this.addPolyfill( 'fragCoord', 'vec2 fragCoord = vec2( viewportCoordinate.x, viewportResolution.y - viewportCoordinate.y );' );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
parseFunction() {
|
||
|
|
|
||
|
|
const node = super.parseFunction();
|
||
|
|
|
||
|
|
if ( node.name === 'mainImage' ) {
|
||
|
|
|
||
|
|
node.params = []; // remove default parameters
|
||
|
|
node.type = 'vec4';
|
||
|
|
node.layout = false; // for now
|
||
|
|
|
||
|
|
const fragColor = new Accessor( 'fragColor' );
|
||
|
|
|
||
|
|
for ( const subNode of node.body ) {
|
||
|
|
|
||
|
|
if ( subNode.isReturn ) {
|
||
|
|
|
||
|
|
subNode.value = fragColor;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
node.body.unshift( new VariableDeclaration( 'vec4', 'fragColor' ) );
|
||
|
|
node.body.push( new Return( fragColor ) );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
return node;
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
export default ShaderToyDecoder;
|