112 lines
4.1 KiB
Java
112 lines
4.1 KiB
Java
package dev.mateuszkowalczyk.ffm.app;
|
|
|
|
import dev.mateuszkowalczyk.ffm.data.database.photo.Photo;
|
|
import dev.mateuszkowalczyk.ffm.data.database.photo.PhotoDAO;
|
|
import dev.mateuszkowalczyk.ffm.image.FaceDetector;
|
|
import dev.mateuszkowalczyk.ffm.image.ThumbnailService;
|
|
import dev.mateuszkowalczyk.ffm.utils.PropertiesLoader;
|
|
import dev.mateuszkowalczyk.ffm.utils.Property;
|
|
import javafx.application.Platform;
|
|
import javafx.scene.image.Image;
|
|
import javafx.scene.image.ImageView;
|
|
import javafx.scene.image.PixelWriter;
|
|
import javafx.scene.image.WritableImage;
|
|
|
|
import java.awt.image.BufferedImage;
|
|
import java.io.IOException;
|
|
import java.nio.file.Files;
|
|
import java.nio.file.Paths;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
|
|
public class DirectoryScanner implements Runnable {
|
|
private WorkspaceService workspaceService = WorkspaceService.getInstance();
|
|
private PropertiesLoader propertiesLoader = PropertiesLoader.getInstance();
|
|
private FaceDetector faceDetector = FaceDetector.getInstance();
|
|
private ThumbnailService thumbnailService = new ThumbnailService();
|
|
private PhotoDAO photoDAO = PhotoDAO.getInstance();
|
|
|
|
@Override
|
|
public void run() {
|
|
System.out.println("Run photos reader");
|
|
this.readImages();
|
|
}
|
|
|
|
private ArrayList<ImageView> readImages() {
|
|
ArrayList<ImageView> imageViews = new ArrayList<>();
|
|
try {
|
|
Files.walk(Paths.get(this.propertiesLoader.get(Property.PATH_TO_DIRECTORY)))
|
|
.filter(Files::isRegularFile)
|
|
.forEach(path -> {
|
|
String filename = path.getFileName().toString();
|
|
ImageView imageView;
|
|
|
|
if(this.isImagePath(filename) && !path.toString().contains(".cache")) {
|
|
Photo photo;
|
|
|
|
photo = this.photoDAO.findByPath(path.toString());
|
|
|
|
if (photo == null) {
|
|
photo = new Photo();
|
|
|
|
photo.setPath(path.toString());
|
|
photo.setFileName(filename);
|
|
|
|
BufferedImage image = this.thumbnailService.createThumbnail(photo);
|
|
this.photoDAO.add(photo);
|
|
|
|
WritableImage wr = null;
|
|
if (image != null) {
|
|
wr = new WritableImage(image.getWidth(), image.getHeight());
|
|
PixelWriter pw = wr.getPixelWriter();
|
|
for (int x = 0; x < image.getWidth(); x++) {
|
|
for (int y = 0; y < image.getHeight(); y++) {
|
|
pw.setArgb(x, y, image.getRGB(x, y));
|
|
}
|
|
}
|
|
}
|
|
|
|
this.faceDetector.run(photo);
|
|
|
|
imageView = new ImageView(wr);
|
|
imageViews.add(imageView);
|
|
|
|
Platform.runLater(() -> WorkspaceService.getInstance().addImage(imageView));
|
|
}
|
|
}
|
|
});
|
|
} catch (IOException e) {
|
|
e.printStackTrace();
|
|
}
|
|
|
|
return imageViews;
|
|
}
|
|
|
|
private boolean isImagePath(String filename) {
|
|
List<String> whiteListExtension = new ArrayList<String>();
|
|
whiteListExtension.add("jpg");
|
|
whiteListExtension.add("jpeg");
|
|
whiteListExtension.add("png");
|
|
|
|
for (String extension: whiteListExtension) {
|
|
var fileExtension = filename.substring(filename.length() - extension.length());
|
|
|
|
if (fileExtension.toLowerCase().equals(extension)){
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
|
|
// filename.endsWith("jpeg");
|
|
//
|
|
// switch (filename.substring(filename.length() - 3).toLowerCase()) {
|
|
// case "jpeg":
|
|
// case "jpg":
|
|
// return true;
|
|
// default:
|
|
// return false;
|
|
// }
|
|
}
|
|
}
|