def parse_links(dc, links):
from glue.core.link_helpers import MultiLink
from glue.core import ComponentLink
data = dict((d.label, d) for d in dc)
result = []
def find_cid(s):
dlabel, clabel = s.split('.')
d = data[dlabel]
c = d.find_component_id(clabel)
if c is None:
raise ValueError("Invalid link (no component named %s)" % s)
return c
for link in links:
f, t = link[0:2] # from and to component names
u = u2 = None
if len(link) >= 3: # forward translation function
u = link[2]
if len(link) == 4: # reverse translation function
u2 = link[3]
# component names -> component IDs
if isinstance(f, str):
f = [find_cid(f)]
else:
f = [find_cid(item) for item in f]
if isinstance(t, str):
t = find_cid(t)
result.append(ComponentLink(f, t, u))
else:
t = [find_cid(item) for item in t]
result += MultiLink(f, t, u, u2)
return result
def __init__(self, in_lon, in_lat, out_lon, out_lat):
MultiLink.__init__(self, [in_lon, in_lat],
[out_lon, out_lat],
self.forward, self.backward)
def __init__(self, in_lon, in_lat, out_lon, out_lat):
MultiLink.__init__(self, [in_lon, in_lat], [out_lon, out_lat],
self.forward, self.backward)
def test_pixel_selection_subset_state():
data1 = Data(x=np.ones((2, 4, 3)))
data2 = Data(y=np.ones((4, 3, 2)))
data3 = Data(z=np.ones((2, 3)))
y_id = data2.main_components[0]
z_id = data3.main_components[0]
slice_1d = [slice(1, 2), slice(None), slice(None)]
slice_2d = [slice(None), slice(2, 3), slice(1, 2)]
slice_3d = [slice(1, 2), slice(2, 3), slice(1, 2)]
state_1d = PixelSubsetState(data1, slice_1d)
state_2d = PixelSubsetState(data1, slice_2d)
state_3d = PixelSubsetState(data1, slice_3d)
states = [state_1d, state_2d, state_3d]
dc = DataCollection([data1, data2, data3])
# Calling to_array with reference data should work by default, and not work
# with unlinked datasets.
for data in dc:
for state in states:
cid = data.main_components[0]
if data is data1:
assert_array_equal(state.to_array(data, cid),
data[cid][state.slices])
else:
with pytest.raises(IncompatibleAttribute):
state.to_array(data, cid)
# Add one set of links
dc.add_link(
LinkSame(data1.pixel_component_ids[0], data2.pixel_component_ids[2]))
dc.add_link(
LinkSame(data1.pixel_component_ids[0], data3.pixel_component_ids[0]))
assert_array_equal(state_1d.to_array(data2, y_id), data2[y_id][:, :, 1:2])
with pytest.raises(IncompatibleAttribute):
state_2d.to_array(data2, y_id)
with pytest.raises(IncompatibleAttribute):
state_3d.to_array(data2, y_id)
assert_array_equal(state_1d.to_array(data3, z_id), data3[z_id][1:2])
with pytest.raises(IncompatibleAttribute):
state_2d.to_array(data3, z_id)
with pytest.raises(IncompatibleAttribute):
state_3d.to_array(data3, z_id)
# Add links with multiple components, in this case linking two cids with two cids
def forwards(x, y):
return x + y, x - y
def backwards(x, y):
return 0.5 * (x + y), 0.5 * (x - y)
dc.add_link(
MultiLink(data1.pixel_component_ids[1:],
data2.pixel_component_ids[:2],
forwards=forwards,
backwards=backwards))
assert_array_equal(state_1d.to_array(data2, y_id), data2[y_id][:, :, 1:2])
assert_array_equal(state_2d.to_array(data2, y_id), data2[y_id][2:3,
1:2, :])
assert_array_equal(state_3d.to_array(data2, y_id), data2[y_id][2:3, 1:2,
1:2])
assert_array_equal(state_1d.to_array(data3, z_id), data3[z_id][1:2])
with pytest.raises(IncompatibleAttribute):
state_2d.to_array(data3, z_id)
with pytest.raises(IncompatibleAttribute):
state_3d.to_array(data3, z_id)