pandas-js/src/js/core/structs.js

Pandas for JavaScript

'use strict';

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.Sheet = exports.Workbook = undefined;

var _classCallCheck2 = require('babel-runtime/helpers/classCallCheck');

var _classCallCheck3 = _interopRequireDefault(_classCallCheck2);

var _createClass2 = require('babel-runtime/helpers/createClass');

var _createClass3 = _interopRequireDefault(_createClass2);

var _immutable = require('immutable');

var _immutable2 = _interopRequireDefault(_immutable);

function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }

// import XLSX from 'xlsx'; TODO figure out if this is the best package

// This does not match the Python pandas

var Workbook = exports.Workbook = function () {
  /**
   * @param {Sheets} sheets
   */
  function Workbook() {
    var sheets = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : new Sheets({});
    (0, _classCallCheck3.default)(this, Workbook);

    this._sheets = sheets;
  }

  /**
   * @returns {Sheets}
   */


  (0, _createClass3.default)(Workbook, [{
    key: 'copy',


    /**
     * Create a copy of the Workbook
     *
     * @returns {Workbook}
     */
    value: function copy() {
      return new Workbook(this._sheets);
    }

    /**
     * Add a values to the Workbook
     *
     * @param {string} sheetName
     * @param {Sheet} sheetContent
     */

  }, {
    key: 'addSheet',
    value: function addSheet(sheetName, sheetContent) {
      console.log(sheetContent.values);
      this._sheets.addSheet(sheetName, sheetContent);
    }
  }, {
    key: 'writeWorkbook',
    value: function writeWorkbook() {
      throw new Error('Workbook writing is not yet implemented');
      // return XLSX.write(this, {bookType: 'xlsx', bookSST: true, type: 'binary'});
    }
  }, {
    key: 'sheets',
    get: function get() {
      return this._sheets;
    }
  }, {
    key: 'SheetNames',
    get: function get() {
      return this.sheets.sheetNames.toArray();
    }
  }, {
    key: 'Sheets',
    get: function get() {
      return this.sheets.sheets.toJS();
    }
  }]);
  return Workbook;
}(); /**
      * structs
      *
      * Description:
      * Primary author(s):
      * Secondary author(s):
      *
      * Notes:
      *
      * January 12, 2017
      * StratoDem Analytics, LLC
      */

var Sheets = function () {
  function Sheets() {
    var initialSheets = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {};
    (0, _classCallCheck3.default)(this, Sheets);

    this._sheets = _immutable2.default.Map(initialSheets);
  }

  /**
   * List of values names in the Sheets
   *
   * @returns {List<string>}
   */


  (0, _createClass3.default)(Sheets, [{
    key: 'addSheet',


    /**
     *
     * @param {string} sheetName
     * @param {Sheet} sheet
     */
    value: function addSheet(sheetName, sheet) {
      this._sheets = this._sheets.set(sheetName, sheet.values);
    }
  }, {
    key: 'sheetNames',
    get: function get() {
      return this._sheets.keySeq().toList();
    }

    /**
     * Map of sheetName, Sheet objects
     *
     * @returns {Map<string, Sheet>}
     */

  }, {
    key: 'sheets',
    get: function get() {
      return this._sheets;
    }
  }]);
  return Sheets;
}();

var Sheet = exports.Sheet = function () {
  /**
   * Construct a Sheet object from a List of Lists of data (from DataFrame.values, e.g.)
   *
   * @param {List<List>} data
   */
  function Sheet(data) {
    (0, _classCallCheck3.default)(this, Sheet);

    this._sheet = this._sheet_from_list_of_lists(data);
  }

  (0, _createClass3.default)(Sheet, [{
    key: '_sheet_from_list_of_lists',


    /**
     * Get a Sheet object's content from a List of Lists
     * As inspiration: view-source:http://sheetjs.com/demos/writexlsx.html
     *
     * @param data
     * @private
     */
    value: function _sheet_from_list_of_lists(data) {
      var range = { s: { c: 10000000, r: 10000000 }, e: { c: 0, r: 0 } };
      var ws = {};

      function datenum(v, date1904) {
        var epoch = Date.parse(date1904 ? v + 1462 : v);
        return (epoch - new Date(Date.UTC(1899, 11, 30))) / (24 * 60 * 60 * 1000);
      }

      data.forEach(function (row, idxRow) {
        row.forEach(function (v, idxCol) {
          if (v === null) return;

          if (range.s.r > idxRow) range.s.r = idxRow;
          if (range.s.c > idxCol) range.s.c = idxCol;
          if (range.e.r < idxRow) range.e.r = idxRow;
          if (range.e.c < idxCol) range.e.c = idxCol;

          var cell = { v: v };
          throw new Error('Sheet not yet implemented');
          // const cell_ref = XLSX.utils.encode_cell({c: idxCol, r: idxRow}); TODO

          if (typeof cell.v === 'number') cell.t = 'n';else if (cell.v instanceof Date) {
            cell.t = 'n';cell.z = XLSX.SSF._table[14];
            cell.v = datenum(cell.v);
          } else cell.t = 's';

          ws[cell_ref] = cell;
        });
      });

      // if (range.s.c < 10000000) ws['!ref'] = XLSX.utils.encode_range(range); TODO

      return ws;
    }
  }, {
    key: 'values',
    get: function get() {
      return this._sheet;
    }
  }]);
  return Sheet;
}();