libz-sys 1.1.29

Low-level bindings to the system libz library (also known as zlib).
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

/* lsxintrin_ext.h
 * Copyright (C) 2025 Vladislav Shchapov <vladislav@shchapov.ru>
 * For conditions of distribution and use, see copyright notice in zlib.h
 */
#ifndef LSXINTRIN_EXT_H
#define LSXINTRIN_EXT_H

#include <lsxintrin.h>


static inline __m128i lsx_sad_bu(__m128i a, __m128i b) {
    __m128i tmp = __lsx_vabsd_bu(a, b);
    tmp = __lsx_vhaddw_hu_bu(tmp, tmp);
    tmp = __lsx_vhaddw_wu_hu(tmp, tmp);
    return __lsx_vhaddw_du_wu(tmp, tmp);
}

static inline int lsx_movemask_b(__m128i v) {
    return __lsx_vpickve2gr_w(__lsx_vmskltz_b(v), 0);
}

static inline __m128i lsx_shuffle_b(__m128i a, __m128i b) {
    /* most significant bit is set - negative 8-bit integer */
    __m128i msb_mask = __lsx_vslti_b(b, 0);

    /* shuffle, clear msb in indices vector b */
    __m128i dst = __lsx_vshuf_b(a, a, __lsx_vandi_b(b, 0xF));

    /* invert and apply mask - clear dst-element if b-msb is set */
    return __lsx_vand_v(dst, __lsx_vnor_v(msb_mask, msb_mask));
}

#endif // include guard LSXINTRIN_EXT_H