@lovelace_lol/embody 0.0.3 → 0.0.5

package/dist/index.cjs

@@ -4229,16 +4229,6 @@ function positiveFiniteOr(value, fallback) {
 }
 
 // src/core/WasmRuntimeCore.ts
-var AU_MORPH_BINDING_STRIDE = 5;
-var VISEME_MORPH_BINDING_STRIDE = 4;
-var AXIS_GROUP_NEGATIVE = 0;
-var AXIS_GROUP_POSITIVE = 1;
-var AXIS_GROUP_PLAIN = 2;
-var BONE_SIDE_NONE = 0;
-var BONE_SIDE_LEFT = 1;
-var BONE_SIDE_RIGHT = 2;
-var ROTATION_CHANNELS = { rx: 0, ry: 1, rz: 2 };
-var TRANSLATION_CHANNELS = { tx: 0, ty: 1, tz: 2 };
 var WasmRuntimeCore = class {
   constructor(options) {
     __publicField(this, "wasm");
@@ -4252,12 +4242,11 @@ var WasmRuntimeCore = class {
     if (!RuntimeCtor) {
       throw new Error("Embody Wasm module does not export RuntimeCore");
     }
-    this.core = new RuntimeCtor(getProfileVisemeSlots(options.profile).length);
+    this.core = new RuntimeCtor(0);
     this.reloadBindings();
   }
   setProfile(profile) {
     this.profile = profile;
-    this.core.set_viseme_slot_count(getProfileVisemeSlots(profile).length);
     this.reloadBindings();
   }
   setModelDescriptor(model) {
@@ -4302,191 +4291,9 @@ var WasmRuntimeCore = class {
     };
   }
   reloadBindings() {
-    const { auBindings, mixedAus, visemeBindings } = compileMorphBindings(this.profile, this.model);
-    this.core.load_au_morph_bindings(auBindings);
-    this.core.load_viseme_morph_bindings(visemeBindings);
-    this.core.set_mixed_aus(mixedAus);
-    const boneBindings = compileBoneBindings(this.profile, this.model);
-    this.core.load_bone_rest_transforms(boneBindings.restTransforms);
-    this.core.load_composite_axes(boneBindings.compositeAxes);
-    this.core.load_bone_translations(boneBindings.translations);
-    this.core.load_jaw_binding(boneBindings.jawBinding);
-    this.core.load_viseme_jaw_amounts(boneBindings.visemeJawAmounts);
+    this.core.configure_with_profile(JSON.stringify(this.profile), JSON.stringify(this.model));
   }
 };
-function compileMorphBindings(profile, model) {
-  const auRows = [];
-  const mixed = [];
-  for (const [auIdText, entry] of Object.entries(profile.auToMorphs || {})) {
-    const auId = Number(auIdText);
-    if (Number.isNaN(auId) || !entry) continue;
-    const meshNames = getMeshNamesForAUProfile(profile, auId);
-    const hasMorphs = !!(entry.left?.length || entry.right?.length || entry.center?.length);
-    if (hasMorphs && profile.auToBones?.[auId]?.length) {
-      mixed.push(auId);
-    }
-    pushAuSide(auRows, auId, 0, entry.left || [], meshNames, model, profile);
-    pushAuSide(auRows, auId, 1, entry.right || [], meshNames, model, profile);
-    pushAuSide(auRows, auId, 2, entry.center || [], meshNames, model, profile);
-  }
-  const visemeRows = [];
-  const visemeMeshNames = getMeshNamesForVisemeProfile(profile);
-  const slotCount = getProfileVisemeSlots(profile).length;
-  for (let index = 0; index < slotCount; index += 1) {
-    for (const bindingTarget of getVisemeBindingTargets(profile, index)) {
-      for (const resolved of resolveMorphTargets(model, profile, bindingTarget.morph, visemeMeshNames)) {
-        visemeRows.push(
-          index,
-          Number(resolved.mesh.id),
-          Number(resolved.morphTarget.id),
-          bindingTarget.weight
-        );
-      }
-    }
-  }
-  return {
-    auBindings: Float32Array.from(auRows),
-    visemeBindings: Float32Array.from(visemeRows),
-    mixedAus: Uint32Array.from(mixed)
-  };
-}
-function compileBoneBindings(profile, model) {
-  const referencedBones = /* @__PURE__ */ new Map();
-  const findBone = (nodeKey) => {
-    const bone = findBoneDescriptor(profile, model, nodeKey);
-    if (bone) referencedBones.set(bone.id, bone);
-    return bone;
-  };
-  const boneSideForAU = (auId, nodeKey) => {
-    const side = profile.auToBones?.[auId]?.find((candidate) => candidate.node === nodeKey)?.side;
-    if (side === "left") return BONE_SIDE_LEFT;
-    if (side === "right") return BONE_SIDE_RIGHT;
-    return BONE_SIDE_NONE;
-  };
-  const compositeRows = [];
-  for (const composite of profile.compositeRotations || []) {
-    const bone = findBone(composite.node);
-    if (!bone) continue;
-    const axes = [
-      { axis: 1, config: composite.yaw },
-      { axis: 0, config: composite.pitch },
-      { axis: 2, config: composite.roll }
-    ];
-    for (const { axis, config } of axes) {
-      if (!config) continue;
-      const negativeAUs = toAUList(config.negative);
-      const positiveAUs = toAUList(config.positive);
-      const hasDirectional = negativeAUs.length > 0 && positiveAUs.length > 0;
-      const valueRows = [];
-      if (hasDirectional) {
-        for (const auId of negativeAUs) {
-          valueRows.push([auId, AXIS_GROUP_NEGATIVE, boneSideForAU(auId, composite.node)]);
-        }
-        for (const auId of positiveAUs) {
-          valueRows.push([auId, AXIS_GROUP_POSITIVE, boneSideForAU(auId, composite.node)]);
-        }
-      } else {
-        for (const auId of config.aus) {
-          valueRows.push([auId, AXIS_GROUP_PLAIN, boneSideForAU(auId, composite.node)]);
-        }
-      }
-      const bindingRows = [];
-      const pushBindingRows = (auIds, group) => {
-        for (const auId of auIds) {
-          const binding = profile.auToBones?.[auId]?.find(
-            (candidate) => candidate.node === composite.node
-          );
-          if (!binding?.maxDegrees) continue;
-          const channel = ROTATION_CHANNELS[binding.channel];
-          if (channel === void 0) continue;
-          bindingRows.push([
-            auId,
-            group,
-            boneSideForAU(auId, composite.node),
-            channel,
-            binding.scale,
-            binding.maxDegrees
-          ]);
-        }
-      };
-      pushBindingRows(negativeAUs, AXIS_GROUP_NEGATIVE);
-      pushBindingRows(positiveAUs, AXIS_GROUP_POSITIVE);
-      pushBindingRows(config.aus, AXIS_GROUP_PLAIN);
-      compositeRows.push(
-        Number(bone.id),
-        axis,
-        hasDirectional ? 1 : 0,
-        valueRows.length,
-        bindingRows.length,
-        0,
-        0,
-        0
-      );
-      for (const row of valueRows) compositeRows.push(...row);
-      for (const row of bindingRows) compositeRows.push(...row);
-    }
-  }
-  const translationRows = [];
-  for (const [auIdText, bindings] of Object.entries(profile.auToBones || {})) {
-    const auId = Number(auIdText);
-    if (Number.isNaN(auId)) continue;
-    for (const binding of bindings || []) {
-      const channel = TRANSLATION_CHANNELS[binding.channel];
-      if (channel === void 0 || binding.maxUnits === void 0) continue;
-      const bone = findBone(binding.node);
-      if (!bone) continue;
-      translationRows.push(auId, Number(bone.id), channel, binding.scale, binding.maxUnits);
-    }
-  }
-  const jawRow = [];
-  const jawBinding = findAutoVisemeJawBinding(profile, model);
-  if (jawBinding?.maxDegrees) {
-    const bone = findBone(jawBinding.node);
-    const channel = ROTATION_CHANNELS[jawBinding.channel];
-    if (bone && channel !== void 0) {
-      jawRow.push(Number(bone.id), channel, jawBinding.scale, jawBinding.maxDegrees);
-    }
-  }
-  const restRows = [];
-  for (const bone of referencedBones.values()) {
-    const position = bone.restTransform?.position;
-    const rotation = bone.restTransform?.rotation;
-    restRows.push(
-      Number(bone.id),
-      position?.x ?? 0,
-      position?.y ?? 0,
-      position?.z ?? 0,
-      rotation?.x ?? 0,
-      rotation?.y ?? 0,
-      rotation?.z ?? 0,
-      rotation?.w ?? 1
-    );
-  }
-  return {
-    restTransforms: Float32Array.from(restRows),
-    compositeAxes: Float32Array.from(compositeRows),
-    translations: Float32Array.from(translationRows),
-    jawBinding: Float32Array.from(jawRow),
-    visemeJawAmounts: Float32Array.from(getVisemeJawAmounts(profile) ?? [])
-  };
-}
-function findBoneDescriptor(profile, model, nodeKey) {
-  const configuredName = profile.boneNodes?.[nodeKey] || nodeKey;
-  const prefix = profile.bonePrefix ?? "";
-  const suffix = profile.boneSuffix ?? "";
-  const prefixedName = prefix && !configuredName.startsWith(prefix) ? `${prefix}${configuredName}` : configuredName;
-  const fullName = suffix && !prefixedName.endsWith(suffix) ? `${prefixedName}${suffix}` : prefixedName;
-  const candidates = /* @__PURE__ */ new Set([nodeKey, configuredName, fullName]);
-  return model.bones.find((bone) => candidates.has(bone.name));
-}
-function findAutoVisemeJawBinding(profile, model) {
-  const isRotation = (binding) => binding?.channel === "rx" || binding?.channel === "ry" || binding?.channel === "rz";
-  const candidates = [
-    profile.auToBones?.[103]?.find(isRotation),
-    profile.auToBones?.[26]?.find(isRotation)
-  ].filter((binding) => !!binding);
-  return candidates.find((binding) => !!findBoneDescriptor(profile, model, binding.node)) ?? candidates[0] ?? null;
-}
 function unpackBoneFrameDelta(values) {
   const stride = PACKED_BONE_FRAME_DELTA_STRIDE;
   const writes = [];
@@ -4527,46 +4334,6 @@ function unpackMorphFrameDelta(values) {
   }
   return writes;
 }
-function pushAuSide(rows, auId, side, morphs, meshNames, model, profile) {
-  for (const morph of morphs) {
-    for (const resolved of resolveMorphTargets(model, profile, morph, meshNames)) {
-      rows.push(
-        auId,
-        side,
-        Number(resolved.mesh.id),
-        Number(resolved.morphTarget.id),
-        1
-      );
-    }
-  }
-}
-function resolveMorphTargets(model, profile, morph, meshNames) {
-  const result = [];
-  for (const meshName of meshNames) {
-    const mesh = model.meshes.find((candidate) => candidate.name === meshName);
-    if (!mesh) continue;
-    const morphTargets = mesh.morphTargetIds.map((id) => model.morphTargets.find((candidate) => candidate.id === id)).filter((target2) => !!target2);
-    const target = typeof morph === "number" ? morphTargets.find((candidate) => candidate.hostIndex === morph) : resolveMorphTargetByName2(morphTargets, morph, profile);
-    if (target) {
-      result.push({ mesh, morphTarget: target });
-    }
-  }
-  return result;
-}
-function resolveMorphTargetByName2(targets, key, profile) {
-  const prefix = profile.morphPrefix || "";
-  const suffix = profile.morphSuffix || "";
-  const fullName = prefix + key + suffix;
-  const exact = targets.find((target) => target.name === fullName);
-  if (exact) return exact;
-  const suffixRegex = profile.suffixPattern ? new RegExp(profile.suffixPattern) : null;
-  if (!suffixRegex) return void 0;
-  return targets.find((target) => {
-    if (!target.name.startsWith(fullName)) return false;
-    const candidateSuffix = target.name.slice(fullName.length);
-    return candidateSuffix === "" || suffixRegex.test(candidateSuffix);
-  });
-}
 
 // src/hair/appearance.ts
 var HAIR_COLOR_PRESETS = {
@@ -10437,6 +10204,16 @@ var BLENDING_MODES = {
   "None": 0
   // THREE.NoBlending
 };
+
+// src/profiles/rustResolvedProfile.ts
+function resolveRustProfileView(wasm, profile) {
+  return JSON.parse(wasm.resolve_profile_view(JSON.stringify(profile)));
+}
+function resolveRustEmbeddedProfileView(wasm, presetId, profileOverride = null) {
+  return JSON.parse(
+    wasm.resolve_embedded_profile_view(presetId, profileOverride ? JSON.stringify(profileOverride) : "")
+  );
+}
 var DEFAULT_HEAD_BONE_NAMES = ["CC_Base_Head", "Head", "head", "Bip01_Head"];
 var DEFAULT_REFERENCE_HEIGHT = 1.8;
 var DEFAULT_FORWARD_OFFSET = 0.08;
@@ -15254,7 +15031,6 @@ Object.defineProperty(exports, "resetEmbodyCoreForTests", {
 exports.AU_INFO = AU_INFO;
 exports.AU_MAPPING_CONFIG = AU_MAPPING_CONFIG;
 exports.AU_MIX_DEFAULTS = AU_MIX_DEFAULTS;
-exports.AU_MORPH_BINDING_STRIDE = AU_MORPH_BINDING_STRIDE;
 exports.AU_TO_MORPHS = AU_TO_MORPHS;
 exports.AnimationController = ThreeAnimationSystem;
 exports.BETTA_FISH_PRESET = BETTA_FISH_PRESET;
@@ -15315,7 +15091,6 @@ exports.TsClipCompiler = TsClipCompiler;
 exports.TsRuntimeCore = TsRuntimeCore;
 exports.VISEME_JAW_AMOUNTS = VISEME_JAW_AMOUNTS;
 exports.VISEME_KEYS = VISEME_KEYS;
-exports.VISEME_MORPH_BINDING_STRIDE = VISEME_MORPH_BINDING_STRIDE;
 exports.WasmRuntimeCore = WasmRuntimeCore;
 exports.analyzeModel = analyzeModel;
 exports.applyAUBoneBindingUpdate = applyAUBoneBindingUpdate;
@@ -15328,8 +15103,6 @@ exports.buildFrameApplierBindings = buildFrameApplierBindings;
 exports.buildMappingEditorModel = buildMappingEditorModel;
 exports.classifyAuAsJointControl = classifyAuAsJointControl;
 exports.collectMorphMeshes = collectMorphMeshes2;
-exports.compileBoneBindings = compileBoneBindings;
-exports.compileMorphBindings = compileMorphBindings;
 exports.compileVisemeKeys = compileVisemeKeys;
 exports.composeTemplateSkeletonFitTransform = composeTemplateSkeletonFitTransform;
 exports.computeCameraRelativeGazeOffset = computeCameraRelativeGazeOffset;
@@ -15417,6 +15190,8 @@ exports.resolvePresetWithOverrides = resolvePresetWithOverrides;
 exports.resolveProfileFromPreset = resolveProfileFromPreset;
 exports.resolveRegionCameraAngle = resolveRegionCameraAngle;
 exports.resolveRegionVisibilityCameraAngle = resolveRegionVisibilityCameraAngle;
+exports.resolveRustEmbeddedProfileView = resolveRustEmbeddedProfileView;
+exports.resolveRustProfileView = resolveRustProfileView;
 exports.resolveVisemeMeshCategory = resolveVisemeMeshCategory;
 exports.stripConfiguredBoneAffixes = stripConfiguredBoneAffixes;
 exports.suggestBestPreset = suggestBestPreset;