Beispiel #1
0
def mprint(m, fmt):
    """ 
        This function will print a VSIPL matrix or vector suitable for pasting into Octave or Matlab.
        usage: mprint(<vsip matrix/vector>, fmt)
        fmt is a string corresponding to a simple fmt statement. 
        For instance '%6.5f' prints as 6 characters wide with 5 decimal digits.
        Note format converts this statement to '% 6.5f' or '%+6.5f' so keep
        the input simple.
    """
    def _fmt1(c):
        if c != '%':
            return c
        else:
            return '% '

    def _fmt2(c):
        if c != '%':
            return c
        else:
            return '%+'

    def _fmtfunc(fmt1, fmt2, y):
        x = vsip.cscalarToComplex(y)
        if type(x) == complex:
            return fmt1 % x.real + fmt2 % x.imag + "i"
        else:
            return fmt % x

    tm = [
        'mview_d', 'mview_f', 'cmview_d', 'cmview_f', 'mview_i', 'mview_uc',
        'mview_si', 'mview_bl'
    ]
    tv = [
        'vview_d', 'vview_f', 'cvview_d', 'cvview_f', 'vview_i', 'vview_uc',
        'vview_si', 'vview_bl', 'vview_vi', 'vview_mi'
    ]
    t = vsip.getType(m)[1]
    tfmt = [_fmt1(c) for c in fmt]
    fmt1 = "".join(tfmt)
    tfmt = [_fmt2(c) for c in fmt]
    fmt2 = "".join(tfmt)
    if t in tm:
        cl = vsip.getcollength(m)
        rl = vsip.getrowlength(m)
        for i in range(cl):
            M = []
            for j in range(rl):
                M.append(_fmtfunc(fmt1, fmt2, vsip.get(m, (i, j))))
            if i == 0:
                print("[" + " ".join(M) + ";")
            elif i < cl - 1:
                print(" " + " ".join(M) + ";")
            else:
                print(" " + " ".join(M) + "]")
    elif t in tv:
        l = vsip.getlength(m)
        V = [_fmtfunc(fmt1, fmt2, vsip.get(m, i)) for i in range(l)]
        print("[" + " ".join(V) + "]")
    else:
        print('Object not VSIP vector or matrix')
Beispiel #2
0
def mstring(m,fmt):
    """ 
        This function will print a VSIPL matrix or vector suitable for pasting into Octave or Matlab.
        usage: mprint(<vsip matrix/vector>, fmt)
        fmt is a string corresponding to a simple fmt statement. 
        For instance '%6.5f' prints as 6 characters wide with 5 decimal digits.
        Note format converts this statement to '% 6.5f' or '%+6.5f' so keep
        the input simple.
    """
    def _fmt1(c):
        if c != '%':
            return c
        else:
            return '% '
    def _fmt2(c):
        if c != '%':
            return c
        else:
            return '%+'
    def _fmtfunc(fmt1,fmt2,y):
        x = vsip.cscalarToComplex(y)
        if type(x) == complex:
            s = fmt1 % x.real
            s += fmt2 % x.imag
            s += "i"
            return s
        else:
            return fmt1 % x
    tm=['mview_d','mview_f','cmview_d','cmview_f','mview_i','mview_uc','mview_si','mview_bl']
    tv=['vview_d','vview_f','cvview_d','cvview_f','vview_i','vview_uc','vview_si','vview_bl','vview_vi','vview_mi']
    t=vsip.getType(m)[1]
    tfmt=[_fmt1(c) for c in fmt]
    fmt1 = "".join(tfmt)
    tfmt=[_fmt2(c) for c in fmt]
    fmt2 = "".join(tfmt)
    if t in tm:
        cl=vsip.getcollength(m)
        rl=vsip.getrowlength(m)
        s=str()
        for i in range(cl):
            M=[]
            for j in range(rl):
                M.append(_fmtfunc(fmt1,fmt2,vsip.get(m,(i,j))))
            if i == 0:
                s += "["+" ".join(M) + ";\n"
            elif i < cl-1:
                s += " "+" ".join(M) + ";\n"
            else:
                s += " "+" ".join(M) + "]\n"
        return s
    elif t in tv:
        l=vsip.getlength(m)
        V=[_fmtfunc(fmt1,fmt2,vsip.get(m,i)) for i in range(l)]
        return "[" + " ".join(V) + "]\n"
    else:
        print('Object not VSIP vector or matrix')
