TaskUnit.java

import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileWriter;
import java.util.ArrayList;
import java.util.Scanner;
import java.util.StringTokenizer;


public class TaskUnit {

    private ArrayList<TaskSubUnit> subUnits;
    private ArrayList<Integer> schedules;
    private String unitCode;
    private TaskModel parent;

    private String unitTitle;
    private float pretUnitar;
    private String unitateMetric;
    private float cantitate;
    private float ore;
    private float pretTotal;
    private float material;
    private float manopera;
    private float utilaj;
    private float transport;


    public TaskUnit(TaskModel parent) {
        schedules = new ArrayList<Integer>();
        subUnits = new ArrayList<>();
        this.parent = parent;
        this.unitCode = "new";
        initialiseSubUnits();
    }

    public TaskUnit(String projectFileLine, TaskModel parent) throws Exception {   // line = 'unitCode@tablemark1@tablemark2@...'
        schedules = new ArrayList<Integer>();
        processProjectFileLine(projectFileLine);
        this.parent = parent;
        subUnits = new ArrayList<>();
        initialiseSubUnits();
        loadUnit(unitCode);
    }

    public TaskUnit(TaskModel parent, String unitCode){
        schedules = new ArrayList<Integer>();
        this.parent = parent;

        this.unitCode = unitCode;
        subUnits = new ArrayList<>();

        initialiseSubUnits();
        try{
            loadUnit(unitCode);
        }
        catch (FileNotFoundException e){
            this.unitCode = unitCode;
        }
        catch (Exception e) {
            e.printStackTrace();
        }
    }

    public void processProjectFileLine(String line) {

        String[] tokens = line.split(Finals.TOK_D);
        unitCode = tokens[0];

        for (int i = 1; i < tokens.length; i++) {
            schedules.add(Integer.parseInt(tokens[i]));
        }
    }

    public String getProjectFileLine(){
        String res = "" + unitCode + Finals.TOK_D;

        for (Integer i : schedules) {
            res += i + Finals.TOK_D;
        }
        return res;
    }

    public void initialiseSubUnits() {
        for (int i = 0; i < 4; i ++)
        {
            subUnits.add(new TaskSubUnit(i, this));
        }
    }

    public String getHeader(){
//        return Finals.UNIT_INITAL + Finals.TOK_D + unitTitle + Finals.TOK_D + pretUnitar + Finals.TOK_D + unitateMetric
//                + Finals.TOK_D + cantitate + Finals.TOK_D + ore + Finals.TOK_D + pretTotal + Finals.TOK_D
//                + material + Finals.TOK_D + manopera + Finals.TOK_D + utilaj + Finals.TOK_D
//                + transport;
//
        return Finals.UNIT_INITAL + Finals.TOK_D + unitTitle + Finals.TOK_D + unitateMetric
                + Finals.TOK_D + cantitate + Finals.TOK_D + ore + Finals.TOK_D;
    }

    public String[] getTableHeader(){
        String[] res = new String[Finals.LENGTH_OF_UNIT_TABLE];

        res[1] = unitTitle;
        res[2] = unitCode;
        res[3] = "" + (double)Math.round(pretUnitar * 100d) / 100d;
        res[4] = unitateMetric;
        res[5] = "" + cantitate;
        res[6] = "" + (double)Math.round(ore * 100d) / 100d;
        res[7] = "" + (double)Math.round(pretTotal * 100d) / 100d;
        res[8] = "" + (double)Math.round(material * 100d) / 100d;
        res[9] = "" + (double)Math.round(manopera * 100d) / 100d;
        res[10] = "" + (double)Math.round(utilaj * 100d) / 100d;
        res[11] = "" + (double)Math.round(transport * 100d) / 100d;

        return res;
    }

    public void processHeader(String line) throws Exception{

        String[] tokens = line.split(Finals.TOK_D);

        if (tokens.length != Finals.NR_OF_FIELDS_IN_UNIT){
            System.out.println("" + tokens.length);

            for (String token : tokens)
            {
                System.out.println("sor: " + token);
            }

            throw new InvalidUnitHeaderException("incorrect number of fields in header");
        }

        int i = 1;

        unitTitle = tokens[i];
        i++;

//        String nextTok = tokens[i];
//        pretUnitar = Float.parseFloat(nextTok);
//        i++;

        unitateMetric = tokens[i];
        i++;

        String nextTok = tokens[i];
        cantitate = Float.parseFloat(nextTok);
        i++;

        nextTok = tokens[i];
        ore = Float.parseFloat(nextTok);
//        i++;

//        nextTok = tokens[i];
//        pretTotal = Float.parseFloat(nextTok);
//        i++;
//
//        nextTok = tokens[i];
//        material = Float.parseFloat(nextTok);
//        i++;
//
//        nextTok = tokens[i];
//        manopera = Float.parseFloat(nextTok);
//        i++;
//
//        nextTok = tokens[i];
//        utilaj = Float.parseFloat(nextTok);
//        i++;
//
//        nextTok = tokens[i];
//        transport = Float.parseFloat(nextTok);
    }

