package com.kehua.framework.utils;

import java.io.FileOutputStream;

import org.apache.poi.ss.usermodel.Cell;

import org.apache.poi.ss.usermodel.Row;

import org.apache.poi.ss.util.CellRangeAddress;

import org.apache.poi.xddf.usermodel.PresetColor;

import org.apache.poi.xddf.usermodel.XDDFColor;

import org.apache.poi.xddf.usermodel.XDDFLineProperties;

import org.apache.poi.xddf.usermodel.XDDFShapeProperties;

import org.apache.poi.xddf.usermodel.XDDFSolidFillProperties;

import org.apache.poi.xddf.usermodel.chart.*;

import org.apache.poi.xssf.usermodel.XSSFChart;

import org.apache.poi.xssf.usermodel.XSSFClientAnchor;

import org.apache.poi.xssf.usermodel.XSSFDrawing;

import org.apache.poi.xssf.usermodel.XSSFSheet;

import org.apache.poi.xssf.usermodel.XSSFWorkbook;

public class DrawXlsxUtil {

/**

* 根据sheet数据画图(单序列)

* caution:文件打开关闭由外部操作

* @param sourceSheet 数据源sheet表(获取数据的表)

* @param distSheet 目标sheet表(画图的表)

* @param anchorPosition 画板位置

* @param chartTitle 图名称

* @param xAxisTitle x轴标识

* @param yAxisTitle y轴标识

* @param xAxisDataRangeAddress x轴数据位置(根据sheet获取)

* @param yAxisDataRangeAddress y轴数据位置

* @param chartType 图样式：暂支持折线、条形

* @param color 颜色

*/

public static void createDataChart(XSSFSheet sourceSheet, XSSFSheet distSheet, AnchorPosition anchorPosition,

String chartTitle, String xAxisTitle, String yAxisTitle,

CellRangeAddress xAxisDataRangeAddress, CellRangeAddress yAxisDataRangeAddress,

ChartTypes chartType, PresetColor color) {

if (!(ChartTypes.BAR.equals(chartType) || ChartTypes.LINE.equals(chartType))) return;

//Create a panel to draw

XSSFDrawing drawing = distSheet.createDrawingPatriarch();

XSSFClientAnchor anchor = drawing.createAnchor(0, 0, 0, 0,

anchorPosition.getCol1Index(), anchorPosition.getRow1Index(),

anchorPosition.getCol2Index(), anchorPosition.getRow2Index());

XSSFChart chart = drawing.createChart(anchor);

// Use a category axis for the bottom axis.

XDDFCategoryAxis bottomAxis = chart.createCategoryAxis(AxisPosition.BOTTOM);

chart.setTitleText(chartTitle);

bottomAxis.setTitle(xAxisTitle);

XDDFValueAxis leftAxis = chart.createValueAxis(AxisPosition.LEFT);

leftAxis.setTitle(yAxisTitle);

leftAxis.setCrosses(AxisCrosses.AUTO_ZERO);

leftAxis.setCrossBetween(AxisCrossBetween.BETWEEN);

XDDFCategoryDataSource xs = XDDFDataSourcesFactory.fromStringCellRange(sourceSheet, xAxisDataRangeAddress);

XDDFNumericalDataSource ys1 = XDDFDataSourcesFactory.fromNumericCellRange(sourceSheet, yAxisDataRangeAddress);

XDDFChartData data = chart.createData(chartType, bottomAxis, leftAxis);

XDDFChartData.Series series1 = data.addSeries(xs, ys1);

//line

if (data instanceof XDDFLineChartData) {

XDDFLineChartData.Series series = (XDDFLineChartData.Series)series1;

series.setSmooth(false);

series.setMarkerStyle(MarkerStyle.NONE);

}

//bar

if (data instanceof XDDFBarChartData) {

// in order to transform a bar chart into a column chart, you just need to change the bar direction

XDDFBarChartData bar = (XDDFBarChartData) data;

bar.setBarDirection(BarDirection.COL);

}

//don't delete the following line

series1.setTitle("picture", null);

chart.plot(data);

solidFillSeries(data, 0, color);

}

/**

* 设置图属性

* @param data 待画的数据

* @param index 数据序列编号 从0开始

* @param color 颜色

*/

private static void solidFillSeries(XDDFChartData data, int index, PresetColor color){

XDDFSolidFillProperties fill = new XDDFSolidFillProperties(XDDFColor.from(color));

XDDFChartData.Series series = data.getSeries().get(index);

XDDFShapeProperties properties = series.getShapeProperties();

if (properties == null) properties = new XDDFShapeProperties();

//line

if (data instanceof XDDFLineChartData) {

XDDFLineProperties line = new XDDFLineProperties();

line.setFillProperties(fill);

properties.setLineProperties(line);

}

//bar

if (data instanceof XDDFBarChartData) properties.setFillProperties(fill);

series.setShapeProperties(properties);

}

/**

* 画折线图调用参考

* @throws Exception

*/

private static void createLineChartDemo() throws Exception{

XSSFWorkbook wb = new XSSFWorkbook();

XSSFSheet sheet = wb.createSheet("linechart");

final int NUM_OF_ROWS = 3;

final int NUM_OF_COLUMNS = 10;

Row row;

Cell cell;

for (int rowIndex = 0; rowIndex < NUM_OF_ROWS; rowIndex++) {

row = sheet.createRow((short) rowIndex);

for (int colIndex = 0; colIndex < NUM_OF_COLUMNS; colIndex++) {

cell = row.createCell((short) colIndex);

if (rowIndex == 0) cell.setCellValue(String.valueOf(colIndex * (rowIndex + 1.0)));

else cell.setCellValue(colIndex * (rowIndex + 1.0));

}

}

CellRangeAddress xAxisDataRangeAddress = new CellRangeAddress(0, 0, 0, NUM_OF_COLUMNS - 1);

CellRangeAddress yAxisDataRangeAddress = new CellRangeAddress(1, 1, 0, NUM_OF_COLUMNS - 1);

AnchorPosition anchorPosition = new AnchorPosition(5, 5, 15, 15);

createDataChart(sheet, sheet, anchorPosition, "Picture","Time(min)", "Power(kW)",

xAxisDataRangeAddress, yAxisDataRangeAddress, ChartTypes.LINE, PresetColor.YELLOW);

FileOutputStream fileOut = new FileOutputStream("D:/ooxml-chart.xlsx");

wb.write(fileOut);

}

/**

* 设置画板在sheet中的位置

*/

public static class AnchorPosition {

private int col1Index;//起始位置

private int row1Index;

private int col2Index;//结束位置

private int row2Index;

public AnchorPosition(int col1Index, int row1Index, int col2Index, int row2Index) {

this.col1Index = col1Index;

this.row1Index = row1Index;

this.col2Index = col2Index;

this.row2Index = row2Index;

}

public int getCol1Index() {

return col1Index;

}

public void setCol1Index(int col1Index) {

this.col1Index = col1Index;

}

public int getRow1Index() {

return row1Index;

}

public void setRow1Index(int row1Index) {

this.row1Index = row1Index;

}

public int getCol2Index() {

return col2Index;

}

public void setCol2Index(int col2Index) {

this.col2Index = col2Index;

}

public int getRow2Index() {

return row2Index;

}

public void setRow2Index(int row2Index) {

this.row2Index = row2Index;

}

}

}