op_confine()

GpuVisionPass::Completed MayaFlux::Yantra::VisionGpuExecutor::op_confine ( VisionGpuContexts &  contexts, const Kinesis::Vision::ConfineParams &  p  )

private

Arbitrary rectangular sub-region of the current image, optionally resized back to the frame's own size before this step.

Publishes the result as VisionResult::images.confine.

Returns

The resulting image and the image it was derived from.

Definition at line 114 of file VisionImageDispatch.cpp.

{
    auto& pixel_ctx = contexts.pixel;
    auto w = contexts.pass.w;
    auto h = contexts.pass.h;
    auto& foundry = Portal::Graphics::get_shader_foundry();
 
    const auto confine_input = contexts.pass.current;
 
    const auto src_x = static_cast<uint32_t>(p.bounds.x * static_cast<float>(w));
    const auto src_y = static_cast<uint32_t>(p.bounds.y * static_cast<float>(h));
    const auto src_w = std::max<uint32_t>(1U, static_cast<uint32_t>(p.bounds.w * static_cast<float>(w)));
    const auto src_h = std::max<uint32_t>(1U, static_cast<uint32_t>(p.bounds.h * static_cast<float>(h)));
 
    const uint32_t out_w = p.resize_to_source ? w : src_w;
    const uint32_t out_h = p.resize_to_source ? h : src_h;
 
    pixel_ctx.swap_shader({
        .shader_path = "vision_confine.comp.spv",
        .workgroup_size = k_wg2d,
        .push_constant_size = sizeof(ConfinePC),
    });
    pixel_ctx.stage_image(confine_input);
    pixel_ctx.set_push_constants(ConfinePC {
        .src_x = src_x, .src_y = src_y, .src_w = src_w, .src_h = src_h,
        .out_w = out_w, .out_h = out_h });
    pixel_ctx.prepare_output_image(out_w, out_h);
    pixel_ctx.set_output_dimensions(out_w, out_h);
 
    const auto fence = pixel_ctx.dispatch_async({});
    pixel_ctx.clear_output_dimensions();
    foundry.wait_for_fence(fence);
    foundry.release_fence(fence);
 
    auto confined = pixel_ctx.get_output_image(0);
 
    contexts.pass.result.images.confine = confined;
    contexts.pass.current = confined;
    contexts.pass.result.structured = std::monostate {};
    contexts.bound_staged.reset();
 
    contexts.pass.set_geometry(out_w, out_h);
    contexts.pass.storage_w = out_w;
    contexts.pass.storage_h = out_h;
 
    return { .output = confined, .input = confine_input };
}

References MayaFlux::Yantra::VisionGpuContexts::bound_staged, MayaFlux::Kinesis::Vision::ConfineParams::bounds, MayaFlux::Kinesis::Vision::ImageOutputs::confine, MayaFlux::Kinesis::Vision::VisionPass< Handle >::current, fence, MayaFlux::Portal::Graphics::get_shader_foundry(), MayaFlux::Kinesis::Vision::BoundingBox::h, MayaFlux::Kinesis::Vision::VisionPass< Handle >::h, h, MayaFlux::Kinesis::Vision::VisionResult::images, MayaFlux::Yantra::VisionInternal::k_wg2d, out_h, out_w, MayaFlux::Yantra::VisionGpuContexts::pass, MayaFlux::Yantra::VisionGpuContexts::pixel, MayaFlux::Kinesis::Vision::ConfineParams::resize_to_source, MayaFlux::Kinesis::Vision::VisionPass< Handle >::result, MayaFlux::Kinesis::Vision::VisionPass< Handle >::set_geometry(), src_h, src_w, MayaFlux::Yantra::VisionInternal::ConfinePC::src_x, src_x, src_y, MayaFlux::Kinesis::Vision::VisionPass< Handle >::storage_h, MayaFlux::Kinesis::Vision::VisionPass< Handle >::storage_w, MayaFlux::Kinesis::Vision::VisionResult::structured, MayaFlux::Kinesis::Vision::BoundingBox::w, MayaFlux::Kinesis::Vision::VisionPass< Handle >::w, MayaFlux::Kinesis::Vision::BoundingBox::x, and MayaFlux::Kinesis::Vision::BoundingBox::y.

Here is the call graph for this function: