import os
from nptdms import TdmsFile
import numpy as np
import tdsmParserMultis

tdmsDirectory = '/Users/schimmt/Documents/Multis/app/FileAssociation/MULTIS_trials/CMULTIS002-3_associated/Data/'

tdmsFiles = sorted(os.listdir(tdmsDirectory))

def inprogress():
    print ("."),
count = len(tdmsFiles)
print ". " * count

pulseList = []
TdmsFilesNoPulse = []
TdmsFiles_WithPulse = []
lowerleg=[]
lowerarm=[]
upperleg=[]
upperarm=[]



for i in tdmsFiles:
    # print i[-14:-12]
    # if i.endswith('.tdms'):
    if i[0:3]=='006':
        # count -= 1
        inprogress()
        tdmsFileName = tdmsDirectory + i
        # tdmsData = tdsmParserMultisCopy.parseTDMSfile(TdmsDirectory + filename)   # reading all TDMS data
        tdmsFile = TdmsFile(tdmsFileName)
        try:  # check to see if pulse exists, if not, exclue TDMS file from directory

            # pulse1 = tdmsData[u'Sensor.Run Number Pulse Train'][u'Run Number Pulse Train']
            group = tdmsFile.groups()[5]
            # print i
            # print group
            pulse1 = tdmsFile.group_channels(group)[0].data
            pulseList.append(pulse1)
            TdmsFiles_WithPulse.append(i)
            data = tdsmParserMultis.parseTDMSfile(tdmsDirectory + i)

            if i[-14:-12] == 'LL':
                lowerleg.append(i)
            elif i[-14:-12] == 'LA':
                lowerarm.append(i)
            elif i[-14:-12] == 'UL':
                upperleg.append(i)
            elif i[-14:-12] == 'UA':
                upperarm.append(i)

                humerus = np.array(data[u'State.Probe-Humerus'][u'Probe-Humerus Position x'])
                print humerus
        except KeyError:
            TdmsFilesNoPulse.append(i)  # must create new list because some files have no Pulse Train
            # print " File '" + i + "' does not have a pulse train"

print ' '
print 'LL'
for i in lowerleg:
    print i
print 'UL'
for i in upperleg:
    print i
print 'LA'
for i in lowerarm:
    print i
print 'UA'
for i in upperarm:
    print i