chart_gen.py

import math
def read_file(f):
	
	f.readline() #header
	header = f.readline()
	header = header.split(",")[1:]
	data_sets = {h.strip():[] for h in header}
	for line in f.readlines():
		entries = line.split(",")
		# print(header)
		# print(entries)
		for i in range(len(entries))[1:]:
			# data_sets[header[i-1].strip()].append((float(entries[0]), float(entries[i].strip())))
			data_sets[header[i-1].strip()].append((float(entries[0]), process(float(entries[i].strip()))))
	return data_sets

def process(x):
	# return safe_log(x)
	return max(x,1e-6) 
	# return x
def safe_log(x):
	if x == 0:
		return -5
	else:
		return math.log10(x) if math.log10(x) >= -5 else -5

def write_datasets(data_sets, ident):
	X_vals = []
	for pair in list(data_sets.values())[0]:
		X_vals.append(pair[0])
	# X_val = [key for key in list(data_sets.values())[0]]
	print(f"X{ident} = {X_vals};")
	labels = [f"X{ident}"]
	for ds in data_sets:
		Y_vals = []
		for pair in data_sets[ds]:
			Y_vals.append(pair[1])
		labels.append(f"Y{ds}{ident}")
		# if "NoCAD" in labels[-1] and "NoTN" in labels[-1]:
		print(f"Y{ds}{ident} = {Y_vals};")
		
	return labels



		
def write_graphs(labels, ident):
	styles = ["r-.*","b--o", "m:+","r-s","b-d", "m-p"]
	arg_str = ""
	for i in range(1,len(labels)):
		arg_str+=("," if arg_str else "" ) +",".join([labels[0],labels[i], f"'{styles[i]}'"])

	print("figure('units','normalized','outerposition',[0 0 1 1])")
	# print(f"plot({arg_str})")
	print(f"semilogy({arg_str})")
	print(f'title("{ident}")')
	print("set(findall(gcf,'-property','FontSize'),'FontSize',40)")

	legendString = ",".join([f'"{label[1:label.find("Inf")]}"' for label in labels[1:]])
	legend_array = []
	# print(labels[1:])
	for label in labels[1:]:
		legend_array.append("+".join((["CAD"] if not "NoCAD" in label else []) + (["Seg."] if not "NoSeg" in label else [])))
		if not legend_array[-1]:
			legend_array[-1] = "Control"
	# print(legend_array)
	legendString = ",".join([f'"{legend}"' for legend in legend_array])
	# legendString = ",".join([f'"{"+".join(["CAD"] if not "NoCAD" in label else [] + ["Seg."] if not "NoSeg" in label else [])}' for label in labels[1:]])
	print(f"legend([{legendString}],'FontSize',28, 'Location', 'northeast')")
	print("ylabel('Key Rate')")
	print("xlabel('Decoherence prob.')")

def save_graphs(output_dir, ident):
	print("fig = gcf;\nfig.PaperSize;\nfig.PaperPositionMode = 'auto';\nfig_pos = fig.PaperPosition;\nfig.PaperSize = [11 fig_pos(4)-1];")
	print(f"print('{os.getcwd()+'/'+output_dir+ident}.pdf','-dpdf', '-fillpage')")


import os 
def main():
	

	# input_dir = "../data_backup/CADnPooling/data"
	# output_dir = "../data_backup/CADnPooling/data/graphs/"

	input_dir = "CADnPooling/data"
	output_dir = "CADnPooling/data/graphs/"
	try:
		os.mkdir(output_dir)
	except:
		pass
	print("close all")
	all_labels = []
	for filename in os.listdir(input_dir):
	# filename = "test.csv"
		ident = filename[:-4]
		try:
			with open(f"{input_dir}/{filename}","r") as f:
				data_sets = read_file(f)
		except:
			continue
		labels = write_datasets(data_sets, ident)
		all_labels += labels
		write_graphs(labels, ident)
		save_graphs(output_dir, ident)
		print("\n\n\n")
	crosses = [(TN, Type) for Type in ["Global", "Local"] for TN in ["NoTN", "Central", "Diagonal"]]
	
	for cross in crosses:
		temp = []
		for label in all_labels:
			if cross[0] in label and cross[1] in label:
				temp.append(label)
		temp.sort(key = lambda x: x.count("No"))
		temp.sort(key = lambda x: x.count("NoCAD"))
		# print(temp)
		temp = [all_labels[0]] + temp[::-1]
		ident = cross[0]+cross[1]
		write_graphs(temp, ident)
		save_graphs(output_dir, ident)
		print("\n\n\n")
		

main()