Android Open Source - Android-Charts Candle Stick Chart

From Project

Back to project page Android-Charts.
License

The source code is released under:
Apache License??Version 2.0, January 2004??http://www.apache.org/licenses/ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION 1. Definitions. "License" shall mean the terms and condi...
If you think the Android project Android-Charts listed in this page is inappropriate, such as containing malicious code/tools or violating the copyright, please email info at java2s dot com, thanks.
Java Source Code

/*
 * CandleStickChart.java/*from  w w w. j  a  va 2s  .  c  om*/
 * Android-Charts
 *
 * Created by limc on 2011/05/29.
 *
 * Copyright 2011 limc.cn All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package cn.limc.androidcharts.view;

import cn.limc.androidcharts.entity.IMeasurable;
import cn.limc.androidcharts.entity.OHLCEntity;

import android.content.Context;
import android.graphics.Canvas;
import android.graphics.Color;
import android.graphics.Paint;
import android.graphics.PointF;
import android.util.AttributeSet;

/**
 * 
 * <p>
 * CandleStickChart is a kind of graph that draw the OHLCs on a GridChart if you
 * want display some moving average lines on this graph, please use see
 * MACandleStickChart for more information
 * </p>
 * <p>
 * CandleStickChart???GridChart??????????????????????????????????????????????????
 * ?????????????????????????????MACandleStickChart???????????????????????
 * </p>
 * <p>
 * CandleStickChart??GridChart???K??????????????????????????????? MACandleStickChart?
 * </p>
 * 
 * @author limc
 * @version v1.0 2011/05/30 16:29:41
 * @see CandleStickChart
 * @see MACandleStickChart
 * 
 */
public class CandleStickChart extends StickChart {
  /**
   * <p>
   * Default price up stick's border color
   * </p>
   * <p>
   * ???????????????????????????
   * </p>
   * <p>
   * ?????????
   * </p>
   */
  public static final int DEFAULT_POSITIVE_STICK_BORDER_COLOR = Color.RED;

  /**
   * <p>
   * Default price up stick's fill color
   * </p>
   * <p>
   * ???????????????????????
   * </p>
   * <p>
   * ?????????
   * </p>
   */
  public static final int DEFAULT_POSITIVE_STICK_FILL_COLOR = Color.RED;

  /**
   * <p>
   * Default price down stick's border color
   * </p>
   * <p>
   * ????????????????????????
   * </p>
   * <p>
   * ?????????
   * </p>
   */
  public static final int DEFAULT_NEGATIVE_STICK_BORDER_COLOR = Color.GREEN;

  /**
   * <p>
   * Default price down stick's fill color
   * </p>
   * <p>
   * ????????????????????
   * </p>
   * <p>
   * ?????????
   * </p>
   */
  public static final int DEFAULT_NEGATIVE_STICK_FILL_COLOR = Color.GREEN;

  /**
   * <p>
   * Default price no change stick's color (cross-star,T-like etc.)
   * </p>
   * <p>
   * ????????????????????
   * </p>
   * <p>
   * ???????????
   * </p>
   */
  public static final int DEFAULT_CROSS_STAR_COLOR = Color.LTGRAY;

  /**
   * <p>
   * Price up stick's border color
   * </p>
   * <p>
   * ??????????????????
   * </p>
   * <p>
   * ???????
   * </p>
   */
  private int positiveStickBorderColor = DEFAULT_POSITIVE_STICK_BORDER_COLOR;

  /**
   * <p>
   * Price up stick's fill color
   * </p>
   * <p>
   * ??????????????
   * </p>
   * <p>
   * ???????
   * </p>
   */
  private int positiveStickFillColor = DEFAULT_POSITIVE_STICK_FILL_COLOR;

  /**
   * <p>
   * Price down stick's border color
   * </p>
   * <p>
   * ???????????????
   * </p>
   * <p>
   * ???????
   * </p>
   */

  private int negativeStickBorderColor = DEFAULT_NEGATIVE_STICK_BORDER_COLOR;

