tek/app/tek.rs

#![feature(
    adt_const_params,
    associated_type_defaults,
    closure_lifetime_binder,
    if_let_guard,
    impl_trait_in_assoc_type,
    trait_alias,
    type_alias_impl_trait,
    type_changing_struct_update,
)]

#![allow(
    clippy::unit_arg
)]

#[cfg(test)] mod tek_test;

mod tek_bind; pub use self::tek_bind::*;
mod tek_cfg;  pub use self::tek_cfg::*;
mod tek_deps; pub use self::tek_deps::*;
mod tek_mode; pub use self::tek_mode::*;
mod tek_view; pub use self::tek_view::*;

/// Total state
#[derive(Default, Debug)]
pub struct App {
    /// Base color.
    pub color:   ItemTheme,
    /// Must not be dropped for the duration of the process
    pub jack:    Jack<'static>,
    /// Display size
    pub size:    Measure<TuiOut>,
    /// Performance counter
    pub perf:    PerfModel,
    /// Available view modes and input bindings
    pub config:  Config,
    /// Currently selected mode
    pub mode:    Arc<Mode<Arc<str>>>,
    /// Undo history
    pub history: Vec<(AppCommand, Option<AppCommand>)>,
    /// Dialog overlay
    pub dialog:  Dialog,
    /// Contains all recently created clips.
    pub pool:    Pool,
    /// Contains the currently edited musical arrangement
    pub project: Arrangement,
}

audio!(
    |self: App, client, scope|{
        let t0 = self.perf.get_t0();
        self.clock().update_from_scope(scope).unwrap();
        let midi_in = self.project.midi_input_collect(scope);
        if let Some(editor) = &self.editor() {
            let mut pitch: Option<u7> = None;
            for port in midi_in.iter() {
                for event in port.iter() {
                    if let (_, Ok(LiveEvent::Midi {message: MidiMessage::NoteOn {key, ..}, ..}))
                        = event
                    {
                        pitch = Some(key.clone());
                    }
                }
            }
            if let Some(pitch) = pitch {
                editor.set_note_pos(pitch.as_int() as usize);
            }
        }
        let result = self.project.process_tracks(client, scope);
        self.perf.update_from_jack_scope(t0, scope);
        result
    };
    |self, event|{
        use JackEvent::*;
        match event {
            SampleRate(sr) => { self.clock().timebase.sr.set(sr as f64); },
            PortRegistration(_id, true) => {
                //let port = self.jack().port_by_id(id);
                //println!("\rport add: {id} {port:?}");
                //println!("\rport add: {id}");
            },
            PortRegistration(_id, false) => {
                /*println!("\rport del: {id}")*/
            },
            PortsConnected(_a, _b, true) => { /*println!("\rport conn: {a} {b}")*/ },
            PortsConnected(_a, _b, false) => { /*println!("\rport disc: {a} {b}")*/ },
            ClientRegistration(_id, true) => {},
            ClientRegistration(_id, false) => {},
            ThreadInit => {},
            XRun => {},
            GraphReorder => {},
            _ => { panic!("{event:?}"); }
        }
    }
);

// Allow source to be read as Literal string
dsl_ns!(App: Arc<str> {
    literal = |dsl|Ok(dsl.src()?.map(|x|x.into()));
});

// Provide boolean values.
dsl_ns!(App: bool {
    // TODO literal = ...
    word = |app| {
        ":mode/editor"         => app.project.editor.is_some(),
        ":focused/dialog"      => !matches!(app.dialog, Dialog::None),
        ":focused/message"     => matches!(app.dialog, Dialog::Message(..)),
        ":focused/add_device"  => matches!(app.dialog, Dialog::Device(..)),
        ":focused/browser"     => app.dialog.browser().is_some(),
        ":focused/pool/import" => matches!(app.pool.mode, Some(PoolMode::Import(..))),
        ":focused/pool/export" => matches!(app.pool.mode, Some(PoolMode::Export(..))),
        ":focused/pool/rename" => matches!(app.pool.mode, Some(PoolMode::Rename(..))),
        ":focused/pool/length" => matches!(app.pool.mode, Some(PoolMode::Length(..))),
        ":focused/clip"        => !app.editor_focused() && matches!(app.selection(),
            Selection::TrackClip{..}),
        ":focused/track"       => !app.editor_focused() && matches!(app.selection(),
            Selection::Track(..)),
        ":focused/scene"       => !app.editor_focused() && matches!(app.selection(),
            Selection::Scene(..)),
        ":focused/mix"         => !app.editor_focused() && matches!(app.selection(),
            Selection::Mix),
    };
});

