comparison scripts/evaluation_stats.py @ 2066:a13e5cf40feb

Scripts: evaluation_stats shows number of total fragments

 
 import xml.etree.ElementTree as ET
 import os
 
 # XML results files location (modify as needed):
-folder_name = "../saves"    # Looks in 'saves/' folder from 'scripts/' folder
+folder_name = "xml_with_statements"    # Looks in 'saves/' folder from 'scripts/' folder
 
 # Turn number of seconds (int) to '[minutes] min [seconds] s' (string)
 def seconds2timestr(time_in_seconds):
     time_in_minutes = int(time_in_seconds/60)
     remaining_seconds = int(time_in_seconds%60)
     return str(time_in_minutes) + " min " + str(remaining_seconds) + " s"
 
 # stats initialisation
 number_of_XML_files = 0
 number_of_pages = 0
+number_of_fragments = 0
 time_per_page_accum = 0
 
 # arrays initialisation
 page_names = []
 page_count = []

             page_name = audioholder.get('id')               # get page name
             
             if page_name is None: # ignore 'empty' audio_holders
                 print "WARNING: " + file + " contains empty audio holder. (evaluation_stats.py)"
                 break # move on to next
-                
+            
             # keep list of audioholder ids and count how many times each audioholder id was tested
             if page_name in page_names:
                 page_index = page_names.index(page_name) # get index
                 page_count[page_index] += 1
-            else: 
+            else:
                 page_names.append(page_name)
                 page_count.append(1)
             
             # 'testTime' keeps total duration: subtract time so far for duration of this audioholder
             duration = float(audioholder.find("./metric/metricresult[@id='testTime']").text) - total_duration

             # keep track of duration in function of page index
             if len(duration_order)>page_number:
                 duration_order[page_number].append(duration)
             else:
                 duration_order.append([duration])
+
+            # number of audio elements
+            audioelements = audioholder.findall("./audioelement") # get audioelements
+            number_of_fragments += len(audioelements) # add length of this list to total
             
             page_number += 1 # increase page count for this specific test
             number_of_pages += 1 # increase total number of pages
             time_per_page_accum += duration # total duration (for average time spent per page)
-        
-        # print total duration of this test
-        print "    TOTAL: " + seconds2timestr(total_duration)
-                    
+
+    # print total duration of this test
+    print "    TOTAL: " + seconds2timestr(total_duration)
+
+
 # PRINT EVERYTHING
 
 print "Number of XML files: " + str(number_of_XML_files)
 print "Number of pages: " + str(number_of_pages)
+print "Number of fragments: " + str(number_of_fragments)
 print "Average time per page: " + seconds2timestr(time_per_page_accum/number_of_pages)
 page_count_strings = list(str(x) for x in page_count)
 count_list = page_names + page_count_strings
 count_list[::2] = page_names
 count_list[1::2] = page_count_strings
 print "Pages tested: " + str(count_list)
 
 # Average duration for first, second, ... page
 for page_number in range(len(duration_order)): #TODO make maximum page number automatic
     print "Average duration page " + str(page_number+1) + ": " +\
-          seconds2timestr(sum(duration_order[page_number])/len(duration_order[page_number])) +\
-          " ("+str(len(duration_order[page_number]))+" subjects)"
-          
+        seconds2timestr(sum(duration_order[page_number])/len(duration_order[page_number])) +\
+            " ("+str(len(duration_order[page_number]))+" subjects)"
+
 
 #TODO
 # time per page in function of number of fragments (plot)
 # time per participant in function of number of pages
 # plot total time for each participant
-# plot total time 
+# plot total time
 # show 'count' per page (in order)
 
 # clear up page_index <> page_count <> page_number confusion