  /**
   * <p>
   * Price down stick's fill color
   * </p>
   * <p>
   * ???????????
   * </p>
   * <p>
   * ???????
   * </p>
   */
  private int negativeStickFillColor = DEFAULT_NEGATIVE_STICK_FILL_COLOR;

  /**
   * <p>
   * Price no change stick's color (cross-star,T-like etc.)
   * </p>
   * <p>
   * ?????????????????????
   * </p>
   * <p>
   * ???????????????,????,T??????
   * </p>
   */
  private int crossStarColor = DEFAULT_CROSS_STAR_COLOR;

  /*
   * (non-Javadoc)
   * 
   * @param context
   * 
   * @see cn.limc.androidcharts.view.AbstractBaseChart#BaseChart(Context)
   */
  public CandleStickChart(Context context) {
    super(context);
  }

  /*
   * (non-Javadoc)
   * 
   * @param context
   * 
   * @param attrs
   * 
   * @param defStyle
   * 
   * @see cn.limc.androidcharts.view.AbstractBaseChart#BaseChart(Context,
   * AttributeSet, int)
   */
  public CandleStickChart(Context context, AttributeSet attrs, int defStyle) {
    super(context, attrs, defStyle);
  }

  /*
   * (non-Javadoc)
   * 
   * @param context
   * 
   * @param attrs
   * 
   * @see cn.limc.androidcharts.view.AbstractBaseChart#BaseChart(Context,
   * AttributeSet)
   */
  public CandleStickChart(Context context, AttributeSet attrs) {
    super(context, attrs);
  }

  @Override
  protected void calcDataValueRange() {
    double maxValue = Double.MIN_VALUE;
    double minValue = Double.MAX_VALUE;
    IMeasurable first;
    if (axisYPosition == AXIS_Y_POSITION_LEFT) {
      first = this.stickData.get(0);
    } else {
      first = this.stickData.get(this.stickData.size() - 1);
    }
    // ???stick????????
    if (first.getHigh() == 0 && first.getLow() == 0) {

    } else {
      maxValue = first.getHigh();
      minValue = first.getLow();
    }
    for (int i = 0; i < this.maxSticksNum; i++) {
      OHLCEntity stick;
      if (axisYPosition == AXIS_Y_POSITION_LEFT) {
        stick = (OHLCEntity) this.stickData.get(i);
      } else {
        stick = (OHLCEntity) this.stickData.get(this.stickData.size()
            - 1 - i);
      }

      if (stick.getOpen() == 0 && stick.getHigh() == 0
          && stick.getLow() == 0) {
        // ???????????
        if (stick.getClose() > 0) {
          if (stick.getClose() < minValue) {
            minValue = stick.getClose();
          }

          if (stick.getClose() > maxValue) {
            maxValue = stick.getClose();
          }
        }
      } else {
        if (stick.getLow() < minValue) {
          minValue = stick.getLow();
        }

        if (stick.getHigh() > maxValue) {
          maxValue = stick.getHigh();
        }
      }
    }

    this.maxValue = maxValue;
    this.minValue = minValue;
  }

  /*
   * (non-Javadoc)
   * 
   * <p>Called when is going to draw this chart<p> <p>???????????????????????<p>
   * <p>???????<p>
   * 
   * @param canvas
   * 
   * @see android.view.View#onDraw(android.graphics.Canvas)
   */
  @Override
  protected void onDraw(Canvas canvas) {
    super.onDraw(canvas);
  }

