/*
 * 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(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