com.zaba37.easyreader.actions.menuBar.ImageBackgroundLoader.java Source code

Introduction

Here is the source code for com.zaba37.easyreader.actions.menuBar.ImageBackgroundLoader.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package com.zaba37.easyreader.actions.menuBar;

import com.zaba37.easyreader.Utils;
import com.zaba37.easyreader.controllers.LoadingWindowSceneController;
import com.zaba37.easyreader.models.EasyReaderItem;
import java.awt.Image;
import java.awt.image.RenderedImage;
import java.io.File;
import java.io.IOException;
import java.net.URISyntaxException;
import java.nio.file.Files;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.imageio.ImageIO;
import org.apache.commons.io.FileUtils;
import org.ghost4j.document.DocumentException;
import org.ghost4j.document.PDFDocument;
import org.ghost4j.renderer.RendererException;
import org.ghost4j.renderer.SimpleRenderer;

/**
 *
 * @author zaba37
 */
public class ImageBackgroundLoader extends AsyncTask {

    private boolean loading;
    private final ArrayList<EasyReaderItem> list;
    private int itemCounter;
    private final String tmpDirPath;
    private final LoadingWindowSceneController controller;

    public ImageBackgroundLoader(LoadingWindowSceneController controller) throws IOException {
        this.controller = controller;
        list = new ArrayList();
        loading = true;
        itemCounter = 0;
        tmpDirPath = createTmpDirectory();
    }

    @Override
    public void onPreExecute() {

    }

    @Override
    public void doInBackground() {
        int fileCounter = 0;
        this.setDaemon(true);

        while (loading) {
            if (fileCounter != this.controller.getFilesList().size()) {

                if (isPDFFile(this.controller.getFilesList().get(fileCounter))) {
                    list.addAll(pdfToEasyReaderItems(this.controller.getFilesList().get(fileCounter)));
                } else {
                    list.add(new EasyReaderItem(this.controller.getFilesList().get(fileCounter)));
                    this.publishProgress((Object) null);
                    itemCounter++;
                }

                fileCounter++;
            }

            if (list.size() == this.controller.getFilesNumber()) {
                loading = false;
            }
        }
    }

    @Override
    public void onPostExecute() {
        controller.closeLoadingWindow();

        if (!controller.isCancel()) {
            Utils.getMainWindowController().initImageListView(list);
        } else {
            Utils.getMainWindowController().initImageListView(new ArrayList());
        }
    }

    @Override
    public void progressCallback(Object... params) {
        this.controller.updateProgressIndycator();
    }

    public void setLoading(boolean loading) {
        this.loading = loading;
    }

    private boolean isPDFFile(File file) {
        String extension = "";
        String fileName = file.getName();

        int i = fileName.lastIndexOf('.');

        if (i > 0) {
            extension = fileName.substring(i + 1);
        }

        if ("pdf".equals(extension)) {
            return true;
        } else {
            return false;
        }
    }

    private List<EasyReaderItem> pdfToEasyReaderItems(File pdfFile) {
        PDFDocument document = new PDFDocument();
        ArrayList<EasyReaderItem> pdfItems = new ArrayList();

        try {
            document.load(pdfFile);
        } catch (IOException ex) {
            Logger.getLogger(ImageBackgroundLoader.class.getName()).log(Level.SEVERE, null, ex);
        }

        SimpleRenderer renderer = new SimpleRenderer();
        renderer.setResolution(300);

        List<Image> images = new ArrayList();

        try {
            images.addAll(renderer.render(document));
        } catch (IOException | RendererException | DocumentException ex) {
            Logger.getLogger(ImageBackgroundLoader.class.getName()).log(Level.SEVERE, null, ex);
        }

        for (int i = 0; i < images.size(); i++) {
            try {
                if (loading) {
                    String filePath = tmpDirPath + File.separator + pdfFile.getName() + (i + 1) + ".png";
                    File imageFile = new File(filePath);

                    ImageIO.write((RenderedImage) images.get(i), "png", imageFile);
                    pdfItems.add(new EasyReaderItem(imageFile));

                    this.publishProgress((Object) null);
                    itemCounter++;
                } else {
                    break;
                }
            } catch (IOException ex) {
                Logger.getLogger(ImageBackgroundLoader.class.getName()).log(Level.SEVERE, null, ex);
            }
        }

        return pdfItems;
    }

    private String createTmpDirectory() {
        String OSType = OSValidation();
        String path = "";

        try {
            path = getClass().getProtectionDomain().getCodeSource().getLocation().toURI().getPath();
        } catch (URISyntaxException ex) {
            Logger.getLogger(ImageBackgroundLoader.class.getName()).log(Level.SEVERE, null, ex);
        }

        switch (OSType) {
        case "win":
            return createWindowsDirecotry(path);
        case "mac":
            return createUnixOrMacDirectory(path);
        case "unix":
            return createUnixOrMacDirectory(path);
        default:
            return "";
        }
    }

    private String createWindowsDirecotry(String mainPath) {
        mainPath = mainPath.substring(0, mainPath.lastIndexOf("/"));
        mainPath = mainPath + "/TmpEasyReaderImageDirectory";
        File file = new File(mainPath);

        try {
            if (file.exists()) {
                FileUtils.forceDelete(file);
            }

            FileUtils.forceMkdir(file);

            Files.setAttribute(file.toPath(), "dos:hidden", true);

        } catch (IOException ex) {
            Logger.getLogger(ImageBackgroundLoader.class.getName()).log(Level.SEVERE, null, ex);
        }

        return file.getPath();
    }

    private String createUnixOrMacDirectory(String mainPath) {
        mainPath = mainPath.substring(0, mainPath.lastIndexOf(File.separator));
        mainPath = mainPath + File.separator + ".TmpEasyReaderImageDirectory";
        File file = new File(mainPath);

        try {
            if (file.exists()) {
                FileUtils.forceDelete(file);
            }

            FileUtils.forceMkdir(file);
        } catch (IOException ex) {
            Logger.getLogger(ImageBackgroundLoader.class.getName()).log(Level.SEVERE, null, ex);
        }

        return file.getPath();
    }

    private String OSValidation() {
        String OS = System.getProperty("os.name").toLowerCase();

        if (OS.indexOf("win") >= 0) {
            System.out.println("Windows");
            return "win";
        } else if (OS.indexOf("mac") >= 0) {
            System.out.println("MaxOS");
            return "mac";
        } else {
            System.out.println("Unix");
            return "unix";
        }
    }
}