substitute_helpers.hpp

Go to the documentation of this file.

#ifndef TATAMI_ISOMETRIC_UNARY_SUBSTITUTE_HELPERS_H
#define TATAMI_ISOMETRIC_UNARY_SUBSTITUTE_HELPERS_H
 
#include "../compare_utils.hpp"
#include <vector>
#include <limits>
 
namespace tatami {
 
template<CompareOperation op_, typename Value_>
bool delayed_substitute_is_sparse(Value_ compared, Value_ substitute) {
    return !delayed_compare<op_, Value_>(0, compared) || substitute == 0;
}
 
template<CompareOperation op_, typename Value_>
void delayed_substitute_run(Value_& val, Value_ compared, Value_ substitute) {
    if (delayed_compare<op_, Value_>(val, compared)) {
        val = substitute;
    }
}
 
template<CompareOperation op_, typename Value_, typename Index_>
void delayed_substitute_run_simple(Value_* buffer, Index_ length, Value_ compared, Value_ substitute) {
    for (Index_ i = 0; i < length; ++i) {
        delayed_substitute_run<op_, Value_>(buffer[i], compared, substitute);
    }
}
template<CompareOperation op_, typename Value_>
class DelayedUnaryIsometricSubstituteScalar {
public:
    DelayedUnaryIsometricSubstituteScalar(Value_ compared, Value_ substitute) : my_compared(compared), my_substitute(substitute) {
        my_sparse = delayed_substitute_is_sparse<op_, Value_>(my_compared, my_substitute);
    }
 
private:
    Value_ my_compared, my_substitute;
    bool my_sparse;
 
public:
    static constexpr bool is_basic = false;
 
    bool is_sparse() const {
        return my_sparse;
    }
public:
    template<typename Index_>
    void dense(bool, Index_, Index_, Index_ length, const Value_*, Value_* output) const {
        delayed_substitute_run_simple<op_, Value_>(output, length, my_compared, my_substitute);
    }
 
    template<typename Index_>
    void dense(bool, Index_, const std::vector<Index_>& indices, const Value_*, Value_* output) const {
        delayed_substitute_run_simple<op_, Value_>(output, static_cast<Index_>(indices.size()), my_compared, my_substitute);
    }
 
    template<typename Index_>
    void sparse(bool, Index_, Index_ number, const Value_*, const Index_*, Value_* output_value) const {
        delayed_substitute_run_simple<op_, Value_>(output_value, number, my_compared, my_substitute);
    }
 
    template<typename, typename, typename Index_>
    Value_ fill(bool, Index_) const {
        if (my_sparse) {
            return 0;
        } else {
            return my_substitute;
        }
    }
};
 
template<CompareOperation op_, typename Value_, typename Vector_>
class DelayedUnaryIsometricSubstituteVector {
public:
    DelayedUnaryIsometricSubstituteVector(Vector_ compared, Vector_ substitute, bool by_row) : 
        my_compared(std::move(compared)), my_substitute(std::move(substitute)), my_by_row(by_row) 
    {
        for (size_t i = 0, end = my_compared.size(); i < end; ++i) {
            if (!delayed_substitute_is_sparse<op_, Value_>(my_compared[i], my_substitute[i])) {
                 my_sparse = false;
                 break;
             }
        }
    }
 
private:
    Vector_ my_compared, my_substitute;
    bool my_by_row;
    bool my_sparse = true;
 
public:
    static constexpr bool is_basic = false;
 
    bool zero_depends_on_row() const {
        return my_by_row;
    }
 
    bool zero_depends_on_column() const {
        return !my_by_row;
    }
 
    bool non_zero_depends_on_row() const {
        return my_by_row;
    }
 
    bool non_zero_depends_on_column() const {
        return !my_by_row;
    }
 
    bool is_sparse() const {
        return my_sparse;
    }
public:
    template<typename Index_>
    void dense(bool row, Index_ idx, Index_ start, Index_ length, const Value_*, Value_* output) const {
        if (row == my_by_row) {
            delayed_substitute_run_simple<op_, Value_>(output, length, my_compared[idx], my_substitute[idx]);
        } else {
            for (Index_ i = 0; i < length; ++i) {
                Index_ is = i + start;
                delayed_substitute_run<op_, Value_>(output[i], my_compared[is], my_substitute[is]);
            }
        }
    }
 
    template<typename Index_>
    void dense(bool row, Index_ idx, const std::vector<Index_>& indices, const Value_*, Value_* output) const {
        if (row == my_by_row) {
            delayed_substitute_run_simple<op_, Value_>(output, static_cast<Index_>(indices.size()), my_compared[idx], my_substitute[idx]);
        } else {
            Index_ length = indices.size();
            for (Index_ i = 0; i < length; ++i) {
                auto ii = indices[i];
                delayed_substitute_run<op_, Value_>(output[i], my_compared[ii], my_substitute[ii]);
            }
        }
    }
 
