Пример #1
0
    def plot2d_unstructured_nodes(self, data, *args, **kwargs):
        method = kwargs.get('plot_function', 'tricontourf')

        x1   = self.x[:,0]
        x2   = self.x[:,1]
        xmin = min(x1)
        xmax = max(x1)
        ymin = min(x2)
        ymax = max(x2)
        if method == 'tricontourf':
            triang = tri.Triangulation(x1, x2)
            plt.tricontourf(triang, data, *args, **kwargs)

        elif method == 'scipy_griddata_pcontourf':
            n  = np.sqrt(len(data))
            xi = np.linspace(xmin, xmax, n)
            yi = np.linspace(ymin, ymax, n)
            zi = sgriddata((x1, x2), q, (xi[None,:], yi[:,None]), method='nearest')
            cs = plt.contourf(xi, yi, zi, *args, **kwargs)

        elif method == 'matplotlib_griddata_pcontourf':
            n  = np.sqrt(len(data))
            xi = np.linspace(xmin, xmax, n)
            yi = np.linspace(ymin, ymax, n)
            zi = mgriddata(x1, x2, q, xi, yi, interp='linear')
            cs = plt.contourf(xi, yi, zi, *args, **kwargs)
Пример #2
0
def plot_tchiphase2(filename):
    '''
    Plot Phase diagram of T*chi - 2D version.
    '''
    #params
    Gamma=0.5/pi
    NS=50
    #plot
    ion()
    fig=figure(figsize=(6,4))
    data=loadtxt(filename)
    Umesh=data[:,0]
    edmesh=data[:,1]
    gmoU=Gamma/Umesh
    edoU=edmesh/Umesh+0.5
    tchimesh=data[:,2]
    #interpolating
    X,Y=meshgrid(linspace(0,0.55,200),linspace(0,0.55,200),indexing='ij')
    #Z=mgriddata(gmoU,edoU,tchimesh,X,Y,interp='linear')
    Z=mgriddata(gmoU,edoU,tchimesh,Y,X,interp='nn')
    pcolormesh(X,Y,Z,rasterized=True)
    pcolormesh(-X,Y,Z,rasterized=True)
    xlabel('$\\tilde{\epsilon}_f/U$',fontsize=18)
    xticks([-0.5,0,0.5],fontsize=14)
    ylabel('$\Gamma/U$',fontsize=18)
    yticks([0,0.5],fontsize=14)
    xlim(-0.5,0.5)
    ylim(0,0.55)
    cb=colorbar()
    cb.set_ticks([0,0.25])
    text(0.,0.2,'LM',fontsize=18,color='w',horizontalalignment='center')
    text(0.35,0.4,'ASC',fontsize=18,color='w',horizontalalignment='center')
    tight_layout()
    pdb.set_trace()
    savefig('%s.eps'%filename[:-4])
    savefig('%s.png'%filename[:-4])
