summaryrefslogtreecommitdiff
path: root/src/toBeSorted/unk_flag_stuff.cpp
blob: f5a3f1af1bffaef92aeb8b04015899ef172b3ba3 (plain)
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
#pragma once

#include "toBeSorted/unk_flag_stuff.h"

u16 UnkFlagStuff::calculateMask(s32 shift) {
    return (2 << (shift & 0xF)) - 1;
}

void UnkFlagStuff::doNothing(u32 unused) {}

UnkFlagStuff::UnkFlagStuff(u16 count, UnkFlagDefinition* definitions) {
    this->mDefinitionsCount = count;
    this->mpDefinitions = definitions;
}

void UnkFlagStuff::prepareIndexShiftMask(u16 counterIdx, u32 unused, u16* indexPtr, u8* shiftPtr, u16* maskPtr) {
    doNothing(counterIdx);
    *indexPtr = mpDefinitions[counterIdx].mIndex;
    *shiftPtr = mpDefinitions[counterIdx].mShiftMask >> 4;
    *maskPtr = calculateMask(mpDefinitions[counterIdx].mShiftMask);
}

u16 UnkFlagStuff::maskForIdx(u16 index) {
    doNothing(index);

    return calculateMask(mpDefinitions[index].mShiftMask);
}

u32 UnkFlagStuff::getCounterOrFlag(u16 counterIdx, u16* dataPtr, u32 param4) {
    u16 index = 0;
    u8 shift = 0;
    u16 mask = 0;
    prepareIndexShiftMask(counterIdx, param4, &index, &shift, &mask);
    return mask & (dataPtr[index] >> shift);
}

void UnkFlagStuff::setCounterOrFlag(u16 counterIdx, u16* dataPtr, u32 flagCount, u32 value) {
    u16 index = 0;
    u8 shift = 0;
    u16 mask = 0;
    prepareIndexShiftMask(counterIdx, flagCount, &index, &shift, &mask);
    dataPtr[index] &= ~(mask << shift);
    dataPtr[index] = (u16)(value << shift) | dataPtr[index];
}

inline bool isLessThan(int a, int b) {
    return a <= b;
}

u32 UnkFlagStuff::checkFlagValid(u16 counterIdx, u32 flagCount) {
    if (mpDefinitions == nullptr) {
        return 0;
    }
    if (mDefinitionsCount <= counterIdx || mpDefinitions[counterIdx].mIndex == 0xFF) {
        return 0;
    } else {
        if (isLessThan(flagCount, mpDefinitions[counterIdx].mIndex)) {
            return 0;
        } else {
            return 1;
        }
    }
}