// grabscreen.cpp+
−
// +
−
// Copyright (c) 2008 - 2010 Accenture. All rights reserved.+
−
// This component and the accompanying materials are made available+
−
// under the terms of the "Eclipse Public License v1.0"+
−
// which accompanies this distribution, and is available+
−
// at the URL "http://www.eclipse.org/legal/epl-v10.html".+
−
// +
−
// Initial Contributors:+
−
// Accenture - Initial contribution+
−
//+
−
+
−
#include <fshell/ioutils.h>+
−
#include <w32std.h>+
−
#include <fbs.h>+
−
#include <bitdev.h>+
−
#include <imageconversion.h>+
−
+
−
using namespace IoUtils;+
−
+
−
+
−
class CCmdGrabscreen : public CCommandBase+
−
	{+
−
public:+
−
	static CCommandBase* NewLC();+
−
	~CCmdGrabscreen();+
−
private:+
−
	enum TFormat+
−
		{+
−
		EJpeg,+
−
		EGif,+
−
		EBmp,+
−
		EPng,+
−
		ERaw+
−
		};+
−
private:+
−
	CCmdGrabscreen();+
−
	void WriteEncodedL(TUid aEncodingUid);+
−
	void WriteRawL();+
−
private: // From CCommandBase.+
−
	virtual const TDesC& Name() const;+
−
	virtual void DoRunL();+
−
	virtual void ArgumentsL(RCommandArgumentList& aArguments);+
−
	virtual void OptionsL(RCommandOptionList& aOptions);+
−
	+
−
	virtual void RunL();+
−
private:+
−
	TFileName2 iFileName;+
−
	CImageEncoder* iImageEncoder;+
−
	HBufC8* iImageData;+
−
	CFrameImageData* iFrameImageData;+
−
	RFile iFile;+
−
	TFormat iFormat;+
−
	RArray<TInt> iRect;+
−
	TInt iScreen;+
−
	CFbsBitmap* iBitMap;+
−
	CWsScreenDevice* iDev;+
−
	RWsSession iWsSession;+
−
	TInt iQuality;+
−
	TBool iGreyscale;+
−
	};+
−
+
−
+
−
CCommandBase* CCmdGrabscreen::NewLC()+
−
	{+
−
	CCmdGrabscreen* self = new(ELeave) CCmdGrabscreen();+
−
	CleanupStack::PushL(self);+
−
	self->BaseConstructL();+
−
	return self;+
−
	}+
−
+
−
CCmdGrabscreen::~CCmdGrabscreen()+
−
	{+
−
	delete iImageEncoder;+
−
	delete iFrameImageData;+
−
	delete iDev;+
−
	delete iBitMap;+
−
	delete iImageData;+
−
	iFile.Close();+
−
	iRect.Close();+
−
	iWsSession.Close();+
−
	REComSession::FinalClose();+
−
	}+
−
+
−
CCmdGrabscreen::CCmdGrabscreen() +
−
	: CCommandBase(EManualComplete), iQuality(100)+
−
	{+
−
	}+
−
+
−
const TDesC& CCmdGrabscreen::Name() const+
−
	{+
−
	_LIT(KName, "grabscreen");	+
−
	return KName;+
−
	}+
