giuliomoro@15: #!/usr/bin/python
giuliomoro@15: # -*- coding: utf-8 -*-
giuliomoro@15: 
giuliomoro@15: import xml.etree.ElementTree as ET
giuliomoro@15: import os       # for getting files from directory
giuliomoro@15: import operator # for sorting data with multiple keys
giuliomoro@15: import sys      # for accessing command line arguments
giuliomoro@15: import subprocess # for calling pdflatex
giuliomoro@15: import shlex # for calling pdflatex
giuliomoro@15: import matplotlib.pyplot as plt # plots
giuliomoro@15: import numpy as np # numbers
giuliomoro@15: 
giuliomoro@15: # Command line arguments
giuliomoro@15: assert len(sys.argv)<4, "generate_report takes at most 2 command line arguments\n"+\
giuliomoro@15:                         "Use: python generate_report.py [results_folder] [no_render | -nr]"
giuliomoro@15: 
giuliomoro@15: render_figures = True
giuliomoro@15: 
giuliomoro@15: # XML results files location
giuliomoro@15: if len(sys.argv) == 1:
giuliomoro@15:     folder_name = "../saves/"    # Looks in 'saves/' folder from 'scripts/' folder
giuliomoro@15:     print("Use: python generate_report.py [results_folder] [no_render | -nr]")
giuliomoro@15:     print("Using default path: " + folder_name)
giuliomoro@15: elif len(sys.argv) == 2:
giuliomoro@15:     folder_name = sys.argv[1]   # First command line argument is folder
giuliomoro@15: elif len(sys.argv) == 3:
giuliomoro@15:     folder_name = sys.argv[1]   # First command line argument is folder
giuliomoro@15:     assert sys.argv[2] in ('no_render','-nr'), "Second argument not recognised. \n" +\
giuliomoro@15:            "Use: python generate_report.py [results_folder] [no_render | -nr]"
giuliomoro@15:     # Second command line argument is [no_render | -nr]
giuliomoro@15:     render_figures = False
giuliomoro@15: 
giuliomoro@15: def isNaN(num):
giuliomoro@15:     return num != num
giuliomoro@15: 
giuliomoro@15: # Turn number of seconds (int) to '[minutes] min [seconds] s' (string)
giuliomoro@15: def seconds2timestr(time_in_seconds):
giuliomoro@15:     if time_in_seconds is not None and not isNaN(time_in_seconds): 
giuliomoro@15:         time_in_minutes = int(time_in_seconds/60)
giuliomoro@15:         remaining_seconds = int(time_in_seconds%60)
giuliomoro@15:         return str(time_in_minutes) + " min " + str(remaining_seconds) + " s"
giuliomoro@15:     else:
giuliomoro@15:         return 'N/A'
giuliomoro@15: 
giuliomoro@15: # stats initialisation
giuliomoro@15: number_of_XML_files  = 0
giuliomoro@15: number_of_pages      = 0
giuliomoro@15: number_of_fragments  = 0
giuliomoro@15: total_empty_comments = 0
giuliomoro@15: total_not_played     = 0
giuliomoro@15: total_not_moved      = 0
giuliomoro@15: time_per_page_accum  = 0
giuliomoro@15: 
giuliomoro@15: # arrays initialisation
giuliomoro@15: page_names       = []
giuliomoro@15: real_page_names  = [] # regardless of differing numbers of fragments
giuliomoro@15: subject_count    = [] # subjects per page name
giuliomoro@15: page_count       = []
giuliomoro@15: duration_page    = []      # duration of experiment in function of page content
giuliomoro@15: duration_order   = []      # duration of experiment in function of page number
giuliomoro@15: fragments_per_page = []    # number of fragments for corresponding page
giuliomoro@15: 
giuliomoro@15: # survey stats
giuliomoro@15: gender = []
giuliomoro@15: age    = []
giuliomoro@15: 
giuliomoro@15: # get username if available
giuliomoro@15: for name in ('LOGNAME', 'USER', 'LNAME', 'USERNAME'):
giuliomoro@15:     user = os.environ.get(name)
giuliomoro@15:     if user:
giuliomoro@15:         break
giuliomoro@15:     else:
giuliomoro@15:         user = ''
giuliomoro@15: 
giuliomoro@15: 
giuliomoro@15: # begin LaTeX document
giuliomoro@15: header = r'''\documentclass[11pt, oneside]{article} 
giuliomoro@15:           \usepackage{geometry}
giuliomoro@15:           \geometry{a4paper}
giuliomoro@15:           \usepackage[parfill]{parskip} % empty line instead of indent
giuliomoro@15:           \usepackage{graphicx}    % figures
giuliomoro@15:           \usepackage[space]{grffile} % include figures with spaces in paths
giuliomoro@15:           \usepackage{hyperref}
giuliomoro@15:           \usepackage{tikz}           % pie charts
giuliomoro@15:           \title{Report}
giuliomoro@15:           \author{'''+\
giuliomoro@15:           user+\
giuliomoro@15:           r'''}
giuliomoro@15:           \graphicspath{{'''+\
giuliomoro@15:           folder_name+\
giuliomoro@15:           r'''}}
giuliomoro@15:           %\setcounter{section}{-1} % Summary section 0 so number of sections equals number of files
giuliomoro@15:           \begin{document}
giuliomoro@15:           \maketitle
giuliomoro@15:           This is an automatically generated report using the `generate\_report.py' Python script 
giuliomoro@15:           included with the Web Audio Evaluation Tool \cite{WAET} distribution which can be found 
giuliomoro@15:           at \texttt{code.soundsoftware.ac.uk/projects/webaudioevaluationtool}.
giuliomoro@15:           \tableofcontents
giuliomoro@15:           
giuliomoro@15:           '''
giuliomoro@15:           
giuliomoro@15: footer = '\n\t\t'+r'''\begin{thebibliography}{9}
giuliomoro@15:          \bibitem{WAET} % reference to accompanying publication
giuliomoro@15:         Nicholas Jillings, Brecht De Man, David Moffat and Joshua D. Reiss, 
giuliomoro@15:         ``Web Audio Evaluation Tool: A browser-based listening test environment,'' 
giuliomoro@15:         presented at the 12th Sound and Music Computing Conference, July 2015.
giuliomoro@15:         \end{thebibliography}
giuliomoro@15:         \end{document}'''
giuliomoro@15: 
giuliomoro@15: body = ''
giuliomoro@15: 
giuliomoro@15: # make sure folder_name ends in '/'
giuliomoro@15: folder_name = os.path.join(folder_name, '')
giuliomoro@15: 
giuliomoro@15: # generate images for later use
giuliomoro@15: if render_figures:
giuliomoro@15:     subprocess.call("python timeline_view_movement.py '"+folder_name+"'", shell=True)
giuliomoro@15:     subprocess.call("python score_parser.py '"+folder_name+"'", shell=True)
giuliomoro@15:     subprocess.call("python score_plot.py '"+folder_name+"ratings/'", shell=True)
giuliomoro@15: 
giuliomoro@15: # get every XML file in folder
giuliomoro@15: files_list = os.listdir(folder_name)
giuliomoro@15: for file in files_list: # iterate over all files in files_list
giuliomoro@15:     if file.endswith(".xml"): # check if XML file
giuliomoro@15:         number_of_XML_files += 1
giuliomoro@15:         tree = ET.parse(folder_name + file)
giuliomoro@15:         root = tree.getroot()
giuliomoro@15:         
giuliomoro@15:         # PRINT name as section
giuliomoro@15:         body+= '\n\section{'+file[:-4].capitalize()+'}\n' # make section header from name without extension
giuliomoro@15:         
giuliomoro@15:         # reset for new subject
giuliomoro@15:         total_duration = 0
giuliomoro@15:         page_number = 0
giuliomoro@15:         
giuliomoro@15:         individual_table = '\n' # table with stats for this individual test file
giuliomoro@15:         timeline_plots = '' # plots of timeline (movements and plays)
giuliomoro@15:         
giuliomoro@15:         # DEMO survey stats
giuliomoro@15:         # get gender
giuliomoro@15:         this_subjects_gender = root.find("./posttest/radio/[@id='gender']")
giuliomoro@15:         if this_subjects_gender is not None:
giuliomoro@15:             gender.append(this_subjects_gender.get("name"))
giuliomoro@15:         else:
giuliomoro@15:             gender.append('UNAVAILABLE')
giuliomoro@15:         # get age
giuliomoro@15:         this_subjects_age = root.find("./posttest/number/[@id='age']")
giuliomoro@15:         if this_subjects_age is not None:
giuliomoro@15:             age.append(this_subjects_age.text)
giuliomoro@15:         #TODO add plot of age
giuliomoro@15:                 
giuliomoro@15:         # get list of all page names
giuliomoro@15:         for page in root.findall("./page"):   # iterate over pages
giuliomoro@15:             page_name = page.get('ref')               # get page name
giuliomoro@15:             
giuliomoro@15:             if page_name is None: # ignore 'empty' audio_holders
giuliomoro@15:                 print("WARNING: " + file + " contains empty audio holder. (evaluation_stats.py)")
giuliomoro@15:                 break # move on to next
giuliomoro@15:             
giuliomoro@15:             number_of_comments = 0 # for this page
giuliomoro@15:             number_of_missing_comments = 0 # for this page
giuliomoro@15:             not_played = [] # for this page
giuliomoro@15:             not_moved = [] # for this page
giuliomoro@15:             
giuliomoro@15:             if page.find("./metric/metricresult[@id='testTime']") is not None: # check if time is included
giuliomoro@15:                 # 'testTime' keeps total duration: subtract time so far for duration of this page
giuliomoro@15:                 duration = float(page.find("./metric/metricresult[@id='testTime']").text)# - total_duration
giuliomoro@15:             
giuliomoro@15:                 # total duration of test
giuliomoro@15:                 total_duration += duration
giuliomoro@15:             else: 
giuliomoro@15:                 duration = float('nan')
giuliomoro@15:                 total_duration = float('nan')
giuliomoro@15:             
giuliomoro@15:             # number of audio elements
giuliomoro@15:             audioelements = page.findall("./audioelement") # get audioelements
giuliomoro@15:             number_of_fragments += len(audioelements) # add length of this list to total
giuliomoro@15:             
giuliomoro@15:             # number of comments (interesting if comments not mandatory)
giuliomoro@15:             for audioelement in audioelements:
giuliomoro@15:                 response = audioelement.find("./comment/response")
giuliomoro@15:                 was_played = audioelement.find("./metric/metricresult/[@name='elementFlagListenedTo']")
giuliomoro@15:                 was_moved = audioelement.find("./metric/metricresult/[@name='elementFlagMoved']")
giuliomoro@15:                 if response is not None and response.text is not None and len(response.text) > 1: 
giuliomoro@15:                     number_of_comments += 1
giuliomoro@15:                 else: 
giuliomoro@15:                     number_of_missing_comments += 1
giuliomoro@15:                 if was_played is not None and was_played.text == 'false': 
giuliomoro@15:                     not_played.append(audioelement.get('name'))
giuliomoro@15:                 if was_moved is not None and was_moved.text == 'false': 
giuliomoro@15:                     not_moved.append(audioelement.get('name'))
giuliomoro@15:             
giuliomoro@15:             # update global counters
giuliomoro@15:             total_empty_comments += number_of_missing_comments
giuliomoro@15:             total_not_played += len(not_played)
giuliomoro@15:             total_not_moved += len(not_moved)
giuliomoro@15:             
giuliomoro@15:             # PRINT alerts when elements not played or markers not moved
giuliomoro@15:             # number of audio elements not played
giuliomoro@15:             if len(not_played) > 1:
giuliomoro@15:                 body += '\t\t\\emph{\\textbf{ATTENTION: '+str(len(not_played))+\
giuliomoro@15:                         ' fragments were not listened to in '+page_name+'! }}'+\
giuliomoro@15:                         ', '.join(not_played)+'\\\\ \n'
giuliomoro@15:             if len(not_played) == 1: 
giuliomoro@15:                 body += '\t\t\\emph{\\textbf{ATTENTION: one fragment was not listened to in '+page_name+'! }}'+\
giuliomoro@15:                         not_played[0]+'\\\\ \n'
giuliomoro@15:             
giuliomoro@15:             # number of audio element markers not moved
giuliomoro@15:             if len(not_moved) > 1:
giuliomoro@15:                 body += '\t\t\\emph{\\textbf{ATTENTION: '+str(len(not_moved))+\
giuliomoro@15:                         ' markers were not moved in '+page_name+'! }}'+\
giuliomoro@15:                         ', '.join(not_moved)+'\\\\ \n'
giuliomoro@15:             if len(not_moved) == 1: 
giuliomoro@15:                 body += '\t\t\\emph{\\textbf{ATTENTION: one marker was not moved in '+page_name+'! }}'+\
giuliomoro@15:                         not_moved[0]+'\\\\ \n'
giuliomoro@15:             
giuliomoro@15:             # PRINT song-specific statistic
giuliomoro@15:             individual_table += '\t\t'+page_name+'&'+\
giuliomoro@15:                                 str(number_of_comments) + '/' +\
giuliomoro@15:                                 str(number_of_comments+number_of_missing_comments)+'&'+\
giuliomoro@15:                                 seconds2timestr(duration)+'\\\\\n'
giuliomoro@15:             
giuliomoro@15:             # get timeline for this page
giuliomoro@15:             img_path = 'timelines_movement/'+file[:-4]+'-'+page_name+'.pdf'
giuliomoro@15:             
giuliomoro@15:             # check if available
giuliomoro@15:             if os.path.isfile(folder_name+img_path):
giuliomoro@15:                 # SHOW timeline image
giuliomoro@15:                 timeline_plots += '\\includegraphics[width=\\textwidth]{'+\
giuliomoro@15:                          folder_name+img_path+'}\n\t\t'
giuliomoro@15:             
giuliomoro@15:             # keep track of duration in function of page index
giuliomoro@15:             if len(duration_order)>page_number:
giuliomoro@15:                 duration_order[page_number].append(duration)
giuliomoro@15:             else:
giuliomoro@15:                 duration_order.append([duration])
giuliomoro@15:             
giuliomoro@15:             # keep list of page ids and count how many times each page id
giuliomoro@15:             # was tested, how long it took, and how many fragments there were 
giuliomoro@15:             # (if number of fragments is different, store as different page id)
giuliomoro@15:             if page_name in page_names: 
giuliomoro@15:                 page_index = page_names.index(page_name) # get index
giuliomoro@15:                 # check if number of audioelements the same
giuliomoro@15:                 if len(audioelements) == fragments_per_page[page_index]: 
giuliomoro@15:                     page_count[page_index] += 1
giuliomoro@15:                     duration_page[page_index].append(duration)
giuliomoro@15:                 else: # make new entry
giuliomoro@15:                     alt_page_name = page_name+"("+str(len(audioelements))+")"
giuliomoro@15:                     if alt_page_name in page_names: # if already there
giuliomoro@15:                         alt_page_index = page_names.index(alt_page_name) # get index
giuliomoro@15:                         page_count[alt_page_index] += 1
giuliomoro@15:                         duration_page[alt_page_index].append(duration)
giuliomoro@15:                     else: 
giuliomoro@15:                         page_names.append(alt_page_name)
giuliomoro@15:                         page_count.append(1)
giuliomoro@15:                         duration_page.append([duration])
giuliomoro@15:                         fragments_per_page.append(len(audioelements))
giuliomoro@15:             else: 
giuliomoro@15:                 page_names.append(page_name)
giuliomoro@15:                 page_count.append(1)
giuliomoro@15:                 duration_page.append([duration])
giuliomoro@15:                 fragments_per_page.append(len(audioelements))
giuliomoro@15:             
giuliomoro@15:             # number of subjects per page regardless of differing numbers of 
giuliomoro@15:             # fragments (for inclusion in box plots)
giuliomoro@15:             if page_name in real_page_names:
giuliomoro@15:                 page_index = real_page_names.index(page_name) # get index
giuliomoro@15:                 subject_count[page_index] += 1
giuliomoro@15:             else: 
giuliomoro@15:                 real_page_names.append(page_name)
giuliomoro@15:                 subject_count.append(1)
giuliomoro@15:             
giuliomoro@15:             # bookkeeping
giuliomoro@15:             page_number += 1 # increase page count for this specific test
giuliomoro@15:             number_of_pages += 1 # increase total number of pages
giuliomoro@15:             time_per_page_accum += duration # total duration (for average time spent per page)
giuliomoro@15: 
giuliomoro@15:         # PRINT table with statistics about this test
giuliomoro@15:         body += '\t\t'+r'''\begin{tabular}{|p{3.5cm}|c|p{2.5cm}|}
giuliomoro@15:                  \hline
giuliomoro@15:                  \textbf{Song name} & \textbf{Comments} & \textbf{Duration} \\ \hline '''+\
giuliomoro@15:                  individual_table+'\t\t'+\
giuliomoro@15:                  r'''\hline
giuliomoro@15:                   \textbf{TOTAL} & & \textbf{'''+\
giuliomoro@15:                   seconds2timestr(total_duration)+\
giuliomoro@15:                  r'''}\\
giuliomoro@15:                   \hline 
giuliomoro@15:                   \end{tabular}
giuliomoro@15:                   
giuliomoro@15:                   '''
giuliomoro@15:         # PRINT timeline plots
giuliomoro@15:         body += timeline_plots
giuliomoro@15: 
giuliomoro@15: # join to footer
giuliomoro@15: footer = body + footer
giuliomoro@15: 
giuliomoro@15: # empty body again
giuliomoro@15: body = ''
giuliomoro@15: 
giuliomoro@15: # PRINT summary of everything (at start) 
giuliomoro@15: #       unnumbered so that number of sections equals number of files
giuliomoro@15: body += '\section*{Summary}\n\t\t\\addcontentsline{toc}{section}{Summary}\n'
giuliomoro@15: 
giuliomoro@15: # PRINT table with statistics
giuliomoro@15: body += '\t\t\\begin{tabular}{ll}\n\t\t\t'
giuliomoro@15: body += r'Number of XML files: &' + str(number_of_XML_files) + r'\\'+'\n\t\t\t'
giuliomoro@15: body += r'Number of pages: &' + str(number_of_pages) + r'\\'+'\n\t\t\t'
giuliomoro@15: body += r'Number of fragments: &' + str(number_of_fragments) + r'\\'+'\n\t\t\t'
giuliomoro@15: body += r'Number of empty comments: &' + str(total_empty_comments) +\
giuliomoro@15:       " (" + str(round(100.0*total_empty_comments/number_of_fragments,2)) + r"\%)\\"+'\n\t\t\t'
giuliomoro@15: body += r'Number of unplayed fragments: &' + str(total_not_played) +\
giuliomoro@15:       " (" + str(round(100.0*total_not_played/number_of_fragments,2)) + r"\%)\\"+'\n\t\t\t'
giuliomoro@15: body += r'Number of unmoved markers: &' + str(total_not_moved) +\
giuliomoro@15:       " (" + str(round(100.0*total_not_moved/number_of_fragments,2)) + r"\%)\\"+'\n\t\t\t'
giuliomoro@15: body += r'Average time per page: &' + seconds2timestr(time_per_page_accum/number_of_pages) + r"\\"+'\n\t\t'
giuliomoro@15: body += '\\end{tabular} \\vspace{1.5cm} \\\\ \n'
giuliomoro@15: 
giuliomoro@15: # Average duration for first, second, ... page
giuliomoro@15: body += "\t\t\\vspace{.5cm} \n\n\t\tAverage duration per page (see also Figure \\ref{fig:avgtimeperpage}): \\\\ \n\t\t"
giuliomoro@15: body += r'''\begin{tabular}{lll}
giuliomoro@15:                     \textbf{Page} & \textbf{Duration} & \textbf{\# subjects}\\'''
giuliomoro@15: tpp_averages = [] # store average time per page
giuliomoro@15: for page_number in range(len(duration_order)): 
giuliomoro@15:     body += '\n\t\t\t'+str(page_number+1) + "&" +\
giuliomoro@15:         seconds2timestr(sum(duration_order[page_number])/len(duration_order[page_number])) +\
giuliomoro@15:             "&"+str(len(duration_order[page_number]))+r"\\"
giuliomoro@15:     tpp_averages.append(sum(duration_order[page_number])/len(duration_order[page_number]))
giuliomoro@15:             
giuliomoro@15: body += '\n\t\t\\end{tabular} \\vspace{1.5cm} \\\\ \n\n\t\t'
giuliomoro@15: 
giuliomoro@15: # SHOW bar plot of average time per page
giuliomoro@15: plt.bar(range(1,len(duration_order)+1), np.array(tpp_averages)/60)
giuliomoro@15: plt.xlabel('Page order')
giuliomoro@15: plt.xlim(.8, len(duration_order)+1)
giuliomoro@15: plt.xticks(np.arange(1,len(duration_order)+1)+.4, range(1,len(duration_order)+1))
giuliomoro@15: plt.ylabel('Average time [minutes]')
giuliomoro@15: plt.savefig(folder_name+"time_per_page.pdf", bbox_inches='tight')
giuliomoro@15: plt.close()
giuliomoro@15: #TODO add error bars
giuliomoro@15: 
giuliomoro@15: 
giuliomoro@15: # Sort pages by number of audioelements, then by duration
giuliomoro@15: 
giuliomoro@15: # average duration and number of subjects per page
giuliomoro@15: average_duration_page = []
giuliomoro@15: number_of_subjects_page = []
giuliomoro@15: for line in duration_page:
giuliomoro@15:     number_of_subjects_page.append(len(line))
giuliomoro@15:     average_duration_page.append(sum(line)/len(line))
giuliomoro@15: 
giuliomoro@15: # combine and sort in function of number of audioelements and duration
giuliomoro@15: combined_list = [page_names, average_duration_page, fragments_per_page, number_of_subjects_page]
giuliomoro@15: combined_list = sorted(zip(*combined_list), key=operator.itemgetter(1, 2)) # sort
giuliomoro@15: 
giuliomoro@15: # Show average duration for all songs
giuliomoro@15: body += r'''\vspace{.5cm}
giuliomoro@15:                 Average duration per page (see also Figure \ref{fig:avgtimeperpage}): \\
giuliomoro@15:                 \begin{tabular}{llll}
giuliomoro@15:                         \textbf{Audioholder} & \textbf{Duration} & \textbf{\# subjects} & \textbf{\# fragments} \\'''
giuliomoro@15: page_names_ordered = []
giuliomoro@15: average_duration_page_ordered = []
giuliomoro@15: number_of_subjects = []
giuliomoro@15: for page_index in range(len(page_names)):
giuliomoro@15:     page_names_ordered.append(combined_list[page_index][0])
giuliomoro@15:     average_duration_page_ordered.append(combined_list[page_index][1])
giuliomoro@15:     number_of_subjects.append(combined_list[page_index][3])
giuliomoro@15:     body +=  '\n\t\t\t'+combined_list[page_index][0] + "&" +\
giuliomoro@15:              seconds2timestr(combined_list[page_index][1]) + "&" +\
giuliomoro@15:              str(combined_list[page_index][3]) + "&" +\
giuliomoro@15:              str(combined_list[page_index][2]) + r"\\"
giuliomoro@15: body += '\n\t\t\\end{tabular}\n'
giuliomoro@15: 
giuliomoro@15: # SHOW bar plot of average time per page
giuliomoro@15: plt.bar(range(1,len(page_names_ordered)+1), np.array(average_duration_page_ordered)/60)
giuliomoro@15: plt.xlabel('Audioholder')
giuliomoro@15: plt.xlim(.8, len(page_names_ordered)+1)
giuliomoro@15: plt.xticks(np.arange(1,len(page_names_ordered)+1)+.4, page_names_ordered, rotation=90)
giuliomoro@15: plt.ylabel('Average time [minutes]')
giuliomoro@15: plt.savefig(folder_name+"time_per_page.pdf", bbox_inches='tight')
giuliomoro@15: plt.close()
giuliomoro@15: 
giuliomoro@15: # SHOW bar plot of average time per page
giuliomoro@15: plt.bar(range(1,len(page_names_ordered)+1), number_of_subjects)
giuliomoro@15: plt.xlabel('Audioholder')
giuliomoro@15: plt.xlim(.8, len(page_names_ordered)+1)
giuliomoro@15: plt.xticks(np.arange(1,len(page_names_ordered)+1)+.4, page_names_ordered, rotation=90)
giuliomoro@15: plt.ylabel('Number of subjects')
giuliomoro@15: ax = plt.gca()
giuliomoro@15: ylims = ax.get_ylim()
giuliomoro@15: yint = np.arange(int(np.floor(ylims[0])), int(np.ceil(ylims[1]))+1)
giuliomoro@15: plt.yticks(yint)
giuliomoro@15: plt.savefig(folder_name+"subjects_per_page.pdf", bbox_inches='tight')
giuliomoro@15: plt.close()
giuliomoro@15: 
giuliomoro@15: # SHOW both figures
giuliomoro@15: body += r'''
giuliomoro@15:          \begin{figure}[htbp]
giuliomoro@15:          \begin{center}
giuliomoro@15:          \includegraphics[width=.65\textwidth]{'''+\
giuliomoro@15:          folder_name+'time_per_page.pdf'+\
giuliomoro@15:         r'''}
giuliomoro@15:         \caption{Average time spent per page.}
giuliomoro@15:         \label{fig:avgtimeperpage}
giuliomoro@15:          \end{center}
giuliomoro@15:          \end{figure}
giuliomoro@15:          
giuliomoro@15:          '''
giuliomoro@15: body += r'''\begin{figure}[htbp]
giuliomoro@15:          \begin{center}
giuliomoro@15:          \includegraphics[width=.65\textwidth]{'''+\
giuliomoro@15:          folder_name+'time_per_page.pdf'+\
giuliomoro@15:         r'''}
giuliomoro@15:         \caption{Average time spent per page.}
giuliomoro@15:         \label{fig:avgtimeperpage}
giuliomoro@15:          \end{center}
giuliomoro@15:          \end{figure}
giuliomoro@15:          
giuliomoro@15:          '''
giuliomoro@15: body += r'''\begin{figure}[htbp]
giuliomoro@15:          \begin{center}
giuliomoro@15:          \includegraphics[width=.65\textwidth]{'''+\
giuliomoro@15:          folder_name+'subjects_per_page.pdf'+\
giuliomoro@15:         r'''}
giuliomoro@15:         \caption{Number of subjects per page.}
giuliomoro@15:         \label{fig:subjectsperpage}
giuliomoro@15:          \end{center}
giuliomoro@15:          \end{figure}
giuliomoro@15:          
giuliomoro@15:          '''
giuliomoro@15: #TODO add error bars
giuliomoro@15: #TODO layout of figures
giuliomoro@15: 
giuliomoro@15: # SHOW boxplot per page
giuliomoro@15: #TODO order in decreasing order of participants
giuliomoro@15: for page_name in page_names: # get each name
giuliomoro@15:     # plot boxplot if exists (not so for the 'alt' names)
giuliomoro@15:     if os.path.isfile(folder_name+'ratings/'+page_name+'-ratings-box.pdf'):
giuliomoro@15:         body += r'''\begin{figure}[htbp]
giuliomoro@15:              \begin{center}
giuliomoro@15:              \includegraphics[width=.65\textwidth]{'''+\
giuliomoro@15:              folder_name+"ratings/"+page_name+'-ratings-box.pdf'+\
giuliomoro@15:             r'''}
giuliomoro@15:             \caption{Box plot of ratings for page '''+\
giuliomoro@15:             page_name+' ('+str(subject_count[real_page_names.index(page_name)])+\
giuliomoro@15:             ''' participants).}
giuliomoro@15:             \label{fig:boxplot'''+page_name.replace(" ", "")+'''}
giuliomoro@15:              \end{center}
giuliomoro@15:              \end{figure}
giuliomoro@15:              
giuliomoro@15:              '''
giuliomoro@15: 
giuliomoro@15: # DEMO pie chart of gender distribution among subjects
giuliomoro@15: genders = ['male', 'female', 'other', 'preferNotToSay', 'UNAVAILABLE']
giuliomoro@15: # TODO: get the above automatically
giuliomoro@15: gender_distribution = ''
giuliomoro@15: for item in genders:
giuliomoro@15:     number = gender.count(item)
giuliomoro@15:     if number>0:
giuliomoro@15:         gender_distribution += str("{:.2f}".format((100.0*number)/len(gender)))+\
giuliomoro@15:                                '/'+item.capitalize()+' ('+str(number)+'),\n'
giuliomoro@15: 
giuliomoro@15: body += r'''
giuliomoro@15:         % Pie chart of gender distribution
giuliomoro@15:         \def\angle{0}
giuliomoro@15:         \def\radius{3}
giuliomoro@15:         \def\cyclelist{{"orange","blue","red","green"}}
giuliomoro@15:         \newcount\cyclecount \cyclecount=-1
giuliomoro@15:         \newcount\ind \ind=-1
giuliomoro@15:         \begin{figure}[htbp]
giuliomoro@15:         \begin{center}\begin{tikzpicture}[nodes = {font=\sffamily}]
giuliomoro@15:         \foreach \percent/\name in {'''+\
giuliomoro@15:         gender_distribution+\
giuliomoro@15:         r'''} {\ifx\percent\empty\else               % If \percent is empty, do nothing
giuliomoro@15:         \global\advance\cyclecount by 1     % Advance cyclecount
giuliomoro@15:         \global\advance\ind by 1            % Advance list index
giuliomoro@15:         \ifnum6<\cyclecount                 % If cyclecount is larger than list
giuliomoro@15:           \global\cyclecount=0              %   reset cyclecount and
giuliomoro@15:           \global\ind=0                     %   reset list index
giuliomoro@15:         \fi
giuliomoro@15:         \pgfmathparse{\cyclelist[\the\ind]} % Get color from cycle list
giuliomoro@15:         \edef\color{\pgfmathresult}         %   and store as \color
giuliomoro@15:         % Draw angle and set labels
giuliomoro@15:         \draw[fill={\color!50},draw={\color}] (0,0) -- (\angle:\radius)
giuliomoro@15:           arc (\angle:\angle+\percent*3.6:\radius) -- cycle;
giuliomoro@15:         \node at (\angle+0.5*\percent*3.6:0.7*\radius) {\percent\,\%};
giuliomoro@15:         \node[pin=\angle+0.5*\percent*3.6:\name]
giuliomoro@15:           at (\angle+0.5*\percent*3.6:\radius) {};
giuliomoro@15:         \pgfmathparse{\angle+\percent*3.6}  % Advance angle
giuliomoro@15:         \xdef\angle{\pgfmathresult}         %   and store in \angle
giuliomoro@15:         \fi
giuliomoro@15:         };
giuliomoro@15:         \end{tikzpicture}
giuliomoro@15:         \caption{Representation of gender across subjects}
giuliomoro@15:         \label{default}
giuliomoro@15:         \end{center}
giuliomoro@15:         \end{figure}
giuliomoro@15:         
giuliomoro@15:         '''
giuliomoro@15: # problem: some people entered twice? 
giuliomoro@15: 
giuliomoro@15: #TODO
giuliomoro@15: # time per page in function of number of fragments (plot)
giuliomoro@15: # time per participant in function of number of pages
giuliomoro@15: # plot total time for each participant
giuliomoro@15: # show 'count' per page (in order)
giuliomoro@15: 
giuliomoro@15: # clear up page_index <> page_count <> page_number confusion
giuliomoro@15: 
giuliomoro@15: 
giuliomoro@15: texfile = header+body+footer # add bits together
giuliomoro@15: 
giuliomoro@15: # print('pdflatex -output-directory="'+folder_name+'"" "'+ folder_name + 'Report.tex"')# DEBUG
giuliomoro@15: 
giuliomoro@15: # write TeX file
giuliomoro@15: with open(folder_name + 'Report.tex','w') as f:
giuliomoro@15:     f.write(texfile)
giuliomoro@15: proc=subprocess.Popen(shlex.split('pdflatex -output-directory="'+folder_name+'" "'+ folder_name + 'Report.tex"'))
giuliomoro@15: proc.communicate()
giuliomoro@15: # run again
giuliomoro@15: proc=subprocess.Popen(shlex.split('pdflatex -output-directory="'+folder_name+'" "'+ folder_name + 'Report.tex"'))
giuliomoro@15: proc.communicate()
giuliomoro@15: 
giuliomoro@15: #TODO remove auxiliary LaTeX files
giuliomoro@15: try:
giuliomoro@15:     os.remove(folder_name + 'Report.aux')
giuliomoro@15:     os.remove(folder_name + 'Report.log')
giuliomoro@15:     os.remove(folder_name + 'Report.out')
giuliomoro@15:     os.remove(folder_name + 'Report.toc')
giuliomoro@15: except OSError:
giuliomoro@15:     pass
giuliomoro@15: