|
|
|
|
@ -365,7 +365,7 @@ public class ExcelKit {
|
|
|
|
|
return data;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public static List<List<String>> getChartData(String docPath, int chartNumber, int skipRowCount) throws IOException, InvalidFormatException, DocumentException, InterruptedException {
|
|
|
|
|
public static List<List<String>> readChart(String docPath, int chartNumber, int skipRowCount) throws IOException, InvalidFormatException, DocumentException, InterruptedException {
|
|
|
|
|
return getChartDataByXml(docPath, chartNumber, skipRowCount);
|
|
|
|
|
|
|
|
|
|
// InputStream is = new FileInputStream(docPath);
|
|
|
|
|
@ -548,129 +548,125 @@ public class ExcelKit {
|
|
|
|
|
UnCompress(sourceDoc);
|
|
|
|
|
List<List<String>> matrix = new ArrayList<>();
|
|
|
|
|
String xml = DocxUnzipDirectory + "word\\charts\\chart" + chartNumber + ".xml";
|
|
|
|
|
|
|
|
|
|
if (!(new File(xml).exists())) {
|
|
|
|
|
System.out.println(sourceDoc + " 没有找到第" + chartNumber + "个图表");
|
|
|
|
|
return matrix;
|
|
|
|
|
}
|
|
|
|
|
//3、开始读取
|
|
|
|
|
// 创建 SAXReader 对象，读取 XML 文件
|
|
|
|
|
SAXReader reader = new SAXReader();
|
|
|
|
|
Document document = reader.read(new File(xml));
|
|
|
|
|
// 获取根元素
|
|
|
|
|
Element root = document.getRootElement();
|
|
|
|
|
//折线图
|
|
|
|
|
//将xml用IDEA打开，搜索关键的数据值，然后右键查看XPATH完整路径可以获取到下面的路径
|
|
|
|
|
///c:chartSpace/c:chart/c:plotArea/c:lineChart/c:ser/c:cat/c:numRef/c:numCache/c:pt/c:v
|
|
|
|
|
|
|
|
|
|
//声明一个数组,图表的所有类型
|
|
|
|
|
return readChart(xml);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 将一个小数形式的字符串，转为保留两位小数的字符串
|
|
|
|
|
*
|
|
|
|
|
* @param v
|
|
|
|
|
* @return
|
|
|
|
|
*/
|
|
|
|
|
public static String doubleWith2f(String v) {
|
|
|
|
|
// 保留两位小数
|
|
|
|
|
try {
|
|
|
|
|
double d = Double.parseDouble(v);
|
|
|
|
|
if (d == (int) d) {
|
|
|
|
|
v = String.valueOf((int) d);
|
|
|
|
|
} else {
|
|
|
|
|
v = String.format("%.2f", d);
|
|
|
|
|
}
|
|
|
|
|
} catch (Exception err) {
|
|
|
|
|
//do nothing
|
|
|
|
|
}
|
|
|
|
|
return v;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
/**
|
|
|
|
|
* 从XML中获取图表数据
|
|
|
|
|
*
|
|
|
|
|
* @param xmlPath
|
|
|
|
|
* @return
|
|
|
|
|
* @throws DocumentException
|
|
|
|
|
*/
|
|
|
|
|
public static List<List<String>> readChart(String xmlPath) throws DocumentException {
|
|
|
|
|
//图表类型
|
|
|
|
|
String[] CHART_TYPES = {"lineChart", "barChart"};//折线，柱状
|
|
|
|
|
|
|
|
|
|
// 创建一个一维列表，用于存储转换后的列
|
|
|
|
|
List<List<String>> transposed = new ArrayList<>();
|
|
|
|
|
SAXReader reader = new SAXReader(); // 创建 SAXReader 对象，读取 XML 文件
|
|
|
|
|
Document document = reader.read(new File(xmlPath));
|
|
|
|
|
Element root = document.getRootElement();// 获取根元素
|
|
|
|
|
|
|
|
|
|
//行数据库，不管是X轴的年份，还是Y轴的数值，都按行保存到一个二维列表中
|
|
|
|
|
List<List<String>> tList = new ArrayList<>();
|
|
|
|
|
|
|
|
|
|
List<Object> listObject = new ArrayList<>();
|
|
|
|
|
List<Element> xList = new ArrayList<>();//横坐标，一般是年份
|
|
|
|
|
List<Element> children = root.element("chart").element("plotArea").elements();//工作区
|
|
|
|
|
|
|
|
|
|
boolean hadReadXList = false;//是不是已经读取过横坐标也就是年份的数据了
|
|
|
|
|
for (Element child : children) {
|
|
|
|
|
//将CHART_TYPES数组转换为List
|
|
|
|
|
List<String> CHART_TYPES_LIST = Arrays.asList(CHART_TYPES);
|
|
|
|
|
List<String> CHART_TYPES_LIST = Arrays.asList(CHART_TYPES);//将CHART_TYPES数组转换为List
|
|
|
|
|
if (CHART_TYPES_LIST.contains(child.getName())) {//如果当前遍历到的子元素是折线图或者柱状图
|
|
|
|
|
System.out.println("找到图表类型：" + child.getName());
|
|
|
|
|
//System.out.println("找到图表类型：" + child.getName());
|
|
|
|
|
String type = child.getName();
|
|
|
|
|
//ce:chart element
|
|
|
|
|
Element ce = root.element("chart").element("plotArea").element(type);
|
|
|
|
|
|
|
|
|
|
System.out.println("Count=" + ce.elements("ser").size());
|
|
|
|
|
Set<Element> set=new HashSet<>();
|
|
|
|
|
for (int m = 0; m < ce.elements("ser").size(); m++) {
|
|
|
|
|
Element ser = ce.elements("ser").get(m);
|
|
|
|
|
//System.out.println(ser);
|
|
|
|
|
for (int i = 0; i < ce.elements("ser").size(); i++) {
|
|
|
|
|
Element ser = ce.elements("ser").get(i);
|
|
|
|
|
//cat 标签
|
|
|
|
|
Element cat = ser.element("cat");
|
|
|
|
|
Element numRef = cat.element("numRef");
|
|
|
|
|
Element f = numRef.element("f");
|
|
|
|
|
System.out.println(f);
|
|
|
|
|
System.out.println(f.getText());
|
|
|
|
|
//数据
|
|
|
|
|
Element numCache = numRef.element("numCache");
|
|
|
|
|
xList = numCache.elements("pt");
|
|
|
|
|
System.out.println(xList);
|
|
|
|
|
//多个数据组，xList(横坐标年份数据)一定是一样的，并且它是最全的,只读取一次即可
|
|
|
|
|
if (!hadReadXList) {
|
|
|
|
|
List<Element> xList = numCache.elements("pt");
|
|
|
|
|
List<String> yearList = new ArrayList<>();
|
|
|
|
|
for (Element pt : xList) {
|
|
|
|
|
String v = pt.element("v").getText();
|
|
|
|
|
yearList.add(v);
|
|
|
|
|
}
|
|
|
|
|
//添加到横向的表中
|
|
|
|
|
tList.add(yearList);
|
|
|
|
|
hadReadXList = true;//标识读取过了
|
|
|
|
|
}
|
|
|
|
|
//======================================================
|
|
|
|
|
//val标签
|
|
|
|
|
Element val = ser.element("val");
|
|
|
|
|
numRef = val.element("numRef");
|
|
|
|
|
numCache = numRef.element("numCache");
|
|
|
|
|
List<Element> yList = numCache.elements("pt");
|
|
|
|
|
//不是每个索引号都一定有数据的，所以，先将存在的索引号保存到HashMap中,后面再遍历每个年份的索引号时，有的就读取数值，没有的就写成null
|
|
|
|
|
Map<String, String> _map = new HashMap<>();
|
|
|
|
|
for (Element pt : yList) {
|
|
|
|
|
String idx = pt.attribute("idx").getValue();
|
|
|
|
|
String v = pt.element("v").getText();
|
|
|
|
|
//保留两位小数的字符串
|
|
|
|
|
v = doubleWith2f(v);
|
|
|
|
|
//保存到map中，一会再遍历时就知道哪个有，哪个没有，没有的需要填充null
|
|
|
|
|
_map.put(idx, v);
|
|
|
|
|
}
|
|
|
|
|
//枚举一遍全的xList
|
|
|
|
|
List<String> vList = new ArrayList<>();
|
|
|
|
|
//首行就是年份行，它的size()就是标准宽度
|
|
|
|
|
for (int j = 0; j < tList.getFirst().size(); j++) {
|
|
|
|
|
vList.add(_map.getOrDefault(String.valueOf(j), null));
|
|
|
|
|
}
|
|
|
|
|
tList.add(vList);
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
///c:chartSpace/c:chart/c:plotArea/c:lineChart/c:ser/c:val/c:numRef/c:numCache
|
|
|
|
|
// for (Element ser : root.element("chart").element("plotArea").element(type).elements("ser")) {
|
|
|
|
|
// List<Element> yList = ser.element("val").element("numRef").element("numCache").elements("pt");
|
|
|
|
|
//
|
|
|
|
|
// //记录都有哪些有效数值和索引号
|
|
|
|
|
// Map<Integer, Element> map = new HashMap<>();
|
|
|
|
|
// for (Element e : yList) {
|
|
|
|
|
// map.put(Integer.parseInt(e.attribute("idx").getValue()), e);
|
|
|
|
|
// }
|
|
|
|
|
// List<Object> list = new ArrayList<>();
|
|
|
|
|
// for (int i = 0; i < xList.size(); i++) {
|
|
|
|
|
// list.add(map.getOrDefault(i, null));
|
|
|
|
|
// }
|
|
|
|
|
// listObject.add(list);
|
|
|
|
|
// }
|
|
|
|
|
//
|
|
|
|
|
//
|
|
|
|
|
// //数据是按行读取的
|
|
|
|
|
// for (int i = 0; i < listObject.size(); i++) {
|
|
|
|
|
// List<String> row = new ArrayList<>();
|
|
|
|
|
// List<Element> lo = (List<Element>) listObject.get(i);
|
|
|
|
|
// for (Element e : lo) {
|
|
|
|
|
// if (e == null) {
|
|
|
|
|
// row.add(null);
|
|
|
|
|
// } else {
|
|
|
|
|
// //如果x可以转为小数，那么需要保留两位小数，如果可以转换为整数，那么直接返回整数，如果是字符串，就不变
|
|
|
|
|
// String x = null;
|
|
|
|
|
// try {
|
|
|
|
|
// x = e.element("v").getText();
|
|
|
|
|
// double d = Double.parseDouble(x);
|
|
|
|
|
// if (d == (int) d) {
|
|
|
|
|
// x = String.valueOf((int) d);
|
|
|
|
|
// } else {
|
|
|
|
|
// x = String.format("%.2f", d);
|
|
|
|
|
// }
|
|
|
|
|
// } catch (Exception err) {
|
|
|
|
|
// //do nothing
|
|
|
|
|
// }
|
|
|
|
|
// row.add(x);
|
|
|
|
|
// }
|
|
|
|
|
// }
|
|
|
|
|
// matrix.add(row);
|
|
|
|
|
// }
|
|
|
|
|
//
|
|
|
|
|
// printTable(matrix);
|
|
|
|
|
//
|
|
|
|
|
//
|
|
|
|
|
// //上面生成的数据格式需要行转列,横坐标是年份，纵坐标是数据
|
|
|
|
|
// int rowCount = matrix.size();// 计算行数和列数
|
|
|
|
|
// int colCount = matrix.getFirst().size();
|
|
|
|
|
//
|
|
|
|
|
// //System.out.println("行数：" + rowCount + " " + "列数：" + colCount);
|
|
|
|
|
//
|
|
|
|
|
// // 遍历每一列
|
|
|
|
|
// for (int col = 0; col < colCount; col++) {
|
|
|
|
|
// // 创建一个新的内部列表，用于存储当前列的所有行
|
|
|
|
|
// List<String> column = new ArrayList<>();
|
|
|
|
|
// // 遍历每一行，将当前列的值添加到新的内部列表中
|
|
|
|
|
// for (int row = 0; row < rowCount; row++) {
|
|
|
|
|
// String x = matrix.get(row).get(col);
|
|
|
|
|
//
|
|
|
|
|
// column.add(x);
|
|
|
|
|
// }
|
|
|
|
|
// // 将当前列添加到结果列表中
|
|
|
|
|
// transposed.add(column);
|
|
|
|
|
// }
|
|
|
|
|
//实现行列转置
|
|
|
|
|
// 创建一个一维列表，用于存储转换后的列
|
|
|
|
|
List<List<String>> res = new ArrayList<>();
|
|
|
|
|
//上面生成的数据格式需要行转列,横坐标是年份，纵坐标是数据
|
|
|
|
|
int rowCount = tList.size();// 计算行数和列数
|
|
|
|
|
int colCount = tList.getFirst().size();
|
|
|
|
|
// 遍历每一列
|
|
|
|
|
for (int col = 0; col < colCount; col++) {
|
|
|
|
|
// 创建一个新的内部列表，用于存储当前列的所有行
|
|
|
|
|
List<String> column = new ArrayList<>();
|
|
|
|
|
// 遍历每一行，将当前列的值添加到新的内部列表中
|
|
|
|
|
for (int row = 0; row < rowCount; row++) {
|
|
|
|
|
String x = tList.get(row).get(col);
|
|
|
|
|
column.add(x);
|
|
|
|
|
}
|
|
|
|
|
// 将当前列添加到结果列表中
|
|
|
|
|
res.add(column);
|
|
|
|
|
}
|
|
|
|
|
// // 在每一行的第一列插入年份
|
|
|
|
|
// for (
|
|
|
|
|
// int i = 0; i < transposed.size(); i++) {
|
|
|
|
|
// List<String> column = transposed.get(i);
|
|
|
|
|
// column.addFirst(xList.get(i).element("v").getText()); // 在每行的开始插入序号
|
|
|
|
|
// }
|
|
|
|
|
return transposed;
|
|
|
|
|
return res;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
