select.hpp (1713B)
1 /** @addtogroup basic_math 2 * @{ 3 */ 4 /* 5 Copyright (C) 2016-2023 Dan Cazarin (https://www.kfrlib.com) 6 This file is part of KFR 7 8 KFR is free software: you can redistribute it and/or modify 9 it under the terms of the GNU General Public License as published by 10 the Free Software Foundation, either version 2 of the License, or 11 (at your option) any later version. 12 13 KFR is distributed in the hope that it will be useful, 14 but WITHOUT ANY WARRANTY; without even the implied warranty of 15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 16 GNU General Public License for more details. 17 18 You should have received a copy of the GNU General Public License 19 along with KFR. 20 21 If GPL is not suitable for your project, you must purchase a commercial license to use KFR. 22 Buying a commercial license is mandatory as soon as you develop commercial activities without 23 disclosing the source code of your own applications. 24 See https://www.kfrlib.com for details. 25 */ 26 #pragma once 27 28 #include "impl/select.hpp" 29 30 namespace kfr 31 { 32 inline namespace CMT_ARCH_NAME 33 { 34 35 /** 36 * @brief Returns x if m is true, otherwise return y. Order of the arguments is same as in ternary operator. 37 * @code 38 * return m ? x : y 39 * @endcode 40 */ 41 template <typename T1, size_t N, typename T2, typename T3, KFR_ENABLE_IF(is_numeric_args<T1, T2, T3>), 42 typename Tout = subtype<std::common_type_t<T2, T3>>> 43 KFR_INTRINSIC vec<Tout, N> select(const mask<T1, N>& m, const T2& x, const T3& y) 44 { 45 return intrinsics::select(bitcast<Tout>(cast<itype<Tout>>(bitcast<itype<T1>>(m.asvec()))).asmask(), 46 broadcastto<Tout>(x), broadcastto<Tout>(y)); 47 } 48 } // namespace CMT_ARCH_NAME 49 } // namespace kfr