xref: /haiku/src/apps/drivesetup/DiskView.h (revision 441ad0b87ec9d4bbf62f769ab6abf7bd6d902605)

/*
 * Copyright 2007-2008 Haiku Inc. All rights reserved.
 * Distributed under the terms of the MIT license.
 */
#ifndef DISK_VIEW_H
#define DISK_VIEW_H


#include <DiskDevice.h>
#include <View.h>

#include "Support.h"


class DiskView : public BView {
	typedef BView Inherited;
public:
								DiskView(const BRect& frame,
									uint32 resizeMode,
									SpaceIDMap& spaceIDMap);
	virtual						~DiskView();

	// BView interface
	virtual	void				Draw(BRect updateRect);

			void				SetDiskCount(int32 count);
			void				SetDisk(BDiskDevice* disk,
									partition_id selectedPartition);

			void				ForceUpdate();
private:
			int32				fDiskCount;
			BDiskDevice*		fDisk;
			SpaceIDMap&			fSpaceIDMap;

			class PartitionLayout;
			PartitionLayout*	fPartitionLayout;
};


#endif // DISK_VIEW_H