ObjectARX实现CAD截图

admin

1. //框选区域
   
2. ads_point p1, p2;
   
3. if (RTNORM != ads_getpoint(NULL, _T("\n第一角点"), p1))
   
4.     return;
   
5. if (RTNORM != ads_getcorner(p1, _T("\n第二角点"), p2))
   
6.     return;
   
7. AcGePoint3d pt1, pt2;
   
8. pt1 = asPnt3d(p1);
   
9. pt2 = asPnt3d(p2);
   
10. //用户坐标系转世界坐标系
    
11. pt1 = PublicFunction::TransUcs2Wcs(pt1);
    
12. pt2 = PublicFunction::TransUcs2Wcs(pt2);
    
13. ads_name sset;
    
14. if (RTNORM != acedSSGet(_T("W"), p1, p2, NULL, sset))
    
15. {
    
16.     return;
    
17. }
    
18. AcDbObjectIdArray idArray = PublicFunction::GetObjectids(sset);
    
19. ads_ssfree(sset);
    
20. if (idArray.isEmpty())
    
21.     return;
    
22. //bmp
    
23. POINT cp1 = GetPointInAcadCoordinate(pt1);
    
24. POINT cp2 = GetPointInAcadCoordinate(pt2);
    
25. if (cp1.x > cp2.x)
    
26.     std::swap(cp1, cp2);
    
27. WriteToBmp(cp1, cp2);
    
28. //wblock
    
29. AcDbDatabase* pShortCutDwg = NULL;
    
30. acdbCurDwg()->wblock(pShortCutDwg, idArray, AcGePoint3d(0.0, 0.0, 0.0));
    
31. pShortCutDwg->saveAs(_T("D:\\a.dwg"));
    
32. delete pShortCutDwg;
    
33. pShortCutDwg = NULL;
    
34. 
    
35. 
    
36. 
    
37. POINT GetPointInAcadCoordinate(const AcGePoint3d &pt)
    
38. {
    
39.     ads_point pnt;
    
40.     ads_point_set(asDblArray(pt), pnt);
    
41.     //获取视口编号
    
42.     int wndNum = 0;
    
43.     struct resbuf v;
    
44.     if (acedGetVar(_T("cvport"), &v) == RTNORM)
    
45.     {
    
46.         wndNum = v.resval.rint;
    
47.     }
    
48.     //将pt坐标转为pixel屏幕坐标
    
49.     CPoint ptInScreen;
    
50.     acedCoordFromWorldToPixel(wndNum, pnt, ptInScreen);
    
51.     //将ptInScreen转为在view上的点p3
    
52.     POINT p3;
    
53.     p3.x = ptInScreen.x; p3.y = ptInScreen.y;
    
54.     ::ClientToScreen(acedGetAcadDwgView()->m_hWnd, &p3);
    
55.     return p3;
    
56. }
    
57. 
    
58. void WriteToBmp(const POINT& pt1, const POINT& pt2)
    