// TODO: provide colors here
dsl_ns!(App: ItemTheme {});

dsl_ns!(App: Dialog {
    word = |app| {
        ":dialog/none"          => Dialog::None,
        ":dialog/options"       => Dialog::Options,
        ":dialog/device"        => Dialog::Device(0),
        ":dialog/device/prev"   => Dialog::Device(0),
        ":dialog/device/next"   => Dialog::Device(0),
        ":dialog/help"          => Dialog::Help(0),
        ":dialog/save"          => Dialog::Browse(BrowseTarget::SaveProject,
            Browse::new(None).unwrap().into()),
        ":dialog/load"          => Dialog::Browse(BrowseTarget::LoadProject,
            Browse::new(None).unwrap().into()),
        ":dialog/import/clip"   => Dialog::Browse(BrowseTarget::ImportClip(Default::default()),
            Browse::new(None).unwrap().into()),
        ":dialog/export/clip"   => Dialog::Browse(BrowseTarget::ExportClip(Default::default()),
            Browse::new(None).unwrap().into()),
        ":dialog/import/sample" => Dialog::Browse(BrowseTarget::ImportSample(Default::default()),
            Browse::new(None).unwrap().into()),
        ":dialog/export/sample" => Dialog::Browse(BrowseTarget::ExportSample(Default::default()),
            Browse::new(None).unwrap().into()),
    };
});

dsl_ns!(App: Selection {
    word = |app| {
        ":select/scene"      => app.selection().select_scene(app.tracks().len()),
        ":select/scene/next" => app.selection().select_scene_next(app.scenes().len()),
        ":select/scene/prev" => app.selection().select_scene_prev(),
        ":select/track"      => app.selection().select_track(app.tracks().len()),
        ":select/track/next" => app.selection().select_track_next(app.tracks().len()),
        ":select/track/prev" => app.selection().select_track_prev(),
    };
});

dsl_ns!(App: Color {
    word = |app| {
        ":color/bg" => Color::Rgb(28, 32, 36),
    };
    expr = |app| {
        "g"   (n: u8) => Color::Rgb(n, n, n),
        "rgb" (r: u8, g: u8, b: u8) => Color::Rgb(r, g, b),
    };
});

dsl_ns!(App: Option<u7> {
    word = |app| {
        ":editor/pitch" => Some(
            (app.editor().as_ref().map(|e|e.get_note_pos()).unwrap() as u8).into()
        )
    };
});

dsl_ns!(App: Option<usize> {
    word = |app| {
        ":selected/scene" => app.selection().scene(),
        ":selected/track" => app.selection().track(),
    };
});

dsl_ns!(App: Option<Arc<RwLock<MidiClip>>> {
    word = |app| {
        ":selected/clip" => if let Selection::TrackClip { track, scene } = app.selection() {
            app.scenes()[*scene].clips[*track].clone()
        } else {
            None
        }
    };
});

dsl_ns!(App: u8 {
    literal = |dsl|Ok(if let Some(src) = dsl.src()? {
        Some(to_number(src)? as u8)
    } else {
        None
    });
});

dsl_ns!(App: u16 {
    literal = |dsl|Ok(if let Some(src) = dsl.src()? {
        Some(to_number(src)? as u16)
    } else {
        None
    });
    word    = |app| {
        ":w/sidebar"       => app.project.w_sidebar(app.editor().is_some()),
        ":h/sample-detail" => 6.max(app.height() as u16 * 3 / 9),
    };
});

dsl_ns!(App: usize {
    literal = |dsl|Ok(if let Some(src) = dsl.src()? {
        Some(to_number(src)? as usize)
    } else {
        None
    });
    word    = |app| {
        ":scene-count"      => app.scenes().len(),
        ":track-count"      => app.tracks().len(),
        ":device-kind"      => app.dialog.device_kind().unwrap_or(0),
        ":device-kind/next" => app.dialog.device_kind_next().unwrap_or(0),
        ":device-kind/prev" => app.dialog.device_kind_prev().unwrap_or(0),
    };
});

dsl_ns!(App: isize {
    literal = |dsl|Ok(if let Some(src) = dsl.src()? {
        Some(to_number(src)? as isize)
    } else {
        None
    });
});

