1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149

use std::sync::{Arc, Weak};

use crate::prelude::{Tab, Track};
use notation_proto::prelude::{
    Duration, Entry, EntryPassMode, FrettedEntry4, FrettedEntry6, ProtoEntry, TrackKind, Units,
};

#[derive(Copy, Clone, Debug)]
pub struct ModelEntryProps {
    pub index: usize,
    pub tied_units: Units,
}

#[derive(Debug)]
pub struct ModelEntry {
    pub track: Weak<Track>,
    pub proto: Arc<ProtoEntry>,
    pub props: ModelEntryProps,
}
impl ModelEntry {
    pub fn new(
        track: Weak<Track>,
        proto: Arc<ProtoEntry>,
        index: usize,
        tied_units: Units,
    ) -> Self {
        let props = ModelEntryProps { index, tied_units };
        Self {
            track,
            proto,
            props,
        }
    }
}
impl Entry for ModelEntry {
    fn duration(&self) -> notation_proto::prelude::Duration {
        self.proto.duration()
    }
    fn prev_is_tie(&self) -> bool {
        self.prev().map(|x| x.proto.is_core_tie()).unwrap_or(false)
    }
    fn next_is_tie(&self) -> bool {
        self.next().map(|x| x.proto.is_core_tie()).unwrap_or(false)
    }
    fn tied_units(&self) -> Units {
        self.props.tied_units
    }
    fn pass_mode(&self) -> EntryPassMode {
        self.proto.pass_mode()
    }
}
impl ModelEntry {
    pub fn track(&self) -> Option<Arc<Track>> {
        self.track.upgrade().map(|x| x.clone())
    }
    pub fn tab(&self) -> Option<Arc<Tab>> {
        self.track().and_then(|x| x.tab())
    }
    pub fn as_fretted6(&self) -> Option<&FrettedEntry6> {
        self.proto.as_fretted6()
    }
    pub fn as_fretted4(&self) -> Option<&FrettedEntry4> {
        self.proto.as_fretted4()
    }
    pub fn prev(&self) -> Option<Arc<ModelEntry>> {
        if self.props.index == 0 {
            None
        } else if let Some(track) = self.track.upgrade() {
            track.entries.get(self.props.index - 1).map(|x| x.clone())
        } else {
            None
        }
    }
    pub fn next(&self) -> Option<Arc<ModelEntry>> {
        if let Some(track) = self.track.upgrade() {
            track.entries.get(self.props.index + 1).map(|x| x.clone())
        } else {
            None
        }
    }
    pub fn prev_as_mark(&self) -> Option<String> {
        if let Some(entry) = self.prev() {
            entry.proto.as_mark().map(|x| x.clone())
        } else {
            None
        }
    }
    pub fn get_tied_prev(&self) -> Option<Arc<ModelEntry>> {
        if self.props.index <= 1 {
            return None;
        }
        if let Some(track) = self.track.upgrade() {
            if let Some(prev) = track.entries.get(self.props.index - 1) {
                if prev.proto.is_core_tie() {
                    for i in self.props.index - 2..=0 {
                        let entry = track.entries.get(i).unwrap();
                        if entry.duration() != Duration::Zero {
                            return Some(entry.clone());
                        }
                    }
                }
            }
        }
        None
    }
    pub fn get_tied_next(&self) -> Option<Arc<ModelEntry>> {
        if let Some(track) = self.track.upgrade() {
            if let Some(next) = track.entries.get(self.props.index + 1) {
                if next.proto.is_core_tie() {
                    for i in self.props.index + 2..track.entries.len() {
                        let entry = track.entries.get(i).unwrap();
                        if entry.duration() != Duration::Zero {
                            return Some(entry.clone());
                        }
                    }
                }
            }
        }
        None
    }
    pub fn track_id(&self) -> String {
        if let Some(track) = self.track.upgrade() {
            track.id.clone()
        } else {
            "".to_owned()
        }
    }
    pub fn track_kind(&self) -> TrackKind {
        if let Some(track) = self.track.upgrade() {
            track.kind.clone()
        } else {
            TrackKind::Custom("".to_owned())
        }
    }
    pub fn track_index(&self) -> Option<usize> {
        if let Some(track) = self.track.upgrade() {
            Some(track.props.index)
        } else {
            None
        }
    }
    pub fn get_track_entry<T, F: Fn(&ModelEntry) -> Option<T>>(&self, predicate: &F) -> Option<T> {
        if let Some(track) = self.track.upgrade() {
            track.get_entry(predicate)
        } else {
            None
        }
    }
}