lib_ts_chainalign 2.1.2

A chaining-based sequence-to-sequence aligner that accounts for template switches
Documentation 
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

use std::fmt::Display;

use crate::{
    alignment::{
        coordinates::AlignmentCoordinates,
        ts_kind::{TsDescendant, TsKind},
    },
    anchors::secondary::SecondaryAnchor,
};

/// A primary anchor.
///
/// This is an anchor between the two sequences in forward direction.
///
/// The anchor is ordered by its minimum ordinate first, then by its first ordinate and finally by its second ordinate.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub struct PrimaryAnchor {
    pub(super) seq1: usize,
    pub(super) seq2: usize,
}

impl PrimaryAnchor {
    pub fn new(seq1: usize, seq2: usize) -> Self {
        Self { seq1, seq2 }
    }

    pub fn new_from_start(alignment_coordinates: &AlignmentCoordinates) -> Self {
        Self::new(
            alignment_coordinates.primary_ordinate_a().unwrap(),
            alignment_coordinates.primary_ordinate_b().unwrap(),
        )
    }

    pub fn new_from_end(alignment_coordinates: &AlignmentCoordinates, k: usize) -> Self {
        Self::new(
            alignment_coordinates
                .primary_ordinate_a()
                .unwrap()
                .checked_sub(k)
                .unwrap(),
            alignment_coordinates
                .primary_ordinate_b()
                .unwrap()
                .checked_sub(k)
                .unwrap(),
        )
    }

    pub fn start(&self) -> AlignmentCoordinates {
        AlignmentCoordinates::Primary {
            a: self.seq1,
            b: self.seq2,
        }
    }

    pub fn end(&self, k: usize) -> AlignmentCoordinates {
        AlignmentCoordinates::Primary {
            a: self.seq1 + k,
            b: self.seq2 + k,
        }
    }

    pub fn chaining_gaps(&self, second: &Self, k: usize) -> Option<(usize, usize)> {
        let gap_start = self.end(k);
        let gap_end = second.start();
        primary_chaining_gaps(gap_start, gap_end)
    }

    pub fn chaining_gaps_from_start(&self, start: AlignmentCoordinates) -> (usize, usize) {
        let gap_end = self.start();
        primary_chaining_gaps(start, gap_end)
            .unwrap_or_else(|| panic!("self: {self}, start: {start}"))
    }

    pub fn chaining_gaps_to_end(&self, end: AlignmentCoordinates, k: usize) -> (usize, usize) {
        let gap_start = self.end(k);
        primary_chaining_gaps(gap_start, end)
            .unwrap_or_else(|| panic!("self: {self}, end: {end}, k: {k}"))
    }

    /// Returns the gap in the descendant for the 12-jump from this anchor to the given anchor.
    pub fn chaining_jump_gap(
        &self,
        second: &SecondaryAnchor,
        ts_kind: TsKind,
        k: usize,
    ) -> Option<usize> {
        let gap_start = self.end(k);
        let gap_end = second.start(ts_kind);

        let gap_start = match ts_kind.descendant {
            TsDescendant::Seq1 => gap_start.primary_ordinate_a().unwrap(),
            TsDescendant::Seq2 => gap_start.primary_ordinate_b().unwrap(),
        };
        let gap_end = gap_end.secondary_ordinate_descendant().unwrap();

        gap_end.checked_sub(gap_start)
    }

    pub fn is_direct_predecessor_of(&self, successor: &Self) -> bool {
        self.seq1 + 1 == successor.seq1 && self.seq2 + 1 == successor.seq2
    }
}

fn primary_chaining_gaps(
    gap_start: AlignmentCoordinates,
    gap_end: AlignmentCoordinates,
) -> Option<(usize, usize)> {
    let gap1 = gap_end
        .primary_ordinate_a()
        .unwrap()
        .checked_sub(gap_start.primary_ordinate_a().unwrap())?;
    let gap2 = gap_end
        .primary_ordinate_b()
        .unwrap()
        .checked_sub(gap_start.primary_ordinate_b().unwrap())?;

    Some((gap1, gap2))
}

impl Display for PrimaryAnchor {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "({}, {})", self.seq1, self.seq2)
    }
}

impl From<(usize, usize)> for PrimaryAnchor {
    fn from(value: (usize, usize)) -> Self {
        Self::new(value.0, value.1)
    }
}

impl Ord for PrimaryAnchor {
    fn cmp(&self, other: &Self) -> std::cmp::Ordering {
        self.seq1
            .min(self.seq2)
            .cmp(&other.seq1.min(other.seq2))
            .then_with(|| self.seq1.cmp(&other.seq1))
            .then_with(|| self.seq2.cmp(&other.seq2))
    }
}

impl PartialOrd for PrimaryAnchor {
    fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> {
        Some(self.cmp(other))
    }
}