    template<typename Index_>
    void sparse(bool row, Index_ idx, Index_ number, const Value_*, const Index_* indices, Value_* output_value) const {
        if (row == my_by_row) {
            delayed_substitute_run_simple<op_, Value_>(output_value, number, my_compared[idx], my_substitute[idx]);
        } else {
            for (Index_ i = 0; i < number; ++i) {
                auto ii = indices[i];
                delayed_substitute_run<op_, Value_>(output_value[i], my_compared[ii], my_substitute[ii]);
            }
        }
    }
 
    template<typename, typename, typename Index_>
    Value_ fill(bool row, Index_ idx) const {
        if (row == my_by_row) {
            auto sub = my_substitute[idx];
            if (!delayed_compare<op_, Value_>(0, my_compared[idx])) {
                return 0;
            } else {
                return sub;
            }
        } else {
            // We should only get to this point if it's sparse, otherwise no
            // single fill value would work across the length of my_compared.
            return 0;
        }
    }
};
 
template<typename Value_ = double>
DelayedUnaryIsometricSubstituteScalar<CompareOperation::EQUAL, Value_> 
    make_DelayedUnaryIsometricSubstituteEqualScalar(Value_ compared, Value_ substitute) 
{
    return DelayedUnaryIsometricSubstituteScalar<CompareOperation::EQUAL, Value_>(std::move(compared), std::move(substitute));
}
 
template<typename Value_ = double>
DelayedUnaryIsometricSubstituteScalar<CompareOperation::GREATER_THAN, Value_> 
    make_DelayedUnaryIsometricSubstituteGreaterThanScalar(Value_ compared, Value_ substitute)
{
    return DelayedUnaryIsometricSubstituteScalar<CompareOperation::GREATER_THAN, Value_>(std::move(compared), std::move(substitute));
}
 
template<typename Value_ = double>
DelayedUnaryIsometricSubstituteScalar<CompareOperation::LESS_THAN, Value_>
    make_DelayedUnaryIsometricSubstituteLessThanScalar(Value_ compared, Value_ substitute)
{
    return DelayedUnaryIsometricSubstituteScalar<CompareOperation::LESS_THAN, Value_>(std::move(compared), std::move(substitute));
}
 
template<typename Value_ = double>
DelayedUnaryIsometricSubstituteScalar<CompareOperation::GREATER_THAN_OR_EQUAL, Value_> 
    make_DelayedUnaryIsometricSubstituteGreaterThanOrEqualScalar(Value_ compared, Value_ substitute)
{
    return DelayedUnaryIsometricSubstituteScalar<CompareOperation::GREATER_THAN_OR_EQUAL, Value_>(std::move(compared), std::move(substitute));
}
 
template<typename Value_ = double>
DelayedUnaryIsometricSubstituteScalar<CompareOperation::LESS_THAN_OR_EQUAL, Value_>
    make_DelayedUnaryIsometricSubstituteLessThanOrEqualScalar(Value_ compared, Value_ substitute)
{
    return DelayedUnaryIsometricSubstituteScalar<CompareOperation::LESS_THAN_OR_EQUAL, Value_>(std::move(compared), std::move(substitute));
}
 
template<typename Value_ = double>
DelayedUnaryIsometricSubstituteScalar<CompareOperation::NOT_EQUAL, Value_>
    make_DelayedUnaryIsometricSubstituteNotEqualScalar(Value_ compared, Value_ substitute) 
{
    return DelayedUnaryIsometricSubstituteScalar<CompareOperation::NOT_EQUAL, Value_>(std::move(compared), std::move(substitute));
}
 
template<typename Value_ = double, typename Vector_ = std::vector<Value_> >
DelayedUnaryIsometricSubstituteVector<CompareOperation::EQUAL, Value_, Vector_> 
    make_DelayedUnaryIsometricSubstituteEqualVector(Vector_ compared, Vector_ substitute, bool by_row) 
{
    return DelayedUnaryIsometricSubstituteVector<CompareOperation::EQUAL, Value_, Vector_>(std::move(compared), std::move(substitute), by_row);
}
 
template<typename Value_ = double, typename Vector_ = std::vector<Value_> >
DelayedUnaryIsometricSubstituteVector<CompareOperation::GREATER_THAN, Value_, Vector_> 
    make_DelayedUnaryIsometricSubstituteGreaterThanVector(Vector_ compared, Vector_ substitute, bool by_row)
{
    return DelayedUnaryIsometricSubstituteVector<CompareOperation::GREATER_THAN, Value_, Vector_>(std::move(compared), std::move(substitute), by_row);
}
 
template<typename Value_ = double, typename Vector_ = std::vector<Value_> >
DelayedUnaryIsometricSubstituteVector<CompareOperation::LESS_THAN, Value_, Vector_> 
    make_DelayedUnaryIsometricSubstituteLessThanVector(Vector_ compared, Vector_ substitute, bool by_row) 
{
    return DelayedUnaryIsometricSubstituteVector<CompareOperation::LESS_THAN, Value_, Vector_>(std::move(compared), std::move(substitute), by_row);
}
 
template<typename Value_ = double, typename Vector_ = std::vector<Value_> >
DelayedUnaryIsometricSubstituteVector<CompareOperation::GREATER_THAN_OR_EQUAL, Value_, Vector_> 
    make_DelayedUnaryIsometricSubstituteGreaterThanOrEqualVector(Vector_ compared, Vector_ substitute, bool by_row) 
{
    return DelayedUnaryIsometricSubstituteVector<CompareOperation::GREATER_THAN_OR_EQUAL, Value_, Vector_>(std::move(compared), std::move(substitute), by_row);
}
 
template<typename Value_ = double, typename Vector_ = std::vector<Value_> >
DelayedUnaryIsometricSubstituteVector<CompareOperation::LESS_THAN_OR_EQUAL, Value_, Vector_> 
    make_DelayedUnaryIsometricSubstituteLessThanOrEqualVector(Vector_ compared, Vector_ substitute, bool by_row) 
{
    return DelayedUnaryIsometricSubstituteVector<CompareOperation::LESS_THAN_OR_EQUAL, Value_, Vector_>(std::move(compared), std::move(substitute), by_row);
}
 
template<typename Value_ = double, typename Vector_ = std::vector<Value_> >
DelayedUnaryIsometricSubstituteVector<CompareOperation::NOT_EQUAL, Value_, Vector_> 
    make_DelayedUnaryIsometricSubstituteNotEqualVector(Vector_ compared, Vector_ substitute, bool by_row) 
{
    return DelayedUnaryIsometricSubstituteVector<CompareOperation::NOT_EQUAL, Value_, Vector_>(std::move(compared), std::move(substitute), by_row);
}
 
template<SpecialCompareOperation op_, bool pass_, typename Value_>
bool delayed_special_substitute_is_sparse(Value_ substitute) {
    return !delayed_special_compare<op_, pass_, Value_>(0) || substitute == 0;
}
 
template<SpecialCompareOperation op_, bool pass_, typename Value_>
void delayed_special_substitute_run(Value_& val, Value_ substitute) {
    if (delayed_special_compare<op_, pass_, Value_>(val)) {
        val = substitute;
    }
}
 
template<SpecialCompareOperation op_, bool pass_, typename Value_, typename Index_>
void delayed_special_substitute_run_simple(Value_* buffer, Index_ length, Value_ substitute) {
    for (Index_ i = 0; i < length; ++i) {
        delayed_special_substitute_run<op_, pass_, Value_>(buffer[i], substitute);
    }
}
template<SpecialCompareOperation op_, bool pass_, typename Value_>
class DelayedUnaryIsometricSpecialSubstitute {
public:
    DelayedUnaryIsometricSpecialSubstitute(Value_ substitute) : my_substitute(substitute) {
        my_sparse = delayed_special_substitute_is_sparse<op_, pass_, Value_>(my_substitute);
    }
 
private:
    Value_ my_substitute;
    bool my_sparse;
 
public:
    static constexpr bool is_basic = false;
 
