From 97ea6d973f70dc718dc0a5046eddd24f387d994b Mon Sep 17 00:00:00 2001
From: zengchen <1737460608@qq.com>
Date: Wed, 22 Dec 2021 13:02:47 +0800
Subject: [PATCH] Add cwrapper/wrapp.py

---
 cwrapper/wrapp.py | 46 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 cwrapper/wrapp.py

diff --git a/cwrapper/wrapp.py b/cwrapper/wrapp.py
new file mode 100644
index 0000000..02591e2
--- /dev/null
+++ b/cwrapper/wrapp.py
@@ -0,0 +1,46 @@
+#!/usr/bin/python3
+import argparse
+import subprocess
+import os
+
+RUNNER = "wrapper_runner.py"
+OUT_PATH = "fixed/output.java"
+N_TRIES = 10
+
+def wrapper(inp_file_path, out_file_path=OUT_PATH, times_to_try_compile=N_TRIES):
+	subprocess.run(['./'+RUNNER, inp_file_path, out_file_path, str(times_to_try_compile)], stderr=subprocess.PIPE).stderr.decode('utf-8')
+
+def tmp_run(inp_file_path, out_file_path):
+	#print("beginerror",["clang", inp_file_path, "-emit-llvm -S -c -o", out_file_path])
+	#cmd = "clang " + inp_file_path + " -emit-llvm -S -c -o " + out_file_path
+	cmd = ["clang", inp_file_path, "-emit-llvm", "-S", "-c", "-o", out_file_path]
+	#a = os.system(cmd)
+	print("ok")
+	return subprocess.run(cmd, stderr=subprocess.PIPE).stderr.decode('utf-8')
+
+def run():
+	input_prefix = "benchmark/bigclonebenchdata_partial/"
+	out_prefix = "benchmark/output/e"
+	for idx in range(100000000):
+		input_file = input_prefix + str(idx) + ".java"
+		if not os.path.exists(input_file):
+			continue
+		out_file = out_prefix + str(idx) + ".java"
+		TIMES_TO_TRY_COMPILE = 20
+		wrapper(input_file, out_file, TIMES_TO_TRY_COMPILE)
+
+if __name__ == '__main__':
+	
+	parser = argparse.ArgumentParser()
+
+	parser.add_argument('-f', required=True, help="path to input c code snippet")
+	parser.add_argument('-o', default=OUT_PATH, help="path to output c file. default=" + OUT_PATH)
+	parser.add_argument('-n', type=int, default=N_TRIES, help="max number of times to try compile. default=10")
+	args = parser.parse_args()
+
+	INPUT_FILE = args.f
+	OUTPUT_FILE = args.o
+	TIMES_TO_TRY_COMPILE = int(args.n)
+	
+
+	wrapper(INPUT_FILE, OUTPUT_FILE, TIMES_TO_TRY_COMPILE)
\ No newline at end of file