pose2sim/Pose2Sim/S00_Demo_BatchSession/tests.py

#! /usr/bin/env python
# -*- coding: utf-8 -*-


'''
    ########################################
    ## Pose2Sim tests                     ##
    ########################################
    
    - BATCH SESSION:
        - Calibration conversion from .qca.txt
        - Single person:
            - synchronization
            - person association
            - triangulation
            - filtering
            - marker augmentation
        - Multi-person:
            - synchronization
            - person association
            - triangulation
            - filtering
            - marker augmentation
            
    - SINGLE TRIAL:
        - calibration
        - synchronization
        - person association
        - triangulation
        - filtering
        - marker augmentation

    N.B.: Calibration from scene dimensions is not tested, as it requires the 
    user to click points on the image. 
    Not all possible configuration parameters are extensively tested.
    Pose estimation, scaling, inverse kinematics are not tested yet.
    
    Usage: 
    from Pose2Sim.S00_Demo_BatchSession import tests; tests.test_workflow()
    python tests.py 
'''

## INIT
import os
import toml
from Pose2Sim import Pose2Sim


## AUTHORSHIP INFORMATION
__author__ = "David Pagnon"
__copyright__ = "Copyright 2021, Pose2Sim"
__credits__ = ["David Pagnon"]
__license__ = "BSD 3-Clause License"
__version__ = "0.8.2"
__maintainer__ = "David Pagnon"
__email__ = "contact@david-pagnon.com"
__status__ = "Development"


## FUNCTIONS
def test_workflow():
    '''
    - BATCH SESSION:
        - Calibration
        - Single person:
            - synchronization
            - person association
            - triangulation
            - filtering
            - marker augmentation
        - Multi-person:
            - synchronization
            - person association
            - triangulation
            - filtering
            - marker augmentation
            
    - SINGLE TRIAL:
        - calibration
        - synchronization
        - person association
        - triangulation
        - filtering
        - marker augmentation

    N.B.: Calibration from scene dimensions is not tested, as it requires the 
    user to click points on the image. 
    Not all possible configuration parameters are extensively tested.
    Pose estimation, scaling, inverse kinematics are not tested yet.
    
    Usage: 
    from Pose2Sim.S00_Demo_BatchSession import tests; tests.test_workflow()
    python tests.py 
    '''
    
    config_dict = toml.load('Config.toml')
    
    
    #################
    # BATCH SESSION #
    #################
    
    ###############
    # Calibration #
    ###############
    
    config_dict.get("project").update({"project_dir":"."})
    Pose2Sim.calibration(config_dict)
    # edit config_dict for calibration based on keypoints
    # Pose2Sim.calibration(config_dict)
    
    
    #################
    # Single person #
    #################
    
    # STATIC TRIAL
    project_dir = os.path.join("S00_P00_SingleParticipant","S00_P00_T00_StaticTrial")
    config_dict.get("project").update({"project_dir":project_dir})
    config_dict.get("synchronization").update({"reset_sync":True})
    # Pose2Sim.poseEstimation(config_dict)
    Pose2Sim.synchronization(config_dict)
    Pose2Sim.personAssociation(config_dict)
    Pose2Sim.triangulation(config_dict)
    Pose2Sim.filtering(config_dict)
    Pose2Sim.markerAugmentation(config_dict)
    # Pose2Sim.kinematics(config_dict)

    # BALANCING TRIAL
    project_dir = os.path.join("S00_P00_SingleParticipant","S00_P00_T01_BalancingTrial")
    config_dict.get("project").update({"project_dir":project_dir})
    config_dict['filtering']['display_figures'] = False
    # Pose2Sim.poseEstimation(config_dict)
    Pose2Sim.synchronization(config_dict)
    Pose2Sim.personAssociation(config_dict)
    Pose2Sim.triangulation(config_dict)
    Pose2Sim.filtering(config_dict)
    Pose2Sim.markerAugmentation(config_dict)
    # Pose2Sim.kinematics(config_dict)
    
    
    ################
    # Multi person #
    ################
    
    # STATIC TRIAL PERSON 1
    project_dir = os.path.join("S00_P01_MultiParticipants","S00_P01_T00_StaticTrialParticipant1")
    config_dict.get("project").update({"project_dir":project_dir})
    config_dict.get("markerAugmentation").update({"participant_height":1.21})
    config_dict.get("markerAugmentation").update({"participant_mass":25.0})
    # Pose2Sim.poseEstimation(config_dict)
    Pose2Sim.synchronization(config_dict)
    Pose2Sim.personAssociation(config_dict)
    Pose2Sim.triangulation(config_dict)
    Pose2Sim.filtering(config_dict)
    Pose2Sim.markerAugmentation(config_dict)
    # Pose2Sim.kinematics(config_dict)

    # STATIC TRIAL PERSON 2
    project_dir = os.path.join("S00_P01_MultiParticipants","S00_P01_T01_StaticTrialParticipant2")
    config_dict.get("project").update({"project_dir":project_dir})
    config_dict.get("markerAugmentation").update({"participant_height":1.72})
    config_dict.get("markerAugmentation").update({"participant_mass":70.0})
    # Pose2Sim.poseEstimation(config_dict)
    Pose2Sim.synchronization(config_dict)
    Pose2Sim.personAssociation(config_dict)
    Pose2Sim.triangulation(config_dict)
    Pose2Sim.filtering(config_dict)
    Pose2Sim.markerAugmentation(config_dict)
    # Pose2Sim.kinematics(config_dict)

    # BALANCING & YOGA TRIAL
    project_dir = os.path.join("S00_P01_MultiParticipants","S00_P01_T02_Participants1-2")
    config_dict.get("project").update({"project_dir":project_dir})
    config_dict.get("project").update({"multi_person":True})
    config_dict.get("markerAugmentation").update({"participant_height":[1.21, 1.72]})
    config_dict.get("markerAugmentation").update({"participant_mass":[25.0, 70.0]})
    config_dict['triangulation']['reorder_trc'] = False
    # Pose2Sim.poseEstimation(config_dict)
    Pose2Sim.synchronization(config_dict)
    Pose2Sim.personAssociation(config_dict)
    Pose2Sim.triangulation(config_dict)
    Pose2Sim.filtering(config_dict)
    Pose2Sim.markerAugmentation(config_dict)
    # Pose2Sim.kinematics(config_dict)
    
    
    #################
    # SINGLE TRIAL  #
    #################
    
    config_dict = toml.load('../S01_Demo_SingleTrial/Config.toml')
    project_dir = os.path.join("../S01_Demo_SingleTrial")
    os.chdir(project_dir)
    config_dict.get("project").update({"project_dir":project_dir})
    config_dict.get("synchronization").update({"display_sync_plots":False})
    config_dict['filtering']['display_figures'] = False
    Pose2Sim.calibration(config_dict)
    # Pose2Sim.poseEstimation(config_dict)
    Pose2Sim.synchronization(config_dict)
    Pose2Sim.personAssociation(config_dict)
    Pose2Sim.triangulation(config_dict)
    Pose2Sim.filtering(config_dict)
    Pose2Sim.markerAugmentation(config_dict)
    # Pose2Sim.kinematics(config_dict)
    
    
if __name__ == '__main__':
    test_workflow()