nlreader-example.cc

#include "mp/nl-reader.h"

enum Expr {OTHER, CONST};

struct ExprCounter : mp::NullNLHandler<Expr> {
  int num_divs;
  ExprCounter() : num_divs(0) {}
  Expr OnBinary(mp::expr::Kind kind, Expr lhs, Expr rhs) {
    if (kind == mp::expr::DIV && rhs != CONST)
      ++num_divs;
  }
  Expr OnNumericConstant(double value) { return CONST; }
};

int main(int argc, char **argv) {
  for (int i = 1; i < argc; ++i) {
    ExprCounter counter;
    mp::ReadNLFile(argv[i], counter);
    if (counter.num_divs != 0)
      fmt::print("{}: {} divisions\n", argv[i], counter.num_divs);
  }
}