Moved around most of the files so that source code is in src/, documentation

is in doc/, etc.

1 files changed, 255 insertions, 0 deletions

diff --git a/src/balance.cc b/src/balance.cc
new file mode 100644
index 00000000..bac8d40c
--- /dev/null
+++ b/src/balance.cc
@@ -0,0 +1,255 @@
+/*
+ * Copyright (c) 2003-2008, John Wiegley.  All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are
+ * met:
+ *
+ * - Redistributions of source code must retain the above copyright
+ *   notice, this list of conditions and the following disclaimer.
+ *
+ * - Redistributions in binary form must reproduce the above copyright
+ *   notice, this list of conditions and the following disclaimer in the
+ *   documentation and/or other materials provided with the distribution.
+ *
+ * - Neither the name of New Artisans LLC nor the names of its
+ *   contributors may be used to endorse or promote products derived from
+ *   this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include "balance.h"
+
+namespace ledger {
+
+balance_t& balance_t::operator+=(const balance_t& bal)
+{
+  foreach (const amounts_map::value_type& pair, bal.amounts)
+    *this += pair.second;
+  return *this;
+}
+
+balance_t& balance_t::operator+=(const amount_t& amt)
+{
+  if (amt.is_null())
+    throw_(balance_error,
+	   "Cannot add an uninitialized amount to a balance");
+
+  if (amt.is_realzero())
+    return *this;
+
+  amounts_map::iterator i = amounts.find(&amt.commodity());
+  if (i != amounts.end())
+    i->second += amt;
+  else
+    amounts.insert(amounts_map::value_type(&amt.commodity(), amt));
+
+  return *this;
+}
+
+balance_t& balance_t::operator-=(const balance_t& bal)
+{
+  foreach (const amounts_map::value_type& pair, bal.amounts)
+    *this -= pair.second;
+  return *this;
+}
+
+balance_t& balance_t::operator-=(const amount_t& amt)
+{
+  if (amt.is_null())
+    throw_(balance_error,
+	   "Cannot subtract an uninitialized amount from a balance");
+
+  if (amt.is_realzero())
+    return *this;
+
+  amounts_map::iterator i = amounts.find(&amt.commodity());
+  if (i != amounts.end()) {
+    i->second -= amt;
+    if (i->second.is_realzero())
+      amounts.erase(i);
+  } else {
+    amounts.insert(amounts_map::value_type(&amt.commodity(), amt.negate()));
+  }
+  return *this;
+}
+
+balance_t& balance_t::operator*=(const amount_t& amt)
+{
+  if (amt.is_null())
+    throw_(balance_error,
+	   "Cannot multiply a balance by an uninitialized amount");
+
+  if (is_realzero()) {
+    ;
+  }
+  else if (amt.is_realzero()) {
+    *this = amt;
+  }
+  else if (! amt.commodity()) {
+    // Multiplying by an amount with no commodity causes all the
+    // component amounts to be increased by the same factor.
+    foreach (amounts_map::value_type& pair, amounts)
+      pair.second *= amt;
+  }
+  else if (amounts.size() == 1) {
+    // Multiplying by a commoditized amount is only valid if the sole
+    // commodity in the balance is of the same kind as the amount's
+    // commodity.
+    if (*amounts.begin()->first == amt.commodity())
+      amounts.begin()->second *= amt;
+    else
+      throw_(balance_error,
+	     "Cannot multiply a balance with annotated commodities by a commoditized amount");
+  }
+  else {
+    assert(amounts.size() > 1);
+    throw_(balance_error,
+	   "Cannot multiply a multi-commodity balance by a commoditized amount");
+  }
+  return *this;
+}
+
+balance_t& balance_t::operator/=(const amount_t& amt)
+{
+  if (amt.is_null())
+    throw_(balance_error,
+	   "Cannot divide a balance by an uninitialized amount");
+
+  if (is_realzero()) {
+    ;
+  }
+  else if (amt.is_realzero()) {
+    throw_(balance_error, "Divide by zero");
+  }
+  else if (! amt.commodity()) {
+    // Dividing by an amount with no commodity causes all the
+    // component amounts to be divided by the same factor.
+    foreach (amounts_map::value_type& pair, amounts)
+      pair.second /= amt;
+  }
+  else if (amounts.size() == 1) {
+    // Dividing by a commoditized amount is only valid if the sole
+    // commodity in the balance is of the same kind as the amount's
+    // commodity.
+    if (*amounts.begin()->first == amt.commodity())
+      amounts.begin()->second /= amt;
+    else
+      throw_(balance_error,
+	     "Cannot divide a balance with annotated commodities by a commoditized amount");
+  }
+  else {
+    assert(amounts.size() > 1);
+    throw_(balance_error,
+	   "Cannot divide a multi-commodity balance by a commoditized amount");
+  }
+  return *this;
+}
+
+optional<balance_t>
+balance_t::value(const optional<datetime_t>& moment) const
+{
+  optional<balance_t> temp;
+
+  foreach (const amounts_map::value_type& pair, amounts)
+    if (optional<amount_t> val = pair.second.value(moment)) {
+      if (! temp)
+	temp = balance_t();
+      *temp += *val;
+    }
+
+  return temp;
+}
+
+optional<amount_t>
+balance_t::commodity_amount(const optional<const commodity_t&>& commodity) const
+{
+  // jww (2007-05-20): Needs work
+  if (! commodity) {
+    if (amounts.size() == 1) {
+      amounts_map::const_iterator i = amounts.begin();
+      return i->second;
+    }
+    else if (amounts.size() > 1) {
+      // Try stripping annotations before giving an error.
+      balance_t temp(strip_annotations());
+      if (temp.amounts.size() == 1)
+	return temp.commodity_amount(commodity);
+
+      throw_(amount_error,
+	     "Requested amount of a balance with multiple commodities: " << temp);
+    }
+  }
+  else if (amounts.size() > 0) {
+    amounts_map::const_iterator i = amounts.find(&*commodity);
+    if (i != amounts.end())
+      return i->second;
+  }
+  return none;
+}
+
+balance_t balance_t::strip_annotations(const bool keep_price,
+				       const bool keep_date,
+				       const bool keep_tag) const
+{
+  balance_t temp;
+
+  foreach (const amounts_map::value_type& pair, amounts)
+    temp += pair.second.strip_annotations(keep_price, keep_date, keep_tag);
+
+  return temp;
+}
+
+void balance_t::print(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;
+
+  typedef std::vector<const amount_t *> amounts_array;
+  amounts_array sorted;
+
+  foreach (const amounts_map::value_type& pair, amounts)
+    if (pair.second)
+      sorted.push_back(&pair.second);
+
+  std::stable_sort(sorted.begin(), sorted.end(), compare_amount_commodities());
+
+  foreach (const amount_t * amount, sorted) {
+    int width;
+    if (! first) {
+      out << std::endl;
+      width = lwidth;
+    } else {
+      first = false;
+      width = first_width;
+    }
+
+    out.width(width);
+    out.fill(' ');
+    out << std::right << *amount;
+  }
+
+  if (first) {
+    out.width(first_width);
+    out.fill(' ');
+    out << std::right << "0";
+  }
+}
+
+} // namespace ledger