#include "nmPebiResultSnapshotRenderAdapter.h"

#include <vtkAlgorithmOutput.h>
#include <vtkCellData.h>
#include <vtkDoubleArray.h>
#include <vtkTrivialProducer.h>
#include <vtkUnstructuredGrid.h>

nmPebiResultSnapshotRenderAdapter::nmPebiResultSnapshotRenderAdapter()
    : m_pRenderGrid(vtkSmartPointer<vtkUnstructuredGrid>::New()),
      m_pPressureBuffer(vtkSmartPointer<vtkDoubleArray>::New()),
      m_pProducer(vtkSmartPointer<vtkTrivialProducer>::New()),
      m_eBindingMode(SharedPressureFrame),
      m_nCurrentFrameIndex(-1)
{
}

nmPebiResultSnapshotRenderAdapter::~nmPebiResultSnapshotRenderAdapter()
{
    clear();
}

bool nmPebiResultSnapshotRenderAdapter::initialize(
        const QSharedPointer<const nmPebiResultSnapshot>& pSnapshot,
        PressureBindingMode eBindingMode)
{
    clear();
    if(pSnapshot.isNull() || !pSnapshot->isComplete() ||
       (eBindingMode != SharedPressureFrame &&
        eBindingMode != CopiedPressureFrame))
    {
        return false;
    }

    m_pSnapshot = pSnapshot;
    m_eBindingMode = eBindingMode;
    if(!m_pSnapshot->copyResultGridStructure(m_pRenderGrid))
    {
        clear();
        return false;
    }

    // TrivialProducer 只负责把适配器网格接入管道，不会主动修改输出对象。
    m_pProducer->SetOutput(m_pRenderGrid);
    m_pProducer->Modified();
    return true;
}

void nmPebiResultSnapshotRenderAdapter::clear()
{
    // 先断开管道和 CellData，及时释放对旧快照压力数组的临时 VTK 引用。
    if(m_pProducer != NULL)
    {
        m_pProducer->SetOutput(NULL);
        m_pProducer->Modified();
    }
    if(m_pRenderGrid != NULL)
    {
        m_pRenderGrid->GetCellData()->Initialize();
        m_pRenderGrid->Initialize();
    }
    if(m_pPressureBuffer != NULL)
    {
        m_pPressureBuffer->Initialize();
    }
    m_pSnapshot.clear();
    m_eBindingMode = SharedPressureFrame;
    m_nCurrentFrameIndex = -1;
}

bool nmPebiResultSnapshotRenderAdapter::setPressureFrame(int nFrameIndex)
{
    if(m_pSnapshot.isNull() || m_pRenderGrid == NULL ||
       nFrameIndex < 0 ||
       nFrameIndex >= m_pSnapshot->getPressureFrameCount())
    {
        return false;
    }

    bool bBound = false;
    if(m_eBindingMode == SharedPressureFrame)
    {
        bBound = m_pSnapshot->bindPressureFrame(
                    nFrameIndex, m_pRenderGrid);
    }
    else
    {
        bBound = m_pSnapshot->copyPressureFrame(
                    nFrameIndex, m_pPressureBuffer);
        if(bBound)
        {
            m_pRenderGrid->GetCellData()->SetScalars(m_pPressureBuffer);
            m_pRenderGrid->Modified();
        }
    }

    if(!bBound)
    {
        return false;
    }
    m_nCurrentFrameIndex = nFrameIndex;
    m_pProducer->Modified();
    return true;
}

vtkAlgorithmOutput*
nmPebiResultSnapshotRenderAdapter::getOutputPort() const
{
    return m_pSnapshot.isNull() || m_pProducer == NULL
            ? NULL : m_pProducer->GetOutputPort();
}

int nmPebiResultSnapshotRenderAdapter::getCurrentFrameIndex() const
{
    return m_nCurrentFrameIndex;
}

vtkIdType nmPebiResultSnapshotRenderAdapter::getCellCount() const
{
    return m_pRenderGrid == NULL ? 0 :
            m_pRenderGrid->GetNumberOfCells();
}

bool nmPebiResultSnapshotRenderAdapter::isUsingSharedPressure() const
{
    return !m_pSnapshot.isNull() &&
            m_eBindingMode == SharedPressureFrame;
}