Beispiel #1
0
def sort_key_list(key_list):
	ke_pair_list = []
	for key in key_list:
		energy = integrals.mat_element(key, key)
		ke_pair_list.append((energy, key))
	ke_pair_list.sort()
	sorted_key_list = []
	for ke in ke_pair_list:
		sorted_key_list.append(ke[1])
	return sorted_key_list
Beispiel #2
0
def draw_H():
	
	import integrals
	import ctrl_panel
	
	dim = len(ctrl_panel.key_list)
	H = []
	for key_i in ctrl_panel.key_list:
		vec = []
		for key_j in ctrl_panel.key_list:
			if key_i != key_j:
				entry = integrals.mat_element(key_i, key_j, integrals.ref_energy)
				vec.append(abs(entry))
			else:
				vec.append(0)
		vec.append(0)
		H.append(vec)
	H.append([0] * (dim+1))
	plot_once_2D(dim, H)