From e13bcbb5f4cd412dd96ec243782b117b0dd24495 Mon Sep 17 00:00:00 2001
From: Zengtudor <Zengtudor@outlook.com>
Date: Fri, 16 Aug 2024 13:54:46 +0800
Subject: [PATCH] update

---
 src/main.cpp | 47 ++++++++++++++++++++++++++++++++++++++++-------
 tests/1.in   |  1 +
 2 files changed, 41 insertions(+), 7 deletions(-)
 create mode 100644 tests/1.in

diff --git a/src/main.cpp b/src/main.cpp
index fadd8ca..463bee0 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -1,5 +1,6 @@
 #include "json/reader.h"
 #include "json/value.h"
+#include <chrono>
 #include <cstdlib>
 #include <filesystem>
 #include <fstream>
@@ -8,7 +9,9 @@
 #include<json/json.h>
 #include <ostream>
 #include <queue>
+#include <sstream>
 #include <string>
+#include <thread>
 
 #ifdef _WIN32
     #include<windows.h>
@@ -16,7 +19,7 @@
     #include "unistd.h"
 #endif
 
-using std::cerr,std::endl,std::cout,std::ifstream,std::string,std::vector,std::queue,std::filesystem::path;
+using std::cerr,std::endl,std::cout,std::ifstream,std::string,std::vector,std::queue,std::filesystem::path,std::thread,std::chrono::time_point,std::chrono::steady_clock,std::chrono::high_resolution_clock;
 
 #define AS_EQ(a,b){if((a)!=(b)){cerr<<"assert eq failed :"<<endl<<#a<<":"<<(a)<<endl<<#b<<":"<<(b)<<endl;exit(1);}}
 #define AS_NE(a,b){if((a)==(b)){cerr<<"assert not eq failed :"<<endl<<#a<<":"<<(a)<<endl<<#b<<":"<<(b)<<endl;exit(1);}}
@@ -47,6 +50,7 @@ std::filesystem::path getAbsolutePath(std::filesystem::path rootPath,std::filesy
 string getProtectPath(const std::filesystem::path &p);
 
 struct Test{
+    unsigned int id;
     std::filesystem::path in;
     std::filesystem::path ans;
     friend std::ostream& operator<<(std::ostream& os,Test &t){
@@ -55,6 +59,11 @@ struct Test{
     }
 };
 
+struct ProgramRet{
+    time_point<steady_clock> start;
+    time_point<steady_clock> end;
+};
+
 template<typename T>
 T getValueOrPanic(T value);
 
@@ -116,20 +125,44 @@ int main(int argc, char* argv[]) {
     if(jsonValue["tests"].size()==0)LOG("cannot find tests");
     LOG("parsing tests")
     vector<Test> tests;
-    for(auto i:jsonValue["tests"]){
+    auto &jsonTests = jsonValue["tests"];
+    for(unsigned int i=1;i<=jsonTests.size();i++){
         tests.push_back({
-            .in=getAbsolutePath(projectPath, getValueOrPanic(i["in"]).asString()),
-            .ans=getAbsolutePath(projectPath,getValueOrPanic(i["ans"]).asString())
+            .id = i,
+            .in=getAbsolutePath(projectPath, getValueOrPanic(jsonTests[i-1]["in"]).asString()),
+            .ans=getAbsolutePath(projectPath,getValueOrPanic(jsonTests[i-1]["ans"]).asString()),
         });
     }
-    queue<std::function<void()>> tasks;
+    queue<std::function<ProgramRet()>> tasks;
+    const auto testsPath = projectPath/"tests";
+    printValue(testsPath)
     const string exePathStr = getProtectPath(exePath);
     for(auto i:tests){
-        tasks.push([i,&exePathStr](){
+        tasks.push([i,&exePathStr,&testsPath]()->ProgramRet{
             string commands;
-            commands = commands+exePathStr+" < "+getProtectPath(i.in)+" > ";
+            std::stringstream ss;
+            ss<<i.id;
+            string id_str;
+            ss>>id_str;
+            path outPath = testsPath/(id_str+".out") ;
+            commands = commands+exePathStr+" < "+getProtectPath(i.in)+" > "+getProtectPath(outPath);
+            LOG(commands)
+            auto start = high_resolution_clock::now();
+            system(commands.c_str());
+            auto end=high_resolution_clock::now();
+            return ProgramRet{
+                .start=start,
+                .end = end
+            };
         });
     }
+    while(tasks.size()>0){
+        auto i = tasks.front();
+        tasks.pop();
+        thread newThread(i);
+        AS_EQ(newThread.joinable(),true);
+        newThread.join();
+    }
 }
 
 string getProtectPath(const std::filesystem::path &p){
diff --git a/tests/1.in b/tests/1.in
new file mode 100644
index 0000000..56a6051
--- /dev/null
+++ b/tests/1.in
@@ -0,0 +1 @@
+1
\ No newline at end of file