コード例 #1
0
ファイル: nodes.py プロジェクト: cygnine/spyctral
def scale_nodes(L,N,delta=0.5,s=1.,t=0.):

    from spyctral.wiener.quad import gq
    from spyctral.common.scaling import scale_factor

    [x,w] = gq(N,s=s,t=t)

    return scale_factor(L,x,delta=delta)
コード例 #2
0
ファイル: fft.py プロジェクト: cygnine/spyctral
def fft_collocation_overhead(N,s=1.,t=0.,shift=0.,scale=1.):

    from numpy import sqrt
    from spyctral.fourier.fft import fft_overhead
    from spyctral.wiener.weights import sqrt_weight_bias as wx_sqrt
    from spyctral.wiener.quad import gq

    overhead = fft_overhead(N,gamma=s-1.,delta=t)
    x = gq(N,s=1.,t=0.,shift=shift,scale=scale)[0]
    premult = sqrt(scale)/wx_sqrt(x,s=s,t=t,shift=shift,scale=scale)

    return [premult,overhead]
コード例 #3
0
ファイル: fft.py プロジェクト: cygnine/spyctral
def fft_collocation(f,s=1.,t=0.,shift=0.,scale=1.):

    from spyctral.fourier.fft import fft as fft_Psi
    from spyctral.wiener.weights import sqrt_weight_bias as wx_sqrt
    from spyctral.wiener.quad import gq
    from numpy import sqrt

    # This stuff can all be overhead
    N = f.size
    x = gq(N,s=1.,t=0.,shift=shift,scale=scale)[0]
    modes = f/wx_sqrt(x,s=s,t=t,shift=shift,scale=scale)

    temp = fft_Psi(modes,gamma=s-1.,delta=t)*float(sqrt(scale))
    temp[-s:] = 0
    temp[:s] = 0

    return temp
コード例 #4
0
ファイル: fft.py プロジェクト: cygnine/spyctral
def ifft_collocation(F,s=1.,t=0.,shift=0.,scale=1.):

    from spyctral.fourier.fft import ifft as ifft_Psi
    from spyctral.wiener.weights import sqrt_weight_bias as wx_sqrt
    from spyctral.wiener.quad import gq
    from numpy import sqrt

    from spyctral.fourier.fft import fft_overhead, ifft_online

    N = F.size

    #overhead = fft_overhead(N,g=s-1.,d=t)
    #fx = ifft_online(f,overhead)
    fx = ifft_Psi(F,gamma=s-1.,delta=t)
    
    x = gq(N,s=1.,t=0.,shift=shift,scale=scale)[0]
    fx *= wx_sqrt(x,s=s,t=t,shift=shift,scale=scale)/float(sqrt(scale))
    #print "hi"

    return fx