has!(Jack<'static>:      |self: App|self.jack);
has!(Pool:               |self: App|self.pool);
has!(Dialog:             |self: App|self.dialog);
has!(Clock:              |self: App|self.project.clock);
has!(Option<MidiEditor>: |self: App|self.project.editor);
has!(Selection:          |self: App|self.project.selection);
has!(Vec<MidiInput>:     |self: App|self.project.midi_ins);
has!(Vec<MidiOutput>:    |self: App|self.project.midi_outs);
has!(Vec<Scene>:         |self: App|self.project.scenes);
has!(Vec<Track>:         |self: App|self.project.tracks);
has!(Measure<TuiOut>:    |self: App|self.size);
has_clips!(              |self: App|self.pool.clips);
maybe_has!(Track: |self: App| { MaybeHas::<Track>::get(&self.project) };
                              { MaybeHas::<Track>::get_mut(&mut self.project) });
maybe_has!(Scene: |self: App| { MaybeHas::<Scene>::get(&self.project) };
                              { MaybeHas::<Scene>::get_mut(&mut self.project) });

impl HasClipsSize for App {
    fn clips_size   (&self) -> &Measure<TuiOut> { &self.project.size_inner }
}
impl HasTrackScroll for App {
    fn track_scroll (&self) -> usize { self.project.track_scroll() }
}
impl HasSceneScroll for App {
    fn scene_scroll (&self) -> usize { self.project.scene_scroll() }
}
impl HasJack<'static> for App {
    fn jack         (&self) -> &Jack<'static> { &self.jack }
}
impl ScenesView for App {
    fn w_side       (&self) -> u16 { 20 }
    fn w_mid        (&self) -> u16 { (self.width() as u16).saturating_sub(self.w_side()) }
    fn h_scenes     (&self) -> u16 { (self.height() as u16).saturating_sub(20) }
}


impl App {

    pub fn editor_focused (&self) -> bool {
        false
    }

    pub fn toggle_dialog (&mut self, mut dialog: Dialog) -> Dialog {
        std::mem::swap(&mut self.dialog, &mut dialog);
        dialog
    }

    pub fn toggle_editor (&mut self, value: Option<bool>) {
        //FIXME: self.editing.store(value.unwrap_or_else(||!self.is_editing()), Relaxed);
        let value = value.unwrap_or_else(||!self.editor().is_some());
        if value {
            // Create new clip in pool when entering empty cell
            if let Selection::TrackClip { track, scene } = *self.selection()
            && let Some(scene) = self.project.scenes.get_mut(scene)
            && let Some(slot)  = scene.clips.get_mut(track)
            && slot.is_none()
            && let Some(track) = self.project.tracks.get_mut(track)
            {
                let (index, mut clip) = self.pool.add_new_clip();
                // autocolor: new clip colors from scene and track color
                let color = track.color.base.mix(scene.color.base, 0.5);
                clip.write().unwrap().color = ItemColor::random_near(color, 0.2).into();
                if let Some(editor) = &mut self.project.editor {
                    editor.set_clip(Some(&clip));
                }
                *slot = Some(clip.clone());
                //Some(clip)
            } else {
                //None
            }
        } else if let Selection::TrackClip { track, scene } = *self.selection()
            && let Some(scene) = self.project.scenes.get_mut(scene)
            && let Some(slot) = scene.clips.get_mut(track)
            && let Some(clip) = slot.as_mut()
        {
            // Remove clip from arrangement when exiting empty clip editor
            let mut swapped = None;
            if clip.read().unwrap().count_midi_messages() == 0 {
                std::mem::swap(&mut swapped, slot);
            }
            if let Some(clip) = swapped {
                self.pool.delete_clip(&clip.read().unwrap());
            }
        }
    }

    pub fn browser (&self) -> Option<&Browse> {
        if let Dialog::Browse(_, ref b) = self.dialog { Some(b) } else { None }
    }

    pub fn device_pick (&mut self, index: usize) {
        self.dialog = Dialog::Device(index);
    }

