three.js, 사용자 정의 TSL 노드 (add 노드를 예시로…)

three.js의 TSL에서 재사용이 가능한 코드를 TSL 노드화하여 제공하면 DX를 향상시킬 수 있습니다. 이미 제공하고 있는 add 노드와 동일하게 작동하는 myAdd 노드를 예시로 사용자 정의 노드를 만드는 코드입니다.

두가지 방식이 가능하며 첫번째는 프리셋 노드를 활용하는 방식입니다.

class MyAddNode extends THREE.TempNode {
  static get type() {
    return 'MyAddNode';
  }

  constructor(aNode, bNode) {
    super('float'); // 출력 타입을 float로 고정

    this.aNode = aNode;
    this.bNode = bNode;
  }

  // 실제 연산 로직 - 기존 TSL 연산자를 조합해서 구성
  setup() {
    const { aNode, bNode } = this;

    return float(aNode).add(float(bNode));
  }
}

export default MyAddNode;

이렇게 정의된 노들 함수형 문법과 메서드 체이닝 문법으로 사용될 수 있도록 하는 코드는 다음과 같습니다.

export default MyAddNode;

// 함수형 문법으로 노출: myAdd( a, b )
export const myAdd = /*@__PURE__*/ nodeProxy(MyAddNode).setParameterLength(2).setName('myAdd');

// 메서드 체이닝 지원: a.myAdd( b )
addMethodChaining('myAdd', myAdd);

프리셋 노드를 사용하는 방식은 커스텀 노드 개발이 간단하다는 장점은 있으나 원하는 기능을 고진 프리셋 노드가 없을 경우 WGSL 코드나 WGSL 코드를 직접 이용하는 방식이며 앞서 봤던 것과 동일한 기능을 가진 노드의 클래스는 다음과 같습니다.

class MyAddNode extends THREE.TempNode {
  static get type() {
    return 'MyAddNode';
  }

  constructor(aNode, bNode) {
    super('float');
    this.aNode = aNode;
    this.bNode = bNode;
  }

  generate(builder, output) {
    const type = this.getNodeType(builder); // 'float'

    const a = this.aNode.build(builder, type);
    const b = this.bNode.build(builder, type);

    return builder.format(`( ${a} + ${b} )`, type, output);
  }
}