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 TypeRange::TypeRange(OperandRange values)
19     : TypeRange(values.begin().getBase(), values.size()) {}
20 TypeRange::TypeRange(ResultRange values)
21     : TypeRange(values.getBase()->getResultTypes().slice(values.getStartIndex(),
22                                                          values.size())) {}
23 TypeRange::TypeRange(ArrayRef<Value> values)
24     : TypeRange(values.data(), values.size()) {}
25 TypeRange::TypeRange(ValueRange values) : TypeRange(OwnerT(), values.size()) {
26   detail::ValueRangeOwner owner = values.begin().getBase();
27   if (auto *op = reinterpret_cast<Operation *>(owner.ptr.dyn_cast<void *>()))
28     this->base = op->getResultTypes().drop_front(owner.startIndex).data();
29   else if (auto *operand = owner.ptr.dyn_cast<OpOperand *>())
30     this->base = operand;
31   else
32     this->base = owner.ptr.get<const Value *>();
33 }
34 
35 /// See `llvm::detail::indexed_accessor_range_base` for details.
36 TypeRange::OwnerT TypeRange::offset_base(OwnerT object, ptrdiff_t index) {
37   if (const auto *value = object.dyn_cast<const Value *>())
38     return {value + index};
39   if (auto *operand = object.dyn_cast<OpOperand *>())
40     return {operand + index};
41   return {object.dyn_cast<const Type *>() + index};
42 }
43 /// See `llvm::detail::indexed_accessor_range_base` for details.
44 Type TypeRange::dereference_iterator(OwnerT object, ptrdiff_t index) {
45   if (const auto *value = object.dyn_cast<const Value *>())
46     return (value + index)->getType();
47   if (auto *operand = object.dyn_cast<OpOperand *>())
48     return (operand + index)->get().getType();
49   return object.dyn_cast<const Type *>()[index];
50 }
51