/////////////////////////////////////////////////////////////////////////////////
//
//  Copyright (C) Statoil ASA
//  Copyright (C) Ceetron Solutions AS
// 
//  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 "RivNNCGeometryGenerator.h"

#include <cmath>

#include "cvfDrawableGeo.h"
#include "cvfPrimitiveSetIndexedUInt.h"
#include "cvfScalarMapper.h"
#include "RigNNCData.h"
#include "RigCell.h"
#include "RigMainGrid.h"


//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
RivNNCGeometryGenerator::RivNNCGeometryGenerator(const RigNNCData* nncData, const cvf::Vec3d& offset, const cvf::Array<size_t>* nncIndexes )
   : m_nncData(nncData),
   m_nncIndexes(nncIndexes),
   m_offset(offset)
{

}

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

//--------------------------------------------------------------------------------------------------
/// Generate surface drawable geo from the specified region
//--------------------------------------------------------------------------------------------------
cvf::ref<cvf::DrawableGeo> RivNNCGeometryGenerator::generateSurface()
{
    computeArrays();

    CVF_ASSERT(m_vertices.notNull());

    if (m_vertices->size() == 0) return NULL;

    cvf::ref<cvf::DrawableGeo> geo = new cvf::DrawableGeo;
    geo->setFromTriangleVertexArray(m_vertices.p());

    return geo;
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
void RivNNCGeometryGenerator::computeArrays()
{
    std::vector<cvf::Vec3f> vertices;
    std::vector<size_t> triangleToNNC;

    const cvf::Vec3d offset = m_offset;
    long long numConnections = static_cast<long long>(m_nncIndexes.isNull()? m_nncData->connections().size(): m_nncIndexes->size());

    bool isVisibilityCalcActive = m_cellVisibility.notNull() && m_grid.notNull();
    std::vector<RigCell>* allCells = NULL;
    if (isVisibilityCalcActive)
    {
        allCells =  &(m_grid->mainGrid()->globalCellArray());
    }

#pragma omp parallel for ordered
    for (long long nIdx = 0; nIdx < numConnections; ++nIdx)
    {
        size_t conIdx = m_nncIndexes.isNull()? nIdx: (*m_nncIndexes)[nIdx];

        const RigConnection& conn = m_nncData->connections()[conIdx];

        if (conn.m_polygon.size())
        {
            bool isVisible = true;
            if (isVisibilityCalcActive)
            {
                bool cell1Visible = false;
                bool cell2Visible = false;

                if ((*allCells)[conn.m_c1GlobIdx].hostGrid() == m_grid.p())
                {
                    size_t cell1GridLocalIdx = (*allCells)[conn.m_c1GlobIdx].gridLocalCellIndex();
                    cell1Visible = (*m_cellVisibility)[cell1GridLocalIdx];
                }

                if ((*allCells)[conn.m_c2GlobIdx].hostGrid() == m_grid.p())
                {
                    size_t cell2GridLocalIdx = (*allCells)[conn.m_c2GlobIdx].gridLocalCellIndex();
                    cell2Visible = (*m_cellVisibility)[cell2GridLocalIdx];
                }

                isVisible = cell1Visible || cell2Visible;

            }

            if (isVisible)
            {
                cvf::Vec3f vx1 = cvf::Vec3f(conn.m_polygon[0] - offset);
                cvf::Vec3f vx2;
                cvf::Vec3f vx3 = cvf::Vec3f(conn.m_polygon[1] - offset);

                for (size_t vxIdx = 2; vxIdx < conn.m_polygon.size() ; ++vxIdx)
                {
                    vx2 = vx3;
                    vx3 = cvf::Vec3f( conn.m_polygon[vxIdx] - offset);
#pragma omp critical
                    {
                        vertices.push_back(vx1);
                        vertices.push_back(vx2);
                        vertices.push_back(vx3);
                        triangleToNNC.push_back(conIdx);
                    }
                }
            }
        }
    }

    m_vertices = new cvf::Vec3fArray;
    m_vertices->assign(vertices);
    m_triangleIndexToNNCIndex = new cvf::Array<size_t>;
    m_triangleIndexToNNCIndex->assign(triangleToNNC);
}

//--------------------------------------------------------------------------------------------------
/// Calculates the texture coordinates in a "nearly" one dimensional texture. 
/// Undefined values are coded with a y-texture coordinate value of 1.0 instead of the normal 0.5
//--------------------------------------------------------------------------------------------------
void RivNNCGeometryGenerator::textureCoordinates(cvf::Vec2fArray* textureCoords, const cvf::ScalarMapper* mapper, size_t scalarResultIndex) const
{
    size_t numVertices = m_vertices->size();

    textureCoords->resize(numVertices);
    cvf::Vec2f* rawPtr = textureCoords->ptr();
   
    const std::vector<double>* nncResultVals = m_nncData->connectionScalarResult(scalarResultIndex);
    if (!nncResultVals)
    {
        textureCoords->setAll(cvf::Vec2f(0.0f, 1.0f));
        return;
    }

    double cellScalarValue;
    cvf::Vec2f texCoord;

#pragma omp parallel for private(texCoord, cellScalarValue)
    for (int tIdx = 0; tIdx < static_cast<int>(m_triangleIndexToNNCIndex->size()); tIdx++)
    {
        cellScalarValue = (*nncResultVals)[(*m_triangleIndexToNNCIndex)[tIdx]];
        texCoord = mapper->mapToTextureCoord(cellScalarValue);
        if (cellScalarValue == HUGE_VAL || cellScalarValue != cellScalarValue) // a != a is true for NAN's
        {
            texCoord[1] = 1.0f;
        }

        size_t j;
        for (j = 0; j < 3; j++)
        {   
            rawPtr[tIdx*3 + j] = texCoord;
        }
    }
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
void RivNNCGeometryGenerator::setCellVisibility(const cvf::UByteArray* cellVisibility, const RigGridBase * grid)
{
    m_cellVisibility = cellVisibility;
    m_grid = grid;
}

//--------------------------------------------------------------------------------------------------
/// 
//--------------------------------------------------------------------------------------------------
cvf::ref<cvf::Array<size_t> > RivNNCGeometryGenerator::triangleToNNCIndex() const
{
    return m_triangleIndexToNNCIndex;
}