#!/usr/bin/env python

# -*- coding: utf-8 -*-

import argparse, time, os, instruments, inspect

#==============================================================================

# Path
PATH = os.getcwd()
# Sampling time acquisition
SAMPLING_TIME = 1
# File duration
FILE_DURATION = 3600*24
# Default instrument
INSTRUMENT = None
# Default instrument adress
ADRESS = None
#Default val type
VAL_TYPE = None

#==============================================================================

def parse():
    """
    Specific parsing procedure for transfering data from any abstract instrument.
    :returns: populated namespace (parser)
    """

    parser = argparse.ArgumentParser(description = 'Acquire data from INSTRUMENT',
                                     epilog = 'Example: \'./datalogger.py -i myInstrument -st 10\' logs myInstrument every 10 seconds to output file YYYYMMDD-HHMMSS-INSTRUMENT.dat')

    parser.add_argument('-l',
                        action='store_true',
                        dest='list',
                        default=False,
                        help='List all available instruments')

    parser.add_argument('-I',
                        action='store',
                        dest='instLog',
                        default=INSTRUMENT,
                        help='Instrument to log (default '+str(INSTRUMENT)+')')

    parser.add_argument('-ip',
                        action='store',
                        dest='adress',
                        default=ADRESS,
                        help='Adress of instrument (IP, USB...) (default '+str(ADRESS)+')')

    parser.add_argument('-v',
                        action='store',
                        dest='vtype',
                        default=VAL_TYPE,
                        help='Value type to measure (default '+str(VAL_TYPE)+')')

    parser.add_argument('-st',
                        action='store',
                        dest='samplingtime',
                        default=SAMPLING_TIME,
                        help='Sampling time acquistion (default '+str(SAMPLING_TIME)+' second)')

    parser.add_argument('-fd',
                        action='store',
                        dest='fileduration',
                        default=FILE_DURATION,
                        help='File duration (infinite : \'-fd -1\') (default '+str(FILE_DURATION)+' second)')

    parser.add_argument('-p',
                        action='store',
                        dest='path',
                        default=PATH,
                        help='Absolute path (default '+PATH+')')

    args = parser.parse_args()
    return args

#==============================================================================

def acq_routine(instrument, path, samplingtime, fileduration):
    instrument.connect()
    t0 = time.time()
    filename = time.strftime("%Y%m%d-%H%M%S", time.gmtime(t0)) + '-' + instrument.model() + '.dat'
    print('Opening %s' %filename)
    try:
        year = time.strftime("%Y", time.gmtime(t0))
        month = time.strftime("%Y-%m", time.gmtime(t0))
        os.chdir(path + '/' + year + '/' + month)
    except:
        try:
            os.chdir(path + '/' + year)
            os.mkdir(month)
            os.chdir(path + '/' + year + '/' + month)
        except:
            os.chdir(path)
            os.mkdir(year)
            os.chdir(path + '/' + year)
            os.mkdir(month)
            os.chdir(path + '/' + year + '/' + month)

    data_file = open(filename, 'wr', 0)

    # Infinite loop
    while True:
        # tic
        tic = time.time()

        if time.time() - t0 >= fileduration:
            t0 = time.time()
            print('Closing %s\n' %filename)
            data_file.close()

            try:
                year = time.strftime("%Y", time.gmtime(t0))
                month = time.strftime("%Y-%m", time.gmtime(t0))
                os.chdir(path + '/' + year + '/' + month)
            except:
                try:
                    os.chdir(path + '/' + year)
                    os.mkdir(month)
                    os.chdir(path + '/' + year + '/' + month)
                except:
                    os.chdir(path)
                    os.mkdir(year)
                    os.chdir(path + '/' + year)
                    os.mkdir(month)
                    os.chdir(path + '/' + year + '/' + month)

            filename = time.strftime("%Y%m%d-%H%M%S", time.gmtime(t0)) + '-' + instrument.model() + '.dat'
            print('Opening %s\n' %filename)
            data_file = open(filename, 'wr', 0)

        try:
            try:
                #epoch time
                epoch = time.time()
                #MJD time
                mjd = epoch / 86400.0 + 40587
                # Meas values
                meas = instrument.getValue()
                meas = meas.replace(",", "\t")
                meas = meas.replace(";", "\t")
                meas = meas.replace("+", "")

                string = "%f\t%f\t%s" % (epoch, mjd, meas)
                data_file.write(string) # Write in a file
                print(string)

                # Sleep until sampletime
                time.sleep(samplingtime - (time.time() - tic))

            except Exception as ex:
                print 'Exception during controler data reading: ' + str(ex)

        except KeyboardInterrupt:
            print '\n  --> Disconnected'
            instrument.disconnect()
            data_file.close()

            # To stop the loop in a clean way
            break

#==============================================================================

def main():
    """
    Main script
    """
    # Parse command line
    args = parse()
    # path
    path = args.path
    # Sampling time
    samplingtime=float(args.samplingtime)
    # File duration
    fileduration=int(args.fileduration)
    # Instrument to log
    instLog = args.instLog
    # instrument adress
    adress = args.adress
    # val type
    vtype = args.vtype

    try:
        if args.list:
            print('\nInstruments:')
            for name, obj in inspect.getmembers(instruments):
                if inspect.ismodule(obj) and name.startswith('__') == False and name.startswith('abstract') == False:
                    print('\n' + name)
                    exec('print("\t" + instruments.%s.ALL_VAL_TYPE)'%name)

        else:
            if instLog == None:
                raise Exception('No instrument selected !')

            if adress == None and vtype == None:
                exec('myInstrument = instruments.%s.%s()'%(instLog, instLog))
            elif adress == None and vtype != None:
                exec('myInstrument = instruments.%s.%s(vtype="%s")'%(instLog, instLog, vtype))
            elif adress != None and vtype != None:
                exec('myInstrument = instruments.%s.%s(adress="%s", vtype="%s")'%(instLog, instLog, adress, vtype))
            acq_routine(myInstrument, path, samplingtime, fileduration)

    except Exception as ex:
            print 'Oops: '+str(ex)

#==============================================================================

if __name__ == "__main__":
    main()