import WimPyDD as WD
import numpy as np

#to print isospin base and proton-neutron base at the sametime

exp=WD.XENON_1T_2018
hamiltonian=WD.eft_hamiltonian('C_1_3_alpha_1_3',{1:lambda:[1,1],3:lambda:[1,1],(1,'qm2'):lambda q:[1/q**2,1/q**2],(3,'qm2'):lambda q:[1/q**2,1/q**2]}) #qm2 means q to the minus 2, 1/q**2 for massless long-range propagator

n_bin=0 #XENON1T has only one bin
vmin,delta_eta0=WD.streamed_halo_function() #Standard Halo model
mchi=10

mapping=WD.get_mapping(hamiltonian)
m=WD.wimp_dd_matrix(exp,hamiltonian,n_bin,vmin,delta_eta0,mchi)

mapping_pn=WD.get_mapping(hamiltonian,pn=True)
U=WD.rotation_from_isospin_to_pn(hamiltonian)
m_pn=np.dot(U,np.dot(m,U)) #matrix in proton-neutron base

print('='*15)
print('mapping in isospin base:')
print(mapping)
print('matrix in isospin base:')
with np.printoptions(linewidth=280):
    print(m)
print('mapping in proton-neutron base:')
print(mapping_pn)
print('matrix in proton-neutron base:')
with np.printoptions(linewidth=280):
    print(m_pn)
print('='*15)