    pub fn add_device (&mut self, index: usize) -> Usually<()> {
        match index {
            0 => {
                let name    = self.jack.with_client(|c|c.name().to_string());
                let midi    = self.project.track().expect("no active track").sequencer.midi_outs[0].port_name();
                let track   = self.track().expect("no active track");
                let port    = format!("{}/Sampler", &track.name);
                let connect = Connect::exact(format!("{name}:{midi}"));
                let sampler = if let Ok(sampler) = Sampler::new(
                    &self.jack, &port, &[connect], &[&[], &[]], &[&[], &[]]
                ) {
                    self.dialog = Dialog::None;
                    Device::Sampler(sampler)
                } else {
                    self.dialog = Dialog::Message("Failed to add device.".into());
                    return Err("failed to add device".into())
                };
                let track   = self.track_mut().expect("no active track");
                track.devices.push(sampler);
                Ok(())
            },
            1 => {
                todo!();
                Ok(())
            },
            _ => unreachable!(),
        }
    }

    pub fn update_clock (&self) {
        ViewCache::update_clock(&self.project.clock.view_cache, self.clock(), self.size.w() > 80)
    }
}

/// Various possible dialog modes.
#[derive(Debug, Clone, Default, PartialEq)]
pub enum Dialog {
    #[default] None,
    Help(usize),
    Menu(usize, MenuItems),
    Device(usize),
    Message(Arc<str>),
    Browse(BrowseTarget, Arc<Browse>),
    Options,
}

#[derive(Debug, Clone, Default, PartialEq)]
pub struct MenuItems(pub Arc<[MenuItem]>);

impl AsRef<Arc<[MenuItem]>> for MenuItems {
    fn as_ref (&self) -> &Arc<[MenuItem]> {
        &self.0
    }
}

#[derive(Clone)]
pub struct MenuItem(
    /// Label
    pub Arc<str>,
    /// Callback
    pub Arc<Box<dyn Fn(&mut App)->Usually<()> + Send + Sync>>
);

impl Default for MenuItem {
    fn default () -> Self { Self("".into(), Arc::new(Box::new(|_|Ok(())))) }
}

impl_debug!(MenuItem |self, w| { write!(w, "{}", &self.0) });

impl PartialEq for MenuItem {
    fn eq (&self, other: &Self) -> bool {
        self.0 == other.0
    }
}

impl Dialog {
    pub fn welcome () -> Self {
        Self::Menu(1, MenuItems([
            MenuItem("Resume session".into(),     Arc::new(Box::new(|_|Ok(())))),
            MenuItem("Create new session".into(), Arc::new(Box::new(|app|Ok({
                app.dialog = Dialog::None;
                app.mode = app.config.modes.clone().read().unwrap().get(":arranger").cloned().unwrap();
            })))),
            MenuItem("Load old session".into(),   Arc::new(Box::new(|_|Ok(())))),
        ].into()))
    }
    pub fn menu_next (&self) -> Self {
        match self {
            Self::Menu(index, items) => Self::Menu(wrap_inc(*index, items.0.len()), items.clone()),
            _ => Self::None
        }
    }
    pub fn menu_prev (&self) -> Self {
        match self {
            Self::Menu(index, items) => Self::Menu(wrap_dec(*index, items.0.len()), items.clone()),
            _ => Self::None
        }
    }
    pub fn menu_selected (&self) -> Option<usize> {
        if let Self::Menu(selected, _) = self { Some(*selected) } else { None }
    }
    pub fn device_kind (&self) -> Option<usize> {
        if let Self::Device(index) = self { Some(*index) } else { None }
    }
    pub fn device_kind_next (&self) -> Option<usize> {
        self.device_kind().map(|index|(index + 1) % device_kinds().len())
    }
    pub fn device_kind_prev (&self) -> Option<usize> {
        self.device_kind().map(|index|index.overflowing_sub(1).0.min(device_kinds().len().saturating_sub(1)))
    }
    pub fn message (&self) -> Option<&str> {
        todo!()
    }
    pub fn browser (&self) -> Option<&Arc<Browse>> {
        todo!()
    }
    pub fn browser_target (&self) -> Option<&BrowseTarget> {
        todo!()
    }
}

///////////////////////////////////////////////////////////////////////////////////////////////////
//has_editor!(|self: App|{
    //editor = self.editor;
    //editor_w = {
        //let size      = self.size.w();
        //let editor    = self.editor.as_ref().expect("missing editor");
        //let time_len  = editor.time_len().get();
        //let time_zoom = editor.time_zoom().get().max(1);
        //(5 + (time_len / time_zoom)).min(size.saturating_sub(20)).max(16)
    //};
    //editor_h = 15;
    //is_editing = self.editor.is_some();
//});