giuliomoro@0: #!/usr/bin/python
giuliomoro@0: # -*- coding: utf-8 -*-
giuliomoro@0: 
giuliomoro@0: import xml.etree.ElementTree as ET
giuliomoro@0: import os       # for getting files from directory
giuliomoro@0: import operator # for sorting data with multiple keys
giuliomoro@0: import sys      # for accessing command line arguments
giuliomoro@0: 
giuliomoro@0: # Command line arguments
giuliomoro@0: assert len(sys.argv)<3, "evaluation_stats takes at most 1 command line argument\n"+\
giuliomoro@0:                         "Use: python evaluation_stats.py [results_folder]"
giuliomoro@0: 
giuliomoro@0: # XML results files location
giuliomoro@0: if len(sys.argv) == 1:
giuliomoro@0:     folder_name = "../saves"    # Looks in 'saves/' folder from 'scripts/' folder
giuliomoro@0:     print "Use: python evaluation_stats.py [results_folder]"
giuliomoro@0:     print "Using default path: " + folder_name
giuliomoro@0: elif len(sys.argv) == 2:
giuliomoro@0:     folder_name = sys.argv[1]   # First command line argument is folder
giuliomoro@0: 
giuliomoro@0: # Turn number of seconds (int) to '[minutes] min [seconds] s' (string)
giuliomoro@0: def seconds2timestr(time_in_seconds):
giuliomoro@0:     time_in_minutes = int(time_in_seconds/60)
giuliomoro@0:     remaining_seconds = int(time_in_seconds%60)
giuliomoro@0:     return str(time_in_minutes) + " min " + str(remaining_seconds) + " s"
giuliomoro@0: 
giuliomoro@0: # stats initialisation
giuliomoro@0: number_of_XML_files  = 0
giuliomoro@0: number_of_pages      = 0
giuliomoro@0: number_of_fragments  = 0
giuliomoro@0: total_empty_comments = 0
giuliomoro@0: total_not_played     = 0
giuliomoro@0: total_not_moved      = 0
giuliomoro@0: time_per_page_accum  = 0
giuliomoro@0: 
giuliomoro@0: # arrays initialisation
giuliomoro@0: page_names       = []
giuliomoro@0: page_count       = []
giuliomoro@0: duration_page    = []      # duration of experiment in function of page content
giuliomoro@0: duration_order   = []      # duration of experiment in function of page number
giuliomoro@0: fragments_per_page = []    # number of fragments for corresponding page
giuliomoro@0: 
giuliomoro@0: # get every XML file in folder
giuliomoro@0: files_list = os.listdir(folder_name)
giuliomoro@0: for file in files_list: # iterate over all files in files_list
giuliomoro@0:     if file.endswith(".xml"): # check if XML file
giuliomoro@0:         number_of_XML_files += 1
giuliomoro@0:         tree = ET.parse(folder_name + '/' + file)
giuliomoro@0:         root = tree.getroot()
giuliomoro@0:         
giuliomoro@0:         print file # print file name (subject name)
giuliomoro@0:         
giuliomoro@0:         # reset for new subject
giuliomoro@0:         total_duration = 0
giuliomoro@0:         page_number = 0
giuliomoro@0:         
giuliomoro@0:         # get list of all page names
giuliomoro@0:         for audioholder in root.findall("./page"):   # iterate over pages
giuliomoro@0:             page_name = audioholder.get('id')               # get page name
giuliomoro@0:             
giuliomoro@0:             if page_name is None: # ignore 'empty' audio_holders
giuliomoro@0:                 print "WARNING: " + file + " contains empty audio holder. (evaluation_stats.py)"
giuliomoro@0:                 break # move on to next
giuliomoro@0:             
giuliomoro@0:             number_of_comments = 0 # for this page
giuliomoro@0:             number_of_missing_comments = 0 # for this page
giuliomoro@0:             not_played = 0 # for this page
giuliomoro@0:             not_moved = 0 # for this page
giuliomoro@0:             
giuliomoro@0:             # 'testTime' keeps total duration: subtract time so far for duration of this audioholder
giuliomoro@0:             duration = float(audioholder.find("./metric/metricresult[@id='testTime']").text) - total_duration
giuliomoro@0:             
giuliomoro@0:             # total duration of test
giuliomoro@0:             total_duration += duration
giuliomoro@0:             
giuliomoro@0:             # number of audio elements
giuliomoro@0:             audioelements = audioholder.findall("./audioelement") # get audioelements
giuliomoro@0:             number_of_fragments += len(audioelements) # add length of this list to total
giuliomoro@0:             
giuliomoro@0:             # number of comments (interesting if comments not mandatory)
giuliomoro@0:             for audioelement in audioelements:
giuliomoro@0:                 response = audioelement.find("./comment/response")
giuliomoro@0:                 was_played = audioelement.find("./metric/metricresult/[@name='elementFlagListenedTo']")
giuliomoro@0:                 was_moved = audioelement.find("./metric/metricresult/[@name='elementFlagMoved']")
giuliomoro@0:                 if response.text is not None and len(response.text) > 1: 
giuliomoro@0:                     number_of_comments += 1
giuliomoro@0:                 else: 
giuliomoro@0:                     number_of_missing_comments += 1
giuliomoro@0:                 if was_played is not None and was_played.text == 'false': 
giuliomoro@0:                     not_played += 1
giuliomoro@0:                 if was_moved is not None and was_moved.text == 'false': 
giuliomoro@0:                     not_moved += 1
giuliomoro@0:             
giuliomoro@0:             # update global counters
giuliomoro@0:             total_empty_comments += number_of_missing_comments
giuliomoro@0:             total_not_played += not_played
giuliomoro@0:             total_not_moved += not_moved
giuliomoro@0:             
giuliomoro@0:             # print audioholder id and duration
giuliomoro@0:             print "    " + page_name + ": " + seconds2timestr(duration) + ", "\
giuliomoro@0:                   + str(number_of_comments)+"/"\
giuliomoro@0:                   +str(number_of_comments+number_of_missing_comments)+" comments"
giuliomoro@0:             
giuliomoro@0:             # number of audio elements not played
giuliomoro@0:             if not_played > 1:
giuliomoro@0:                 print 'ATTENTION: '+str(not_played)+' fragments were not listened to!'
giuliomoro@0:             if not_played == 1: 
giuliomoro@0:                 print 'ATTENTION: one fragment was not listened to!'
giuliomoro@0:             
giuliomoro@0:             # number of audio element markers not moved
giuliomoro@0:             if not_moved > 1:
giuliomoro@0:                 print 'ATTENTION: '+str(not_moved)+' markers were not moved!'
giuliomoro@0:             if not_moved == 1: 
giuliomoro@0:                 print 'ATTENTION: one marker was not moved!'
giuliomoro@0:             
giuliomoro@0:             # keep track of duration in function of page index
giuliomoro@0:             if len(duration_order)>page_number:
giuliomoro@0:                 duration_order[page_number].append(duration)
giuliomoro@0:             else:
giuliomoro@0:                 duration_order.append([duration])
giuliomoro@0:                 
giuliomoro@0:             # keep list of audioholder ids and count how many times each audioholder id
giuliomoro@0:             # was tested, how long it took, and how many fragments there were (if number of 
giuliomoro@0:             # fragments is different, store as different audioholder id)
giuliomoro@0:             if page_name in page_names: 
giuliomoro@0:                 page_index = page_names.index(page_name) # get index
giuliomoro@0:                 # check if number of audioelements the same
giuliomoro@0:                 if len(audioelements) == fragments_per_page[page_index]: 
giuliomoro@0:                     page_count[page_index] += 1
giuliomoro@0:                     duration_page[page_index].append(duration)
giuliomoro@0:                 else: # make new entry
giuliomoro@0:                     alt_page_name = page_name+"("+str(len(audioelements))+")"
giuliomoro@0:                     if alt_page_name in page_names: # if already there
giuliomoro@0:                         alt_page_index = page_names.index(alt_page_name) # get index
giuliomoro@0:                         page_count[alt_page_index] += 1
giuliomoro@0:                         duration_page[alt_page_index].append(duration)
giuliomoro@0:                     else: 
giuliomoro@0:                         page_names.append(alt_page_name)
giuliomoro@0:                         page_count.append(1)
giuliomoro@0:                         duration_page.append([duration])
giuliomoro@0:                         fragments_per_page.append(len(audioelements))
giuliomoro@0:             else: 
giuliomoro@0:                 page_names.append(page_name)
giuliomoro@0:                 page_count.append(1)
giuliomoro@0:                 duration_page.append([duration])
giuliomoro@0:                 fragments_per_page.append(len(audioelements))
giuliomoro@0:                 
giuliomoro@0:             # bookkeeping
giuliomoro@0:             page_number += 1 # increase page count for this specific test
giuliomoro@0:             number_of_pages += 1 # increase total number of pages
giuliomoro@0:             time_per_page_accum += duration # total duration (for average time spent per page)
giuliomoro@0: 
giuliomoro@0:         # print total duration of this test
giuliomoro@0:         print "    TOTAL: " + seconds2timestr(total_duration)
giuliomoro@0: 
giuliomoro@0: 
giuliomoro@0: # PRINT EVERYTHING
giuliomoro@0: 
giuliomoro@0: print "Number of XML files: " + str(number_of_XML_files)
giuliomoro@0: print "Number of pages: " + str(number_of_pages)
giuliomoro@0: print "Number of fragments: " + str(number_of_fragments)
giuliomoro@0: print "Number of empty comments: " + str(total_empty_comments) +\
giuliomoro@0:       " (" + str(round(100.0*total_empty_comments/number_of_fragments,2)) + "%)"
giuliomoro@0: print "Number of unplayed fragments: " + str(total_not_played) +\
giuliomoro@0:       " (" + str(round(100.0*total_not_played/number_of_fragments,2)) + "%)"
giuliomoro@0: print "Number of unmoved markers: " + str(total_not_moved) +\
giuliomoro@0:       " (" + str(round(100.0*total_not_moved/number_of_fragments,2)) + "%)"
giuliomoro@0: print "Average time per page: " + seconds2timestr(time_per_page_accum/number_of_pages)
giuliomoro@0: 
giuliomoro@0: # Pages and number of times tested
giuliomoro@0: page_count_strings = list(str(x) for x in page_count)
giuliomoro@0: count_list = page_names + page_count_strings
giuliomoro@0: count_list[::2] = page_names
giuliomoro@0: count_list[1::2] = page_count_strings
giuliomoro@0: print "Pages tested: " + str(count_list)
giuliomoro@0: 
giuliomoro@0: # Average duration for first, second, ... page
giuliomoro@0: print "Average duration per page:"
giuliomoro@0: for page_number in range(len(duration_order)): 
giuliomoro@0:     print "        page " + str(page_number+1) + ": " +\
giuliomoro@0:         seconds2timestr(sum(duration_order[page_number])/len(duration_order[page_number])) +\
giuliomoro@0:             " ("+str(len(duration_order[page_number]))+" subjects)"
giuliomoro@0: 
giuliomoro@0: 
giuliomoro@0: # Sort pages by number of audioelements, then by duration
giuliomoro@0: 
giuliomoro@0: # average duration and number of subjects per page
giuliomoro@0: average_duration_page = []
giuliomoro@0: number_of_subjects_page = []
giuliomoro@0: for line in duration_page:
giuliomoro@0:     number_of_subjects_page.append(len(line))
giuliomoro@0:     average_duration_page.append(sum(line)/len(line))
giuliomoro@0: 
giuliomoro@0: # combine and sort in function of number of audioelements and duration
giuliomoro@0: combined_list = [page_names, average_duration_page, fragments_per_page, number_of_subjects_page]
giuliomoro@0: combined_list = sorted(zip(*combined_list), key=operator.itemgetter(1, 2)) # sort
giuliomoro@0: 
giuliomoro@0: # Show average duration for all songs
giuliomoro@0: print "Average duration per audioholder:"
giuliomoro@0: for page_index in range(len(page_names)):
giuliomoro@0:     print "        "+combined_list[page_index][0] + ": " \
giuliomoro@0:           + seconds2timestr(combined_list[page_index][1]) \
giuliomoro@0:           + " (" + str(combined_list[page_index][3]) + " subjects, " \
giuliomoro@0:           + str(combined_list[page_index][2]) + " fragments)"
giuliomoro@0: 
giuliomoro@0: 
giuliomoro@0: #TODO
giuliomoro@0: # time per page in function of number of fragments (plot)
giuliomoro@0: # time per participant in function of number of pages
giuliomoro@0: # plot total time for each participant
giuliomoro@0: # plot total time
giuliomoro@0: # show 'count' per page (in order)
giuliomoro@0: 
giuliomoro@0: # clear up page_index <> page_count <> page_number confusion
giuliomoro@0: 
giuliomoro@0: # LaTeX -> PDF print out