1 //===-- Utils.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 // Coding style: https://mlir.llvm.org/getting_started/DeveloperGuide/
10 //
11 //===----------------------------------------------------------------------===//
12
13 #include "flang/Optimizer/Support/Utils.h"
14 #include "flang/Optimizer/Dialect/FIROps.h"
15 #include "flang/Optimizer/Dialect/FIRType.h"
16 #include "flang/Optimizer/Support/InternalNames.h"
17
lookupTypeInfoOp(fir::RecordType recordType,mlir::ModuleOp module,const mlir::SymbolTable * symbolTable)18 fir::TypeInfoOp fir::lookupTypeInfoOp(fir::RecordType recordType,
19 mlir::ModuleOp module,
20 const mlir::SymbolTable *symbolTable) {
21 // fir.type_info was created with the mangled name of the derived type.
22 // It is the same as the name in the related fir.type, except when a pass
23 // lowered the fir.type (e.g., when lowering fir.boxproc type if the type has
24 // pointer procedure components), in which case suffix may have been added to
25 // the fir.type name. Get rid of them when looking up for the fir.type_info.
26 llvm::StringRef originalMangledTypeName =
27 fir::NameUniquer::dropTypeConversionMarkers(recordType.getName());
28 return fir::lookupTypeInfoOp(originalMangledTypeName, module, symbolTable);
29 }
30
lookupTypeInfoOp(llvm::StringRef name,mlir::ModuleOp module,const mlir::SymbolTable * symbolTable)31 fir::TypeInfoOp fir::lookupTypeInfoOp(llvm::StringRef name,
32 mlir::ModuleOp module,
33 const mlir::SymbolTable *symbolTable) {
34 if (symbolTable)
35 if (auto typeInfo = symbolTable->lookup<fir::TypeInfoOp>(name))
36 return typeInfo;
37 return module.lookupSymbol<fir::TypeInfoOp>(name);
38 }
39
getComponentLowerBoundsIfNonDefault(fir::RecordType recordType,llvm::StringRef component,mlir::ModuleOp module,const mlir::SymbolTable * symbolTable)40 std::optional<llvm::ArrayRef<int64_t>> fir::getComponentLowerBoundsIfNonDefault(
41 fir::RecordType recordType, llvm::StringRef component,
42 mlir::ModuleOp module, const mlir::SymbolTable *symbolTable) {
43 fir::TypeInfoOp typeInfo =
44 fir::lookupTypeInfoOp(recordType, module, symbolTable);
45 if (!typeInfo || typeInfo.getComponentInfo().empty())
46 return std::nullopt;
47 for (auto componentInfo :
48 typeInfo.getComponentInfo().getOps<fir::DTComponentOp>())
49 if (componentInfo.getName() == component)
50 return componentInfo.getLowerBounds();
51 return std::nullopt;
52 }
53