make_DelayedSubset.hpp

Go to the documentation of this file.

#ifndef TATAMI_MAKE_DELAYED_SUBSET_HPP
#define TATAMI_MAKE_DELAYED_SUBSET_HPP
 
#include "DelayedSubsetSortedUnique.hpp"
#include "DelayedSubsetSorted.hpp"
#include "DelayedSubsetUnique.hpp"
#include "DelayedSubset.hpp"
#include "DelayedSubsetBlock.hpp"
#include "../utils/ArrayView.hpp"
 
#include <algorithm>
#include <memory>
 
namespace tatami {
 
template<typename Value_, typename Index_, class SubsetStorage_>
std::shared_ptr<Matrix<Value_, Index_> > make_DelayedSubset(std::shared_ptr<const Matrix<Value_, Index_> > matrix, SubsetStorage_ subset, bool by_row) {
    bool is_unsorted = false;
    for (Index_ i = 0, end = subset.size(); i < end; ++i) {
        if (i) {
            if (subset[i] < subset[i-1]) {
                is_unsorted = true;
                break;
            }
        }
    }
 
    if (!is_unsorted) {
        bool has_duplicates = false;
        for (Index_ i = 0, end = subset.size(); i < end; ++i) {
            if (i) {
                if (subset[i] == subset[i-1]) {
                    has_duplicates = true;
                    break;
                }
            }
        }
 
        if (!has_duplicates) {
            bool consecutive = true;
            for (Index_ i = 0, end = subset.size(); i < end; ++i) {
                if (i) {
                    if (subset[i] > subset[i-1] + 1) {
                        consecutive = false;
                        break;
                    }
                }
            }
 
            if (consecutive) {
                auto start = (subset.size() ? subset[0] : 0);
                return std::shared_ptr<Matrix<Value_, Index_> >(
                    new DelayedSubsetBlock<Value_, Index_>(std::move(matrix), start, subset.size(), by_row)
                );
            } else {
                return std::shared_ptr<Matrix<Value_, Index_> >(
                    new DelayedSubsetSortedUnique<Value_, Index_, SubsetStorage_>(std::move(matrix), std::move(subset), by_row, false)
                );
            }
        } else {
            return std::shared_ptr<Matrix<Value_, Index_> >(
                new DelayedSubsetSorted<Value_, Index_, SubsetStorage_>(std::move(matrix), std::move(subset), by_row, false)
            );
        }
    }
 
    bool has_duplicates = false;
    std::vector<unsigned char> accumulated(by_row ? matrix->nrow() : matrix->ncol());
    for (Index_ i = 0, end = subset.size(); i < end; ++i) {
        auto& found = accumulated[subset[i]];
        if (found) {
            has_duplicates = true;
            break;
        } else {
            found = 1;
        }
    }
 
    if (!has_duplicates) {
        return std::shared_ptr<Matrix<Value_, Index_> >(
            new DelayedSubsetUnique<Value_, Index_, SubsetStorage_>(std::move(matrix), std::move(subset), by_row, false)
        );
    } else {
        return std::shared_ptr<Matrix<Value_, Index_> >(
            new DelayedSubset<Value_, Index_, SubsetStorage_>(std::move(matrix), std::move(subset), by_row)
        );
    }
}
 
template<typename Value_, typename Index_, class SubsetStorage_>
std::shared_ptr<Matrix<Value_, Index_> > make_DelayedSubset(std::shared_ptr<Matrix<Value_, Index_> > matrix, SubsetStorage_ subset, bool by_row) {
    return make_DelayedSubset<Value_, Index_, SubsetStorage_>(std::shared_ptr<const Matrix<Value_, Index_> >(std::move(matrix)), std::move(subset), by_row);
}
// Back-compatibility only.
template<int margin_, typename Value_, typename Index_, class SubsetStorage_>
std::shared_ptr<Matrix<Value_, Index_> > make_DelayedSubset(std::shared_ptr<const Matrix<Value_, Index_> > matrix, SubsetStorage_ subset) {
    return make_DelayedSubset<Value_, Index_, SubsetStorage_>(std::move(matrix), std::move(subset), margin_ == 0);
}
 
template<int margin_, typename Value_, typename Index_, class SubsetStorage_>
std::shared_ptr<Matrix<Value_, Index_> > make_DelayedSubset(std::shared_ptr<Matrix<Value_, Index_> > matrix, SubsetStorage_ subset) {
    return make_DelayedSubset<Value_, Index_, SubsetStorage_>(std::move(matrix), std::move(subset), margin_ == 0);
}
}
 
#endif