−
+
−
void CCmdGrabscreen::DoRunL()+
−
	{+
−
	if ((iFormat == ERaw) && (!iArguments.IsPresent(0)))+
−
		{+
−
		PrintError(KErrArgument, _L("STDOUT writing is not supported for raw format; please specify a filename"));+
−
		DisplayHelp();+
−
		User::Leave(KErrArgument);+
−
		}+
−
	+
−
	LeaveIfErr(iWsSession.Connect(), _L("Couldn't connect to windowserver"));+
−
	iDev = new(ELeave) CWsScreenDevice(iWsSession);+
−
#ifdef SYMBIAN_WSERV_AND_CONE_MULTIPLE_SCREENS+
−
	const TInt numScreens = iWsSession.NumberOfScreens();+
−
	if (iScreen >= numScreens)+
−
		{+
−
		if (numScreens == 1)+
−
			{+
−
			LeaveIfErr(KErrArgument, _L("Invalid screen number - there is only 1 screen"));+
−
			}+
−
		else+
−
			{+
−
			LeaveIfErr(KErrArgument, _L("Invalid screen number - there are only %d screens"));+
−
			}+
−
		}+
−
#endif+
−
	LeaveIfErr(iDev->Construct(iScreen), _L("Couldn't construct CWsScreenDevice for screen %d"), iScreen);+
−
	iBitMap = new(ELeave) CFbsBitmap();+
−
+
−
	TRect rect;+
−
	if (iOptions.IsPresent(&iRect))+
−
		{+
−
		if (iRect.Count() != 4)+
−
			{+
−
			LeaveIfErr(KErrArgument, _L("Invalid --rect option - four integers must be specifed"));+
−
			}+
−
		rect.SetRect(iRect[0], iRect[1], iRect[2], iRect[3]);+
−
		}+
−
	else+
−
		{+
−
		rect.SetRect(TPoint(0, 0), iDev->SizeInPixels());+
−
		}+
−
	+
−
	LeaveIfErr(iBitMap->Create(rect.Size(), iDev->DisplayMode()), _L("Couldn't create fbsbitmap of size (%d,%d) for display mode %d"), rect.Width(), rect.Height(), iDev->DisplayMode());+
−
	User::LeaveIfError(iDev->CopyScreenToBitmap(iBitMap, rect));+
−
+
−
	switch (iFormat)+
−
		{+
−
		case EJpeg:+
−
			WriteEncodedL(KImageTypeJPGUid);+
−
			break;+
−
		case EGif:+
−
			WriteEncodedL(KImageTypeGIFUid);+
−
			break;+
−
		case EBmp:+
−
			WriteEncodedL(KImageTypeBMPUid);+
−
			break;+
−
		case EPng:+
−
			WriteEncodedL(KImageTypePNGUid);+
−
			break;+
−
		case ERaw:+
−
			WriteRawL();+
−
			break;+
−
		}+
−
	}+
−
+
−
void CCmdGrabscreen::WriteEncodedL(TUid aEncodingUid)+
−
	{+
−
	if (aEncodingUid == KImageTypeJPGUid)+
−
		{+
−
		TJpegImageData* imageData = new(ELeave) TJpegImageData();+
−
		CleanupStack::PushL(imageData);+
−
		imageData->iSampleScheme = iGreyscale ? TJpegImageData::EMonochrome : TJpegImageData::EColor444;+
−
		imageData->iQualityFactor = iQuality;+
−
		TRAPL(iFrameImageData = CFrameImageData::NewL(), _L("Couldn't create CFrameImageData"));+
−
		LeaveIfErr(iFrameImageData->AppendImageData(imageData), _L("Failed to write image data to TJpegImageData"));+
−
		CleanupStack::Pop(imageData);+
−
		}+
−
	else if (aEncodingUid == KImageTypeGIFUid)+
−
		{+
−
		if (iGreyscale)+
−
			{+
−
			PrintWarning(_L("Greyscale not supported for GIF images."));+
−
			}+
−
		if (iOptions.IsPresent(&iQuality))+
−
			{+
−
			PrintWarning(_L("Quality control not supported for GIF images."));+
−
			}+
−
		}+
−
	else if (aEncodingUid == KImageTypeBMPUid)+
−
		{+
−
		if (iGreyscale)+
−
			{+
−
			PrintWarning(_L("Greyscale not supported for BMP images."));+
−
			}+
−
		if (iOptions.IsPresent(&iQuality))+
−
			{+
−
			PrintWarning(_L("Quality control not supported for BMP images."));+
−
			}+
−
		}+
−
	else if (aEncodingUid == KImageTypePNGUid)+
−
		{+
−
		if (iOptions.IsPresent(&iQuality))+
−
			{+
−
			PrintWarning(_L("Quality control not supported for PNG images."));+
−
			}+
−
+
−
		TPngEncodeData* imageData = new(ELeave) TPngEncodeData();+
−
		CleanupStack::PushL(imageData);+
−
		imageData->iBitsPerPixel = iGreyscale ? 4 : 8;+
−
		imageData->iColor = !iGreyscale;+
−
		imageData->iLevel = TPngEncodeData::EBestCompression;+
−
		imageData->iPaletted = EFalse;+
−
		iFrameImageData = CFrameImageData::NewL();+
−
		User::LeaveIfError(iFrameImageData->AppendFrameData(imageData));+
−
		CleanupStack::Pop(imageData);+
−
		}+
−
		+
−
	if (iArguments.IsPresent(0))+
−
		{+
−
		LeaveIfErr(iFile.Replace(FsL(), iFileName, EFileWrite), _L("Unable to open \"%S\" for writing"), &iFileName);+
−
		iImageEncoder = CImageEncoder::FileNewL(iFile, CImageEncoder::EOptionNone, aEncodingUid);+
−
		}+
−
	else+
−
		{+
−
		iImageEncoder = CImageEncoder::DataNewL(iImageData, CImageEncoder::EOptionNone, aEncodingUid);+
−
		}+
−
	iImageEncoder->Convert(&iStatus, *iBitMap, iFrameImageData);+
−
	SetActive();+
−
	}+
