pkg/vector/alg/common.go - develop - githubmirror/photoprism

pkg/vector/alg/common.go

114 строк2 KB

Michael Mayer

Faces: Remove unused OPTICS clusterer from pkg/vector/alg #5628

30 май 2026, 10:15

30 май 2026, 10:15418a018

package alg

import (
	"container/heap"
	"crypto/rand"
	"math"
	"math/big"
	"sync"
)

// struct denoting start and end indices of database portion to be scanned for nearest neighbors by workers in DBSCAN
type rangeJob struct {
	a, b int
}

// priority queue
type pItem struct {
	v int
	p float64
	i int
}

type priorityQueue []*pItem

func newPriorityQueue(size int) priorityQueue {
	q := make(priorityQueue, 0, size)
	heap.Init(&q)

	return q
}

func (pq priorityQueue) Len() int { return len(pq) }

func (pq priorityQueue) Less(i, j int) bool {
	return pq[i].p > pq[j].p
}

func (pq priorityQueue) Swap(i, j int) {
	pq[i], pq[j] = pq[j], pq[i]
	pq[i].i = i
	pq[j].i = j
}

func (pq *priorityQueue) Push(x any) {
	n := len(*pq)
	item := x.(*pItem)
	item.i = n
	*pq = append(*pq, item)
	heap.Fix(pq, item.i)
}

func (pq *priorityQueue) Pop() any {
	old := *pq
	n := len(old)
	item := old[n-1]
	item.i = -1
	*pq = old[0 : n-1]
	return item
}

func (pq *priorityQueue) NotEmpty() bool {
	return len(*pq) > 0
}

func (pq *priorityQueue) Update(item *pItem, value int, priority float64) {
	item.v = value
	item.p = priority
	heap.Fix(pq, item.i)
}

func bounds(data [][]float64) []*[2]float64 {
	var (
		wg sync.WaitGroup

		l = len(data[0])
		r = make([]*[2]float64, l)
	)

	for i := range l {
		r[i] = &[2]float64{
			data[0][i],
			data[0][i],
		}
	}

	wg.Add(l)

	for i := range l {
		go func(n int) {
			defer wg.Done()

			for j := range data {
				if data[j][n] < r[n][0] {
					r[n][0] = data[j][n]
				} else if data[j][n] > r[n][1] {
					r[n][1] = data[j][n]
				}
			}
		}(i)
	}

	wg.Wait()

	return r
}

func uniform(data *[2]float64) float64 {
	n, err := rand.Int(rand.Reader, big.NewInt(math.MaxInt64))
	if err != nil {
		return data[0]
	}
	r := float64(n.Int64()) / float64(math.MaxInt64)
	return r*(data[1]-data[0]) + data[0]
}