Beispiel #3
0
def FFTdataGenRamp(a):
    """ FFTdataGenRamp(v) expects argument 'v' to ba a complex VSIPL vector.
        The vector is filled with data which, when the fft is applied,
        should turn into a ramp starting a 0 with increment 1.
        This function is normally used for testing FFT codes.
    """
    from math import cos as ccos, sin as ssin, pi

    def _gen(a):
        N = int(vsip.getlength(a))
        npm = N / 2
        c = pi / float(N)
        vsip.put(a, 0, (float(N) - 1.0) / 2.0)
        indx = 0
        if N % 2:
            for i in range(npm):
                indx = i + 1
                x = c * float(i) + c
                x = -0.5j * ccos(x) / ssin(x) - 0.5
                vsip.put(a, indx, x)
            for i in range(npm, 0, -1):
                indx += 1
                x = vsip.get(a, i)
                x.i = -x.i
                vsip.put(a, indx, x)
        else:
            npm -= 1
            for i in range(npm):
                indx = i + 1
                x = c * float(i) + c
                x = -0.5j * ccos(x) / ssin(x) - 0.5
                vsip.put(a, indx, x)
            x = -0.5
            indx += 1
            vsip.put(a, indx, x)
            for i in range(npm, 0, -1):
                indx += 1
                x = vsip.get(a, i)
                x.i = -x.i
                vsip.put(a, indx, x)

    t = vsip.getType(a)
    sz = ()
    if t[0]:
        sz = size(a)
    if len(sz) == 3 and (t[1] == 'cvview_f' or t[1] == 'cvview_d'):
        _gen(a)
        return a
    else:
        print(
            'Not a supported type. Should be a vsip complex float vector vector'
        )
        return False
Beispiel #4
0
def FFTdataGenRamp(a):
    """ FFTdataGenRamp(v) expects argument 'v' to ba a complex VSIPL vector.
        The vector is filled with data which, when the fft is applied,
        should turn into a ramp starting a 0 with increment 1.
        This function is normally used for testing FFT codes.
    """
    from math import cos as ccos,sin as ssin,pi
    def _gen(a):
        N=int(vsip.getlength(a))
        npm=N/2
        c=pi/float(N)
        vsip.put(a,0,(float(N) - 1.0)/2.0)
        indx=0;
        if N % 2:
            for i in range(npm):
                indx = i+1
                x = c * float(i) + c
                x = -0.5j * ccos(x)/ssin(x)  - 0.5
                vsip.put(a,indx,x)
            for i in range(npm,0,-1):
                indx +=1
                x=vsip.get(a,i)
                x.i=-x.i
                vsip.put(a,indx,x)
        else:
            npm -= 1
            for i in range(npm):
               indx = i+1
               x = c * float(i) + c
               x = -0.5j * ccos(x)/ssin(x)  - 0.5
               vsip.put(a,indx,x)
            x=-0.5
            indx +=1
            vsip.put(a,indx,x)
            for i in range(npm,0,-1):
                indx +=1
                x=vsip.get(a,i)
                x.i=-x.i
                vsip.put(a,indx,x)
    t=vsip.getType(a)
    sz=()
    if t[0]:
        sz=size(a)
    if len(sz) == 3 and (t[1] == 'cvview_f' or t[1] == 'cvview_d'):
        _gen(a)
        return a
    else:
        print('Not a supported type. Should be a vsip complex float vector vector')
        return False  
