#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# random-muscle-activation.py -- Generate random muscle activation data in CSV
#
# Copyright (C) 2013 Tobias Klauser <tklauser@distanz.ch>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as
# published by the Free Software Foundation.

import os
import sys
import getopt
import numpy as np
import plotcsv

DEFAULT_N = 1000
DEFAULT_XMIN = 0.0
DEFAULT_XMAX = 100.0
DEFAULT_YMIN = 0.0
DEFAULT_YMAX = 50.0
DEFAULT_YDMIN = -0.5
DEFAULT_YDMAX = 0.5

def usage():
    print("""usage: {} [OPTION...] FILE...

Generate random muscle activation data in CSV.

options:
    -f    force overwrite of existing file
    -n N  set number of data points to N (default: {})
    -p    plot data after generating it
    -x N  set minimum x value to N (default: {})
    -X N  set maximum x value to N (default: {})
    -y N  set minimum y value to N (default: {})
    -Y N  set maximum y value to N (default: {})
    -d N  set minimum y change value to N (default: {})
    -D N  set maximum y change value to N (default: {})""".format(
        os.path.basename(sys.argv[0]), DEFAULT_N,
        DEFAULT_XMIN, DEFAULT_XMAX,
        DEFAULT_YMIN, DEFAULT_YMAX,
        DEFAULT_YDMIN, DEFAULT_YDMAX))

def generate_activation_data(N, xmin, xmax, ymin, ymax, ydmin, ydmax):
    X = np.array([ ((x / float(N)) * (xmax - xmin)) + xmin for x in range(0, N) ])
    Y = np.zeros(N)
    # Y = np.array([ ((y / float(N)) * (ymax - ymin)) + ymin for y in range(0, N) ])

    Y[0] = (ymax - ymin) * np.random.random_sample() + ymin
    for y in range(1, N):
        Y[y] = Y[y-1] + ((ydmax - ydmin) * np.random.random_sample() + ydmin)
        if Y[y] < ymin:
            Y[y] = ymin
        if Y[y] > ymax:
            Y[y] = ymax

    act = np.array((X, Y)).transpose()

    return act

def main():
    try:
        opts, args = getopt.getopt(sys.argv[1:], "fn:px:X:y:Y:d:D:h")
    except getopt.GetoptError as err:
        print(str(err))
        usage()
        sys.exit(-1)

    do_overwrite = False
    do_plot = False
    N = DEFAULT_N
    xmin = DEFAULT_XMIN
    xmax = DEFAULT_XMAX
    ymin = DEFAULT_YMIN
    ymax = DEFAULT_YMAX
    ydmin = DEFAULT_YDMIN
    ydmax = DEFAULT_YDMAX
    for o, a in opts:
        if o == '-f':
            do_overwrite = True
        elif o == '-n':
            try:
                N = int(a)
            except ValueError:
                print("Error: number of data points must be an integer > 0")
                usage()
                sys.exit(-1)
        elif o == '-p':
            do_plot = True
        elif o == '-x':
            try:
                xmin = float(a)
            except ValueError:
                print("Error: xmin must be a float")
                usage()
                sys.exit(-1)
        elif o == '-X':
            try:
                xmax = float(a)
            except ValueError:
                print("Error: xmax must be a float")
                usage()
                sys.exit(-1)
        elif o == '-y':
            try:
                ymin = float(a)
            except ValueError:
                print("Error: ymin must be a float")
                usage()
                sys.exit(-1)
        elif o == '-Y':
            try:
                ymax = float(a)
            except ValueError:
                print("Error: ymax must be a float")
                usage()
                sys.exit(-1)
        elif o == '-d':
            try:
                ydmin = float(a)
            except ValueError:
                print("Error: ydmin must be a float")
                usage()
                sys.exit(-1)
        elif o == '-D':
            try:
                ydmax = float(a)
            except ValueError:
                print("Error: ydmax must be a float")
                usage()
                sys.exit(-1)
        elif o == '-h':
            usage()
            sys.exit(0)

    if N <= 0:
        print("Error: invalid number of data points ({}), must be > 0".format(N))
        usage()
        sys.exit(-1)

    if len(args) < 1:
        print("Error: no output file(s) specified")
        usage()
        sys.exit(-1)

    for a in args:
        if not do_overwrite and os.path.exists(a):
            print("Error: output file {} already exists, use -f to overwrite".format(a))
            sys.exit(-1)

    print("Generating {} files with {} data points each...".format(len(args), N))
    for a in args:
        with open(a, 'w') as f:
            act = generate_activation_data(N, xmin, xmax, ymin, ymax, ydmin, ydmax)
            f.write('x,y\n')
            for x, y in act:
                f.write('{},{}\n'.format(x, y))
            if do_plot:
                print("Plotting data for {}, please close window to continue...".format(a))
                plotcsv.plot(act[:,0], act[:,1], xmax, ymax, 'x', 'y', a, False, False)

    sys.exit(0)

if __name__ == '__main__':
    main()