summaryrefslogtreecommitdiff
path: root/valexpr.h
blob: b1be9a6cb1e11b78382f875781b12a51f2a15390 (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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
#ifndef _EXPR_H
#define _EXPR_H

#include "ledger.h"
#include "value.h"
#include "error.h"

namespace ledger {

class mask_t
{
 public:
  bool        exclude;
  std::string pattern;
  void *      regexp;

  explicit mask_t(const std::string& pattern);
  mask_t(const mask_t&);

  ~mask_t();

  bool match(const std::string& str) const;
};


struct details_t
{
  const entry_t *	entry;
  const transaction_t * xact;
  const account_t *     account;

  details_t(const entry_t * _entry)
    : entry(_entry), xact(NULL), account(NULL) {}

  details_t(const transaction_t * _xact)
    : entry(_xact->entry), xact(_xact), account(_xact->account) {}

  details_t(const account_t * _account)
    : entry(NULL), xact(NULL), account(_account) {}
};

struct value_expr_t
{
  enum kind_t {
    // Constants
    CONSTANT_I,
    CONSTANT_T,
    CONSTANT_A,

    // Item details
    AMOUNT,
    COST,
    DATE,
    CLEARED,
    REAL,
    INDEX,
    DEPTH,

    // Item totals
    COUNT,
    TOTAL,
    COST_TOTAL,

    // Relating to format_t
    VALUE_EXPR,
    TOTAL_EXPR,

    // Functions
    F_ARITH_MEAN,
    F_VALUE,
    F_NEG,
    F_ABS,
    F_STRIP,
    F_PAYEE_MASK,
    F_ACCOUNT_MASK,
    F_SHORT_ACCOUNT_MASK,

    // Binary operators
    O_ADD,
    O_SUB,
    O_MUL,
    O_DIV,
    O_EQ,
    O_LT,
    O_LTE,
    O_GT,
    O_GTE,
    O_NOT,
    O_AND,
    O_OR,
    O_QUES,
    O_COL,

    LAST
  };

  kind_t	 kind;
  value_expr_t * left;
  value_expr_t * right;

  union {
    std::time_t	 constant_t;
    unsigned int constant_i;
  };
  amount_t	 constant_a;
  mask_t *	 mask;

  value_expr_t(const kind_t _kind)
    : kind(_kind), left(NULL), right(NULL), mask(NULL) {}

  ~value_expr_t() {
    if (mask)  delete mask;
    if (left)  delete left;
    if (right) delete right;
  }

  void compute(value_t& result, const details_t& details,
	       value_t::type_t type = value_t::ANY) const;

  void compute(balance_t& result, const details_t& details) const {
    value_t value;
    compute(value, details, value_t::BALANCE);
    result = value.operator balance_t();
  }
};

value_expr_t * parse_value_expr(std::istream& in);

inline value_expr_t * parse_value_expr(const char * p) {
  std::istringstream stream(p);
  return parse_value_expr(stream);
}

inline value_expr_t * parse_value_expr(const std::string& str) {
  return parse_value_expr(str.c_str());
}

#ifdef DEBUG_ENABLED
void dump_value_expr(std::ostream& out, const value_expr_t * node);
#endif

template <typename T>
class item_predicate
{
  const value_expr_t * predicate;

 public:
  item_predicate(const std::string& _predicate) {
    predicate = NULL;
    if (! _predicate.empty()) {
      try {
#ifdef DEBUG_ENABLED
	DEBUG_CLASS("valexpr.predicate.parse");

	DEBUG_PRINT_("parsing: '" << _predicate << "'");
#endif
	predicate = parse_value_expr(_predicate);

#ifdef DEBUG_ENABLED
	if (DEBUG_() && ledger::debug_stream) {
	  *ledger::debug_stream << "dump: ";
	  dump_value_expr(*ledger::debug_stream, predicate);
	  *ledger::debug_stream << std::endl;
	}
#endif
      }
      catch (const value_expr_error& err) {
	std::cerr << "Error in predicate '" << _predicate << "': "
		  << err.what() << std::endl;
	std::exit(1);
      }
    }
  }
  item_predicate(const value_expr_t * _predicate)
    : predicate(_predicate) {}

  ~item_predicate() {
    if (predicate)
      delete predicate;
  }

  bool operator()(const T * item) const {
    if (predicate) {
      value_t result;
      predicate->compute(result, details_t(item));
      return result;
    } else {
      return true;
    }
  }
};

} // namespace report

#endif // _REPORT_H