assignable.hpp

Go to the documentation of this file.

//=======================================================================
// Copyright (c) 2014-2023 Baptiste Wicht
// Distributed under the terms of the MIT License.
// (See accompanying file LICENSE or copy at
//  http://opensource.org/licenses/MIT)
//=======================================================================

#pragma once

#include <concepts>
#include "etl/impl/transpose.hpp"
#include "etl/impl/fft.hpp"

namespace etl {

template <typename D, typename V>
struct assignable {
    using derived_t  = D; 
    using value_type = V; 

    template <etl_expr E>
    derived_t& operator=(E&& e) {
        validate_assign(as_derived(), e);

        if constexpr (!decay_traits<E>::is_linear) {
            if (e.alias(as_derived())) {
                auto tmp = etl::force_temporary_dim_only(as_derived());

                e.assign_to(tmp);

                as_derived() = tmp;
            } else {
                e.assign_to(as_derived());
            }
        } else {
            e.assign_to(as_derived());
        }

        return as_derived();
    }

    derived_t& operator=(const value_type& v) {
        if constexpr (decay_traits<derived_t>::is_direct) {
            direct_fill(as_derived(), v);
        } else {
            std::fill(as_derived().begin(), as_derived().end(), v);
        }

        return as_derived();
    }

    template <std_container Container>
    derived_t& operator=(const Container& vec) requires(std::convertible_to<typename Container::value_type, value_type>) {
        validate_assign(as_derived(), vec);

        std::copy(vec.begin(), vec.end(), as_derived().begin());

        return as_derived();
    }

private:
    derived_t& as_derived() noexcept {
        return *static_cast<derived_t*>(this);
    }
};

} //end of namespace etl