polygonjs-engine/dist/src/core/operations/sop/AttribCreate.js

node-based webgl 3D engine https://polygonjs.com

import {BaseSopOperation} from "./_Base";
import {Vector2 as Vector22} from "three/src/math/Vector2";
import {Vector3 as Vector32} from "three/src/math/Vector3";
import {Vector4 as Vector42} from "three/src/math/Vector4";
import {ATTRIBUTE_CLASSES, AttribClass, AttribType, ATTRIBUTE_TYPES} from "../../geometry/Constant";
import {InputCloneMode as InputCloneMode2} from "../../../engine/poly/InputCloneMode";
import {TypeAssert} from "../../../engine/poly/Assert";
import {CoreAttribute} from "../../geometry/Attribute";
export class AttribCreateSopOperation extends BaseSopOperation {
  static type() {
    return "attribCreate";
  }
  cook(input_contents, params) {
    const core_group = input_contents[0];
    if (params.name && params.name.trim() != "") {
      this._add_attribute(ATTRIBUTE_CLASSES[params.class], core_group, params);
    } else {
      this.states?.error.set("attribute name is not valid");
    }
    return core_group;
  }
  async _add_attribute(attrib_class, core_group, params) {
    const attrib_type = ATTRIBUTE_TYPES[params.type];
    switch (attrib_class) {
      case AttribClass.VERTEX:
        await this.add_point_attribute(attrib_type, core_group, params);
        return;
      case AttribClass.OBJECT:
        await this.add_object_attribute(attrib_type, core_group, params);
        return;
    }
    TypeAssert.unreachable(attrib_class);
  }
  async add_point_attribute(attrib_type, core_group, params) {
    const core_objects = core_group.coreObjects();
    switch (attrib_type) {
      case AttribType.NUMERIC: {
        for (let i = 0; i < core_objects.length; i++) {
          await this.add_numeric_attribute_to_points(core_objects[i], params);
        }
        return;
      }
      case AttribType.STRING: {
        for (let i = 0; i < core_objects.length; i++) {
          await this.add_string_attribute_to_points(core_objects[i], params);
        }
        return;
      }
    }
    TypeAssert.unreachable(attrib_type);
  }
  async add_object_attribute(attrib_type, core_group, params) {
    const core_objects = core_group.coreObjectsFromGroup(params.group);
    switch (attrib_type) {
      case AttribType.NUMERIC:
        await this.add_numeric_attribute_to_object(core_objects, params);
        return;
      case AttribType.STRING:
        await this.add_string_attribute_to_object(core_objects, params);
        return;
    }
    TypeAssert.unreachable(attrib_type);
  }
  async add_numeric_attribute_to_points(core_object, params) {
    const core_geometry = core_object.coreGeometry();
    if (!core_geometry) {
      return;
    }
    const value = [params.value1, params.value2, params.value3, params.value4][params.size - 1];
    core_object.addNumericVertexAttrib(params.name, params.size, value);
  }
  async add_numeric_attribute_to_object(core_objects, params) {
    const value = [params.value1, params.value2, params.value3, params.value4][params.size - 1];
    for (let core_object of core_objects) {
      core_object.setAttribValue(params.name, value);
    }
  }
  async add_string_attribute_to_points(core_object, params) {
    const points = core_object.pointsFromGroup(params.group);
    const value = params.string;
    const string_values = new Array(points.length);
    for (let i = 0; i < points.length; i++) {
      string_values[i] = value;
    }
    const index_data = CoreAttribute.array_to_indexed_arrays(string_values);
    const geometry = core_object.coreGeometry();
    if (geometry) {
      geometry.setIndexedAttribute(params.name, index_data["values"], index_data["indices"]);
    }
  }
  async add_string_attribute_to_object(core_objects, params) {
    const value = params.string;
    for (let core_object of core_objects) {
      core_object.setAttribValue(params.name, value);
    }
  }
}
AttribCreateSopOperation.DEFAULT_PARAMS = {
  group: "",
  class: ATTRIBUTE_CLASSES.indexOf(AttribClass.VERTEX),
  type: ATTRIBUTE_TYPES.indexOf(AttribType.NUMERIC),
  name: "new_attrib",
  size: 1,
  value1: 0,
  value2: new Vector22(0, 0),
  value3: new Vector32(0, 0, 0),
  value4: new Vector42(0, 0, 0, 0),
  string: ""
};
AttribCreateSopOperation.INPUT_CLONED_STATE = InputCloneMode2.FROM_NODE;