initial commit: adx loop parser

adx/adx.py

@@ -0,0 +1,44 @@
+import glob
+import json
+import os
+import traceback
+from .adx_parser import adx_file
+
+def extract_loop_data_from_dir(dir_in, fout):
+	if fout is None:
+		fout = "adx_loop.json"
+	files = glob.glob(os.path.join(dir_in, "**/*.adx"), recursive=True)
+	extract_loop_data_from_files(files, fout)
+
+def extract_loop_data_from_files(files_in, fout):
+	if not files_in:
+		raise FileNotFoundError("No input files found")
+	collect = {}
+	for x in files_in:
+		res = extract_loop_data_from_file(x)
+		if res is not None:
+			collect[os.path.splitext(os.path.basename(x))[0]] = res
+	with open(fout, 'w') as out:
+		json.dump(collect, out, ensure_ascii=False, indent="\t", sort_keys=True)
+
+def extract_loop_data_from_file(file_in, fout=None):
+	if not file_in.endswith(".adx"):
+		raise ValueError("Input file not .adx")
+
+	try:
+		with open(file_in, 'rb') as x:
+			data = x.read(64)
+			ADX = adx_file(data)
+			if ADX.l_exists and ADX.is_valid:
+				res = ADX.parse_loop_data()
+			else:
+				res = None
+	except Exception as e:
+		traceback.print_exc()
+		print("Error processing file: " + file_in)
+		return None
+
+	if fout is not None and res is not None:
+		with open(fout, 'w') as out:
+			json.dump(res, out, ensure_ascii=False, indent="\t", sort_keys=True)
+	return res