/*
 * Copyright 2007, Haiku. All rights reserved.
 * Distributed under the terms of the MIT License.
 *
 * Authors:
 *		Michael Pfeiffer
 */

#ifndef _PICTURE_TEST_H
#define _PICTURE_TEST_H

#include <InterfaceKit.h>
#include <String.h>

typedef void (draw_func)(BView *view, BRect frame);

class PictureTest {

public:
	PictureTest();
	virtual ~PictureTest();
	
	void SetColorSpace(color_space colorSpace) { fColorSpace = colorSpace; }

	bool Test(draw_func* func, BRect frame);
	
	const char *ErrorMessage() const { return fErrorMessage.String(); }

	BBitmap *DirectBitmap(bool detach = false);
	BBitmap *BitmapFromPicture(bool detach = false);
	BBitmap *BitmapFromRestoredPicture(bool detach = false);

protected:
	virtual BPicture *SaveAndRestore(BPicture *picture) = 0;
	void SetErrorMessage(const char* message);

private:
	
	void CleanUp();
	
	BPicture *RecordPicture(draw_func* func, BRect frame);

	BBitmap *CreateBitmap(draw_func* func, BRect frame);
	BBitmap *CreateBitmap(BPicture *picture, BRect frame);

	bool IsSame(BBitmap *bitmap1, BBitmap *bitmap2, BString &reason);

	color_space fColorSpace;

	BBitmap *fDirectBitmap;
	BBitmap *fBitmapFromPicture;
	BBitmap *fBitmapFromRestoredPicture;
	
	BString fErrorMessage;
};

class FlattenPictureTest : public PictureTest
{
public:
	FlattenPictureTest();

protected:
	BPicture *SaveAndRestore(BPicture *picture);
};

class ArchivePictureTest : public PictureTest
{
public:
	ArchivePictureTest();

protected:
	BPicture *SaveAndRestore(BPicture *picture);
};

#endif