From 059aa2984cb5ebe79d172ba35aba849e538d14fd Mon Sep 17 00:00:00 2001
From: zhangxin <zhangx@bao.ac.cn>
Date: Wed, 7 May 2025 11:23:00 +0800
Subject: [PATCH] update merge code

---
 mergeFiles.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/mergeFiles.py b/mergeFiles.py
index 3337fab..c362b5f 100644
--- a/mergeFiles.py
+++ b/mergeFiles.py
@@ -1,8 +1,15 @@
 import numpy as np
 import os
 from astropy.table import Table
+import sys
 
-multiFileDir = '/nfsdata/share/CSSOSDataProductsSims/C10_refCat/1/'
+if len(sys.argv) < 3:
+	# print('usage:\n'+sys.argv[0]+' survey_result.txt')
+	sys.exit(0)
+
+multiFileDir = str(sys.argv[1])
+outDir = str(sys.argv[2])
+# multiFileDir = '/nfsdata/share/CSSOSDataProductsSims/C10_refCat/1/'
 
 
 mergeKeys = ['parallax1',
@@ -51,7 +58,7 @@ for fn in allFiles:
         for key in mergeKeys:
             mergeTable[key] = mergeTable[key] + tTab[key]
 
-mergeTable.write(outFn,overwrite = True)
+mergeTable.write(os.path.join(outDir,outFn),overwrite = True)
 
 
 
-- 
GitLab