59 lines · cpp
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-exception6//7//===----------------------------------------------------------------------===//8 9#include "mlir/IR/TypeRange.h"10 11using namespace mlir;12 13//===----------------------------------------------------------------------===//14// TypeRange15//===----------------------------------------------------------------------===//16 17TypeRange::TypeRange(ArrayRef<Type> types)18 : TypeRange(types.data(), types.size()) {19 assert(llvm::all_of(types, [](Type t) { return t; }) &&20 "attempting to construct a TypeRange with null types");21}22TypeRange::TypeRange(OperandRange values)23 : TypeRange(values.begin().getBase(), values.size()) {}24TypeRange::TypeRange(ResultRange values)25 : TypeRange(values.getBase(), values.size()) {}26TypeRange::TypeRange(ValueRange values) : TypeRange(OwnerT(), values.size()) {27 if (count == 0)28 return;29 ValueRange::OwnerT owner = values.begin().getBase();30 if (auto *result = llvm::dyn_cast_if_present<detail::OpResultImpl *>(owner))31 this->base = result;32 else if (auto *operand = llvm::dyn_cast_if_present<OpOperand *>(owner))33 this->base = operand;34 else35 this->base = cast<const Value *>(owner);36}37 38/// See `llvm::detail::indexed_accessor_range_base` for details.39TypeRange::OwnerT TypeRange::offset_base(OwnerT object, ptrdiff_t index) {40 if (const auto *value = llvm::dyn_cast_if_present<const Value *>(object))41 return {value + index};42 if (auto *operand = llvm::dyn_cast_if_present<OpOperand *>(object))43 return {operand + index};44 if (auto *result = llvm::dyn_cast_if_present<detail::OpResultImpl *>(object))45 return {result->getNextResultAtOffset(index)};46 return {llvm::dyn_cast_if_present<const Type *>(object) + index};47}48 49/// See `llvm::detail::indexed_accessor_range_base` for details.50Type TypeRange::dereference_iterator(OwnerT object, ptrdiff_t index) {51 if (const auto *value = llvm::dyn_cast_if_present<const Value *>(object))52 return (value + index)->getType();53 if (auto *operand = llvm::dyn_cast_if_present<OpOperand *>(object))54 return (operand + index)->get().getType();55 if (auto *result = llvm::dyn_cast_if_present<detail::OpResultImpl *>(object))56 return result->getNextResultAtOffset(index)->getType();57 return llvm::dyn_cast_if_present<const Type *>(object)[index];58}59