#define _CRT_SECURE_NO_WARNINGS
#define _AFXDLL
#include<iostream>


#include <afxwin.h>
void Screen(char filename[])
{
	CDC *pDC;//螢幕DC
	pDC = CDC::FromHandle(GetDC(NULL));//獲取當前整個螢幕DC
	int BitPerPixel = pDC->GetDeviceCaps(BITSPIXEL);//獲得顏色模式
	int Width = pDC->GetDeviceCaps(HORZRES);
	int Height = pDC->GetDeviceCaps(VERTRES);
	printf("當前螢幕色彩模式為%d位色彩\n", BitPerPixel);
	printf("螢幕寬度：%d\n", Width);
	printf("螢幕高度：%d\n", Height);
	CDC memDC;//記憶體DC
	memDC.CreateCompatibleDC(pDC);
	CBitmap memBitmap, *oldmemBitmap;//建立和螢幕相容的bitmap
	memBitmap.CreateCompatibleBitmap(pDC, Width, Height);
	oldmemBitmap = memDC.SelectObject(&memBitmap);//將memBitmap選入記憶體DC
	memDC.BitBlt(0, 0, Width, Height, pDC, 0, 0, SRCCOPY);//複製螢幕影象到記憶體DC
														  //以下程式碼儲存memDC中的點陣圖到檔案
	BITMAP bmp;
	memBitmap.GetBitmap(&bmp);//獲得點陣圖資訊
	FILE *fp = fopen(filename, "w+b");
	BITMAPINFOHEADER bih = { 0 };//點陣圖資訊頭
	bih.biBitCount = bmp.bmBitsPixel;//每個畫素位元組大小
	bih.biCompression = BI_RGB;
	bih.biHeight = bmp.bmHeight;//高度
	bih.biPlanes = 1;
	bih.biSize = sizeof(BITMAPINFOHEADER);
	bih.biSizeImage = bmp.bmWidthBytes * bmp.bmHeight;//影象資料大小
	bih.biWidth = bmp.bmWidth;//寬度
	BITMAPFILEHEADER bfh = { 0 };//點陣圖檔案頭
	bfh.bfOffBits = sizeof(BITMAPFILEHEADER) + sizeof(BITMAPINFOHEADER);//到點陣圖資料的偏移量
	bfh.bfSize = bfh.bfOffBits + bmp.bmWidthBytes * bmp.bmHeight;//檔案總的大小
	bfh.bfType = (WORD)0x4d42;
	fwrite(&bfh, 1, sizeof(BITMAPFILEHEADER), fp);//寫入點陣圖檔案頭
	fwrite(&bih, 1, sizeof(BITMAPINFOHEADER), fp);//寫入點陣圖資訊頭
	byte * p = new byte[bmp.bmWidthBytes * bmp.bmHeight];//申請記憶體儲存點陣圖資料
	GetDIBits(memDC.m_hDC, (HBITMAP)memBitmap.m_hObject, 0, Height, p,
		(LPBITMAPINFO)&bih, DIB_RGB_COLORS);//獲取點陣圖資料
	fwrite(p, 1, bmp.bmWidthBytes * bmp.bmHeight, fp);//寫入點陣圖資料
	delete[] p;
	fclose(fp);
	memDC.SelectObject(oldmemBitmap);
}
int main()
{
	Screen("test.png");

	system("pause");
	return 0;
}