  /**
   * <p>
   * draw sticks
   * </p>
   * <p>
   * ??????????
   * </p>
   * <p>
   * ??????
   * </p>
   * 
   * @param canvas
   */
  @Override
  protected void drawSticks(Canvas canvas) {
    if (null == stickData) {
      return;
    }
    if (stickData.size() <= 0) {
      return;
    }

    Paint mPaintPositive = new Paint();
    mPaintPositive.setColor(positiveStickFillColor);

    Paint mPaintNegative = new Paint();
    mPaintNegative.setColor(negativeStickFillColor);

    Paint mPaintCross = new Paint();
    mPaintCross.setColor(crossStarColor);

    float stickWidth = dataQuadrant.getQuadrantPaddingWidth() / maxSticksNum
        - stickSpacing;

    if (axisYPosition == AXIS_Y_POSITION_LEFT) {
      float stickX = dataQuadrant.getQuadrantPaddingStartX();

      for (int i = 0; i < stickData.size(); i++) {
        OHLCEntity ohlc = (OHLCEntity) stickData.get(i);
        float openY = (float) ((1f - (ohlc.getOpen() - minValue)
            / (maxValue - minValue))
            * (dataQuadrant.getQuadrantPaddingHeight()) + dataQuadrant.getQuadrantPaddingStartY());
        float highY = (float) ((1f - (ohlc.getHigh() - minValue)
            / (maxValue - minValue))
            * (dataQuadrant.getQuadrantPaddingHeight()) + dataQuadrant.getQuadrantPaddingStartY());
        float lowY = (float) ((1f - (ohlc.getLow() - minValue)
            / (maxValue - minValue))
            * (dataQuadrant.getQuadrantPaddingHeight()) + dataQuadrant.getQuadrantPaddingStartY());
        float closeY = (float) ((1f - (ohlc.getClose() - minValue)
            / (maxValue - minValue))
            * (dataQuadrant.getQuadrantPaddingHeight()) + dataQuadrant.getQuadrantPaddingStartY());

        if (ohlc.getOpen() < ohlc.getClose()) {
          // stick or line
          if (stickWidth >= 2f) {
            canvas.drawRect(stickX, closeY, stickX + stickWidth,
                openY, mPaintPositive);
          }
          canvas.drawLine(stickX + stickWidth / 2f, highY, stickX
              + stickWidth / 2f, lowY, mPaintPositive);
        } else if (ohlc.getOpen() > ohlc.getClose()) {
          // stick or line
          if (stickWidth >= 2f) {
            canvas.drawRect(stickX, openY, stickX + stickWidth,
                closeY, mPaintNegative);
          }
          canvas.drawLine(stickX + stickWidth / 2f, highY, stickX
              + stickWidth / 2f, lowY, mPaintNegative);
        } else {
          // line or point
          if (stickWidth >= 2f) {
            canvas.drawLine(stickX, closeY, stickX + stickWidth,
                openY, mPaintCross);
          }
          canvas.drawLine(stickX + stickWidth / 2f, highY, stickX
              + stickWidth / 2f, lowY, mPaintCross);
        }

        // next x
        stickX = stickX + stickSpacing + stickWidth;
      }
    } else {
      float stickX = dataQuadrant.getQuadrantPaddingEndX() - stickWidth;
      for (int i = stickData.size() - 1; i >= 0; i--) {
        OHLCEntity ohlc = (OHLCEntity) stickData.get(i);
        float openY = (float) ((1f - (ohlc.getOpen() - minValue)
            / (maxValue - minValue))
            * (dataQuadrant.getQuadrantPaddingHeight()) + dataQuadrant.getQuadrantPaddingStartY());
        float highY = (float) ((1f - (ohlc.getHigh() - minValue)
            / (maxValue - minValue))
            * (dataQuadrant.getQuadrantPaddingHeight()) + dataQuadrant.getQuadrantPaddingStartY());
        float lowY = (float) ((1f - (ohlc.getLow() - minValue)
            / (maxValue - minValue))
            * (dataQuadrant.getQuadrantPaddingHeight()) + dataQuadrant.getQuadrantPaddingStartY());
        float closeY = (float) ((1f - (ohlc.getClose() - minValue)
            / (maxValue - minValue))
            * (dataQuadrant.getQuadrantPaddingHeight()) + dataQuadrant.getQuadrantPaddingStartY());

        if (ohlc.getOpen() < ohlc.getClose()) {
          // stick or line
          if (stickWidth >= 2f) {
            canvas.drawRect(stickX, closeY, stickX + stickWidth,
                openY, mPaintPositive);
          }
          canvas.drawLine(stickX + stickWidth / 2f, highY, stickX
              + stickWidth / 2f, lowY, mPaintPositive);
        } else if (ohlc.getOpen() > ohlc.getClose()) {
          // stick or line
          if (stickWidth >= 2f) {
            canvas.drawRect(stickX, openY, stickX + stickWidth,
                closeY, mPaintNegative);
          }
          canvas.drawLine(stickX + stickWidth / 2f, highY, stickX
              + stickWidth / 2f, lowY, mPaintNegative);
        } else {
          // line or point
          if (stickWidth >= 2f) {
            canvas.drawLine(stickX, closeY, stickX + stickWidth,
                openY, mPaintCross);
          }
          canvas.drawLine(stickX + stickWidth / 2f, highY, stickX
              + stickWidth / 2f, lowY, mPaintCross);
        }

        // next x
        stickX = stickX - stickSpacing - stickWidth;
      }
    }
  }
  
