44 lines
2.0 KiB
JavaScript
44 lines
2.0 KiB
JavaScript
import { parseFreeMoCapCsv } from '../src/studio/freemocapImport.js';
|
|
import { done, ok, section } from './harness.mjs';
|
|
|
|
const points = [
|
|
'left_ear', 'right_ear', 'left_shoulder', 'right_shoulder',
|
|
'left_elbow', 'right_elbow', 'left_wrist', 'right_wrist',
|
|
'left_hip', 'right_hip', 'left_knee', 'right_knee',
|
|
'left_ankle', 'right_ankle', 'left_heel', 'right_heel',
|
|
'left_foot_index', 'right_foot_index',
|
|
];
|
|
|
|
section('FreeMoCap tidy XYZ data imports into MediaPipe landmark order');
|
|
{
|
|
const lines = ['frame,keypoint,x,y,z,model,trajectory,reprojection_error'];
|
|
for (const trajectory of ['3d_xyz', 'rigid_3d_xyz']) {
|
|
for (const frame of [0, 1]) {
|
|
for (const [index, point] of points.entries()) {
|
|
const base = trajectory === 'rigid_3d_xyz' ? 200 : 100;
|
|
lines.push(`${frame},${point},${base + index},20,300,mediapipe_body,${trajectory},0.5`);
|
|
}
|
|
}
|
|
}
|
|
// Non-body rows in the tidy all-data file must not contaminate the pose.
|
|
lines.push('0,left_wrist,999,999,999,mediapipe_left_hand,rigid_3d_xyz,0.5');
|
|
const result = parseFreeMoCapCsv(lines.join('\n'));
|
|
ok(result.frames.length === 2, 'all complete body frames import');
|
|
ok(result.trajectory === 'rigid_3d_xyz', 'rigid body trajectory is preferred when available');
|
|
ok(result.tracker === 'MediaPipe', 'tracker metadata is reported');
|
|
const leftEar = result.frames[0].landmarks[7];
|
|
ok(leftEar.x === 200 && leftEar.y === -300 && leftEar.z === -20, 'FreeMoCap Z-up axes convert for the Tilt retargeter');
|
|
}
|
|
|
|
section('FreeMoCap per-trajectory CSV and RTMPose toe names are accepted');
|
|
{
|
|
const rtmposePoints = points.map((point) => point.replace('foot_index', 'big_toe'));
|
|
const lines = ['frame,keypoint,x,y,z'];
|
|
for (const [index, point] of rtmposePoints.entries()) lines.push(`4,${point},${index},10,20`);
|
|
const result = parseFreeMoCapCsv(lines.join('\r\n'));
|
|
ok(result.frames[0].frame === 4, 'source frame numbers survive import');
|
|
ok(result.frames[0].landmarks[31].visibility === 1, 'RTMPose big toe maps to the Tilt foot-index landmark');
|
|
}
|
|
|
|
done('FreeMoCap import');
|