1 /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
2 /* vim:set ts=2 sw=2 sts=2 et cindent tw=79 ft=cpp: */
4 * Copyright (C) 2007 Sergey Yanovich <ynvich@gmail.com>
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License as
8 * published by the Free Software Foundation; either version 2 of the
9 * License, or (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * General Public License for more details.
16 * You should have received a copy of the GNU General Public
17 * License along with this program; if not, write to the
18 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
19 * Boston, MA 02111-1307, USA.
22 #include <abstract/aacore.h>
25 #include "nsServiceManagerUtils.h"
26 #include "nsStringAPI.h"
27 #include "nsTextFormatter.h"
30 #include "unstable/nsIArray.h"
31 #include "unstable/nsArrayUtils.h"
32 #include "unstable/nsITreeColumns.h"
33 #include "unstable/nsIAtom.h"
34 #include "unstable/nsIAtomService.h"
36 /* Project includes */
37 #include <abstract/base/aaIEntity.h>
38 #include <abstract/base/aaIFlow.h>
39 #include <abstract/base/aaIResource.h>
40 #include <abstract/base/aaIBalance.h>
41 #include <abstract/storage/aaILoadQuery.h>
42 #include <abstract/view/aaViewUtils.h>
43 #include "aaTreeView.h"
44 #include "aaTreeViewBalance.h"
46 aaTreeViewBalance::aaTreeViewBalance()
50 aaTreeViewBalance::~aaTreeViewBalance()
54 NS_IMPL_ISUPPORTS_INHERITED0(aaTreeViewBalance
, aaTreeView
)
58 aaTreeViewBalance::getColumn0(aaIBalance
*balance
, nsAString
& _retval
)
62 nsCOMPtr
<aaIFlow
> flow
;
63 rv
= balance
->GetFlow(getter_AddRefs( flow
));
64 NS_ENSURE_TRUE(flow
, NS_OK
);
66 return flow
->GetTag(_retval
);
70 aaTreeViewBalance::getColumn1(aaIBalance
*balance
, nsAString
& _retval
)
74 nsCOMPtr
<aaIFlow
> flow
;
75 rv
= balance
->GetFlow(getter_AddRefs( flow
));
76 NS_ENSURE_TRUE(flow
, NS_OK
);
78 nsCOMPtr
<aaIEntity
> entity
;
79 rv
= flow
->GetEntity(getter_AddRefs( entity
));
83 return entity
->GetTag(_retval
);
87 aaTreeViewBalance::getColumn2(aaIBalance
*balance
, nsAString
& _retval
)
91 nsCOMPtr
<aaIResource
> resource
;
92 rv
= balance
->GetResource(getter_AddRefs( resource
));
96 return resource
->GetTag(_retval
);
100 aaTreeViewBalance::getColumn3(aaIBalance
*balance
, nsAString
& _retval
)
104 balance
->GetSide(&side
);
107 balance
->GetValue(&value
);
108 assignDouble(_retval
, value
);
113 aaTreeViewBalance::getColumn4(aaIBalance
*balance
, nsAString
& _retval
)
117 balance
->GetSide(&side
);
120 balance
->GetValue(&value
);
121 assignDouble(_retval
, value
);
126 aaTreeViewBalance::colFuncs
[])(aaIBalance
*, nsAString
&) = {
127 &aaTreeViewBalance::getColumn0
,
128 &aaTreeViewBalance::getColumn1
,
129 &aaTreeViewBalance::getColumn2
,
130 &aaTreeViewBalance::getColumn3
,
131 &aaTreeViewBalance::getColumn4
134 /* Private functions */
136 aaTreeViewBalance::doGetCellText(PRInt32 row
, nsITreeColumn
*col
,
142 PRInt32 ci
= getColumnIndex(col
, sizeof(colFuncs
));
143 NS_ENSURE_TRUE(ci
>= 0, NS_OK
);
145 nsCOMPtr
<aaIBalance
> balance(do_QueryElementAt(mDataSet
, row
, &rv
));
146 NS_ENSURE_SUCCESS(rv
, rv
);
147 NS_ENSURE_TRUE(balance
, NS_OK
);
149 return colFuncs
[ci
](balance
, _retval
);