Beispiel #1
0
def _set_xmode(exec_mode=None):
    'Save programmer for typing exe= in every call'
    global glob_exe

    if exec_mode != None:
        ivlad.chk_par_in_list(exec_mode, ['x', 'g', 'p', 'l'])
        glob_exe = exec_mode
Beispiel #2
0
def _set_xmode(exec_mode=None):
    'Save programmer for typing exe= in every call'
    global glob_exe
    
    if exec_mode != None:
        ivlad.chk_par_in_list(exec_mode,['x','g','p','l'])
        glob_exe = exec_mode
Beispiel #3
0
def main(par):

    inp = par.string('inp') # input file
    out = par.string('out') # output file

    verb = par.bool('verb', False) # if y, print system commands, outputs

    ivlad.chk_file_dims(inp, 2)

    n1 = ivlad.getout('sfget',['parform=n','n1'], inp, verb)
    n2 = ivlad.getout('sfget',['parform=n','n2'], inp, verb)
    n1 = int(n1)
    n2 = int(n2)

    h = par.int('h', 768)  # output height
    w = par.int('w', 1024) # output width
    has_h = par.string('h')
    has_w = par.string('w')

    if has_h and has_w: # both w and h were read. Check for sanity:
        ivlad.chk_param_limit(w, 'w')
        ivlad.chk_param_limit(h, 'h')
        if (h,w) == (n1,n2):
            ivlad.msg('Change h or w if you want out != inp')
            sf.cp(inp, out, verb)
            return ivlad.unix_success

    # Transform h and w to pixels, if they are not
    # No default value for par.string -- Quirk of rsf, replicated in rsfbak
    unit = par.string('unit', 'px') # unit of h and w. Can be: px, mm, cm, in
    ivlad.chk_par_in_list(unit,['mm','cm','in','px'])

    if unit != 'px':
        ppi = par.int('ppi') # outp. resolution (px/in). Necessary when unit!=px
        ivlad.chk_param_limit(ppi, 'ppi')
        # Transform w and h to px
        if unit == 'in':
            scale = 1
        elif unit == 'mm':
            scale = 254
        elif unit == 'cm':
            scale = 25.4
        w *= ppi / float(scale)
        h *= ppi / float(scale)
        # Don't worry, we'll convert to int after prar block
        del scale

    # Now h and w are in pixels

    # If prar=y, then h and/or w define a bounding box.
    # Find the dimensions of the image inside this box
    prar = par.bool('prar', True) # if y, PReserve Aspect Ratio of input
    if prar: # preserve aspect ratio
        if has_h and not has_w:
            w = n2 * float(h) / n1
        elif has_w and not has_h:
            h = n1 * float(w) / n2
        else: # Full bounding box specified
            hscale = float(h) / n1
            wscale = float(w) / n2
            if hscale < wscale:
                w = n2 * hscale
            else:
                h = n1 * wscale

    h = int(h)
    w = int(w)

    assert h > 1
    assert w > 1

    h = ivlad.valswitch(h, n1, None)
    w = ivlad.valswitch(w, n2, None)
    tmp = tempfile.mktemp(dir=rsf.path.datapath())

    # Interpolation and, if needed, bandpass 
    if h != None:
        d1 = ivlad.getout('sfget', ['parform=n','d1'], inp, verb)
        d1 = float(d1) * (n1-1)/float(h-1)
        if h < n1:
            ready_for_remap_1 = tmp + '1'
            sf.bandpass(inp, ready_for_remap_1, fhi=0.5/d1, verb=verb)
            rem2del_junk1 = True
        else:
            ready_for_remap_1 = inp
            rem2del_junk1 = False
        if w:
            out_remap1 = tmp + '2'
            rem2del_junk2 = True
        else:
            out_remap1 = out
            rem2del_junk2 = False
        sf.remap1(ready_for_remap_1, out_remap1, n1=h, d1=d1, verb=verb)
        if rem2del_junk1:
            sf.rm(ready_for_remap_1, verb)
    else: # no action on axis 1
        out_remap1 = inp
        rem2del_junk2 = False

    if w != None:
        d2 = ivlad.getout('sfget', ['parform=n','d2'], inp, verb)
        d2 = float(d2) * (n2-1)/float(w-1)
        out_transp1 = tmp + '3'
        sf.transp(out_remap1, out_transp1, verb=verb)
        if rem2del_junk2:
            sf.rm(out_remap1, verb)
        if w < n2:
            ready_for_remap_2 = tmp + '4'
            sf.bandpass(out_transp1, ready_for_remap_2, fhi=0.5/d2, verb=verb)
            rem2del_junk4 = True
        else:
            ready_for_remap_2 = out_transp1
            rem2del_junk4 = False
        ready_for_transp2 = tmp + '5'
        sf.remap1(ready_for_remap_2,ready_for_transp2,n1=w,d1=d2,verb=verb)
        sf.rm(out_transp1, verb)
        if rem2del_junk4:
            sf.rm(ready_for_remap_2, verb)
        sf.transp(ready_for_transp2, out, verb=verb)
        sf.rm(ready_for_transp2, verb)

    return ivlad.unix_success
