生産がす: DIB(8) - 直線描画

2011年4月22日金曜日
DIB(8) - 直線描画

　直線を引くプログラムです。
大雑把に言えば、
横に1px進むごとに縦に0.なんぼかpx動かし点を打っていけば直線がかける寸法です。
ただし、横に1px進むごとに縦に1px以上動くようだと線がとぎれとぎれになるので、
そういう場合は縦に1px進むごとに横に0.なんぼかpx動く、のように変更して線が途切れないようにする必要があります。

■サンプル

#include <windows.h>
#include <algorithm>
const int WINDOW_WIDTH = 640;
const int WINDOW_HEIGHT = 480;
 
 
 
class DIB32
{
public:
  DIB32()
    : m_pixel( NULL )
  {
    ZeroMemory( &m_bmi, sizeof(m_bmi) );
  }
 
  virtual ~DIB32()
  {
    release();
  }
 
  virtual bool create( LONG width, LONG height )
  {
    if ( width <= 0 || height <= 0 ) return false;
    release();
 
    m_pixel = new DWORD[ width * height ];
    if ( !m_pixel ) return false;
 
    m_bmi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
    m_bmi.bmiHeader.biWidth = width;
    m_bmi.bmiHeader.biHeight = height;
    m_bmi.bmiHeader.biPlanes = 1;
    //32bit固定にする
    m_bmi.bmiHeader.biBitCount = 32;
    m_bmi.bmiHeader.biCompression = BI_RGB;
    //96dpiだと3780らしい。0の場合もあるとのこと
    m_bmi.bmiHeader.biXPelsPerMeter = 3780;
    m_bmi.bmiHeader.biYPelsPerMeter = 3780;
 
    return true;
  }
 
 
  virtual void release()
  {
    if ( m_pixel )
    {
      delete [] m_pixel;
      m_pixel = NULL;
    }
    ZeroMemory( &m_bmi, sizeof(m_bmi) );
  }
 
  bool render( HDC hdc, HWND hWnd ) const
  {
    RECT rect;
    GetClientRect( hWnd, &rect );
    return GDI_ERROR != StretchDIBits(
      hdc, 0, 0, rect.right, rect.bottom,
      0, 0, getWidth(), getHeight(),
      m_pixel, &m_bmi, DIB_RGB_COLORS, SRCCOPY );
  }
 
 
 
  bool line( LONG x1, LONG y1, LONG x2, LONG y2, DWORD color )
  {
    const LONG width = abs( x2 - x1 );
    const LONG height = abs( y2 - y1 );
 
    // check error
    if ( !m_pixel ||
      !width ||
      !height ||
      x1 < 0 && x2 < 0 ||
      y1 < 0 && y2 < 0 ||
      x1 >= getWidth() && x2 >= getWidth() ||
      y1 >= getHeight() && y2 >= getHeight() )
    {
      return false;
    }
    // 長い方を基準にする
    else if ( width >= height )
    {
      if ( x1 > x2 )
      {
        std::swap( x1, x2 );
        std::swap( y1, y2 );
      }
 
      
      LONG startX = x1;
      LONG endX = x2;
      const float addY = static_cast<float>(y2-y1) / static_cast<float>(x2-x1);
      float fy = static_cast<float>( y1 );
 
      // clipping-left
      if ( startX < 0 )
      {
        fy += (addY * static_cast<float>(-startX));
        startX = 0;
      }
 
      // clioping-right
      if ( endX >= getWidth() )
      {
        endX = getWidth()-1;
      }
 
      for (LONG x=startX; x<endX; ++x)
      {
        const LONG y = static_cast<LONG>( fy );
        if ( y >= 0 && y < getHeight() )
        {
          *getPixelAddr(x, y) = color;
        }
 
        fy += addY;
      }
    }
    else if ( width < height )
    {
      if ( y1 > y2 )
      {
        std::swap( x1, x2 );
        std::swap( y1, y2 );
      }
 
      const float addX = static_cast<float>(x2-x1) / static_cast<float>(y2-y1);
      float fx = static_cast<float>( x1 );
 
      // clipping-top
      if ( y1 < 0 )
      {
        fx += (addX * static_cast<float>(-y1));
        y1 = 0;
      }
      // clipping-bottom
      if ( y2 >= getHeight() )
      {
        y2 = getHeight()-1;
      }
 
      for (LONG y=y1; y<=y2; ++y)
      {
        const LONG x = static_cast<LONG>( fx );
        if ( x >= 0 && x < getWidth() )
        {
          *getPixelAddr(x, y) = color;
        }
 
        fx += addX;
      }
    }
    
    return true;
  }
 
  
  LONG getWidth() const { return m_bmi.bmiHeader.biWidth; }
  LONG getHeight() const { return m_bmi.bmiHeader.biHeight; }
  LPDWORD getPixelAddr( LONG x, LONG y) { return m_pixel + (getHeight()-1-y)*getWidth() + x; }
 
protected:
  LPDWORD m_pixel;
  BITMAPINFO m_bmi;
};
 
 
 
 
 