    public void loadUnit(String unitCode) throws Exception{

        int currentSubUnit = -1;

        Scanner scan = new Scanner(new File(Finals.UNITS_PATH + unitCode + ".txt"));

        if(scan.hasNextLine()) {
            String line = scan.nextLine();

            if(!line.substring(0,1).equals(Finals.UNIT_INITAL)){
                throw new MissingUnitHeaderException("nincs hader");
            }
            else {

                processHeader(line);
            }
        }
        else
        {
            throw new MissingUnitHeaderException("URES FILE");
        }

        while(scan.hasNextLine()){
            String line = scan.nextLine();
            if(line.equals(Finals.UNIT_END_STRING)){
                break;
            }

            if (line.substring(0,1).equals(Finals.SUB_UNIT_INITAL)){
                currentSubUnit ++;
                subUnits.get(currentSubUnit).processHeader(line);
            }
            else{
                if (currentSubUnit == -1){
                    throw new MissingSubUnitException("missing subunit");
                }

                try {
                    subUnits.get(currentSubUnit).loadLine(line);
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        }
        scan.close();
        clculateAll();
    }

    public void saveUnit() throws Exception{

        //FileWriter fw = new FileWriter(Finals.UNITS_PATH + unitCode + "_" + unitTitle + ".txt");
        //ennel az a baj ugye hgoy mi ugy mentenenk el ha ez nem lenne kicommentelve hogy
        //unitCode_unitTitle.txt, viszont a load az arra keres hogy unitCode.txt

        FileWriter fw = new FileWriter(Finals.UNITS_PATH + unitCode + ".txt");


        fw.write(getHeader() + "\n");


        for (int i = 0; i < Finals.NUMBER_OF_SUBUNITS; i++){
            String currentLine = subUnits.get(i).saveLine();

            while(!currentLine.equals(Finals.NO_MORE_ITEMS))
            {
                fw.write(currentLine + "\n");
                currentLine = subUnits.get(i).saveLine();
            }
        }
        fw.write(Finals.UNIT_END_STRING);
        fw.close();
    }

//    public void insertSubUnit(int subUnitIndex){
//
//        if(subUnitIndex == subUnits.size()){
//            subUnits.add(new TaskSubUnit());
//        }
//        else{
//            subUnits.add(subUnitIndex,new TaskSubUnit());
//        }
//    }

    public void deleteSubUnit(int subUnitIndex){
        subUnits.remove(subUnitIndex);
    }

    public void insertRow(int subUnitIndex, int rowIndex){

        subUnits.get(subUnitIndex).insertRow(rowIndex);
    }

    public void deleteRow(int subUnitIndex, int rowIndex){

        subUnits.get(subUnitIndex).deleteRow(rowIndex);
    }

    public String getUnitCode() {
        return unitCode;
    }

//    private void setUnitCode(String unitCode) {
//        this.unitCode = unitCode;
//    }

//.........................setters

    public void setUnitCode(String unitCode) {
        this.unitCode = unitCode;
    }

    public void setUnitTitle(String unitTitle) {
        this.unitTitle = unitTitle;
    }

    public void setUnitateMetric(String unitateMetric) {
        this.unitateMetric = unitateMetric;
    }

    public void setCantitate(float cantitate) {
        this.cantitate = cantitate;

        calculatePretTotal();
        calculateMaterial();
        calculateManopera();
        calculateUtilaj();
        calculateTransport();

        for (TaskSubUnit su : subUnits){
            su.calculateAll();
        }

//        for (TaskSubUnit su : subUnits){
//            for (TaskRow tr : su.getTaskRows())
//        }
    }

    public void setOre(float ore) {
        this.ore = ore;
    }

//.........................calculations
    public void calculatePretUnitar(){
        float res = 0;

        for (TaskSubUnit tsu : subUnits){
            res += tsu.getSumPretTotalUnitar();
        }

        pretUnitar = res;
        calculatePretTotal();
    }

    public void calculatePretTotal() {
        pretTotal = cantitate * pretUnitar;
    }

    public void calculateMaterial(){
        material = subUnits.get(0).getSumPretTotalUnitar() * cantitate;
        parent.calculateMaterial();
    }

    public void calculateManopera(){
        manopera = subUnits.get(1).getSumPretTotalUnitar() * cantitate / 60;
        parent.calculateManopera();
    }

    public void calculateUtilaj(){
        utilaj = subUnits.get(2).getSumPretTotalUnitar() * cantitate;
        parent.calculateUtilaj();
    }

    public void calculateTransport(){
        transport = subUnits.get(3).getSumPretTotalUnitar() * cantitate;
        parent.calculateTransport();
    }

    public void clculateAll() {
        for (TaskSubUnit tsu : subUnits){
            tsu.calculateAll();
        }

        calculatePretUnitar();
    }
//.............................GETTERS

    public String getUnitTitle() {
        return unitTitle;
    }

    public float getPretUnitar() {
        return pretUnitar;
    }

    public String getUnitateMetric() {
        return unitateMetric;
    }

    public float getCantitate() {
        return cantitate;
    }

    public float getOre() {
        return ore;
    }

    public float getPretTotal() {
        return pretTotal;
    }

    public float getMaterial() {
        return material;
    }

    public float getManopera() {
        return manopera;
    }

    public float getUtilaj() {
        return utilaj;
    }

    public float getTransport() {
        return transport;
    }

    public ArrayList<TaskSubUnit> getSubUnits() {
        return subUnits;
    }

    public ArrayList<Integer> getSchedules() {
        return schedules;
    }
}