123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298 |
- /* Copyright (C) 2002-2004 RealVNC Ltd. All Rights Reserved.
- *
- * This 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 2 of the License, or
- * (at your option) any later version.
- *
- * This software 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 for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this software; if not, write to the Free Software
- * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
- * USA.
- */
-
- // -=- DeviceFrameBuffer.cxx
- //
- // The DeviceFrameBuffer class encapsulates the pixel data of the system
- // display.
-
- #define WIN32_LEAN_AND_MEAN
- #include <windows.h>
- #include <assert.h>
-
- #include <vector>
-
- #include <rfb/LogWriter.h>
- #include <rfb/Exception.h>
- #include <rdr/types.h>
- #include <rfb/VNCServer.h>
- #include <rfb_win32/DeviceFrameBuffer.h>
- #include <rfb_win32/Win32Util.h>
- #include <rfb_win32/OSVersion.h>
-
- namespace rfb {
-
- namespace win32 {
-
- static LogWriter vlog("FrameBuffer");
-
-
- // -=- DeviceFrameBuffer class
-
- DeviceFrameBuffer::DeviceFrameBuffer(HDC deviceContext, const Rect& wRect)
- : DIBSectionBuffer(deviceContext), device(deviceContext), cursorBm(deviceContext),
- ignoreGrabErrors(false)
- {
-
- // -=- Firstly, let's check that the device has suitable capabilities
-
- int capabilities = GetDeviceCaps(device, RASTERCAPS);
- if (!(capabilities & RC_BITBLT)) {
- throw Exception("device does not support BitBlt");
- }
- if (!(capabilities & RC_DI_BITMAP)) {
- throw Exception("device does not support GetDIBits");
- }
- /*
- if (GetDeviceCaps(device, PLANES) != 1) {
- throw Exception("device does not support planar displays");
- }
- */
-
- // -=- Get the display dimensions and pixel format
-
- // Get the display dimensions
- RECT cr;
- if (!GetClipBox(device, &cr))
- throw rdr::SystemException("GetClipBox", GetLastError());
- deviceCoords = Rect(cr.left, cr.top, cr.right, cr.bottom);
- if (!wRect.is_empty())
- deviceCoords = wRect.translate(deviceCoords.tl);
- int w = deviceCoords.width();
- int h = deviceCoords.height();
-
- // We can't handle uneven widths :(
- if (w % 2) w--;
-
- // Configure the underlying DIB to match the device
- DIBSectionBuffer::setPF(DeviceContext::getPF(device));
- DIBSectionBuffer::setSize(w, h);
-
- // Configure the cursor buffer
- cursorBm.setPF(format);
-
- // Set up a palette if required
- if (!format.trueColour)
- updateColourMap();
- }
-
- DeviceFrameBuffer::~DeviceFrameBuffer() {
- }
-
-
- void
- DeviceFrameBuffer::setPF(const PixelFormat &pf) {
- throw Exception("setPF not supported");
- }
-
- void
- DeviceFrameBuffer::setSize(int w, int h) {
- throw Exception("setSize not supported");
- }
-
-
- #ifndef CAPTUREBLT
- #define CAPTUREBLT 0x40000000
- #endif
-
- void
- DeviceFrameBuffer::grabRect(const Rect &rect) {
- BitmapDC tmpDC(device, bitmap);
-
- // Map the rectangle coords from VNC Desktop-relative to device relative - usually (0,0)
- Point src = desktopToDevice(rect.tl);
-
- // Note: Microsoft's documentation lies directly about CAPTUREBLT and claims it works on 98/ME
- // If you try CAPTUREBLT on 98 then you get blank output...
- if (!::BitBlt(tmpDC, rect.tl.x, rect.tl.y, rect.width(), rect.height(), device, src.x, src.y,
- osVersion.isPlatformNT ? CAPTUREBLT | SRCCOPY : SRCCOPY)) {
- if (ignoreGrabErrors)
- vlog.error("BitBlt failed:%ld", GetLastError());
- else
- throw rdr::SystemException("BitBlt failed", GetLastError());
- }
- }
-
- void
- DeviceFrameBuffer::grabRegion(const Region &rgn) {
- std::vector<Rect> rects;
- std::vector<Rect>::const_iterator i;
- rgn.get_rects(&rects);
- for(i=rects.begin(); i!=rects.end(); i++) {
- grabRect(*i);
- }
- ::GdiFlush();
- }
-
-
- void copyDevicePaletteToDIB(HDC dc, DIBSectionBuffer* dib) {
- // - Fetch the system palette for the framebuffer
- PALETTEENTRY syspalette[256];
- UINT entries = ::GetSystemPaletteEntries(dc, 0, 256, syspalette);
-
- if (entries == 0) {
- vlog.info("resorting to standard 16 color palette");
- for (unsigned int i=0;i<256;i++) {
- int v = (i%16) >= 8 ? 127 : 255;
- syspalette[i].peRed = i & 1 ? v : 0;
- syspalette[i].peGreen = i & 2 ? v : 0;
- syspalette[i].peBlue = i & 4 ? v : 0;
- }
- } else {
- vlog.info("framebuffer has %u palette entries", entries);
- }
-
- // - Update the bitmap's stored copy of the palette
- for (unsigned int i=0;i<256;i++) {
- int r, g, b;
- r = (syspalette[i].peRed << 8) + 0x80;
- g = (syspalette[i].peGreen << 8) + 0x80;
- b = (syspalette[i].peBlue << 8) + 0x80;
- dib->setColour(i, r, g, b);
- }
-
- // - Update the DIB section to use the palette
- dib->refreshPalette();
- }
-
-
- void DeviceFrameBuffer::setCursor(HCURSOR hCursor, VNCServer* server)
- {
- // - If hCursor is null then there is no cursor - clear the old one
-
- if (hCursor == 0) {
- server->setCursor(0, 0, 0, 0, 0, 0);
- return;
- }
-
- try {
-
- // - Get the size and other details about the cursor.
-
- IconInfo iconInfo((HICON)hCursor);
-
- BITMAP maskInfo;
- if (!GetObject(iconInfo.hbmMask, sizeof(BITMAP), &maskInfo))
- throw rdr::SystemException("GetObject() failed", GetLastError());
-
- assert(maskInfo.bmPlanes == 1 && maskInfo.bmBitsPixel == 1);
-
- // - Create the cursor pixel buffer and mask storage
- // NB: The cursor pixel buffer is NOT used here. Instead, we
- // pass the cursorBm.data pointer directly, to save overhead.
-
- cursor.setSize(maskInfo.bmWidth, maskInfo.bmHeight);
- cursor.setPF(format);
- cursor.hotspot = Point(iconInfo.xHotspot, iconInfo.yHotspot);
-
- // - Get the AND and XOR masks. There is only an XOR mask if this is not a
- // colour cursor.
-
- if (!iconInfo.hbmColor)
- cursor.setSize(cursor.width(), cursor.height() / 2);
- rdr::U8Array mask(maskInfo.bmWidthBytes * maskInfo.bmHeight);
- rdr::U8* xorMask = mask.buf + cursor.height() * maskInfo.bmWidthBytes;
-
- if (!GetBitmapBits(iconInfo.hbmMask,
- maskInfo.bmWidthBytes * maskInfo.bmHeight, mask.buf))
- throw rdr::SystemException("GetBitmapBits failed", GetLastError());
-
- // Configure the cursor bitmap
- cursorBm.setSize(cursor.width(), cursor.height());
-
- // Copy the palette into it if required
- if (format.bpp <= 8)
- copyDevicePaletteToDIB(device, &cursorBm);
-
- // Draw the cursor into the bitmap
- BitmapDC dc(device, cursorBm.bitmap);
- if (!DrawIconEx(dc, 0, 0, hCursor, 0, 0, 0, NULL, DI_NORMAL | DI_COMPAT))
- throw rdr::SystemException("unable to render cursor", GetLastError());
-
- // Replace any XORed pixels with xorColour, because RFB doesn't support
- // XORing of cursors. XORing is used for the I-beam cursor, which is most
- // often used over a white background, but also sometimes over a black
- // background. We set the XOR'd pixels to black, then draw a white outline
- // around the whole cursor.
-
- // *** should we replace any pixels not set in mask to zero, to ensure
- // that irrelevant data doesn't screw compression?
-
- bool doOutline = false;
- if (!iconInfo.hbmColor) {
- Pixel xorColour = format.pixelFromRGB(0, 0, 0, cursorBm.getColourMap());
- for (int y = 0; y < cursor.height(); y++) {
- bool first = true;
- for (int x = 0; x < cursor.width(); x++) {
- int byte = y * maskInfo.bmWidthBytes + x / 8;
- int bit = 7 - x % 8;
- if ((mask.buf[byte] & (1 << bit)) && (xorMask[byte] & (1 << bit)))
- {
- mask.buf[byte] &= ~(1 << bit);
-
- switch (format.bpp) {
- case 8:
- ((rdr::U8*)cursorBm.data)[y * cursor.width() + x] = xorColour; break;
- case 16:
- ((rdr::U16*)cursorBm.data)[y * cursor.width() + x] = xorColour; break;
- case 32:
- ((rdr::U32*)cursorBm.data)[y * cursor.width() + x] = xorColour; break;
- }
-
- doOutline = true;
- }
- }
- }
- }
-
- // Finally invert the AND mask so it's suitable for RFB and pack it into
- // the minimum number of bytes per row.
-
- int maskBytesPerRow = (cursor.width() + 7) / 8;
-
- for (int j = 0; j < cursor.height(); j++) {
- for (int i = 0; i < maskBytesPerRow; i++)
- cursor.mask.buf[j * maskBytesPerRow + i]
- = ~mask.buf[j * maskInfo.bmWidthBytes + i];
- }
-
- if (doOutline) {
- vlog.debug("drawing cursor outline!");
- memcpy(cursor.data, cursorBm.data, cursor.dataLen());
- cursor.drawOutline(format.pixelFromRGB(0xffff, 0xffff, 0xffff, cursorBm.getColourMap()));
- memcpy(cursorBm.data, cursor.data, cursor.dataLen());
- }
-
- server->setCursor(cursor.width(), cursor.height(),
- cursor.hotspot.x, cursor.hotspot.y,
- cursorBm.data, cursor.mask.buf);
- } catch (rdr::Exception& e) {
- vlog.error(e.str());
- }
- }
-
-
- void
- DeviceFrameBuffer::updateColourMap() {
- if (!format.trueColour)
- copyDevicePaletteToDIB(device, this);
- }
-
- }; // namespace win32
-
- }; // namespace rfb
|