cpclib-disc 0.6.0

use crate::cfg::DiscConfig;
use crate::edsk::{ExtendedDsk, Head};

/// Generate an edsk from the given configuration
pub fn build_disc_from_cfg(cfg: &DiscConfig) -> ExtendedDsk {
    let mut edsk = ExtendedDsk::default();

    // Feed the disc info table
    edsk.disc_information_bloc.creator_name = "RUST CPC - BND".to_string();
    assert_eq!(edsk.disc_information_bloc.creator_name.len(), 14);
    edsk.disc_information_bloc.number_of_heads = cfg.nb_heads;
    edsk.disc_information_bloc.number_of_tracks = cfg.nb_tracks;
    edsk.disc_information_bloc.track_size_table = cfg
        .track_idx_iterator()
        .map(|idx| {
            let info = cfg
                .track_information_for_track(*idx.0, idx.1)
                .unwrap_or_else(|| panic!("Unable to acquire information for track {:?}", idx));
            let sectors_size = info.sector_size as usize * info.number_of_sectors();
            let header_size = 256;
            ((sectors_size + header_size) / 256) as u8
        })
        .collect::<Vec<_>>();

    // Create the empty tracks -- to be filled in the next loop
    for (head, track_idx) in cfg.track_idx_iterator() {
        let mut track = edsk.track_list.add_empty_track();
        track.track_number = track_idx;

        track.head_number = match *head {
            Head::A | Head::Unspecified => Head::A,
            Head::B => Head::B
        }
        .into();
        track.track_size =
            u16::from(edsk.disc_information_bloc.track_size_table[track_idx as usize]) * 256 + 512;
        // XXX Need to see why I had to add these 512 bytes ...
    }

    // Update the tracks stuff
    for (&head, track_idx) in cfg.track_idx_iterator() {
        let track_info = edsk
            .get_track_information_mut(head, track_idx)
            .unwrap_or_else(|| {
                panic!(
                    "Unable to acquire track {} on head {:?} on the dsk",
                    track_idx, head
                )
            });
        let track_model = cfg.track_information_for_track(head, track_idx).unwrap();

        track_info.track_number = track_idx;
        track_info.head_number = track_model.head.into();
        track_info.data_rate = 1.into();
        track_info.recording_mode = 2.into();
        track_info.sector_size = track_model.sector_size_dsk_format();
        track_info.number_of_sectors = track_model.sector_id.len() as _;
        track_info.gap3_length = track_model.gap3; // TODO ensure a 8buts value is in the cfg
        track_info.filler_byte = 0xE5;
        track_info.sector_information_list.fill_with(
            &track_model.sector_id,
            &track_model.sector_id_head,
            track_idx,
            track_model.sector_size_dsk_format(),
            track_info.filler_byte
        )
    }
    edsk
}

#[allow(missing_docs)]
impl From<DiscConfig> for ExtendedDsk {
    fn from(config: DiscConfig) -> Self {
        build_disc_from_cfg(&config)
    }
}

#[allow(missing_docs)]
impl From<&DiscConfig> for ExtendedDsk {
    fn from(config: &DiscConfig) -> Self {
        build_disc_from_cfg(config)
    }
}

/// Generate a standard singled sided amsdos disc
pub fn single_head_data_dsk() -> ExtendedDsk {
    DiscConfig::single_head_data_format().into()
}

/// Generate a double sided amsdos disc
pub fn single_head_data42_dsk() -> ExtendedDsk {
    DiscConfig::single_head_data42_format().into()
}