lib_tsalign 2.0.2

A 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

use compact_genome::interface::sequence::GenomeSequence;

use crate::a_star_aligner::template_switch_distance::{
    AlignmentType, Context, Identifier, TemplateSwitchPrimary,
};

use super::{AlignmentStrategy, AlignmentStrategySelector, primary_match::PrimaryMatchStrategy};

pub trait TemplateSwitchDescendantStrategy: AlignmentStrategy {
    /// True if a TSM with this descendant can be started.
    fn is_descendant_allowed(&self, descendant: TemplateSwitchPrimary) -> bool;
}

#[derive(Debug, Clone, Copy, Eq, PartialEq)]
pub struct AnyTemplateSwitchDescendantStrategy;

#[derive(Debug, Clone, Copy, Eq, PartialEq)]
pub struct OnlyEqualTemplateSwitchDescendantStrategy {
    allowed_descendant: Option<TemplateSwitchPrimary>,
}

impl TemplateSwitchDescendantStrategy for AnyTemplateSwitchDescendantStrategy {
    fn is_descendant_allowed(&self, _descendant: TemplateSwitchPrimary) -> bool {
        true
    }
}

impl TemplateSwitchDescendantStrategy for OnlyEqualTemplateSwitchDescendantStrategy {
    fn is_descendant_allowed(&self, descendant: TemplateSwitchPrimary) -> bool {
        if let Some(allowed_descendant) = self.allowed_descendant {
            allowed_descendant == descendant
        } else {
            true
        }
    }
}

impl AlignmentStrategy for AnyTemplateSwitchDescendantStrategy {
    fn create_root<
        SubsequenceType: GenomeSequence<Strategies::Alphabet, SubsequenceType> + ?Sized,
        Strategies: AlignmentStrategySelector,
    >(
        _context: &Context<'_, '_, SubsequenceType, Strategies>,
    ) -> Self {
        Self
    }

    fn generate_successor<
        SubsequenceType: GenomeSequence<Strategies::Alphabet, SubsequenceType> + ?Sized,
        Strategies: AlignmentStrategySelector,
    >(
        &self,
        _identifier: Identifier<
            <<Strategies as AlignmentStrategySelector>::PrimaryMatch as PrimaryMatchStrategy<
                <Strategies as AlignmentStrategySelector>::Cost,
            >>::IdentifierPrimaryExtraData,
        >,
        _alignment_type: AlignmentType,
        _context: &Context<'_, '_, SubsequenceType, Strategies>,
    ) -> Self {
        *self
    }
}

impl AlignmentStrategy for OnlyEqualTemplateSwitchDescendantStrategy {
    fn create_root<
        SubsequenceType: GenomeSequence<Strategies::Alphabet, SubsequenceType> + ?Sized,
        Strategies: AlignmentStrategySelector,
    >(
        _context: &Context<'_, '_, SubsequenceType, Strategies>,
    ) -> Self {
        Self {
            allowed_descendant: None,
        }
    }

    fn generate_successor<
        SubsequenceType: GenomeSequence<Strategies::Alphabet, SubsequenceType> + ?Sized,
        Strategies: AlignmentStrategySelector,
    >(
        &self,
        identifier: Identifier<
            <<Strategies as AlignmentStrategySelector>::PrimaryMatch as PrimaryMatchStrategy<
                <Strategies as AlignmentStrategySelector>::Cost,
            >>::IdentifierPrimaryExtraData,
        >,
        _alignment_type: AlignmentType,
        _context: &Context<'_, '_, SubsequenceType, Strategies>,
    ) -> Self {
        let mut successor = *self;
        if let Identifier::TemplateSwitchEntrance {
            template_switch_primary: descendant,
            ..
        } = identifier
        {
            if let Some(allowed_descendant) = successor.allowed_descendant {
                assert_eq!(allowed_descendant, descendant);
            }

            successor.allowed_descendant = Some(descendant);
        }

        successor
    }
}