Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(432)

Side by Side Diff: tools/python_charts/webrtc/data_helper.py

Issue 2580763002: Remove unused items in tools/ (Closed)
Patch Set: Created 4 years ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « tools/python_charts/webrtc/__init__.py ('k') | tools/python_charts/webrtc/data_helper_test.py » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(Empty)
1 #!/usr/bin/env python
2 # Copyright (c) 2011 The WebRTC project authors. All Rights Reserved.
3 #
4 # Use of this source code is governed by a BSD-style license
5 # that can be found in the LICENSE file in the root of the source
6 # tree. An additional intellectual property rights grant can be found
7 # in the file PATENTS. All contributing project authors may
8 # be found in the AUTHORS file in the root of the source tree.
9
10 __author__ = 'kjellander@webrtc.org (Henrik Kjellander)'
11
12 class DataHelper(object):
13 """
14 Helper class for managing table data.
15 This class does not verify the consistency of the data tables sent into it.
16 """
17
18 def __init__(self, data_list, table_description, names_list, messages):
19 """ Initializes the DataHelper with data.
20
21 Args:
22 data_list: List of one or more data lists in the format that the
23 Google Visualization Python API expects (list of dictionaries, one
24 per row of data). See the gviz_api.DataTable documentation for more
25 info.
26 table_description: dictionary describing the data types of all
27 columns in the data lists, as defined in the gviz_api.DataTable
28 documentation.
29 names_list: List of strings of what we're going to name the data
30 columns after. Usually different runs of data collection.
31 messages: List of strings we might append error messages to.
32 """
33 self.data_list = data_list
34 self.table_description = table_description
35 self.names_list = names_list
36 self.messages = messages
37 self.number_of_datasets = len(data_list)
38 self.number_of_frames = len(data_list[0])
39
40 def CreateData(self, field_name, start_frame=0, end_frame=0):
41 """ Creates a data structure for a specified data field.
42
43 Creates a data structure (data type description dictionary and a list
44 of data dictionaries) to be used with the Google Visualization Python
45 API. The frame_number column is always present and one column per data
46 set is added and its field name is suffixed by _N where N is the number
47 of the data set (0, 1, 2...)
48
49 Args:
50 field_name: String name of the field, must be present in the data
51 structure this DataHelper was created with.
52 start_frame: Frame number to start at (zero indexed). Default: 0.
53 end_frame: Frame number to be the last frame. If zero all frames
54 will be included. Default: 0.
55
56 Returns:
57 A tuple containing:
58 - a dictionary describing the columns in the data result_data_table below.
59 This description uses the name for each data set specified by
60 names_list.
61
62 Example with two data sets named 'Foreman' and 'Crew':
63 {
64 'frame_number': ('number', 'Frame number'),
65 'ssim_0': ('number', 'Foreman'),
66 'ssim_1': ('number', 'Crew'),
67 }
68 - a list containing dictionaries (one per row) with the frame_number
69 column and one column of the specified field_name column per data
70 set.
71
72 Example with two data sets named 'Foreman' and 'Crew':
73 [
74 {'frame_number': 0, 'ssim_0': 0.98, 'ssim_1': 0.77 },
75 {'frame_number': 1, 'ssim_0': 0.81, 'ssim_1': 0.53 },
76 ]
77 """
78
79 # Build dictionary that describes the data types
80 result_table_description = {'frame_number': ('string', 'Frame number')}
81 for dataset_index in range(self.number_of_datasets):
82 column_name = '%s_%s' % (field_name, dataset_index)
83 column_type = self.table_description[field_name][0]
84 column_description = self.names_list[dataset_index]
85 result_table_description[column_name] = (column_type, column_description)
86
87 # Build data table of all the data
88 result_data_table = []
89 # We're going to have one dictionary per row.
90 # Create that and copy frame_number values from the first data set
91 for source_row in self.data_list[0]:
92 row_dict = {'frame_number': source_row['frame_number']}
93 result_data_table.append(row_dict)
94
95 # Pick target field data points from the all data tables
96 if end_frame == 0: # Default to all frames
97 end_frame = self.number_of_frames
98
99 for dataset_index in range(self.number_of_datasets):
100 for row_number in range(start_frame, end_frame):
101 column_name = '%s_%s' % (field_name, dataset_index)
102 # Stop if any of the data sets are missing the frame
103 try:
104 result_data_table[row_number][column_name] = \
105 self.data_list[dataset_index][row_number][field_name]
106 except IndexError:
107 self.messages.append("Couldn't find frame data for row %d "
108 "for %s" % (row_number, self.names_list[dataset_index]))
109 break
110 return result_table_description, result_data_table
111
112 def GetOrdering(self, table_description): # pylint: disable=R0201
113 """ Creates a list of column names, ordered alphabetically except for the
114 frame_number column which always will be the first column.
115
116 Args:
117 table_description: A dictionary of column definitions as defined by the
118 gviz_api.DataTable documentation.
119 Returns:
120 A list of column names, where frame_number is the first and the
121 remaining columns are sorted alphabetically.
122 """
123 # The JSON data representation generated from gviz_api.DataTable.ToJSon()
124 # must have frame_number as its first column in order for the chart to
125 # use it as it's X-axis value series.
126 # gviz_api.DataTable orders the columns by name by default, which will
127 # be incorrect if we have column names that are sorted before frame_number
128 # in our data table.
129 columns_ordering = ['frame_number']
130 # add all other columns:
131 for column in sorted(table_description.keys()):
132 if column != 'frame_number':
133 columns_ordering.append(column)
134 return columns_ordering
135
136 def CreateConfigurationTable(self, configurations): # pylint: disable=R0201
137 """ Combines multiple test data configurations for display.
138
139 Args:
140 configurations: List of one ore more configurations. Each configuration
141 is required to be a list of dictionaries with two keys: 'name' and
142 'value'.
143 Example of a single configuration:
144 [
145 {'name': 'name', 'value': 'VP8 software'},
146 {'name': 'test_number', 'value': '0'},
147 {'name': 'input_filename', 'value': 'foreman_cif.yuv'},
148 ]
149 Returns:
150 A tuple containing:
151 - a dictionary describing the columns in the configuration table to be
152 displayed. All columns will have string as data type.
153 Example:
154 {
155 'name': 'string',
156 'test_number': 'string',
157 'input_filename': 'string',
158 }
159 - a list containing dictionaries (one per configuration) with the
160 configuration column names mapped to the value for each test run:
161
162 Example matching the columns above:
163 [
164 {'name': 'VP8 software',
165 'test_number': '12',
166 'input_filename': 'foreman_cif.yuv' },
167 {'name': 'VP8 hardware',
168 'test_number': '5',
169 'input_filename': 'foreman_cif.yuv' },
170 ]
171 """
172 result_description = {}
173 result_data = []
174
175 for configuration in configurations:
176 data = {}
177 result_data.append(data)
178 for values in configuration:
179 name = values['name']
180 value = values['value']
181 result_description[name] = 'string'
182 data[name] = value
183 return result_description, result_data
OLDNEW
« no previous file with comments | « tools/python_charts/webrtc/__init__.py ('k') | tools/python_charts/webrtc/data_helper_test.py » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698