chart/cl.draw.kbar.js

/**
 * Copyright (c) 2018-present clchart Contributors.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 *
 */

// 以下是 ClLineKBar 的实体定义

import {
  _drawBegin,
  _drawEnd,
  // _setTransColor,
  _drawKBar
} from '../util/cl.draw'
import getValue from '../data/cl.data.tools'
import {
  initCommonInfo
} from '../chart/cl.chart.init'

// 创建时必须带入父类，后面的运算定位都会基于父节点进行；
// 这个类仅仅是画图, 因此需要把可以控制的rect传入进来
/**
 * Class representing ClDrawKBar
 * @export
 * @class ClDrawKBar
 */
export default class ClDrawKBar {
  /**

   * Creates an instance of ClDrawKBar.
   * @param {Object} father
   * @param {Object} rectMain
   */
  constructor (father, rectMain) {
    initCommonInfo(this, father)
    this.rectMain = rectMain

    this.linkInfo = father.father.linkInfo
    this.source = father.father
    this.maxmin = father.maxmin
  }
  /**
   * paint
   * @param {String} key
   * @memberof ClDrawKBar
   */
  onPaint (key) {
    if (key !== undefined) this.hotKey = key
    this.data = this.source.getData(this.hotKey)
    this.codeInfo = this.source.getData('INFO')

    let clr = this.color.red
    // clr = _setTransColor(clr, 1, 'adjust')
    // clr = _setTransColor(clr, 0.7)
    _drawBegin(this.context, clr)
    let open, close
    let before = getValue(this.codeInfo, 'before', 0)
    for (let k = 0, idx = this.linkInfo.minIndex; idx <= this.linkInfo.maxIndex; k++, idx++) {
      if (idx > 0) before = getValue(this.data, 'close', idx - 1)
      open = getValue(this.data, 'open', idx)
      close = getValue(this.data, 'close', idx)
      if (open < close || (open === close && close >= before)) {
      // getValue(this.data, 'high', idx),
      // getValue(this.data, 'low', idx));
        _drawKBar(this.context, {
          filled: this.color.sys === 'white',
          index: k,
          spaceX: this.linkInfo.spaceX,
          unitX: this.linkInfo.unitX,
          unitY: this.maxmin.unitY,
          maxmin: this.maxmin,
          rect: this.rectMain,
          fillclr: clr
        },
        [close,
          getValue(this.data, 'high', idx),
          getValue(this.data, 'low', idx),
          open
        ])
      }
    }
    _drawEnd(this.context)
    clr = this.color.green
    // clr = _setTransColor(clr, 1, 'adjust')
    // clr = _setTransColor(clr, 0.7)
    _drawBegin(this.context, clr)
    for (let k = 0, idx = this.linkInfo.minIndex; idx <= this.linkInfo.maxIndex; k++, idx++) {
      if (idx > 0) before = getValue(this.data, 'close', idx - 1)
      open = getValue(this.data, 'open', idx)
      close = getValue(this.data, 'close', idx)
      if (open > close || (open === close && close < before)) {
        // getValue(this.data, 'high', idx),
        // getValue(this.data, 'low', idx));
        _drawKBar(this.context, {
          filled: true,
          index: k,
          spaceX: this.linkInfo.spaceX,
          unitX: this.linkInfo.unitX,
          unitY: this.maxmin.unitY,
          maxmin: this.maxmin,
          rect: this.rectMain,
          fillclr: clr
        },
        [open,
          getValue(this.data, 'high', idx),
          getValue(this.data, 'low', idx),
          close
        ])
      }
    }
    _drawEnd(this.context)
  }
}