paraview-glance/dist/itk/writeImageEmscriptenFSFile.cjs

Web application for Visualizing Scientific and Medical datasets

"use strict";

const getMatrixElement = require('./getMatrixElement.js');

const imageJSComponentToIOComponent = require('./imageJSComponentToIOComponent.js');

const imageJSPixelTypeToIOPixelType = require('./imageJSPixelTypeToIOPixelType.js');

const writeImageEmscriptenFSFile = (module, useCompression, image, filePath) => {
  const imageIO = new module.ITKImageIO();
  imageIO.SetFileName(filePath);

  if (!imageIO.CanWriteFile(filePath)) {
    throw new Error('Could not write file: ' + filePath);
  }

  const dimension = image.imageType.dimension;
  imageIO.SetNumberOfDimensions(dimension);
  const ioComponentType = imageJSComponentToIOComponent(module, image.imageType.componentType);
  imageIO.SetComponentType(ioComponentType);
  const ioPixelType = imageJSPixelTypeToIOPixelType(module, image.imageType.pixelType);
  imageIO.SetPixelType(ioPixelType);
  imageIO.SetNumberOfComponents(image.imageType.components);

  for (let ii = 0; ii < dimension; ++ii) {
    imageIO.SetDimensions(ii, image.size[ii]);
    imageIO.SetSpacing(ii, image.spacing[ii]);
    imageIO.SetOrigin(ii, image.origin[ii]);
    const directionColumn = new module.AxisDirectionType();
    directionColumn.resize(dimension, 0.0);

    for (let jj = 0; jj < dimension; ++jj) {
      directionColumn.set(jj, getMatrixElement(image.direction, jj, ii));
    }

    imageIO.SetDirection(ii, directionColumn);
  }

  imageIO.SetUseCompression(useCompression); // Copy data to Emscripten heap (directly accessed from Module.HEAPU8)

  const numberOfBytes = image.data.length * image.data.BYTES_PER_ELEMENT;

  const dataPtr = module._malloc(numberOfBytes);

  const dataHeap = new Uint8Array(module.HEAPU8.buffer, dataPtr, numberOfBytes);
  dataHeap.set(new Uint8Array(image.data.buffer)); // The ImageIO's also call WriteImageInformation() because
  // itk::ImageFileWriter only calls Write()

  imageIO.Write(dataHeap.byteOffset);

  module._free(dataHeap.byteOffset);
};

module.exports = writeImageEmscriptenFSFile;