poved/src/data/DataLoader.java @ a353614c

package data;

import java.io.*;
import java.nio.charset.Charset;
import java.nio.file.Files;
import java.util.ArrayList;
import java.util.List;

/**
 *
 * Třída pro základní čtení a parsovani zdrojových souborů
 * - issue #7315
 * @author Daniel Stus, Marek Sobota, Lukas Scurko, Jan Jirman
 * @version 1.0
 */
public class DataLoader {

    //--------------------------------------- ATRIBUTY: -------------------------------------------------//
    //---------------------------------------------------------------------------------------------------//

    /** Jméno načteného souboru */
    private  String fileName;
    /** Hashmapa načtených dat */
    private ArrayList<String[]> loadedData = new ArrayList<>();
    /** Proměnná souboru */
    private File file;
    /** Proměnná root (JDF) složky */
    private File directory;

    //-------------------------------------- KONSTRUKTOR: -----------------------------------------------//
    //---------------------------------------------------------------------------------------------------//

    /**
     * Konstruktor DataLoaderu - Inicializuje slozku se zdrojovymi soubory pro cteni
     * @param directory Slozka se zdrojovymi soubory pro cteni
     */
    public DataLoader(File directory) {
        this.directory = directory;
        this.file = null;
        this.fileName = "";
    }

    //------------------------------------ METODY: Get() ------------------------------------------------//
    //---------------------------------------------------------------------------------------------------//

    /**
     * Vrací pole dat souboru
     * @return pole dat souboru
     */
    public ArrayList<String[]> getLoadedData() {
        return loadedData;
    }

    /**
     * Vraci hodnotu atributu fileName obsahujici jmeno souboru
     *
     * @return jméno souboru zpracovávaného metodami
     */
    public String getFileName() {
        return fileName;
    }

    /**
     * Vrací instanci souboru
     * @return intance souboru
     */
    public File getFile() {
        return file;
    }

    /**
     * Vrací instanci root (JDF) složky
     * @return intance root (JDF) složka
     */
    public File getDirectory(){
        return directory;
    }

    //----------------------------------------- METODY: -------------------------------------------------//
    //---------------------------------------------------------------------------------------------------//

    /**
     * Metoda načte všechna data ze souboru
     * @param fileName jméno souboru, ze kterého se bude číst
     */
    public ArrayList<String[]> loadData(String fileName){
        ArrayList<String[]> dataArrayList;
        String currLine;

        //soubor
        File file = new File(directory.getAbsolutePath()+"/"+fileName);
        if(!file.isFile()){
            System.out.println("Soubor neexistuje..");
            return null;
        }

        //nastavení atributů
        setAttributes(file, fileName);

        //čtení souboru
        try {
            BufferedReader BuffReader = new BufferedReader(new InputStreamReader(new FileInputStream(file), "Cp1250"));
            while((currLine = BuffReader.readLine()) != null) {
                //celý záznam - jedná řádka
                String[] record = parseLine(currLine);
                //přídání do arraylistu loadedData
                loadedData.add(record);
            }
            BuffReader.close();

        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (UnsupportedEncodingException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        return loadedData;
    }

    /**
     * Načte seznam linek z uživatelského soubor a vrátí jej jako pole seznamů
     * kde jeden první seznam je seznam linek
     * a druhý seznam je seznamem pseudo linek ke generování bitmapy
     * pouze podle časových a pevných kódů
     *
     * @param lineFile cesta k uživatelskému souboru se seznamem linek
     */
    public List[] loadLines(File lineFile) {

        List[] lineList = new ArrayList[2];
        List<String> lines = new ArrayList<>();
        List<String> pseudoLines = new ArrayList<>();

        //soubor
        if(!lineFile.isFile()){
            System.out.println("Soubor neexistuje..");
            return null;
        }

        //nastavení atributů
        setAttributes(lineFile, fileName);


        try {
            Files.lines(lineFile.toPath()).forEach(line -> {
                if(line.charAt(2)=='-') pseudoLines.add(line);
                else if (Character.isDigit(line.charAt(0))) lines.add(line);
            });

        } catch (IOException e) {
            e.printStackTrace();
        }

        lineList[0] = lines;
        lineList[1] = pseudoLines;
      return lineList;
    }

    /**,
     * Nastavení zbývajících atributů pro DataLoader
     * @param file instance souboru
     * @param fileName název souborů
     */
    private void setAttributes(File file, String fileName){
        this.file = file;
        this.fileName = fileName;
    }

    /**
     * Zpracuje a vrátí jeden záznam (řádku) jako pole hodnot
     *
     * @param line řádka ke zpracování
     * @return pole se záznamem
     */
    public static String[] parseLine(String line) {
        int startIndex = 1;
        int shift = 2;

        /*
         Odstraní ze záznamu úvodní uvozovku a konečnou uvozovku se středníkem
         příprava pro následující rozdělení záznamu pomocí metody split
         */
        String subString = line.substring(startIndex, line.length()-shift);

        String[] entry = subString.split("\",\"");

        return entry;
    }
}