on_attach()

void MayaFlux::Buffers::RelaxationStepProcessor::on_attach ( const std::shared_ptr< Buffer > &  buffer )

overrideprotectedvirtual

Cover the grid with a 2D dispatch and write GridExtent.

Parameters

buffer

The attached buffer, expected to be a RelaxationGridBuffer.

The local size is fixed at 16x16x1 and the group counts are ceil(width / 16) by ceil(height / 16). The workgroup_x constructor argument and ShaderSpec::workgroup_size are not consulted once attached to a grid.

Reimplemented from MayaFlux::Buffers::BufferProcessor.

Definition at line 91 of file RelaxationStepProcessor.cpp.

{
    ComputeProcessor::on_attach(buffer);
    m_grid = std::dynamic_pointer_cast<RelaxationGridBuffer>(buffer);
 
    if (m_grid) {
        set_workgroup_size(16, 16, 1);
        set_dispatch_mode(ShaderDispatchConfig::DispatchMode::MANUAL);
        uint32_t gx = (m_grid->get_grid_width() + 15) / 16;
        uint32_t gy = (m_grid->get_grid_height() + 15) / 16;
        set_manual_dispatch(gx, gy, 1);
        write_grid_extent_constants();
    }
}

References buffer, m_grid, MayaFlux::Buffers::ShaderDispatchConfig::MANUAL, MayaFlux::Buffers::BufferProcessor::on_attach(), MayaFlux::Buffers::ComputeProcessor::set_dispatch_mode(), MayaFlux::Buffers::ComputeProcessor::set_manual_dispatch(), MayaFlux::Buffers::ComputeProcessor::set_workgroup_size(), and write_grid_extent_constants().

Here is the call graph for this function: