itk-wasm
Version:
High-performance spatial analysis in a web browser, Node.js, and reproducible execution across programming languages and hardware architectures.
250 lines • 14 kB
JavaScript
import Image from './interface-types/image.js';
import PixelTypes from './interface-types/pixel-types.js';
import IntTypes from './interface-types/int-types.js';
import FloatTypes from './interface-types/float-types.js';
/**
* Cast an image to another PixelType and/or ComponentType
*
* @param {Image} image - The input image
* @param {CastImageOptions} options - specify the componentType and/or pixelType of the output
*/
function castImage(inputImage, options) {
const outputImageType = { ...inputImage.imageType };
if (typeof options !== 'undefined' &&
typeof options.pixelType !== 'undefined') {
outputImageType.pixelType = options.pixelType;
if (options.pixelType === PixelTypes.Scalar &&
outputImageType.components !== 1) {
throw new Error('Cannot cast multi-component image to a scalar image');
}
}
if (typeof options !== 'undefined' &&
typeof options.componentType !== 'undefined' &&
options.componentType !== inputImage.imageType.componentType) {
outputImageType.componentType = options.componentType;
}
const outputImage = new Image(outputImageType);
outputImage.name = inputImage.name;
outputImage.origin = Array.from(inputImage.origin);
outputImage.spacing = Array.from(inputImage.spacing);
outputImage.direction = inputImage.direction.slice();
outputImage.size = Array.from(inputImage.size);
// Deep copy the map
outputImage.metadata = new Map(JSON.parse(JSON.stringify(Array.from(inputImage.metadata))));
if (inputImage.data !== null) {
if (typeof options !== 'undefined' &&
typeof options.componentType !== 'undefined' &&
options.componentType !== inputImage.imageType.componentType) {
switch (inputImage.imageType.componentType) {
case IntTypes.UInt8:
case IntTypes.Int8:
case IntTypes.UInt16:
case IntTypes.Int16:
case IntTypes.UInt32:
case IntTypes.Int32:
case FloatTypes.Float32:
case FloatTypes.Float64:
switch (outputImage.imageType.componentType) {
case IntTypes.UInt8:
if (inputImage.data instanceof BigInt64Array ||
inputImage.data instanceof BigUint64Array) {
outputImage.data = new Uint8Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = Number(inputImage.data[idx]);
}
}
else {
outputImage.data = new Uint8Array(inputImage.data);
}
break;
case IntTypes.Int8:
if (inputImage.data instanceof BigInt64Array ||
inputImage.data instanceof BigUint64Array) {
outputImage.data = new Int8Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = Number(inputImage.data[idx]);
}
}
else {
outputImage.data = new Int8Array(inputImage.data);
}
break;
case IntTypes.UInt16:
if (inputImage.data instanceof BigInt64Array ||
inputImage.data instanceof BigUint64Array) {
outputImage.data = new Uint16Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = Number(inputImage.data[idx]);
}
}
else {
outputImage.data = new Uint16Array(inputImage.data);
}
break;
case IntTypes.Int16:
if (inputImage.data instanceof BigInt64Array ||
inputImage.data instanceof BigUint64Array) {
outputImage.data = new Int16Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = Number(inputImage.data[idx]);
}
}
else {
outputImage.data = new Int16Array(inputImage.data);
}
break;
case IntTypes.UInt32:
if (inputImage.data instanceof BigInt64Array ||
inputImage.data instanceof BigUint64Array) {
outputImage.data = new Uint32Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = Number(inputImage.data[idx]);
}
}
else {
outputImage.data = new Uint32Array(inputImage.data);
}
break;
case IntTypes.Int32:
if (inputImage.data instanceof BigInt64Array ||
inputImage.data instanceof BigUint64Array) {
outputImage.data = new Int32Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = Number(inputImage.data[idx]);
}
}
else {
outputImage.data = new Int32Array(inputImage.data);
}
break;
case FloatTypes.Float32:
if (inputImage.data instanceof BigInt64Array ||
inputImage.data instanceof BigUint64Array) {
outputImage.data = new Float32Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = Number(inputImage.data[idx]);
}
}
else {
outputImage.data = new Float32Array(inputImage.data);
}
break;
case FloatTypes.Float64:
if (inputImage.data instanceof BigInt64Array ||
inputImage.data instanceof BigUint64Array) {
outputImage.data = new Float64Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = Number(inputImage.data[idx]);
}
}
else {
outputImage.data = new Float64Array(inputImage.data);
}
break;
case IntTypes.UInt64:
outputImage.data = new BigUint64Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = BigInt.asIntN(64, BigInt(inputImage.data[idx]));
}
break;
case IntTypes.Int64:
outputImage.data = new BigInt64Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = BigInt.asUintN(64, BigInt(inputImage.data[idx]));
}
break;
}
break;
case IntTypes.UInt64:
case IntTypes.Int64:
switch (outputImage.imageType.componentType) {
case IntTypes.UInt8:
outputImage.data = new Uint8Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = Number(inputImage.data[idx]);
}
break;
case IntTypes.Int8:
outputImage.data = new Int8Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = Number(inputImage.data[idx]);
}
break;
case IntTypes.UInt16:
outputImage.data = new Uint16Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = Number(inputImage.data[idx]);
}
break;
case IntTypes.Int16:
outputImage.data = new Int16Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = Number(inputImage.data[idx]);
}
break;
case IntTypes.UInt32:
outputImage.data = new Uint32Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = Number(inputImage.data[idx]);
}
break;
case IntTypes.Int32:
outputImage.data = new Int32Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = Number(inputImage.data[idx]);
}
break;
case FloatTypes.Float32:
outputImage.data = new Float32Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = Number(inputImage.data[idx]);
}
break;
case FloatTypes.Float64:
outputImage.data = new Float64Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = Number(inputImage.data[idx]);
}
break;
case IntTypes.UInt64:
if (inputImage.data instanceof BigInt64Array ||
inputImage.data instanceof BigUint64Array) {
outputImage.data = new BigUint64Array(inputImage.data);
}
else {
outputImage.data = new BigUint64Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = BigInt.asUintN(64, BigInt(inputImage.data[idx]));
}
}
break;
case IntTypes.Int64:
if (inputImage.data instanceof BigInt64Array ||
inputImage.data instanceof BigUint64Array) {
outputImage.data = new BigInt64Array(inputImage.data);
}
else {
outputImage.data = new BigInt64Array(inputImage.data.length);
for (let idx = 0; idx < outputImage.data.length; idx++) {
outputImage.data[idx] = BigInt.asIntN(64, BigInt(inputImage.data[idx]));
}
}
break;
}
break;
}
}
else {
// copy
const CTor = inputImage.data.constructor;
outputImage.data = new CTor(inputImage.data.length);
if (outputImage.data != null) {
// @ts-expect-error: error TS2345: Argument of type 'TypedArray' is not assignable to parameter of type 'ArrayLike<number> & ArrayLike<bigint>'
outputImage.data.set(inputImage.data, 0);
}
}
}
return outputImage;
}
export default castImage;
//# sourceMappingURL=cast-image.js.map