haiku.git

/*
 * Copyright 2006-2008 Haiku Inc. All rights reserved.
 * Distributed under the terms of the MIT license.
 *
 * Authors:
 *		Ithamar R. Adema <ithamar@unet.nl>
 *		James Urquhart
 *		Stephan Aßmus <superstippi@gmx.de>
 */
#ifndef PARTITIONLIST_H
#define PARTITIONLIST_H


#include <ColumnListView.h>
#include <ColumnTypes.h>
#include <Partition.h>


class BPartition;


// A field type displaying both a bitmap and a string so that the
// tree display looks nicer (both text and bitmap are indented)
class BBitmapStringField : public BStringField {
	typedef BStringField Inherited;
public:
								BBitmapStringField(BBitmap* bitmap,
									const char* string);
	virtual						~BBitmapStringField();

			void				SetBitmap(BBitmap* bitmap);
			const BBitmap*		Bitmap() const
									{ return fBitmap; }

private:
			BBitmap*			fBitmap;
};


// BColumn for PartitionListView which knows how to render
// a BBitmapStringField
class PartitionColumn : public BTitledColumn {
	typedef BTitledColumn Inherited;
public:
								PartitionColumn(const char* title,
									float width, float minWidth,
									float maxWidth, uint32 truncateMode,
									alignment align = B_ALIGN_LEFT);

	virtual	void				DrawField(BField* field, BRect rect,
									BView* parent);
	virtual float				GetPreferredWidth(BField* field, BView* parent) const;

	virtual	bool				AcceptsField(const BField* field) const;

	static	void				InitTextMargin(BView* parent);

private:
			uint32				fTruncateMode;
	static	float				sTextMargin;
};


// BRow for the PartitionListView
class PartitionListRow : public BRow {
	typedef BRow Inherited;
public:
								PartitionListRow(BPartition* partition);
								PartitionListRow(partition_id parentID,
									partition_id id, off_t offset, off_t size);

			partition_id		ID() const
									{ return fPartitionID; }
			partition_id		ParentID() const
									{ return fParentID; }
			off_t				Offset() const
									{ return fOffset; }
			off_t				Size() const
									{ return fSize; }

			const char*			DevicePath();

private:
			partition_id		fPartitionID;
			partition_id		fParentID;
			off_t				fOffset;
			off_t				fSize;
};


class PartitionListView : public BColumnListView {
	typedef BColumnListView Inherited;
public:
								PartitionListView();

	virtual	void				AttachedToWindow();

	virtual	bool				InitiateDrag(BPoint rowPoint, bool wasSelected);
	virtual	void 				ExpandOrCollapse(BRow* row, bool expand);

			PartitionListRow*	FindRow(partition_id id,
									PartitionListRow* parent = NULL);
			PartitionListRow*	AddPartition(BPartition* partition);
			PartitionListRow*	AddSpace(partition_id parent,
									partition_id id, off_t offset, off_t size);

	virtual BSize				PreferredSize();

private:
			int32				_InsertIndexForOffset(PartitionListRow* parent,
									off_t offset) const;
};


#endif // PARTITIONLIST_H