ISFPRadarScreen.cpp

#include "pch.h"
#include "ISFPRadarScreen.h"
#include "CESPlugIn.h"
#include <string>

extern CESPlugIn* pMyPlugIn;

static ULONG_PTR g_gdiPlusToken = 0;
static bool g_bGdiPlusStarted = false;

CISFPRadarScreen::CISFPRadarScreen(void)
    : m_hBrush(NULL)
    , m_nOpacity(100)
    , m_bHasImage(false)
{
    m_wszImagePath[0] = 0;

    if (!g_bGdiPlusStarted)
    {
        Gdiplus::GdiplusStartupInput si;
        si.GdiplusVersion = 1;
        si.SuppressBackgroundThread = TRUE;
        Gdiplus::GdiplusStartup(&g_gdiPlusToken, &si, NULL);
        g_bGdiPlusStarted = true;
    }

    if (pMyPlugIn)
        pMyPlugIn->RegisterScreen(this);
}

CISFPRadarScreen::~CISFPRadarScreen(void)
{
    if (pMyPlugIn)
        pMyPlugIn->UnregisterScreen(this);
    ClearBackgroundImage();
}

void CISFPRadarScreen::OnRefresh(HDC hDC, int Phase)
{
    if (Phase == REFRESH_PHASE_BACK_BITMAP && m_bHasImage && m_hBrush)
    {
        RECT rc = GetRadarArea();
        FillRect(hDC, &rc, m_hBrush);
    }

    if (Phase == REFRESH_PHASE_AFTER_LISTS)
    {
        RECT radarArea = GetRadarArea();

        RECT watermarkArea;
        watermarkArea.left   = radarArea.right - 400;
        watermarkArea.top    = radarArea.bottom - 130;
        watermarkArea.right  = radarArea.right - 120;
        watermarkArea.bottom = radarArea.bottom - 80;

        COLORREF oldTextColor = SetTextColor(hDC, RGB(200, 200, 200));
        int oldBkMode = SetBkMode(hDC, TRANSPARENT);

        HFONT hFont = CreateFontA(
            20, 0, 0, 0, FW_BOLD, FALSE, FALSE, FALSE,
            DEFAULT_CHARSET, OUT_DEFAULT_PRECIS, CLIP_DEFAULT_PRECIS,
            DEFAULT_QUALITY, DEFAULT_PITCH | FF_SWISS, "Arial");

        HFONT oldFont = (HFONT)SelectObject(hDC, hFont);

        DrawTextA(hDC, "ISFP-eLink", -1, &watermarkArea, DT_RIGHT | DT_TOP | DT_SINGLELINE);

        RECT line2Area = watermarkArea;
        line2Area.top += 24;
        line2Area.bottom += 24;
        DrawTextA(hDC, "Powered by ISFP", -1, &line2Area, DT_RIGHT | DT_TOP | DT_SINGLELINE);

        SelectObject(hDC, oldFont);
        DeleteObject(hFont);
        SetBkMode(hDC, oldBkMode);
        SetTextColor(hDC, oldTextColor);
    }
}

void CISFPRadarScreen::OnAsrContentToBeClosed(void)
{
    delete this;
}

