102 lines
3.0 KiB
Java
102 lines
3.0 KiB
Java
package ocr;
|
|
import java.awt.image.BufferedImage;
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.nio.file.Files;
|
|
import java.nio.file.Path;
|
|
import java.nio.file.Paths;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.stream.Collectors;
|
|
import java.util.stream.Stream;
|
|
|
|
import javax.imageio.ImageIO;
|
|
|
|
import org.apache.pdfbox.pdmodel.PDDocument;
|
|
import org.apache.pdfbox.pdmodel.PDPage;
|
|
import org.apache.pdfbox.rendering.ImageType;
|
|
import org.apache.pdfbox.rendering.PDFRenderer;
|
|
|
|
public class PdfToImage {
|
|
|
|
public List<String> listAllFiles(String directory, String extension) {
|
|
// https://www.mkyong.com/java/java-how-to-list-all-files-in-a-directory/
|
|
List<String> files = new ArrayList<String>();
|
|
try (Stream<Path> walk = Files.walk(Paths.get(directory))) {
|
|
// voir simplification si necessaire
|
|
files = walk.map(x -> x.toString()).filter(f -> f.endsWith(extension)).collect(Collectors.toList());
|
|
} catch (IOException ioe) {
|
|
ioe.printStackTrace();
|
|
}
|
|
return files;
|
|
}
|
|
|
|
public BufferedImage blackWhiteConvert(BufferedImage image) {
|
|
// Convertit une image en image en noir et blanc
|
|
// TODO : voir recursivite
|
|
int width = image.getWidth();
|
|
int height = image.getHeight();
|
|
for (int x = 0; x < width; x++) {
|
|
for (int y = 0; y < height; y++) {
|
|
if (image.getRGB(x, y) < 128) {
|
|
image.setRGB(x, y, 0);
|
|
} else {
|
|
image.setRGB(x, y, 255);
|
|
}
|
|
}
|
|
}
|
|
return image;
|
|
}
|
|
|
|
public boolean isBlackWhite(BufferedImage image) {
|
|
// verifie si une image est en noir et blanc
|
|
// TODO : voir recursivite
|
|
int width = image.getWidth();
|
|
int height = image.getHeight();
|
|
for (int x = 0; x < width; x++) {
|
|
for (int y = 0; y < height; y++) {
|
|
if ((image.getRGB(x, y) != 0) || (image.getRGB(x, y) != 255)) {
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
public ArrayList<BufferedImage> convertPagesToBWJPG(PDDocument document) {
|
|
// convertit chaque page d'un document pdf en image noir et blanc
|
|
// retourne une array liste d'images
|
|
ArrayList<BufferedImage> images = new ArrayList<BufferedImage>();
|
|
PDFRenderer pdfRenderer = new PDFRenderer(document);
|
|
try {
|
|
int pageCounter = 0;
|
|
for (PDPage page : document.getPages()) {
|
|
System.out.println("page.getRotation() : " + page.getRotation());
|
|
System.out.println("pageCounter : " + pageCounter);
|
|
BufferedImage bim = pdfRenderer.renderImageWithDPI(pageCounter++, 300, ImageType.BINARY); // BINARY =
|
|
// noir et
|
|
// blanc
|
|
images.add(bim);
|
|
System.out.println("Ajout n°" + pageCounter);
|
|
}
|
|
// document.close();
|
|
} catch (IOException ioe) {
|
|
ioe.printStackTrace();
|
|
}
|
|
return images;
|
|
}
|
|
|
|
public void saveOnDisk(ArrayList<BufferedImage> images, String originalFileDir) {
|
|
// sauvegarde sur le disque les images
|
|
int pageCounter = 0;
|
|
try {
|
|
for (BufferedImage img : images) {
|
|
ImageIO.write(img, "JPEG", new File(originalFileDir + "img_" + pageCounter++ + ".jpg"));
|
|
}
|
|
} catch (IOException ioe) {
|
|
ioe.printStackTrace();
|
|
}
|
|
}
|
|
|
|
}
|