Пример #3
0
def plot_Fig15(filename1,filename2,filename11,filename22):
    '''
    Plot Phase diagram of T*chi - 2D version-two in one version.
    '''
    nzcomp=[0,1,3]
    which='s'
    #params
    D=1.
    Gap=0.1
    #plot
    gs=GridSpec(2,11,hspace=0.01)
    ymax=0.2
    ymin=-0.15 if which=='d' else -0.2
    ion()
    fig=figure(figsize=(9,6))
    fig1=subplot(gs[0,:5])
    colormap=cm.rainbow(linspace(0.2,0.8,len(nzcomp)))
    plts=[]
    data=loadtxt(filename1)
    wlist=data[:,0]
    dlv0=data[:,1:5]/pi
    dlv=data[:,5:]/pi
    for color,i in zip(colormap,nzcomp):
        mask1=wlist<-Gap
        mask2=wlist>Gap
        plts+=plot(concatenate([wlist[mask1],[-Gap,Gap],wlist[mask2]]),concatenate([dlv0[mask1,i],zeros(2),dlv0[mask2,i]]),lw=2,color=color)
    for color,i in zip(colormap,nzcomp):
        #scatter(wlist[mask],dlv[mask,i],s=30,lw=1,edgecolors=colormap[i],facecolors='none')
        plts.append(scatter(wlist[::3],dlv[::3,i],s=30,lw=1,edgecolors=color,facecolors='none'))
    legend(plts,[r'$d_0$',r'$d_x$',r'$d_z$',r"$d'_0$",r"$d'_x$",r"$d'_z$"],ncol=2,loc=1 if which=='d' else 3,fontsize=12)
    ylim(ymin,ymax)
    yticks([-0.2,-0.1,0,0.1,0.2])
    xlim(-1,1)
    xlabel(r'$\omega/\omega_U$',fontsize=16)
    xticks([-1,0,1],fontsize=14)
    text(0.8,0.17,'(a)',fontsize=14)
    axhline(y=0,color='#999999',ls=':',lw=2)
 
    paramstring=r'\usepackage{bm}'
    matplotlib.rcParams['text.latex.preamble'] = paramstring
    matplotlib.rcParams['text.usetex'] = True
    #ylabel(r'$\bm{\mathcal{D}}(\omega)$',fontsize=16)
    ylabel(r'$\mathcal{D}(\omega)$',fontsize=16)
    xlabel(r'$\omega$/\omega_U',fontsize=16)
    xticks([-1,0,1],fontsize=14)
    ylim(ymin,ymax)
    xlim(-1,1)
    text(0.8,0.17,'(a)',fontsize=14)
    axhline(y=0,color='#999999',ls=':',lw=2)

    fig2=subplot(gs[0,5:10])
    data=loadtxt(filename2)
    wlist=data[:,0]
    dlv0=data[:,1:5]
    dlv=data[:,5:]
    for color,i in zip(colormap,nzcomp):
        mask1=wlist<-Gap
        mask2=wlist>Gap
        plot(concatenate([wlist[mask1],[-Gap,Gap],wlist[mask2]]),concatenate([dlv0[mask1,i],zeros(2),dlv0[mask2,i]]),lw=2,color=color)
    for color,i in zip(colormap,nzcomp):
        mask1=wlist<-Gap
        mask2=wlist>Gap
        mask1[1::2]=0
        mask2[0::2]=0
        scatter(wlist[mask1|mask2],dlv[mask1|mask2,i],s=30,lw=1,edgecolors=color,facecolors='none')
    ylim(ymin,ymax)
    xlim(-1,1)
    xlabel(r'$\omega$',fontsize=16)
    xticks([-1,0,1],fontsize=14)
    yticks([])
    text(0.8,0.17,'(b)',fontsize=14)
    axhline(y=0,color='#999999',ls=':',lw=2)

    #params
    Gamma=0.5/pi
    NS=50
    #plot
    fig1=subplot(gs[1,:5])
    text(0.48,0.2,'(c)',fontsize=14,color='w')
    data=loadtxt(filename11)
    Umesh=data[:,0]
    edmesh=data[:,1]
    gmoU=Gamma/Umesh
    edoU=edmesh/Umesh+0.5
    tchimesh=data[:,2]
    #interpolating
    X,Y=meshgrid(linspace(-0.6,0.6,200),linspace(0,1/4.5,200),indexing='ij')
    Z=mgriddata(gmoU,edoU,tchimesh,Y,X,interp='nn')
    pcolormesh(X,Y,Z,rasterized=True)
    plot([0,0],[0,0.3],ls='--',color='k')
    xlabel('$\epsilon_f/U$',fontsize=16)
    xticks([-0.5,-0.3,0,0.3,0.5])
    ylabel('$\Gamma/U$',fontsize=16)
    #yticks([0,0.5],fontsize=14)
    xlim(-0.6,0.6)
    ylim(0,1/4.5)
    text(0.,0.1,'doublet',fontsize=18,color='w',horizontalalignment='center')
    text(-0.4,0.2,'singlet',fontsize=18,color='w',horizontalalignment='center')

    fig2=subplot(gs[1,5:])
    text(0.48,0.2,'(d)',fontsize=14,color='w')
    data=loadtxt(filename22)
    Umesh=data[:,0]
    edmesh=data[:,1]
    gmoU=Gamma/Umesh
    edoU=edmesh/Umesh+0.5
    tchimesh=data[:,2]
    #interpolating
    Z=mgriddata(gmoU,edoU,tchimesh,Y,X,interp='nn')
    pcolormesh(X,Y,Z,rasterized=True)
    xlabel('$\epsilon_f/U$',fontsize=18)
    yticks([])
    xlim(-0.6,0.6)
    xticks([-0.5,-0.3,0,0.3,0.5])
    ylim(0,1/4.5)
    plot([0,0.43],[0,0.5],ls='--',color='k')
    #cax = fig.add_axes([0.9, 0.1, 0.03, 0.8])
    #cb=fig.colorbar(img,cax=cax)
    cb=colorbar()
    cb.set_ticks([0,0.25])
    text(0.,0.1,'doublet',fontsize=18,color='w',horizontalalignment='center')
    text(-0.4,0.2,'singlet',fontsize=18,color='w',horizontalalignment='center')
    y0=1./2/pi
    plot([-0.25],[y0],'wo')
    offsetx=0.03
    plot([0.25],[y0],'wo')
    text(0.25+offsetx,y0,'D',fontsize=14,color='w')
    text(-0.25+offsetx,y0,'S',fontsize=14,color='w')
 
    gs.tight_layout(fig,pad=0.3,w_pad=0.6)
    pdb.set_trace()
    savefig('allinone.eps')
    savefig('allinone.png')
Пример #4
0
def plot_Fig5(filename1,filename2):
    '''
    Plot Phase diagram of T*chi - 2D version-two in one version.
    '''
    #params
    Gamma=0.5/pi
    NS=50
    #plot
    ion()
    fig=figure(figsize=(8,3.5))
    gs=GridSpec(1,11,hspace=0.01)
    fig1=subplot(gs[:5])
    text(0.4,0.2,'(a)',fontsize=14,color='w')
    data=loadtxt(filename1)
    Umesh=data[:,0]
    edmesh=data[:,1]
    gmoU=Gamma/Umesh
    edoU=edmesh/Umesh+0.5
    tchimesh=data[:,2]
    #interpolating
    X,Y=meshgrid(linspace(-0.6,0.6,200),linspace(0,1/4.5,200),indexing='ij')
    Z=mgriddata(gmoU,edoU,tchimesh,Y,X,interp='nn')
    pcolormesh(X,Y,Z,rasterized=True)
    plot([0,0],[0,0.3],ls='--',color='k')
    xlabel('$\epsilon_f/U$',fontsize=16)
    xticks([-0.5,-0.3,0,0.3,0.5])
    ylabel('$\Gamma/U$',fontsize=16)
    #yticks([0,0.5],fontsize=14)
    xlim(-0.6,0.6)
    ylim(0,1/4.5)
    text(0.,0.1,'doublet',fontsize=18,color='w',horizontalalignment='center')
    text(-0.4,0.2,'singlet',fontsize=18,color='w',horizontalalignment='center')
    #y0=1./2/pi
    #xs,xd=0.35,0.
    #plot([xs],[y0],'wo')
    #offsetx=0.03
    #plot([xd],[y0],'wo')
    #text(xd+offsetx,y0,'D1',fontsize=14,color='w')
    #text(xs+offsetx,y0,'S1',fontsize=14,color='w')

    fig2=subplot(gs[5:])
    text(0.4,0.2,'(b)',fontsize=14,color='w')
    data=loadtxt(filename2)
    Umesh=data[:,0]
    edmesh=data[:,1]
    gmoU=Gamma/Umesh
    edoU=edmesh/Umesh+0.5
    tchimesh=data[:,2]
    #interpolating
    Z=mgriddata(gmoU,edoU,tchimesh,Y,X,interp='nn')
    pcolormesh(X,Y,Z,rasterized=True)
    xlabel('$\epsilon_f/U$',fontsize=18)
    yticks([])
    xlim(-0.6,0.6)
    xticks([-0.5,-0.3,0,0.3,0.5])
    ylim(0,1/4.5)
    plot([0,0.43],[0,0.5],ls='--',color='k')
    #cax = fig.add_axes([0.9, 0.1, 0.03, 0.8])
    #cb=fig.colorbar(img,cax=cax)
    cb=colorbar()
    cb.set_ticks([0,0.25])
    text(0.,0.1,'doublet',fontsize=18,color='w',horizontalalignment='center')
    text(-0.4,0.2,'singlet',fontsize=18,color='w',horizontalalignment='center')
    y0=1./2/pi
    #plot([-0.25],[y0],'wo')
    plot([0],[y0],'wo')
    offsetx=0.03
    text(0.+offsetx,y0,'D',fontsize=14,color='w')
    #text(-0.25+offsetx,y0,'S2',fontsize=14,color='w')
 
    gs.tight_layout(fig,pad=0.3,w_pad=0.6)
    pdb.set_trace()
    savefig('tchiskew.eps')
    savefig('tchiskew.png')
