pdf.js/worker/canvas.js

/* -*- Mode: Java; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set shiftwidth=2 tabstop=2 autoindent cindent expandtab: */

'use strict';

var JpegStreamProxyCounter = 0;
// WebWorker Proxy for JpegStream.
var JpegStreamProxy = (function() {
  function constructor(bytes, dict) {
    this.id = JpegStreamProxyCounter++;
    this.dict = dict;

    // Tell the main thread to create an image.
    postMessage({
      action: 'jpeg_stream',
      data: {
        id: this.id,
        raw: bytesToString(bytes)
      }
    });
  }

  constructor.prototype = {
    getImage: function() {
      return this;
    },
    getChar: function() {
      error('internal error: getChar is not valid on JpegStream');
    }
  };

  return constructor;
})();

// Really simple GradientProxy. There is currently only one active gradient at
// the time, meaning you can't create a gradient, create a second one and then
// use the first one again. As this isn't used in pdf.js right now, it's okay.
function GradientProxy(cmdQueue, x0, y0, x1, y1) {
  cmdQueue.push(['$createLinearGradient', [x0, y0, x1, y1]]);
  this.addColorStop = function(i, rgba) {
    cmdQueue.push(['$addColorStop', [i, rgba]]);
  };
}

// Really simple PatternProxy.
var patternProxyCounter = 0;
function PatternProxy(cmdQueue, object, kind) {
  this.id = patternProxyCounter++;

  if (!(object instanceof CanvasProxy)) {
    throw 'unkown type to createPattern';
  }

  // Flush the object here to ensure it's available on the main thread.
  // TODO: Make some kind of dependency management, such that the object
  // gets flushed only if needed.
  object.flush();
  cmdQueue.push(['$createPatternFromCanvas', [this.id, object.id, kind]]);
}

var canvasProxyCounter = 0;
function CanvasProxy(width, height) {
  this.id = canvasProxyCounter++;

  // The `stack` holds the rendering calls and gets flushed to the main thead.
  var cmdQueue = this.cmdQueue = [];

  // Dummy context that gets exposed.
  var ctx = {};
  this.getContext = function(type) {
    if (type != '2d') {
      throw 'CanvasProxy can only provide a 2d context.';
    }
    return ctx;
  };

  // Expose only the minimum of the canvas object - there is no dom to do
  // more here.
  this.width = width;
  this.height = height;
  ctx.canvas = this;

  // Setup function calls to `ctx`.
  var ctxFunc = [
  'createRadialGradient',
  'arcTo',
  'arc',
  'fillText',
  'strokeText',
  'createImageData',
  'drawWindow',
  'save',
  'restore',
  'scale',
  'rotate',
  'translate',
  'transform',
  'setTransform',
  'clearRect',
  'fillRect',
  'strokeRect',
  'beginPath',
  'closePath',
  'moveTo',
  'lineTo',
  'quadraticCurveTo',
  'bezierCurveTo',
  'rect',
  'fill',
  'stroke',
  'clip',
  'measureText',
  'isPointInPath',

  // These functions are necessary to track the rendering currentX state.
  // The exact values can be computed on the main thread only, as the
  // worker has no idea about text width.
  '$setCurrentX',
  '$addCurrentX',
  '$saveCurrentX',
  '$restoreCurrentX',
  '$showText',
  '$setFont'
  ];

  function buildFuncCall(name) {
    return function() {
      // console.log("funcCall", name)
      cmdQueue.push([name, Array.prototype.slice.call(arguments)]);
    };
  }
  var name;
  for (var i = 0; i < ctxFunc.length; i++) {
    name = ctxFunc[i];
    ctx[name] = buildFuncCall(name);
  }

  // Some function calls that need more work.

  ctx.createPattern = function(object, kind) {
    return new PatternProxy(cmdQueue, object, kind);
  };

  ctx.createLinearGradient = function(x0, y0, x1, y1) {
    return new GradientProxy(cmdQueue, x0, y0, x1, y1);
  };

  ctx.getImageData = function(x, y, w, h) {
    return {
      width: w,
      height: h,
      data: Uint8ClampedArray(w * h * 4)
    };
  };

  ctx.putImageData = function(data, x, y, width, height) {
    cmdQueue.push(['$putImageData', [data, x, y, width, height]]);
  };

  ctx.drawImage = function(image, x, y, width, height,
                           sx, sy, swidth, sheight) {
    if (image instanceof CanvasProxy) {
      // Send the image/CanvasProxy to the main thread.
      image.flush();
      cmdQueue.push(['$drawCanvas', [image.id, x, y, sx, sy, swidth, sheight]]);
    } else if (image instanceof JpegStreamProxy) {
      cmdQueue.push(['$drawImage', [image.id, x, y, sx, sy, swidth, sheight]]);
    } else {
      throw 'unkown type to drawImage';
    }
  };

  // Setup property access to `ctx`.
  var ctxProp = {
    // "canvas"
    'globalAlpha': '1',
    'globalCompositeOperation': 'source-over',
    'strokeStyle': '#000000',
    'fillStyle': '#000000',
    'lineWidth': '1',
    'lineCap': 'butt',
    'lineJoin': 'miter',
    'miterLimit': '10',
    'shadowOffsetX': '0',
    'shadowOffsetY': '0',
    'shadowBlur': '0',
    'shadowColor': 'rgba(0, 0, 0, 0)',
    'font': '10px sans-serif',
    'textAlign': 'start',
    'textBaseline': 'alphabetic',
    'mozTextStyle': '10px sans-serif',
    'mozImageSmoothingEnabled': 'true'
  };

  function buildGetter(name) {
    return function() {
      return ctx['$' + name];
    };
  }

  function buildSetter(name) {
    return function(value) {
      cmdQueue.push(['$', name, value]);
      return (ctx['$' + name] = value);
    };
  }

  // Setting the value to `stroke|fillStyle` needs special handling, as it
  // might gets an gradient/pattern.
  function buildSetterStyle(name) {
    return function(value) {
      if (value instanceof GradientProxy) {
        cmdQueue.push(['$' + name + 'Gradient']);
      } else if (value instanceof PatternProxy) {
        cmdQueue.push(['$' + name + 'Pattern', [value.id]]);
      } else {
        cmdQueue.push(['$', name, value]);
        return (ctx['$' + name] = value);
      }
    };
  }

  for (var name in ctxProp) {
    ctx['$' + name] = ctxProp[name];
    ctx.__defineGetter__(name, buildGetter(name));

    // Special treatment for `fillStyle` and `strokeStyle`: The passed style
    // might be a gradient. Need to check for that.
    if (name == 'fillStyle' || name == 'strokeStyle') {
      ctx.__defineSetter__(name, buildSetterStyle(name));
    } else {
      ctx.__defineSetter__(name, buildSetter(name));
    }
  }
}

/**
* Sends the current cmdQueue of the CanvasProxy over to the main thread and
* resets the cmdQueue.
*/
CanvasProxy.prototype.flush = function() {
  postMessage({
    action: 'canvas_proxy_cmd_queue',
    data: {
      id: this.id,
      cmdQueue: this.cmdQueue,
      width: this.width,
      height: this.height
    }
  });
  this.cmdQueue.length = 0;
};