    bool is_sparse() const {
        return my_sparse;
    }
public:
    template<typename Index_>
    void dense(bool, Index_, Index_, Index_ length, const Value_*, Value_* output) const {
        delayed_special_substitute_run_simple<op_, pass_, Value_>(output, length, my_substitute);
    }
 
    template<typename Index_>
    void dense(bool, Index_, const std::vector<Index_>& indices, const Value_*, Value_* output) const {
        delayed_special_substitute_run_simple<op_, pass_, Value_>(output, static_cast<Index_>(indices.size()), my_substitute);
    }
 
    template<typename Index_>
    void sparse(bool, Index_, Index_ number, const Value_*, const Index_*, Value_* output_value) const {
        delayed_special_substitute_run_simple<op_, pass_, Value_>(output_value, number, my_substitute);
    }
 
    template<typename, typename, typename Index_>
    Value_ fill(bool, Index_) const {
        if (my_sparse) {
            return 0;
        } else {
            return my_substitute;
        }
    }
};
 
template<bool pass_ = true, typename Value_ = double>
DelayedUnaryIsometricSpecialSubstitute<SpecialCompareOperation::ISNAN, pass_, Value_> make_DelayedUnaryIsometricSubstituteIsnan(Value_ substitute) {
    return DelayedUnaryIsometricSpecialSubstitute<SpecialCompareOperation::ISNAN, pass_, Value_>(substitute);
}
 
template<bool pass_ = true, typename Value_ = double>
DelayedUnaryIsometricSpecialSubstitute<SpecialCompareOperation::ISINF, pass_, Value_> make_DelayedUnaryIsometricSubstituteIsinf(Value_ substitute) {
    return DelayedUnaryIsometricSpecialSubstitute<SpecialCompareOperation::ISINF, pass_, Value_>(substitute);
}
 
template<bool pass_ = true, typename Value_ = double>
DelayedUnaryIsometricSpecialSubstitute<SpecialCompareOperation::ISFINITE, pass_, Value_> make_DelayedUnaryIsometricSubstituteIsfinite(Value_ substitute) {
    return DelayedUnaryIsometricSpecialSubstitute<SpecialCompareOperation::ISFINITE, pass_, Value_>(substitute);
}
 
}
 
#endif