//
//  DiskController.hpp
//  Clock Signal
//
//  Created by Thomas Harte on 14/01/2020.
//  Copyright © 2020 Thomas Harte. All rights reserved.
//

#ifndef DiskController_h
#define DiskController_h

namespace Oric {

class DiskController: public WD::WD1770 {
	public:
		DiskController(WD::WD1770::Personality personality, int clock_rate) :
			WD::WD1770(personality), clock_rate_(clock_rate) {}

		void set_disk(std::shared_ptr<Storage::Disk::Disk> disk, int d) {
			const size_t drive = size_t(d);
			if(!drives_[drive]) {
				drives_[drive] = std::make_unique<Storage::Disk::Drive>(clock_rate_, 300, 2);
				if(drive == selected_drive_) set_drive(drives_[drive]);
			}
			drives_[drive]->set_disk(disk);
		}

		enum PagingFlags {
			/// Indicates that overlay RAM is enabled, implying no ROM is visible.
			OverlayRAMEnable		=	(1 << 0),

			/// Indicates that the BASIC ROM is disabled, implying that the disk
			/// controller's ROM fills its space.
			BASICDisable			=	(1 << 1)
		};

		struct Delegate: public WD1770::Delegate {
			virtual void disk_controller_did_change_paging_flags(DiskController *controller) = 0;
		};
		inline void set_delegate(Delegate *delegate)	{
			delegate_ = delegate;
			WD1770::set_delegate(delegate);
			if(delegate) delegate->disk_controller_did_change_paging_flags(this);
		}
		inline int get_paging_flags() {
			return paging_flags_;
		}

	protected:
		inline void set_paging_flags(int new_flags) {
			if(paging_flags_ == new_flags) return;
			paging_flags_ = new_flags;
			if(delegate_) {
				delegate_->disk_controller_did_change_paging_flags(this);
			}
		}

		std::array<std::shared_ptr<Storage::Disk::Drive>, 4> drives_;
		size_t selected_drive_;
		void select_drive(size_t drive) {
			if(drive != selected_drive_) {
				selected_drive_ = drive;
				set_drive(drives_[selected_drive_]);
			}
		}

	private:
		int paging_flags_ = 0;
		Delegate *delegate_ = nullptr;
		int clock_rate_;

};


};


#endif /* DiskController_h */