package com.example.client.utils;
|
|
import javax.swing.*;
|
import javax.swing.plaf.basic.BasicTableUI;
|
import javax.swing.table.JTableHeader;
|
import javax.swing.table.TableColumnModel;
|
import java.awt.*;
|
|
|
/**
|
* 实现表头渲染UI，主要功能时画出表头的单元格
|
* @author 雷锋
|
* @Date 2019年5月27日
|
*/
|
import javax.swing.*;
|
import javax.swing.table.TableCellRenderer;
|
import javax.swing.table.TableColumnModel;
|
import java.awt.*;
|
import java.awt.event.MouseEvent;
|
|
public class ComplexTableUI extends BasicTableUI {
|
private Object[][] headerRows; // 表头数据，用于判断合并逻辑
|
private JTable table;
|
private int singleRowHeight; // 每行的高度
|
private int headerHeight; // 表头的总高度
|
|
public ComplexTableUI(Object[][] headerRows, JTable table) {
|
this.headerRows = headerRows;
|
this.table = table;
|
this.singleRowHeight = table.getRowHeight(); // 获取单行高度
|
this.headerHeight = singleRowHeight * headerRows.length; // 计算表头总高度
|
}
|
|
@Override
|
public void installUI(JComponent c) {
|
super.installUI(c);
|
// 禁用列的拖动，因为合并单元格可能导致布局混乱
|
table.getTableHeader().setReorderingAllowed(false);
|
}
|
|
@Override
|
public void paint(Graphics g, JComponent c) {
|
super.paint(g, c); // 绘制默认的表格内容
|
|
// 遍历表格主体的每一行和每一列
|
for (int row = 0; row < table.getRowCount(); row++) {
|
for (int col = 0; col < table.getColumnCount(); col++) {
|
if (isMergedCell(row, col)) {
|
continue; // 跳过被合并的单元格
|
}
|
|
// 获取单元格的绘制区域
|
Rectangle rect = getCellRect(row, col);
|
// 获取单元格内容
|
Object value = table.getValueAt(row, col);
|
String text = value == null ? "" : value.toString();
|
|
// 绘制单元格
|
paintCell(g, rect, text);
|
}
|
}
|
}
|
|
/**
|
* 判断单元格是否被合并
|
*/
|
private boolean isMergedCell(int row, int col) {
|
// 示例：假设合并逻辑基于表格主体数据
|
Object cellValue = table.getValueAt(row, col);
|
return cellValue == ComplexTable.mergeCellX || cellValue == ComplexTable.mergeCellY;
|
}
|
|
/**
|
* 获取单元格的绘制区域
|
*/
|
private Rectangle getCellRect(int row, int col) {
|
// 获取默认单元格区域
|
Rectangle rect = table.getCellRect(row, col, false);
|
|
// 调整 y 坐标，考虑表头高度
|
rect.y += headerHeight;
|
|
// 如果需要合并单元格，调整区域大小
|
if (isMergedCell(row, col)) {
|
int mergeRowNum = getMergeRowNum(row, col);
|
int mergeColNum = getMergeColNum(row, col);
|
rect.height = getCellHeight(mergeRowNum);
|
rect.width = getCellWidth(col, mergeColNum);
|
}
|
|
return rect;
|
}
|
|
/**
|
* 获取合并的行数
|
*/
|
private int getMergeRowNum(int row, int col) {
|
int mergeRowNum = 1;
|
for (int nextRow = row + 1; nextRow < table.getRowCount(); nextRow++) {
|
Object nextRowCell = table.getValueAt(nextRow, col);
|
if (nextRowCell == ComplexTable.mergeCellY) {
|
mergeRowNum++;
|
} else {
|
break;
|
}
|
}
|
return mergeRowNum;
|
}
|
|
/**
|
* 获取合并的列数
|
*/
|
private int getMergeColNum(int row, int col) {
|
int mergeColNum = 1;
|
for (int nextCol = col + 1; nextCol < table.getColumnCount(); nextCol++) {
|
Object nextCell = table.getValueAt(row, nextCol);
|
if (nextCell == ComplexTable.mergeCellX) {
|
mergeColNum++;
|
} else {
|
break;
|
}
|
}
|
return mergeColNum;
|
}
|
|
/**
|
* 根据合并行数计算单元格高度
|
*/
|
private int getCellHeight(int mergeRowNum) {
|
return mergeRowNum * singleRowHeight;
|
}
|
|
/**
|
* 根据合并列数计算单元格宽度
|
*/
|
private int getCellWidth(int col, int mergeColNum) {
|
TableColumnModel colModel = table.getColumnModel();
|
int width = 0;
|
for (int i = 0; i < mergeColNum; i++) {
|
width += colModel.getColumn(col + i).getWidth();
|
}
|
return width;
|
}
|
|
/**
|
* 绘制单元格
|
*/
|
private void paintCell(Graphics g, Rectangle rect, String text) {
|
Component component = getComponent(text);
|
rendererPane.paintComponent(g, component, table, rect.x, rect.y,
|
rect.width, rect.height, true);
|
}
|
|
/**
|
* 创建用于绘制的组件
|
*/
|
private JLabel getComponent(String text) {
|
JLabel label = new JLabel(text, JLabel.CENTER);
|
Font font = new Font("宋体", Font.PLAIN, 16);
|
label.setFont(font);
|
label.setBorder(UIManager.getBorder("Table.cellBorder"));
|
return label;
|
}
|
}
|
