JAVA将Excel中的报表导出为图片格式(三)换一种实现 - 丁丁·全栈工程师

时间:2024-02-19 08:43:27

JAVA将Excel中的报表导出为图片格式(三)换一种实现

上一篇介绍了使用Java的Robot机器人实现截图,然后将剪贴板上的数据流生成PNG图片

但是经过博主的不断测试,在完全依赖远程桌面的没有终端显示器的服务器上

使用截图方式是不可行的,因为一旦使用了远程桌面,再断开桌面,系统后台执行的截图程序会全部截到黑屏

所以博主不得已去用另一种通用方式生成PNG

这就是要使用Apache公司的POI工具,对整个excel进行解析

将excel的每一个单元格,数据,格式,边框全都读取出来

其中需要考虑到单元格的合并,行列的隐藏等等复杂的问题

最终使用JAVA 2d将读取到的数据绘制到PNG图片上。

第一张是Excel原样截图

 下面一张是使用JAVA 2D绘制出来的报表:

贴上源代码,其中涉及到使用POI读取单元格背景色,字体前景色等颜色操作,还是比较复杂的。

package com.newflyig.jpg;

/**
 * width:pix=getColumnWidthInPixels*1.15
 * height:pix=getHeightInPoints*96/72
 * 
 * 本示例用来读取Excel报表文件,并力图将报表无差别转化成PNG图片
 * 使用POI读取Excel各项数据
 * 使用JAVA 2D绘制PNG
 * TODO 本示例基本实现了常见Excel的所有样式输出,但Office2007以后的版本添加了条件样式功能,因为POI的API无法读取条件样式,所以无法实现
 * 今后可以通过关键字标注等方法,在Excel中需要加入条件样式的单元格用注解标明,使用JAVA计算得出样式再绘制出来
 */

import java.awt.BasicStroke;
import java.awt.Color;
import java.awt.Font;
import java.awt.FontMetrics;
import java.awt.Graphics2D;
import java.awt.image.BufferedImage;
import java.io.File;
import java.text.DecimalFormat;
import java.util.ArrayList;
import java.util.List;

import javax.imageio.ImageIO;

import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.ss.usermodel.WorkbookFactory;
import org.apache.poi.ss.util.CellRangeAddress;
import org.apache.poi.xssf.usermodel.XSSFFont;

import sun.awt.SunHints;

public class DrawFromExcel {

