Financial-Calculator/EntryDataModel.cpp at master · jBeller/Financial-Calculator · GitHub

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
80
81
82
83
84
85
86
#include "EntryDataModel.h"

EntryDataModel::EntryDataModel()
{
	m_root = new EntryDataModelNode(NULL, "Entries");

	m_income = new EntryDataModelNode(m_root, "Income");
	m_root->m_insert(m_income);
	m_expense = new EntryDataModelNode(m_root, "Expenses");
	m_root->m_insert(m_expense);
}
void EntryDataModel::GetValue(wxVariant &variant, const wxDataViewItem &item, unsigned int col) const
{
	wxASSERT(item.IsOk());

	EntryDataModelNode* node = (EntryDataModelNode*)item.GetID();
	if (col == 0)
		variant = node->m_name;
	else if (col == 1)
		variant = node->m_value_text;
}
bool EntryDataModel::SetValue(const wxVariant &variant, const wxDataViewItem &item, unsigned int col)
{
	wxASSERT(item.IsOk());

	EntryDataModelNode* node = (EntryDataModelNode*)item.GetID();
	if (col == 0)
	{
		node->m_name = variant.GetString();
		return true;
	}
	else if (col == 1)
	{
		node->m_value_text = variant.GetString();
		return true;
	}
	return false;
}
wxDataViewItem EntryDataModel::GetParent(const wxDataViewItem &item) const
{
	if (!item.IsOk())
		return wxDataViewItem(0);

	EntryDataModelNode* node = (EntryDataModelNode*)item.GetID();
	if (node == m_root)
		return wxDataViewItem(0);
	return wxDataViewItem((void*) node->m_get_parent());
}
bool EntryDataModel::IsContainer(const wxDataViewItem &item) const
{
	if (!item.IsOk())
		return true;

	EntryDataModelNode* node = (EntryDataModelNode*)item.GetID();
	return node->m_is_container();
}
unsigned int EntryDataModel::GetChildren(const wxDataViewItem &parent, wxDataViewItemArray &array) const
{
	EntryDataModelNode* node = (EntryDataModelNode*)parent.GetID();
	if (!node)
	{
		array.Add(wxDataViewItem((void*)m_root));
		return 1;
	}
	if (node->m_get_child_count() == 0)
		return 0;

	unsigned int count = node->m_get_child_count();
	for (unsigned int pos = 0; pos < count; pos++)
	{
		EntryDataModelNode* child = node->m_get_nth_child(pos);
		array.Add(wxDataViewItem((void*)child));
	}

	return count;
}
void EntryDataModel::InsertIncome(wxString &str, wxString &val)
{
	EntryDataModelNode* child_node = new EntryDataModelNode(m_income, str, val);
	m_income->m_insert(child_node);
}
void EntryDataModel::InsertExpense(wxString &str, wxString &val)
{
	EntryDataModelNode* child_node = new EntryDataModelNode(m_expense, str, val);
	m_expense->m_insert(child_node);
}