ericw-tools/include/vis/leafbits.hh

85 lines
2.3 KiB
C++

/* Copyright (C) 2012-2013 Kevin Shanahan
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
See file, 'COPYING', for details.
*/
#ifndef VIS_LEAFBITS_H
#define VIS_LEAFBITS_H
#include <stdlib.h>
#include <string.h>
#include <common/cmdlib.hh>
/* Use some GCC builtins */
#if !defined(ffsl) && defined(__GNUC__)
#define ffsl __builtin_ffsl
#elif defined(WIN32)
#include <intrin.h>
inline int ffsl(long int val)
{
unsigned long indexout;
unsigned char res = _BitScanForward(&indexout, val);
if (!res) return 0;
else return indexout + 1;
}
#endif
#ifndef offsetof
#define offsetof(type, member) __builtin_offsetof(type, member)
#endif
typedef unsigned long leafblock_t;
typedef struct {
int numleafs;
leafblock_t bits[]; /* Variable Sized */
} leafbits_t;
#define QBYTESHIFT(x) ((x) == 8 ? 6 : ((x) == 4 ? 5 : 0 ))
#define LEAFSHIFT QBYTESHIFT(sizeof(leafblock_t))
static_assert(LEAFSHIFT != 0, "unsupported sizeof(unsigned long)");
#define LEAFMASK ((sizeof(leafblock_t) << 3) - 1UL)
static inline int
TestLeafBit(const leafbits_t *bits, int leafnum)
{
return !!(bits->bits[leafnum >> LEAFSHIFT] & (1UL << (leafnum & LEAFMASK)));
}
static inline void
SetLeafBit(leafbits_t *bits, int leafnum)
{
bits->bits[leafnum >> LEAFSHIFT] |= 1UL << (leafnum & LEAFMASK);
}
static inline void
ClearLeafBit(leafbits_t *bits, int leafnum)
{
bits->bits[leafnum >> LEAFSHIFT] &= ~(1UL << (leafnum & LEAFMASK));
}
static inline size_t
LeafbitsSize(int numleafs)
{
int numblocks = (numleafs + LEAFMASK) >> LEAFSHIFT;
return sizeof(leafbits_t) + (sizeof(leafblock_t) * numblocks);
}
#endif /* VIS_LEAFBITS_H */