/*
 * Copyright (c) 2023 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef MAPLEBE_INCLUDE_CG_DATAINFO_H
#define MAPLEBE_INCLUDE_CG_DATAINFO_H
#include "maple_string.h"
#include "common_utils.h"
#include "mempool.h"
#include "mempool_allocator.h"

namespace maplebe {
class DataInfo {
public:
    DataInfo(uint32 bitNum, MapleAllocator &alloc) : info(alloc.Adapter())
    {
        info.resize(bitNum / kWordSize + 1, 0);
    }
    DataInfo(const DataInfo &other, MapleAllocator &alloc) : info(other.info, alloc.Adapter()) {}
    DataInfo &Clone(MapleAllocator &alloc)
    {
        auto *dataInfo = alloc.New<DataInfo>(*this, alloc);
        return *dataInfo;
    }

    ~DataInfo() = default;

    void SetBit(int64 bitNO)
    {
        DEBUG_ASSERT(static_cast<size_t>(bitNO) < info.size() * kWordSize, "Out of Range");
        info[static_cast<size_t>(bitNO / kWordSize)] |= (1ULL << static_cast<uint64>((bitNO % kWordSize)));
    }

    void ResetBit(uint32 bitNO)
    {
        info[bitNO / kWordSize] &= (~(1ULL << (bitNO % kWordSize)));
    }

    bool TestBit(uint32 bitNO) const
    {
        return (info[bitNO / kWordSize] & (1ULL << (bitNO % kWordSize))) != 0ULL;
    }

    const uint64 &GetElem(uint32 index) const
    {
        DEBUG_ASSERT(index < info.size(), "out of range");
        return info[index];
    }

    void SetElem(uint32 index, uint64 val)
    {
        DEBUG_ASSERT(index < info.size(), "out of range");
        info[index] = val;
    }

    bool NoneBit() const
    {
        for (auto &data : info) {
            if (data != 0ULL) {
                return false;
            }
        }
        return true;
    }

    size_t Size() const
    {
        return info.size() * kWordSize;
    }

    const MapleVector<uint64> &GetInfo() const
    {
        return info;
    }

    bool IsEqual(const DataInfo &secondInfo) const
    {
        auto infoSize = static_cast<const uint32>(info.size());
        DEBUG_ASSERT(infoSize == secondInfo.GetInfo().size(), "two dataInfo's size different");
        for (uint32 i = 0; i != infoSize; ++i) {
            if (info[i] != secondInfo.GetElem(i)) {
                return false;
            }
        }
        return true;
    }

    bool IsEqual(const MapleVector<uint64> &LiveInfoBak) const
    {
        size_t infoSize = info.size();
        DEBUG_ASSERT(infoSize == LiveInfoBak.size(), "two dataInfo's size different");
        for (size_t i = 0; i != infoSize; ++i) {
            if (info[i] != LiveInfoBak[i]) {
                return false;
            }
        }
        return true;
    }

    void AndBits(const DataInfo &secondInfo)
    {
        auto infoSize = static_cast<const uint32>(info.size());
        DEBUG_ASSERT(infoSize == secondInfo.GetInfo().size(), "two dataInfo's size different");
        for (uint32 i = 0; i != infoSize; ++i) {
            info[i] &= secondInfo.GetElem(i);
        }
    }

    void OrBits(const DataInfo &secondInfo)
    {
        auto infoSize = static_cast<const uint32>(info.size());
        DEBUG_ASSERT(infoSize == secondInfo.GetInfo().size(), "two dataInfo's size different");
        for (uint32 i = 0; i != infoSize; i++) {
            info[i] |= secondInfo.GetElem(i);
        }
    }

    void OrDesignateBits(const DataInfo &secondInfo, uint32 infoIndex)
    {
        DEBUG_ASSERT(infoIndex < secondInfo.GetInfo().size(), "out of secondInfo's range");
        DEBUG_ASSERT(infoIndex < info.size(), "out of secondInfo's range");
        info[infoIndex] |= secondInfo.GetElem(infoIndex);
    }

    void EorBits(const DataInfo &secondInfo)
    {
        auto infoSize = static_cast<const uint32>(info.size());
        DEBUG_ASSERT(infoSize == secondInfo.GetInfo().size(), "two dataInfo's size different");
        for (uint32 i = 0; i != infoSize; i++) {
            info[i] ^= secondInfo.GetElem(i);
        }
    }

    /* if bit in secondElem is 1, bit in current DataInfo is set 0 */
    void Difference(const DataInfo &secondInfo)
    {
        auto infoSize = static_cast<const uint32>(info.size());
        DEBUG_ASSERT(infoSize == secondInfo.GetInfo().size(), "two dataInfo's size different");
        for (uint32 i = 0; i != infoSize; i++) {
            info[i] &= (~(secondInfo.GetElem(i)));
        }
    }

    void ResetAllBit()
    {
        for (auto &data : info) {
            data = 0ULL;
        }
    }

    void EnlargeCapacityToAdaptSize(uint32 bitNO)
    {
        /* add one more size for each enlarge action */
        info.resize(bitNO / kWordSize + 1, 0);
    }

    void GetNonZeroElemsIndex(std::set<uint32> &index)
    {
        auto infoSize = static_cast<const int32>(info.size());
        for (int32 i = 0; i < infoSize; i++) {
            if (info[i] != 0ULL) {
                (void)index.insert(i);
            }
        }
    }

    template <typename T>
    void GetBitsOfInfo(T &wordRes) const
    {
        wordRes.clear();
        for (size_t i = 0; i != info.size(); ++i) {
            uint32 result = 0;
            uint64 word = info[i];
            uint32 offset = 0;
            uint32 baseWord = 0;
            bool firstTime = true;
            while (word) {
                int32 index = __builtin_ffsll(static_cast<int64>(word));
                if (index == 0) {
                    continue;
                }
                if (index == k64BitSize) {
                    /* when the highest bit is 1, the shift operation will cause error, need special treatment. */
                    result = i * kWordSize + (index - 1);
                    (void)wordRes.insert(result);
                    break;
                }
                if (firstTime) {
                    offset = static_cast<uint32>(index - 1);
                    baseWord = i * kWordSize;
                    firstTime = false;
                } else {
                    offset = static_cast<uint32>(index);
                    baseWord = 0;
                }
                result += baseWord + offset;
                (void)wordRes.insert(result);
                word = word >> static_cast<uint64>(index);
            }
        }
    }

    void ClearDataInfo()
    {
        info.clear();
    }

private:
    /* long type has 8 bytes, 64 bits */
    static constexpr int32 kWordSize = 64;
    MapleVector<uint64> info;
};
} /* namespace maplebe */
#endif /* MAPLEBE_INCLUDE_CG_INSN_H */