从Julia更新C结构的字段值

时间:2016-11-07 04:59:55

标签: c julia ffi

我的问题很简单,但我不知道最好的方法(或者Julia目前还没提供这样的方式):我如何设置C结构的字段值朱莉娅?

想象一下,您有一个结构类型来表示C库中树的节点:

typedef struct node_s
{
    int type;
    node_t* next;
    node_t* children;
    node_t* parent;
} node_t;

并将其复制到朱莉娅:

immutable node_t
    typ::Cint
    next::Ptr{node_t}
    children::Ptr{node_t}
    parent::Ptr{node_t}
end

现在假设您有一个指向C中分配的node_t的指针,并希望更新Julia中的parent字段。我知道我们有unsafe_store!来更新指针指向的值,但计算parent字段的指针偏移量很麻烦(在这种情况下,它会是sizeof(Int) + sizeof(Ptr{node_t}) * 2在我的64位机器上)。有没有更简单的方法来做同样的事情?

1 个答案:

答案 0 :(得分:4)

为此用例提供了fieldoffset函数:

julia> immutable node_t
           typ::Cint
           next::Ptr{node_t}
           children::Ptr{node_t}
           parent::Ptr{node_t}
       end

julia> fieldoffset(node_t, 1)
0x0000000000000000

julia> fieldoffset(node_t, 2)
0x0000000000000008

julia> fieldoffset(node_t, 3)
0x0000000000000010

julia> fieldoffset(node_t, 4)
0x0000000000000018

但也不要担心简单地存储整个immutable,而是改变一个字段;它将被优化掉。

julia> k = Ptr{node_t}(Libc.malloc(sizeof(node_t)))
Ptr{node_t} @0x00000000036194c0

julia> unsafe_load(k)
node_t(29544064,Ptr{node_t} @0x3038662d34363a34,Ptr{node_t} @0x3a386e2d3832313a,Ptr{node_t} @0x34363a32333a3631)

julia> update_parent(x::node_t, n::Ptr{node_t}) =
           node_t(x.typ, x.next, x.children, n)
update_parent (generic function with 1 method)

julia> mutate!(k) = unsafe_store!(k, update_parent(unsafe_load(k), Ptr{node_t}(0)))
mutate! (generic function with 1 method)

julia> @code_llvm mutate!(k)

define %node_t* @"julia_mutate!_70963"(%node_t*) #0 {
top:
  %1 = load %node_t, %node_t* %0, align 1
  %.fca.3.insert = insertvalue %node_t %1, %node_t* null, 3
  store %node_t %.fca.3.insert, %node_t* %0, align 1
  ret %node_t* %0
}