补充完整的索引
zhanghai
2 years ago
parent
1dde1a9e3b
commit
b5d3a7a526
@ -0,0 +1,330 @@
|
||||
package com.docus.sw.souyin;
|
||||
|
||||
import com.alibaba.excel.EasyExcel;
|
||||
import com.alibaba.excel.util.FileUtils;
|
||||
import com.docus.sw.Config;
|
||||
import com.docus.sw.fenpan.Document;
|
||||
import com.docus.sw.fenpan.FileTypeEnum;
|
||||
import com.docus.sw.fenpan.Pieces;
|
||||
import com.docus.sw.fenpan.Roll;
|
||||
import com.docus.sw.word.GetPicsDocx;
|
||||
import com.docus.sw.word.PdfBoxUtils;
|
||||
import com.docus.sw.word.ReadImgDoc;
|
||||
import com.google.gson.Gson;
|
||||
import org.apache.commons.imaging.ImageInfo;
|
||||
import org.apache.commons.imaging.ImageReadException;
|
||||
import org.apache.commons.imaging.Imaging;
|
||||
|
||||
import javax.imageio.ImageIO;
|
||||
import java.awt.image.BufferedImage;
|
||||
import java.io.File;
|
||||
import java.io.IOException;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.StandardCopyOption;
|
||||
import java.util.*;
|
||||
|
||||
public class CheckService {
|
||||
|
||||
//读取分盘后的结果。
|
||||
//宗，盘，卷，件
|
||||
//输出目录文件
|
||||
//生成片头，片尾
|
||||
//统计出需要使用的 胶片量
|
||||
|
||||
|
||||
public static void main(String[] args) throws IOException {
|
||||
Config.loadConfig();
|
||||
String path = "C:\\Users\\zhanghai\\Desktop\\桌面\\test\\ppp\\ppp";
|
||||
new CheckService().index(path);
|
||||
}
|
||||
|
||||
public void index(String path) throws IOException {
|
||||
Map<String, IndexPlate> map = new HashMap<>();
|
||||
findAllDir(path, map);
|
||||
|
||||
//计算出宗
|
||||
|
||||
//计算出每个盘的目录。
|
||||
for (IndexPlate indexPlate : map.values()) {
|
||||
String absolutePath = indexPlate.getAbsolutePath();
|
||||
File file = new File(absolutePath);
|
||||
File[] files = file.listFiles();
|
||||
|
||||
List<Roll> rollList = new ArrayList<>();
|
||||
//卷
|
||||
for (File rollFile : files) {
|
||||
if (rollFile.getName().equals("片头")) {
|
||||
//填充卷
|
||||
Roll roll = new Roll(rollFile.getName(), rollFile.getAbsolutePath());
|
||||
|
||||
rollList.add(roll);
|
||||
} else if (rollFile.getName().equals("片尾")) {
|
||||
//填充卷
|
||||
Roll roll = new Roll(rollFile.getName(), rollFile.getAbsolutePath());
|
||||
|
||||
rollList.add(roll);
|
||||
} else {
|
||||
//
|
||||
List<Pieces> pieceList = new ArrayList<>();
|
||||
File[] piecesFile = rollFile.listFiles();
|
||||
for (File piece : piecesFile) {
|
||||
|
||||
//区分word和pdf
|
||||
if (piece.getName().endsWith(".doc")) {
|
||||
Pieces pieces = new Pieces(FileTypeEnum.DOC, piece.getAbsolutePath(), piece.getName());
|
||||
|
||||
String teamName = "tempIndex/" + UUID.randomUUID();
|
||||
File temp = new File(teamName);
|
||||
if (!temp.exists()) {
|
||||
temp.mkdirs();
|
||||
}
|
||||
|
||||
ReadImgDoc.readPicture(piece.getAbsolutePath(), temp.getAbsolutePath());
|
||||
File[] listFiles = temp.listFiles();
|
||||
genPieces(pieces, listFiles, FileTypeEnum.DOC);
|
||||
pieceList.add(pieces);
|
||||
|
||||
//删除对应的temp 文件
|
||||
FileUtils.delete(temp);
|
||||
} else if (piece.getName().endsWith(".docx")) {
|
||||
Pieces pieces = new Pieces(FileTypeEnum.DOCX, piece.getAbsolutePath(), piece.getName());
|
||||
|
||||
String teamName = "tempIndex/" + UUID.randomUUID();
|
||||
File temp = new File(teamName);
|
||||
if (!temp.exists()) {
|
||||
temp.mkdirs();
|
||||
}
|
||||
|
||||
GetPicsDocx.getPics(piece.getAbsolutePath(), temp.getAbsolutePath());
|
||||
File[] listFiles = temp.listFiles();
|
||||
genPieces(pieces, listFiles, FileTypeEnum.DOCX);
|
||||
pieceList.add(pieces);
|
||||
//删除对应的temp 文件
|
||||
FileUtils.delete(temp);
|
||||
} else if (piece.getName().endsWith(".pdf")) {
|
||||
Pieces pieces = new Pieces(FileTypeEnum.DOC, piece.getAbsolutePath(), piece.getName());
|
||||
|
||||
String teamName = "tempIndex/" + UUID.randomUUID();
|
||||
File temp = new File(teamName);
|
||||
if (!temp.exists()) {
|
||||
temp.mkdirs();
|
||||
}
|
||||
|
||||
try {
|
||||
PdfBoxUtils.pdf2image(piece.getAbsolutePath(), temp.getAbsolutePath());
|
||||
File[] listFiles = temp.listFiles();
|
||||
genPieces(pieces, listFiles, FileTypeEnum.DOC);
|
||||
pieceList.add(pieces);
|
||||
|
||||
} catch (IOException e) {
|
||||
|
||||
}
|
||||
//删除对应的temp 文件
|
||||
// temp.delete();
|
||||
FileUtils.delete(temp);
|
||||
} else {
|
||||
Pieces pieces = new Pieces(FileTypeEnum.JPG, piece.getAbsolutePath(), piece.getName());
|
||||
File[] listFiles = piece.listFiles();
|
||||
genPieces(pieces, listFiles, FileTypeEnum.JPG);
|
||||
pieceList.add(pieces);
|
||||
}
|
||||
|
||||
}
|
||||
//填充卷
|
||||
Roll roll = new Roll(rollFile.getName(), rollFile.getAbsolutePath());
|
||||
roll.putAll(pieceList);
|
||||
rollList.add(roll);
|
||||
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
indexPlate.putAll(rollList);
|
||||
|
||||
}
|
||||
|
||||
FileUtils.delete(new File("tempIndex"));
|
||||
|
||||
//生成索引目录
|
||||
for (IndexPlate indexPlate : map.values()) {
|
||||
List<Roll> rollList = indexPlate.getRollList();
|
||||
for(Roll roll:rollList){
|
||||
List<Pieces> piecesList = roll.getPiecesList();
|
||||
for(Pieces pieces:piecesList){
|
||||
List<Document> documentList = pieces.getDocumentList();
|
||||
for(Document document:documentList){
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
private static void genPieces(Pieces pieces, File[] listFiles, FileTypeEnum fileTypeEnum) {
|
||||
List<Document> documentList = new ArrayList<>();
|
||||
|
||||
for (File docfile : listFiles) {
|
||||
//非图片模式，跳过。
|
||||
if (!(docfile.getName().endsWith(".jpg") || docfile.getName().endsWith(".png")
|
||||
|| docfile.getName().endsWith(".jpeg") || docfile.getName().endsWith(".tif")
|
||||
|| docfile.getName().endsWith(".tiff") || docfile.getName().endsWith(".jp2") || docfile.getName().endsWith(".jpm")) || docfile.getName().endsWith(".gif")) {
|
||||
continue;
|
||||
}
|
||||
if (docfile.getName().endsWith(".jp2") || docfile.getName().endsWith(".jpm")) {
|
||||
// 读取 JPEG 2000 图像文件
|
||||
|
||||
try {
|
||||
BufferedImage image = ImageIO.read(docfile);
|
||||
int height = image.getHeight();
|
||||
int width = image.getWidth();
|
||||
Document document = new Document(width, height, 300, docfile.getName());
|
||||
documentList.add(document);
|
||||
} catch (IOException e) {
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
|
||||
} else {
|
||||
try {
|
||||
ImageInfo imageInfo = Imaging.getImageInfo(docfile);
|
||||
int height = imageInfo.getHeight();
|
||||
int width = imageInfo.getWidth();
|
||||
int physicalHeightDpi = imageInfo.getPhysicalHeightDpi();
|
||||
Document document = new Document(width, height, physicalHeightDpi, docfile.getName());
|
||||
documentList.add(document);
|
||||
} catch (IOException e) {
|
||||
throw new RuntimeException("非图片格式", e);
|
||||
} catch (ImageReadException e) {
|
||||
throw new RuntimeException(e);
|
||||
} catch (IllegalArgumentException e) {
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
pieces.put(documentList);
|
||||
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
private List<IndexPageRow> indexGen(IndexPlate indexPlate, List<Roll> rollList, Integer pianPageNum, Integer totalSize) {
|
||||
List<IndexPageRow> indexPageRows = new ArrayList<>();
|
||||
int start = 1;
|
||||
//添加片头
|
||||
for (Roll roll : rollList) {
|
||||
|
||||
//跳过片头
|
||||
if (roll.getName().equals("片头")) {
|
||||
File file = new File(roll.getAbsolutePath());
|
||||
int fileasize = file.listFiles().length;
|
||||
fileasize += pianPageNum;
|
||||
IndexPageRow indexPageRow = new IndexPageRow(indexPlate.getZongName(), indexPlate.getName(), roll.getName(), "", fileasize, start);
|
||||
start += fileasize;
|
||||
indexPageRows.add(indexPageRow);
|
||||
start += 1;
|
||||
}
|
||||
}
|
||||
|
||||
//处理偏中
|
||||
for (Roll roll : rollList) {
|
||||
|
||||
//跳过片头
|
||||
if (roll.getName().equals("片头")) {
|
||||
continue;
|
||||
}
|
||||
//跳过片尾
|
||||
if (roll.getName().equals("片尾")) {
|
||||
continue;
|
||||
}
|
||||
List<Pieces> piecesList = roll.getPiecesList();
|
||||
for (Pieces pieces : piecesList) {
|
||||
IndexPageRow indexPageRow = new IndexPageRow(indexPlate.getZongName(), indexPlate.getName(), roll.getName(), pieces.getName(), pieces.getDocumentList().size(), start);
|
||||
start += pieces.getDocumentList().size();
|
||||
indexPageRows.add(indexPageRow);
|
||||
}
|
||||
start += 1;
|
||||
}
|
||||
|
||||
//添加片尾
|
||||
for (Roll roll : rollList) {
|
||||
|
||||
//跳过片头
|
||||
if (roll.getName().equals("片尾")) {
|
||||
File file = new File(roll.getAbsolutePath());
|
||||
int fileasize = file.listFiles().length;
|
||||
fileasize += pianPageNum;
|
||||
IndexPageRow indexPageRow = new IndexPageRow(indexPlate.getZongName(), indexPlate.getName(), roll.getName(), "", fileasize, start);
|
||||
start += fileasize;
|
||||
indexPageRows.add(indexPageRow);
|
||||
start += 1;
|
||||
}
|
||||
}
|
||||
|
||||
//加上统计
|
||||
IndexPageRow indexPageRow = new IndexPageRow("", "", "", "有效画幅数", totalSize, null);
|
||||
indexPageRows.add(indexPageRow);
|
||||
|
||||
return indexPageRows;
|
||||
}
|
||||
|
||||
|
||||
private void findAllDir(String absolutePath, Map<String, IndexPlate> map) {
|
||||
|
||||
|
||||
File sourceFile = new File(absolutePath);
|
||||
File[] files = sourceFile.listFiles();
|
||||
for (File o : files) {
|
||||
if (o.isDirectory()) {
|
||||
findAllDir(o.getAbsolutePath(), map);
|
||||
}
|
||||
|
||||
if (o.isFile()) {
|
||||
//判断是word,pdf,pic
|
||||
if (o.getName().endsWith(".pdf")) {
|
||||
|
||||
File rofile = o.getParentFile().getParentFile();
|
||||
// new Plate()
|
||||
if (map.get(rofile.getAbsolutePath()) == null) {
|
||||
IndexPlate indexPlate = new IndexPlate(rofile.getAbsolutePath(), rofile.getName(), rofile.getParentFile().getName());
|
||||
map.put(rofile.getAbsolutePath(), indexPlate);
|
||||
}
|
||||
|
||||
} else if (o.getName().endsWith(".docx") || o.getName().endsWith(".doc")) {
|
||||
File rofile = o.getParentFile().getParentFile();
|
||||
// new Plate()
|
||||
if (map.get(rofile.getAbsolutePath()) == null) {
|
||||
IndexPlate indexPlate = new IndexPlate(rofile.getAbsolutePath(), rofile.getName(), rofile.getParentFile().getName());
|
||||
map.put(rofile.getAbsolutePath(), indexPlate);
|
||||
}
|
||||
} else if (o.getName().endsWith(".jpg") || o.getName().endsWith(".png")
|
||||
|| o.getName().endsWith(".jpeg") || o.getName().endsWith(".tif")
|
||||
|| o.getName().endsWith(".tiff")) {
|
||||
|
||||
if (o.getParentFile().getName().equals("片头") || o.getParentFile().getName().equals("片尾")) {
|
||||
continue;
|
||||
}
|
||||
|
||||
|
||||
//件
|
||||
File parentFile = o.getParentFile()
|
||||
//卷
|
||||
.getParentFile()
|
||||
//盘
|
||||
.getParentFile();
|
||||
|
||||
if (map.get(parentFile.getAbsolutePath()) == null) {
|
||||
IndexPlate indexPlate = new IndexPlate(parentFile.getAbsolutePath(), parentFile.getName(), parentFile.getParentFile().getName());
|
||||
map.put(parentFile.getAbsolutePath(), indexPlate);
|
||||
}
|
||||
|
||||
} else {
|
||||
|
||||
System.out.println("请删除无效的文件：" + o.getAbsolutePath());
|
||||
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue