tema.src.js 13.3 KB
/**
 * @license Highstock JS v7.2.0 (2019-09-03)
 *
 * Indicator series type for Highstock
 *
 * (c) 2010-2019 Rafal Sebestjanski
 *
 * License: www.highcharts.com/license
 */
'use strict';
(function (factory) {
    if (typeof module === 'object' && module.exports) {
        factory['default'] = factory;
        module.exports = factory;
    } else if (typeof define === 'function' && define.amd) {
        define('highcharts/indicators/tema', ['highcharts', 'highcharts/modules/stock'], function (Highcharts) {
            factory(Highcharts);
            factory.Highcharts = Highcharts;
            return factory;
        });
    } else {
        factory(typeof Highcharts !== 'undefined' ? Highcharts : undefined);
    }
}(function (Highcharts) {
    var _modules = Highcharts ? Highcharts._modules : {};
    function _registerModule(obj, path, args, fn) {
        if (!obj.hasOwnProperty(path)) {
            obj[path] = fn.apply(null, args);
        }
    }
    _registerModule(_modules, 'mixins/indicator-required.js', [_modules['parts/Globals.js']], function (H) {
        /**
         *
         *  (c) 2010-2019 Daniel Studencki
         *
         *  License: www.highcharts.com/license
         *
         *  !!!!!!! SOURCE GETS TRANSPILED BY TYPESCRIPT. EDIT TS FILE ONLY. !!!!!!!
         *
         * */
        var error = H.error;
        /* eslint-disable no-invalid-this, valid-jsdoc */
        var requiredIndicatorMixin = {
            /**
             * Check whether given indicator is loaded, else throw error.
             * @private
             * @param {Highcharts.Indicator} indicator
             *        Indicator constructor function.
             * @param {string} requiredIndicator
             *        Required indicator type.
             * @param {string} type
             *        Type of indicator where function was called (parent).
             * @param {Highcharts.IndicatorConstructorFunction} callback
             *        Callback which is triggered if the given indicator is loaded.
             *        Takes indicator as an argument.
             * @param {string} errMessage
             *        Error message that will be logged in console.
             * @return {boolean}
             *         Returns false when there is no required indicator loaded.
             */
            isParentLoaded: function (indicator, requiredIndicator, type, callback, errMessage) {
                if (indicator) {
                    return callback ? callback(indicator) : true;
                }
                error(errMessage || this.generateMessage(type, requiredIndicator));
                return false;
            },
            /**
             * @private
             * @param {string} indicatorType
             *        Indicator type
             * @param {string} required
             *        Required indicator
             * @return {string}
             *         Error message
             */
            generateMessage: function (indicatorType, required) {
                return 'Error: "' + indicatorType +
                    '" indicator type requires "' + required +
                    '" indicator loaded before. Please read docs: ' +
                    'https://api.highcharts.com/highstock/plotOptions.' +
                    indicatorType;
            }
        };

        return requiredIndicatorMixin;
    });
    _registerModule(_modules, 'indicators/tema.src.js', [_modules['parts/Globals.js'], _modules['parts/Utilities.js'], _modules['mixins/indicator-required.js']], function (H, U, requiredIndicatorMixin) {
        /* *
         *
         *  License: www.highcharts.com/license
         *
         * */



        var isArray = U.isArray;


        var EMAindicator = H.seriesTypes.ema,
            requiredIndicator = requiredIndicatorMixin,
            correctFloat = H.correctFloat;

        /**
         * The TEMA series type.
         *
         * @private
         * @class
         * @name Highcharts.seriesTypes.tema
         *
         * @augments Highcharts.Series
         */
        H.seriesType(
            'tema',
            'ema',
            /**
             * Triple exponential moving average (TEMA) indicator. This series requires
             * `linkedTo` option to be set and should be loaded after the
             * `stock/indicators/indicators.js` and `stock/indicators/ema.js`.
             *
             * Requires `https://code.highcharts.com/stock/indicators/ema.js`.
             *
             * @sample {highstock} stock/indicators/tema
             *         TEMA indicator
             *
             * @extends      plotOptions.ema
             * @since        7.0.0
             * @product      highstock
             * @excluding    allAreas, colorAxis, compare, compareBase, joinBy, keys,
             *               navigatorOptions, pointInterval, pointIntervalUnit,
             *               pointPlacement, pointRange, pointStart, showInNavigator,
             *               stacking
             * @optionparent plotOptions.tema
             */
            {},
            /**
             * @lends Highcharts.Series#
             */
            {
                init: function () {
                    var args = arguments,
                        ctx = this;

                    requiredIndicator.isParentLoaded(
                        EMAindicator,
                        'ema',
                        ctx.type,
                        function (indicator) {
                            indicator.prototype.init.apply(ctx, args);
                        }
                    );
                },
                getEMA: function (
                    yVal,
                    prevEMA,
                    SMA,
                    index,
                    i,
                    xVal
                ) {
                    return EMAindicator.prototype.calculateEma(
                        xVal || [],
                        yVal,
                        i === undefined ? 1 : i,
                        this.chart.series[0].EMApercent,
                        prevEMA,
                        index === undefined ? -1 : index,
                        SMA
                    );
                },
                getTemaPoint: function (
                    xVal,
                    tripledPeriod,
                    EMAlevels,
                    i
                ) {
                    var TEMAPoint = [
                        xVal[i - 3],
                        correctFloat(
                            3 * EMAlevels.level1 -
                            3 * EMAlevels.level2 + EMAlevels.level3
                        )
                    ];

                    return TEMAPoint;
                },
                getValues: function (series, params) {
                    var period = params.period,
                        doubledPeriod = 2 * period,
                        tripledPeriod = 3 * period,
                        xVal = series.xData,
                        yVal = series.yData,
                        yValLen = yVal ? yVal.length : 0,
                        index = -1,
                        accumulatePeriodPoints = 0,
                        SMA = 0,
                        TEMA = [],
                        xDataTema = [],
                        yDataTema = [],
                        // EMA of previous point
                        prevEMA,
                        prevEMAlevel2,
                        // EMA values array
                        EMAvalues = [],
                        EMAlevel2values = [],
                        i,
                        TEMAPoint,
                        // This object contains all EMA EMAlevels calculated like below
                        // EMA = level1
                        // EMA(EMA) = level2,
                        // EMA(EMA(EMA)) = level3,
                        EMAlevels = {};

                    series.EMApercent = (2 / (period + 1));

                    // Check period, if bigger than EMA points length, skip
                    if (yValLen < 3 * period - 2) {
                        return false;
                    }

                    // Switch index for OHLC / Candlestick / Arearange
                    if (isArray(yVal[0])) {
                        index = params.index ? params.index : 0;
                    }

                    // Accumulate first N-points
                    accumulatePeriodPoints =
                      EMAindicator.prototype.accumulatePeriodPoints(
                          period,
                          index,
                          yVal
                      );

                    // first point
                    SMA = accumulatePeriodPoints / period;
                    accumulatePeriodPoints = 0;

                    // Calculate value one-by-one for each period in visible data
                    for (i = period; i < yValLen + 3; i++) {
                        if (i < yValLen + 1) {
                            EMAlevels.level1 = this.getEMA(
                                yVal,
                                prevEMA,
                                SMA,
                                index,
                                i
                            )[1];
                            EMAvalues.push(EMAlevels.level1);
                        }
                        prevEMA = EMAlevels.level1;

                        // Summing first period points for ema(ema)
                        if (i < doubledPeriod) {
                            accumulatePeriodPoints += EMAlevels.level1;
                        } else {
                            // Calculate dema
                            // First dema point
                            if (i === doubledPeriod) {
                                SMA = accumulatePeriodPoints / period;
                                accumulatePeriodPoints = 0;
                            }
                            EMAlevels.level1 = EMAvalues[i - period - 1];
                            EMAlevels.level2 = this.getEMA(
                                [EMAlevels.level1],
                                prevEMAlevel2,
                                SMA
                            )[1];
                            EMAlevel2values.push(EMAlevels.level2);
                            prevEMAlevel2 = EMAlevels.level2;
                            // Summing first period points for ema(ema(ema))
                            if (i < tripledPeriod) {
                                accumulatePeriodPoints += EMAlevels.level2;
                            } else {
                                // Calculate tema
                                // First tema point
                                if (i === tripledPeriod) {
                                    SMA = accumulatePeriodPoints / period;
                                }
                                if (i === yValLen + 1) {
                                    // Calculate the last ema and emaEMA points
                                    EMAlevels.level1 = EMAvalues[i - period - 1];
                                    EMAlevels.level2 = this.getEMA(
                                        [EMAlevels.level1],
                                        prevEMAlevel2,
                                        SMA
                                    )[1];
                                    EMAlevel2values.push(EMAlevels.level2);
                                }
                                EMAlevels.level1 = EMAvalues[i - period - 2];
                                EMAlevels.level2 = EMAlevel2values[i - 2 * period - 1];
                                EMAlevels.level3 = this.getEMA(
                                    [EMAlevels.level2],
                                    EMAlevels.prevLevel3,
                                    SMA
                                )[1];
                                TEMAPoint = this.getTemaPoint(
                                    xVal,
                                    tripledPeriod,
                                    EMAlevels,
                                    i
                                );
                                // Make sure that point exists (for TRIX oscillator)
                                if (TEMAPoint) {
                                    TEMA.push(TEMAPoint);
                                    xDataTema.push(TEMAPoint[0]);
                                    yDataTema.push(TEMAPoint[1]);
                                }
                                EMAlevels.prevLevel3 = EMAlevels.level3;
                            }
                        }
                    }

                    return {
                        values: TEMA,
                        xData: xDataTema,
                        yData: yDataTema
                    };
                }
            }
        );

        /**
         * A `TEMA` series. If the [type](#series.ema.type) option is not
         * specified, it is inherited from [chart.type](#chart.type).
         *
         * @extends   series,plotOptions.ema
         * @since     7.0.0
         * @product   highstock
         * @excluding allAreas, colorAxis, compare, compareBase, dataParser, dataURL,
         *            joinBy, keys, navigatorOptions, pointInterval, pointIntervalUnit,
         *            pointPlacement, pointRange, pointStart, showInNavigator, stacking
         * @apioption series.tema
         */

    });
    _registerModule(_modules, 'masters/indicators/tema.src.js', [], function () {


    });
}));