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