diff --git a/src/main/java/no/ntnu/idatt1002/demo/controller/IncomeExpenseController.java b/src/main/java/no/ntnu/idatt1002/demo/controller/IncomeExpenseController.java index 0052c5d6274233f43796c2c4fa83c1be88eadb5e..5e5760a068ee7650d75bd7241acd97d01ec5efc3 100644 --- a/src/main/java/no/ntnu/idatt1002/demo/controller/IncomeExpenseController.java +++ b/src/main/java/no/ntnu/idatt1002/demo/controller/IncomeExpenseController.java @@ -209,14 +209,21 @@ public class IncomeExpenseController implements FinanceController { */ @Override public void handleAddBtn(javafx.event.ActionEvent event) { + int sizeBf = (expenseRegister.getItems().size() + incomeRegister.getItems().size()); + if (event.getSource() == addIncome) { handleAddIncome(); } else if (event.getSource() == addExpense){ handleAddExpense(); } - refreshTableView(); - refreshPieCharts(); - refreshProgress(); + + int sizeAf = (expenseRegister.getItems().size() + incomeRegister.getItems().size()); + + if (sizeAf != sizeBf) { + refreshTableView(); + refreshPieCharts(); + refreshProgress(); + } } /** @@ -234,14 +241,14 @@ public class IncomeExpenseController implements FinanceController { handleEditIncome(chosenIncome); } else if (chosenExpense != null) { handleEditExpense(chosenExpense); - } + } else return; } else if (event.getSource() == deleteBtn) { if (chosenIncome != null) { handleDeleteIncome(chosenIncome); } else if (chosenExpense != null) { handleDeleteExpense(chosenExpense); - } - } else return; + } else return; + } //Updates the tableview and pie chart using the register refreshTableView(); refreshPieCharts();