1 //===- TypeRange.cpp ------------------------------------------------------===//
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 #include "mlir/IR/TypeRange.h"
10 #include "mlir/IR/Operation.h"
11 using namespace mlir;
12 
13 //===----------------------------------------------------------------------===//
14 // TypeRange
15 
16 TypeRange::TypeRange(ArrayRef<Type> types)
17     : TypeRange(types.data(), types.size()) {
18   assert(llvm::all_of(types, [](Type t) { return t; }) &&
19          "attempting to construct a TypeRange with null types");
20 }
21 TypeRange::TypeRange(OperandRange values)
22     : TypeRange(values.begin().getBase(), values.size()) {}
23 TypeRange::TypeRange(ResultRange values)
24     : TypeRange(values.getBase()->getResultTypes().slice(values.getStartIndex(),
25                                                          values.size())) {}
26 TypeRange::TypeRange(ArrayRef<Value> values)
27     : TypeRange(values.data(), values.size()) {}
28 TypeRange::TypeRange(ValueRange values) : TypeRange(OwnerT(), values.size()) {
29   detail::ValueRangeOwner owner = values.begin().getBase();
30   if (auto *op = reinterpret_cast<Operation *>(owner.ptr.dyn_cast<void *>()))
31     this->base = op->getResultTypes().drop_front(owner.startIndex).data();
32   else if (auto *operand = owner.ptr.dyn_cast<OpOperand *>())
33     this->base = operand;
34   else
35     this->base = owner.ptr.get<const Value *>();
36 }
37 
38 /// See `llvm::detail::indexed_accessor_range_base` for details.
39 TypeRange::OwnerT TypeRange::offset_base(OwnerT object, ptrdiff_t index) {
40   if (const auto *value = object.dyn_cast<const Value *>())
41     return {value + index};
42   if (auto *operand = object.dyn_cast<OpOperand *>())
43     return {operand + index};
44   return {object.dyn_cast<const Type *>() + index};
45 }
46 /// See `llvm::detail::indexed_accessor_range_base` for details.
47 Type TypeRange::dereference_iterator(OwnerT object, ptrdiff_t index) {
48   if (const auto *value = object.dyn_cast<const Value *>())
49     return (value + index)->getType();
50   if (auto *operand = object.dyn_cast<OpOperand *>())
51     return (operand + index)->get().getType();
52   return object.dyn_cast<const Type *>()[index];
53 }
54