−
	+
−
void CCmdGrabscreen::RunL()+
−
	{+
−
	LeaveIfErr(iStatus.Int(), _L("Error returned from CImageEncoder::Convert"));+
−
	if (!iArguments.IsPresent(0))+
−
		{+
−
		User::LeaveIfNull(iImageData);+
−
		+
−
		Stdout().SetModeL(RIoReadWriteHandle::EBinary);+
−
		+
−
		// we need to expand the 8 bit data when writing to stdout+
−
		static const TInt KBufferSize = 0x100;+
−
		TBuf<KBufferSize> buf;+
−
		TPtr8 imageData(iImageData->Des());+
−
		while (imageData.Length())+
−
			{+
−
			TInt thisCopyLen = Min(KBufferSize, imageData.Length());+
−
			buf.Copy(imageData.Left(thisCopyLen));+
−
			imageData.Set(imageData.MidTPtr(thisCopyLen));+
−
			Stdout().WriteL(buf);+
−
			}+
−
		}+
−
	Complete(KErrNone);+
−
	}+
−
+
−
void CCmdGrabscreen::WriteRawL()+
−
	{+
−
	if (iGreyscale)+
−
		{+
−
		PrintWarning(_L("Greyscale not supported for raw images."));+
−
		}+
−
	if (iOptions.IsPresent(&iQuality))+
−
		{+
−
		PrintWarning(_L("Quality control not supported for raw images."));+
−
		}+
−
	LeaveIfErr(iBitMap->Save(iFileName), _L("Unable to write \"%S\""), &iFileName);+
−
	Complete(KErrNone);+
−
	}+
−
+
−
void CCmdGrabscreen::ArgumentsL(RCommandArgumentList& aArguments)+
−
	{+
−
	_LIT(KArg, "file_name");+
−
	aArguments.AppendFileNameL(iFileName, KArg);+
−
	}+
−
+
−
void CCmdGrabscreen::OptionsL(RCommandOptionList& aOptions)+
−
	{+
−
	_LIT(KOptFormat, "format");+
−
	aOptions.AppendEnumL((TInt&)iFormat, KOptFormat);+
−
+
−
	_LIT(KOptRect, "rect");+
−
	aOptions.AppendIntL(iRect, KOptRect);+
−
+
−
	_LIT(KOptScreen, "screen");+
−
	aOptions.AppendIntL(iScreen, KOptScreen);+
−
+
−
	_LIT(KOptQuality, "quality");+
−
	aOptions.AppendIntL(iQuality, KOptQuality);+
−
+
−
	_LIT(KOptGreyscale, "greyscale");+
−
	aOptions.AppendBoolL(iGreyscale, KOptGreyscale);+
−
	}+
−
+
−
+
−
EXE_BOILER_PLATE(CCmdGrabscreen)+
−
+
−