diff options
| -rw-r--r-- | src/argaze/utils/export_tobii_segment_plots.py | 25 |
1 files changed, 1 insertions, 24 deletions
diff --git a/src/argaze/utils/export_tobii_segment_plots.py b/src/argaze/utils/export_tobii_segment_plots.py index 94e88e7..9b519f5 100644 --- a/src/argaze/utils/export_tobii_segment_plots.py +++ b/src/argaze/utils/export_tobii_segment_plots.py @@ -76,9 +76,7 @@ def main(): subplot = figure.add_subplot(711) subplot.set_title('Pupil diameter', loc='left') subplot.set_ylim(0, 10) - patches = tobii_segment_data['PupilDiameter'].plot(names=['value'], colors=['#FFD800']) - subplot.legend(handles=patches, loc='upper left') # Annotate events @@ -90,66 +88,45 @@ def main(): subplot.annotate(f'{event_type}\n{event_tag}', xy=(ts, 7), horizontalalignment="left", verticalalignment="top") subplot.vlines(ts, 0, 6, color="tab:red", linewidth=1) - # Access to timestamped pupil center data buffer - df_ts_pupil_centers = tobii_segment_data['PupilCenter'].as_dataframe(split={'value':['x','y','z']}) - # Plot pupil center data subplot = figure.add_subplot(712) subplot.set_title('Pupil center', loc='left') subplot.set_ylim(-40, -20) - patches = tobii_segment_data['PupilCenter'].plot(names=['x','y','z'], colors=['#276FB6','#9427B6','#888888'], split={'value':['x','y','z']}) - subplot.legend(handles=patches, loc='upper left') - # Access to timestamped gaze position data buffer - df_ts_gaze_positions = tobii_segment_data['GazePosition'].as_dataframe(split={'value':['x','y']}) - # Plot gaze position data subplot = figure.add_subplot(713) subplot.set_title('Gaze position', loc='left') subplot.set_ylim(0., 1.) - patches = tobii_segment_data['GazePosition'].plot(names=['x','y'], colors=['#276FB6','#9427B6'], split={'value':['x','y']}) - subplot.legend(handles=patches, loc='upper left') # Plot gaze direction data subplot = figure.add_subplot(714) subplot.set_title('Gaze direction', loc='left') - patches = tobii_segment_data['GazeDirection'].plot(names=['x','y','z'], colors=['#276FB6','#9427B6','#888888'], split={'value':['x','y','z']}) - subplot.legend(handles=patches, loc='upper left') # Plot gaze direction data subplot = figure.add_subplot(715) subplot.set_title('Gaze position 3D', loc='left') - patches = tobii_segment_data['GazePosition3D'].plot(names=['x','y','z'], colors=['#276FB6','#9427B6','#888888'], split={'value':['x','y','z']}) - subplot.legend(handles=patches, loc='upper left') # Plot accelerometer data subplot = figure.add_subplot(716) subplot.set_title('Accelerometer', loc='left') - patches = tobii_segment_data['Accelerometer'].plot(names=['x','y','z'], colors=['#276FB6','#9427B6','#888888'], split={'value':['x','y','z']}) - subplot.legend(handles=patches, loc='upper left') - # Access to timestamped gyroscope data buffer - df_ts_gyroscopes = tobii_segment_data['Gyroscope'].as_dataframe(split={'value':['x','y', 'z']}) - # Plot accelerometer data subplot = figure.add_subplot(717) subplot.set_title('Gyroscope', loc='left') - patches = tobii_segment_data['Gyroscope'].plot(names=['x','y','z'], colors=['#276FB6','#9427B6','#888888'], split={'value':['x','y','z']}) - subplot.legend(handles=patches, loc='upper left') - # Export data plots + # Export figure mpyplot.tight_layout() mpyplot.savefig(data_plots_filepath) mpyplot.close('all') |