	public static void main(String[] args) throws Exception {
		// 给定两个初始值,标志出导出区域,两个行列组合的单元格
		int[] fromIndex = { 0, 0 };
		int[] toIndex = { 17, 20 };

		int imageWidth = 0;
		int imageHeight = 0;

		Workbook wb = WorkbookFactory.create(new File("d:/2014年1月营销活动报表140116.xlsx"));
		Sheet sheet = wb.getSheet("test");
		List<CellRangeAddress> rangeAddress = sheet.getMergedRegions(); // 获取整个sheet中合并单元格组合的集合

		// 首先做初步的边界检测,如果指定区域是不合法的则抛出异常
		int rowSum = sheet.getPhysicalNumberOfRows();
		int colSum = sheet.getRow(0).getPhysicalNumberOfCells();
		if (fromIndex[0] > rowSum || fromIndex[0] > toIndex[0] || toIndex[0] > rowSum) {
			throw new Exception("the rowIndex of the area is wrong!");
		}
		if (fromIndex[1] > colSum || fromIndex[1] > toIndex[1] || toIndex[1] > colSum) {
			throw new Exception("the colIndex of the area is wrong!");
		}

		// 计算实际需要载入内存的二维Cell数组的大小,剔除隐藏行列
		int rowSize = toIndex[0]+1;
		int colSize = toIndex[1]+1;

		// 遍历需要扫描的区域
		
		UserCell[][] cells = new UserCell[rowSize][colSize];
		int[] rowPixPos = new int[rowSize + 1];
		rowPixPos[0] = 0;
		int[] colPixPos = new int[colSize + 1];
		colPixPos[0] = 0;
		for (int i = 0; i < rowSize; i++) {

			for (int j = 0; j < colSize; j++) {

				cells[i][j] = new UserCell();
				cells[i][j].setCell(sheet.getRow(i).getCell(j));
				cells[i][j].setRow(i);
				cells[i][j].setCol(j);
				boolean ifShow=(i>=fromIndex[0]) && (j>=fromIndex[1]);	//首先行列要在指定区域之间
				ifShow=ifShow && !(sheet.isColumnHidden(j) || sheet.getRow(i).getZeroHeight());	//其次行列不可以隐藏
				cells[i][j].setShow(ifShow);

				// 计算所求区域宽度
				float widthPix = !ifShow ? 0 : sheet.getColumnWidthInPixels(j); // 如果该单元格是隐藏的,则置宽度为0
				if (i == fromIndex[0]) {
					imageWidth += widthPix;
				}
				
				colPixPos[j+1] = (int) (widthPix * 1.15 + colPixPos[j]);

			}

			// 计算所求区域高度
			boolean ifShow=(i>=fromIndex[0]);	//行序列在指定区域中间
			ifShow=ifShow && !sheet.getRow(i).getZeroHeight();	//行序列不能隐藏
			float heightPoint = !ifShow ? 0 : sheet.getRow(i).getHeightInPoints(); // 如果该单元格是隐藏的,则置高度为0
			imageHeight += heightPoint;			
			rowPixPos[i+1] = (int) (heightPoint * 96 / 72) + rowPixPos[i];

		}

		imageHeight = imageHeight * 96 / 72;
		imageWidth = imageWidth * 115 / 100;

		wb.close();

		List<Grid> grids = new ArrayList<Grid>();
		for (int i = 0; i < rowSize; i++) {
			for (int j = 0; j < colSize; j++) {
				Grid grid = new Grid();
				// 设置坐标和宽高
				grid.setX(colPixPos[j]);
				grid.setY(rowPixPos[i]);
				grid.setWidth(colPixPos[j + 1] - colPixPos[j]);
				grid.setHeight(rowPixPos[i + 1] - rowPixPos[i]);
				grid.setRow(cells[i][j].getRow());
				grid.setCol(cells[i][j].getCol());
				grid.setShow(cells[i][j].isShow());

				// 判断是否为合并单元格
				int[] isInMergedStatus = isInMerged(grid.getRow(), grid.getCol(), rangeAddress);

				if (isInMergedStatus[0] == 0 && isInMergedStatus[1] == 0) {
					// 此单元格是合并单元格,并且不是第一个单元格,需要跳过本次循环,不进行绘制
					continue;
				} else if (isInMergedStatus[0] != -1 && isInMergedStatus[1] != -1) {
					// 此单元格是合并单元格,并且属于第一个单元格,则需要调整网格大小					
					int lastRowPos=isInMergedStatus[0]>rowSize-1?rowSize-1:isInMergedStatus[0];
					int lastColPos=isInMergedStatus[1]>colSize-1?colSize-1:isInMergedStatus[1];					
					
					grid.setWidth(colPixPos[lastColPos + 1] - colPixPos[j]);
					grid.setHeight(rowPixPos[lastRowPos + 1] - rowPixPos[i]);
					
					
				}

				// 单元格背景颜色
				CellStyle cs = cells[i][j].getCell().getCellStyle();
				if (cs.getFillPattern() == CellStyle.SOLID_FOREGROUND)
					grid.setBgColor(cells[i][j].getCell().getCellStyle().getFillForegroundColorColor());

				// 设置字体
				org.apache.poi.ss.usermodel.Font font = wb.getFontAt(cs.getFontIndex());
				grid.setFont(font);

				// 设置字体前景色
				if (font instanceof XSSFFont) {
					XSSFFont xf = (XSSFFont) font;

					grid.setFtColor(xf.getXSSFColor());
				}

				// 设置文本
				String strCell = "";
				switch (cells[i][j].getCell().getCellType()) {
				case HSSFCell.CELL_TYPE_NUMERIC:										
					strCell = String.valueOf(cells[i][j].getCell().getNumericCellValue());
					break;
				case HSSFCell.CELL_TYPE_STRING:
					strCell = cells[i][j].getCell().getStringCellValue();
					break;
				case HSSFCell.CELL_TYPE_BOOLEAN:
					strCell = String.valueOf(cells[i][j].getCell().getBooleanCellValue());
					break;
				case HSSFCell.CELL_TYPE_FORMULA:
					
					try {
						strCell = String.valueOf(cells[i][j].getCell().getNumericCellValue());
					} catch (IllegalStateException e) {
						strCell = String.valueOf(cells[i][j].getCell().getRichStringCellValue());
					}
					break;
				default:
					strCell = "";
				}
				
				if(cells[i][j].getCell().getCellStyle().getDataFormatString().contains("0.00%")){
					try{
						double dbCell=Double.valueOf(strCell);
						strCell=new DecimalFormat("#.00").format(dbCell*100)+"%";
					}catch(NumberFormatException e){}					
				}
								
				grid.setText(strCell.matches("\\w*\\.0") ? strCell.substring(0, strCell.length() - 2) : strCell);

				grids.add(grid);
			}
		}

		BufferedImage image = new BufferedImage(imageWidth, imageHeight, BufferedImage.TYPE_INT_RGB);
		Graphics2D g2d = image.createGraphics();
		// 平滑字体
		g2d.setRenderingHint(SunHints.KEY_ANTIALIASING, SunHints.VALUE_ANTIALIAS_OFF);
		g2d.setRenderingHint(SunHints.KEY_TEXT_ANTIALIASING, SunHints.VALUE_TEXT_ANTIALIAS_DEFAULT);
		g2d.setRenderingHint(SunHints.KEY_STROKE_CONTROL, SunHints.VALUE_STROKE_DEFAULT);
		g2d.setRenderingHint(SunHints.KEY_TEXT_ANTIALIAS_LCD_CONTRAST, 140);
		g2d.setRenderingHint(SunHints.KEY_FRACTIONALMETRICS, SunHints.VALUE_FRACTIONALMETRICS_OFF);
		g2d.setRenderingHint(SunHints.KEY_RENDERING, SunHints.VALUE_RENDER_DEFAULT);

		g2d.setColor(Color.white);
		g2d.fillRect(0, 0, imageWidth, imageHeight);

		// 绘制表格
		for (Grid g : grids) {
			if (!g.isShow())
				continue;

			// 绘制背景色
			g2d.setColor(g.getBgColor() == null ? Color.white : g.getBgColor());
			g2d.fillRect(g.getX(), g.getY(), g.getWidth(), g.getHeight());

			// 绘制边框
			g2d.setColor(Color.black);
			g2d.setStroke(new BasicStroke(1));
			g2d.drawRect(g.getX(), g.getY(), g.getWidth(), g.getHeight());

			// 绘制文字,居中显示
			g2d.setColor(g.getFtColor());
			Font font = g.getFont();
			FontMetrics fm = g2d.getFontMetrics(font);
			int strWidth = fm.stringWidth(g.getText());// 获取将要绘制的文字宽度
			g2d.setFont(font);
			g2d.drawString(g.getText(),
					g.getX() + (g.getWidth() - strWidth) / 2,
					g.getY() + (g.getHeight() - font.getSize()) / 2 + font.getSize());
		}

		g2d.dispose();
		ImageIO.write(image, "png", new File("d:/test.png"));
		
		System.out.println("Output to PNG file Success!");
	}

	/**
	 * 判断Excel中的单元格是否为合并单元格
	 * 
	 * @param row
	 * @param col
	 * @param rangeAddress
	 * @return 如果不是合并单元格返回{-1,-1},如果是合并单元格并且是一个单元格返回{lastRow,lastCol},
	 *         如果是合并单元格并且不是第一个格子返回{0,0}
	 */
	private static int[] isInMerged(int row, int col, List<CellRangeAddress> rangeAddress) {
		int[] isInMergedStatus = { -1, -1 };
		for (CellRangeAddress cra : rangeAddress) {
			if (row == cra.getFirstRow() && col == cra.getFirstColumn()) {
				isInMergedStatus[0] = cra.getLastRow();
				isInMergedStatus[1] = cra.getLastColumn();
				return isInMergedStatus;
			}
			if (row >= cra.getFirstRow() && row <= cra.getLastRow()) {
				if (col >= cra.getFirstColumn() && col <= cra.getLastColumn()) {
					isInMergedStatus[0] = 0;
					isInMergedStatus[1] = 0;
					return isInMergedStatus;
				}
			}
		}
		return isInMergedStatus;
	}
}