// NewLabels returns labelblk Labels, a representation of externally usable subvolume
// or slice data, given some geometry and optional image data.
// If img is passed in, the function will initialize Voxels with data from the image.
// Otherwise, it will allocate a zero buffer of appropriate size.
//
// TODO : Unlike the standard imageblk.NewVoxels, the labelblk version can modify the
// labels based on the z-coordinate of the given geometry for Raveler labeling.
// This will be removed when Raveler-specific labels are moved outside DVID.
func (d *Data) NewLabels(geom dvid.Geometry, img interface{}) (*Labels, error) {
bytesPerVoxel := d.Properties.Values.BytesPerElement()
stride := geom.Size().Value(0) * bytesPerVoxel
var data []byte
if img == nil {
numVoxels := geom.NumVoxels()
if numVoxels <= 0 {
return nil, fmt.Errorf("Illegal geometry requested: %s", geom)
}
requestSize := int64(bytesPerVoxel) * numVoxels
if requestSize > server.MaxDataRequest {
return nil, fmt.Errorf("Requested payload (%d bytes) exceeds this DVID server's set limit (%d)",
requestSize, server.MaxDataRequest)
}
data = make([]byte, requestSize)
} else {
switch t := img.(type) {
case image.Image:
var inputBytesPerVoxel, actualStride int32
var err error
data, inputBytesPerVoxel, actualStride, err = dvid.ImageData(t)
if err != nil {
return nil, err
}
if actualStride != stride {
// Need to do some conversion here.
switch d.Labeling {
case Standard64bit:
data, err = d.convertTo64bit(geom, data, int(inputBytesPerVoxel), int(actualStride))
if err != nil {
return nil, err
}
case RavelerLabel:
data, err = d.addLabelZ(geom, data, actualStride)
if err != nil {
return nil, err
}
default:
return nil, fmt.Errorf("unexpected label type in labelblk: %s", d.Labeling)
}
}
case []byte:
data = t
actualLen := int64(len(data))
expectedLen := int64(bytesPerVoxel) * geom.NumVoxels()
if actualLen != expectedLen {
return nil, fmt.Errorf("labels data was %d bytes, expected %d bytes for %s",
actualLen, expectedLen, geom)
}
default:
return nil, fmt.Errorf("unexpected image type given to NewVoxels(): %T", t)
}
}
labels := &Labels{
imageblk.NewVoxels(geom, d.Properties.Values, data, stride),
}
return labels, nil
}
// Prints RGBA of first n x n pixels of image with header string.
func debugData(img image.Image, message string) {
data, _, stride, _ := dvid.ImageData(img)
var n = 3 // neighborhood to write
fmt.Printf("%s>\n", message)
for y := 0; y < n; y++ {
for x := 0; x < n; x++ {
i := y*int(stride) + x*4
fmt.Printf("[%3d %3d %3d %3d] ", data[i], data[i+1], data[i+2], data[i+3])
}
fmt.Printf("\n")
}
}
// NewVoxels returns Voxels with given geometry and optional image data.
// If img is passed in, the function will initialize the Voxels with data from the image.
// Otherwise, it will allocate a zero buffer of appropriate size.
func (d *Data) NewVoxels(geom dvid.Geometry, img interface{}) (*Voxels, error) {
bytesPerVoxel := d.Properties.Values.BytesPerElement()
stride := geom.Size().Value(0) * bytesPerVoxel
voxels := &Voxels{
Geometry: geom,
values: d.Properties.Values,
stride: stride,
}
if img == nil {
numVoxels := geom.NumVoxels()
if numVoxels <= 0 {
return nil, fmt.Errorf("Illegal geometry requested: %s", geom)
}
requestSize := int64(bytesPerVoxel) * numVoxels
if requestSize > server.MaxDataRequest {
return nil, fmt.Errorf("Requested payload (%d bytes) exceeds this DVID server's set limit (%d)",
requestSize, server.MaxDataRequest)
}
voxels.data = make([]uint8, requestSize)
} else {
switch t := img.(type) {
case image.Image:
var actualStride int32
var err error
voxels.data, _, actualStride, err = dvid.ImageData(t)
if err != nil {
return nil, err
}
if actualStride < stride {
return nil, fmt.Errorf("Too little data in input image (expected stride %d)", stride)
}
voxels.stride = actualStride
case []byte:
voxels.data = t
actualLen := int64(len(voxels.data))
expectedLen := int64(bytesPerVoxel) * geom.NumVoxels()
if actualLen != expectedLen {
return nil, fmt.Errorf("voxels data was %d bytes, expected %d bytes for %s",
actualLen, expectedLen, geom)
}
default:
return nil, fmt.Errorf("Unexpected image type given to NewVoxels(): %T", t)
}
}
return voxels, nil
}