LRESULT CALLBACK wndProc(
  HWND hWnd,
  UINT msg,
  WPARAM wParam,
  LPARAM lParam )
{
  static DIB32 image;
  switch (msg)
  {
  case WM_DESTROY:
    ShowWindow( hWnd, SW_HIDE );
    PostQuitMessage(0);
    break;
  case WM_CREATE:
    image.create( WINDOW_WIDTH, WINDOW_HEIGHT );
    image.line( -50, -20, 690, 500, 0xFFFF8080 );
    break;
  case WM_PAINT:
    {
      PAINTSTRUCT ps = {0};
      HDC hdc = BeginPaint( hWnd, &ps );
      image.render( hdc, hWnd );
      EndPaint( hWnd, &ps );
    }
    break;
  default:
    return DefWindowProc( hWnd, msg, wParam, lParam );
  }
 
  return 0;
}
 
 
 
 
int WINAPI WinMain(
  HINSTANCE hInstance,
  HINSTANCE, PSTR, int )
{
  LPCTSTR WINDOW_NAME = TEXT("sample");
 
  WNDCLASSEX wc;
  wc.style    = CS_HREDRAW | CS_VREDRAW;
  wc.lpfnWndProc  = reinterpret_cast<WNDPROC>( wndProc );
  wc.cbClsExtra  = 0;
  wc.cbWndExtra  = 0;
  wc.cbSize    = sizeof( WNDCLASSEX );
  wc.hInstance  = hInstance;
  wc.hIcon    = NULL;
  wc.hIconSm    = NULL;
  wc.hCursor    = LoadCursor( NULL, IDC_ARROW );
  wc.hbrBackground= reinterpret_cast<HBRUSH>( GetStockObject(WHITE_BRUSH) );
  wc.lpszMenuName  = NULL;
  wc.lpszClassName= WINDOW_NAME;
  if ( !RegisterClassEx(&wc) ) return 0;
 
  LONG winWidth = WINDOW_WIDTH
    + GetSystemMetrics(SM_CXEDGE)
    + GetSystemMetrics(SM_CXBORDER)
    + GetSystemMetrics(SM_CXDLGFRAME);
  LONG winHeight = WINDOW_HEIGHT
    + GetSystemMetrics(SM_CYEDGE)
    + GetSystemMetrics(SM_CYBORDER)
    + GetSystemMetrics(SM_CYDLGFRAME)
    + GetSystemMetrics(SM_CYCAPTION);
  HWND hWnd = CreateWindowEx(
    0, WINDOW_NAME, NULL, WS_OVERLAPPEDWINDOW ^ WS_THICKFRAME,
    CW_USEDEFAULT, CW_USEDEFAULT, winWidth, winHeight,
    NULL, NULL, hInstance, NULL);
  if ( !hWnd ) return -1;
 
  ShowWindow( hWnd, SW_SHOWNORMAL );
  UpdateWindow( hWnd );
  
  MSG msg;
  for (;;)
  {
    if ( !GetMessage(&msg, NULL, 0, 0) ) break;
    TranslateMessage( &msg );
    DispatchMessage( &msg );
  }
 
  
  UnregisterClass( WINDOW_NAME, hInstance );
  return msg.wParam;
}
0 件のコメント:

コメントを投稿