dlls/gdi32/pen.c

   1 /*
   2  * GDI pen objects
   3  *
   4  * Copyright 1993 Alexandre Julliard
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include "config.h"
  22
  23 #include <stdarg.h>
  24 #include <stdlib.h>
  25 #include <string.h>
  26 #include <assert.h>
  27
  28 #include "windef.h"
  29 #include "winbase.h"
  30 #include "wingdi.h"
  31 #include "gdi_private.h"
  32 #include "wine/debug.h"
  33
  34 WINE_DEFAULT_DEBUG_CHANNEL(gdi);
  35
  36   /* GDI logical pen object */
  37 typedef struct
  38 {
  39     struct brush_pattern pattern;
  40     EXTLOGPEN            logpen;
  41 } PENOBJ;
  42
  43
  44 static HGDIOBJ PEN_SelectObject( HGDIOBJ handle, HDC hdc );
  45 static INT PEN_GetObject( HGDIOBJ handle, INT count, LPVOID buffer );
  46 static BOOL PEN_DeleteObject( HGDIOBJ handle );
  47
  48 static const struct gdi_obj_funcs pen_funcs =
  49 {
  50     PEN_SelectObject,  /* pSelectObject */
  51     PEN_GetObject,     /* pGetObjectA */
  52     PEN_GetObject,     /* pGetObjectW */
  53     NULL,              /* pUnrealizeObject */
  54     PEN_DeleteObject   /* pDeleteObject */
  55 };
  56
  57
  58 /***********************************************************************
  59  *           CreatePen    (GDI32.@)
  60  */
  61 HPEN WINAPI CreatePen( INT style, INT width, COLORREF color )
  62 {
  63     LOGPEN logpen;
  64
  65     TRACE("%d %d %06x\n", style, width, color );
  66
  67     logpen.lopnStyle = style;
  68     logpen.lopnWidth.x = width;
  69     logpen.lopnWidth.y = 0;
  70     logpen.lopnColor = color;
  71
  72     return CreatePenIndirect( &logpen );
  73 }
  74
  75
  76 /***********************************************************************
  77  *           CreatePenIndirect    (GDI32.@)
  78  */
  79 HPEN WINAPI CreatePenIndirect( const LOGPEN * pen )
  80 {
  81     PENOBJ * penPtr;
  82     HPEN hpen;
  83
  84     if (pen->lopnStyle == PS_NULL)
  85     {
  86         hpen = GetStockObject(NULL_PEN);
  87         if (hpen) return hpen;
  88     }
  89
  90     if (!(penPtr = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*penPtr) ))) return 0;
  91
  92     penPtr->logpen.elpPenStyle = pen->lopnStyle;
  93     penPtr->logpen.elpWidth = abs(pen->lopnWidth.x);
  94     penPtr->logpen.elpColor = pen->lopnColor;
  95     penPtr->logpen.elpBrushStyle = BS_SOLID;
  96
  97     switch (pen->lopnStyle)
  98     {
  99     case PS_SOLID:
 100     case PS_DASH:
 101     case PS_DOT:
 102     case PS_DASHDOT:
 103     case PS_DASHDOTDOT:
 104     case PS_INSIDEFRAME:
 105         break;
 106     case PS_NULL:
 107         penPtr->logpen.elpWidth = 1;
 108         penPtr->logpen.elpColor = 0;
 109         break;
 110     default:
 111         penPtr->logpen.elpPenStyle = PS_SOLID;
 112         break;
 113     }
 114
 115     if (!(hpen = alloc_gdi_handle( penPtr, OBJ_PEN, &pen_funcs )))
 116         HeapFree( GetProcessHeap(), 0, penPtr );
 117     return hpen;
 118 }
 119
 120 /***********************************************************************
 121  *           ExtCreatePen    (GDI32.@)
 122  */
 123
 124 HPEN WINAPI ExtCreatePen( DWORD style, DWORD width,
 125                               const LOGBRUSH * brush, DWORD style_count,
 126                               const DWORD *style_bits )
 127 {
 128     PENOBJ *penPtr = NULL;
 129     HPEN hpen;
 130     LOGBRUSH logbrush;
 131
 132     if ((style_count || style_bits) && (style & PS_STYLE_MASK) != PS_USERSTYLE)
 133         goto invalid;
 134
 135     switch (style & PS_STYLE_MASK)
 136     {
 137     case PS_NULL:
 138         return CreatePen( PS_NULL, 0, brush->lbColor );
 139
 140     case PS_SOLID:
 141     case PS_DASH:
 142     case PS_DOT:
 143     case PS_DASHDOT:
 144     case PS_DASHDOTDOT:
 145         break;
 146
 147     case PS_USERSTYLE:
 148         if (((INT)style_count) <= 0) return 0;
 149
 150         if ((style_count > 16) || !style_bits) goto invalid;
 151
 152         if ((style & PS_TYPE_MASK) == PS_GEOMETRIC)
 153         {
 154             UINT i;
 155             BOOL has_neg = FALSE, all_zero = TRUE;
 156
 157             for(i = 0; (i < style_count) && !has_neg; i++)
 158             {
 159                 has_neg = has_neg || (((INT)(style_bits[i])) < 0);
 160                 all_zero = all_zero && (style_bits[i] == 0);
 161             }
 162
 163             if (all_zero || has_neg) goto invalid;
 164         }
 165         break;
 166
 167     case PS_INSIDEFRAME:  /* applicable only for geometric pens */
 168         if ((style & PS_TYPE_MASK) != PS_GEOMETRIC) goto invalid;
 169         break;
 170
 171     case PS_ALTERNATE:  /* applicable only for cosmetic pens */
 172         if ((style & PS_TYPE_MASK) == PS_GEOMETRIC) goto invalid;
 173         break;
 174
 175     default:
 176         SetLastError(ERROR_INVALID_PARAMETER);
 177         return 0;
 178     }
 179
 180     if ((style & PS_TYPE_MASK) == PS_GEOMETRIC)
 181     {
 182         if (brush->lbStyle == BS_NULL) return CreatePen( PS_NULL, 0, 0 );
 183     }
 184     else
 185     {
 186         if (width != 1) goto invalid;
 187         if (brush->lbStyle != BS_SOLID) goto invalid;
 188     }
 189
 190     if (!(penPtr = HeapAlloc(GetProcessHeap(), 0, FIELD_OFFSET(PENOBJ,logpen.elpStyleEntry[style_count]))))
 191         return 0;
 192
 193     logbrush = *brush;
 194     if (!store_brush_pattern( &logbrush, &penPtr->pattern )) goto invalid;
 195     if (logbrush.lbStyle == BS_DIBPATTERN) logbrush.lbStyle = BS_DIBPATTERNPT;
 196
 197     penPtr->logpen.elpPenStyle = style;
 198     penPtr->logpen.elpWidth = abs((int)width);
 199     penPtr->logpen.elpBrushStyle = logbrush.lbStyle;
 200     penPtr->logpen.elpColor = logbrush.lbColor;
 201     penPtr->logpen.elpHatch = brush->lbHatch;
 202     penPtr->logpen.elpNumEntries = style_count;
 203     memcpy(penPtr->logpen.elpStyleEntry, style_bits, style_count * sizeof(DWORD));
 204
 205     if (!(hpen = alloc_gdi_handle( penPtr, OBJ_EXTPEN, &pen_funcs )))
 206     {
 207         free_brush_pattern( &penPtr->pattern );
 208         HeapFree( GetProcessHeap(), 0, penPtr );
 209     }
 210     return hpen;
 211
 212 invalid:
 213     HeapFree( GetProcessHeap(), 0, penPtr );
 214     SetLastError( ERROR_INVALID_PARAMETER );
 215     return 0;
 216 }
 217
 218 /***********************************************************************
 219  *           PEN_SelectObject
 220  */
 221 static HGDIOBJ PEN_SelectObject( HGDIOBJ handle, HDC hdc )
 222 {
 223     PENOBJ *pen;
 224     HGDIOBJ ret = 0;
 225     DC *dc = get_dc_ptr( hdc );
 226     WORD type;
 227
 228     if (!dc)
 229     {
 230         SetLastError( ERROR_INVALID_HANDLE );
 231         return 0;
 232     }
 233
 234     if ((pen = get_any_obj_ptr( handle, &type )))
 235     {
 236         struct brush_pattern *pattern;
 237         PHYSDEV physdev = GET_DC_PHYSDEV( dc, pSelectPen );
 238
 239         switch (type)
 240         {
 241         case OBJ_PEN:
 242             pattern = NULL;
 243             break;
 244         case OBJ_EXTPEN:
 245             pattern = &pen->pattern;
 246             if (!pattern->info) pattern = NULL;
 247             break;
 248         default:
 249             GDI_ReleaseObj( handle );
 250             release_dc_ptr( dc );
 251             return 0;
 252         }
 253
 254         GDI_inc_ref_count( handle );
 255         GDI_ReleaseObj( handle );
 256
 257         if (!physdev->funcs->pSelectPen( physdev, handle, pattern ))
 258         {
 259             GDI_dec_ref_count( handle );
 260         }
 261         else
 262         {
 263             ret = dc->hPen;
 264             dc->hPen = handle;
 265             GDI_dec_ref_count( ret );
 266         }
 267     }
 268     release_dc_ptr( dc );
 269     return ret;
 270 }
 271
 272
 273 /***********************************************************************
 274  *           PEN_DeleteObject
 275  */
 276 static BOOL PEN_DeleteObject( HGDIOBJ handle )
 277 {
 278     PENOBJ *pen = free_gdi_handle( handle );
 279
 280     if (!pen) return FALSE;
 281     free_brush_pattern( &pen->pattern );
 282     return HeapFree( GetProcessHeap(), 0, pen );
 283 }
 284
 285
 286 /***********************************************************************
 287  *           PEN_GetObject
 288  */
 289 static INT PEN_GetObject( HGDIOBJ handle, INT count, LPVOID buffer )
 290 {
 291     WORD type;
 292     PENOBJ *pen = get_any_obj_ptr( handle, &type );
 293     INT ret = 0;
 294
 295     if (!pen) return 0;
 296
 297     switch (type)
 298     {
 299     case OBJ_PEN:
 300     {
 301         LOGPEN *lp;
 302
 303         if (!buffer) ret = sizeof(LOGPEN);
 304         else if (count < sizeof(LOGPEN)) ret = 0;
 305         else if ((pen->logpen.elpPenStyle & PS_STYLE_MASK) == PS_NULL && count == sizeof(EXTLOGPEN))
 306         {
 307             EXTLOGPEN *elp = buffer;
 308             *elp = pen->logpen;
 309             elp->elpWidth = 0;
 310             ret = sizeof(EXTLOGPEN);
 311         }
 312         else
 313         {
 314             lp = buffer;
 315             lp->lopnStyle = pen->logpen.elpPenStyle;
 316             lp->lopnColor = pen->logpen.elpColor;
 317             lp->lopnWidth.x = pen->logpen.elpWidth;
 318             lp->lopnWidth.y = 0;
 319             ret = sizeof(LOGPEN);
 320         }
 321         break;
 322     }
 323
 324     case OBJ_EXTPEN:
 325         ret = sizeof(EXTLOGPEN) + pen->logpen.elpNumEntries * sizeof(DWORD) - sizeof(pen->logpen.elpStyleEntry);
 326         if (buffer)
 327         {
 328             if (count < ret) ret = 0;
 329             else memcpy(buffer, &pen->logpen, ret);
 330         }
 331         break;
 332     }
 333     GDI_ReleaseObj( handle );
 334     return ret;
 335 }