1//===- CombinerUtils.h ----------------------------------------------------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9/// \file Utility functions used by both Combiner backends.
10//
11//===----------------------------------------------------------------------===//
12
13#ifndef LLVM_UTILS_TABLEGEN_COMMON_GLOBALISEL_COMBINERUTILS_H
14#define LLVM_UTILS_TABLEGEN_COMMON_GLOBALISEL_COMBINERUTILS_H
15
16#include "llvm/ADT/StringRef.h"
17#include "llvm/TableGen/Record.h"
18
19namespace llvm {
20
21/// A convenience function to check that an Init refers to a specific def. This
22/// is primarily useful for testing for defs and similar in DagInit's since
23/// DagInit's support any type inside them.
24inline bool isSpecificDef(const Init &N, StringRef Def) {
25 if (const DefInit *OpI = dyn_cast<DefInit>(Val: &N))
26 if (OpI->getDef()->getName() == Def)
27 return true;
28 return false;
29}
30
31/// A convenience function to check that an Init refers to a def that is a
32/// subclass of the given class and coerce it to a def if it is. This is
33/// primarily useful for testing for subclasses of GIDefKind and similar in
34/// DagInit's since DagInit's support any type inside them.
35inline const Record *getDefOfSubClass(const Init &N, StringRef Cls) {
36 if (const DefInit *OpI = dyn_cast<DefInit>(Val: &N))
37 if (OpI->getDef()->isSubClassOf(Name: Cls))
38 return OpI->getDef();
39 return nullptr;
40}
41
42/// A convenience function to check that an Init refers to a dag whose operator
43/// is a specific def and coerce it to a dag if it is. This is primarily useful
44/// for testing for subclasses of GIDefKind and similar in DagInit's since
45/// DagInit's support any type inside them.
46inline const DagInit *getDagWithSpecificOperator(const Init &N,
47 StringRef Name) {
48 if (const DagInit *I = dyn_cast<DagInit>(Val: &N))
49 if (I->getNumArgs() > 0)
50 if (const DefInit *OpI = dyn_cast<DefInit>(Val: I->getOperator()))
51 if (OpI->getDef()->getName() == Name)
52 return I;
53 return nullptr;
54}
55
56/// A convenience function to check that an Init refers to a dag whose operator
57/// is a def that is a subclass of the given class and coerce it to a dag if it
58/// is. This is primarily useful for testing for subclasses of GIDefKind and
59/// similar in DagInit's since DagInit's support any type inside them.
60inline const DagInit *getDagWithOperatorOfSubClass(const Init &N,
61 StringRef Cls) {
62 if (const DagInit *I = dyn_cast<DagInit>(Val: &N))
63 if (const DefInit *OpI = dyn_cast<DefInit>(Val: I->getOperator()))
64 if (OpI->getDef()->isSubClassOf(Name: Cls))
65 return I;
66 return nullptr;
67}
68
69/// Copies a StringRef into a static pool to preserve it.
70StringRef insertStrRef(StringRef S);
71
72} // namespace llvm
73
74#endif // LLVM_UTILS_TABLEGEN_COMMON_GLOBALISEL_COMBINERUTILS_H
75