CartoDB-SQL-API/lib/models/bin-encoder.js

'use strict';

function ArrayBufferSer(type, data, options) {
  if(type === undefined) {
      throw "ArrayBufferSer should be created with a type";
  }
  this.options = options || {};
  this._initFunctions();
  this.headerSize = 8;
  this.data = data;
  this.type = type = Math.min(type, ArrayBufferSer.BUFFER);
  var size = this._sizeFor(this.headerSize, data);
  this.buffer = new Buffer(this.headerSize + size);
  this.buffer.writeUInt32BE(type, 0); // this could be one byte but for byte padding is better to be 4 bytes
  this.buffer.writeUInt32BE(size, 4);
  this.offset = this.headerSize;

  var w = this.writeFn[type];

  var i;
  if(!this.options.delta) {
    for(i = 0; i < data.length; ++i) {
      this[w](data[i]);
    }
  } else {
    this[w](data[0]);
    for(i = 1; i < data.length; ++i) {
      this[w](data[i] - data[i - 1]);
    }
  }
}

//
// constants
//
ArrayBufferSer.INT8 = 1;
ArrayBufferSer.UINT8 = 2;
ArrayBufferSer.UINT8_CLAMP = 3;
ArrayBufferSer.INT16 = 4;
ArrayBufferSer.UINT16 = 5;
ArrayBufferSer.INT32 = 6;
ArrayBufferSer.UINT32 = 7;
ArrayBufferSer.FLOAT32 = 8;
//ArrayBufferSer.FLOAT64 = 9; not supported
ArrayBufferSer.STRING = 10;
ArrayBufferSer.BUFFER = 11;

ArrayBufferSer.MAX_PADDING = ArrayBufferSer.INT32;


ArrayBufferSer.typeNames = {
  'int8': ArrayBufferSer.INT8,
  'uint8': ArrayBufferSer.UINT8,
  'uintclamp': ArrayBufferSer.UINT8_CLAMP,
  'int16': ArrayBufferSer.INT16,
  'uint16': ArrayBufferSer.UINT16,
  'int32': ArrayBufferSer.INT32,
  'uint32': ArrayBufferSer.UINT32,
  'float32': ArrayBufferSer.FLOAT32,
  'string': ArrayBufferSer.STRING,
  'buffer': ArrayBufferSer.BUFFER
};

ArrayBufferSer.prototype = {

  // 0 not used
  sizes: [NaN, 1, 1, 1, 2, 2, 4, 4, 4, 8],

  _paddingFor: function(off, type) {
    var s = this.sizes[type];
    if(s) {
      var r = off % s;
      return r === 0 ? 0 : s - r;
    }
    return 0;
  },

  _sizeFor: function(offset, t) {
    var self = this;
    var s = this.sizes[this.type];
    if(s) {
      return s*t.length;
    }
    s = 0;
    if(this.type === ArrayBufferSer.STRING) {
      // calculate size with padding
      t.forEach(function(arr) {
        var pad = self._paddingFor(offset, ArrayBufferSer.MAX_PADDING);
        s += pad;
        offset += pad;
        var len = (self.headerSize + arr.length*2);
        s += len;
        offset += len;
      });
    } else {
      t.forEach(function(arr) {
        var pad = self._paddingFor(offset, ArrayBufferSer.MAX_PADDING);
        s += pad;
        offset += pad;
        s += arr.getSize();
        offset += arr.getSize();
      });
    }
    return s;
  },

  getDataSize: function() {
    return this._sizeFor(0, this.data);
  },

  getSize: function() {
    return this.headerSize + this._sizeFor(this.headerSize, this.data);
  },

  writeFn: [
      '',
      'writeInt8',
      'writeUInt8',
      'writeUInt8Clamp',
      'writeInt16LE',
      'writeUInt16LE',
      'writeUInt32LE',
      'writeUInt32LE',
      'writeFloatLE',
      'writeDoubleLE',
      'writeString',
      'writteBuffer'
  ],

  _initFunctions: function() {
    var self = this;
    this.writeFn.forEach(function(fn) {
      if(self[fn] === undefined) {
        self[fn] = function(d) {
          self.buffer[fn](d, self.offset);
          self.offset += self.sizes[self.type];
        };
      }
    });
  },

  writeUInt8Clamp: function(c) {
    this.buffer.writeUInt8(Math.min(255, c), this.offset);
    this.offset += 1;
  },

  writeString: function(s) {
    var arr = [];
    for(var i = 0, len = s.length; i < len; ++i) {
      arr.push(s.charCodeAt(i));
    }
    var str = new ArrayBufferSer(ArrayBufferSer.UINT16, arr);
    this.writteBuffer(str);
  },

  writteBuffer: function(b) {
    this.offset += this._paddingFor(this.offset, ArrayBufferSer.MAX_PADDING);
    // copy header
    b.buffer.copy(this.buffer, this.offset);
    this.offset += b.buffer.length;
  }

};


module.exports = ArrayBufferSer;