//
//  DiskController.hpp
//  Clock Signal
//
//  Created by Thomas Harte on 14/07/2016.
//  Copyright 2016 Thomas Harte. All rights reserved.
//

#pragma once

#include "../Drive.hpp"
#include "../DPLL/DigitalPhaseLockedLoop.hpp"
#include "../Track/PCMSegment.hpp"

#include "../../../ClockReceiver/ClockReceiver.hpp"
#include "../../../ClockReceiver/ClockingHintSource.hpp"

namespace Storage::Disk {

/*!
	Provides the shell for emulating a disk controller: something that is connected to a disk drive and uses a
	phase locked loop ('PLL') to decode a bit stream from the surface of the disk.

	Partly abstract; it is expected that subclasses will provide methods to deal with receiving a newly-recognised
	bit from the PLL and with crossing the index hole.

	TODO: communication of head size and permissible stepping extents, appropriate simulation of gain.
*/
class Controller:
	public ClockingHint::Source,
	private Drive::EventDelegate,
	private ClockingHint::Observer {
	protected:
		/*!
			Constructs a @c Controller that will be run at @c clock_rate.
		*/
		Controller(Cycles clock_rate);

		/*!
			Communicates to the PLL the expected length of a bit as a fraction of a second.
		*/
		void set_expected_bit_length(Time bit_length);

		/*!
			Advances the drive by @c number_of_cycles cycles.
		*/
		void run_for(const Cycles cycles);

		/*!
			Sets the current drive(s), by bit mask. Normally this will be exactly one, but some
			machines allow zero or multiple drives to be attached, with useless results.

			E.g. supply 1 to select drive 0, 2 to select drive 1, 4 to select drive 2, etc.
		*/
		void set_drive(int index_mask);

		/*!
			Adds a new drive to the drive list, returning its index.
		*/
		template<typename... Args> size_t emplace_drive(Args&&... args) {
			drives_.emplace_back(new Drive(std::forward<Args>(args)...));
			drives_.back()->set_clocking_hint_observer(this);
			return drives_.size() - 1;
		}

		/*!
			Adds @c count new drives to the drive list, returning the index of the final one added.
		*/
		template<typename... Args> size_t emplace_drives(size_t count, Args&&... args) {
			while(count--) {
				emplace_drive(std::forward<Args>(args)...);
			}
			return drives_.size() - 1;
		}

		/*!
			Should be implemented by subclasses; communicates each bit that the PLL recognises.
		*/
		virtual void process_input_bit(int value) = 0;

		/*!
			Should be implemented by subclasses; communicates that the index hole has been reached.
		*/
		virtual void process_index_hole() = 0;

		/*!
			Should be implemented by subclasses if they implement writing; communicates that
			all bits supplied to write_bit have now been written.
		*/
		virtual void process_write_completed() override;

		/*!
			Puts the controller and the drive returned by get_drive() into write mode, supplying to
			the drive the current bit length.

			While the controller is in write mode it disconnects the PLL. So subclasses will not
			receive any calls to @c process_input_bit.

			@param clamp_to_index_hole If @c true then writing will automatically be truncated by
			the index hole. Writing will continue over the index hole otherwise.
		*/
		void begin_writing(bool clamp_to_index_hole);

		/*!
			Puts the drive returned by get_drive() out of write mode, and marks the controller
			as no longer being in write mode.
		*/
		void end_writing();

		/*!
			@returns @c true if the controller is in reading mode; @c false otherwise.
		*/
		bool is_reading();

		/*!
			Returns the connected drive or, if none is connected, an invented one. No guarantees are
			made about the lifetime or the exclusivity of the invented drive.
		*/
		Drive &get_drive();

		Drive &get_drive(size_t index) {
			return *drives_[index];
		}

		void for_all_drives(const std::function<void(Drive &, size_t)> &func) {
			size_t index = 0;
			for(auto &drive: drives_) {
				func(*drive, index);
				++index;
			}
		}

		/*!
			As per ClockingHint::Source.
		*/
		ClockingHint::Preference preferred_clocking() const override;

	private:
		Time bit_length_;
		Cycles::IntType clock_rate_multiplier_ = 1;
		Cycles::IntType clock_rate_ = 1;

		bool is_reading_ = true;

		DigitalPhaseLockedLoop<Controller> pll_;
		friend DigitalPhaseLockedLoop<Controller>;

		Drive empty_drive_;
		std::vector<std::unique_ptr<Drive>> drives_;
		Drive *drive_;
		int drive_selection_mask_ = 0xff;

		// ClockingHint::Observer.
		void set_component_prefers_clocking(ClockingHint::Source *component, ClockingHint::Preference clocking) final;

		// for Drive::EventDelegate
		void process_event(const Drive::Event &event) final;
		void advance(const Cycles cycles) final;

		// to satisfy DigitalPhaseLockedLoop::Delegate
		void digital_phase_locked_loop_output_bit(int value);
};

}