From 7168f51d42fc9dc5e681717a0dff6839ab8c707d Mon Sep 17 00:00:00 2001
From: shuaiqing <s_q@zju.edu.cn>
Date: Mon, 10 Jul 2023 22:11:38 +0800
Subject: [PATCH] update the entrypoint

---
 apps/mocap/run.py                          |   8 +-
 scripts/publish/sync_Learning_EasyMocap.py | 200 +++++++++++++++++++++
 2 files changed, 207 insertions(+), 1 deletion(-)
 create mode 100644 scripts/publish/sync_Learning_EasyMocap.py

diff --git a/apps/mocap/run.py b/apps/mocap/run.py
index f436dd6..a88460d 100644
--- a/apps/mocap/run.py
+++ b/apps/mocap/run.py
@@ -35,10 +35,14 @@ def update_exp_by_args(cfg_exp, args):
                 opts_alias.append(cfg_exp.alias[args.opt_exp[i*2]])
                 opts_alias.append(args.opt_exp[i*2+1])
         cfg_exp.merge_from_list(opts_alias)
-    if args.skip_vis:
+    if args.skip_vis or args.skip_vis_step:
         for key, val in cfg_exp.args.at_step.items():
             if key.startswith('vis'):
                 val.skip = True
+    if args.skip_vis or args.skip_vis_final:
+        for key, val in cfg_exp.args.at_final.items():
+            if key.startswith('vis'):
+                val.skip = True    
 
 def load_cfg_from_file(cfg, args):
     cfg = Config.load(cfg)
@@ -71,6 +75,8 @@ def main_entrypoint():
     parser.add_argument('--cameras', type=str, default=None, help='Camera file path')
     parser.add_argument('--out', type=str, default=None)
     parser.add_argument('--skip_vis', action='store_true')
+    parser.add_argument('--skip_vis_step', action='store_true')
+    parser.add_argument('--skip_vis_final', action='store_true')
     parser.add_argument('--debug', action='store_true')
     args = parser.parse_args()
 
diff --git a/scripts/publish/sync_Learning_EasyMocap.py b/scripts/publish/sync_Learning_EasyMocap.py
new file mode 100644
index 0000000..5e5302b
--- /dev/null
+++ b/scripts/publish/sync_Learning_EasyMocap.py
@@ -0,0 +1,200 @@
+filetree_l = '''
+apps:
+    - mocap:
+        - run.py
+myeasymocap:
+    - datasets:
+        - basedata.py
+        - mv1p.py
+        - sv1p.py
+      io:
+        - model.py
+        - video.py
+        - vis.py
+        - vis3d.py
+        - write.py
+      operations:
+        - init.py
+        - loss.py
+        - match.py
+        - merge.py
+        - optimizer.py
+        - select.py
+        - smooth.py
+        - triangulate.py
+        - match_base.py
+        - iterative_triangulate.py
+      stages:
+        - basestage.py
+        - collect.py
+      backbone:
+        - basetopdown.py
+        - topdown_keypoints.py
+        - yolo:
+            - yolo.py
+          vitpose:
+            - layers.py
+            - vit_moe.py
+          hrnet:
+            - __init__.py
+            - modules.py
+            - hrnet.py
+            - myhrnet.py
+          pare:
+            - pare.py
+            - config.py
+            - constants.py
+            - backbone:
+                - __init__.py
+                - resnet.py
+                - mobilenet.py
+                - hrnet.py
+                - utils.py
+              head:
+                - __init__.py
+                - pare_head.py
+                - hmr_head.py
+                - smpl_head.py
+                - smpl_cam_head.py
+              layers:
+                - __init__.py
+                - locallyconnected2d.py
+                - interpolate.py
+                - nonlocalattention.py
+                - keypoint_attention.py
+                - coattention.py
+                - softargmax.py
+                - non_local:
+                  - __init__.py
+                  - dot_product.py
+              utils:
+                - geometry.py
+                - kp_utils.py
+          mediapipe:
+            - hand.py
+          hand2d:
+            - __init__.py
+            - hand2d.py
+            - resnet.py
+          hmr:
+            - __init__.py
+            - hmr.py
+            - models.py
+            - hmr_api.py        
+config:
+    - datasets:
+        - mvimage.yml
+        - svimage.yml
+      mv1p:
+        - detect_triangulate.yml
+        - detect_triangulate_fitSMPL.yml
+        - detect_hand_triangulate.yml
+        - detect_hand_triangulate_fitMANO.yml
+      1v1p:
+        - hrnet_pare_finetune.yml
+        - hand_detect_finetune.yml
+        - fixhand.yml
+'''
+
+filetree_e = '''
+3rdparty:
+  - eigen-3.3.7
+  - pybind11
+library:
+    - pymatch:
+        - setup.py
+        - CMakeLists.txt
+        - python:
+            - __init__.py
+            - CMakeLists.txt
+            - pymatchlr.cpp
+        - include:
+            - base.h
+            - projfunc.hpp
+            - Timer.hpp
+            - visualize.hpp
+            - matchSVT.hpp
+        - pymatchlr:
+            - __init__.py
+apps:
+    - calibration:
+        - vis_camera_by_open3d.py
+easymocap:
+    - multistage:
+        - gmm.py
+        - gmm_08.pkl
+      mytools:
+        - vis_base.py
+        - camera_utils.py
+'''
+
+sync_config = {
+    'EasyMocap': {
+        'src': '../EasyMocapPublic',
+        'dst': '.',
+        'filetree': filetree_e
+    },
+    'Learn': {
+        'src': '../Learning_EasyMocap',
+        'dst': '.',
+        'filetree': filetree_l
+    }
+}
+import os
+import shutil
+import yaml
+from os.path import join
+from easymocap.mytools.debug_utils import log, mywarn
+
+def compare_files(file1, file2):
+    with open(file1, 'r') as f1, open(file2, 'r') as f2:
+        content1 = f1.read()
+        content2 = f2.read()
+        if content1 == content2:
+            return True
+        else:
+            return False
+
+def copy_node(dir, nodes):
+    for node in nodes:
+        if isinstance(node, str):
+            srcname = join(SRC, dir, node)
+            dstname = join(DST, dir, node)
+            if not os.path.exists(srcname):
+                mywarn('Not exists {}'.format(srcname))
+                continue
+            if os.path.exists(dstname):
+                if os.path.isdir(srcname):
+                    mywarn('Current not support overwrite folders: {}'.format(dstname))
+                else:
+                    if compare_files(srcname, dstname):
+                        pass
+                    else:
+                        mywarn('Overwrite file: {}'.format(dstname))
+                        shutil.copyfile(srcname, dstname)
+            else:
+                if os.path.isdir(srcname):
+                    log('Copy dir: {}'.format(dstname))
+                    shutil.copytree(srcname, dstname)
+                else:
+                    os.makedirs(join(DST, dir), exist_ok=True)
+                    log('Copy file: {}'.format(dstname))
+                    shutil.copyfile(srcname, dstname)
+        elif isinstance(node, dict):
+            for subdir, subnode in node.items():
+                copy_node(join(dir, subdir), subnode)
+    
+if __name__ == '__main__':
+    for cfg_name, cfg in sync_config.items():
+        SRC = cfg['src']
+        DST = cfg['dst']
+        filetree = cfg['filetree']
+        filetree = yaml.safe_load(filetree)
+        SRC = os.path.abspath(SRC)
+        DST = os.path.abspath(DST)
+        os.chdir(SRC)
+        cmd = 'git pull origin master'
+        os.system(cmd)
+        os.chdir(DST)
+        for dir, files in filetree.items():
+            copy_node(dir, files)