| //===- CharacterTest.cpp -- CharacterExprHelper unit tests ----------------===// |
| // |
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| // See https://llvm.org/LICENSE.txt for license information. |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "flang/Optimizer/Builder/Character.h" |
| #include "gtest/gtest.h" |
| #include "flang/Optimizer/Builder/BoxValue.h" |
| #include "flang/Optimizer/Builder/FIRBuilder.h" |
| #include "flang/Optimizer/Support/InitFIR.h" |
| #include "flang/Optimizer/Support/KindMapping.h" |
| |
| struct CharacterTest : public testing::Test { |
| public: |
| void SetUp() override { |
| fir::KindMapping kindMap(&context, |
| "i10:80,l3:24,a1:8,r54:Double,c20:X86_FP80,r11:PPC_FP128," |
| "r12:FP128,r13:X86_FP80,r14:Double,r15:Float,r16:Half,r23:BFloat"); |
| mlir::OpBuilder builder(&context); |
| auto loc = builder.getUnknownLoc(); |
| |
| // Set up a Module with a dummy function operation inside. |
| // Set the insertion point in the function entry block. |
| mlir::ModuleOp mod = builder.create<mlir::ModuleOp>(loc); |
| mlir::FuncOp func = mlir::FuncOp::create( |
| loc, "func1", builder.getFunctionType(llvm::None, llvm::None)); |
| auto *entryBlock = func.addEntryBlock(); |
| mod.push_back(mod); |
| builder.setInsertionPointToStart(entryBlock); |
| |
| fir::support::loadDialects(context); |
| firBuilder = std::make_unique<fir::FirOpBuilder>(mod, kindMap); |
| } |
| |
| fir::FirOpBuilder &getBuilder() { return *firBuilder; } |
| |
| mlir::MLIRContext context; |
| std::unique_ptr<fir::FirOpBuilder> firBuilder; |
| }; |
| |
| static void checkIntegerConstant(mlir::Value value, mlir::Type ty, int64_t v) { |
| EXPECT_TRUE(mlir::isa<ConstantOp>(value.getDefiningOp())); |
| auto cstOp = dyn_cast<ConstantOp>(value.getDefiningOp()); |
| EXPECT_EQ(ty, cstOp.getType()); |
| auto valueAttr = cstOp.getValue().dyn_cast_or_null<IntegerAttr>(); |
| EXPECT_EQ(v, valueAttr.getInt()); |
| } |
| |
| TEST_F(CharacterTest, smallUtilityFunctions) { |
| auto builder = getBuilder(); |
| auto loc = builder.getUnknownLoc(); |
| llvm::StringRef strValue("onestringliteral"); |
| auto strLit = fir::factory::createStringLiteral(builder, loc, strValue); |
| EXPECT_TRUE( |
| fir::factory::CharacterExprHelper::hasConstantLengthInType(strLit)); |
| auto ty = strLit.getCharBox()->getAddr().getType(); |
| EXPECT_TRUE(fir::factory::CharacterExprHelper::isCharacterScalar(ty)); |
| EXPECT_EQ(fir::factory::CharacterExprHelper::getCharacterOrSequenceKind(ty), |
| fir::factory::CharacterExprHelper::getCharacterKind(ty)); |
| } |
| |
| TEST_F(CharacterTest, createConcatenate) { |
| auto builder = getBuilder(); |
| auto loc = builder.getUnknownLoc(); |
| auto charHelper = fir::factory::CharacterExprHelper(builder, loc); |
| llvm::StringRef lhs("rightsideofconcat"); |
| llvm::StringRef rhs("leftsideofconcat"); |
| auto strLitLhs = fir::factory::createStringLiteral(builder, loc, lhs); |
| auto strLitRhs = fir::factory::createStringLiteral(builder, loc, rhs); |
| auto concat = charHelper.createConcatenate( |
| *strLitRhs.getCharBox(), *strLitLhs.getCharBox()); |
| EXPECT_TRUE(mlir::isa<arith::AddIOp>(concat.getLen().getDefiningOp())); |
| auto addOp = dyn_cast<arith::AddIOp>(concat.getLen().getDefiningOp()); |
| EXPECT_TRUE(mlir::isa<ConstantOp>(addOp.lhs().getDefiningOp())); |
| auto lhsCstOp = dyn_cast<ConstantOp>(addOp.lhs().getDefiningOp()); |
| EXPECT_TRUE(mlir::isa<ConstantOp>(addOp.rhs().getDefiningOp())); |
| auto rhsCstOp = dyn_cast<ConstantOp>(addOp.rhs().getDefiningOp()); |
| checkIntegerConstant(lhsCstOp, builder.getCharacterLengthType(), 16); |
| checkIntegerConstant(rhsCstOp, builder.getCharacterLengthType(), 17); |
| } |
| |
| TEST_F(CharacterTest, createSubstring) { |
| auto builder = getBuilder(); |
| auto loc = builder.getUnknownLoc(); |
| auto charHelper = fir::factory::CharacterExprHelper(builder, loc); |
| llvm::StringRef data("a dummy string to test substring"); |
| auto str = fir::factory::createStringLiteral(builder, loc, data); |
| auto lb = builder.createIntegerConstant(loc, builder.getI64Type(), 18); |
| auto ub = builder.createIntegerConstant(loc, builder.getI64Type(), 22); |
| auto substr = charHelper.createSubstring(*str.getCharBox(), {lb, ub}); |
| EXPECT_FALSE( |
| fir::factory::CharacterExprHelper::hasConstantLengthInType(substr)); |
| EXPECT_FALSE(charHelper.getCharacterType(substr).hasConstantLen()); |
| EXPECT_FALSE(fir::factory::CharacterExprHelper::isArray( |
| charHelper.getCharacterType(substr))); |
| } |