void CISFPRadarScreen::RegenerateBrush()
{
    if (m_hBrush)
    {
        DeleteObject(m_hBrush);
        m_hBrush = NULL;
    }
    m_bHasImage = false;

    if (m_wszImagePath[0] == 0)
        return;

    Gdiplus::Bitmap gdiBmp(m_wszImagePath);
    if (gdiBmp.GetLastStatus() != Gdiplus::Ok)
        return;

    UINT sw = gdiBmp.GetWidth();
    UINT sh = gdiBmp.GetHeight();
    if (sw == 0 || sh == 0)
        return;

    Gdiplus::BitmapData bd;
    Gdiplus::Rect bdRc(0, 0, sw, sh);
    if (gdiBmp.LockBits(&bdRc, Gdiplus::ImageLockModeRead, PixelFormat32bppARGB, &bd) != Gdiplus::Ok)
        return;

    RECT radarArea = GetRadarArea();
    int nDstW = radarArea.right - radarArea.left;
    int nDstH = radarArea.bottom - radarArea.top;
    if (nDstW <= 0 || nDstH <= 0)
    {
        gdiBmp.UnlockBits(&bd);
        return;
    }

    float fOp = m_nOpacity / 100.0f;
    BYTE* pSrc = (BYTE*)bd.Scan0;
    int nSrcStride = bd.Stride;

    BITMAPINFO bmi = {};
    bmi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
    bmi.bmiHeader.biWidth = nDstW;
    bmi.bmiHeader.biHeight = -nDstH;
    bmi.bmiHeader.biPlanes = 1;
    bmi.bmiHeader.biBitCount = 24;
    bmi.bmiHeader.biCompression = BI_RGB;

    void* pDstBits = NULL;
    HDC hdcScreen = GetDC(NULL);
    HBITMAP hDib = CreateDIBSection(hdcScreen, &bmi, DIB_RGB_COLORS, &pDstBits, NULL, 0);
    ReleaseDC(NULL, hdcScreen);

    if (!hDib || !pDstBits)
    {
        gdiBmp.UnlockBits(&bd);
        return;
    }

    BYTE* pDst = (BYTE*)pDstBits;
    int nDstStride = ((nDstW * 3 + 3) / 4) * 4;

    for (int y = 0; y < nDstH; y++)
    {
        int srcY = y * (int)sh / nDstH;
        if (srcY >= (int)sh) srcY = (int)sh - 1;

        BYTE* pSrcRow = pSrc + srcY * nSrcStride;
        BYTE* pDstRow = pDst + y * nDstStride;

        for (int x = 0; x < nDstW; x++)
        {
            int srcX = x * (int)sw / nDstW;
            if (srcX >= (int)sw) srcX = (int)sw - 1;

            BYTE* pSrcPx = pSrcRow + srcX * 4;

            pDstRow[x * 3 + 0] = (BYTE)(pSrcPx[0] * fOp);
            pDstRow[x * 3 + 1] = (BYTE)(pSrcPx[1] * fOp);
            pDstRow[x * 3 + 2] = (BYTE)(pSrcPx[2] * fOp);
        }
    }

    gdiBmp.UnlockBits(&bd);

    m_hBrush = CreatePatternBrush(hDib);
    DeleteObject(hDib);

    if (!m_hBrush)
        return;

    m_bHasImage = true;
}

void CISFPRadarScreen::SetBackgroundImage(const wchar_t* lpszPath)
{
    if (!lpszPath || lpszPath[0] == 0)
        return;

    wcscpy_s(m_wszImagePath, MAX_PATH, lpszPath);
    RegenerateBrush();

    if (pMyPlugIn)
    {
        char szMsg[256];
        if (m_bHasImage)
            sprintf_s(szMsg, sizeof(szMsg), "Image loaded, opacity=%d%% (PatternBrush)", m_nOpacity);
        else
            sprintf_s(szMsg, sizeof(szMsg), "Failed to create pattern brush");
        pMyPlugIn->DisplayUserMessage("ISFP-eLink", "Background", szMsg, 1, 0, 0, 0, 0);
    }

    CPosition ld, ru;
    GetDisplayArea(&ld, &ru);
    SetDisplayArea(ld, ru);
    RequestRefresh();
}

void CISFPRadarScreen::SetBackgroundOpacity(int nOpacity)
{
    if (nOpacity < 0) nOpacity = 0;
    if (nOpacity > 100) nOpacity = 100;
    m_nOpacity = nOpacity;

    RegenerateBrush();

    if (pMyPlugIn)
    {
        char szMsg[128];
        sprintf_s(szMsg, sizeof(szMsg), "Opacity set to %d%%", m_nOpacity);
        pMyPlugIn->DisplayUserMessage("ISFP-eLink", "Background", szMsg, 1, 0, 0, 0, 0);
    }

    CPosition ld, ru;
    GetDisplayArea(&ld, &ru);
    SetDisplayArea(ld, ru);
    RequestRefresh();
}

void CISFPRadarScreen::ClearBackgroundImage()
{
    if (m_hBrush)
    {
        DeleteObject(m_hBrush);
        m_hBrush = NULL;
    }
    m_bHasImage = false;
    m_wszImagePath[0] = 0;
}