Optimize converting internal Direct2D renderer bitmap to wxImage.

Get access to bitmap bits with IWICBitmap::Lock instead of IWICBitmapSource::CopyPixels because in-place processing is more effective in terms of speed and memory consumption (see IWICBitmap interface description in MSDN).

See #17311.

src/msw/graphicsd2d.cpp

@@ -2336,17 +2336,41 @@ protected:
 private:
     void FlushRenderTargetToImage()
     {
-        int width = m_resultImage->GetWidth();
+        const int width = m_resultImage->GetWidth();
-        int height = m_resultImage->GetHeight();
+        const int height = m_resultImage->GetHeight();
-        int bufferSize = 4 * width * height;
 
-        BYTE* buffer = new BYTE[bufferSize];
+        WICRect rcLock = { 0, 0, width, height };
-        m_wicBitmap->CopyPixels(NULL, 4 * width, bufferSize, buffer);
+        IWICBitmapLock *pLock = NULL;
-        unsigned char* dest = m_resultImage->GetData();
+        HRESULT hr = m_wicBitmap->Lock(&rcLock, WICBitmapLockRead, &pLock);
-        unsigned char* destAlpha = m_resultImage->GetAlpha();
+        wxCHECK_HRESULT_RET(hr);
+
+        UINT rowStride = 0;
+        hr = pLock->GetStride(&rowStride);
+        if ( FAILED(hr) )
+        {
+            pLock->Release();
+            wxFAILED_HRESULT_MSG(hr);
+            return;
+        }
+
+        UINT bufferSize = 0;
+        BYTE *pBmpBuffer = NULL;
+        hr = pLock->GetDataPointer(&bufferSize, &pBmpBuffer);
+        if ( FAILED(hr) )
+        {
+            pLock->Release();
+            wxFAILED_HRESULT_MSG(hr);
+            return;
+        }
 
         WICPixelFormatGUID pixelFormat;
-        m_wicBitmap->GetPixelFormat(&pixelFormat);
+        hr = pLock->GetPixelFormat(&pixelFormat);
+        if ( FAILED(hr) )
+        {
+            pLock->Release();
+            wxFAILED_HRESULT_MSG(hr);
+            return;
+        }
         wxASSERT_MSG( pixelFormat == GUID_WICPixelFormat32bppPBGRA ||
                   pixelFormat == GUID_WICPixelFormat32bppBGR,
                   wxS("Unsupported pixel format") );
@@ -2354,21 +2378,29 @@ private:
         // Only premultiplied ARGB bitmaps are supported.
         const bool hasAlpha = pixelFormat == GUID_WICPixelFormat32bppPBGRA;
 
-        int k = 0;
+        unsigned char* destRGB = m_resultImage->GetData();
-        while (k < width * height)
+        unsigned char* destAlpha = m_resultImage->GetAlpha();
+        for( int y = 0; y < height; y++ )
         {
-            wxPBGRAColor color = wxPBGRAColor(buffer + k * 4);
+            BYTE *pPixByte = pBmpBuffer;
+            for ( int x = 0; x < width; x++ )
+            {
+                wxPBGRAColor color = wxPBGRAColor(pPixByte);
                 unsigned char a =  hasAlpha ? color.a : 255;
                 // Undo premultiplication for ARGB bitmap
-            dest[k * 3 + 0] = (a > 0 && a < 255) ? ( color.r * 255 ) / a : color.r;
+                *destRGB++ = (a > 0 && a < 255) ? ( color.r * 255 ) / a : color.r;
-            dest[k * 3 + 1] = (a > 0 && a < 255) ? ( color.g * 255 ) / a : color.g;
+                *destRGB++ = (a > 0 && a < 255) ? ( color.g * 255 ) / a : color.g;
-            dest[k * 3 + 2] = (a > 0 && a < 255) ? ( color.b * 255 ) / a : color.b;
+                *destRGB++ = (a > 0 && a < 255) ? ( color.b * 255 ) / a : color.b;
                 if ( destAlpha )
                     *destAlpha++ = a;
-            ++k;
+
+                pPixByte += 4;
             }
 
-        delete[] buffer;
+            pBmpBuffer += rowStride;
+        }
+
+        pLock->Release();
    }
 
 private: