1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
|
#include "mkql_toindexdict.h"
#include <yql/essentials/minikql/computation/mkql_computation_node_holders.h>
#include <yql/essentials/minikql/computation/mkql_computation_node_codegen.h> // Y_IGNORE
#include <yql/essentials/minikql/mkql_node_cast.h>
namespace NKikimr {
namespace NMiniKQL {
namespace {
class TToIndexDictWrapper : public TMutableCodegeneratorNode<TToIndexDictWrapper> {
typedef TMutableCodegeneratorNode<TToIndexDictWrapper> TBaseComputation;
public:
TToIndexDictWrapper(TComputationMutables& mutables, IComputationNode* list)
: TBaseComputation(mutables, list->GetRepresentation())
, List(list)
{
}
NUdf::TUnboxedValuePod DoCalculate(TComputationContext& ctx) const {
return ctx.HolderFactory.ToIndexDict(ctx.Builder, List->GetValue(ctx).Release());
}
#ifndef MKQL_DISABLE_CODEGEN
Value* DoGenerateGetValue(const TCodegenContext& ctx, BasicBlock*& block) const {
auto& context = ctx.Codegen.GetContext();
const auto indexType = Type::getInt32Ty(context);
const auto first = GetElementPtrInst::CreateInBounds(GetCompContextType(context), ctx.Ctx, {ConstantInt::get(indexType, 0), ConstantInt::get(indexType, 0)}, "first", block);
const auto fourth = GetElementPtrInst::CreateInBounds(GetCompContextType(context), ctx.Ctx, {ConstantInt::get(indexType, 0), ConstantInt::get(indexType, 3)}, "fourth", block);
const auto structPtrType = PointerType::getUnqual(StructType::get(context));
const auto factory = new LoadInst(structPtrType, first, "factory", block);
const auto builder = new LoadInst(structPtrType, fourth, "builder", block);
const auto func = ConstantInt::get(Type::getInt64Ty(context), GetMethodPtr(&THolderFactory::ToIndexDict));
const auto list = GetNodeValue(List, ctx, block);
const auto funType = FunctionType::get(list->getType(), {factory->getType(), builder->getType(), list->getType()}, false);
const auto funcPtr = CastInst::Create(Instruction::IntToPtr, func, PointerType::getUnqual(funType), "function", block);
const auto result = CallInst::Create(funType, funcPtr, {factory, builder, list}, "result", block);
return result;
}
#endif
private:
void RegisterDependencies() const final {
DependsOn(List);
}
IComputationNode* const List;
};
}
IComputationNode* WrapToIndexDict(TCallable& callable, const TComputationNodeFactoryContext& ctx) {
MKQL_ENSURE(callable.GetInputsCount() == 1, "Expected 1 args");
return new TToIndexDictWrapper(ctx.Mutables, LocateNode(ctx.NodeLocator, callable, 0));
}
}
}
|
