58 lines
1.8 KiB
Java
58 lines
1.8 KiB
Java
package core.gui;
|
|
|
|
import core.State;
|
|
import core.StateActionTable;
|
|
|
|
import javax.swing.*;
|
|
import java.awt.*;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
public class QTableFrame<A extends Enum> extends JFrame {
|
|
private JLabel stateCountLabel;
|
|
private StateActionTable<A> stateActionTable;
|
|
private List<StateActionRow<A>> rows;
|
|
private JPanel areaWrapper;
|
|
|
|
public QTableFrame(StateActionTable<A> stateActionTable) {
|
|
super("Q-Table");
|
|
this.stateActionTable = stateActionTable;
|
|
rows = new ArrayList<>(10);
|
|
setDefaultCloseOperation(WindowConstants.HIDE_ON_CLOSE);
|
|
setLayout(new BorderLayout());
|
|
setPreferredSize(new Dimension(500, 500));
|
|
stateCountLabel = new JLabel();
|
|
add(BorderLayout.NORTH, stateCountLabel);
|
|
areaWrapper = new JPanel();
|
|
areaWrapper.setLayout(new BoxLayout(areaWrapper, BoxLayout.Y_AXIS));
|
|
for(int i = 0; i < 10; ++i) {
|
|
StateActionRow<A> a = new StateActionRow<>();
|
|
rows.add(a);
|
|
areaWrapper.add(a);
|
|
}
|
|
add(BorderLayout.CENTER, areaWrapper);
|
|
setVisible(false);
|
|
pack();
|
|
}
|
|
|
|
private void refreshAllTextAreas(){
|
|
for(StateActionRow<A> row : rows){
|
|
row.refreshLabels();
|
|
}
|
|
}
|
|
protected void refreshQTable() {
|
|
System.out.println("ref");
|
|
int stateCount = stateActionTable.getStateCount();
|
|
stateCountLabel.setText("Total states: " + stateCount);
|
|
int idx = -1;
|
|
for(Map.Entry<State, Map<A, Double>> entry : stateActionTable.getFirstStateEntriesForView()) {
|
|
if(++idx > rows.size() -1) break;
|
|
StateActionRow<A> row = rows.get(idx);
|
|
row.setState(entry.getKey());
|
|
row.setActionValues(entry.getValue());
|
|
}
|
|
refreshAllTextAreas();
|
|
}
|
|
}
|