//---------------------------------------------------------------------------
#include <vcl.h>
#include <jpeg.hpp>
#pragma hdrstop

#include "Main.h"
#include "Config.h"
#include "About.h"
#include "Socket.cpp"
//---------------------------------------------------------------------------
#pragma resource "*.dfm"

#define CM_ENTIRESCREEN 201   //捕獲全屏
#define CM_ACTIVEWINDOW 202   //捕獲當前窗口

TMainForm *MainForm;
//---------------------------------------------------------------------------

__fastcall TMainForm::TMainForm(TComponent *Owner)
	: TForm(Owner)
{
}
//---------------------------------------------------------------------------

// 捕獲當前屏幕并保存到imagestream中
void CaptureImage(int options, int level, int cq, TMemoryStream* imgstream)
{
	LONG width,height;
	RECT capRect;
	HDC DesktopDC;
	switch (options) {
		case CM_ENTIRESCREEN:  // 捕獲整個屏幕
			// 取得桌面的矩形區域范圍
			GetWindowRect(GetDesktopWindow(),&capRect);  
		break;
		case CM_ACTIVEWINDOW:  // 捕獲當前窗口
			HWND ForegWin;
			ForegWin = GetForegroundWindow();  // 取得當前窗口句柄
			if (!ForegWin)
				ForegWin = GetDesktopWindow();
			GetWindowRect(ForegWin,&capRect);  // 取得當前窗口的矩形區域范圍
		break;
	}

	DesktopDC = GetDC(GetDesktopWindow());  // 創建內存設備描述表
	width = capRect.right - capRect.left;
	height = capRect.bottom - capRect.top;
	Graphics::TBitmap *bBitmap;  // 定義位圖變量
	try {
		bBitmap = new Graphics::TBitmap(); // 創建位圖
		bBitmap->Width=width;
		bBitmap->Height=height;
		if ((level>0)&&(level<8))
			bBitmap->PixelFormat = TPixelFormat(level);  // 設定色深
		// 拷貝屏幕的指定區域到位圖
		BitBlt(bBitmap->Canvas->Handle,0,0,width,height,DesktopDC,
			capRect.left,capRect.top,SRCCOPY);

		if (cq>=0) {
			TJPEGImage *jpeg;
			try {
				jpeg = new TJPEGImage;  // 創建JPEG圖象
				jpeg->CompressionQuality = cq;  // 設定圖象品質
				jpeg->Assign(bBitmap);  // 將位圖轉化為JPEG格式
				jpeg->SaveToStream(imgstream);  // 保存JPEG圖象信息
			}
			 __finally {
				delete jpeg;  // 釋放資源
			}
		}
		else {
			bBitmap->SaveToStream(imgstream);  // 保存位圖信息
		}
	}
	__finally {
		delete bBitmap;  // 釋放資源
	}
}

// 新建觀察窗口
void __fastcall TMainForm::CreateMDIChild(AnsiString sAddress, int CL, int CQ)
{
	TMDIChild *Child;

	//--- create a new MDI child window ----
	Child = new TMDIChild(Application);
	Child->RemoteAddress = sAddress;  // 主機IP
	Child->CL = CL;  // 色深
	Child->CQ = CQ;  // 品質
	Child->Caption = "來自'" + sAddress+"'";
	Child->Image1DblClick(Owner);  // 捕獲第一屏
}
//---------------------------------------------------------------------------

void __fastcall TMainForm::FileNew1Execute(TObject *Sender)
{
	ConfigForm->ShowModal();
	if (ConfigForm->Execute) {
		// 新建觀察窗口
		CreateMDIChild(ConfigForm->ComboBox1->Text,  // 主機IP
		ConfigForm->TrackBar1->Position,  // 色深
		ConfigForm->TrackBar2->Position*10);  // 品質
	}
}
//---------------------------------------------------------------------------


void __fastcall TMainForm::HelpAbout1Execute(TObject *Sender)
{
	AboutBox->ShowModal();
}
//---------------------------------------------------------------------------

void __fastcall TMainForm::FileExit1Execute(TObject *Sender)
{
	Close();
}
//---------------------------------------------------------------------------

// 捕獲并發送自己的屏幕圖象
void __fastcall TMainForm::ServerSocket1ClientRead(TObject *Sender,
      TCustomWinSocket *Socket)
{
	AnsiString sRecvString = Socket->ReceiveText();  // 保存接收到的字符串
	AnsiString sRemoteAddress = Socket->RemoteAddress;  // 保存對方IP
	int CL,CQ;
	u_short port;

	// 將接收到的字符串分解為接收端口、色深、品質3個參數
	int pos = sRecvString.Pos("\n");
	// 接收端口
	port = u_short(StrToIntDef(sRecvString.SubString(1,pos-1),0));
	sRecvString = sRecvString.SubString(pos+1,sRecvString.Length()-pos);

	pos = sRecvString.Pos("\n");
	// 色深
	CL = StrToIntDef(sRecvString.SubString(1,pos-1),0);
	sRecvString = sRecvString.SubString(pos+1,sRecvString.Length()-pos);

	pos = sRecvString.Pos("\n");
	// 品質
	CQ = StrToIntDef(sRecvString.SubString(1,pos-1),0);

	if (port) {
		TMemoryStream *ImageStream;  // 定義數據流
		try {
			ImageStream = new TMemoryStream;  // 分配內存
			// 捕獲當前屏幕并保存到ImageStream中
			CaptureImage(CM_ENTIRESCREEN, CL, CQ, ImageStream);
			// 發送ImageStream到接收端口
			if (!SendStream(sRemoteAddress, port, ImageStream))
				MessageBox(0,"發送數據流失敗","冰河",MB_ICONERROR);
		}
		__finally {
			delete ImageStream;  // 釋放資源
		}
	}
}
//---------------------------------------------------------------------------

void __fastcall TMainForm::ServerSocket1ClientError(TObject *Sender,
      TCustomWinSocket *Socket, TErrorEvent ErrorEvent, int &ErrorCode)
{
	ErrorCode = 0;
	MessageBox(0,"遠程連接出錯","冰河",MB_ICONERROR);  // 顯示出錯信息
}
//---------------------------------------------------------------------------

void __fastcall TMainForm::FormCreate(TObject *Sender)
{
	ServerSocket1->Port = LISTENPORT;  // 指定監聽端口
	ServerSocket1->Open();  // 開始監聽
}
//---------------------------------------------------------------------------






void __fastcall TMainForm::N2Click(TObject *Sender)
{
	// 是否打開端口允許別人抓屏
	N2->Checked = !N2->Checked;
	ServerSocket1->Active = N2->Checked;
}
//---------------------------------------------------------------------------