Query.h

Go to the documentation of this file.

//===--- Query.h ------------------------------------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
 
#ifndef MLIR_TOOLS_MLIRQUERY_QUERY_H
#define MLIR_TOOLS_MLIRQUERY_QUERY_H
 
#include "Matcher/VariantValue.h"
#include "mlir/Support/LogicalResult.h"
#include "llvm/ADT/IntrusiveRefCntPtr.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/LineEditor/LineEditor.h"
#include <string>
 
namespace mlir::query {
 
enum class QueryKind { Invalid, NoOp, Help, Match, Quit };
 
class QuerySession;
 
struct Query : llvm::RefCountedBase<Query> {
  Query(QueryKind kind) : kind(kind) {}
  virtual ~Query();
 
  // Perform the query on qs and print output to os.
  virtual mlir::LogicalResult run(llvm::raw_ostream &os,
                                  QuerySession &qs) const = 0;
 
  llvm::StringRef remainingContent;
  const QueryKind kind;
};
 
typedef llvm::IntrusiveRefCntPtr<Query> QueryRef;
 
QueryRef parse(llvm::StringRef line, const QuerySession &qs);
 
std::vector<llvm::LineEditor::Completion>
complete(llvm::StringRef line, size_t pos, const QuerySession &qs);
 
// Any query which resulted in a parse error. The error message is in ErrStr.
struct InvalidQuery : Query {
  InvalidQuery(const llvm::Twine &errStr)
      : Query(QueryKind::Invalid), errStr(errStr.str()) {}
  mlir::LogicalResult run(llvm::raw_ostream &os,
                          QuerySession &qs) const override;
 
  std::string errStr;
 
  static bool classof(const Query *query) {
    return query->kind == QueryKind::Invalid;
  }
};
 
// No-op query (i.e. a blank line).
struct NoOpQuery : Query {
  NoOpQuery() : Query(QueryKind::NoOp) {}
  mlir::LogicalResult run(llvm::raw_ostream &os,
                          QuerySession &qs) const override;
 
  static bool classof(const Query *query) {
    return query->kind == QueryKind::NoOp;
  }
};
 
// Query for "help".
struct HelpQuery : Query {
  HelpQuery() : Query(QueryKind::Help) {}
  mlir::LogicalResult run(llvm::raw_ostream &os,
                          QuerySession &qs) const override;
 
  static bool classof(const Query *query) {
    return query->kind == QueryKind::Help;
  }
};
 
// Query for "quit".
struct QuitQuery : Query {
  QuitQuery() : Query(QueryKind::Quit) {}
  mlir::LogicalResult run(llvm::raw_ostream &os,
                          QuerySession &qs) const override;
 
  static bool classof(const Query *query) {
    return query->kind == QueryKind::Quit;
  }
};
 
// Query for "match MATCHER".
struct MatchQuery : Query {
  MatchQuery(llvm::StringRef source, const matcher::DynMatcher &matcher)
      : Query(QueryKind::Match), matcher(matcher), source(source) {}
  mlir::LogicalResult run(llvm::raw_ostream &os,
                          QuerySession &qs) const override;
 
  const matcher::DynMatcher matcher;
 
  llvm::StringRef source;
 
  static bool classof(const Query *query) {
    return query->kind == QueryKind::Match;
  }
};
 
} // namespace mlir::query
 
#endif