Beispiel #4
0
def main(par):

    inp = par.string('inp')  # input file
    out = par.string('out')  # output file

    verb = par.bool('verb', False)  # if y, print system commands, outputs

    ivlad.chk_file_dims(inp, 2)

    n1 = ivlad.getout('sfget', ['parform=n', 'n1'], inp, verb)
    n2 = ivlad.getout('sfget', ['parform=n', 'n2'], inp, verb)
    n1 = int(n1)
    n2 = int(n2)

    h = par.int('h', 768)  # output height
    w = par.int('w', 1024)  # output width
    has_h = par.string('h')
    has_w = par.string('w')

    if has_h and has_w:  # both w and h were read. Check for sanity:
        ivlad.chk_param_limit(w, 'w')
        ivlad.chk_param_limit(h, 'h')
        if (h, w) == (n1, n2):
            ivlad.msg('Change h or w if you want out != inp')
            sf.cp(inp, out, verb)
            return ivlad.unix_success

    # Transform h and w to pixels, if they are not
    # No default value for par.string -- Quirk of rsf, replicated in rsfbak
    unit = par.string('unit', 'px')  # unit of h and w. Can be: px, mm, cm, in
    ivlad.chk_par_in_list(unit, ['mm', 'cm', 'in', 'px'])

    if unit != 'px':
        ppi = par.int(
            'ppi')  # outp. resolution (px/in). Necessary when unit!=px
        ivlad.chk_param_limit(ppi, 'ppi')
        # Transform w and h to px
        if unit == 'in':
            scale = 1
        elif unit == 'mm':
            scale = 254
        elif unit == 'cm':
            scale = 25.4
        w *= ppi / float(scale)
        h *= ppi / float(scale)
        # Don't worry, we'll convert to int after prar block
        del scale

    # Now h and w are in pixels

    # If prar=y, then h and/or w define a bounding box.
    # Find the dimensions of the image inside this box
    prar = par.bool('prar', True)  # if y, PReserve Aspect Ratio of input
    if prar:  # preserve aspect ratio
        if has_h and not has_w:
            w = n2 * float(h) / n1
        elif has_w and not has_h:
            h = n1 * float(w) / n2
        else:  # Full bounding box specified
            hscale = float(h) / n1
            wscale = float(w) / n2
            if hscale < wscale:
                w = n2 * hscale
            else:
                h = n1 * wscale

    h = int(h)
    w = int(w)

    assert h > 1
    assert w > 1

    h = ivlad.valswitch(h, n1, None)
    w = ivlad.valswitch(w, n2, None)
    tmp = tempfile.mktemp(dir=rsf.path.datapath())

    # Interpolation and, if needed, bandpass
    if h != None:
        d1 = ivlad.getout('sfget', ['parform=n', 'd1'], inp, verb)
        d1 = float(d1) * (n1 - 1) / float(h - 1)
        if h < n1:
            ready_for_remap_1 = tmp + '1'
            sf.bandpass(inp, ready_for_remap_1, fhi=0.5 / d1, verb=verb)
            rem2del_junk1 = True
        else:
            ready_for_remap_1 = inp
            rem2del_junk1 = False
        if w:
            out_remap1 = tmp + '2'
            rem2del_junk2 = True
        else:
            out_remap1 = out
            rem2del_junk2 = False
        sf.remap1(ready_for_remap_1, out_remap1, n1=h, d1=d1, verb=verb)
        if rem2del_junk1:
            sf.rm(ready_for_remap_1, verb)
    else:  # no action on axis 1
        out_remap1 = inp
        rem2del_junk2 = False

    if w != None:
        d2 = ivlad.getout('sfget', ['parform=n', 'd2'], inp, verb)
        d2 = float(d2) * (n2 - 1) / float(w - 1)
        out_transp1 = tmp + '3'
        sf.transp(out_remap1, out_transp1, verb=verb)
        if rem2del_junk2:
            sf.rm(out_remap1, verb)
        if w < n2:
            ready_for_remap_2 = tmp + '4'
            sf.bandpass(out_transp1,
                        ready_for_remap_2,
                        fhi=0.5 / d2,
                        verb=verb)
            rem2del_junk4 = True
        else:
            ready_for_remap_2 = out_transp1
            rem2del_junk4 = False
        ready_for_transp2 = tmp + '5'
        sf.remap1(ready_for_remap_2, ready_for_transp2, n1=w, d1=d2, verb=verb)
        sf.rm(out_transp1, verb)
        if rem2del_junk4:
            sf.rm(ready_for_remap_2, verb)
        sf.transp(ready_for_transp2, out, verb=verb)
        sf.rm(ready_for_transp2, verb)

    return ivlad.unix_success
