summaryrefslogtreecommitdiffstats
path: root/scene/3d/xr_face_modifier_3d.cpp
diff options
context:
space:
mode:
authorMalcolm Nixon <Malcolm.nixon@gmail.com>2024-02-11 20:23:52 -0500
committerMalcolm Nixon <Malcolm.nixon@gmail.com>2024-02-17 20:12:47 -0500
commit7d1a1abe7680f75b83034cbd6438690b75c83718 (patch)
tree01f4774c0cefce197aad26a0e121233e1064f9a5 /scene/3d/xr_face_modifier_3d.cpp
parent4e990cd7e51d17cf24f854cc33b2715eaa27200f (diff)
downloadredot-engine-7d1a1abe7680f75b83034cbd6438690b75c83718.tar.gz
Initial XRFaceTrackingProvider and XRFaceTracker work.
Updated to XRFaceModifier3D.
Diffstat (limited to 'scene/3d/xr_face_modifier_3d.cpp')
-rw-r--r--scene/3d/xr_face_modifier_3d.cpp615
1 files changed, 615 insertions, 0 deletions
diff --git a/scene/3d/xr_face_modifier_3d.cpp b/scene/3d/xr_face_modifier_3d.cpp
new file mode 100644
index 0000000000..be92a587b0
--- /dev/null
+++ b/scene/3d/xr_face_modifier_3d.cpp
@@ -0,0 +1,615 @@
+/**************************************************************************/
+/* xr_face_modifier_3d.cpp */
+/**************************************************************************/
+/* This file is part of: */
+/* GODOT ENGINE */
+/* https://godotengine.org */
+/**************************************************************************/
+/* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */
+/* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */
+/* */
+/* Permission is hereby granted, free of charge, to any person obtaining */
+/* a copy of this software and associated documentation files (the */
+/* "Software"), to deal in the Software without restriction, including */
+/* without limitation the rights to use, copy, modify, merge, publish, */
+/* distribute, sublicense, and/or sell copies of the Software, and to */
+/* permit persons to whom the Software is furnished to do so, subject to */
+/* the following conditions: */
+/* */
+/* The above copyright notice and this permission notice shall be */
+/* included in all copies or substantial portions of the Software. */
+/* */
+/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
+/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
+/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. */
+/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
+/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
+/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
+/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
+/**************************************************************************/
+
+#include "xr_face_modifier_3d.h"
+
+#include "servers/xr/xr_face_tracker.h"
+#include "servers/xr_server.h"
+
+// This method takes the name of a mesh blend shape and returns the
+// corresponding XRFaceTracker blend shape. If no match is
+// found then the function returns -1.
+static int find_face_blend_shape(const StringName &p_name) {
+ // Entry for blend shape name table.
+ struct blend_map_entry {
+ int blend;
+ const char *name[4];
+ };
+
+ // Table of blend shape names.
+ //
+ // This table consists of the XRFaceTracker blend shape and
+ // the corresponding names (lowercase and no underscore) of:
+ // - The Unified Expression blend shape name.
+ // - The ARKit blend shape name (if present and different).
+ // - The SRanipal blend shape name (if present and different).
+ // - The Meta blend shape name (if present and different).
+ static constexpr blend_map_entry blend_map[] = {
+ { XRFaceTracker::FT_EYE_LOOK_OUT_RIGHT,
+ { "eyelookoutright", "eyerightright", "eyeslookoutr" } },
+ { XRFaceTracker::FT_EYE_LOOK_IN_RIGHT,
+ { "eyelookinright", "eyerightleft", "eyeslookinr" } },
+ { XRFaceTracker::FT_EYE_LOOK_UP_RIGHT,
+ { "eyelookupright", "eyerightlookup", "eyeslookupr" } },
+ { XRFaceTracker::FT_EYE_LOOK_DOWN_RIGHT,
+ { "eyelookdownright", "eyerightlookdown", "eyeslookdownr" } },
+ { XRFaceTracker::FT_EYE_LOOK_OUT_LEFT,
+ { "eyelookoutleft", "eyeleftleft", "eyeslookoutl" } },
+ { XRFaceTracker::FT_EYE_LOOK_IN_LEFT,
+ { "eyelookinleft", "eyeleftright", "eyeslookinl" } },
+ { XRFaceTracker::FT_EYE_LOOK_UP_LEFT,
+ { "eyelookupleft", "eyeleftlookup", "eyeslookupl" } },
+ { XRFaceTracker::FT_EYE_LOOK_DOWN_LEFT,
+ { "eyelookdownleft", "eyeleftlookdown", "eyeslookdownl" } },
+ { XRFaceTracker::FT_EYE_CLOSED_RIGHT,
+ { "eyeclosedright", "eyeblinkright", "eyerightblink", "eyesclosedr" } },
+ { XRFaceTracker::FT_EYE_CLOSED_LEFT,
+ { "eyeclosedleft", "eyeblinkleft", "eyeleftblink", "eyesclosedl" } },
+ { XRFaceTracker::FT_EYE_SQUINT_RIGHT,
+ { "eyesquintright", "eyessquintr" } },
+ { XRFaceTracker::FT_EYE_SQUINT_LEFT,
+ { "eyesquintleft", "eyessquintl" } },
+ { XRFaceTracker::FT_EYE_WIDE_RIGHT,
+ { "eyewideright", "eyerightwide", "eyeswidenr" } },
+ { XRFaceTracker::FT_EYE_WIDE_LEFT,
+ { "eyewideleft", "eyeleftwide", "eyeswidenl" } },
+ { XRFaceTracker::FT_EYE_DILATION_RIGHT,
+ { "eyedilationright", "eyerightdilation" } },
+ { XRFaceTracker::FT_EYE_DILATION_LEFT,
+ { "eyedilationleft", "eyeleftdilation" } },
+ { XRFaceTracker::FT_EYE_CONSTRICT_RIGHT,
+ { "eyeconstrictright", "eyerightconstrict" } },
+ { XRFaceTracker::FT_EYE_CONSTRICT_LEFT,
+ { "eyeconstrictleft", "eyeleftconstrict" } },
+ { XRFaceTracker::FT_BROW_PINCH_RIGHT,
+ { "browpinchright" } },
+ { XRFaceTracker::FT_BROW_PINCH_LEFT,
+ { "browpinchleft" } },
+ { XRFaceTracker::FT_BROW_LOWERER_RIGHT,
+ { "browlowererright" } },
+ { XRFaceTracker::FT_BROW_LOWERER_LEFT,
+ { "browlowererleft" } },
+ { XRFaceTracker::FT_BROW_INNER_UP_RIGHT,
+ { "browinnerupright", "innerbrowraiserr" } },
+ { XRFaceTracker::FT_BROW_INNER_UP_LEFT,
+ { "browinnerupleft", "innerbrowraiserl" } },
+ { XRFaceTracker::FT_BROW_OUTER_UP_RIGHT,
+ { "browouterupright", "outerbrowraiserr" } },
+ { XRFaceTracker::FT_BROW_OUTER_UP_LEFT,
+ { "browouterupleft", "outerbrowraiserl" } },
+ { XRFaceTracker::FT_NOSE_SNEER_RIGHT,
+ { "nosesneerright", "nosewrinklerr" } },
+ { XRFaceTracker::FT_NOSE_SNEER_LEFT,
+ { "nosesneerleft", "nosewrinklerl" } },
+ { XRFaceTracker::FT_NASAL_DILATION_RIGHT,
+ { "nasaldilationright" } },
+ { XRFaceTracker::FT_NASAL_DILATION_LEFT,
+ { "nasaldilationleft" } },
+ { XRFaceTracker::FT_NASAL_CONSTRICT_RIGHT,
+ { "nasalconstrictright" } },
+ { XRFaceTracker::FT_NASAL_CONSTRICT_LEFT,
+ { "nasalconstrictleft" } },
+ { XRFaceTracker::FT_CHEEK_SQUINT_RIGHT,
+ { "cheeksquintright", "cheekraiserr" } },
+ { XRFaceTracker::FT_CHEEK_SQUINT_LEFT,
+ { "cheeksquintleft", "cheekraiserl" } },
+ { XRFaceTracker::FT_CHEEK_PUFF_RIGHT,
+ { "cheekpuffright", "cheekpuffr" } },
+ { XRFaceTracker::FT_CHEEK_PUFF_LEFT,
+ { "cheekpuffleft", "cheekpuffl" } },
+ { XRFaceTracker::FT_CHEEK_SUCK_RIGHT,
+ { "cheeksuckright", "cheeksuckr" } },
+ { XRFaceTracker::FT_CHEEK_SUCK_LEFT,
+ { "cheeksuckleft", "cheeksuckl" } },
+ { XRFaceTracker::FT_JAW_OPEN,
+ { "jawopen", "jawdrop" } },
+ { XRFaceTracker::FT_MOUTH_CLOSED,
+ { "mouthclosed", "mouthclose", "mouthapeshape", "lipstoward" } },
+ { XRFaceTracker::FT_JAW_RIGHT,
+ { "jawright", "jawsidewaysright" } },
+ { XRFaceTracker::FT_JAW_LEFT,
+ { "jawleft", "jawsidewaysleft" } },
+ { XRFaceTracker::FT_JAW_FORWARD,
+ { "jawforward", "jawthrust" } },
+ { XRFaceTracker::FT_JAW_BACKWARD,
+ { "jawbackward" } },
+ { XRFaceTracker::FT_JAW_CLENCH,
+ { "jawclench" } },
+ { XRFaceTracker::FT_JAW_MANDIBLE_RAISE,
+ { "jawmandibleraise" } },
+ { XRFaceTracker::FT_LIP_SUCK_UPPER_RIGHT,
+ { "lipsuckupperright", "lipsuckrt" } },
+ { XRFaceTracker::FT_LIP_SUCK_UPPER_LEFT,
+ { "lipsuckupperleft", "lipsucklt" } },
+ { XRFaceTracker::FT_LIP_SUCK_LOWER_RIGHT,
+ { "lipsucklowerright", "lipsuckrb" } },
+ { XRFaceTracker::FT_LIP_SUCK_LOWER_LEFT,
+ { "lipsucklowerleft", "lipsucklb" } },
+ { XRFaceTracker::FT_LIP_SUCK_CORNER_RIGHT,
+ { "lipsuckcornerright" } },
+ { XRFaceTracker::FT_LIP_SUCK_CORNER_LEFT,
+ { "lipsuckcornerleft" } },
+ { XRFaceTracker::FT_LIP_FUNNEL_UPPER_RIGHT,
+ { "lipfunnelupperright", "lipfunnelerrt" } },
+ { XRFaceTracker::FT_LIP_FUNNEL_UPPER_LEFT,
+ { "lipfunnelupperleft", "lipfunnelerlt" } },
+ { XRFaceTracker::FT_LIP_FUNNEL_LOWER_RIGHT,
+ { "lipfunnellowerright", "lipsuckrb" } },
+ { XRFaceTracker::FT_LIP_FUNNEL_LOWER_LEFT,
+ { "lipfunnellowerleft", "lipsucklb" } },
+ { XRFaceTracker::FT_LIP_PUCKER_UPPER_RIGHT,
+ { "lippuckerupperright" } },
+ { XRFaceTracker::FT_LIP_PUCKER_UPPER_LEFT,
+ { "lippuckerupperleft" } },
+ { XRFaceTracker::FT_LIP_PUCKER_LOWER_RIGHT,
+ { "lippuckerlowerright" } },
+ { XRFaceTracker::FT_LIP_PUCKER_LOWER_LEFT,
+ { "lippuckerlowerleft" } },
+ { XRFaceTracker::FT_MOUTH_UPPER_UP_RIGHT,
+ { "mouthupperupright", "upperlipraiserr" } },
+ { XRFaceTracker::FT_MOUTH_UPPER_UP_LEFT,
+ { "mouthupperupleft", "upperlipraiserl" } },
+ { XRFaceTracker::FT_MOUTH_LOWER_DOWN_RIGHT,
+ { "mouthlowerdownright", "mouthlowerupright", "lowerlipdepressorr" } },
+ { XRFaceTracker::FT_MOUTH_LOWER_DOWN_LEFT,
+ { "mouthlowerdownleft", "mouthlowerupleft", "lowerlipdepressorl" } },
+ { XRFaceTracker::FT_MOUTH_UPPER_DEEPEN_RIGHT,
+ { "mouthupperdeepenright" } },
+ { XRFaceTracker::FT_MOUTH_UPPER_DEEPEN_LEFT,
+ { "mouthupperdeepenleft" } },
+ { XRFaceTracker::FT_MOUTH_UPPER_RIGHT,
+ { "mouthupperright" } },
+ { XRFaceTracker::FT_MOUTH_UPPER_LEFT,
+ { "mouthupperleft" } },
+ { XRFaceTracker::FT_MOUTH_LOWER_RIGHT,
+ { "mouthlowerright" } },
+ { XRFaceTracker::FT_MOUTH_LOWER_LEFT,
+ { "mouthlowerleft" } },
+ { XRFaceTracker::FT_MOUTH_CORNER_PULL_RIGHT,
+ { "mouthcornerpullright" } },
+ { XRFaceTracker::FT_MOUTH_CORNER_PULL_LEFT,
+ { "mouthcornerpullleft" } },
+ { XRFaceTracker::FT_MOUTH_CORNER_SLANT_RIGHT,
+ { "mouthcornerslantright" } },
+ { XRFaceTracker::FT_MOUTH_CORNER_SLANT_LEFT,
+ { "mouthcornerslantleft" } },
+ { XRFaceTracker::FT_MOUTH_FROWN_RIGHT,
+ { "mouthfrownright", "lipcornerdepressorr" } },
+ { XRFaceTracker::FT_MOUTH_FROWN_LEFT,
+ { "mouthfrownleft", "lipcornerdepressorl" } },
+ { XRFaceTracker::FT_MOUTH_STRETCH_RIGHT,
+ { "mouthstretchright", "lipstretcherr" } },
+ { XRFaceTracker::FT_MOUTH_STRETCH_LEFT,
+ { "mouthstretchleft", "lipstretcherl" } },
+ { XRFaceTracker::FT_MOUTH_DIMPLE_RIGHT,
+ { "mouthdimplerright", "mouthdimpleright", "dimplerr" } },
+ { XRFaceTracker::FT_MOUTH_DIMPLE_LEFT,
+ { "mouthdimplerleft", "mouthdimpleleft", "dimplerl" } },
+ { XRFaceTracker::FT_MOUTH_RAISER_UPPER,
+ { "mouthraiserupper", "mouthshrugupper", "chinraisert" } },
+ { XRFaceTracker::FT_MOUTH_RAISER_LOWER,
+ { "mouthraiserlower", "mouthshruglower", "mouthloweroverlay", "chinraiserb" } },
+ { XRFaceTracker::FT_MOUTH_PRESS_RIGHT,
+ { "mouthpressright", "lippressorr" } },
+ { XRFaceTracker::FT_MOUTH_PRESS_LEFT,
+ { "mouthpressleft", "lippressorl" } },
+ { XRFaceTracker::FT_MOUTH_TIGHTENER_RIGHT,
+ { "mouthtightenerright", "liptightenerr" } },
+ { XRFaceTracker::FT_MOUTH_TIGHTENER_LEFT,
+ { "mouthtightenerleft", "liptightenerl" } },
+ { XRFaceTracker::FT_TONGUE_OUT,
+ { "tongueout", "tonguelongstep2" } },
+ { XRFaceTracker::FT_TONGUE_UP,
+ { "tongueup" } },
+ { XRFaceTracker::FT_TONGUE_DOWN,
+ { "tonguedown" } },
+ { XRFaceTracker::FT_TONGUE_RIGHT,
+ { "tongueright" } },
+ { XRFaceTracker::FT_TONGUE_LEFT,
+ { "tongueleft" } },
+ { XRFaceTracker::FT_TONGUE_ROLL,
+ { "tongueroll" } },
+ { XRFaceTracker::FT_TONGUE_BLEND_DOWN,
+ { "tongueblenddown" } },
+ { XRFaceTracker::FT_TONGUE_CURL_UP,
+ { "tonguecurlup" } },
+ { XRFaceTracker::FT_TONGUE_SQUISH,
+ { "tonguesquish" } },
+ { XRFaceTracker::FT_TONGUE_FLAT,
+ { "tongueflat" } },
+ { XRFaceTracker::FT_TONGUE_TWIST_RIGHT,
+ { "tonguetwistright" } },
+ { XRFaceTracker::FT_TONGUE_TWIST_LEFT,
+ { "tonguetwistleft" } },
+ { XRFaceTracker::FT_SOFT_PALATE_CLOSE,
+ { "softpalateclose" } },
+ { XRFaceTracker::FT_THROAT_SWALLOW,
+ { "throatswallow" } },
+ { XRFaceTracker::FT_NECK_FLEX_RIGHT,
+ { "neckflexright" } },
+ { XRFaceTracker::FT_NECK_FLEX_LEFT,
+ { "neckflexleft" } },
+ { XRFaceTracker::FT_EYE_CLOSED,
+ { "eyeclosed" } },
+ { XRFaceTracker::FT_EYE_WIDE,
+ { "eyewide" } },
+ { XRFaceTracker::FT_EYE_SQUINT,
+ { "eyesquint" } },
+ { XRFaceTracker::FT_EYE_DILATION,
+ { "eyedilation" } },
+ { XRFaceTracker::FT_EYE_CONSTRICT,
+ { "eyeconstrict" } },
+ { XRFaceTracker::FT_BROW_DOWN_RIGHT,
+ { "browdownright", "browlowererr" } },
+ { XRFaceTracker::FT_BROW_DOWN_LEFT,
+ { "browdownleft", "browlowererl" } },
+ { XRFaceTracker::FT_BROW_DOWN,
+ { "browdown" } },
+ { XRFaceTracker::FT_BROW_UP_RIGHT,
+ { "browupright" } },
+ { XRFaceTracker::FT_BROW_UP_LEFT,
+ { "browupleft" } },
+ { XRFaceTracker::FT_BROW_UP,
+ { "browup" } },
+ { XRFaceTracker::FT_NOSE_SNEER,
+ { "nosesneer" } },
+ { XRFaceTracker::FT_NASAL_DILATION,
+ { "nasaldilation" } },
+ { XRFaceTracker::FT_NASAL_CONSTRICT,
+ { "nasalconstrict" } },
+ { XRFaceTracker::FT_CHEEK_PUFF,
+ { "cheekpuff" } },
+ { XRFaceTracker::FT_CHEEK_SUCK,
+ { "cheeksuck" } },
+ { XRFaceTracker::FT_CHEEK_SQUINT,
+ { "cheeksquint" } },
+ { XRFaceTracker::FT_LIP_SUCK_UPPER,
+ { "lipsuckupper", "mouthrollupper", "mouthupperinside" } },
+ { XRFaceTracker::FT_LIP_SUCK_LOWER,
+ { "lipsucklower", "mouthrolllower", "mouthlowerinside" } },
+ { XRFaceTracker::FT_LIP_SUCK,
+ { "lipsuck" } },
+ { XRFaceTracker::FT_LIP_FUNNEL_UPPER,
+ { "lipfunnelupper", "mouthupperoverturn" } },
+ { XRFaceTracker::FT_LIP_FUNNEL_LOWER,
+ { "lipfunnellower", "mouthloweroverturn" } },
+ { XRFaceTracker::FT_LIP_FUNNEL,
+ { "lipfunnel", "mouthfunnel" } },
+ { XRFaceTracker::FT_LIP_PUCKER_UPPER,
+ { "lippuckerupper" } },
+ { XRFaceTracker::FT_LIP_PUCKER_LOWER,
+ { "lippuckerlower" } },
+ { XRFaceTracker::FT_LIP_PUCKER,
+ { "lippucker", "mouthpucker", "mouthpout" } },
+ { XRFaceTracker::FT_MOUTH_UPPER_UP,
+ { "mouthupperup" } },
+ { XRFaceTracker::FT_MOUTH_LOWER_DOWN,
+ { "mouthlowerdown" } },
+ { XRFaceTracker::FT_MOUTH_OPEN,
+ { "mouthopen" } },
+ { XRFaceTracker::FT_MOUTH_RIGHT,
+ { "mouthright" } },
+ { XRFaceTracker::FT_MOUTH_LEFT,
+ { "mouthleft" } },
+ { XRFaceTracker::FT_MOUTH_SMILE_RIGHT,
+ { "mouthsmileright", "lipcornerpullerr" } },
+ { XRFaceTracker::FT_MOUTH_SMILE_LEFT,
+ { "mouthsmileleft", "lipcornerpullerl" } },
+ { XRFaceTracker::FT_MOUTH_SMILE,
+ { "mouthsmile" } },
+ { XRFaceTracker::FT_MOUTH_SAD_RIGHT,
+ { "mouthsadright" } },
+ { XRFaceTracker::FT_MOUTH_SAD_LEFT,
+ { "mouthsadleft" } },
+ { XRFaceTracker::FT_MOUTH_SAD,
+ { "mouthsad" } },
+ { XRFaceTracker::FT_MOUTH_STRETCH,
+ { "mouthstretch" } },
+ { XRFaceTracker::FT_MOUTH_DIMPLE,
+ { "mouthdimple" } },
+ { XRFaceTracker::FT_MOUTH_TIGHTENER,
+ { "mouthtightener" } },
+ { XRFaceTracker::FT_MOUTH_PRESS,
+ { "mouthpress" } }
+ };
+
+ // Convert the name to lower-case and strip non-alphanumeric characters.
+ const String name = String(p_name).to_lower().replace("_", "");
+
+ // Iterate through the blend map.
+ for (const blend_map_entry &entry : blend_map) {
+ for (const char *n : entry.name) {
+ if (n == nullptr) {
+ break;
+ }
+
+ if (name == n) {
+ return entry.blend;
+ }
+ }
+ }
+
+ // Blend shape not found.
+ return -1;
+}
+
+// This method adds all the identified XRFaceTracker blend shapes of
+// the mesh to the p_blend_mapping map. The map is indexed by the
+// XRFaceTracker blend shape, and the value is the index of the mesh
+// blend shape.
+static void identify_face_blend_shapes(RBMap<int, int> &p_blend_mapping, const Ref<Mesh> &mesh) {
+ // Find all blend shapes.
+ const int count = mesh->get_blend_shape_count();
+ for (int i = 0; i < count; i++) {
+ const int blend = find_face_blend_shape(mesh->get_blend_shape_name(i));
+ if (blend >= 0) {
+ p_blend_mapping[blend] = i;
+ }
+ }
+}
+
+// This method removes any unified blend shapes from the p_blend_mapping map
+// if all the individual blend shapes are found and going to be driven.
+static void remove_driven_unified_blend_shapes(RBMap<int, int> &p_blend_mapping) {
+ // Entry for unified blend table.
+ struct unified_blend_entry {
+ int unified;
+ int individual[4];
+ };
+
+ // Table of unified blend shapes.
+ //
+ // This table consists of:
+ // - The XRFaceTracker unified blend shape.
+ // - The individual blend shapes that make up the unified blend shape.
+ static constexpr unified_blend_entry unified_blends[] = {
+ { XRFaceTracker::FT_EYE_CLOSED,
+ { XRFaceTracker::FT_EYE_CLOSED_RIGHT, XRFaceTracker::FT_EYE_CLOSED_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_EYE_WIDE,
+ { XRFaceTracker::FT_EYE_WIDE_RIGHT, XRFaceTracker::FT_EYE_WIDE_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_EYE_SQUINT,
+ { XRFaceTracker::FT_EYE_SQUINT_RIGHT, XRFaceTracker::FT_EYE_SQUINT_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_EYE_DILATION,
+ { XRFaceTracker::FT_EYE_DILATION_RIGHT, XRFaceTracker::FT_EYE_DILATION_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_EYE_CONSTRICT,
+ { XRFaceTracker::FT_EYE_CONSTRICT_RIGHT, XRFaceTracker::FT_EYE_CONSTRICT_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_BROW_DOWN_RIGHT,
+ { XRFaceTracker::FT_BROW_LOWERER_RIGHT, XRFaceTracker::FT_BROW_PINCH_RIGHT, -1, -1 } },
+ { XRFaceTracker::FT_BROW_DOWN_LEFT,
+ { XRFaceTracker::FT_BROW_LOWERER_LEFT, XRFaceTracker::FT_BROW_PINCH_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_BROW_DOWN,
+ { XRFaceTracker::FT_BROW_LOWERER_RIGHT, XRFaceTracker::FT_BROW_PINCH_RIGHT, XRFaceTracker::FT_BROW_LOWERER_LEFT, XRFaceTracker::FT_BROW_PINCH_LEFT } },
+ { XRFaceTracker::FT_BROW_UP_RIGHT,
+ { XRFaceTracker::FT_BROW_INNER_UP_RIGHT, XRFaceTracker::FT_BROW_OUTER_UP_RIGHT, -1, -1 } },
+ { XRFaceTracker::FT_BROW_UP_LEFT,
+ { XRFaceTracker::FT_BROW_INNER_UP_LEFT, XRFaceTracker::FT_BROW_OUTER_UP_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_BROW_UP,
+ { XRFaceTracker::FT_BROW_INNER_UP_RIGHT, XRFaceTracker::FT_BROW_OUTER_UP_RIGHT, XRFaceTracker::FT_BROW_INNER_UP_LEFT, XRFaceTracker::FT_BROW_OUTER_UP_LEFT } },
+ { XRFaceTracker::FT_NOSE_SNEER,
+ { XRFaceTracker::FT_NOSE_SNEER_RIGHT, XRFaceTracker::FT_NOSE_SNEER_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_NASAL_DILATION,
+ { XRFaceTracker::FT_NASAL_DILATION_RIGHT, XRFaceTracker::FT_NASAL_DILATION_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_NASAL_CONSTRICT,
+ { XRFaceTracker::FT_NASAL_CONSTRICT_RIGHT, XRFaceTracker::FT_NASAL_CONSTRICT_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_CHEEK_PUFF,
+ { XRFaceTracker::FT_CHEEK_PUFF_RIGHT, XRFaceTracker::FT_CHEEK_PUFF_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_CHEEK_SUCK,
+ { XRFaceTracker::FT_CHEEK_SUCK_RIGHT, XRFaceTracker::FT_CHEEK_SUCK_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_CHEEK_SQUINT,
+ { XRFaceTracker::FT_CHEEK_SQUINT_RIGHT, XRFaceTracker::FT_CHEEK_SQUINT_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_LIP_SUCK_UPPER,
+ { XRFaceTracker::FT_LIP_SUCK_UPPER_RIGHT, XRFaceTracker::FT_LIP_SUCK_UPPER_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_LIP_SUCK_LOWER,
+ { XRFaceTracker::FT_LIP_SUCK_LOWER_RIGHT, XRFaceTracker::FT_LIP_SUCK_LOWER_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_LIP_SUCK,
+ { XRFaceTracker::FT_LIP_SUCK_UPPER_RIGHT, XRFaceTracker::FT_LIP_SUCK_UPPER_LEFT, XRFaceTracker::FT_LIP_SUCK_LOWER_RIGHT, XRFaceTracker::FT_LIP_SUCK_LOWER_LEFT } },
+ { XRFaceTracker::FT_LIP_FUNNEL_UPPER,
+ { XRFaceTracker::FT_LIP_FUNNEL_UPPER_RIGHT, XRFaceTracker::FT_LIP_FUNNEL_UPPER_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_LIP_FUNNEL_LOWER,
+ { XRFaceTracker::FT_LIP_FUNNEL_LOWER_RIGHT, XRFaceTracker::FT_LIP_FUNNEL_LOWER_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_LIP_FUNNEL,
+ { XRFaceTracker::FT_LIP_FUNNEL_UPPER_RIGHT, XRFaceTracker::FT_LIP_FUNNEL_UPPER_LEFT, XRFaceTracker::FT_LIP_FUNNEL_LOWER_RIGHT, XRFaceTracker::FT_LIP_FUNNEL_LOWER_LEFT } },
+ { XRFaceTracker::FT_LIP_PUCKER_UPPER,
+ { XRFaceTracker::FT_LIP_PUCKER_UPPER_RIGHT, XRFaceTracker::FT_LIP_PUCKER_UPPER_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_LIP_PUCKER_LOWER,
+ { XRFaceTracker::FT_LIP_PUCKER_LOWER_RIGHT, XRFaceTracker::FT_LIP_PUCKER_LOWER_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_LIP_PUCKER,
+ { XRFaceTracker::FT_LIP_PUCKER_UPPER_RIGHT, XRFaceTracker::FT_LIP_PUCKER_UPPER_LEFT, XRFaceTracker::FT_LIP_PUCKER_LOWER_RIGHT, XRFaceTracker::FT_LIP_PUCKER_LOWER_LEFT } },
+ { XRFaceTracker::FT_MOUTH_UPPER_UP,
+ { XRFaceTracker::FT_MOUTH_UPPER_UP_RIGHT, XRFaceTracker::FT_MOUTH_UPPER_UP_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_MOUTH_LOWER_DOWN,
+ { XRFaceTracker::FT_MOUTH_LOWER_DOWN_RIGHT, XRFaceTracker::FT_MOUTH_LOWER_DOWN_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_MOUTH_OPEN,
+ { XRFaceTracker::FT_MOUTH_UPPER_UP_RIGHT, XRFaceTracker::FT_MOUTH_UPPER_UP_LEFT, XRFaceTracker::FT_MOUTH_LOWER_DOWN_RIGHT, XRFaceTracker::FT_MOUTH_LOWER_DOWN_LEFT } },
+ { XRFaceTracker::FT_MOUTH_RIGHT,
+ { XRFaceTracker::FT_MOUTH_UPPER_RIGHT, XRFaceTracker::FT_MOUTH_LOWER_RIGHT, -1, -1 } },
+ { XRFaceTracker::FT_MOUTH_LEFT,
+ { XRFaceTracker::FT_MOUTH_UPPER_LEFT, XRFaceTracker::FT_MOUTH_LOWER_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_MOUTH_SMILE_RIGHT,
+ { XRFaceTracker::FT_MOUTH_CORNER_PULL_RIGHT, XRFaceTracker::FT_MOUTH_CORNER_SLANT_RIGHT, -1, -1 } },
+ { XRFaceTracker::FT_MOUTH_SMILE_LEFT,
+ { XRFaceTracker::FT_MOUTH_CORNER_PULL_LEFT, XRFaceTracker::FT_MOUTH_CORNER_SLANT_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_MOUTH_SMILE,
+ { XRFaceTracker::FT_MOUTH_CORNER_PULL_RIGHT, XRFaceTracker::FT_MOUTH_CORNER_SLANT_RIGHT, XRFaceTracker::FT_MOUTH_CORNER_PULL_LEFT, XRFaceTracker::FT_MOUTH_CORNER_SLANT_LEFT } },
+ { XRFaceTracker::FT_MOUTH_SAD_RIGHT,
+ { XRFaceTracker::FT_MOUTH_FROWN_RIGHT, XRFaceTracker::FT_MOUTH_STRETCH_RIGHT, -1, -1 } },
+ { XRFaceTracker::FT_MOUTH_SAD_LEFT,
+ { XRFaceTracker::FT_MOUTH_FROWN_LEFT, XRFaceTracker::FT_MOUTH_STRETCH_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_MOUTH_SAD,
+ { XRFaceTracker::FT_MOUTH_FROWN_RIGHT, XRFaceTracker::FT_MOUTH_STRETCH_RIGHT, XRFaceTracker::FT_MOUTH_FROWN_LEFT, XRFaceTracker::FT_MOUTH_STRETCH_LEFT } },
+ { XRFaceTracker::FT_MOUTH_STRETCH,
+ { XRFaceTracker::FT_MOUTH_STRETCH_RIGHT, XRFaceTracker::FT_MOUTH_STRETCH_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_MOUTH_DIMPLE,
+ { XRFaceTracker::FT_MOUTH_DIMPLE_RIGHT, XRFaceTracker::FT_MOUTH_DIMPLE_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_MOUTH_TIGHTENER,
+ { XRFaceTracker::FT_MOUTH_TIGHTENER_RIGHT, XRFaceTracker::FT_MOUTH_TIGHTENER_LEFT, -1, -1 } },
+ { XRFaceTracker::FT_MOUTH_PRESS,
+ { XRFaceTracker::FT_MOUTH_PRESS_RIGHT, XRFaceTracker::FT_MOUTH_PRESS_LEFT, -1, -1 } }
+ };
+
+ // Remove unified blend shapes if individual blend shapes are found.
+ for (const unified_blend_entry &entry : unified_blends) {
+ // Check if all individual blend shapes are found.
+ bool found = true;
+ for (const int i : entry.individual) {
+ if (i >= 0 && !p_blend_mapping.find(i)) {
+ found = false;
+ break;
+ }
+ }
+
+ // If all individual blend shapes are found then remove the unified blend shape.
+ if (found) {
+ p_blend_mapping.erase(entry.unified);
+ }
+ }
+}
+
+void XRFaceModifier3D::_bind_methods() {
+ ClassDB::bind_method(D_METHOD("set_face_tracker", "tracker_name"), &XRFaceModifier3D::set_face_tracker);
+ ClassDB::bind_method(D_METHOD("get_face_tracker"), &XRFaceModifier3D::get_face_tracker);
+ ADD_PROPERTY(PropertyInfo(Variant::STRING, "face_tracker", PROPERTY_HINT_ENUM_SUGGESTION, "/user/head"), "set_face_tracker", "get_face_tracker");
+
+ ClassDB::bind_method(D_METHOD("set_target", "target"), &XRFaceModifier3D::set_target);
+ ClassDB::bind_method(D_METHOD("get_target"), &XRFaceModifier3D::get_target);
+ ADD_PROPERTY(PropertyInfo(Variant::NODE_PATH, "target", PROPERTY_HINT_NODE_PATH_VALID_TYPES, "MeshInstance3D"), "set_target", "get_target");
+}
+
+void XRFaceModifier3D::set_face_tracker(const StringName &p_tracker_name) {
+ tracker_name = p_tracker_name;
+}
+
+StringName XRFaceModifier3D::get_face_tracker() const {
+ return tracker_name;
+}
+
+void XRFaceModifier3D::set_target(const NodePath &p_target) {
+ target = p_target;
+
+ if (is_inside_tree()) {
+ _get_blend_data();
+ }
+}
+
+NodePath XRFaceModifier3D::get_target() const {
+ return target;
+}
+
+MeshInstance3D *XRFaceModifier3D::get_mesh_instance() const {
+ if (!has_node(target)) {
+ return nullptr;
+ }
+
+ Node *node = get_node(target);
+ if (!node) {
+ return nullptr;
+ }
+
+ return Object::cast_to<MeshInstance3D>(node);
+}
+
+void XRFaceModifier3D::_get_blend_data() {
+ // This method constructs the blend mapping from the XRFaceTracker
+ // blend shapes to the available blend shapes of the target mesh. It does this
+ // by:
+ //
+ // 1. Identifying the blend shapes of the target mesh and identifying what
+ // XRFaceTracker blend shape they correspond to. The results are
+ // placed in the blend_mapping map.
+ // 2. Prevent over-driving facial blend-shapes by removing any unified blend
+ // shapes from the map if all the individual blend shapes are already
+ // found and going to be driven.
+
+ blend_mapping.clear();
+
+ // Get the target MeshInstance3D.
+ const MeshInstance3D *mesh_instance = get_mesh_instance();
+ if (!mesh_instance) {
+ return;
+ }
+
+ // Get the mesh.
+ const Ref<Mesh> mesh = mesh_instance->get_mesh();
+ if (mesh.is_null()) {
+ return;
+ }
+
+ // Identify all face blend shapes and populate the map.
+ identify_face_blend_shapes(blend_mapping, mesh);
+
+ // Remove the unified blend shapes if all the individual blend shapes are found.
+ remove_driven_unified_blend_shapes(blend_mapping);
+}
+
+void XRFaceModifier3D::_update_face_blends() const {
+ // Get the XR Server.
+ const XRServer *xr_server = XRServer::get_singleton();
+ if (!xr_server) {
+ return;
+ }
+
+ // Get the face tracker.
+ const Ref<XRFaceTracker> p = xr_server->get_face_tracker(tracker_name);
+ if (!p.is_valid()) {
+ return;
+ }
+
+ // Get the face mesh.
+ MeshInstance3D *mesh_instance = get_mesh_instance();
+ if (!mesh_instance) {
+ return;
+ }
+
+ // Get the blend weights.
+ const PackedFloat32Array weights = p->get_blend_shapes();
+
+ // Apply all the face blend weights to the mesh.
+ for (const KeyValue<int, int> &it : blend_mapping) {
+ mesh_instance->set_blend_shape_value(it.value, weights[it.key]);
+ }
+}
+
+void XRFaceModifier3D::_notification(int p_what) {
+ switch (p_what) {
+ case NOTIFICATION_ENTER_TREE: {
+ _get_blend_data();
+ set_process_internal(true);
+ } break;
+ case NOTIFICATION_EXIT_TREE: {
+ set_process_internal(false);
+ blend_mapping.clear();
+ } break;
+ case NOTIFICATION_INTERNAL_PROCESS: {
+ _update_face_blends();
+ } break;
+ default: {
+ } break;
+ }
+}