Source code for bin2csv

#!/usr/bin/env python

"""
bin2csv
input raw binary, output csv

usage: bin2csv.py [-h] [--nchan NCHAN] [--word WORD] [--outroot OUTROOT]
                  [--out OUT] [--paste PASTE]
                  binfiles [binfiles ...]

bin2csv

positional arguments:
  binfiles           file[s] to convert

optional arguments:
  -h, --help         show this help message and exit
  --nchan NCHAN      number of channels
  --word WORD        int16|int32
  --outroot OUTROOT  output root directory
  --out OUT          explicit output name
  --paste PASTE      1: paste multiple files * 1 chan
"""

import csv
import argparse
import numpy as np
import os

[docs]def get_word_type(wtype): if wtype == 'int16': return np.int16 elif wtype == 'int32': return np.int32 else: print("ERROR, undefined word type {}".format(wtype)) exit(1)
[docs]def csv_name(args, binfile): if len(args.out) > 0: basename = args.out else: basename, extn = os.path.splitext(binfile) return "{}{}{}.csv".format(args.outroot, os.sep if len(args.outroot)>0 else '', basename)
[docs]def bin2csv_onesource_manychan(args): for src in args.binfiles: raw = np.fromfile(src, args.wtype) nrows = len(raw)/args.nchan chx = np.reshape(raw, (nrows, args.nchan)) with open(csv_name(args, src), 'w' ) as fout: writer = csv.writer(fout) for row in range(0, nrows): writer.writerow(chx[row,:])
[docs]def bin2csv_many_onechan_sources(args): chx = list() for binf in args.binfiles: chx.append(np.fromfile(binf, args.wtype)) lens = [ len(u) for u in chx ] nrows = lens[0] chxx = np.vstack(chx) csvf = csv_name(args, args.binfiles[0]) with open(csvf, 'w' ) as fout: writer = csv.writer(fout) for row in range(0, nrows): writer.writerow(chxx[:,row])
[docs]def bin2csv(args): args.wtype = get_word_type(args.word) if args.paste: bin2csv_many_onechan_sources(args) else: bin2csv_onesource_manychan(args)
[docs]def run_main(): parser = argparse.ArgumentParser(description='bin2csv') parser.add_argument('--nchan', default=1, type=int, help="number of channels") parser.add_argument('--word', default='int16', help="int16|int32") parser.add_argument('--outroot', default='', help="output root directory") parser.add_argument('--out', default='', help="explicit output name") parser.add_argument('--paste', default=0, type=int, help="1: paste multiple files * 1 chan") parser.add_argument('binfiles', nargs='+', help="file[s] to convert") bin2csv(parser.parse_args())
# execution starts here if __name__ == '__main__': run_main()