package world

import (
	"github.com/Tnze/go-mc/save"
)

const ChunkSectionCount = 16

type ChunkData struct {
	Pos      ChunkPos                        `json:"pos"`
	Sections [ChunkSectionCount]ChunkSection `json:"sections"`
}

func (cd *ChunkData) SectionFor(pos BlockPos) *ChunkSection {
	return &cd.Sections[pos.Y%ChunkSectionCount]
}

func (cd *ChunkData) IndexToBlockPos(index int) BlockPos {
	posX := index % 16
	posZ := ((index - posX) % 256) / 16
	posY := ((index - posZ) % 4096) / 256
	return BlockPos{
		X: posX + (cd.Pos.X * 16),
		Y: uint(posY),
		Z: posZ + (cd.Pos.Z * 16),
	}
}

func extractPaletteIndexes(compressed int64) [16]byte {
	var outputs [16]byte
	var outputIndex int

	for index := 0; index < 64; index += 4 {
		shifted := compressed >> index
		// Mask off the lowest four bits
		shifted &= 0xf
		outputs[outputIndex] = byte(shifted)
		outputIndex += 1
	}

	return outputs
}

func (cd *ChunkData) FromMCAChunk(other save.Chunk) {
	// Load the chunk's position
	cd.Pos = ChunkPos{X: int(other.XPos), Z: int(other.ZPos)}

	// Load the data from the chunk
	for sectionIndex, section := range other.Sections {
		// TODO: Enable chunks to have more than 16 sections
		if sectionIndex >= ChunkSectionCount {
			break
		}

		var currentSection ChunkSection

		paletteIndexes := []int{}
		for _, compress := range section.BlockStates.Data {
			indexes := extractPaletteIndexes(int64(compress))

			converted := make([]int, 16)
			for i := 0; i < 16; i++ {
				converted[i] = int(indexes[i])
			}

			paletteIndexes = append(paletteIndexes, converted...)
		}

		for blockIndex, paletteIndex := range paletteIndexes {
			var state BlockID
			if section.BlockStates.Palette[paletteIndex].Name == "minecraft:air" {
				state = Empty
			} else {
				cd.Sections[sectionIndex].BlockCount += 1
				state = Generic
			}

			// TODO: Remove this workaround for larger bit sizes in palettes
			if blockIndex < 4096 {
				currentSection.BlockStates[blockIndex] = state
			}
		}

		cd.Sections[sectionIndex] = currentSection
	}
}