swifft.hpp

Go to the documentation of this file.

/*
 * Copyright (C) 2020 Yaron Gvili and Gvili Tech Ltd.
 *
 * See the accompanying LICENSE.txt file for licensing information.
 */
#ifndef __LIBSWIFFT_SWIFFT_HPP__
#define __LIBSWIFFT_SWIFFT_HPP__

#include "libswifft/swifft.h"
#include <string.h>

namespace LibSwifft {

struct SWIFFT_ALIGN SwifftInput;
SwifftInput& Copy(SwifftInput &dst, const SwifftInput &src);
SwifftInput & Set(SwifftInput &lhs, const BitSequence value);

struct SWIFFT_ALIGN SwifftOutput;
SwifftOutput& Copy(SwifftOutput &dst, const SwifftOutput &src);
SwifftOutput & Set(SwifftOutput &lhs, const int16_t value);

struct SWIFFT_ALIGN SwifftCompact;
SwifftCompact& Copy(SwifftCompact &dst, const SwifftCompact &src);
SwifftCompact & Set(SwifftCompact &lhs, const BitSequence value);

struct SWIFFT_ALIGN SwifftInput {
    BitSequence data[SWIFFT_INPUT_BLOCK_SIZE];

    LIBSWIFFT_INLINE SwifftInput & operator=(const SwifftInput &rhs) { Copy(*this, rhs); return *this; }
    LIBSWIFFT_INLINE SwifftInput & operator=(const BitSequence value) { Set(*this, value); return *this; }

};

struct SWIFFT_ALIGN SwifftOutput {
    BitSequence data[SWIFFT_OUTPUT_BLOCK_SIZE];

    LIBSWIFFT_INLINE SwifftOutput & operator=(const SwifftOutput &rhs) { Copy(*this, rhs); return *this; }
    LIBSWIFFT_INLINE SwifftOutput & operator=(const int16_t value) { Set(*this, value); return *this; }

};

struct SWIFFT_ALIGN SwifftCompact {
    BitSequence data[SWIFFT_COMPACT_BLOCK_SIZE];

    LIBSWIFFT_INLINE SwifftCompact & operator=(const SwifftCompact &rhs) { Copy(*this, rhs); return *this; }
    LIBSWIFFT_INLINE SwifftCompact & operator=(const BitSequence value) { Set(*this, value); return *this; }

};

LIBSWIFFT_INLINE bool operator==(const SwifftInput &si1, const SwifftInput &si2) {
    return memcmp(si1.data, si2.data, sizeof(SwifftInput)) == 0;
}

LIBSWIFFT_INLINE bool operator!=(const SwifftInput &si1, const SwifftInput &si2) {
    return !(si1 == si2);
}

LIBSWIFFT_INLINE bool operator==(const SwifftOutput &so1, const SwifftOutput &so2) {
    return memcmp(so1.data, so2.data, sizeof(SwifftOutput)) == 0;
}

LIBSWIFFT_INLINE bool operator!=(const SwifftOutput &so1, const SwifftOutput &so2) {
    return !(so1 == so2);
}

LIBSWIFFT_INLINE bool operator==(const SwifftCompact &sc1, const SwifftCompact &sc2) {
    return memcmp(sc1.data, sc2.data, sizeof(SwifftCompact)) == 0;
}

LIBSWIFFT_INLINE bool operator!=(const SwifftCompact &sc1, const SwifftCompact &sc2) {
    return !(sc1 == sc2);
}

LIBSWIFFT_INLINE SwifftInput& Copy(SwifftInput &dst, const SwifftInput &src) {
    memcpy(dst.data, src.data, sizeof(dst.data));
    return dst;
}

LIBSWIFFT_INLINE SwifftOutput& Copy(SwifftOutput &dst, const SwifftOutput &src) {
    memcpy(dst.data, src.data, sizeof(dst.data));
    return dst;
}

LIBSWIFFT_INLINE SwifftCompact& Copy(SwifftCompact &dst, const SwifftCompact &src) {
    memcpy(dst.data, src.data, sizeof(dst.data));
    return dst;
}

LIBSWIFFT_INLINE SwifftInput & Set(SwifftInput &lhs, BitSequence value) {
    memset(lhs.data, value, sizeof(lhs.data));
    return lhs;
}

LIBSWIFFT_INLINE SwifftOutput & Set(SwifftOutput &lhs, const int16_t value) {
    SWIFFT_ConstSet(lhs.data, value);
    return lhs;
}

LIBSWIFFT_INLINE SwifftCompact & Set(SwifftCompact &lhs, BitSequence value) {
    memset(lhs.data, value, sizeof(lhs.data));
    return lhs;
}

LIBSWIFFT_INLINE SwifftOutput & operator+=(SwifftOutput &lhs, const int16_t value) {
    SWIFFT_ConstAdd(lhs.data, value);
    return lhs;
}

LIBSWIFFT_INLINE SwifftOutput & operator-=(SwifftOutput &lhs, const int16_t value) {
    SWIFFT_ConstSub(lhs.data, value);
    return lhs;
}

LIBSWIFFT_INLINE SwifftOutput & operator*=(SwifftOutput &lhs, const int16_t value) {
    SWIFFT_ConstMul(lhs.data, value);
    return lhs;
}

LIBSWIFFT_INLINE SwifftOutput & operator+=(SwifftOutput &lhs, const SwifftOutput &rhs) {
    SWIFFT_Add(lhs.data, rhs.data);
    return lhs;
}

LIBSWIFFT_INLINE SwifftOutput & operator-=(SwifftOutput &lhs, const SwifftOutput &rhs) {
    SWIFFT_Sub(lhs.data, rhs.data);
    return lhs;
}

LIBSWIFFT_INLINE SwifftOutput & operator*=(SwifftOutput &lhs, const SwifftOutput &rhs) {
    SWIFFT_Mul(lhs.data, rhs.data);
    return lhs;
}

} // end namespace LibSwifft

#endif // __LIBSWIFFT_SWIFFT_HPP__