Also check for Trinity-v4 spine0 (#766)

Summary:
Pull Request resolved: #766

Update `getUpperBodyJoints` to handle more skeleton definitions.

Reviewed By: cdtwigg

Differential Revision: D85957988

fbshipit-source-id: 61cf21ddfd386942be6ba4312c1d40761c4bdfec

Author: Guillermo Arto Sanchez

pymomentum/geometry/momentum_geometry.cpp

@@ -550,7 +550,11 @@ std::vector<bool> listToBitset(const std::vector<size_t>& list, const size_t sz)
 
 std::vector<size_t> getUpperBodyJoints(const momentum::Skeleton& skeleton) {
   auto upperBodyRoot_idx = skeleton.getJointIdByName("b_spine0");
-  MT_THROW_IF(upperBodyRoot_idx == momentum::kInvalidIndex, "Missing 'b_spine0' joint.");
+  if (upperBodyRoot_idx == momentum::kInvalidIndex) { // Check for different skeleton definition
+    upperBodyRoot_idx = skeleton.getJointIdByName("c_spine0");
+  }
+  MT_THROW_IF(
+      upperBodyRoot_idx == momentum::kInvalidIndex, "Missing 'b_spine0' & 'c_spine0' joint.");
 
   std::vector<size_t> result;