/
StockMarketTableModel.java
79 lines (65 loc) · 2.03 KB
/
StockMarketTableModel.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
package com.jamesshore.finances.ui;
import javax.swing.table.*;
import com.jamesshore.finances.domain.*;
import com.jamesshore.finances.util.*;
public class StockMarketTableModel extends AbstractTableModel {
private static final long serialVersionUID = 1L;
private static final String[] COLUMN_TITLES = { "Year", "Starting Balance", "Cost Basis", "Sales", "Growth", "Ending Balance" };
private static final Class<?>[] COLUMN_CLASSES = { Year.class, SelfRenderable.class, SelfRenderable.class, SelfRenderable.class, SelfRenderable.class, SelfRenderable.class };
private StockMarketProjection projection;
public StockMarketTableModel(StockMarketProjection projection) {
this.projection = projection;
}
public void setProjection(StockMarketProjection projection) {
this.projection = projection;
this.fireTableDataChanged();
}
public StockMarketProjection stockMarketProjection() {
return projection;
}
@Override
public int getColumnCount() {
return COLUMN_TITLES.length;
}
@Override
public String getColumnName(int column) {
return COLUMN_TITLES[column];
}
@Override
public Class<?> getColumnClass(int column) {
return COLUMN_CLASSES[column];
}
@Override
public int getRowCount() {
return projection.numberOfYears();
}
@Override
public Object getValueAt(int rowIndex, int columnIndex) {
StockMarketYear currentYear = projection.getYearOffset(rowIndex);
switch (columnIndex) {
case 0:
return currentYear.year();
case 1:
return currentYear.startingBalance();
case 2:
return currentYear.startingCostBasis();
case 3:
return currentYear.totalSold();
case 4:
return currentYear.growth();
case 5:
return currentYear.endingBalance();
default:
throw new UnreachableCodeException();
}
}
public Dollars startingBalance() {
return projection.getYearOffset(0).startingBalance();
}
public Dollars startingCostBasis() {
return projection.getYearOffset(0).startingCostBasis();
}
public Dollars yearlySpending() {
return projection.getYearOffset(0).totalSellOrders();
}
}