8
8
import pandas as pd
9
9
10
10
11
- class _Transform (Protocol ):
11
+ class _MatplotlibTransform (Protocol ):
12
12
def transform (self , verts ):
13
13
...
14
14
15
- def __sub__ (self , other ) -> "_Transform " :
15
+ def __sub__ (self , other ) -> "_MatplotlibTransform " :
16
16
...
17
17
18
18
@@ -35,7 +35,7 @@ class DataContainer(Protocol):
35
35
def query (
36
36
self ,
37
37
# TODO 3D?!!
38
- transform : _Transform ,
38
+ coord_transform : _MatplotlibTransform ,
39
39
size : Tuple [int , int ],
40
40
) -> Tuple [Dict [str , Any ], Union [str , int ]]:
41
41
"""
@@ -46,7 +46,7 @@ def query(
46
46
47
47
Parameters
48
48
----------
49
- transform : matplotlib.transform.Transform
49
+ coord_transform : matplotlib.transform.Transform
50
50
Must go from axes fraction space -> data space
51
51
52
52
size : 2 integers
@@ -88,7 +88,7 @@ def __init__(self, **data):
88
88
89
89
def query (
90
90
self ,
91
- transform : _Transform ,
91
+ coord_transform : _MatplotlibTransform ,
92
92
size : Tuple [int , int ],
93
93
) -> Tuple [Dict [str , Any ], Union [str , int ]]:
94
94
return dict (self ._data ), self ._cache_key
@@ -113,7 +113,7 @@ def __init__(self, **shapes):
113
113
114
114
def query (
115
115
self ,
116
- transform : _Transform ,
116
+ coord_transform : _MatplotlibTransform ,
117
117
size : Tuple [int , int ],
118
118
) -> Tuple [Dict [str , Any ], Union [str , int ]]:
119
119
return {k : np .random .randn (* d .shape ) for k , d in self ._desc .items ()}, str (uuid .uuid4 ())
@@ -166,26 +166,26 @@ def _split(input_dict):
166
166
167
167
def query (
168
168
self ,
169
- transform : _Transform ,
169
+ coord_transform : _MatplotlibTransform ,
170
170
size : Tuple [int , int ],
171
171
) -> Tuple [Dict [str , Any ], Union [str , int ]]:
172
172
# TODO find a better way to compute the hash key, this is not sentative to
173
173
# scale changes, only limit changes
174
- data_bounds = tuple (transform .transform ([[0 , 0 ], [1 , 1 ]]).flatten ())
174
+ data_bounds = tuple (coord_transform .transform ([[0 , 0 ], [1 , 1 ]]).flatten ())
175
175
hash_key = hash ((data_bounds , size ))
176
176
if hash_key in self ._cache :
177
177
return self ._cache [hash_key ], hash_key
178
178
179
179
xpix , ypix = size
180
- x_data , _ = transform .transform (
180
+ x_data , _ = coord_transform .transform (
181
181
np .vstack (
182
182
[
183
183
np .linspace (0 , 1 , int (xpix ) * 2 ),
184
184
np .zeros (int (xpix ) * 2 ),
185
185
]
186
186
).T
187
187
).T
188
- _ , y_data = transform .transform (
188
+ _ , y_data = coord_transform .transform (
189
189
np .vstack (
190
190
[
191
191
np .zeros (int (ypix ) * 2 ),
@@ -218,11 +218,11 @@ def __init__(self, raw_data, num_bins: int):
218
218
219
219
def query (
220
220
self ,
221
- transform : _Transform ,
221
+ coord_transform : _MatplotlibTransform ,
222
222
size : Tuple [int , int ],
223
223
) -> Tuple [Dict [str , Any ], Union [str , int ]]:
224
224
dmin , dmax = self ._full_range
225
- xmin , ymin , xmax , ymax = transform .transform ([[0 , 0 ], [1 , 1 ]]).flatten ()
225
+ xmin , ymin , xmax , ymax = coord_transform .transform ([[0 , 0 ], [1 , 1 ]]).flatten ()
226
226
227
227
xmin , xmax = np .clip ([xmin , xmax ], dmin , dmax )
228
228
hash_key = hash ((xmin , xmax ))
@@ -266,7 +266,7 @@ def __init__(self, series: pd.Series, *, index_name: str, col_name: str):
266
266
267
267
def query (
268
268
self ,
269
- transform : _Transform ,
269
+ coord_transform : _MatplotlibTransform ,
270
270
size : Tuple [int , int ],
271
271
) -> Tuple [Dict [str , Any ], Union [str , int ]]:
272
272
return {self ._index_name : self ._data .index .values , self ._col_name : self ._data .values }, self ._hash_key
@@ -305,7 +305,7 @@ def __init__(
305
305
306
306
def query (
307
307
self ,
308
- transform : _Transform ,
308
+ coord_transform : _MatplotlibTransform ,
309
309
size : Tuple [int , int ],
310
310
) -> Tuple [Dict [str , Any ], Union [str , int ]]:
311
311
ret = {}
@@ -328,10 +328,10 @@ def __init__(self, data: DataContainer, mapping: Dict[str, str]):
328
328
329
329
def query (
330
330
self ,
331
- transform : _Transform ,
331
+ coord_transform : _MatplotlibTransform ,
332
332
size : Tuple [int , int ],
333
333
) -> Tuple [Dict [str , Any ], Union [str , int ]]:
334
- base , cache_key = self ._data .query (transform , size )
334
+ base , cache_key = self ._data .query (coord_transform , size )
335
335
return {v : base [k ] for k , v in self ._mapping .items ()}, cache_key
336
336
337
337
def describe (self ):
@@ -346,13 +346,13 @@ def __init__(self, *data: DataContainer):
346
346
347
347
def query (
348
348
self ,
349
- transform : _Transform ,
349
+ coord_transform : _MatplotlibTransform ,
350
350
size : Tuple [int , int ],
351
351
) -> Tuple [Dict [str , Any ], Union [str , int ]]:
352
352
cache_keys = []
353
353
ret = {}
354
354
for data in self ._datas :
355
- base , cache_key = data .query (transform , size )
355
+ base , cache_key = data .query (coord_transform , size )
356
356
ret .update (base )
357
357
cache_keys .append (cache_key )
358
358
return ret , hash (tuple (cache_keys ))
@@ -364,7 +364,7 @@ def describe(self):
364
364
class WebServiceContainer :
365
365
def query (
366
366
self ,
367
- transform : _Transform ,
367
+ coord_transform : _MatplotlibTransform ,
368
368
size : Tuple [int , int ],
369
369
) -> Tuple [Dict [str , Any ], Union [str , int ]]:
370
370
def hit_some_database ():
0 commit comments