diff options
-rw-r--r-- | julia/CrystFEL/src/image.jl | 15 | ||||
-rw-r--r-- | julia/partial_sim.jl | 4 |
2 files changed, 17 insertions, 2 deletions
diff --git a/julia/CrystFEL/src/image.jl b/julia/CrystFEL/src/image.jl index 020117e7..33af3252 100644 --- a/julia/CrystFEL/src/image.jl +++ b/julia/CrystFEL/src/image.jl @@ -217,6 +217,21 @@ function Base.push!(image::Image, cr::Crystal) end +function Base.push!(image::Image, cr::Crystal, reflections::RefList{UnmergedReflection}) + @ccall libcrystfel.image_add_crystal_refls(image.internalptr::Ptr{InternalImage}, + cr.internalptr::Ptr{InternalCrystal}, + reflections.internalptr::Ptr{InternalRefList})::Cvoid + idata = unsafe_load(image.internalptr) + ncryst = idata.n_crystals + pairptr = unsafe_load(idata.crystals, ncryst) + pairptr.owns_crystal = 0 + pairptr.owns_reflist = 0 + unsafe_store!(idata.crystals, pairptr, ncryst) + push!(getfield(image, :crystals), cr) + push!(getfield(image, :reflists), reflections) +end + + """ Image(dtempl::DataTemplate) diff --git a/julia/partial_sim.jl b/julia/partial_sim.jl index 4ddec347..11384f47 100644 --- a/julia/partial_sim.jl +++ b/julia/partial_sim.jl @@ -10,8 +10,8 @@ let st = Stream("partials.stream", "w", dtempl) image.filename = "simulation_" * string(i) image.ev = "//" cr = Crystal(rotatecell(cell)) - cr.reflections = predictreflections(cr, image) - push!(image, cr) + reflections = predictreflections(cr, image) + push!(image, cr, reflections) chunkwrite(st, image) end end |