/*

 * Copyright (c) 2007, 2019, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * This code 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

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

#include "D3DPipeline.h"

#include "jlong.h"

#include "GraphicsPrimitiveMgr.h"

#include "D3DContext.h"

#include "D3DSurfaceData.h"

#include "D3DBufImgOps.h"

#include "D3DPaints.h"

#include "D3DRenderQueue.h"

#include "D3DShaders.h"

#include "D3DTextRenderer.h"

#include "D3DPipelineManager.h"

#include "D3DGlyphCache.h"

typedef struct {

    D3DBLEND src;

    D3DBLEND dst;

} D3DBlendRule;

/**

 * This table contains the standard blending rules (or Porter-Duff compositing

 * factors) used in SetRenderState(), indexed by the rule constants from the

 * AlphaComposite class.

 */

D3DBlendRule StdBlendRules[] = {

    { D3DBLEND_ZERO,         D3DBLEND_ZERO        }, /* 0 - Nothing      */

    { D3DBLEND_ZERO,         D3DBLEND_ZERO        }, /* 1 - RULE_Clear   */

    { D3DBLEND_ONE,          D3DBLEND_ZERO        }, /* 2 - RULE_Src     */

    { D3DBLEND_ONE,          D3DBLEND_INVSRCALPHA }, /* 3 - RULE_SrcOver */

    { D3DBLEND_INVDESTALPHA, D3DBLEND_ONE         }, /* 4 - RULE_DstOver */

    { D3DBLEND_DESTALPHA,    D3DBLEND_ZERO        }, /* 5 - RULE_SrcIn   */

    { D3DBLEND_ZERO,         D3DBLEND_SRCALPHA    }, /* 6 - RULE_DstIn   */

    { D3DBLEND_INVDESTALPHA, D3DBLEND_ZERO        }, /* 7 - RULE_SrcOut  */

    { D3DBLEND_ZERO,         D3DBLEND_INVSRCALPHA }, /* 8 - RULE_DstOut  */

    { D3DBLEND_ZERO,         D3DBLEND_ONE         }, /* 9 - RULE_Dst     */

    { D3DBLEND_DESTALPHA,    D3DBLEND_INVSRCALPHA }, /*10 - RULE_SrcAtop */

    { D3DBLEND_INVDESTALPHA, D3DBLEND_SRCALPHA    }, /*11 - RULE_DstAtop */

    { D3DBLEND_INVDESTALPHA, D3DBLEND_INVSRCALPHA }, /*12 - RULE_AlphaXor*/

};

void

D3DUtils_SetOrthoMatrixOffCenterLH(D3DMATRIX *m,

                                   float width, float height)

{

    ZeroMemory(m, sizeof(D3DMATRIX));

    m->_11 =  2.0f/width;

    m->_22 = -2.0f/height;

    m->_33 =  0.5f;

    m->_44 =  1.0f;

    m->_41 = -1.0f;

    m->_42 =  1.0f;

    m->_43 =  0.5f;

}

void

D3DUtils_SetIdentityMatrix(D3DMATRIX *m)

{

    m->_12 = m->_13 = m->_14 = m->_21 = m->_23 = m->_24 = 0.0f;

    m->_31 = m->_32 = m->_34 = m->_41 = m->_42 = m->_43 = 0.0f;

    m->_11 = m->_22 = m->_33 = m->_44 = 1.0f;

}

// the following methods are copies of the AffineTransform's class

// corresponding methods, with these changes to the indexes:

// 00 -> 11

// 11 -> 22

// 01 -> 21

// 10 -> 12

// 02 -> 41

// 12 -> 42

void

D3DUtils_2DConcatenateM(D3DMATRIX *m, D3DMATRIX *m1)

{

    float M0, M1;

    float T00, T10, T01, T11;

    float T02, T12;

    T00 = m1->_11; T01 = m1->_21; T02 = m1->_41;

    T10 = m1->_12; T11 = m1->_22; T12 = m1->_42;

    M0 = m->_11;

    M1 = m->_21;

    m->_11  = T00 * M0 + T10 * M1;

    m->_21  = T01 * M0 + T11 * M1;

    m->_41 += T02 * M0 + T12 * M1;

    M0 = m->_12;

    M1 = m->_22;

    m->_12  = T00 * M0 + T10 * M1;

    m->_22  = T01 * M0 + T11 * M1;

    m->_42 += T02 * M0 + T12 * M1;

}

#ifdef UPDATE_TX

void

D3DUtils_2DScaleM(D3DMATRIX *m, float sx, float sy)

{

    m->_11 *= sx;

    m->_22 *= sy;

}

void

D3DUtils_2DInvertM(D3DMATRIX *m)

