Program Listing for File generate_data_matrix.hpp

↰ Return to documentation for file (fwdpp/ts/generate_data_matrix.hpp)

#ifndef FWDPP_TS_GENERATE_DATA_MATRIX_HPP
#define FWDPP_TS_GENERATE_DATA_MATRIX_HPP

#include <vector>
#include <type_traits>
#include <cstdint>
#include <stdexcept>
#include <fwdpp/data_matrix.hpp>
#include "site_visitor.hpp"
#include "exceptions.hpp"
#include "marginal_tree_functions/samples.hpp"
#include "detail/generate_data_matrix_details.hpp"

namespace fwdpp
{
    namespace ts
    {
        template <typename TableCollectionType, typename Samples>
        data_matrix
        generate_data_matrix(const TableCollectionType& tables, Samples&& samples,
                             const bool record_neutral, const bool record_selected,
                             const bool skip_fixed, const double start,
                             const double stop)
        {
            if (!(stop > start))
                {
                    throw std::invalid_argument("invalid position range");
                }
            std::size_t samplesize = samples.size();
            std::vector<std::int8_t> genotypes(samplesize);
            site_visitor<TableCollectionType> sv(tables, samples);
            data_matrix rv(samplesize);
            decltype(sv()) itr;
            while ((itr = sv()) != end(sv))
                {
                    if (itr->position >= start && itr->position < stop)
                        {
                            const auto& tree = sv.current_tree();
                            detail::process_site_range(tree, itr, sv.get_mutations(),
                                                       record_neutral, record_selected,
                                                       skip_fixed, genotypes, rv);
                        }
                }
            return rv;
        }

        template <typename TableCollectionType, typename Samples>
        data_matrix
        generate_data_matrix(const TableCollectionType& tables, Samples&& samples,
                             const bool record_neutral, const bool record_selected,
                             const bool skip_fixed)
        {
            return generate_data_matrix(tables, std::forward<Samples>(samples),
                                        record_neutral, record_selected, skip_fixed, 0.,
                                        tables.genome_length());
        }

    } // namespace ts
} // namespace fwdpp

#endif