djmoffat@718: #!/usr/bin/python djmoffat@718: djmoffat@718: import xml.etree.ElementTree as ET djmoffat@718: import os # list files in directory djmoffat@718: import sys # command line arguments djmoffat@718: import matplotlib.pyplot as plt # plots djmoffat@718: import matplotlib.patches as patches # rectangles djmoffat@718: djmoffat@718: # COMMAND LINE ARGUMENTS djmoffat@718: djmoffat@718: assert len(sys.argv)<3, "timeline_view takes at most 1 command line argument\n"+\ djmoffat@718: "Use: python timeline_view.py [XML_files_location]" djmoffat@718: djmoffat@718: # XML results files location djmoffat@718: if len(sys.argv) == 1: djmoffat@718: folder_name = "../saves" # Looks in 'saves/' folder from 'scripts/' folder djmoffat@718: print "Use: python timeline_view.py [XML_files_location]" djmoffat@718: print "Using default path: " + folder_name djmoffat@718: elif len(sys.argv) == 2: djmoffat@718: folder_name = sys.argv[1] # First command line argument is folder djmoffat@718: djmoffat@718: # check if folder_name exists djmoffat@718: if not os.path.exists(folder_name): djmoffat@718: #the file is not there djmoffat@718: print "Folder '"+folder_name+"' does not exist." djmoffat@718: sys.exit() # terminate script execution djmoffat@718: elif not os.access(os.path.dirname(folder_name), os.W_OK): djmoffat@718: #the file does exist but write privileges are not given djmoffat@718: print "No write privileges in folder '"+folder_name+"'." djmoffat@718: djmoffat@718: djmoffat@718: # CONFIGURATION djmoffat@718: djmoffat@718: # Folder where to store timelines djmoffat@718: timeline_folder = folder_name + '/timelines/' # Stores in 'saves/timelines/' djmoffat@718: djmoffat@718: # Font settings djmoffat@718: font = {'weight' : 'bold', djmoffat@718: 'size' : 16} djmoffat@718: plt.rc('font', **font) djmoffat@718: djmoffat@718: # Colormap for to cycle through djmoffat@718: colormap = ['b', 'r', 'g', 'c', 'm', 'y', 'k'] djmoffat@718: djmoffat@718: # if enabled, x-axis shows time per audioholder, not total test time djmoffat@718: show_audioholder_time = True djmoffat@718: djmoffat@718: # bar height (<1 to avoid overlapping) djmoffat@718: bar_height = 0.6 djmoffat@718: djmoffat@718: # figure size djmoffat@718: fig_width = 25 djmoffat@718: fig_height = 5 djmoffat@718: djmoffat@718: djmoffat@718: # CODE djmoffat@718: djmoffat@718: # create timeline_folder if not yet created djmoffat@718: if not os.path.exists(timeline_folder): djmoffat@718: os.makedirs(timeline_folder) djmoffat@718: djmoffat@718: # get every XML file in folder djmoffat@718: for file in os.listdir(folder_name): djmoffat@718: if file.endswith(".xml"): djmoffat@718: tree = ET.parse(folder_name + '/' + file) djmoffat@718: root = tree.getroot() djmoffat@718: subject_id = file[:-4] # drop '.xml' djmoffat@718: djmoffat@718: time_offset = 0 # test starts at zero djmoffat@718: djmoffat@718: # ONE TIMELINE PER PAGE - make new plot per page djmoffat@718: djmoffat@718: # get list of all page names djmoffat@718: for audioholder in root.findall("./audioholder"): # iterate over pages djmoffat@718: page_name = audioholder.get('id') # get page name djmoffat@718: plot_empty = True # check if any data is plotted djmoffat@718: djmoffat@718: if page_name is None: # ignore 'empty' audio_holders djmoffat@718: break djmoffat@718: djmoffat@718: # SORT AUDIO ELEMENTS ALPHABETICALLY djmoffat@718: audioelements = audioholder.findall("./audioelement") djmoffat@718: djmoffat@718: data = [] djmoffat@718: for elem in audioelements: # from http://effbot.org/zone/element-sort.htm djmoffat@718: key = elem.get("id") djmoffat@718: data.append((key, elem)) djmoffat@718: data.sort() djmoffat@718: djmoffat@718: N_audioelements = len(audioelements) # number of audio elements for this page djmoffat@718: increment = 0 # increased for every new audioelement djmoffat@718: audioelements_names = [] # store names of audioelements djmoffat@718: djmoffat@718: # get axes handle djmoffat@718: fig = plt.figure(figsize=(fig_width, fig_height)) djmoffat@718: ax = fig.add_subplot(111) #, aspect='equal' djmoffat@718: djmoffat@718: # for page [page_name], print comments related to fragment [id] djmoffat@718: for tuple in data: djmoffat@718: audioelement = tuple[1] djmoffat@718: if audioelement is not None: # Check it exists djmoffat@718: audio_id = str(audioelement.get('id')) djmoffat@718: audioelements_names.append(audio_id) djmoffat@718: djmoffat@718: # for this audioelement, loop over all listen events djmoffat@718: listen_events = audioelement.findall("./metric/metricresult/[@name='elementListenTracker']/event") djmoffat@718: for event in listen_events: djmoffat@718: # mark this plot as not empty djmoffat@718: plot_empty = False djmoffat@718: djmoffat@718: # get testtime: start and stop djmoffat@718: start_time = float(event.find('testtime').get('start'))-time_offset djmoffat@718: stop_time = float(event.find('testtime').get('stop'))-time_offset djmoffat@718: # event lines: djmoffat@718: ax.plot([start_time, start_time], # x-values djmoffat@718: [0, N_audioelements+1], # y-values djmoffat@718: color='k' djmoffat@718: ) djmoffat@718: ax.plot([stop_time, stop_time], # x-values djmoffat@718: [0, N_audioelements+1], # y-values djmoffat@718: color='k' djmoffat@718: ) djmoffat@718: # plot time: djmoffat@718: ax.add_patch( djmoffat@718: patches.Rectangle( djmoffat@718: (start_time, N_audioelements-increment-bar_height/2), # (x, y) djmoffat@718: stop_time - start_time, # width djmoffat@718: bar_height, # height djmoffat@718: color=colormap[increment%len(colormap)] # colour djmoffat@718: ) djmoffat@718: ) djmoffat@718: djmoffat@718: increment+=1 # to next audioelement djmoffat@718: djmoffat@718: # subtract total audioholder length from subsequent audioholder event times djmoffat@718: audioholder_time = audioholder.find("./metric/metricresult/[@id='testTime']") djmoffat@718: if audioholder_time is not None and show_audioholder_time: djmoffat@718: time_offset = float(audioholder_time.text) djmoffat@718: djmoffat@718: if not plot_empty: djmoffat@718: # set plot parameters djmoffat@718: plt.title('Timeline ' + file + ": "+page_name) djmoffat@718: plt.xlabel('Time [seconds]') djmoffat@718: plt.ylabel('Fragment') djmoffat@718: plt.ylim(0, N_audioelements+1) djmoffat@718: djmoffat@718: #y-ticks: fragment IDs, top to bottom djmoffat@718: plt.yticks(range(N_audioelements, 0, -1), audioelements_names) # show fragment names djmoffat@718: djmoffat@718: djmoffat@718: #plt.show() # uncomment to show plot; comment when just saving djmoffat@718: #exit() djmoffat@718: djmoffat@718: plt.savefig(timeline_folder+subject_id+"-"+page_name+".pdf", bbox_inches='tight') djmoffat@718: plt.close() djmoffat@718: djmoffat@718: #TODO: if 'nonsensical' or unknown: dashed line until next event djmoffat@718: #TODO: Vertical lines for fragment looping point djmoffat@718: