def test_encoded_stream(self):
data = numpy.random.randint(0,255, (3, 100, 200)).astype(numpy.uint8)
codec = VoxelsNddataCodec( data.dtype )
stream = codec.create_encoded_stream_from_ndarray(data)
roundtrip_data = codec.decode_to_ndarray(stream, data.shape)
assert roundtrip_data.flags['F_CONTIGUOUS']
self._assert_matching(roundtrip_data, data)
def test_encoded_stream(self):
data = numpy.random.randint(0, 255, (3, 100, 200)).astype(numpy.uint8)
codec = VoxelsNddataCodec(data.dtype)
stream = codec.create_encoded_stream_from_ndarray(data)
roundtrip_data = codec.decode_to_ndarray(stream, data.shape)
assert roundtrip_data.flags['F_CONTIGUOUS']
self._assert_matching(roundtrip_data, data)
def post_ndarray(connection,
uuid,
data_name,
access_type,
voxels_metadata,
start,
stop,
new_data,
throttle=False):
_validate_query_bounds(start,
stop,
voxels_metadata.shape,
allow_overflow_extents=True)
codec = VoxelsNddataCodec(voxels_metadata.dtype)
rest_query = _format_subvolume_rest_uri(uuid,
data_name,
access_type,
start,
stop,
format="",
query_args=None,
throttle=throttle)
body_data_stream = codec.create_encoded_stream_from_ndarray(new_data)
# Slightly tricky here:
# The httplib docs say that we can only send a stream that has a fileno() method,
# but it turns out we can send any stream as long as we provide our own content-length header.
headers = {
"Content-Type": VoxelsNddataCodec.VOLUME_MIMETYPE,
"Content-Length": str(codec.calculate_buffer_len(new_data.shape))
}
connection.request("POST",
rest_query,
body=body_data_stream,
headers=headers)
with contextlib.closing(connection.getresponse()) as response:
#if response.status != httplib.NO_CONTENT:
if response.status != httplib.OK:
raise DvidHttpError("subvolume post",
response.status, response.reason,
response.read(), "POST", rest_query,
"<binary data>", headers)
# Something (either dvid or the httplib) gets upset if we don't read the full response.
response.read()
def post_ndarray( connection, uuid, data_name, access_type, voxels_metadata, start, stop, new_data, throttle=False ):
_validate_query_bounds( start, stop, voxels_metadata.shape, allow_overflow_extents=True )
codec = VoxelsNddataCodec( voxels_metadata.dtype )
rest_query = _format_subvolume_rest_uri( uuid, data_name, access_type, start, stop, format="", query_args=None, throttle=throttle )
body_data_stream = codec.create_encoded_stream_from_ndarray(new_data)
# Slightly tricky here:
# The httplib docs say that we can only send a stream that has a fileno() method,
# but it turns out we can send any stream as long as we provide our own content-length header.
headers = { "Content-Type" : VoxelsNddataCodec.VOLUME_MIMETYPE,
"Content-Length" : str(codec.calculate_buffer_len(new_data.shape)) }
connection.request( "POST", rest_query, body=body_data_stream, headers=headers )
with contextlib.closing( connection.getresponse() ) as response:
#if response.status != httplib.NO_CONTENT:
if response.status != httplib.OK:
raise DvidHttpError(
"subvolume post", response.status, response.reason, response.read(),
"POST", rest_query, "<binary data>", headers)
# Something (either dvid or the httplib) gets upset if we don't read the full response.
response.read()
