def data_to_features()

in src/featuresets.py [0:0]

• 22 lines of code
• 9 McCabe index (conditional complexity)

    def data_to_features(self):
        features = []
        l_finger_idx = self._get_lfinger_idx()
        r_finger_idx = self._get_rfinger_idx()
        for idx in self._get_joint_idx():
            if idx in l_finger_idx:
                features.append(d.JointLocalPosition(idx, ref_idx=self.l_wrist_idx))
            elif idx in r_finger_idx:
                features.append(d.JointLocalPosition(idx, ref_idx=self.r_wrist_idx))
            elif idx == self.head_idx:
                features.append(d.PositionNoOcc(idx))
            else:
                features.append(d.JointLocalPosition(idx, ref_idx=self.head_idx))

        if self.replace_occluded is not None:
            for idx in self._get_joint_idx():
                if idx == self.head_idx:
                    features.append(d.NeverOccluded(idx))
                elif self.use_binary_occ:
                    features.append(d.IsOccluded(idx))
                else:
                    features.append(d.IsOccludedSinceZeroOffset(idx))

        return d.FeaturePacker(features, self.replace_occluded, name=self.file_name())