Mercurial > hg > numerics
diff numerics/smooth.py @ 29:c2c92c8da611
add simple smoother
author | Jeff Hammel <k0scist@gmail.com> |
---|---|
date | Thu, 09 Oct 2014 15:01:01 -0700 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/numerics/smooth.py Thu Oct 09 15:01:01 2014 -0700 @@ -0,0 +1,70 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +""" +smoothing +""" + +# imports +import argparse +import os +import subprocess +import sys + +# module globals +__all__ = ['main', 'smooth', 'SmoothingParser'] + +def smooth(iterations, *values): + assert len(values) >= 2 + while iterations > 0: + inner = [sum(i) for i in zip(values[1:], values[:-1])] + left = [inner[0]] + inner[:] + right = inner[:] + [inner[-1]] + values = [0.25*sum(i) for i in zip(left, right)] + iterations -= 1 + return values + +class SmoothingParser(argparse.ArgumentParser): + """`smooth` CLI option parser""" + def __init__(self, **kwargs): + kwargs.setdefault('description', __doc__) + argparse.ArgumentParser.__init__(self, **kwargs) + self.add_argument('input', nargs='?', + type=argparse.FileType('r'), default=sys.stdin, + help='input file, or read from stdin if ommitted') + self.add_argument('-o', '--output', dest='output', + type=argparse.FileType('a'), default=sys.stdout, + help="output file to write to, or stdout") + self.add_argument('-n', '--iterations', dest='iterations', + type=int, default=1, + help="number of iterations to apply [DEFAULT: %(default)s]") + self.options = None + + def parse_args(self, *args, **kw): + options = argparse.ArgumentParser.parse_args(self, *args, **kw) + self.validate(options) + self.options = options + return options + + def validate(self, options): + """validate options""" + +def main(args=sys.argv[1:]): + """CLI""" + + # parse command line options + parser = SmoothingParser() + options = parser.parse_args(args) + + # read data + data = options.input.read().strip().split() + data = [float(i) for i in data] + + smoothed = smooth(options.iterations, *data) + + # write data + options.output.write('\n'.join([str(i) for i in smoothed])) + +if __name__ == '__main__': + main() +