59. {
    
60.     //获得屏幕的HWND
    
61.     HWND hWnd = ::GetDesktopWindow();
    
62.     //获得屏幕的HDC
    
63.     HDC hScreenDC = ::GetDC(hWnd);
    
64.     HDC MemDC = ::CreateCompatibleDC(hScreenDC);
    
65.     RECT rect;
    
66.     ::GetWindowRect(hWnd, &rect);
    
67.     SIZE screensize;
    
68.     screensize.cx = abs(pt1.x - pt2.x);
    
69.     screensize.cy = abs(pt1.y - pt2.y);
    
70.     HBITMAP hBitmap = ::CreateCompatibleBitmap(hScreenDC, screensize.cx, screensize.cy);
    
71.     HGDIOBJ hOldBMP = ::SelectObject(MemDC, hBitmap);
    
72.     ::BitBlt(MemDC, 0, 0, screensize.cx, screensize.cy, hScreenDC, pt1.x, pt1.y, SRCCOPY);
    
73.     ::SelectObject(MemDC, hOldBMP);
    
74.     ::DeleteObject(MemDC);
    
75.     ::ReleaseDC(hWnd, hScreenDC);
    
76. 
    
77.     HDC hDC = ::CreateDC(_T("DISPLAY"), NULL, NULL, NULL);
    
78.     //当前分辨率下每个像素所占字节数
    
79.     int iBits = ::GetDeviceCaps(hDC, BITSPIXEL) * ::GetDeviceCaps(hDC, PLANES);
    
80.     ::DeleteDC(hDC);
    
81.     //位图中每个像素所占字节数
    
82.     WORD wBitCount;
    
83.     if (iBits <= 1)
    
84.         wBitCount = 1;
    
85.     else if (iBits <= 4)
    
86.         wBitCount = 4;
    
87.     else if (iBits <= 8)
    
88.         wBitCount = 8;
    
89.     else if (iBits <= 24)
    
90.         wBitCount = 24;
    
91.     else
    
92.         wBitCount = iBits;
    
93.     //调色板大小， 位图中像素字节大小
    
94.     DWORD dwPaletteSize = 0;
    
95.     if (wBitCount <= 8)
    
96.         dwPaletteSize = (1 << wBitCount) * sizeof(RGBQUAD);
    
97. 
    
98.     //位图属性结构
    
99.     BITMAP bm;
    
100.     ::GetObject(hBitmap, sizeof(bm), (LPSTR)&bm);
     
101. 
     
102.     //位图信息头结构
     
103.     BITMAPINFOHEADER bi;
     
104.     bi.biSize = sizeof(BITMAPINFOHEADER);
     
105.     bi.biWidth = bm.bmWidth;
     
106.     bi.biHeight = bm.bmHeight;
     
107.     bi.biPlanes = 1;
     
108.     bi.biBitCount = wBitCount;
     
109.     //BI_RGB表示位图没有压缩
     
110.     bi.biCompression = BI_RGB;
     
111.     bi.biSizeImage = 0;
     
112.     bi.biXPelsPerMeter = 0;
     
113.     bi.biYPelsPerMeter = 0;
     
114.     bi.biClrUsed = 0;
     
115.     bi.biClrImportant = 0;
     
116. 
     
117.     DWORD dwBmBitsSize = ((bm.bmWidth * wBitCount + 31) / 32) * 4 * bm.bmHeight;
     
118.     //为位图内容分配内存
     
119.     HANDLE hDib = ::GlobalAlloc(GHND, dwBmBitsSize + dwPaletteSize + sizeof(BITMAPINFOHEADER));
     
120.     LPBITMAPINFOHEADER lpbi = (LPBITMAPINFOHEADER)GlobalLock(hDib);
     
121.     *lpbi = bi;
     
122.     // 处理调色板
     
123.     HANDLE hPal = ::GetStockObject(DEFAULT_PALETTE);
     
124.     HANDLE  hOldPal = NULL;
     
125.     if (hPal)
     
126.     {
     
127.         hDC = ::GetDC(NULL);
     
128.         hOldPal = SelectPalette(hDC, (HPALETTE)hPal, FALSE);
     
129.         RealizePalette(hDC);
     
130.     }
     
131.     // 获取该调色板下新的像素值
     
132.     ::GetDIBits(hDC, hBitmap, 0, (UINT)bm.bmHeight, (LPSTR)lpbi + sizeof(BITMAPINFOHEADER) + dwPaletteSize, (BITMAPINFO*)lpbi, DIB_RGB_COLORS);
     
133.     //恢复调色板
     
134.     if (hOldPal)
     
135.     {
     
136.         SelectPalette(hDC, (HPALETTE)hOldPal, TRUE);
     
137.         RealizePalette(hDC);
     
138.         ::ReleaseDC(NULL, hDC);
     
139.     }
     
140.     //位图文件头结构
     
141.     BITMAPFILEHEADER bmfHdr;
     
142.     // "BM"
     
143.     bmfHdr.bfType = 0x4D42;
     
144.     // 设置位图文件头
     
145.     DWORD dwDIBSize = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER) + dwPaletteSize + dwBmBitsSize;
     
146.     bmfHdr.bfSize = dwDIBSize;
     
147.     bmfHdr.bfReserved1 = 0;
     
148.     bmfHdr.bfReserved2 = 0;
     
149.     bmfHdr.bfOffBits = (DWORD)sizeof(BITMAPFILEHEADER) + (DWORD)sizeof(BITMAPINFOHEADER) + dwPaletteSize;
     
150.     //创建位图文件
     
151.     HANDLE hFile = CreateFile(_T("D:\\a.bmp"), GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL | FILE_FLAG_SEQUENTIAL_SCAN, NULL);
     
152.     DWORD dwWritten;
     
153.     // 写入位图文件头
     
154.     WriteFile(hFile, (LPSTR)&bmfHdr, sizeof(BITMAPFILEHEADER), &dwWritten, NULL);
     
155.     // 写入位图文件其余内容
     
156.     WriteFile(hFile, (LPSTR)lpbi, dwDIBSize, &dwWritten, NULL);
     
157.     //清除
     
158.     GlobalUnlock(hDib);
     
159.     GlobalFree(hDib);
     
160.     CloseHandle(hFile);
     
161. }
     
162. 
     
163. 
     

复制代码