MPAndroidChart的K线图上添加均线

MPAndroidChart的K线图上添加均线

均线计算方法：

源码

Demo Code

@Override
public void drawData(Canvas c) {

CandleData candleData = mChart.getCandleData();

for (CandleDataSet set : candleData.getDataSets()) {

if (set.isVisible() && set.getEntryCount() > 0)
drawDataSet(c, set);
}
}

protected void drawDataSet(Canvas c, CandleDataSet dataSet) {

Transformer trans = mChart.getTransformer(dataSet.getAxisDependency());

float phaseX = mAnimator.getPhaseX();
float phaseY = mAnimator.getPhaseY();

int dataSetIndex = mChart.getCandleData().getIndexOfDataSet(dataSet);

List<CandleEntry> entries = dataSet.getYVals();

int minx = Math.max(mMinX, 0);
int maxx = Math.min(mMaxX + 1, entries.size());

int range = (maxx - minx) * 4;
int to = (int) Math.ceil((maxx - minx) * phaseX + minx);

CandleBodyBuffer bodyBuffer = mBodyBuffers[dataSetIndex];
bodyBuffer.setBodySpace(dataSet.getBodySpace());
bodyBuffer.setPhases(phaseX, phaseY);
bodyBuffer.limitFrom(minx);
bodyBuffer.limitTo(maxx);
bodyBuffer.feed(entries);

trans.pointValuesToPixel(bodyBuffer.buffer);

CandleShadowBuffer shadowBuffer = mShadowBuffers[dataSetIndex];
shadowBuffer.setPhases(phaseX, phaseY);
shadowBuffer.limitFrom(minx);
shadowBuffer.limitTo(maxx);
shadowBuffer.feed(entries);

trans.pointValuesToPixel(shadowBuffer.buffer);

mRenderPaint.setStrokeWidth(dataSet.getShadowWidth());

ArrayList<Float> closes = new ArrayList<Float>();
ArrayList<Float> jx = new ArrayList<Float>();
// draw the body
for (int j = 0; j < range; j += 4) {

// get the entry
CandleEntry e = entries.get(j / 4 + minx);

if (!fitsBounds(e.getXIndex(), mMinX, to))
continue;

if (dataSet.getShadowColorSameAsCandle()) {

if (e.getOpen() > e.getClose())
mRenderPaint.setColor(
dataSet.getDecreasingColor() == ColorTemplate.COLOR_NONE ?
dataSet.getColor(j) :
dataSet.getDecreasingColor()
);

else if (e.getOpen() < e.getClose())
mRenderPaint.setColor(
dataSet.getIncreasingColor() == ColorTemplate.COLOR_NONE ?
dataSet.getColor(j) :
dataSet.getIncreasingColor()
);

else
mRenderPaint.setColor(
dataSet.getShadowColor() == ColorTemplate.COLOR_NONE ?
dataSet.getColor(j) :
dataSet.getShadowColor()
);

} else {
mRenderPaint.setColor(
dataSet.getShadowColor() == ColorTemplate.COLOR_NONE ?
dataSet.getColor(j) :
dataSet.getShadowColor()
);
}

mRenderPaint.setStyle(Paint.Style.STROKE);

// draw the shadow
c.drawLine(shadowBuffer.buffer[j], shadowBuffer.buffer[j + 1],
shadowBuffer.buffer[j + 2], shadowBuffer.buffer[j + 3],
mRenderPaint);

float leftBody = bodyBuffer.buffer[j];
float open = bodyBuffer.buffer[j + 1];
float rightBody = bodyBuffer.buffer[j + 2];
float close = bodyBuffer.buffer[j + 3];
closes.add(close);
jx.add((leftBody + rightBody) / 2);

// draw body differently for increasing and decreasing entry
if (open > close) { // decreasing

if (dataSet.getDecreasingColor() == ColorTemplate.COLOR_NONE) {
mRenderPaint.setColor(dataSet.getColor(j / 4 + minx));
} else {
mRenderPaint.setColor(dataSet.getDecreasingColor());
}

mRenderPaint.setStyle(dataSet.getDecreasingPaintStyle());
// draw the body
c.drawRect(leftBody, close, rightBody, open, mRenderPaint);

} else if (open < close) {

if (dataSet.getIncreasingColor() == ColorTemplate.COLOR_NONE) {
mRenderPaint.setColor(dataSet.getColor(j / 4 + minx));
} else {
mRenderPaint.setColor(dataSet.getIncreasingColor());
}

mRenderPaint.setStyle(dataSet.getIncreasingPaintStyle());
// draw the body
c.drawRect(leftBody, open, rightBody, close, mRenderPaint);
} else { // equal values

mRenderPaint.setColor(dataSet.getShadowColor());
c.drawLine(leftBody, open, rightBody, close, mRenderPaint);
}

// 5节点均线
drawLine(c, 5, closes, jx);
// 10节点均线
drawLine(c, 10, closes, jx);
// 30节点均线
drawLine(c, 30, closes, jx);
}
}

/**
* 画均线
* @param c
* @param i
* @param closes
* @param jx
*/
private void drawLine(Canvas c, int i, ArrayList<Float> closes, ArrayList<Float> jx) {
Paint paint = new Paint();
// 画线宽度
paint.setStrokeWidth(1);
// 画线颜色
if (5 == i) {
paint.setColor(Color.BLUE);
} else if (10 == i) {
paint.setColor(Color.RED);
} else if (30 == i) {
paint.setColor(Color.BLACK);
} else {
paint.setColor(Color.GREEN);
}

if (closes.size() >= i + 1) {
float startX = jx.get(closes.size() - 2);
float startY = 0;
for (int x = 0; x < i; x++) {
startY += closes.get(closes.size() - 2 - x);
}
startY = startY / i;
float endX = jx.get(closes.size() - 1);
float endY = 0;
for (int y = 0; y < i; y++) {
endY += closes.get(closes.size() - 1 - y);
}
endY = endY / i;
c.drawLine(startX, startY, endX, endY, paint);
}
}

图形联动