package util

import (
	"fmt"
	"strings"
)

const (
	bitSize = 8
)

var bitmask = []byte{1, 1 << 1, 1 << 2, 1 << 3, 1 << 4, 1 << 5, 1 << 6, 1 << 7}

// Bitmap 首字母小写 只能调用 工厂函数 创建
type Bitmap struct {
	bits     []byte
	bitCount uint64 // 已填入数字的数量
	capacity uint64 // 容量
}

// NewBitmap 创建工厂函数
func NewBitmap(maxnum uint64) *Bitmap {
	return &Bitmap{bits: make([]byte, (maxnum+7)/bitSize), bitCount: 0, capacity: maxnum}
}

// Set 填入数字
func (this *Bitmap) Set(num uint64) {
	byteIndex, bitPos := this.offset(num)
	// 1 左移 bitPos 位 进行 按位或 (置为 1)
	this.bits[byteIndex] |= bitmask[bitPos]
	this.bitCount++
}

// Reset 清除填入的数字
func (this *Bitmap) Reset(num uint64) {
	byteIndex, bitPos := this.offset(num)
	// 重置为空位 (重置为 0)
	this.bits[byteIndex] &= ^bitmask[bitPos]
	this.bitCount--
}

// Test 数字是否在位图中
func (this *Bitmap) Test(num uint64) bool {
	byteIndex := num / bitSize
	if byteIndex >= uint64(len(this.bits)) {
		return false
	}
	bitPos := num % bitSize
	// 右移 bitPos 位 和 1 进行 按位与
	return !(this.bits[byteIndex]&bitmask[bitPos] == 0)
}

func (this *Bitmap) offset(num uint64) (byteIndex uint64, bitPos byte) {
	byteIndex = num / bitSize // 字节索引
	if byteIndex >= uint64(len(this.bits)) {
		panic(fmt.Sprintf(" runtime error: index value %d out of range", byteIndex))
		return
	}
	bitPos = byte(num % bitSize) // bit位置
	return byteIndex, bitPos
}

// Size 位图的容量
func (this *Bitmap) Size() uint64 {
	return uint64(len(this.bits) * bitSize)
}

// IsEmpty 是否空位图
func (this *Bitmap) IsEmpty() bool {
	return this.bitCount == 0
}

// IsFully 是否已填满
func (this *Bitmap) IsFully() bool {
	return this.bitCount == this.capacity
}

// Count 已填入的数字个数
func (this *Bitmap) Count() uint64 {
	return this.bitCount
}

// GetData 获取填入的数字切片
func (this *Bitmap) GetData() []uint64 {
	var data []uint64
	count := this.Size()
	for index := uint64(0); index < count; index++ {
		if this.Test(index) {
			data = append(data, index)
		}
	}
	return data
}

func (this *Bitmap) String() string {
	var sb strings.Builder
	for index := len(this.bits) - 1; index >= 0; index-- {
		sb.WriteString(byteToBinaryString(this.bits[index]))
		sb.WriteString(" ")
	}
	return sb.String()
}

func byteToBinaryString(data byte) string {
	var sb strings.Builder
	for index := 0; index < bitSize; index++ {
		if (bitmask[7-index] & data) == 0 {
			sb.WriteString("0")
		} else {
			sb.WriteString("1")
		}
	}
	return sb.String()
}