tree_v4_manual.go

// Code generated by automation. DO NOT EDIT

package generics_tree

import (
	"fmt"

	"github.com/kentik/patricia"
)

// this is IPv4 tree code that's not very copy/paste friendly for when we transfer IPv4 code to IPv6

// create a new node in the tree, return its index
func (t *TreeV4[T]) newNode(address patricia.IPv4Address, prefixLength uint) uint {
	availCount := len(t.availableIndexes)
	if availCount > 0 {
		index := t.availableIndexes[availCount-1]
		t.availableIndexes = t.availableIndexes[:availCount-1]
		t.nodes[index] = treeNodeV4[T]{prefix: address.Address, prefixLength: prefixLength}
		return index
	}

	t.nodes = append(t.nodes, treeNodeV4[T]{prefix: address.Address, prefixLength: prefixLength})
	return uint(len(t.nodes) - 1)
}

// Address returns the current IP address for the iterator.
func (iter *TreeIteratorV4[T]) Address() patricia.IPv4Address {
	var prefix uint32
	var prefixLength uint
	for _, i := range iter.nodeHistory {
		prefix, prefixLength = patricia.MergePrefixes32(prefix, prefixLength,
			iter.t.nodes[i].prefix, iter.t.nodes[i].prefixLength)
	}
	prefix, prefixLength = patricia.MergePrefixes32(prefix, prefixLength,
		iter.t.nodes[iter.nodeIndex].prefix, iter.t.nodes[iter.nodeIndex].prefixLength)
	return patricia.NewIPv4Address(prefix, prefixLength)
}

//nolint
func (t *TreeV4[T]) print() {
	buf := make([]T, 0)
	for i := range t.nodes {
		buf = buf[:0]
		fmt.Printf("%d: \tleft: %d, right: %d, prefix: %032b (%d), tags: (%d): %v\n", i, int(t.nodes[i].Left), int(t.nodes[i].Right), int(t.nodes[i].prefix), int(t.nodes[i].prefixLength), t.nodes[i].TagCount, t.tagsForNode(buf, uint(i), nil))
	}
}