soutade/KissCount
1
0
Fork 0
Code Issues Pull Requests Packages Releases 1 Wiki Activity
ebe9ef4a62
KissCount/src/view/grid/GridAccount.cpp

1574 lines
39 KiB
C++
Raw Blame History

/*
Copyright 2010-2011 Grégory Soutadé
This file is part of KissCount.
KissCount is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
KissCount is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with KissCount. If not, see <http://www.gnu.org/licenses/>.
*/
#include "GridAccount.h"
#define SET_ROW_COLOR(row, backcolor, forecolor) for(int i=0; i<NUMBER_COLS_OPS; i++) \
{ \
SetCellBackgroundColour(row, i, backcolor); \
SetCellTextColour(row, i, forecolor); \
}
#define SET_ROW_FONT(row, font) for(int i=0; i<NUMBER_COLS_OPS; i++) \
{ \
SetCellFont(row, i, font); \
}
#define UNESCAPE_CHARS(s) { \
s.Replace(wxT("\\\""), wxT("\""), true); \
s.Replace(wxT("\\\'"), wxT("\'"), true); \
}
BEGIN_EVENT_TABLE(GridAccount, wxGrid)
EVT_GRID_CELL_LEFT_CLICK(GridAccount::OnCellLeftClick )
END_EVENT_TABLE()
enum {GRID_ID};
GridAccount::GridAccount(KissCount* kiss, wxWindow *parent, wxWindowID id,
bool canAddOperation, bool setWeek, bool synchronizeWithDatabase)
: wxGrid(parent, id), _fixCosts(0), _week1(0),
_week2(0), _week3(0), _week4(0), _parent(parent), _kiss(kiss),
_canAddOperation(canAddOperation), _setWeek(setWeek),
_databaseSynchronization(synchronizeWithDatabase), _loadOperations(false),
_curMonth(0), _curYear(0)
{
wxBitmap deleteBitmap(wxT(DELETE_ICON), wxBITMAP_TYPE_PNG);
wxBitmap checkedBitmap(wxT(CHECKED_ICON), wxBITMAP_TYPE_PNG);
DEFAULT_FONT(font);
int i;
User* user = _kiss->GetUser();
std::vector<Account>::iterator accountIt;
std::vector<Category>::iterator categoryIt;
CreateGrid(1, NUMBER_COLS_OPS);
SetColLabelSize(0);
SetRowLabelSize(0);
SetColSize (DESCRIPTION, GetColSize(DESCRIPTION)*3);
SetDefaultCellFont(font);
font.SetWeight(wxFONTWEIGHT_BOLD);
wxString colsName[] = {wxT(""), _("Description"), _("Date"), _("Debit"), _("Credit"), _("Category"), _("Account"), wxT(""), wxT("")};
for(i=0; i<NUMBER_COLS_OPS; i++)
{
SetCellValue(0, i, colsName[i]);
SetCellBackgroundColour(0, i, OWN_CYAN);
SetCellFont(0, i, font);
SetReadOnly(0, i, true);
SetCellAlignment(0, i, wxALIGN_CENTRE, wxALIGN_CENTRE);
}
SetCellRenderer(0, OP_DELETE, new wxGridCellBitmapRenderer(deleteBitmap));
SetCellRenderer(0, CHECKED, new wxGridCellBitmapRenderer(checkedBitmap));
_accounts = new wxString[user->GetAccountsNumber()];
for (i=0,
accountIt = user->_accounts.begin();
accountIt != user->_accounts.end();
accountIt++, i++)
_accounts[i] = accountIt->name;
_categories = new wxString[user->GetCategoriesNumber()] ;
for(i=0, categoryIt = user->_categories.begin();
categoryIt != user->_categories.end();
categoryIt++, i++)
{
_categories[i] = wxGetTranslation(categoryIt->name) ;
}
Connect(id, wxEVT_GRID_CELL_CHANGE, wxGridEventHandler(GridAccount::OnOperationModified), NULL, this);
AutoSizeColumn(TREE, false);
AutoSizeColumn(CATEGORY, false);
AutoSizeColumn(OP_DATE, false);
AutoSizeColumn(ACCOUNT, false);
AutoSizeColumn(OP_DELETE, false);
AutoSizeColumn(CHECKED, false);
}
GridAccount::~GridAccount()
{
delete[] _categories;
delete[] _accounts;
}
wxPen GridAccount::GetColGridLinePen (int col)
{return wxPen(*wxBLACK, 1, wxSOLID);}
wxPen GridAccount::GetRowGridLinePen (int row) {
if (_canAddOperation)
{
if (row == 0 || row == _fixCosts ||
row == _week1 ||
row == _week2 ||
row == _week3 ||
row == _week4)
return wxPen(*wxBLACK, 1, wxSOLID);
}
else
if (row == 0)
return wxPen(*wxBLACK, 1, wxSOLID);
return GetCellBackgroundColour(row, 0);
}
void GridAccount::ResetWeeks()
{
_week1 = _week2 = _week3 = _week4 = 0;
}
void GridAccount::SetWeek(int week, int line) {
switch (week) {
case 1: _week1 = line; break;
case 2: _week2 = line; break;
case 3: _week3 = line; break;
case 4: _week4 = line; break;
}
}
Operation& GridAccount::GetOperation(const wxString& id)
{
std::vector<Operation>::iterator it;
for(it=_operations->begin(); it!=_operations->end(); it++)
if (it->id == id)
return *it;
// I'm aware about the warning, but the method may not fail
}
void GridAccount::UpdateOperation(Operation& op)
{
std::vector<Operation>::iterator it;
int i;
for(i=0; i < (int)_operations->size(); i++)
if ((*_operations)[i].id == op.id)
{
if (_databaseSynchronization)
_kiss->UpdateOperation(op);
(*_operations)[i] = op;
break;
}
}
int GridAccount::GetDisplayedRow(const wxString& id)
{
for(int i=0; i<(int)_displayedOperations.size(); i++)
if (_displayedOperations[i].id == id)
return i;
return -1;
}
void GridAccount::ClearGrid()
{
std::vector<Operation> operations;
LoadOperations(&operations, 0, 0);
}
void GridAccount::LoadOperations(std::vector<Operation>* operations, int month, int year)
{
std::vector<Operation>::iterator it;
User* user = _kiss->GetUser();
int curLine = 0;
Operation NULLop;
NULLop.id = wxT("");
_loadOperations = true;
_operations = operations;
_curMonth = month;
_curYear = year;
_displayedOperations.clear();
_displayedOperations.push_back(NULLop); // Header
_fixCosts = 0;
it = _operations->begin();
if (GetNumberRows () > 1)
DeleteRows(1, GetNumberRows ()-1);
if (_canAddOperation)
{
for (;it != _operations->end() && it->fix_cost; it++)
{
if (it->parent.Length()) continue;
if (_setWeek)
InsertOperationWithWeek(user, *it, ++curLine, true, it->month, it->year);
else
InsertOperation(user, *it, ++curLine, true, it->month, it->year);
}
InsertOperation(user, NULLop, ++curLine, true, month, year);
for (; it != _operations->end(); it++)
{
if (it->parent.Length()) continue;
if (_setWeek)
InsertOperationWithWeek(user, *it, ++curLine, false, it->month, it->year);
else
InsertOperation(user, *it, ++curLine, false, it->month, it->year);
}
InsertOperation(user, NULLop, ++curLine, false, month, year);
}
else
{
for (;it != _operations->end(); it++)
{
if (it->parent.Length()) continue;
InsertOperation(user, *it, ++curLine, it->fix_cost, it->month, it->year);
}
}
AutoSizeColumn(TREE, false);
AutoSizeColumn(CATEGORY, false);
AutoSizeColumn(OP_DATE, false);
AutoSizeColumn(ACCOUNT, false);
AutoSizeColumn(OP_DELETE, false);
AutoSizeColumn(CHECKED, false);
_loadOperations = false;
}
void GridAccount::ComputeWeeks()
{
std::vector<Operation>::iterator it;
int curLine, curWeek, week, i;
if (!_canAddOperation) return;
for (it = _displayedOperations.begin(), curLine=0;
it != _displayedOperations.end();
it++, curLine++)
{
if (it->id.Length() && !it->fix_cost) break;
}
if (it == _displayedOperations.end()) return;
ResetWeeks();
curWeek = wxDateTime(it->day+1, (wxDateTime::Month)it->month, it->year).GetWeekOfMonth();
it++;
for (i=1; it != _displayedOperations.end(); it++, curLine++)
{
if (!it->id.Length() || it->parent.Length()) continue;
week = wxDateTime(it->day+1, (wxDateTime::Month)it->month, it->year).GetWeekOfMonth();
if (week != curWeek)
{
SetWeek(i++, curLine);
curWeek = week;
}
}
}
void GridAccount::InsertOperationWithWeek(User* user, Operation& op, int line, bool fix, int month, int year)
{
InsertOperation(user, op, line, fix, month, year);
if (op.id.Length() && !fix)
ComputeWeeks();
}
void GridAccount::InsertOperation(User* user, Operation& op, int line, bool fix, int month, int year)
{
std::vector<Operation>::iterator it;
std::vector<wxString>::iterator it2;
int r, g, b;
double amount;
wxColour color;
wxDateTime curDate;
wxString description;
wxFont font;
Category cat ;
Operation op2;
curDate.SetToCurrent();
// // First is header
// if (op)
_displayedOperations.insert(_displayedOperations.begin()+line, op);
if (!user->_accounts.size()) return;
InsertRows(line, 1);
if (op.id.Length() && op.meta)
{
SetCellRenderer(line, TREE, new wxGridCellTreeButtonRenderer());
SetCellEditor(line, TREE, new wxGridCellTreeButtonEditor());
SetReadOnly(line, OP_DATE, true);
SetReadOnly(line, CREDIT, true);
SetReadOnly(line, DEBIT, true);
SetReadOnly(line, CATEGORY, true);
SetReadOnly(line, ACCOUNT, true);
}
else
SetReadOnly(line, TREE, true);
SetCellEditor(line, DEBIT, new wxGridCellFloatEditor(wxID_ANY, 2));
SetCellEditor(line, CREDIT, new wxGridCellFloatEditor(wxID_ANY, 2));
wxGridCellChoiceEditor* accountEditor = new wxGridCellChoiceEditor(user->GetAccountsNumber(), _accounts, false);
SetCellEditor(line, ACCOUNT, accountEditor);
wxGridCellChoiceEditor* categoryEditor ;
if (_canAddOperation)
categoryEditor = new wxGridCellChoiceEditor(user->GetCategoriesNumber()-1, _categories+1, false);
else
categoryEditor = new wxGridCellChoiceEditor(user->GetCategoriesNumber(), _categories, false);
SetCellEditor(line, CATEGORY, categoryEditor);
if (fix)
{
SetCellValue(line, CATEGORY, _("Fix"));
SetReadOnly(line, CATEGORY);
_fixCosts++;
}
SetCellEditor(line, DEBIT, new wxGridCellFormulaEditor(op.formula));
SetCellEditor(line, CREDIT, new wxGridCellFormulaEditor(op.formula));
if (op.id.Length())
{
cat = user->GetCategory(op.category);
SetCellEditor(line, OP_DATE, new CalendarEditor(op.day, month, year));
description = op.description;
UNESCAPE_CHARS(description);
SetCellValue(line, DESCRIPTION, description);
SetCellRenderer(line, DESCRIPTION, new wxGridCellTabStringRenderer ());
SetCellValue(line, OP_DATE, wxString::Format(wxT("%02d/%02d/%d"), op.day+1, month+1, year));
if (op.amount < 0)
SetCellValue(line, DEBIT, wxString::Format(wxT("%.2lf"), -op.amount));
else
SetCellValue(line, CREDIT, wxString::Format(wxT("%.2lf"), op.amount));
if (!op.meta)
SetCellValue(line, ACCOUNT, user->GetAccountName(op.account));
if (!fix && !op.meta)
SetCellValue(line, CATEGORY, wxGetTranslation(cat.name));
SetCellRenderer(line, OP_DELETE, new wxGridCellBoolRenderer ());
SetCellEditor(line, OP_DELETE, new wxGridCellBoolEditor ());
SetCellRenderer(line, CHECKED, new wxGridCellBoolRenderer ());
SetCellEditor(line, CHECKED, new wxGridCellFastBoolEditor ());
if (op.category.Length())
color = cat.backcolor;
else
color = OWN_GREEN;
if (op.checked)
{
r = ((color.Red()*1.5) >= 0xFF) ? 0xFF : color.Red()*1.5 ;
g = ((color.Green()*1.5) >= 0xFF) ? 0xFF : color.Green()*1.5 ;
b = ((color.Blue()*1.5) >= 0xFF) ? 0xFF : color.Blue()*1.5 ;
color.Set(r, g, b, color.Alpha());
SetCellValue(line, CHECKED, wxT("1"));
}
SET_ROW_COLOR(line, color, cat.forecolor);
if (op.category.Length() && cat.font.Length())
{
font = user->GetCategoryFont(cat.id);
SET_ROW_FONT(line, font);
}
if (op.meta && !op.amount)
{
amount = _kiss->MetaPositiveAmount(op.id);
SetCellValue(line, DEBIT, wxString::Format(wxT("%.2lf"), amount));
SetCellValue(line, CREDIT, wxString::Format(wxT("%.2lf"), amount));
}
if (!_loadOperations)
{
if (op.meta)
AutoSizeColumn(TREE, false);
AutoSizeColumn(CATEGORY, false);
AutoSizeColumn(ACCOUNT, false);
}
}
else
{
SetCellEditor(line, OP_DATE, new CalendarEditor(0, month, year));
if (curDate.GetMonth() == month &&
curDate.GetYear() == year)
{
if (!fix)
SetCellValue(line, OP_DATE, wxString::Format(wxT("%02d/%02d/%d"), curDate.GetDay(), month+1, year));
SetCellEditor(line, OP_DATE, new CalendarEditor(curDate.GetDay()-1, month, year));
}
else
SetCellEditor(line, OP_DATE, new CalendarEditor(0, month, year));
if (fix)
{
SET_ROW_COLOR(line, OWN_YELLOW, *wxBLACK);
}
else
{
SET_ROW_COLOR(line, OWN_GREEN, *wxBLACK);
}
SetReadOnly(line, CHECKED, true);
SetReadOnly(line, OP_DELETE, true);
}
SetCellAlignment(line, OP_DATE, wxALIGN_CENTRE, wxALIGN_CENTRE);
SetCellAlignment(line, DEBIT, wxALIGN_RIGHT, wxALIGN_CENTRE);
SetCellAlignment(line, CREDIT, wxALIGN_RIGHT, wxALIGN_CENTRE);
SetCellAlignment(line, OP_DELETE, wxALIGN_CENTRE, wxALIGN_CENTRE);
SetCellAlignment(line, CHECKED, wxALIGN_CENTRE, wxALIGN_CENTRE);
AutoSizeRow(line);
Layout();
SetMinSize(GetMinSize());
}
// From http://nomadsync.cvs.sourceforge.net/nomadsync/nomadsync/src/
void GridAccount::OnCellLeftClick(wxGridEvent& evt)
{
if (evt.GetCol() != TREE && evt.GetCol() != OP_DELETE && evt.GetCol() != CHECKED) { evt.Skip() ; return;}
// This forces the cell to go into edit mode directly
//m_waitForSlowClick = TRUE;
SetGridCursor(evt.GetRow(), evt.GetCol());
// Store the click co-ordinates in the editor if possible
// if an editor has created a ClientData area, we presume it's
// a wxPoint and we store the click co-ordinates
wxGridCellEditor* pEditor = GetCellEditor(evt.GetRow(), evt.GetCol());
wxPoint* pClickPoint = (wxPoint*)pEditor->GetClientData();
if (pClickPoint)
{
*pClickPoint = ClientToScreen(evt.GetPosition());
#ifndef __WINDOWS__
EnableCellEditControl(true);
#endif
}
// hack to prevent selection from being lost when click combobox
if (/*evt.GetCol() == 0 && */IsInSelection(evt.GetRow(), evt.GetCol()))
{
//m_selTemp = m_selection;
m_selection = NULL;
}
pEditor->DecRef();
evt.Skip();
}
void GridAccount::DeleteOperation(const Operation& op)
{
std::vector<Operation>::iterator it;
for (it=_operations->begin(); it!=_operations->end(); it++)
if (it->id == op.id)
{
_operations->erase(it);
break;
}
}
void GridAccount::InsertIntoGrid(Operation& op)
{
int i, a, start;
User* user = _kiss->GetUser();
Operation parent;
// No previous fix operations
if (op.fix_cost && !_displayedOperations[1].id.Length())
i = 1;
else
{
if (op.parent.Length())
start = GetDisplayedRow(op.parent);
else
start = 0;
for(i=start; i<(int)_displayedOperations.size(); i++)
{
if (!_displayedOperations[i].id.Length()) continue;
if (_displayedOperations[i].parent.Length()) continue;
if ((_displayedOperations)[i].fix_cost && !op.fix_cost) continue;
if (!(_displayedOperations)[i].fix_cost && op.fix_cost) break;
if (user->_preferences[wxT("operation_order")] == wxT("ASC"))
{
if ((_displayedOperations)[i].day > op.day)
break;
}
else
{
if ((_displayedOperations)[i].day < op.day)
break;
}
}
if (op.parent.Length())
{
parent = GetOperation(op.parent);
if ((i-start) > (int)(parent.childs.size()))
i = start + parent.childs.size();
if (parent.day >= op.day)
i = start + 1;
}
else if (i == (int)_displayedOperations.size() ||
i == _fixCosts)
i--;
else if (!(_displayedOperations)[i].fix_cost && op.fix_cost)
i --;
}
for (a=0; a<(int)_operations->size(); a++)
{
if ((*_operations)[a].fix_cost && !op.fix_cost) continue;
if (!(*_operations)[a].fix_cost && op.fix_cost)
{
a--;
break;
}
if ((*_operations)[a].day > op.day)
{
a--;
break;
}
}
if (a < 0) a = 0;
_operations->insert(_operations->begin()+a, op);
InsertOperationWithWeek(user, (*_operations)[a], i, op.fix_cost, _curMonth, _curYear);
}
int GridAccount::RemoveMeta(Operation op, int line, bool removeRoot, bool deleteOp)
{
std::vector<Operation*>::iterator it, it2;
wxGridCellTreeButtonRenderer* treeRenderer;
int i, deletedOperations = 0;
Operation op2;
treeRenderer = (wxGridCellTreeButtonRenderer*) GetCellRenderer(line, TREE);
for(i=0; i<(int)op.childs.size(); i++)
{
op2 = GetOperation(op.childs[i]);
if (op2.meta)
RemoveMeta(op2, line+1, true, deleteOp);
else
{
if (treeRenderer->IsCollapsed())
{
DeleteRows(line+1, 1);
deletedOperations++;
if (op2.fix_cost) _fixCosts--;
_displayedOperations.erase(_displayedOperations.begin()+line+1);
}
if (deleteOp)
{
DeleteOperation(op2);
if (_databaseSynchronization)
_kiss->DeleteOperation(op2);
}
}
}
op.childs.clear();
if (removeRoot)
{
DeleteRows(line, 1);
_displayedOperations.erase(_displayedOperations.begin()+line);
if (op.fix_cost) _fixCosts--;
if (deleteOp)
{
DeleteOperation(op);
if (_databaseSynchronization)
_kiss->DeleteOperation(op);
}
deletedOperations++;
}
treeRenderer->DecRef();
return deletedOperations;
}
void GridAccount::CheckMeta(Operation& op, int line, bool check)
{
std::vector<Operation*>::iterator it, it2;
wxGridCellTreeButtonRenderer* treeRenderer;
int i;
Operation op2;
wxColour color ;
unsigned char r, g, b;
User* user = _kiss->GetUser();
treeRenderer = (wxGridCellTreeButtonRenderer*) GetCellRenderer(line, TREE);
for(i=0; i<(int)op.childs.size(); i++)
{
op2 = GetOperation(op.childs[i]);
op2.checked = check;
UpdateOperation(op2);
if (op2.meta)
CheckMeta(op2, line+1, check);
if (treeRenderer->IsCollapsed())
{
SetCellValue(line+i+1, CHECKED, check ? wxT("1") : wxT("0"));
color = user->GetCategory(op2.category).backcolor;
if (check)
{
r = ((color.Red()*1.5) >= 0xFF) ? 0xFF : color.Red()*1.5 ;
g = ((color.Green()*1.5) >= 0xFF) ? 0xFF : color.Green()*1.5 ;
b = ((color.Blue()*1.5) >= 0xFF) ? 0xFF : color.Blue()*1.5 ;
color.Set(r, g, b, color.Alpha());
}
SET_ROW_COLOR(line+i+1, color, user->GetCategory(op2.category).forecolor);
}
}
treeRenderer->DecRef();
}
void GridAccount::OnOperationModified(wxGridEvent& event)
{
User* user = _kiss->GetUser();
int row = event.GetRow();
int col = event.GetCol();
Operation new_op, cur_op, op_tmp, op_tmp2;
int op_complete = 6, i, last_day;
wxString value ;
wxDateTime date;
bool need_insertion = false, fix_op=false;
static bool inModification = false ;
wxColour color ;
unsigned char r, g, b;
wxGridCellTreeButtonRenderer* treeRenderer;
std::vector<wxString>::iterator it;
Operation op, op2;
double amount;
wxFont font;
Category cat ;
// Avoid recursives calls
if (inModification || _loadOperations) return;
inModification = true ;
if (col == TREE)
{
treeRenderer = (wxGridCellTreeButtonRenderer*) GetCellRenderer(row, col);
op = _displayedOperations[row];
if (!treeRenderer->IsCollapsed())
{
for (i=1, it=op.childs.begin(); it!=op.childs.end(); it++, i++)
{
op2 = GetOperation(*it);
InsertOperationWithWeek(user, op2, row+i, op2.fix_cost, op2.month, op2.year);
}
}
else
{
RemoveMeta(op, row, false, false);
}
treeRenderer->DecRef();
ComputeWeeks();
inModification = false;
_parent->Fit();
return;
}
if (col == DEBIT)
SetCellValue(row, CREDIT, wxT(""));
else if (col == CREDIT)
SetCellValue(row, DEBIT, wxT(""));
value = GetCellValue(row, DESCRIPTION);
if (value.Length())
{
new_op.description = value;
op_complete--;
}
value = GetCellValue(row, OP_DATE);
if (value.Length())
{
date.ParseFormat(value, wxT("%d/%m/%Y"));
new_op.day = date.GetDay()-1;
new_op.month = date.GetMonth();
new_op.year = date.GetYear();
op_complete--;
}
if (col == DESCRIPTION &&
(!GetCellValue(row, CATEGORY).Length() ||
!GetCellValue(row, ACCOUNT).Length()))
{
new_op.fix_cost = (row <= _fixCosts);
if (_kiss->SearchPreviousOperation(&op_tmp, new_op, _curMonth, _curYear, _canAddOperation))
{
if (!GetCellValue(row, CATEGORY).Length())
{
new_op.category = op_tmp.category;
SetCellValue(row, CATEGORY, wxGetTranslation(user->GetCategoryName(new_op.category)));
op_complete--;
}
if (!GetCellValue(row, ACCOUNT).Length())
{
new_op.account = op_tmp.account;
SetCellValue(row, ACCOUNT, user->GetAccountName(new_op.account));
op_complete--;
}
col = CATEGORY;
new_op.fix_cost = (new_op.category == user->GetCategoryId(wxT("Fix")));
}
}
value = GetCellValue(row, DEBIT);
if (value.Length())
{
value.ToDouble(&new_op.amount);
if (new_op.amount < 0)
{
new_op.amount *= -1.0;
SetCellValue(row, DEBIT, wxString::Format(wxT("%.2lf"), new_op.amount));
}
if (new_op.amount != 0.0) new_op.amount *= -1.0;
op_complete--;
wxGridCellFormulaEditor* pEditor = (wxGridCellFormulaEditor*) GetCellEditor(row, DEBIT);
new_op.formula = pEditor->GetFormula();
pEditor->DecRef();
}
value = GetCellValue(row, CREDIT);
if (value.Length())
{
value.ToDouble(&new_op.amount);
if (new_op.amount < 0)
{
new_op.amount *= -1.0;
SetCellValue(row, CREDIT, wxString::Format(wxT("%.2lf"), new_op.amount));
}
op_complete--;
wxGridCellFormulaEditor* pEditor = (wxGridCellFormulaEditor*) GetCellEditor(row, CREDIT);
new_op.formula = pEditor->GetFormula();
pEditor->DecRef();
}
value = GetCellValue(row, CATEGORY);
if (value.Length())
{
new_op.category = user->GetCategoryId(value);
op_complete--;
}
value = GetCellValue(row, ACCOUNT);
if (value.Length())
{
new_op.account = user->GetAccountId(value);
op_complete--;
}
value = GetCellValue(row, CHECKED);
if (value.Length() && value != wxT("0"))
new_op.checked = true;
else
new_op.checked = false;
op_complete--;
cur_op = (_displayedOperations)[row] ;
if (col == CHECKED || col == CATEGORY)
{
color = user->GetCategory(new_op.category).backcolor;
if (new_op.checked)
{
r = ((color.Red()*1.5) >= 0xFF) ? 0xFF : color.Red()*1.5 ;
g = ((color.Green()*1.5) >= 0xFF) ? 0xFF : color.Green()*1.5 ;
b = ((color.Blue()*1.5) >= 0xFF) ? 0xFF : color.Blue()*1.5 ;
color.Set(r, g, b, color.Alpha());
}
SET_ROW_COLOR(row, color, user->GetCategory(new_op.category).forecolor);
SET_ROW_FONT(row, user->GetCategoryFont(new_op.category));
if (col == CHECKED)
{
cur_op.checked = new_op.checked;
UpdateOperation(cur_op);
inModification = false;
if (cur_op.meta)
CheckMeta(cur_op, row, cur_op.checked);
else
{
if (cur_op.parent.Length())
{
op2 = GetOperation(cur_op.parent);
UpdateMeta(op2);
int row2 = GetDisplayedRow(op2.id);
SetCellValue(row2, CHECKED, op2.checked ? wxT("1") : wxT("0"));
color = user->GetCategory(op2.category).backcolor;
if (op2.checked)
{
r = ((color.Red()*1.5) >= 0xFF) ? 0xFF : color.Red()*1.5 ;
g = ((color.Green()*1.5) >= 0xFF) ? 0xFF : color.Green()*1.5 ;
b = ((color.Blue()*1.5) >= 0xFF) ? 0xFF : color.Blue()*1.5 ;
color.Set(r, g, b, color.Alpha());
}
SET_ROW_COLOR(row2, color, user->GetCategory(op2.category).forecolor);
}
}
event.Skip();
return;
}
}
if (col == OP_DELETE)
{
wxMessageDialog dialog(this, _("Are you sure want to delete : \n")+new_op.description, wxT("KissCount"), wxYES_NO);
if (dialog.ShowModal() == wxID_NO)
{
SetCellValue(row, col, wxT("0"));
inModification = false;
return;
}
}
// Modify a fix operation
if (row < _fixCosts || !_canAddOperation)
{
if (col == OP_DELETE)
{
if (cur_op.parent.Length())
user->UnGroup(_displayedOperations[row]);
if (cur_op.meta)
RemoveMeta(_displayedOperations[row], row, true, true);
else
{
if (cur_op.parent.Length())
{
op_tmp = GetOperation(cur_op.parent);
for (int a=0; a<(int)op_tmp.childs.size(); a++)
if (op_tmp.childs[a] == op.id)
{
op2.childs.erase(op_tmp.childs.begin()+a);
break;
}
}
DeleteRows(row, 1);
DeleteOperation(cur_op);
if (_databaseSynchronization)
_kiss->DeleteOperation(cur_op);
_displayedOperations.erase(_displayedOperations.begin()+row);
if (cur_op.parent.Length() && op_tmp.childs.size() < 2)
{
if (op_tmp.childs.size() == 1)
{
op_tmp2 = GetOperation(op_tmp.childs[0]);
op_tmp2.parent = wxT("");
UpdateOperation(op_tmp2);
row = GetDisplayedRow(op_tmp2.id);
_displayedOperations[row] = op_tmp2;
}
row = GetDisplayedRow(cur_op.parent);
DeleteRows(row, 1);
DeleteOperation(op_tmp);
if (_databaseSynchronization)
_kiss->DeleteOperation(op_tmp);
_displayedOperations.erase(_displayedOperations.begin()+row);
_fixCosts--;
}
_fixCosts--;
ComputeWeeks();
}
inModification = false ;
event.Skip();
return ;
}
new_op.id = cur_op.id;
new_op.fix_cost = true;
new_op.transfert = cur_op.transfert;
new_op.meta = cur_op.meta;
new_op.parent = cur_op.parent;
new_op.childs = cur_op.childs;
new_op._virtual = cur_op._virtual;
if (cur_op.day != new_op.day)
{
need_insertion = true;
DeleteRows(row, 1);
DeleteOperation(cur_op);
_displayedOperations.erase(_displayedOperations.begin()+row);
_fixCosts--;
}
else
{
UpdateOperation(new_op);
(_displayedOperations)[row] = new_op;
}
fix_op = true;
}
// Add a fixCost
else if (row == _fixCosts)
{
if (op_complete) {
inModification = false ;
return ;
}
need_insertion = true;
fix_op = true;
new_op.fix_cost = true;
new_op.meta = false;
new_op._virtual = false;
for(i=0; i<NUMBER_COLS_OPS; i++)
{
if (i == CATEGORY) continue;
SetCellValue(row, i, wxT(""));
}
DEFAULT_FONT(font);
SET_ROW_COLOR(row, OWN_YELLOW, *wxBLACK);
SET_ROW_FONT(row, font);
new_op.id = _kiss->AddOperation(new_op);
}
// Modify an operation
else if (row < (int)(_displayedOperations.size()-1))
{
new_op.id = cur_op.id;
new_op.fix_cost = false;
new_op.transfert = cur_op.transfert;
new_op.meta = cur_op.meta;
new_op.parent = cur_op.parent;
new_op.childs = cur_op.childs;
new_op._virtual = cur_op._virtual;
if (col == OP_DELETE)
{
if (cur_op.parent.Length())
user->UnGroup(_displayedOperations[row]);
if (cur_op.meta)
RemoveMeta(_displayedOperations[row], row, true, true);
else
{
if (cur_op.parent.Length())
{
op_tmp = GetOperation(cur_op.parent);
for (int a=0; a<(int)op_tmp.childs.size(); a++)
if (op_tmp.childs[a] == op.id)
{
op2.childs.erase(op_tmp.childs.begin()+a);
break;
}
}
DeleteRows(row, 1);
DeleteOperation(cur_op);
_displayedOperations.erase(_displayedOperations.begin()+row);
if (_databaseSynchronization)
_kiss->DeleteOperation(cur_op);
if (cur_op.parent.Length() && op_tmp.childs.size() <= 1)
{
if (op_tmp.childs.size() == 1)
{
op_tmp2 = GetOperation(op_tmp.childs[0]);
op_tmp2.parent = wxT("");
UpdateOperation(op_tmp2);
row = GetDisplayedRow(op_tmp2.id);
_displayedOperations[row] = op_tmp2;
}
row = GetDisplayedRow(cur_op.parent);
DeleteRows(row, 1);
DeleteOperation(op_tmp);
if (_databaseSynchronization)
_kiss->DeleteOperation(op_tmp);
_displayedOperations.erase(_displayedOperations.begin()+row);
}
ComputeWeeks();
}
inModification = false ;
event.Skip();
return ;
}
if (cur_op.day != new_op.day)
{
need_insertion = true;
DeleteRows(row, 1);
DeleteOperation(cur_op);
_displayedOperations.erase(_displayedOperations.begin()+row);
}
else
{
UpdateOperation(new_op);
(_displayedOperations)[row] = new_op;
}
}
// Add an operation
else
{
if (op_complete) {
inModification = false ;
return ;
}
need_insertion = true;
fix_op = false;
new_op.fix_cost = false;
new_op.meta = false;
new_op._virtual = false;
for(i=0; i<NUMBER_COLS_OPS; i++)
{
SetCellValue(row, i, wxT(""));
}
DEFAULT_FONT(font);
SET_ROW_COLOR(row, OWN_GREEN, *wxBLACK);
SET_ROW_FONT(row, font);
new_op.id = _kiss->AddOperation(new_op);
}
if (user->GetAccount(new_op.account).blocked && new_op.amount < 0)
wxMessageBox(_("You made a debit on a blocked account"), _("Warning"), wxICON_WARNING | wxOK);
if (need_insertion)
{
InsertIntoGrid(new_op);
UpdateOperation(new_op);
}
if (new_op.parent.Length())
{
row = GetDisplayedRow(new_op.parent);
new_op = _displayedOperations[row];
last_day = new_op.day;
UpdateMeta(new_op);
_displayedOperations[row] = new_op;
cat = user->GetCategory(new_op.category);
if (new_op.category.Length())
color = cat.backcolor;
else
color = OWN_GREEN;
if (new_op.checked)
{
r = ((color.Red()*1.5) >= 0xFF) ? 0xFF : color.Red()*1.5 ;
g = ((color.Green()*1.5) >= 0xFF) ? 0xFF : color.Green()*1.5 ;
b = ((color.Blue()*1.5) >= 0xFF) ? 0xFF : color.Blue()*1.5 ;
color.Set(r, g, b, color.Alpha());
SetCellValue(row, CHECKED, wxT("1"));
}
else
SetCellValue(row, CHECKED, wxT("0"));
SET_ROW_COLOR(row, color, cat.forecolor);
if (new_op.category.Length() && cat.font.Length())
{
font = user->GetCategoryFont(cat.id);
SET_ROW_FONT(row, font);
}
SetCellValue(row, OP_DATE, wxString::Format(wxT("%02d/%02d/%d"), new_op.day+1, _curMonth+1, _curYear));
if (!_displayedOperations[row].amount)
{
amount = _kiss->MetaPositiveAmount(new_op.id);
SetCellValue(row, DEBIT, wxString::Format(wxT("%.2lf"), amount));
SetCellValue(row, CREDIT, wxString::Format(wxT("%.2lf"), amount));
}
else
{
if (_displayedOperations[row].amount < 0)
{
SetCellValue(row, DEBIT, wxString::Format(wxT("%.2lf"), -new_op.amount));
SetCellValue(row, CREDIT, wxT(""));
}
else
{
SetCellValue(row, DEBIT, wxT(""));
SetCellValue(row, CREDIT, wxString::Format(wxT("%.2lf"), new_op.amount));
}
}
// Move updated meta
if ((int)new_op.day != last_day)
{
int i;
RemoveMeta(new_op, row, true, false);
InsertIntoGrid(new_op);
row = GetDisplayedRow(new_op.id);
for (i=1, it=new_op.childs.begin(); it!=new_op.childs.end(); it++, i++)
{
op2 = GetOperation(*it);
InsertOperationWithWeek(user, op2, row+i, op2.fix_cost, _curMonth, _curYear);
}
treeRenderer = (wxGridCellTreeButtonRenderer*) GetCellRenderer(row, 0);
treeRenderer->Invert();
treeRenderer->DecRef();
}
}
inModification = false ;
event.Skip();
}
void GridAccount::UpdateMeta(Operation& meta)
{
std::vector<wxString>::iterator it;
Operation op ;
wxString category = wxT("");
bool updateCat = false ;
if (!meta.childs.size()) return ;
meta.category = wxT("");
meta.checked = true;
meta.amount = 0;
for(it=meta.childs.begin(); it!=meta.childs.end(); it++)
{
op = GetOperation(*it);
if (op.year <= meta.year && op.month <= meta.month && op.day < meta.day)
{
meta.year = op.year;
meta.month = op.month;
meta.day = op.day;
}
meta.checked &= op.checked;
if (!meta.description.Length() && op.description.Length())
meta.description = op.description;
if (!category.Length())
{
if (op.category.Length())
{
category = op.category;
updateCat = true;
}
}
else
{
if (op.category.Length() && op.category != category)
updateCat = false;
}
op.parent = meta.id;
}
if (updateCat)
meta.category = category;
meta.amount = _kiss->MetaAmount(meta.id);
UpdateOperation(meta);
}
void GridAccount::GetSelectedOperations(std::vector<int>* rows)
{
std::vector<int>::iterator it;
rows->clear();
// Blocks. We always expect btl and bbr to have the same size, since their
// entries are supposed to correspond.
const wxGridCellCoordsArray& btl(GetSelectionBlockTopLeft());
const wxGridCellCoordsArray& bbr(GetSelectionBlockBottomRight());
size_t blockCount = btl.size();
if (blockCount == bbr.size())
{
for (size_t i = 0; i < blockCount; ++i)
{
const wxGridCellCoords& tl = btl[i];
const wxGridCellCoords& br = bbr[i];
for (int row = tl.GetRow(); row <= br.GetRow(); ++row)
{
for (it=rows->begin(); it!=rows->end(); it++)
if (*it == row)
break;
if (it != rows->end() || !row) continue;
rows->push_back(row);
}
}
}
// Singly selected cells.
const wxGridCellCoordsArray& cells(GetSelectedCells());
for (size_t i = 0; i < cells.size(); ++i)
{
const wxGridCellCoords& c = cells[i];
for (it=rows->begin(); it!=rows->end(); it++)
if (*it == c.GetRow())
break;
if (it != rows->end() || !c.GetRow()) continue;
rows->push_back(c.GetRow());
}
}
void GridAccount::Group()
{
std::vector<int> selected, rows;
std::vector<int>::iterator it;
std::vector<Operation> ops;
std::vector<Operation>::iterator it2;
std::vector<wxString>::iterator it3;
wxString parent = wxT("");
Operation op, op2;
int fix = -1, i, a, row;
GetSelectedOperations(&selected);
for (size_t i = 0; i < selected.size(); ++i)
{
op = _displayedOperations[selected[i]] ;
if (op.id.Length())
{
if (!parent.Length())
{
if (op.parent.Length())
{
parent = op.parent;
continue;
}
else if(op.meta)
parent = op.id;
}
else
{
if ((parent.Length() && op.parent.Length() && op.parent != parent))
{
wxMessageBox(_("Cannot group these operations"), _("Error"), wxICON_ERROR | wxOK);
return ;
}
}
if (fix != -1 && ((!fix && op.fix_cost) || (fix && !op.fix_cost)))
{
wxMessageBox(_("Cannot group these operations"), _("Error"), wxICON_ERROR | wxOK);
return ;
}
if (fix == -1)
fix = op.fix_cost ? 1 : 0;
ops.push_back(op);
rows.push_back(selected[i]);
}
}
if (!ops.size()) return;
if (!parent.Length())
{
if (rows.size() < 2) return;
op.parent = wxT("");
op.day = ops[0].day;
op.month = ops[0].month;
op.year = ops[0].year;
op.amount = 0;
op.description = wxT("");
op.category = wxT("");
op.fix_cost = fix;
op.account = wxT("");
op.checked = false;
op.transfert = wxT("");
op.formula = wxT("");
op.meta = true;
op.childs.clear();
op.id = _kiss->AddOperation(op);
}
else
{
if (rows.size() < 1) return;
row = GetDisplayedRow(parent);
op = _displayedOperations[row];
//if (op.id.Length()) return;
}
std::sort(rows.begin(), rows.end());
for(i=0; i<(int)rows.size(); i++)
{
if (ops[i].meta)
RemoveMeta(ops[i], rows[i], true, false);
else
{
if (ops[i].fix_cost) _fixCosts--;
DeleteRows(rows[i], 1);
_displayedOperations.erase(_displayedOperations.begin()+rows[i]);
for(a=i+1; a<(int)rows.size(); a++)
if (rows[a] >= rows[i])
rows[a]--;
}
}
for(it2=ops.begin(); it2!=ops.end(); it2++)
{
if (it2->id == parent) continue;
for (i=0, it3=op.childs.begin(); it3!=op.childs.end(); it3++, i++)
{
op2 = GetOperation(*it3);
if (*it3 == it2->id ||
op2.day > it2->day)
break;
}
if (i) i--;
if (it3 == op.childs.end())
op.childs.push_back(it2->id);
else if (*it3 == it2->id)
continue;
else
op.childs.insert(op.childs.begin()+i, it2->id);
it2->parent = op.id;
UpdateOperation(*it2);
}
UpdateMeta(op);
InsertIntoGrid(op);
UpdateOperation(op);
}
void GridAccount::UnGroup()
{
std::vector<int> rows, selected;
std::vector<int>::iterator it;
std::vector<Operation> ops;
std::vector<wxString> ops2;
std::vector<Operation>::iterator it2;
std::vector<wxString>::iterator it3;
wxString parent = wxT("");
Operation op, op2;
int fix = -1, i, a, line;
GetSelectedOperations(&selected);
for (size_t i = 0; i < selected.size(); ++i)
{
op = _displayedOperations[selected[i]] ;
if (op.id.Length())
{
if ((parent.Length() && op.parent != parent)
|| (!op.parent.Length() && !op.meta))
{
wxMessageBox(_("Cannot ungroup these operations"), _("Error"), wxICON_ERROR | wxOK);
return ;
}
if (fix != -1 && ((!fix && op.fix_cost) || (fix && !op.fix_cost)))
{
wxMessageBox(_("Cannot ungroup these operations"), _("Error"), wxICON_ERROR | wxOK);
return ;
}
if (fix == -1)
fix = op.fix_cost ? 1 : 0;
if(op.meta)
{
parent = op.id;
continue;
}
if (!parent.Length() && op.parent.Length())
parent = op.parent;
ops.push_back(op);
rows.push_back(selected[i]);
}
}
if (!ops.size() && !parent.Length()) return;
removeLastGroup:
// Only one meta is selected
if (!ops.size())
{
line = GetDisplayedRow(parent);
op = _displayedOperations[line];
ops2 = op.childs;
RemoveMeta(op, line, true, false);
for(i=0; i<(int)ops2.size(); i++)
{
op2 = GetOperation(ops2[i]);
op2.parent = wxT("");
UpdateOperation(op2);
InsertIntoGrid(op2);
}
_kiss->DeleteOperation(op);
DeleteOperation(op);
}
else
{
if (!parent.Length()) return;
line = GetDisplayedRow(parent);
op2 = _displayedOperations[line];
for(i=0; i<(int)ops.size(); i++)
{
op = ops[i];
op.parent = wxT("");
UpdateOperation(op);
line = GetDisplayedRow(op.id);
DeleteRows(line, 1);
_displayedOperations.erase(_displayedOperations.begin()+line);
InsertIntoGrid(GetOperation(op.id));
if (op.fix_cost) _fixCosts--;
for (a=0; a<(int)op2.childs.size(); a++)
if (op2.childs[a] == op.id)
{
op2.childs.erase(op2.childs.begin()+a);
break;
}
UpdateMeta(op2);
UpdateOperation(op2);
}
line = GetDisplayedRow(parent);
_displayedOperations[line] = op2;
if (op2.childs.size() < 2)
{
ops.clear();
// Sorry ...
goto removeLastGroup;
}
UpdateOperation(op2);
}
ComputeWeeks();
}
void GridAccount::MassUpdate(std::vector<int>& rows, updateOperationFunc func, void** params)
{
int i, b;
std::vector<Operation>::iterator it;
Operation op, op2;
_parent->Disable();
_parent->SetCursor(wxCursor(wxCURSOR_WAIT));
_parent->Update();
if (rows.size())
{
for(i=0; i<(int)rows.size(); i++)
{
op = _displayedOperations[rows[i]];
func (&op, params);
UpdateOperation(op);
if (op.meta)
{
for(b=0; b<(int)op.childs.size(); b++)
{
op2 = GetOperation(op.childs[b]);
func (&op2, params);
UpdateOperation(op2);
}
}
}
}
else
{
for(it=_operations->begin(); it!=_operations->end(); it++)
{
func (&(*it), params);
if (_databaseSynchronization)
_kiss->UpdateOperation(*it);
}
}
ClearGrid();
LoadOperations(_operations, 0, 0);
Layout();
_parent->Enable();
_parent->SetCursor(wxNullCursor);
}
Reference in New Issue View Git Blame Copy Permalink