parser = argparse.ArgumentParser(description='Process EEG Data.')

parser.add_argument('-i', nargs=1 , required=True, dest='indir',

help='Directory containing EEG Files.')

parser.add_argument('-o', nargs=1, required=True, dest='outdir',

help='Path of processed data and reports. New \

directory with timestamp will be created inside this\

folder.')

parser.add_argument('--type', nargs=1, required=True, dest='intype',

choices = ['raw', 'pam1hz'], help='Select type of\

input data: Raw 512 Hz data or Neurosky\

output data of 1Hz having proprietary fields: \

PoorSignal, Attention and Meditation. ')

parser.add_argument('--interpolate', action='store_true',

help='Interpolate 512Hz time stamps on 512Hz data\

having 1Hz time stamps (raw data available from CMU). \

[raw]')

parser.add_argument('--kernelsvm', action='store_true',

help='Perform Kernelized SVC on interpolated raw data.\

[raw]')

parser.add_argument('--chartsforpaper', action='store_true',

help='Create charts in single page PDF files. [raw]')

parser.add_argument('--plotsubjects', action='store_true',

help='Create charts for 1Hz data of subjects. \

[pam1hz]')

parser.add_argument('--plotavgrolmed', nargs=1, dest='nsamp',

type=int,

help='Plot average rolling median of first NSAMP \

samples. Note: Rolling median is usually downsampled\

to 10Hz, so 1st second = 10 samples. [raw]')

parser.add_argument('--plotavgraw', nargs=1, dest='nsampraw',

type=int,

help='Plot raw signal averaged over all subjects for \

1st NSAMPRAW samples. [raw]')

parser.add_argument('--plotraw', nargs=1, dest='nraw',

type=int,

help='Plot raw signal of all subjects for \

1st NRAW samples. [raw]')

parser.add_argument('--filter', action='store_true',

help='Use Butteworth filter and plot charts for 1Hz\

subject data. [pam1hz]')

parser.add_argument('--stats', action='store_true',

help='General statistics of data [raw]')

parser.add_argument('--clean', action='store_true',

help='Remove data with 0 attention or meditation\

values; Remove data with > 0 poor signal value.\

[pam1hz]')

args = parser.parse_args()

mydir = join(out_dir,

datetime.datetime.now().strftime('%Y-%m-%d_%H-%M-%S'))

try:

os.makedirs(mydir)

os.makedirs(join(mydir,'data'))

os.makedirs(join(mydir,'reports'))

except OSError, e:

if e.errno != 17:

raise # This was not a "directory exist" error