Beispiel #5
0
def __x(exe, g_exe):
    if exe == None:
        return g_exe
    else:
        ivlad.chk_par_in_list(exe, ['x', 'g', 'p', 'l'])
        return exe
Beispiel #6
0
def __x(exe, g_exe):
    if exe == None:
        return g_exe
    else:
        ivlad.chk_par_in_list(exe, ['x','g','p','l'])
        return exe
Beispiel #7
0
def main(par):

    # Input parameters

    delim = par.string('delimiter',',') # Separator between values in input file
    numtype = par.string('dtype', 'float') # Input type    

    ivlad.chk_par_in_list(numtype, ooio.dtype_avl)

    # Process parameters

    verb = par.bool('verb', False) # Whether to echo n1, n2, infill/truncation
    debug = par.bool('debug', False) # Extra verbosity for debugging
    truncate = par.bool('trunc', False) 
    # Truncate or add zeros if nr elems in rows differs

    header = par.bool('header',False) # If the first line is a header

    # Output parameters
    
    o = [
    par.float('o1', 0.), # Origin of axis 1 in output (rows in input)
    par.float('o2', 0.)] # Origin of axis 2 in output (columns in input)

    d = [
    par.float('d1', 1.), # Axis 1 sampling
    par.float('d2', 1.)] # Axis 2 sampling
    
    unit = [
    par.string('unit1', 'unknown'),
    par.string('unit2', 'unknown')]

    lbl = [
    par.string('label1', 'unknown'),
    par.string('label2', 'unknown')]

    ##### End reading parameters #####

    stdin = csv.reader(sys.stdin, delimiter=delim)
 
    # Copy stdin so we can go back through it
    lines = []
    i = 0
    nr_cols_cst = True # Whether the nr of values in rows is constant

    if header: # the first line contains header keys
        line = stdin.next()
        k = 0
        for name in line:
            k += 1
            print('key%d=%s' % (k,name))
        
    # Find max nr of elements in a row
    for line in stdin:
        if line == []: # throw away blank lines
            continue
        curline = [float(x) for x in [x or '0' for x in line]]
        if numtype == 'int':
            curline = [int(x) for x in curline]
        lines.append(curline)
        llen = len(curline)
        i+=1 # We have successfully read line i
        if i==1:
            n2 = llen
        elif llen != n2:
            nr_cols_cst = False
            if (llen < n2 and truncate) or (llen > n2 and not truncate):
                n2 = llen

    n1 = len(lines)

    if not nr_cols_cst: # Truncate or append as necessary
        for i in range(n1):
            line = lines[i]
            lines[i] = ivlad.trunc_or_append(n2, line, 0, verb)

    # Avoiding to add a second dimension of length 1
    if n1 == 1:
        ndim_out = 1
        n = [n2]
        o = [o[0]]
        d = [d[0]]
        unit = [unit[0]]
        lbl = [lbl[0]]
    else:
        ndim_out = 2
        n = [n2, n1]

    out = ooio.RSFfile(ooio.stdout,par,ndim=ndim_out,intent='out',dtype=numtype) 
    out.set_hdr_info(n, o, d, unit, lbl)
    
    if debug:
        out.print_self('out')
        out.hdr.print_self('out.hdr')
        out.dat.print_self('out.dat')
        ivlad.msg(ivlad.hr)

    for line in lines:
        for val in line:
            out.write(val)

    return ivlad.unix_success
