要支持单元格图片的复制,你需要在 copyRow
方法中添加处理图片的逻辑。下面是修改后的代码示例:
private void copyRow(Row source, Row target, Workbook workbook) {
// 遍历源行的每个单元格
for (int i = source.getFirstCellNum(); i <= source.getLastCellNum(); i++) {
Cell sourceCell = source.getCell(i);
// 如果源单元格为null,则不创建新单元格
if (sourceCell != null) {
Cell targetCell = target.createCell(i, sourceCell.getCellType());
// 复制单元格样式(如果需要)
CellStyle sourceCellStyle = sourceCell.getCellStyle();
CellStyle targetCellStyle = workbook.createCellStyle();
targetCellStyle.cloneStyleFrom(sourceCellStyle);
targetCell.setCellStyle(targetCellStyle);
switch (sourceCell.getCellTypeEnum()) {
case STRING:
targetCell.setCellValue(sourceCell.getStringCellValue());
break;
case NUMERIC:
targetCell.setCellValue(sourceCell.getNumericCellValue());
break;
case BOOLEAN:
targetCell.setCellValue(sourceCell.getBooleanCellValue());
break;
case FORMULA:
// 如果单元格是公式,则复制公式本身
targetCell.setCellFormula(sourceCell.getCellFormula());
break;
case BLANK:
// do nothing
break;
default:
// 默认处理
}
// 处理图片(如果有)
Drawing<?> drawing = null;
if (workbook instanceof XSSFWorkbook) { // 处理XLSX格式文件
XSSFSheet sourceSheet = (XSSFSheet) source.getRow(i).getSheet();
XSSFDrawing xssfDrawing = (XSSFDrawing) sourceSheet.getDrawingPatriarch();
if (xssfDrawing != null) {
drawing = targetSheet.createDrawingPatriarch();
}
} else if (workbook instanceof HSSFWorkbook) { // 处理XLS格式文件
HSSFSheet sourceSheet = (HSSFSheet) source.getRow(i).getSheet();
HSSFPatriarch hssfPatriarch = sourceSheet.getDrawingPatriarch();
if (hssfPatriarch != null) {
drawing = targetSheet.createDrawingPatriarch();
}
}
if (drawing != null) {
List<XSSFShape> shapes = xssfDrawing.getShapes(); // XLSX 格式
List<HSSFShape> shapes = hssfPatriarch.getChildren(); // XLS 格式
for (XSSFShape shape : shapes) { // 复制 XLSX 格式图片
XSSFPicture picture = (XSSFPicture) shape;
XSSFPictureData pictureData = picture.getPictureData();
int rowStartIndex = sourceCell.getRowIndex();
int colStartIndex = sourceCell.getColumnIndex();
byte[] pictureBytes = pictureData.getData();
int pictureIndex;
if (workbook instanceof XSSFWorkbook) {
pictureIndex = ((XSSFWorkbook) workbook).addPicture(pictureBytes, Workbook.PICTURE_TYPE_PNG);
drawing.createPicture(new XSSFClientAnchor(), pictureIndex);
}
ClientAnchor anchor;
// 设置图片在目标单元格中的位置和大小,可以根据需要进行调整
if (pictureBytes != null && workbook instanceof XSSFWorkbook){
anchor.setDx1(0); // 图片左上角在单元格内的 x 坐标(单位:EMU)
anchor.setDy1(0); // 图片左上角在单元格内的 y 坐标(单位:EMU)
anchor.setDx2(100); // 图片右下角在单元格内的 x 坐标(单位:EMU)
anchor.setDy2(100); // 图片右下角在单元格内的 y 坐标(单位:EMU)
anchor.setRow1(rowStartIndex);
anchor.setCol1(colStartIndex);
Picture picture = drawing.createPicture(anchor, pictureIndex);
picture.resize(); // 调整图片大小
}
}
for (HSSFShape shape : shapes) { // 复制 XLS 格式图片
if (shape instanceof HSSFPicture) {
HSSFPicture picture = (HSSFPicture) shape;
int rowStartIndex = sourceCell.getRowIndex();
int colStartIndex = sourceCell.getColumnIndex();
byte[] pictureBytes = picture.getPictureData().getData();
int pictureIndex;
if (workbook instanceof HSSFWorkbook){
pictureIndex = ((HSSFWorkbook) workbook).addPicture(pictureBytes, Workbook.PICTURE_TYPE_PNG);
ClientAnchor anchor = new HSSFClientAnchor();
anchor.setDx1(0); // 图片左上角在单元格内的 x 坐标(单位:EMU)
anchor.setDy1(0); // 图片左上角在单元格内的 y 坐标(单位:EMU)
anchor.setDx2(100); // 图片右下角在单元格内的 x 坐标(单位:EMU)
anchor.setDy2(100); // 图片右下角在单元格内的 y 坐标(单位:EMU)
anchor.setRow1(rowStartIndex);
anchor.setCol1(colStartIndex);
Picture picture = drawing.createPicture(anchor, pictureIndex);
picture.resize(); // 调整图片大小
}
}
}
}
}
}
}
请注意,上述代码是一个示例,具体的实现方式可能需要根据你的需求进行调整和适配。
内容由零声教学AI助手提供,问题来源于学员提问