GeometryStage.h

Go to the documentation of this file.

/*
 *  Copyright (C) 2010 - 2015 Leonid Kostrykin
 *
 *  Chair of Medical Engineering (mediTEC)
 *  RWTH Aachen University
 *  Pauwelsstr. 20
 *  52074 Aachen
 *  Germany
 *
 */

#ifndef GEOMETRYSTAGE_H_6014714286
#define GEOMETRYSTAGE_H_6014714286

#include <Carna/base/FrameRenderer.h>
#include <Carna/base/RenderStage.h>
#include <Carna/base/RenderQueue.h>
#include <Carna/base/GeometryFeature.h>
#include <Carna/base/math.h>
#include <memory>
#include <map>

namespace Carna
{

namespace base
{



// ----------------------------------------------------------------------------------
// GeometryStage
// ----------------------------------------------------------------------------------

template< typename RenderableCompare >
class GeometryStage : public RenderStage
{

    typedef GeometryFeature::ManagedInterface VideoResource;

    Node* root;
    std::size_t passesRendered;
    std::map< GeometryFeature*, VideoResource* > acquiredFeatures;

protected:

    RenderQueue< RenderableCompare > rq;

public:

    GeometryStage
        ( unsigned int geometryType
        , unsigned int geometryTypeMask = RenderQueue< RenderableCompare >::EXACT_MATCH_GEOMETRY_TYPE_MASK );
        
    const unsigned int geometryType;     
    const unsigned int geometryTypeMask; 

    virtual ~GeometryStage();

    virtual void prepareFrame( Node& root ) override;

    virtual void renderPass( const math::Matrix4f& viewTransform, RenderTask& rt, const Viewport& vp ) override;

    std::size_t renderedPassesCount() const;
    
    template< typename GeometryFeatureType >
    typename GeometryFeatureType::ManagedInterface& videoResource( GeometryFeatureType& geometryFeature ) const;
    
    template< typename GeometryFeatureType >
    const typename GeometryFeatureType::ManagedInterface& videoResource( const GeometryFeatureType& geometryFeature ) const;

protected:

    void activateGLContext() const;
    
    virtual void buildRenderQueues( Node& root, const math::Matrix4f& viewTransform );

    virtual void rewindRenderQueues();

    virtual void updateRenderQueues( const math::Matrix4f& viewTransform );

    virtual void render( const Renderable& renderable ) = 0;

}; // GeometryStage


template< typename RenderableCompare >
GeometryStage< RenderableCompare >::GeometryStage( unsigned int geometryType, unsigned int geometryTypeMask )
    : root( nullptr )
    , passesRendered( 0 )
    , rq( geometryType, geometryTypeMask )
    , geometryType( geometryType )
    , geometryTypeMask( geometryTypeMask )
{
}


template< typename RenderableCompare >
GeometryStage< RenderableCompare >::~GeometryStage()
{
    activateGLContext();
    std::for_each( acquiredFeatures.begin(), acquiredFeatures.end(),
        [&]( const std::pair< GeometryFeature*, VideoResource* >& entry )
        {
            if( entry.second != nullptr )
            {
                delete entry.second;
            }
        }
    );
}


template< typename RenderableCompare >
void GeometryStage< RenderableCompare >::activateGLContext() const
{
    if( isInitialized() )
    {
        renderer().glContext().makeCurrent();
    }
}


template< typename RenderableCompare >
void GeometryStage< RenderableCompare >::prepareFrame( Node& root )
{
    RenderStage::prepareFrame( root );
    this->root = &root;
    this->passesRendered = 0;
}


template< typename RenderableCompare >
std::size_t GeometryStage< RenderableCompare >::renderedPassesCount() const
{
    return passesRendered;
}


template< typename RenderableCompare >
void GeometryStage< RenderableCompare >::buildRenderQueues( Node& root, const math::Matrix4f& viewTransform )
{
    rq.build( root, viewTransform );
}


template< typename RenderableCompare >
void GeometryStage< RenderableCompare >::rewindRenderQueues()
{
    rq.rewind();
}


template< typename RenderableCompare >
void GeometryStage< RenderableCompare >::updateRenderQueues( const math::Matrix4f& viewTransform )
{
    rq.updateModelViewTransforms( viewTransform );
}


template< typename RenderableCompare >
void GeometryStage< RenderableCompare >::renderPass( const math::Matrix4f& viewTransform, RenderTask& rt, const Viewport& vp )
{
    const bool isFirstPass = passesRendered == 0;
    
    /* Maintain the render queues.
     */
    if( isFirstPass )
    {
        buildRenderQueues( *root, viewTransform );
    }
    else
    {
        rewindRenderQueues();
        if( isViewTransformFixed() )
        {
            updateRenderQueues( viewTransform );
        }
    }

    std::set< GeometryFeature* > usedFeatures;
    while( !rq.isEmpty() )
    {
        const Renderable& renderable = rq.poll();
        if( isFirstPass )
        {
            renderable.geometry().visitFeatures( [&]( GeometryFeature& gf, unsigned int role )
                {
                    /* Denote that the geometry feature was used.
                     */
                    usedFeatures.insert( &gf );

                    /* Check whether video resources need to be acquired.
                     */
                    if( acquiredFeatures.find( &gf ) == acquiredFeatures.end() )
                    {
                        VideoResource* const vr = gf.acquireVideoResource();
                        acquiredFeatures[ &gf ] = vr;
                    }
                }
            );
        }
        render( renderable );
    }
    if( isFirstPass )
    {
        /* Release unused video resources.
         */
        for( auto itr = acquiredFeatures.begin(); itr != acquiredFeatures.end(); )
        {
            if( usedFeatures.find( itr->first ) == usedFeatures.end() )
            {
                if( itr->second != nullptr )
                {
                    delete itr->second;
                }
                acquiredFeatures.erase( itr++ );
            }
            else
            {
                ++itr;
            }
        }
    }
}


template< typename RenderableCompare >
template< typename GeometryFeatureType >
typename GeometryFeatureType::ManagedInterface& GeometryStage< RenderableCompare >
    ::videoResource( GeometryFeatureType& gf ) const
{
    const auto itr = acquiredFeatures.find( &gf );
    CARNA_ASSERT( itr != acquiredFeatures.end() );
    return static_cast< typename GeometryFeatureType::ManagedInterface& >( *itr->second );
}


template< typename RenderableCompare >
template< typename GeometryFeatureType >
const typename GeometryFeatureType::ManagedInterface& GeometryStage< RenderableCompare >
    ::videoResource( const GeometryFeatureType& gf ) const
{
    return videoResource( const_cast< GeometryFeatureType& >( gf ) );
}



}  // namespace Carna :: base

}  // namespace Carna

#endif // GEOMETRYSTAGE_H_6014714286