{

    float M11, M21, M41;

    float M12, M22, M42;

    float det;

    M11 = m->_11; M21 = m->_21; M41 = m->_41;

    M12 = m->_12; M22 = m->_22; M42 = m->_42;

    det = M11 * M22 - M21 * M12;

    if (fabs(det) <= 0.0000000001f) {

        memset(m, 0, sizeof(D3DMATRIX));

        return;

    }

    m->_11 =  M22 / det;

    m->_12 = -M12 / det;

    m->_21 = -M21 / det;

    m->_22 =  M11 / det;

    m->_41 = (M21 * M42 - M22 * M41) / det;

    m->_42 = (M12 * M41 - M11 * M42) / det;

}

void

D3DUtils_2DTranslateM(D3DMATRIX *m, float tx, float ty)

{

    m->_41 = tx * m->_11 + ty * m->_21 + m->_41;

    m->_42 = tx * m->_12 + ty * m->_22 + m->_42;

}

void

D3DUtils_2DTransformXY(D3DMATRIX *m, float *px, float *py)

{

    float x = *px;

    float y = *py;

    *px = x * m->_11 + y * m->_21 + m->_41;

    *py = x * m->_12 + y * m->_22 + m->_42;

}

void

D3DUtils_2DInverseTransformXY(D3DMATRIX *m, float *px, float *py)

{

    float x = *px, y = *py;

    x -= m->_41;

    y -= m->_42;

    float det = m->_11 * m->_22 - m->_21 * m->_12;

    if (fabs(det) < 0.0000000001f) {

        *px = 0.0f;

        *py = 0.0f;

    } else {

        *px = (x * m->_22 - y * m->_21) / det;

        *py = (y * m->_11 - x * m->_12) / det;

    }

}

#endif // UPDATE_TX

static void

D3DContext_DisposeShader(jlong programID)

{

    IDirect3DPixelShader9 *shader =

        (IDirect3DPixelShader9 *)jlong_to_ptr(programID);

    J2dTraceLn(J2D_TRACE_INFO, "D3DContext_DisposeShader");

    SAFE_RELEASE(shader);

}

// static

HRESULT

D3DContext::CreateInstance(IDirect3D9 *pd3d9, UINT adapter, D3DContext **ppCtx)

{

    HRESULT res;

    *ppCtx = new D3DContext(pd3d9, adapter);

    if (FAILED(res = (*ppCtx)->InitContext())) {

        delete *ppCtx;

        *ppCtx = NULL;

    }

    return res;

}

D3DContext::D3DContext(IDirect3D9 *pd3d, UINT adapter)

{

    J2dTraceLn(J2D_TRACE_INFO, "D3DContext::D3DContext");

    J2dTraceLn1(J2D_TRACE_VERBOSE, "  pd3d=0x%x", pd3d);

    pd3dObject = pd3d;

    pd3dDevice = NULL;

    adapterOrdinal = adapter;

    pResourceMgr = NULL;

    pMaskCache = NULL;

    pVCacher = NULL;

    pSyncQuery = NULL;

    pSyncRTRes = NULL;

    pStateBlock = NULL;

    D3DC_INIT_SHADER_LIST(convolvePrograms,   MAX_CONVOLVE);

    D3DC_INIT_SHADER_LIST(rescalePrograms,    MAX_RESCALE);

    D3DC_INIT_SHADER_LIST(lookupPrograms,     MAX_LOOKUP);

    D3DC_INIT_SHADER_LIST(basicGradPrograms,  4);

    D3DC_INIT_SHADER_LIST(linearGradPrograms, 8);

    D3DC_INIT_SHADER_LIST(radialGradPrograms, 8);

    pLCDGlyphCache= NULL;

    pGrayscaleGlyphCache= NULL;

    lcdTextProgram = NULL;

    aaPgramProgram = NULL;

    contextCaps = CAPS_EMPTY;

    bBeginScenePending = FALSE;

    ZeroMemory(&devCaps, sizeof(D3DCAPS9));

    ZeroMemory(&curParams, sizeof(curParams));

    extraAlpha = 1.0f;

}

void D3DContext::ReleaseDefPoolResources()

{

    J2dTraceLn(J2D_TRACE_INFO, "D3DContext::ReleaseDefPoolResources");

    EndScene();

    contextCaps = CAPS_EMPTY;

    SAFE_RELEASE(pSyncQuery);

    SAFE_RELEASE(pStateBlock);

    if (pVCacher != NULL) {

        pVCacher->ReleaseDefPoolResources();

    }

    if (pMaskCache != NULL) {

        pMaskCache->ReleaseDefPoolResources();

    }

    if (pLCDGlyphCache != NULL) {

        pLCDGlyphCache->ReleaseDefPoolResources();

    }

    if (pGrayscaleGlyphCache != NULL) {

        pGrayscaleGlyphCache->ReleaseDefPoolResources();

    }

    if (pResourceMgr != NULL) {

        if (pSyncRTRes != NULL) {

            pResourceMgr->ReleaseResource(pSyncRTRes);

            pSyncRTRes = NULL;

        }

        pResourceMgr->ReleaseDefPoolResources();

    }

    ZeroMemory(lastTexture, sizeof(lastTexture));

    ZeroMemory(lastTextureColorState, sizeof(lastTextureColorState));

}

