import numpy as np
import os 

data_folder= "normal_g2_files"
save_folder = "g2_merged_data"

# cross_EYP-DFB-0780-110mA_25degC_2_emitter_400m_dealy_g2x-186m_ts_130ns_delay_alpha-0p000_20240719_0324
# file_start_string = "cross_EYP-DFB-0780-100mA_25degC_C230_collimated_A230_MOT2N_6p4V_DG10-1500-PO1_A230_30cm_bare_SMF_fibre_near_center_288ns_delay_MPD_20231026_1154_time_flipped"
# length_file_start_string = len(file_start_string)

for filename in os.listdir(data_folder):
	if filename[-4] == "d":
		continue
	print(filename)
	
	# alpha = filename[length_file_start_string:length_file_start_string+3]

	time_flipped_filename = filename[:-3]+"_time_flipped.g2"

	normal_time_arr = np.genfromtxt(os.path.join(data_folder,filename),skip_header=3).T
	flipped_time_arr = np.genfromtxt(os.path.join(data_folder,time_flipped_filename),skip_header=3).T
	flipped_time_arr = np.fliplr(flipped_time_arr)

	flipped_time_arr[0] = -1*flipped_time_arr[0]

	joint_arr = np.concatenate((flipped_time_arr,normal_time_arr),axis=1)

	normal_time_file = open(os.path.join(data_folder,filename))
	normal_time_header_1 = normal_time_file.readline()
	normal_time_header_2 = normal_time_file.readline()
	normal_time_file.close()

	flipped_time_file = open(os.path.join(data_folder,time_flipped_filename))
	flipped_time_header_1 = flipped_time_file.readline()
	flipped_time_header_2 = flipped_time_file.readline()
	flipped_time_file.close()

	file_header = "#normal\n"+normal_time_header_1+normal_time_header_2+"#flipped\n"+flipped_time_header_1+flipped_time_header_2

	np.savetxt(os.path.join(save_folder,"cross_RGG.dat"),joint_arr.T,header=file_header)