i\ddlmZddlmZddlmZddlZeGddZdS)) annotations) dataclass)IterableNcheZdZUded<ded<ded<ded<dd Zedd ZddZddZdS)TightBindingModelstrnameint dimension np.ndarrayonsitez#list[tuple[np.ndarray, np.ndarray]]hoppingsreturnNonectj|jtj|_g}|jD]S\}}|tj|t tj|tjfT||_|jjdks&|jjd|jjdkrtd|j dkrtd|jD]I\}}|j|j fkrtd|j|jjkrtdJdS) Ndtyperzonsite must be a square matrixzdimension must be positivez+displacement vector has the wrong dimensionz,hopping matrix shape must match onsite shape) npasarrayr complex128rappendfloatndimshape ValueErrorr )self normalized displacementhoppings /D:\ml\projects\myproj\tight binding\tb\model.py __post_init__zTightBindingModel.__post_init__sMjBMBBB :< %)]   !L'   J|5999Jwbm<<<     # ; q DK$5a$8DK> > >Q  9:: :%)] Q Q !L'!dn%666 !NOOO} 111 !OPPP2 Q Qc@t|jjdS)Nr)r r r)rs r" num_orbitalszTightBindingModel.num_orbitals's4;$Q'(((r$k_pointIterable[float]ctjt|t}|j|jfkrt d|j}|j D]h\}}tj dtj ||z}|||zz }|| j tj |zz }i|S)Nrzk-point has the wrong dimensiony?)rrtuplerrr rr copyrexpdotconjT)rr'k_vector hamiltonianr r!phases r"r1zTightBindingModel.hamiltonian+s:eGnnE::: >dn. . .>?? ?k&&(( %)] = = !L'F2x > >>??E 7U? *K 7<<>>+bgenn< rBs""""""!!!!!! .B.B.B.B.B.B.B .B.B.Br$