void D3DContext::ReleaseContextResources()

{

    J2dTraceLn1(J2D_TRACE_INFO,

                "D3DContext::ReleaseContextResources: pd3dDevice = 0x%x",

                pd3dDevice);

    ReleaseDefPoolResources();

    // dispose shader lists

    ShaderList_Dispose(&convolvePrograms);

    ShaderList_Dispose(&rescalePrograms);

    ShaderList_Dispose(&lookupPrograms);

    ShaderList_Dispose(&basicGradPrograms);

    ShaderList_Dispose(&linearGradPrograms);

    ShaderList_Dispose(&radialGradPrograms);

    SAFE_DELETE(pLCDGlyphCache);

    SAFE_DELETE(pGrayscaleGlyphCache);

    SAFE_RELEASE(lcdTextProgram);

    SAFE_RELEASE(aaPgramProgram);

    SAFE_DELETE(pVCacher);

    SAFE_DELETE(pMaskCache);

    SAFE_DELETE(pResourceMgr);

}

D3DContext::~D3DContext() {

    J2dTraceLn2(J2D_TRACE_INFO,

                "~D3DContext: pd3dDevice=0x%x, pd3dObject =0x%x",

                pd3dDevice, pd3dObject);

    ReleaseContextResources();

    SAFE_RELEASE(pd3dDevice);

}

HRESULT

D3DContext::InitDevice(IDirect3DDevice9 *pd3dDevice)

{

    HRESULT res = S_OK;

    pd3dDevice->GetDeviceCaps(&devCaps);

    J2dRlsTraceLn1(J2D_TRACE_INFO,

                   "D3DContext::InitDevice: device %d", adapterOrdinal);

    // disable some of the unneeded and costly d3d functionality

    pd3dDevice->SetRenderState(D3DRS_CULLMODE, D3DCULL_NONE);

    pd3dDevice->SetRenderState(D3DRS_SPECULARENABLE, FALSE);

    pd3dDevice->SetRenderState(D3DRS_LIGHTING,  FALSE);

    pd3dDevice->SetRenderState(D3DRS_CLIPPING,  FALSE);

    pd3dDevice->SetRenderState(D3DRS_ZENABLE, D3DZB_FALSE);

    pd3dDevice->SetRenderState(D3DRS_ZWRITEENABLE, D3DZB_FALSE);

    pd3dDevice->SetRenderState(D3DRS_COLORVERTEX, FALSE);

    pd3dDevice->SetRenderState(D3DRS_STENCILENABLE, FALSE);

    // set the default texture addressing mode

    pd3dDevice->SetSamplerState(0, D3DSAMP_ADDRESSU, D3DTADDRESS_CLAMP);

    pd3dDevice->SetSamplerState(0, D3DSAMP_ADDRESSV, D3DTADDRESS_CLAMP);

    // REMIND: check supported filters with

    // IDirect3D9::CheckDeviceFormat with D3DUSAGE_QUERY_FILTER

    pd3dDevice->SetSamplerState(0, D3DSAMP_MAGFILTER, D3DTEXF_POINT);

    pd3dDevice->SetSamplerState(0, D3DSAMP_MINFILTER, D3DTEXF_POINT);

    // these states never change

    pd3dDevice->SetTextureStageState(0, D3DTSS_ALPHAOP, D3DTOP_MODULATE);

    pd3dDevice->SetTextureStageState(0, D3DTSS_COLOROP, D3DTOP_MODULATE);

    pd3dDevice->SetTextureStageState(0, D3DTSS_ALPHAARG2, D3DTA_DIFFUSE);

    pd3dDevice->SetTextureStageState(0, D3DTSS_COLORARG2, D3DTA_DIFFUSE);

    pd3dDevice->SetTextureStageState(1, D3DTSS_ALPHAOP, D3DTOP_MODULATE);

    pd3dDevice->SetTextureStageState(1, D3DTSS_COLOROP, D3DTOP_MODULATE);

    pd3dDevice->SetTextureStageState(1, D3DTSS_ALPHAARG2, D3DTA_CURRENT);

    pd3dDevice->SetTextureStageState(1, D3DTSS_COLORARG2, D3DTA_CURRENT);

    // init the array of latest textures

    ZeroMemory(lastTexture, sizeof(lastTexture));

    ZeroMemory(lastTextureColorState, sizeof(lastTextureColorState));

    opState = STATE_CHANGE;

    if (pResourceMgr == NULL) {

        res = D3DResourceManager::CreateInstance(this, &pResourceMgr);

    } else {

        res = pResourceMgr->Init(this);

    }

    RETURN_STATUS_IF_FAILED(res);

    if (pVCacher == NULL) {

        res = D3DVertexCacher::CreateInstance(this, &pVCacher);

    } else {

        res = pVCacher->Init(this);

    }

    RETURN_STATUS_IF_FAILED(res);

    if (pMaskCache == NULL) {

        res = D3DMaskCache::CreateInstance(this, &pMaskCache);

    } else{

        res = pMaskCache->Init(this);

    }

    RETURN_STATUS_IF_FAILED(res);

    if (pLCDGlyphCache != NULL) {

        if (FAILED(res = pLCDGlyphCache->Init(this))) {

            // we can live without the cache

            SAFE_DELETE(pLCDGlyphCache);

            res = S_OK;

        }

    }

    if (pGrayscaleGlyphCache != NULL) {

        if (FAILED(res = pGrayscaleGlyphCache->Init(this))) {

            // we can live without the cache

            SAFE_DELETE(pGrayscaleGlyphCache);

            res = S_OK;

        }

    }

    D3DMATRIX tx;

    D3DUtils_SetIdentityMatrix(&tx);

    pd3dDevice->SetTransform(D3DTS_WORLD, &tx);

    bIsIdentityTx = TRUE;

    if (pSyncQuery == NULL) {

        // this is allowed to fail, do not propagate the error

        if (FAILED(pd3dDevice->CreateQuery(D3DQUERYTYPE_EVENT, &pSyncQuery))) {

            J2dRlsTraceLn(J2D_TRACE_WARNING,

                          "D3DContext::InitDevice: sync query not available");

            pSyncQuery = NULL;

        }

    }

    if (pSyncRTRes == NULL) {

        D3DFORMAT format;

        if (FAILED(GetResourceManager()->

                   CreateRTSurface(32, 32, TRUE, TRUE, &format, &pSyncRTRes))) {

            J2dRlsTraceLn(J2D_TRACE_WARNING,

                          "D3DContext::InitDevice: "

                          "error creating sync surface");

        }

    }

    bBeginScenePending = FALSE;

    J2dRlsTraceLn1(J2D_TRACE_INFO,

                   "D3DContext::InitDefice: successfully initialized device %d",

                   adapterOrdinal);

    return res;

}

