alignment_trace_matrix_full.hpp

Go to the documentation of this file.

// -----------------------------------------------------------------------------------------------------
// Copyright (c) 2006-2021, Knut Reinert & Freie Universität Berlin
// Copyright (c) 2016-2021, Knut Reinert & MPI für molekulare Genetik
// This file may be used, modified and/or redistributed under the terms of the 3-clause BSD-License
// shipped with this file and also available at: https://github.com/seqan/seqan3/blob/master/LICENSE.md
// -----------------------------------------------------------------------------------------------------
 
#pragma once
 
#include <seqan3/std/iterator>
#include <seqan3/std/ranges>
 
#include <seqan3/alignment/matrix/detail/alignment_matrix_column_major_range_base.hpp>
#include <seqan3/alignment/matrix/detail/alignment_trace_matrix_base.hpp>
#include <seqan3/alignment/matrix/detail/alignment_trace_matrix_proxy.hpp>
#include <seqan3/alignment/matrix/detail/trace_iterator.hpp>
#include <seqan3/utility/views/zip.hpp>
 
namespace seqan3::detail
{
 
template <typename trace_t, bool coordinate_only = false>
class alignment_trace_matrix_full :
    protected alignment_trace_matrix_base<trace_t>,
    public alignment_matrix_column_major_range_base<alignment_trace_matrix_full<trace_t, coordinate_only>>
{
private:
    static_assert(std::same_as<trace_t, trace_directions> || simd_concept<trace_t>,
                  "Value type must either be a trace_directions object or a simd vector.");
 
    using matrix_base_t = alignment_trace_matrix_base<trace_t>;
    using range_base_t = alignment_matrix_column_major_range_base<alignment_trace_matrix_full<trace_t, coordinate_only>>;
 
    friend range_base_t;
 
protected:
    using typename matrix_base_t::element_type;
    using typename matrix_base_t::coordinate_type;
    using typename range_base_t::alignment_column_type;
    using column_data_view_type = std::conditional_t<coordinate_only,
                                        decltype(std::views::iota(coordinate_type{}, coordinate_type{})),
                                        decltype(views::zip(std::declval<std::span<element_type>>(),
                                                                std::declval<std::span<element_type>>(),
                                                                std::views::iota(coordinate_type{}, coordinate_type{})))>;
 
public:
    using value_type = alignment_trace_matrix_proxy<coordinate_type,
                                                    std::conditional_t<coordinate_only,
                                                                       detail::ignore_t const,
                                                                       trace_t>>;
    using reference = value_type;
    using iterator = typename range_base_t::iterator;
    using sentinel = typename range_base_t::sentinel;
    using typename matrix_base_t::size_type;
 
    constexpr alignment_trace_matrix_full() = default; 
    constexpr alignment_trace_matrix_full(alignment_trace_matrix_full const &) = default; 
    constexpr alignment_trace_matrix_full(alignment_trace_matrix_full &&) = default; 
    constexpr alignment_trace_matrix_full & operator=(alignment_trace_matrix_full const &) = default; 
    constexpr alignment_trace_matrix_full & operator=(alignment_trace_matrix_full &&) = default; 
    ~alignment_trace_matrix_full() = default; 
 
    template <std::ranges::forward_range first_sequence_t, std::ranges::forward_range second_sequence_t>
    constexpr alignment_trace_matrix_full(first_sequence_t && first,
                                          second_sequence_t && second,
                                          [[maybe_unused]] trace_t const initial_value = trace_t{})
    {
        matrix_base_t::num_cols = static_cast<size_type>(std::ranges::distance(first) + 1);
        matrix_base_t::num_rows = static_cast<size_type>(std::ranges::distance(second) + 1);
 
        if constexpr (!coordinate_only)
        {
            // Allocate the matrix here.
            matrix_base_t::data = typename matrix_base_t::pool_type{number_rows{matrix_base_t::num_rows},
                                                                    number_cols{matrix_base_t::num_cols}};
            matrix_base_t::cache_left.resize(matrix_base_t::num_rows, initial_value);
        }
    }
 
    auto trace_path(matrix_coordinate const & trace_begin)
    {
        static_assert(!coordinate_only, "Requested trace but storing the trace was disabled!");
 
        using matrix_iter_t = std::ranges::iterator_t<typename matrix_base_t::pool_type>;
        using trace_iterator_t = trace_iterator<matrix_iter_t>;
        using path_t = std::ranges::subrange<trace_iterator_t, std::default_sentinel_t>;
 
        if (trace_begin.row >= matrix_base_t::num_rows || trace_begin.col >= matrix_base_t::num_cols)
            throw std::invalid_argument{"The given coordinate exceeds the matrix in vertical or horizontal direction."};
 
        return path_t{trace_iterator_t{matrix_base_t::data.begin() + matrix_offset{trace_begin}},
                      std::default_sentinel};
    }
 
private:
    constexpr alignment_column_type initialise_column(size_type const column_index) noexcept
    {
        coordinate_type row_begin{column_index_type{column_index}, row_index_type{0u}};
        coordinate_type row_end{column_index_type{column_index}, row_index_type{matrix_base_t::num_rows}};
        if constexpr (coordinate_only)
        {
            return alignment_column_type{*this,
                                         column_data_view_type{std::views::iota(std::move(row_begin),
                                                                                std::move(row_end))}};
        }
        else
        {
            matrix_coordinate current_position{row_index_type{0u}, column_index_type{column_index}};
            auto col = views::zip(std::span<element_type>{std::addressof(matrix_base_t::data[current_position]),
                                                               matrix_base_t::num_rows},
                                  std::span<element_type>{matrix_base_t::cache_left},
                                  std::views::iota(std::move(row_begin), std::move(row_end)));
            return alignment_column_type{*this, column_data_view_type{col}};
        }
    }
 
    template <std::random_access_iterator iter_t>
    constexpr value_type make_proxy(iter_t host_iter) noexcept
    {
        if constexpr (coordinate_only)
        {
            return {*host_iter, std::ignore, std::ignore, std::ignore, std::ignore};
        }
        else
        {
            return {std::get<2>(*host_iter),  // the coordinate.
                    std::get<0>(*host_iter),  // the current entry.
                    std::get<1>(*host_iter),  // the last left cell to read from.
                    std::get<1>(*host_iter),  // the next left cell to write to.
                    matrix_base_t::cache_up,  // the last up cell to read/write from/to.
                    };
        }
    }
};
 
} // namespace seqan3::detail