From a6e1fbf37e6d7b942fbb2d27bc29da756f284ea5 Mon Sep 17 00:00:00 2001
From: marques <20172333133@m.scnu.edu.cn>
Date: Wed, 20 Apr 2022 22:17:22 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=91=BD=E4=BB=A4=E8=A1=8C?=
 =?UTF-8?q?=E4=BC=A0=E5=85=A5=E5=8F=82=E6=95=B0=E6=94=AF=E6=8C=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Main_Quality_Relabel.py | 19 +++++++++++++++----
 README.md               |  7 ++++++-
 2 files changed, 21 insertions(+), 5 deletions(-)

diff --git a/Main_Quality_Relabel.py b/Main_Quality_Relabel.py
index 09d420f..5475c5b 100644
--- a/Main_Quality_Relabel.py
+++ b/Main_Quality_Relabel.py
@@ -7,6 +7,7 @@
 @email:2021022362@m.scnu.edu.cn
 @time:2022/03/28
 """
+import argparse
 from pathlib import Path
 from utils.Quality_Relabel import Quality_Relabel
 
@@ -40,9 +41,19 @@ sampNo = 670
 start_bcg_index = 0
 
 if __name__ == '__main__':
-    qualityRelabel = Quality_Relabel(all_path=all_path, sampNo=sampNo, frequency=frequency, bcg_frequency=bcg_frequency,
+    parser = argparse.ArgumentParser()
+    parser.add_argument('-n', '--sampNo', nargs='?', type=int, default=sampNo, help='samp number')
+    parser.add_argument('-i', '--start_bcg_index', nargs='?', type=int, default=start_bcg_index, help='samp number')
+    parser.add_argument('-f', '--front_add_second', nargs='?', type=int, default=front_add_second,
+                        help='Add x second signal in front of the event')
+    parser.add_argument('-b', '--back_add_second', nargs='?', type=int, default=back_add_second,
+                        help='Add x second signal after the event')
+    opt = parser.parse_args()
+
+    qualityRelabel = Quality_Relabel(all_path=all_path, sampNo=opt.sampNo, frequency=frequency,
+                                     bcg_frequency=bcg_frequency,
                                      focus_event_list=focus_event_list)
 
-    qualityRelabel.show_all_event(start_bcg_index=start_bcg_index,
-                                  front_add_second=front_add_second,
-                                  back_add_second=back_add_second)
+    qualityRelabel.show_all_event(start_bcg_index=opt.start_bcg_index,
+                                  front_add_second=opt.front_add_second,
+                                  back_add_second=opt.back_add_second)
diff --git a/README.md b/README.md
index 2e1a1ae..57cd032 100644
--- a/README.md
+++ b/README.md
@@ -67,4 +67,9 @@ Quality_Relabel                     主目录
 
 2022-04-15 22：06
 
-​	1. 删除了PSG开头部分不存在于心晓中的标签，以后不用计算shifting
+	1. ~~删除了PSG开头部分不存在于心晓中的标签，以后不用计算shifting~~
+
+2022-04-17 20：00
+
+​	1. 添加命令行传入参数功能
+