/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) 2018-     Equinor ASA
//
//  ResInsight 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 3 of the License, or
//  (at your option) any later version.
//
//  ResInsight 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 at <http://www.gnu.org/licenses/gpl.html>
//  for more details.
//
/////////////////////////////////////////////////////////////////////////////////

#include "RiuCvfOverlayItemWidget.h"

#include "RiaGuiApplication.h"

#include "cafTitledOverlayFrame.h"
#include "cafViewer.h"

#include "cvfCamera.h"
#include "cvfFramebufferObject.h"
#include "cvfRenderSequence.h"
#include "cvfRenderbufferObject.h"
#include "cvfRendering.h"
#include "cvfqtUtils.h"

#include <QApplication>
#include <QBoxLayout>
#include <QFrame>
#include <QLabel>
#include <QPixmap>
#include <QPushButton>
#include <QResizeEvent>

#include "glew/GL/glew.h"
//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RiuCvfOverlayItemWidget::RiuCvfOverlayItemWidget( QWidget* parent /*=0*/, QWidget* widgetToSnapTo )
    : RiuDraggableOverlayFrame( parent, widgetToSnapTo )
{
    this->layout()->setMargin( 0 );
    this->layout()->setSpacing( 0 );
}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
RiuCvfOverlayItemWidget::~RiuCvfOverlayItemWidget() {}

//--------------------------------------------------------------------------------------------------
///
//--------------------------------------------------------------------------------------------------
void RiuCvfOverlayItemWidget::updateFromOverlayItem( caf::TitledOverlayFrame* item )
{
    unsigned int width  = item->renderSize().x();
    unsigned int height = item->renderSize().y();

    QGLFormat glFormat;
    glFormat.setDirectRendering( RiaGuiApplication::instance()->useShaders() );

    // Enforce no border to avoid
    item->setBackgroundFrameColor( cvf::Color4f( 0, 0, 0, 0 ) );

    caf::Viewer*        viewer        = new caf::Viewer( glFormat, nullptr );
    cvf::OpenGLContext* cvfOglContext = viewer->cvfOpenGLContext();
    viewer->resize( width, height );

    // Create a rendering

    cvf::ref<cvf::Rendering> rendering = new cvf::Rendering;
    item->setLayoutFixedPosition( {0, 0} );
    rendering->addOverlayItem( item );

    rendering->camera()->setViewport( 0, 0, width, height );
    rendering->camera()->viewport()->setClearColor( {1, 1, 1, 0} );

    cvf::ref<cvf::RenderSequence> renderingSequence = new cvf::RenderSequence;
    renderingSequence->addRendering( rendering.p() );

    if ( RiaGuiApplication::instance()->useShaders() )
    {
        // Set up frame and render buffers

        cvf::ref<cvf::FramebufferObject> fbo = new cvf::FramebufferObject;

        cvf::ref<cvf::RenderbufferObject> rboColor = new cvf::RenderbufferObject( cvf::RenderbufferObject::RGBA,
                                                                                  width,
                                                                                  height );
        cvf::ref<cvf::RenderbufferObject> rboDepth =
            new cvf::RenderbufferObject( cvf::RenderbufferObject::DEPTH_COMPONENT24, width, height );

        fbo->attachDepthRenderbuffer( rboDepth.p() );
        fbo->attachColorRenderbuffer( 0, rboColor.p() );

        fbo->applyOpenGL( cvfOglContext );
        rendering->setTargetFramebuffer( fbo.p() );
        fbo->bind( cvfOglContext );
    }

    renderingSequence->render( cvfOglContext );

    // Read data from framebuffer

    cvf::UByteArray arr( 4 * width * height );
    glReadPixels( 0, 0, static_cast<GLsizei>( width ), static_cast<GLsizei>( height ), GL_RGBA, GL_UNSIGNED_BYTE, arr.ptr() );

    // Create a cvf texture image

    cvf::ref<cvf::TextureImage> img = new cvf::TextureImage;
    img->setData( arr.ptr(), width, height );

    QImage image = cvfqt::Utils::toQImage( *img.p() );
    // image.save("jjsLegendImageTest.png");

    QPixmap pixmap = QPixmap::fromImage( image );

    delete viewer;

    m_overlayItemLabel->setPixmap( pixmap );
    this->setMinimumSize( QSize( width, height ) );
    this->resize( QSize( width, height ) );
}