HRESULT

D3DContext::CheckAndResetDevice()

{

    HRESULT res = E_FAIL;

    J2dTraceLn(J2D_TRACE_INFO, "D3DContext::CheckAndResetDevice");

    if (pd3dDevice != NULL) {

        if (FAILED(res = pd3dDevice->TestCooperativeLevel())) {

            if (res == D3DERR_DEVICELOST) {

                J2dTraceLn1(J2D_TRACE_VERBOSE, "  device %d is still lost",

                            adapterOrdinal);

                // nothing to be done here, wait for D3DERR_DEVICENOTRESET

                return res;

            } else if (res == D3DERR_DEVICENOTRESET) {

                J2dTraceLn1(J2D_TRACE_VERBOSE, "  device %d needs to be reset",

                            adapterOrdinal);

                res = ResetContext();

            } else {

                // some unexpected error

                DebugPrintD3DError(res, "D3DContext::CheckAndResetDevice: "\

                                   "unknown error %x from TestCooperativeLevel");

            }

        } else {

            J2dTraceLn1(J2D_TRACE_VERBOSE, "  device %d is not lost",

                        adapterOrdinal);

        }

    } else {

        J2dTraceLn(J2D_TRACE_VERBOSE, "  null device");

    }

    return res;

}

HRESULT

D3DContext::ResetContext()

{

    HRESULT res = E_FAIL;

    J2dRlsTraceLn(J2D_TRACE_INFO, "D3DContext::ResetContext");

    if (pd3dDevice != NULL) {

        D3DPRESENT_PARAMETERS newParams;

        newParams = curParams;

        if (newParams.Windowed) {

            // reset to the current display mode if we're windowed,

            // otherwise to the display mode we were in when the device

            // was lost

            newParams.BackBufferFormat = D3DFMT_UNKNOWN;

            newParams.FullScreen_RefreshRateInHz = 0;

            newParams.BackBufferWidth = 0;

            newParams.BackBufferHeight = 0;

        }

        res = ConfigureContext(&newParams);

    }

    return res;

}

HRESULT

D3DContext::ConfigureContext(D3DPRESENT_PARAMETERS *pNewParams)

{

    J2dRlsTraceLn1(J2D_TRACE_INFO, "D3DContext::ConfigureContext device %d",

                   adapterOrdinal);

    HRESULT res = S_OK;

    D3DFORMAT stencilFormat;