Пример #5
0
def plot_Fig4(filename1,filename2):
    '''
    Plot Phase diagram of T*chi - 2D version-two in one version.
    '''
    #params
    Gamma=0.5/pi
    NS=50
    #plot
    ion()
    fig=figure(figsize=(8,4))
    gs=GridSpec(1,11)
    subplot(gs[:5])
    text(0.4,0.5,'(a)',fontsize=14,color='w')
    data=loadtxt(filename1)
    Umesh=data[:,0]
    edmesh=data[:,1]
    gmoU=Gamma/Umesh
    edoU=edmesh/Umesh+0.5
    tchimesh=data[:,2]
    #interpolating
    X,Y=meshgrid(linspace(-0.55,0.55,200),linspace(0,0.55,200),indexing='ij')
    gmoU=append(gmoU,gmoU);edoU=append(-edoU,edoU);tchimesh=append(tchimesh,tchimesh)
    #Z=mgriddata(gmoU,edoU,tchimesh,X,Y,interp='linear')
    Z=mgriddata(gmoU,edoU,tchimesh,Y,X,interp='nn')
    pcolormesh(X,Y,Z,rasterized=True)
    #pcolormesh(-X,Y,Z,rasterized=True)
    xlabel('$\epsilon_f/U$',fontsize=16)
    #xticks([-0.5,0,0.5],fontsize=14)
    ylabel('$\Gamma/U$',fontsize=16)
    #yticks([0,0.5],fontsize=14)
    xlim(-0.5,0.5)
    ylim(0,0.55)
    text(0.,0.2,'doublet',fontsize=18,color='w',horizontalalignment='center')
    text(0.35,0.43,'singlet',fontsize=18,color='w',horizontalalignment='center')

    subplot(gs[5:])
    text(0.4,0.5,'(b)',fontsize=14,color='w')
    data=loadtxt(filename2)
    Umesh=data[:,0]
    edmesh=data[:,1]
    gmoU=Gamma/Umesh
    edoU=edmesh/Umesh+0.5
    tchimesh=data[:,2]
    #interpolating
    gmoU=append(gmoU,gmoU);edoU=append(-edoU,edoU);tchimesh=append(tchimesh,tchimesh)
    Z=mgriddata(gmoU,edoU,tchimesh,Y,X,interp='nn')
    pcolormesh(X,Y,Z,rasterized=True)
    #img=pcolormesh(-X,Y,Z,rasterized=True)
    xlabel('$\epsilon_f/U$',fontsize=18)
    #xticks([-0.5,0,0.5],fontsize=14)
    yticks([])
    xlim(-0.5,0.5)
    ylim(0,0.55)
    #cax = fig.add_axes([0.9, 0.1, 0.03, 0.8])
    #cb=fig.colorbar(img,cax=cax)
    cb=colorbar()
    cb.set_ticks([0,0.25])
    text(0.,0.2,'doublet',fontsize=18,color='w',horizontalalignment='center')
    text(0.35,0.43,'singlet',fontsize=18,color='w',horizontalalignment='center')
 
    tight_layout()
    pdb.set_trace()
    savefig('tchi2.eps')
    savefig('tchi2.png')