package com.zh.szf;

import com.badlogic.gdx.utils.ObjectMap;
import com.zh.szf.data.DataFrame;
import com.zh.szf.data.XLSFile;
import com.zh.szf.gui.Button;
import com.zh.szf.gui.DataView;
import com.zh.szf.gui.Signal;
import com.zh.szf.system.Input;
import com.zh.szf.system.Renderer;

import java.util.HashSet;

public class Stage {

    private Button button;
    private DataFrame dataFrame;
    private DataView dataView;

    private double sum;
    private double avg;
    private ObjectMap<String, Integer> counts;

    public Stage() {
        button = new Button(5, 5, 300, 40, "Szöveg");
        dataFrame = new DataFrame(new XLSFile("files/titanic3.xls"));
        dataView = new DataView(5, 60, dataFrame, null, 25);
        dataView.setRowLimits(0, 9);
        HashSet<String> columnFilter = new HashSet<>();
        columnFilter.add("sex");
        columnFilter.add("boat");
        columnFilter.add("fare");
        columnFilter.add("embarked");
        dataView.setColumnFilter(columnFilter);
    }

    public void draw(Renderer renderer) {
        button.draw(renderer);
        dataView.draw(renderer);
    }

    public void update(Input input) {
        button.update(input);
        dataView.update(input);

        if (button.consumeSignal() != Signal.NULL) {
            System.out.println("Button pressed!");
        }
    }

}