itk/readPolyDataArrayBuffer.js

High performance spatial analysis in a JavaScript runtime environment.

import createWebworkerPromise from './createWebworkerPromise';
import mimeToIO from './MimeToPolyDataIO';
import getFileExtension from './getFileExtension';
import extensionToIO from './extensionToPolyDataIO';
import IOTypes from './IOTypes';
import config from './itkConfig';

var readPolyDataArrayBuffer = function readPolyDataArrayBuffer(webWorker, arrayBuffer, fileName, mimeType) {
  var worker = webWorker;
  return createWebworkerPromise('Pipeline', worker).then(function (_ref) {
    var webworkerPromise = _ref.webworkerPromise,
        usedWorker = _ref.worker;
    worker = usedWorker;
    var extension = getFileExtension(fileName);
    var pipelinePath = null;

    if (mimeToIO.has(mimeType)) {
      pipelinePath = mimeToIO.get(mimeType);
    } else if (extensionToIO.has(extension)) {
      pipelinePath = extensionToIO.get(extension);
    }

    if (pipelinePath === null) {
      Promise.reject(Error('Could not find IO for: ' + fileName));
    }

    var args = [fileName, fileName + '.output.json'];
    var outputs = [{
      path: args[1],
      type: IOTypes.vtkPolyData
    }];
    var inputs = [{
      path: args[0],
      type: IOTypes.Binary,
      data: new Uint8Array(arrayBuffer)
    }];
    var transferables = [];
    inputs.forEach(function (input) {
      // Binary data
      if (input.type === IOTypes.Binary) {
        if (input.data.buffer) {
          transferables.push(input.data.buffer);
        } else if (input.data.byteLength) {
          transferables.push(input.data);
        }
      }
    });
    return webworkerPromise.postMessage({
      operation: 'runPolyDataIOPipeline',
      config: config,
      pipelinePath: pipelinePath,
      args: args,
      outputs: outputs,
      inputs: inputs
    }, transferables).then(function (_ref2) {
      var stdout = _ref2.stdout,
          stderr = _ref2.stderr,
          outputs = _ref2.outputs;
      return Promise.resolve({
        polyData: outputs[0].data,
        webWorker: worker
      });
    });
  });
};

export default readPolyDataArrayBuffer;