1//===- CharacterTest.cpp -- CharacterExprHelper unit tests ----------------===//
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 "flang/Optimizer/Builder/Character.h"
10#include "gtest/gtest.h"
11#include "flang/Optimizer/Builder/BoxValue.h"
12#include "flang/Optimizer/Builder/FIRBuilder.h"
13#include "flang/Optimizer/Dialect/Support/KindMapping.h"
14#include "flang/Optimizer/Support/InitFIR.h"
15
16struct CharacterTest : public testing::Test {
17public:
18 void SetUp() override {
19 fir::support::loadDialects(context);
20
21 kindMap = std::make_unique<fir::KindMapping>(&context,
22 "i10:80,l3:24,a1:8,r54:Double,c20:X86_FP80,r11:PPC_FP128,"
23 "r12:FP128,r13:X86_FP80,r14:Double,r15:Float,r16:Half,r23:BFloat");
24 mlir::OpBuilder builder(&context);
25 auto loc = builder.getUnknownLoc();
26
27 // Set up a Module with a dummy function operation inside.
28 // Set the insertion point in the function entry block.
29 moduleOp = builder.create<mlir::ModuleOp>(loc);
30 builder.setInsertionPointToStart(moduleOp->getBody());
31 mlir::func::FuncOp func = builder.create<mlir::func::FuncOp>(
32 loc, "func1", builder.getFunctionType(std::nullopt, std::nullopt));
33 auto *entryBlock = func.addEntryBlock();
34 builder.setInsertionPointToStart(entryBlock);
35
36 firBuilder = std::make_unique<fir::FirOpBuilder>(builder, *kindMap);
37 }
38
39 fir::FirOpBuilder &getBuilder() { return *firBuilder; }
40
41 mlir::MLIRContext context;
42 mlir::OwningOpRef<mlir::ModuleOp> moduleOp;
43 std::unique_ptr<fir::KindMapping> kindMap;
44 std::unique_ptr<fir::FirOpBuilder> firBuilder;
45};
46
47TEST_F(CharacterTest, smallUtilityFunctions) {
48 auto builder = getBuilder();
49 auto loc = builder.getUnknownLoc();
50 llvm::StringRef strValue("onestringliteral");
51 auto strLit = fir::factory::createStringLiteral(builder, loc, strValue);
52 EXPECT_TRUE(
53 fir::factory::CharacterExprHelper::hasConstantLengthInType(strLit));
54 auto ty = strLit.getCharBox()->getAddr().getType();
55 EXPECT_TRUE(fir::factory::CharacterExprHelper::isCharacterScalar(ty));
56 EXPECT_EQ(fir::factory::CharacterExprHelper::getCharacterOrSequenceKind(ty),
57 fir::factory::CharacterExprHelper::getCharacterKind(ty));
58}
59
60TEST_F(CharacterTest, createSubstring) {
61 auto builder = getBuilder();
62 auto loc = builder.getUnknownLoc();
63 auto charHelper = fir::factory::CharacterExprHelper(builder, loc);
64 llvm::StringRef data("a dummy string to test substring");
65 auto str = fir::factory::createStringLiteral(builder, loc, data);
66 auto lb = builder.createIntegerConstant(loc, builder.getI64Type(), 18);
67 auto ub = builder.createIntegerConstant(loc, builder.getI64Type(), 22);
68 auto substr = charHelper.createSubstring(*str.getCharBox(), {lb, ub});
69 EXPECT_FALSE(
70 fir::factory::CharacterExprHelper::hasConstantLengthInType(substr));
71 EXPECT_FALSE(charHelper.getCharacterType(substr).hasConstantLen());
72 EXPECT_FALSE(fir::factory::CharacterExprHelper::isArray(
73 charHelper.getCharacterType(substr)));
74}
75

source code of flang/unittests/Optimizer/Builder/CharacterTest.cpp