cached_container_gpu.cpp

Go to the documentation of this file.

/*
 * This program source code file is part of KiCad, a free EDA CAD application.
 *
 * Copyright 2013-2017 CERN
 * Copyright The KiCad Developers, see AUTHORS.txt for contributors.
 *
 * @author Maciej Suminski <[email protected]>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */
 
#include <gal/opengl/cached_container_gpu.h>
#include <gal/opengl/gpu_oom_error.h>
#include <gal/opengl/vertex_manager.h>
#include <gal/opengl/vertex_item.h>
#include <gal/opengl/shader.h>
#include <gal/opengl/utils.h>
 
#include <wx/log.h>
 
#include <cstring>
#include <list>
#include <memory>
 
#include <core/profile.h>
#include <trace_helpers.h>
 
using namespace KIGFX;

static const wxChar* const traceGalCachedContainerGpu = wxT( "KICAD_GAL_CACHED_CONTAINER_GPU" );
 
 
CACHED_CONTAINER_GPU::CACHED_CONTAINER_GPU( unsigned int aSize ) :
        CACHED_CONTAINER( aSize ),
        m_isMapped( false ),
        m_glBufferHandle( -1 )
{
    m_useCopyBuffer = !!GLAD_GL_ARB_copy_buffer;
 
    wxString vendor( glGetString( GL_VENDOR ) );
 
    // workaround for intel GPU drivers:
    // disable glCopyBuffer, causes crashes/freezes on certain driver versions
    // Note, Intel's GL_VENDOR string varies depending on GPU/driver generation
    // But generally always starts with Intel at least
    if( vendor.StartsWith( "Intel" ) || vendor.Contains( "etnaviv" ) )
    {
        m_useCopyBuffer = false;
    }
 
#ifdef KICAD_GAL_PROFILE
    wxLogTrace( traceGalProfile, "VBO initial size: %u", m_currentSize );
#endif
 
    glGenBuffers( 1, &m_glBufferHandle );
    glBindBuffer( GL_ARRAY_BUFFER, m_glBufferHandle );
    glBufferData( GL_ARRAY_BUFFER, m_currentSize * VERTEX_SIZE, nullptr, GL_DYNAMIC_DRAW );
    glBindBuffer( GL_ARRAY_BUFFER, 0 );
    checkGlError( "allocating video memory for cached container", __FILE__, __LINE__ );
}
 
 
void CACHED_CONTAINER_GPU::Abandon()
{
    // Name 0 is silently ignored by glDeleteBuffers, so the destructor becomes a no-op
    m_isMapped = false;
    m_glBufferHandle = 0;
}
 
 
CACHED_CONTAINER_GPU::~CACHED_CONTAINER_GPU()
{
    if( m_isMapped )
        Unmap();
 
    if( glDeleteBuffers )
        glDeleteBuffers( 1, &m_glBufferHandle );
}
 
 
void CACHED_CONTAINER_GPU::Map()
{
    wxCHECK( !IsMapped(), /*void*/ );
 
    // OpenGL version might suddenly stop being available in Windows when an RDP session is started
    if( !glBindBuffer )
        throw std::runtime_error( "OpenGL no longer available!" );
 
    glBindBuffer( GL_ARRAY_BUFFER, m_glBufferHandle );
    m_vertices = static_cast<VERTEX*>( glMapBuffer( GL_ARRAY_BUFFER, GL_READ_WRITE ) );
 
    if( checkGlError( "mapping vertices buffer", __FILE__, __LINE__ ) == GL_NO_ERROR
        && m_vertices != nullptr )
    {
        m_isMapped = true;
    }
    else
    {
        m_vertices = nullptr;
        glBindBuffer( GL_ARRAY_BUFFER, 0 );
        throw std::runtime_error( "Could not map vertex buffer: glMapBuffer returned null" );
    }
}
 
 
void CACHED_CONTAINER_GPU::Unmap()
{
    wxCHECK( IsMapped(), /*void*/ );
 
    // This gets called from ~CACHED_CONTAINER_GPU.  To avoid throwing an exception from
    // the dtor, catch it here instead.
    try
    {
        glUnmapBuffer( GL_ARRAY_BUFFER );
        checkGlError( "unmapping vertices buffer", __FILE__, __LINE__ );
        glBindBuffer( GL_ARRAY_BUFFER, 0 );
        m_vertices = nullptr;
        checkGlError( "unbinding vertices buffer", __FILE__, __LINE__ );
    }
    catch( const std::runtime_error& err )
    {
        wxLogError( wxT( "OpenGL did not shut down properly.\n\n%s" ), err.what() );
    }
 
    m_isMapped = false;
}
 
 
bool CACHED_CONTAINER_GPU::defragmentResize( unsigned int aNewSize )
{
    // A doubling resize transiently holds the old and the new buffer in video memory at the
    // same time.  On a large board this peak can exceed the driver's budget and trip a fatal
    // out-of-memory abort (e.g. NVIDIA "Error code: 6"), which kills the process before any
    // GL error can be observed.
    const size_t oldBytes = static_cast<size_t>( m_currentSize ) * VERTEX_SIZE;
    const size_t newBytes = static_cast<size_t>( aNewSize ) * VERTEX_SIZE;
 
    switch( KIGFX::chooseResizeStrategy( KIGFX::queryFreeVideoMemoryBytes(), oldBytes, newBytes, 0.15 ) )
    {
    case KIGFX::VRAM_RESIZE_STRATEGY::REFUSE:
        throw KIGFX::GPU_OOM_ERROR( "Insufficient GPU memory to render this board; switching to software rendering." );
 
    case KIGFX::VRAM_RESIZE_STRATEGY::RAM_STAGE:
        return defragmentResizeStaged( aNewSize );
 
    case KIGFX::VRAM_RESIZE_STRATEGY::GPU_COPY:
        break;
    }
 
    if( !m_useCopyBuffer )
        return defragmentResizeMemcpy( aNewSize );
 
    wxCHECK( IsMapped(), false );
 
    wxLogTrace( traceGalCachedContainerGpu,
                wxT( "Resizing & defragmenting container from %d to %d" ), m_currentSize,
                aNewSize );
 
    // No shrinking if we cannot fit all the data
    if( usedSpace() > aNewSize )
        return false;
 
#ifdef KICAD_GAL_PROFILE
    PROF_TIMER totalTime;
#endif /* KICAD_GAL_PROFILE */
 
    GLuint newBuffer;
 
    // glCopyBufferSubData requires a buffer to be unmapped
    glUnmapBuffer( GL_ARRAY_BUFFER );
 
    // Create a new destination buffer
    glGenBuffers( 1, &newBuffer );
 
    // It would be best to use GL_COPY_WRITE_BUFFER here,
    // but it is not available everywhere
#ifdef KICAD_GAL_PROFILE
    GLint eaBuffer = -1;
    glGetIntegerv( GL_ELEMENT_ARRAY_BUFFER_BINDING, &eaBuffer );
    wxASSERT( eaBuffer == 0 );
#endif /* KICAD_GAL_PROFILE */
    glBindBuffer( GL_ELEMENT_ARRAY_BUFFER, newBuffer );
    glBufferData( GL_ELEMENT_ARRAY_BUFFER, aNewSize * VERTEX_SIZE, nullptr, GL_DYNAMIC_DRAW );
    checkGlError( "creating buffer during defragmentation", __FILE__, __LINE__ );
 
    std::set<VERTEX_ITEM*>::iterator it, it_end;
    int                              newOffset = 0;
 
    // Defragmentation
    for( it = m_items.begin(), it_end = m_items.end(); it != it_end; ++it )
    {
        VERTEX_ITEM* item = *it;
        int          itemOffset = item->GetOffset();
        int          itemSize = item->GetSize();
 
        // Move an item to the new container
        glCopyBufferSubData( GL_ARRAY_BUFFER, GL_ELEMENT_ARRAY_BUFFER, itemOffset * VERTEX_SIZE,
                             newOffset * VERTEX_SIZE, itemSize * VERTEX_SIZE );
 
        // Update new offset
        item->setOffset( newOffset );
 
        // Move to the next free space
        newOffset += itemSize;
    }
 
    // Move the current item and place it at the end
    if( m_item->GetSize() > 0 )
    {
        glCopyBufferSubData( GL_ARRAY_BUFFER, GL_ELEMENT_ARRAY_BUFFER,
                             m_item->GetOffset() * VERTEX_SIZE, newOffset * VERTEX_SIZE,
                             m_item->GetSize() * VERTEX_SIZE );
 
        m_item->setOffset( newOffset );
        m_chunkOffset = newOffset;
    }
 
    // Cleanup
    glBindBuffer( GL_ELEMENT_ARRAY_BUFFER, 0 );
    glBindBuffer( GL_ARRAY_BUFFER, 0 );
 
    // Previously we have unmapped the array buffer, now when it is also
    // unbound, it may be officially marked as unmapped
    m_isMapped = false;
    glDeleteBuffers( 1, &m_glBufferHandle );
 
    // Switch to the new vertex buffer
    m_glBufferHandle = newBuffer;
 
    try
    {
        Map();
    }
    catch( const std::runtime_error& )
    {
        // Map() failed, likely due to glMapBuffer returning null.
        // The buffer is valid but we can't map it.
        return false;
    }
 
    checkGlError( "switching buffers during defragmentation", __FILE__, __LINE__ );
 
#ifdef KICAD_GAL_PROFILE
    totalTime.Stop();
 
    wxLogTrace( traceGalCachedContainerGpu, "Defragmented container storing %d vertices / %.1f ms",
                m_currentSize - m_freeSpace, totalTime.msecs() );
#endif /* KICAD_GAL_PROFILE */
 
    m_freeSpace += ( aNewSize - m_currentSize );
    m_currentSize = aNewSize;
 
    wxLogTrace( traceGalProfile, "VBO size %d used %d", m_currentSize, AllItemsSize() );
 
    // Now there is only one big chunk of free memory
    m_freeChunks.clear();
    m_freeChunks.insert( std::make_pair( m_freeSpace, m_currentSize - m_freeSpace ) );
 
    return true;
}
 
 
bool CACHED_CONTAINER_GPU::defragmentResizeMemcpy( unsigned int aNewSize )
{
    wxCHECK( IsMapped(), false );
 
    wxLogTrace( traceGalCachedContainerGpu,
                wxT( "Resizing & defragmenting container (memcpy) from %d to %d" ), m_currentSize,
                aNewSize );
 
    // No shrinking if we cannot fit all the data
    if( usedSpace() > aNewSize )
        return false;
 
#ifdef KICAD_GAL_PROFILE
    PROF_TIMER totalTime;
#endif /* KICAD_GAL_PROFILE */
 
    GLuint  newBuffer;
    VERTEX* newBufferMem;
 
    // Create the destination buffer
    glGenBuffers( 1, &newBuffer );
 
    // It would be best to use GL_COPY_WRITE_BUFFER here,
    // but it is not available everywhere
#ifdef KICAD_GAL_PROFILE
    GLint eaBuffer = -1;
    glGetIntegerv( GL_ELEMENT_ARRAY_BUFFER_BINDING, &eaBuffer );
    wxASSERT( eaBuffer == 0 );
#endif /* KICAD_GAL_PROFILE */
 
    glBindBuffer( GL_ELEMENT_ARRAY_BUFFER, newBuffer );
    glBufferData( GL_ELEMENT_ARRAY_BUFFER, aNewSize * VERTEX_SIZE, nullptr, GL_DYNAMIC_DRAW );
    newBufferMem = static_cast<VERTEX*>( glMapBuffer( GL_ELEMENT_ARRAY_BUFFER, GL_WRITE_ONLY ) );
    checkGlError( "creating buffer during defragmentation", __FILE__, __LINE__ );
 
    if( newBufferMem == nullptr )
    {
        glBindBuffer( GL_ELEMENT_ARRAY_BUFFER, 0 );
        glDeleteBuffers( 1, &newBuffer );
        return false;
    }
 
    defragment( newBufferMem );
 
    // Cleanup
    glUnmapBuffer( GL_ELEMENT_ARRAY_BUFFER );
    glBindBuffer( GL_ELEMENT_ARRAY_BUFFER, 0 );
    Unmap();
    glDeleteBuffers( 1, &m_glBufferHandle );
 
    // Switch to the new vertex buffer
    m_glBufferHandle = newBuffer;
 
    try
    {
        Map();
    }
    catch( const std::runtime_error& )
    {
        // Map() failed, likely due to glMapBuffer returning null.
        // The buffer is valid but we can't map it.
        return false;
    }
 
    checkGlError( "switching buffers during defragmentation", __FILE__, __LINE__ );
 
#ifdef KICAD_GAL_PROFILE
    totalTime.Stop();
 
    wxLogTrace( traceGalCachedContainerGpu, "Defragmented container storing %d vertices / %.1f ms",
                m_currentSize - m_freeSpace, totalTime.msecs() );
#endif /* KICAD_GAL_PROFILE */
 
    m_freeSpace += ( aNewSize - m_currentSize );
    m_currentSize = aNewSize;
 
    wxLogTrace( traceGalProfile, "VBO size %d used: %d", m_currentSize, AllItemsSize() );
 
    // Now there is only one big chunk of free memory
    m_freeChunks.clear();
    m_freeChunks.insert( std::make_pair( m_freeSpace, m_currentSize - m_freeSpace ) );
 
    return true;
}
 
 
bool CACHED_CONTAINER_GPU::defragmentResizeStaged( unsigned int aNewSize )
{
    wxCHECK( IsMapped(), false );
 
    wxLogTrace( traceGalCachedContainerGpu,
                wxT( "Resizing & defragmenting container (RAM staged) from %d to %d" ),
                m_currentSize, aNewSize );
 
    // No shrinking if we cannot fit all the data
    if( usedSpace() > aNewSize )
        return false;
 
    const unsigned int usedVerts = usedSpace();
 
    // Stage the compacted vertices in host memory so the old video buffer can be released
    // before the larger replacement is allocated, keeping the peak VRAM at max(old, new).
    std::unique_ptr<VERTEX[]> staging;
 
    try
    {
        staging.reset( new VERTEX[usedVerts] );
    }
    catch( const std::bad_alloc& )
    {
        throw GPU_OOM_ERROR( "Out of memory while staging a GPU buffer resize; "
                             "switching to software rendering." );
    }
 
    // Reads from the mapped old buffer, so it must run before that buffer is released.
    defragment( staging.get() );
 
    Unmap();
    glDeleteBuffers( 1, &m_glBufferHandle );
 
    GLuint newBuffer;
    glGenBuffers( 1, &newBuffer );
    glBindBuffer( GL_ARRAY_BUFFER, newBuffer );
    glBufferData( GL_ARRAY_BUFFER, aNewSize * VERTEX_SIZE, nullptr, GL_DYNAMIC_DRAW );
    glBindBuffer( GL_ARRAY_BUFFER, 0 );
    checkGlError( "allocating staged buffer during defragmentation", __FILE__, __LINE__ );
 
    m_glBufferHandle = newBuffer;
 
    try
    {
        Map();
    }
    catch( const std::runtime_error& )
    {
        // Map() failed, likely due to glMapBuffer returning null.
        return false;
    }
 
    if( usedVerts > 0 )
        memcpy( m_vertices, staging.get(), usedVerts * VERTEX_SIZE );
 
    checkGlError( "switching buffers during staged defragmentation", __FILE__, __LINE__ );
 
    m_freeSpace += ( aNewSize - m_currentSize );
    m_currentSize = aNewSize;
 
    wxLogTrace( traceGalProfile, "VBO size %d used: %d", m_currentSize, AllItemsSize() );
 
    // Now there is only one big chunk of free memory
    m_freeChunks.clear();
    m_freeChunks.insert( std::make_pair( m_freeSpace, m_currentSize - m_freeSpace ) );
 
    return true;
}
 
 
unsigned int CACHED_CONTAINER_GPU::AllItemsSize() const
{
    unsigned int size = 0;
 
    for( const auto& item : m_items )
    {
        size += item->GetSize();
    }
 
    return size;
}