Option --count sums payees, account, commodities

This commit is contained in:
John Wiegley 2010-05-22 17:04:38 -04:00
parent de3803d027
commit 02e7825516
4 changed files with 39 additions and 18 deletions

View file

@ -236,38 +236,51 @@ void report_accounts::flush()
{ {
std::ostream& out(report.output_stream); std::ostream& out(report.output_stream);
foreach (accounts_pair& entry, accounts) foreach (accounts_pair& entry, accounts) {
if (report.HANDLED(count))
out << entry.second << ' ';
out << *entry.first << '\n'; out << *entry.first << '\n';
}
} }
void report_accounts::operator()(post_t& post) void report_accounts::operator()(post_t& post)
{ {
std::map<account_t *, bool>::iterator i = accounts.find(post.account); std::map<account_t *, std::size_t>::iterator i = accounts.find(post.account);
if (i == accounts.end()) if (i == accounts.end())
accounts.insert(accounts_pair(post.account, true)); accounts.insert(accounts_pair(post.account, 1));
else
(*i).second++;
} }
void report_payees::flush() void report_payees::flush()
{ {
std::ostream& out(report.output_stream); std::ostream& out(report.output_stream);
foreach (payees_pair& entry, payees) foreach (payees_pair& entry, payees) {
if (report.HANDLED(count))
out << entry.second << ' ';
out << entry.first << '\n'; out << entry.first << '\n';
}
} }
void report_payees::operator()(post_t& post) void report_payees::operator()(post_t& post)
{ {
std::map<string, bool>::iterator i = payees.find(post.xact->payee); std::map<string, std::size_t>::iterator i = payees.find(post.xact->payee);
if (i == payees.end()) if (i == payees.end())
payees.insert(payees_pair(post.xact->payee, true)); payees.insert(payees_pair(post.xact->payee, 1));
else
(*i).second++;
} }
void report_commodities::flush() void report_commodities::flush()
{ {
std::ostream& out(report.output_stream); std::ostream& out(report.output_stream);
foreach (commodities_pair& entry, commodities) foreach (commodities_pair& entry, commodities) {
if (report.HANDLED(count))
out << entry.second << ' ';
out << *entry.first << '\n'; out << *entry.first << '\n';
}
} }
void report_commodities::operator()(post_t& post) void report_commodities::operator()(post_t& post)
@ -275,18 +288,22 @@ void report_commodities::operator()(post_t& post)
amount_t temp(post.amount.strip_annotations(report.what_to_keep())); amount_t temp(post.amount.strip_annotations(report.what_to_keep()));
commodity_t& comm(temp.commodity()); commodity_t& comm(temp.commodity());
std::map<commodity_t *, bool>::iterator i = commodities.find(&comm); std::map<commodity_t *, std::size_t>::iterator i = commodities.find(&comm);
if (i == commodities.end()) if (i == commodities.end())
commodities.insert(commodities_pair(&comm, true)); commodities.insert(commodities_pair(&comm, 1));
else
(*i).second++;
if (comm.has_annotation()) { if (comm.has_annotation()) {
annotated_commodity_t& ann_comm(as_annotated_commodity(comm)); annotated_commodity_t& ann_comm(as_annotated_commodity(comm));
if (ann_comm.details.price) { if (ann_comm.details.price) {
std::map<commodity_t *, bool>::iterator i = std::map<commodity_t *, std::size_t>::iterator i =
commodities.find(&ann_comm.details.price->commodity()); commodities.find(&ann_comm.details.price->commodity());
if (i == commodities.end()) if (i == commodities.end())
commodities.insert commodities.insert
(commodities_pair(&ann_comm.details.price->commodity(), true)); (commodities_pair(&ann_comm.details.price->commodity(), 1));
else
(*i).second++;
} }
} }
@ -294,7 +311,9 @@ void report_commodities::operator()(post_t& post)
amount_t temp_cost(post.cost->strip_annotations(report.what_to_keep())); amount_t temp_cost(post.cost->strip_annotations(report.what_to_keep()));
i = commodities.find(&temp_cost.commodity()); i = commodities.find(&temp_cost.commodity());
if (i == commodities.end()) if (i == commodities.end())
commodities.insert(commodities_pair(&temp_cost.commodity(), true)); commodities.insert(commodities_pair(&temp_cost.commodity(), 1));
else
(*i).second++;
} }
} }

View file

@ -108,9 +108,9 @@ class report_accounts : public item_handler<post_t>
protected: protected:
report_t& report; report_t& report;
std::map<account_t *, bool> accounts; std::map<account_t *, std::size_t> accounts;
typedef std::map<account_t *, bool>::value_type accounts_pair; typedef std::map<account_t *, std::size_t>::value_type accounts_pair;
public: public:
report_accounts(report_t& _report) : report(_report) { report_accounts(report_t& _report) : report(_report) {
@ -129,9 +129,9 @@ class report_payees : public item_handler<post_t>
protected: protected:
report_t& report; report_t& report;
std::map<string, bool> payees; std::map<string, std::size_t> payees;
typedef std::map<string, bool>::value_type payees_pair; typedef std::map<string, std::size_t>::value_type payees_pair;
public: public:
report_payees(report_t& _report) : report(_report) { report_payees(report_t& _report) : report(_report) {
@ -150,9 +150,9 @@ class report_commodities : public item_handler<post_t>
protected: protected:
report_t& report; report_t& report;
std::map<commodity_t *, bool> commodities; std::map<commodity_t *, std::size_t> commodities;
typedef std::map<commodity_t *, bool>::value_type commodities_pair; typedef std::map<commodity_t *, std::size_t>::value_type commodities_pair;
public: public:
report_commodities(report_t& _report) : report(_report) { report_commodities(report_t& _report) : report(_report) {

View file

@ -853,6 +853,7 @@ option_t<report_t> * report_t::lookup_option(const char * p)
else OPT(columns_); else OPT(columns_);
else OPT_ALT(basis, cost); else OPT_ALT(basis, cost);
else OPT_(current); else OPT_(current);
else OPT(count);
break; break;
case 'd': case 'd':
OPT(daily); OPT(daily);

View file

@ -454,6 +454,7 @@ public:
}); });
OPTION(report_t, columns_); OPTION(report_t, columns_);
OPTION(report_t, count);
OPTION__(report_t, csv_format_, CTOR(report_t, csv_format_) { OPTION__(report_t, csv_format_, CTOR(report_t, csv_format_) {
on(none, on(none,