From b66e839b32f014db326c0b79a7fb14202b1f8d61 Mon Sep 17 00:00:00 2001
From: YiHui Liu <liuyihui02@gmail.com>
Date: Thu, 7 Jul 2022 02:09:19 +0800
Subject: [PATCH] change: output to python

---
 .gitignore |  4 ++++
 main.cpp   | 32 ++++++++++++++++++++++----------
 2 files changed, 26 insertions(+), 10 deletions(-)

diff --git a/.gitignore b/.gitignore
index e3021eb..85dae8f 100644
--- a/.gitignore
+++ b/.gitignore
@@ -3,8 +3,12 @@
 *.log
 *.csv
 *.json
+*.txt
 2016Q3D/
 
+# env
+venv/
+
 # build cache
 build/
 
diff --git a/main.cpp b/main.cpp
index d00877b..bfe513e 100644
--- a/main.cpp
+++ b/main.cpp
@@ -8,28 +8,40 @@
 using namespace std;
 
 int main() {
-    int n;
+    int n, id, E;
     string run;
     FileHandler *FH;
     BlockHandler *BH;
-    CsvReader cR("config2.csv");
+    CsvReader CR("config1.csv");
+
+    CR.readData();
+    n = CR.rows();
+    E = stoi(CR(0, 3));
 
-    cR.readData();
-    n = cR.rows();
     FH = new FileHandler[n - 1];
+    BH = new BlockHandler[5];
+    for (int i = 0; i < 5; i++) BH[i] = BlockHandler(i);
 
     for (int i = 1; i < n; i++) {
-        run = cR(i, 0);
+        run = CR(i, 0);
         FH[i - 1] = FileHandler("2016Q3D/root/raw/201609Q3D" + run + ".root", 5);
-        FH[i - 1].pX = stoi(cR(i, 5));
+        FH[i - 1].pX = stoi(CR(i, 2));
         FH[i - 1].readData();
+        BH[stoi(CR(i, 1))].addData(FH[i - 1]);
     }
 
-    BH = new BlockHandler[5];
+    for (int i = 0; i < 5; i++) BH[i].splitData();
+
+    string s;
     for (int i = 0; i < 5; i++) {
-        BH[i] = BlockHandler(4 - i);
-        for (int j = i * 7; j < i * 7 + 7; j++) BH[i].addData(FH[j]);
-        BH[i].splitData();
+        for (int j = 0; j < 8; j++) {
+            s = "result/bind/" + to_string(E) + "-" + to_string(i) + "-" + to_string(j) + ".txt";
+            std::ofstream ofs(s);
+            for (int k = 0; k < BH[i].bind[j].data.size(); k++) {
+                auto p = BH[i].bind[j].data.at(k);
+                ofs << p(0) << " " << p(1) << std::endl;
+            }
+        }
     }
 
     return 0;