ops.hpp

Go to the documentation of this file.

// include/nk/builtin/ops.hpp
//
// Arithmetic, relational, logical operators and array transformations (MATLAB parity).
#pragma once
 
#include <memory_resource>
#include <nk/value/value.hpp>
#include <nk/value/span.hpp>
 
namespace nk::builtin {

 

 
// ── Arithmetic Operations ───────────────────────────────────────────────────

Value plus(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value minus(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value times(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value mtimes(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value rdivide(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value ldivide(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value mrdivide(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value mldivide(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value power(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value elementPower(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value mpower(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value uplus(const Value &a, std::pmr::memory_resource *mr = nullptr);

Value uminus(const Value &a, std::pmr::memory_resource *mr = nullptr);

enum class TranspOp {
    None,       
    Transpose,  
    CTranspose  
};

Value pagemtimes(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value pagemtimes(const Value &x, TranspOp tx, const Value &y, TranspOp ty, std::pmr::memory_resource *mr = nullptr);
 
// ── Relational & Comparison Operations ──────────────────────────────────────

Value eq(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value ne(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value lt(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value le(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value gt(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value ge(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);
 
// ── Logical Operations & Predicates ─────────────────────────────────────────

Value logical_and(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

inline Value and_op(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr)
{
    return logical_and(a, b, mr);
}

Value logical_or(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

inline Value or_op(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr)
{
    return logical_or(a, b, mr);
}

Value logical_not(const Value &a, std::pmr::memory_resource *mr = nullptr);

inline Value not_op(const Value &a, std::pmr::memory_resource *mr = nullptr)
{
    return logical_not(a, mr);
}

Value logical_xor(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

inline Value xor_op(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr)
{
    return logical_xor(a, b, mr);
}

Value any(const Value &x, int dim = 0, std::pmr::memory_resource *mr = nullptr);

Value all(const Value &x, int dim = 0, std::pmr::memory_resource *mr = nullptr);
 
// ── Transposition ──────────────────────────────────────────────────────────

Value ctranspose(const Value &a, std::pmr::memory_resource *mr = nullptr);

Value transpose(const Value &a, std::pmr::memory_resource *mr = nullptr);

inline Value logicalAnd(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr) { return logical_and(a, b, mr); }

inline Value logicalOr(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr) { return logical_or(a, b, mr); }

inline Value logicalNot(const Value &a, std::pmr::memory_resource *mr = nullptr) { return logical_not(a, mr); }

inline Value logicalXor(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr) { return logical_xor(a, b, mr); }
 
inline Value transposeNC(const Value &a, std::pmr::memory_resource *mr = nullptr) { return transpose(a, mr); }

Value bitand_(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value bitor_(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value bitxor_(const Value &a, const Value &b, std::pmr::memory_resource *mr = nullptr);

Value bitshift(const Value &a, const Value &k, std::pmr::memory_resource *mr = nullptr);

Value bitcmp(const Value &a, int width = 64, std::pmr::memory_resource *mr = nullptr);

Value bitset(const Value &a, const Value &n, const Value &val = Value(), std::pmr::memory_resource *mr = nullptr);

Value bitget(const Value &a, const Value &n, std::pmr::memory_resource *mr = nullptr);
 

} // namespace nk::builtin