summaryrefslogtreecommitdiff
path: root/balance.cc
blob: eeef3d4986982db56b99a85ba8c9b79b23540826 (plain)
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
#include "ledger.h"

namespace ledger {

amount_t balance_t::amount(const commodity_t * commodity) const
{
  if (! commodity) {
    if (amounts.size() == 1) {
      amounts_map::const_iterator i = amounts.begin();
      return (*i).second;
    }
  }
  else if (amounts.size() > 0) {
    amounts_map::const_iterator i = amounts.find(commodity);
    if (i != amounts.end())
      return (*i).second;
  }
  return amount_t();
}

#if 0
balance_t balance_t::round() const
{
  balance_t temp;

  for (amounts_map::const_iterator i = amounts.begin();
       i != amounts.end();
       i++)
    temp += (*i).second.round();

  return temp;
}
#endif

balance_t balance_t::value(const std::time_t moment) const
{
  balance_t temp;

  for (amounts_map::const_iterator i = amounts.begin();
       i != amounts.end();
       i++)
    temp += (*i).second.value(moment);

#if 1
  return temp;
#else
  return temp.round();
#endif
}

void balance_t::write(std::ostream& out,
		      const int     first_width,
		      const int     latter_width) const
{
  bool first  = true;
  int  lwidth = latter_width;

  if (lwidth == -1)
    lwidth = first_width;

  for (amounts_map::const_iterator i = amounts.begin();
       i != amounts.end();
       i++) {
    if ((*i).second) {
      int width;

      if (! first) {
	out << std::endl;
	width = lwidth;
      } else {
	first = false;
	width = first_width;
      }

      out.width(width);
      out.fill(' ');
      out << std::right << std::string((*i).second);
    }
  }

  if (first) {
    out.width(first_width);
    out.fill(' ');
    out << std::right << "0";
  }
}

balance_pair_t::balance_pair_t(const transaction_t& xact)
  : quantity(xact.amount), cost(xact.cost) {}

balance_pair_t& balance_pair_t::operator+=(const transaction_t& xact)
{
  quantity += xact.amount;
  cost     += xact.cost;
  return *this;
}

balance_pair_t& balance_pair_t::operator-=(const transaction_t& xact)
{
  quantity -= xact.amount;
  cost     -= xact.cost;
  return *this;
}

} // namespace ledger