var_extract.h

// --- Hongce Zhang

#ifndef ILANG_VTARGET_OUT_VAR_EXTRACT_H__
#define ILANG_VTARGET_OUT_VAR_EXTRACT_H__

#include <functional>
#include <string>
#include <vector>

namespace ilang {

class VarExtractor {

public:
  typedef enum { KEEP = 0, ILA_S, ILA_IN, VLG_S, UNKN_S, NUM } token_type;
  typedef std::pair<token_type, std::string> token;
  typedef std::function<bool(const std::string&)> str_j;
  typedef std::function<std::string(const token&)> str_r;

protected:
  std::vector<token> _tokens;
  str_j _is_ila_state;
  str_j _is_ila_input;
  str_j _is_vlg_sig;

public:
  // ---------------------- CONSTRUCTOR ----------------- //
  VarExtractor(str_j is_ila_state, str_j is_ila_input, str_j is_vlg_sig);
  // ---------------------- METHODS ----------------- //
  void
  ParseToExtract(const std::string& in,
                 bool force_vlg_statename =
                     false); // the later is only used in dealing w. invariants
  void ForEachTokenReplace(
      str_r replacer); // you can return the same so it will not be replaced
  std::string GenString() const;
  std::string Replace(const std::string& in, bool force_vlg_statename,
                      str_r replacer) {
    ParseToExtract(in, force_vlg_statename);
    ForEachTokenReplace(replacer);
    return GenString();
  }
  // ---------------------- HELPERS ----------------- //
  static bool contains_mod_inst_name(const std::string& s,
                                     const std::string& mi);

}; // class VarExtractor

}; // namespace ilang

#endif // ILANG_VTARGET_OUT_VAR_EXTRACT_H__