@antv/coord/lib/coordinate.js

"use strict";
var __spreadArray = (this && this.__spreadArray) || function (to, from) {
    for (var i = 0, il = from.length, j = to.length; i < il; i++, j++)
        to[j] = from[i];
    return to;
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.Coordinate = void 0;
var util_1 = require("@antv/util");
var matrix_util_1 = require("@antv/matrix-util");
var utils_1 = require("./utils");
var transforms_1 = require("./transforms");
var Coordinate = /** @class */ (function () {
    /**
     * Create a new Coordinate Object.
     * @param options Custom options
     */
    function Coordinate(options) {
        // 当前的选项
        this.options = {
            x: 0,
            y: 0,
            width: 300,
            height: 150,
            transformations: [],
        };
        // 当前可以使用的变换
        this.transformers = {
            cartesian: transforms_1.cartesian,
            translate: transforms_1.translate,
            custom: transforms_1.custom,
            matrix: transforms_1.matrix,
            polar: transforms_1.polar,
            transpose: transforms_1.transpose,
            scale: transforms_1.scale,
            'shear.x': transforms_1.shearX,
            'shear.y': transforms_1.shearY,
            reflect: transforms_1.reflect,
            'reflect.x': transforms_1.reflectX,
            'reflect.y': transforms_1.reflectY,
            rotate: transforms_1.rotate,
            helix: transforms_1.helix,
            parallel: transforms_1.parallel,
            fisheye: transforms_1.fisheye,
            'fisheye.x': transforms_1.fisheyeX,
            'fisheye.y': transforms_1.fisheyeY,
            'fisheye.circular': transforms_1.fisheyeCircular,
        };
        this.update(options);
    }
    /**
     * Update options and inner state.
     * @param options Options to be updated
     */
    Coordinate.prototype.update = function (options) {
        this.options = util_1.deepMix({}, this.options, options);
        this.recoordinate();
    };
    /**
     * Returns a new Coordinate with same options.
     * @returns Coordinate
     */
    Coordinate.prototype.clone = function () {
        return new Coordinate(this.options);
    };
    /**
     * Returns current options.
     * @returns options
     */
    Coordinate.prototype.getOptions = function () {
        return this.options;
    };
    /**
     * Clear transformations and update.
     */
    Coordinate.prototype.clear = function () {
        this.update({
            transformations: [],
        });
    };
    /**
     * Returns the size of the bounding box of the coordinate.
     * @returns [width, height]
     */
    Coordinate.prototype.getSize = function () {
        var _a = this.options, width = _a.width, height = _a.height;
        return [width, height];
    };
    /**
     * Returns the center of the bounding box of the coordinate.
     * @returns [centerX, centerY]
     */
    Coordinate.prototype.getCenter = function () {
        var _a = this.options, x = _a.x, y = _a.y, width = _a.width, height = _a.height;
        return [(x + width) / 2, (y + height) / 2];
    };
    /**
     * Add selected transformation.
     * @param args transform type and params
     * @returns Coordinate
     */
    Coordinate.prototype.transform = function () {
        var args = [];
        for (var _i = 0; _i < arguments.length; _i++) {
            args[_i] = arguments[_i];
        }
        var transformations = this.options.transformations;
        this.update({
            transformations: __spreadArray(__spreadArray([], transformations), [__spreadArray([], args)]),
        });
        return this;
    };
    /**
     * Apples transformations for the current vector.
     * @param vector original vector2
     * @returns transformed vector2
     */
    Coordinate.prototype.map = function (vector) {
        return this.output(vector);
    };
    /**
     * Apples invert transformations for the current vector.
     * @param vector transformed vector2
     * @param vector original vector2
     */
    Coordinate.prototype.invert = function (vector) {
        return this.input(vector);
    };
    Coordinate.prototype.recoordinate = function () {
        this.output = this.compose();
        this.input = this.compose(true);
    };
    // 将所有的变换合成一个函数
    // 变换有两种类型：矩阵变换和函数变换
    // 处理过程中需要把连续的矩阵变换合成一个变换函数，然后在和其他变换函数合成最终的变换函数
    Coordinate.prototype.compose = function (invert) {
        if (invert === void 0) { invert = false; }
        var transformations = invert ? __spreadArray([], this.options.transformations).reverse() : this.options.transformations;
        var getter = invert ? function (d) { return d.untransform; } : function (d) { return d.transform; };
        var matrixes = [];
        var transforms = [];
        var add = function (transform, extended) {
            if (extended === void 0) { extended = true; }
            return transforms.push(extended ? utils_1.extend(transform) : transform);
        };
        for (var _i = 0, transformations_1 = transformations; _i < transformations_1.length; _i++) {
            var _a = transformations_1[_i], name_1 = _a[0], args = _a.slice(1);
            var createTransformer = this.transformers[name_1];
            if (createTransformer) {
                var _b = this.options, x = _b.x, y = _b.y, width = _b.width, height = _b.height;
                var transformer = createTransformer(__spreadArray([], args), x, y, width, height);
                if (utils_1.isMatrix(transformer)) {
                    // 如果当前变换是矩阵变换，那么先保存下来
                    matrixes.push(transformer);
                }
                else {
                    // 如果当前变换是函数变换，并且之前有没有合成的矩阵变换，那么现将之前的矩阵变换合成
                    if (matrixes.length) {
                        var transform_1 = this.createMatrixTransform(matrixes, invert);
                        add(transform_1);
                        matrixes.splice(0, matrixes.length);
                    }
                    var transform = getter(transformer) || util_1.identity;
                    add(transform, name_1 !== 'parallel'); // 对于非平行坐标系的变换需要扩展
                }
            }
        }
        // 合成剩下的矩阵变换
        if (matrixes.length) {
            var transform = this.createMatrixTransform(matrixes, invert);
            add(transform);
        }
        return utils_1.compose.apply(void 0, transforms);
    };
    // 将连续的矩阵的运算合成一个变换函数
    Coordinate.prototype.createMatrixTransform = function (matrixes, invert) {
        var matrix = matrix_util_1.mat3.create();
        if (invert)
            matrixes.reverse();
        matrixes.forEach(function (m) { return matrix_util_1.mat3.mul(matrix, matrix, m); });
        if (invert) {
            matrix_util_1.mat3.invert(matrix, matrix_util_1.mat3.clone(matrix));
        }
        return function (vector) {
            var vector3 = [vector[0], vector[1], 1];
            matrix_util_1.vec3.transformMat3(vector3, vector3, matrix);
            return [vector3[0], vector3[1]];
        };
    };
    return Coordinate;
}());
exports.Coordinate = Coordinate;
//# sourceMappingURL=coordinate.js.map