1 //===-- Ragged.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 #include "flang/Optimizer/Builder/Runtime/Ragged.h"
10 #include "flang/Optimizer/Builder/FIRBuilder.h"
11 #include "flang/Optimizer/Builder/Runtime/RTBuilder.h"
12 #include "flang/Runtime/ragged.h"
13 
14 using namespace Fortran::runtime;
15 
16 void fir::runtime::genRaggedArrayAllocate(mlir::Location loc,
17                                           fir::FirOpBuilder &builder,
18                                           mlir::Value header, bool asHeaders,
19                                           mlir::Value eleSize,
20                                           mlir::ValueRange extents) {
21   auto i32Ty = builder.getIntegerType(32);
22   auto rank = extents.size();
23   auto i64Ty = builder.getIntegerType(64);
24   auto func =
25       fir::runtime::getRuntimeFunc<mkRTKey(RaggedArrayAllocate)>(loc, builder);
26   auto fTy = func.getType();
27   auto i1Ty = builder.getIntegerType(1);
28   fir::SequenceType::Shape shape = {
29       static_cast<fir::SequenceType::Extent>(rank)};
30   auto extentTy = fir::SequenceType::get(shape, i64Ty);
31   auto refTy = fir::ReferenceType::get(i64Ty);
32   // Position of the bufferPointer in the header struct.
33   auto one = builder.createIntegerConstant(loc, i32Ty, 1);
34   auto eleTy = fir::unwrapSequenceType(fir::unwrapRefType(header.getType()));
35   auto ptrTy = builder.getRefType(eleTy.cast<mlir::TupleType>().getType(1));
36   auto ptr = builder.create<fir::CoordinateOp>(loc, ptrTy, header, one);
37   auto heap = builder.create<fir::LoadOp>(loc, ptr);
38   auto cmp = builder.genIsNull(loc, heap);
39   builder.genIfThen(loc, cmp)
40       .genThen([&]() {
41         auto asHeadersVal = builder.createIntegerConstant(loc, i1Ty, asHeaders);
42         auto rankVal = builder.createIntegerConstant(loc, i64Ty, rank);
43         auto buff = builder.create<fir::AllocMemOp>(loc, extentTy);
44         // Convert all the extents to i64 and pack them in a buffer on the heap.
45         for (auto i : llvm::enumerate(extents)) {
46           auto offset = builder.createIntegerConstant(loc, i32Ty, i.index());
47           auto addr =
48               builder.create<fir::CoordinateOp>(loc, refTy, buff, offset);
49           auto castVal = builder.createConvert(loc, i64Ty, i.value());
50           builder.create<fir::StoreOp>(loc, castVal, addr);
51         }
52         auto args = fir::runtime::createArguments(
53             builder, loc, fTy, header, asHeadersVal, rankVal, eleSize, buff);
54         builder.create<fir::CallOp>(loc, func, args);
55       })
56       .end();
57 }
58 
59 void fir::runtime::genRaggedArrayDeallocate(mlir::Location loc,
60                                             fir::FirOpBuilder &builder,
61                                             mlir::Value header) {
62   auto func = fir::runtime::getRuntimeFunc<mkRTKey(RaggedArrayDeallocate)>(
63       loc, builder);
64   auto fTy = func.getType();
65   auto args = fir::runtime::createArguments(builder, loc, fTy, header);
66   builder.create<fir::CallOp>(loc, func, args);
67 }
68