Beispiel #8
0
def main(par):

    # Input parameters

    delim = par.string('delimiter',',') # Separator between values in input file
    numtype = par.string('dtype', 'float') # Input type    

    ivlad.chk_par_in_list(numtype, ooio.dtype_avl)

    # Process parameters

    verb = par.bool('verb', False) # Whether to echo n1, n2, infill/truncation
    debug = par.bool('debug', False) # Extra verbosity for debugging
    truncate = par.bool('trunc', False) 
    # Truncate or add zeros if nr elems in rows differs

    header = par.bool('header',False) # If the first line is a header

    # Output parameters
    
    o = [
    par.float('o1', 0.), # Origin of axis 1 in output (rows in input)
    par.float('o2', 0.)] # Origin of axis 2 in output (columns in input)

    d = [
    par.float('d1', 1.), # Axis 1 sampling
    par.float('d2', 1.)] # Axis 2 sampling
    
    unit = [
    par.string('unit1', 'unknown'),
    par.string('unit2', 'unknown')]

    lbl = [
    par.string('label1', 'unknown'),
    par.string('label2', 'unknown')]

    ##### End reading parameters #####

    stdin = csv.reader(sys.stdin, delimiter=delim)
 
    # Copy stdin so we can go back through it
    lines = []
    i = 0
    nr_cols_cst = True # Whether the nr of values in rows is constant

    if header: # the first line contains header keys
        line = stdin.next()
        k = 0
        for name in line:
            k += 1
            print 'key%d=%s' % (k,name)
        
    # Find max nr of elements in a row
    for line in stdin:
        if line == []: # throw away blank lines
            continue
        curline = [float(x) for x in map(lambda x: x or '0', line)]
        if numtype == 'int':
            curline = [int(x) for x in curline]
        lines.append(curline)
        llen = len(curline)
        i+=1 # We have successfully read line i
        if i==1:
            n2 = llen
        elif llen != n2:
            nr_cols_cst = False
            if (llen < n2 and truncate) or (llen > n2 and not truncate):
                n2 = llen

    n1 = len(lines)

    if not nr_cols_cst: # Truncate or append as necessary
        for i in range(n1):
            line = lines[i]
            lines[i] = ivlad.trunc_or_append(n2, line, 0, verb)

    # Avoiding to add a second dimension of length 1
    if n1 == 1:
        ndim_out = 1
        n = [n2]
        o = [o[0]]
        d = [d[0]]
        unit = [unit[0]]
        lbl = [lbl[0]]
    else:
        ndim_out = 2
        n = [n2, n1]

    out = ooio.RSFfile(ooio.stdout,par,ndim=ndim_out,intent='out',dtype=numtype) 
    out.set_hdr_info(n, o, d, unit, lbl)
    
    if debug:
        out.print_self('out')
        out.hdr.print_self('out.hdr')
        out.dat.print_self('out.dat')
        ivlad.msg(ivlad.hr)

    for line in lines:
        for val in line:
            out.write(val)

    return ivlad.unix_success