123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167 |
- /* Copyright (C) 2002-2005 RealVNC Ltd. All Rights Reserved.
- * Copyright 2014 Pierre Ossman for Cendio AB
- *
- * 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.
- */
-
- #ifdef HAVE_CONFIG_H
- #include <config.h>
- #endif
-
- #include <rfb_win32/DIBSectionBuffer.h>
- #include <rfb_win32/DeviceContext.h>
- #include <rfb_win32/BitmapInfo.h>
- #include <rfb/LogWriter.h>
-
- using namespace rfb;
- using namespace win32;
-
- static LogWriter vlog("DIBSectionBuffer");
-
-
- DIBSectionBuffer::DIBSectionBuffer(HWND window_)
- : bitmap(0), window(window_), device(0) {
- }
-
- DIBSectionBuffer::DIBSectionBuffer(HDC device_)
- : bitmap(0), window(0), device(device_) {
- }
-
- DIBSectionBuffer::~DIBSectionBuffer() {
- if (bitmap)
- DeleteObject(bitmap);
- }
-
-
- inline void initMaxAndShift(DWORD mask, int* max, int* shift) {
- for ((*shift) = 0; (mask & 1) == 0; (*shift)++) mask >>= 1;
- (*max) = (uint16_t)mask;
- }
-
- void DIBSectionBuffer::initBuffer(const PixelFormat& pf, int w, int h) {
- HBITMAP new_bitmap = 0;
- uint8_t* new_data = 0;
-
- if (!pf.trueColour)
- throw rfb::Exception("palette format not supported");
-
- format = pf;
-
- if (w && h && (format.depth != 0)) {
- BitmapInfo bi;
- memset(&bi, 0, sizeof(bi));
- UINT iUsage = DIB_RGB_COLORS;
- bi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
- bi.bmiHeader.biBitCount = format.bpp;
- bi.bmiHeader.biSizeImage = (format.bpp / 8) * w * h;
- bi.bmiHeader.biPlanes = 1;
- bi.bmiHeader.biWidth = w;
- bi.bmiHeader.biHeight = -h;
- bi.bmiHeader.biCompression = (format.bpp > 8) ? BI_BITFIELDS : BI_RGB;
- bi.mask.red = format.pixelFromRGB((uint16_t)~0, 0, 0);
- bi.mask.green = format.pixelFromRGB(0, (uint16_t)~0, 0);
- bi.mask.blue = format.pixelFromRGB(0, 0, (uint16_t)~0);
-
- // Create a DIBSection to draw into
- if (device)
- new_bitmap = ::CreateDIBSection(device, (BITMAPINFO*)&bi.bmiHeader, iUsage,
- (void**)&new_data, NULL, 0);
- else
- new_bitmap = ::CreateDIBSection(WindowDC(window), (BITMAPINFO*)&bi.bmiHeader, iUsage,
- (void**)&new_data, NULL, 0);
-
- if (!new_bitmap) {
- int err = GetLastError();
- throw rdr::SystemException("unable to create DIB section", err);
- }
-
- vlog.debug("recreateBuffer()");
- } else {
- vlog.debug("one of area or format not set");
- }
-
- if (new_bitmap && bitmap) {
- vlog.debug("preserving bitmap contents");
-
- // Copy the contents across
- if (device) {
- BitmapDC src_dev(device, bitmap);
- BitmapDC dest_dev(device, new_bitmap);
- BitBlt(dest_dev, 0, 0, w, h, src_dev, 0, 0, SRCCOPY);
- } else {
- WindowDC wndDC(window);
- BitmapDC src_dev(wndDC, bitmap);
- BitmapDC dest_dev(wndDC, new_bitmap);
- BitBlt(dest_dev, 0, 0, w, h, src_dev, 0, 0, SRCCOPY);
- }
- }
-
- if (bitmap) {
- // Delete the old bitmap
- DeleteObject(bitmap);
- bitmap = 0;
- setBuffer(0, 0, NULL, 0);
- }
-
- if (new_bitmap) {
- int bpp, depth;
- int redMax, greenMax, blueMax;
- int redShift, greenShift, blueShift;
- int new_stride;
-
- // Set up the new bitmap
- bitmap = new_bitmap;
-
- // Determine the *actual* DIBSection format
- DIBSECTION ds;
- if (!GetObject(bitmap, sizeof(ds), &ds))
- throw rdr::SystemException("GetObject", GetLastError());
-
- // Correct the "stride" of the DIB
- // *** This code DWORD aligns each row - is that right???
- new_stride = w;
- int bytesPerRow = new_stride * format.bpp/8;
- if (bytesPerRow % 4) {
- bytesPerRow += 4 - (bytesPerRow % 4);
- new_stride = (bytesPerRow * 8) / format.bpp;
- vlog.info("adjusting DIB stride: %d to %d", w, new_stride);
- }
-
- setBuffer(w, h, new_data, new_stride);
-
- // Calculate the PixelFormat for the DIB
- bpp = depth = ds.dsBm.bmBitsPixel;
-
- // Get the truecolour format used by the DIBSection
- initMaxAndShift(ds.dsBitfields[0], &redMax, &redShift);
- initMaxAndShift(ds.dsBitfields[1], &greenMax, &greenShift);
- initMaxAndShift(ds.dsBitfields[2], &blueMax, &blueShift);
-
- // Calculate the effective depth
- depth = 0;
- Pixel bits = ds.dsBitfields[0] | ds.dsBitfields[1] | ds.dsBitfields[2];
- while (bits) {
- depth++;
- bits = bits >> 1;
- }
- if (depth > bpp)
- throw Exception("Bad DIBSection format (depth exceeds bpp)");
-
- format = PixelFormat(bpp, depth, false, true,
- redMax, greenMax, blueMax,
- redShift, greenShift, blueShift);
- }
- }
|