1 files changed, 92 insertions, 0 deletions
diff --git a/src/crush/CrushCompiler.h b/src/crush/CrushCompiler.h
new file mode 100644
index 00000000..f035085e
--- /dev/null
+++ b/src/crush/CrushCompiler.h
@@ -0,0 +1,92 @@
+// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
+// vim: ts=8 sw=2 smarttab
+
+#ifndef CEPH_CRUSH_COMPILER_H
+#define CEPH_CRUSH_COMPILER_H
+
+#include "crush/CrushWrapper.h"
+#include "crush/grammar.h"
+
+#include <map>
+#include <iostream>
+
+class CrushCompiler {
+  CrushWrapper& crush;
+  ostream& err;
+  int verbose;
+  bool unsafe_tunables;
+
+  // decompile
+  enum dcb_state_t {
+    DCB_STATE_IN_PROGRESS = 0,
+    DCB_STATE_DONE
+  };
+
+  int decompile_weight_set_weights(crush_weight_set weight_set,
+				   ostream &out);
+  int decompile_weight_set(crush_weight_set *weight_set,
+			   __u32 size,
+			   ostream &out);
+  int decompile_choose_arg(crush_choose_arg *arg,
+			   int bucket_id,
+			   ostream &out);
+  int decompile_ids(int *ids,
+		    __u32 size,
+		    ostream &out);
+  int decompile_choose_arg_map(crush_choose_arg_map arg_map,
+			       ostream &out);
+  int decompile_choose_args(const std::pair<const long unsigned int, crush_choose_arg_map> &i,
+			    ostream &out);
+  int decompile_bucket_impl(int i, ostream &out);
+  int decompile_bucket(int cur,
+		       std::map<int, dcb_state_t>& dcb_states,
+		       ostream &out);
+
+  // compile
+  typedef char const*         iterator_t;
+  typedef tree_match<iterator_t> parse_tree_match_t;
+  typedef parse_tree_match_t::tree_iterator iter_t;
+  typedef parse_tree_match_t::node_t node_t;
+
+  map<string, int> item_id;
+  map<int, string> id_item;
+  map<int, unsigned> item_weight;
+  map<string, int> type_id;
+  map<string, int> rule_id;
+  std::map<int32_t, map<int32_t, int32_t> > class_bucket; // bucket id -> class id -> shadow bucket id
+
+  string string_node(node_t &node);
+  int int_node(node_t &node); 
+  float float_node(node_t &node);
+
+  int parse_tunable(iter_t const& i);
+  int parse_device(iter_t const& i);
+  int parse_bucket_type(iter_t const& i);
+  int parse_bucket(iter_t const& i);
+  int parse_rule(iter_t const& i);
+  int parse_weight_set_weights(iter_t const& i, int bucket_id, crush_weight_set *weight_set);
+  int parse_weight_set(iter_t const& i, int bucket_id, crush_choose_arg *arg);
+  int parse_choose_arg_ids(iter_t const& i, int bucket_id, crush_choose_arg *args);
+  int parse_choose_arg(iter_t const& i, crush_choose_arg *args);
+  int parse_choose_args(iter_t const& i);
+  void find_used_bucket_ids(iter_t const& i);
+  int parse_crush(iter_t const& i);  
+  void dump(iter_t const& i, int ind=1);
+  string consolidate_whitespace(string in);
+  int adjust_bucket_item_place(iter_t const &i);
+
+public:
+  CrushCompiler(CrushWrapper& c, ostream& eo, int verbosity=0)
+    : crush(c), err(eo), verbose(verbosity),
+      unsafe_tunables(false) {}
+  ~CrushCompiler() {}
+
+  void enable_unsafe_tunables() {
+    unsafe_tunables = true;
+  }
+
+  int decompile(ostream& out);
+  int compile(istream& in, const char *infn=0);
+};
+
+#endif