Beispiel #5
0
def size(a):
    """
       size(view) will return a tuple of
           (offset,stride,length)
       if view is a vsipl vector or
           (offset, col_stride, col_length, row_stride, row_length)
       if view is a vsipl matrix
    """
    f = {
        'vview_d': '_vsize(vsip.getattrib(a))',
        'vview_f': '_vsize(vsip.getattrib(a))',
        'cvview_d': '_vsize(vsip.getattrib(a))',
        'cvview_f': '_vsize(vsip.getattrib(a))',
        'vview_vi': '_vsize(vsip.getattrib(a))',
        'vview_mi': '_vsize(vsip.getattrib(a))',
        'vview_bl': '_vsize(vsip.getattrib(a))',
        'vview_i': '_vsize(vsip.getattrib(a))',
        'vview_si': '_vsize(vsip.getattrib(a))',
        'vview_uc': '_vsize(vsip.getattrib(a))',
        'mview_d': '_msize(vsip.getattrib(a))',
        'mview_f': '_msize(vsip.getattrib(a))',
        'cmview_d': '_msize(vsip.getattrib(a))',
        'cmview_f': '_msize(vsip.getattrib(a))',
        'mview_bl': '_msize(vsip.getattrib(a))',
        'mview_i': '_msize(vsip.getattrib(a))',
        'mview_si': '_msize(vsip.getattrib(a))',
        'mview_uc': '_msize(vsip.getattrib(a))'
    }

    def _vsize(attr):
        return (attr.offset, attr.stride, attr.length)

    def _msize(attr):
        return (attr.offset, attr.col_stride, attr.col_length, attr.row_stride,
                attr.row_length)

    t = vsip.getType(a)
    if t[1] in f:
        return eval(f[t[1]])
    else:
        print('Not a supported type')
        return False
Beispiel #6
0
def vView(x,o,s,l):
    """ vView(x,o,s,l) returns the view x with offset, stride and
        length (o,s,l) .
        This could also be done with vsiputils using
        bind(vsip.block(x),(o,s,l))
        except that bind will create a new object that will need to 
        be allocated and destroyed.
        vView will modify the supplied object x.
    """
    t=['vview_f', 'vview_d',
       'cvview_f','cvview_d',
       'vview_i','vview_si','vview_uc','vview_vi',
       'vview_bl','vview_mi']
    myT=vsip.getType(x)[1]
    if myT in t:
        attr=vsip.getattrib(x)
        attr.offset=o; attr.stride=s; attr.length=l
        return vsip.putattrib(x,attr)
    else:
        print('type ' + myT + ' not supported by vView')
Beispiel #7
0
def vView(x, o, s, l):
    """ vView(x,o,s,l) returns the view x with offset, stride and
        length (o,s,l) .
        This could also be done with vsiputils using
        bind(vsip.block(x),(o,s,l))
        except that bind will create a new object that will need to 
        be allocated and destroyed.
        vView will modify the supplied object x.
    """
    t = [
        'vview_f', 'vview_d', 'cvview_f', 'cvview_d', 'vview_i', 'vview_si',
        'vview_uc', 'vview_vi', 'vview_bl', 'vview_mi'
    ]
    myT = vsip.getType(x)[1]
    if myT in t:
        attr = vsip.getattrib(x)
        attr.offset = o
        attr.stride = s
        attr.length = l
        return vsip.putattrib(x, attr)
    else:
        print('type ' + myT + ' not supported by vView')
Beispiel #8
0
def size(a):
    """
       size(view) will return a tuple of
           (offset,stride,length)
       if view is a vsipl vector or
           (offset, col_stride, col_length, row_stride, row_length)
       if view is a vsipl matrix
    """
    f={'vview_d':'_vsize(vsip.getattrib(a))',
       'vview_f': '_vsize(vsip.getattrib(a))',
       'cvview_d':'_vsize(vsip.getattrib(a))',
       'cvview_f':'_vsize(vsip.getattrib(a))',
       'vview_vi':'_vsize(vsip.getattrib(a))',
       'vview_mi':'_vsize(vsip.getattrib(a))',
       'vview_bl':'_vsize(vsip.getattrib(a))',
       'vview_i':'_vsize(vsip.getattrib(a))',
       'vview_si':'_vsize(vsip.getattrib(a))',
       'vview_uc':'_vsize(vsip.getattrib(a))',
       'mview_d':'_msize(vsip.getattrib(a))',
       'mview_f':'_msize(vsip.getattrib(a))',
       'cmview_d':'_msize(vsip.getattrib(a))',
       'cmview_f':'_msize(vsip.getattrib(a))',
       'mview_bl':'_msize(vsip.getattrib(a))',
       'mview_i':'_msize(vsip.getattrib(a))',
       'mview_si':'_msize(vsip.getattrib(a))',
       'mview_uc':'_msize(vsip.getattrib(a))'}
    def _vsize(attr):
        return (attr.offset,attr.stride,attr.length)
    def _msize(attr):
        return (attr.offset,attr.col_stride,attr.col_length,attr.row_stride, attr.row_length)
    t=vsip.getType(a)
    if t[1] in f:
        return eval(f[t[1]])
    else:
        print('Not a supported type')
        return False