  protected PointF calcBindPoint(float x ,float y) {
    float calcX = 0;
    float calcY = 0;
    
    int index = calcSelectedIndex(x,y);
    
    float stickWidth = dataQuadrant.getQuadrantPaddingWidth() / maxSticksNum;
    OHLCEntity stick = (OHLCEntity)stickData.get(index);
    calcY = (float) ((1f - (stick.getClose() - minValue)
        / (maxValue - minValue))
        * (dataQuadrant.getQuadrantPaddingHeight()) + dataQuadrant.getQuadrantPaddingStartY());
    if (axisYPosition == AXIS_Y_POSITION_LEFT) {
      calcX = dataQuadrant.getQuadrantPaddingStartX() + stickWidth * index + stickWidth / 2;
    }else{
      if(stickData.size() - index <= maxSticksNum){
        calcX = dataQuadrant.getQuadrantPaddingEndX() - stickWidth * (stickData.size() - index) + stickWidth /2 ;
      }else{
        calcX = x;
      }
    }
    
    return new PointF(calcX,calcY);
  }

  /**
   * @return the positiveStickBorderColor
   */
  public int getPositiveStickBorderColor() {
    return positiveStickBorderColor;
  }

  /**
   * @param positiveStickBorderColor
   *            the positiveStickBorderColor to set
   */
  public void setPositiveStickBorderColor(int positiveStickBorderColor) {
    this.positiveStickBorderColor = positiveStickBorderColor;
  }

  /**
   * @return the positiveStickFillColor
   */
  public int getPositiveStickFillColor() {
    return positiveStickFillColor;
  }

  /**
   * @param positiveStickFillColor
   *            the positiveStickFillColor to set
   */
  public void setPositiveStickFillColor(int positiveStickFillColor) {
    this.positiveStickFillColor = positiveStickFillColor;
  }

  /**
   * @return the negativeStickBorderColor
   */
  public int getNegativeStickBorderColor() {
    return negativeStickBorderColor;
  }

  /**
   * @param negativeStickBorderColor
   *            the negativeStickBorderColor to set
   */
  public void setNegativeStickBorderColor(int negativeStickBorderColor) {
    this.negativeStickBorderColor = negativeStickBorderColor;
  }

  /**
   * @return the negativeStickFillColor
   */
  public int getNegativeStickFillColor() {
    return negativeStickFillColor;
  }

  /**
   * @param negativeStickFillColor
   *            the negativeStickFillColor to set
   */
  public void setNegativeStickFillColor(int negativeStickFillColor) {
    this.negativeStickFillColor = negativeStickFillColor;
  }

  /**
   * @return the crossStarColor
   */
  public int getCrossStarColor() {
    return crossStarColor;
  }

  /**
   * @param crossStarColor
   *            the crossStarColor to set
   */
  public void setCrossStarColor(int crossStarColor) {
    this.crossStarColor = crossStarColor;
  }

}
Java Source Code List
