13754430621

Committed 10 Mar 2025 12:52AM CUT coverage: 99.436% (+0.005%) from 99.431%

Build # 13754430621

Build Type

Pull #175

github

Committed by

web-flow

Commit Message

Merge 8bfc82421 into 5994e504b

Pull Request Pull Request #175: Add `Query.Count`

Run Details

95 of 96 new or added lines in 3 files covered. (98.96%)

9 existing lines in 1 file now uncovered.

6523 of 6560 relevant lines covered (99.44%)

26772.95 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

99.09

/ecs/query_gen.go

package ecs

// Code generated by go generate; DO NOT EDIT.

type cursor struct {
        archetype int
        table     int
        index     uintptr
        maxIndex  int64
}

type queryResult0 struct {
        Entity Entity
}

// Query0 is a query for 0 components.
// Use a [Filter0] to create one.
//
// Queries are one-time use iterators and must be re-created each time before iterating.
//
// See [Query2] for a usage example.
type Query0 struct {
        world      *World
        filter     *filter
        relations  []RelationID
        lock       uint8
        cursor     cursor
        tables     []tableID
        table      *table
        cache      *cacheEntry
        components []*componentStorage
}

func newQuery0(world *World, filter *filter, relations []RelationID,
        cacheID cacheID, components []*componentStorage) Query0 {
        var cache *cacheEntry
        if cacheID != maxCacheID {
                cache = world.storage.getRegisteredFilter(cacheID)
        }

        return Query0{
                world:      world,
                filter:     filter,
                relations:  relations,
                cache:      cache,
                lock:       world.lock(),
                components: components,
                cursor: cursor{
                        archetype: -1,
                        table:     -1,
                        index:     0,
                        maxIndex:  -1,
                },
        }
}

// Next advances the query's cursor to the next entity.
func (q *Query0) Next() bool {
        q.world.checkQueryNext(&q.cursor)
        if int64(q.cursor.index) < q.cursor.maxIndex {
                q.cursor.index++
                return true
        }
        return q.nextTableOrArchetype()
}

// Entity returns the current entity.
func (q *Query0) Entity() Entity {
        q.world.checkQueryGet(&q.cursor)
        return q.table.GetEntity(q.cursor.index)
}

// Count counts the entities matching this query.
//
// Has some overhead of iterating through archetypes.
// However, this is still much faster than manual counting via iteration.
//
// Does not iterate or close the query.
func (q *Query0) Count() int {
        if q.cache == nil {
                return countQuery(&q.world.storage, q.filter, q.relations)
        }
        return countQueryCache(&q.world.storage, q.cache, q.relations)
}

// Close closes the Query and unlocks the world.
//
// Automatically called when iteration completes.
// Needs to be called only if breaking out of the query iteration or not iterating at all.
func (q *Query0) Close() {
        q.cursor.archetype = -2
        q.cursor.table = -2
        q.tables = nil
        q.table = nil
        q.cache = nil
        q.world.unlock(q.lock)
}

func (q *Query0) nextTableOrArchetype() bool {
        if q.cache != nil {
                return q.nextTable(q.cache.tables)
        }
        if q.cursor.archetype >= 0 && q.nextTable(q.tables) {
                return true
        }
        return q.nextArchetype()
}

func (q *Query0) nextArchetype() bool {
        maxArchIndex := len(q.world.storage.archetypes) - 1
        for q.cursor.archetype < maxArchIndex {
                q.cursor.archetype++
                archetype := &q.world.storage.archetypes[q.cursor.archetype]
                if !q.filter.matches(&archetype.mask) {
                        continue
                }

                if !archetype.HasRelations() {
                        table := &q.world.storage.tables[archetype.tables[0]]
                        if table.Len() > 0 {
                                q.setTable(0, table)
                                return true
                        }
                        continue
                }

                q.tables = archetype.GetTables(q.relations)
                q.cursor.table = -1
                if q.nextTable(q.tables) {
                        return true
                }
        }
        q.Close()
        return false
}

func (q *Query0) nextTable(tables []tableID) bool {
        maxTableIndex := len(tables) - 1
        for q.cursor.table < maxTableIndex {
                q.cursor.table++
                table := &q.world.storage.tables[tables[q.cursor.table]]
                if table.Len() == 0 {
                        continue
                }
                if !table.Matches(q.relations) {
                        continue
                }
                q.setTable(q.cursor.table, table)
                return true
        }
        if q.cache != nil {
                q.Close()
        }
        return false
}

func (q *Query0) setTable(index int, table *table) {
        q.cursor.table = index
        q.table = table
        q.cursor.index = 0
        q.cursor.maxIndex = int64(q.table.Len() - 1)
}

type queryResult1[A any] struct {
        Entity Entity
        A      *A
}

// Query1 is a query for 1 components.
// Use a [Filter1] to create one.
//
// Queries are one-time use iterators and must be re-created each time before iterating.
//
// See [Query2] for a usage example.
type Query1[A any] struct {
        world      *World
        filter     *filter
        relations  []RelationID
        lock       uint8
        cursor     cursor
        tables     []tableID
        table      *table
        cache      *cacheEntry
        components []*componentStorage
        columnA    *column
}

func newQuery1[A any](world *World, filter *filter, ids []ID, relations []RelationID,
        cacheID cacheID, components []*componentStorage) Query1[A] {
        var cache *cacheEntry
        if cacheID != maxCacheID {
                cache = world.storage.getRegisteredFilter(cacheID)
        }

        return Query1[A]{
                world:      world,
                filter:     filter,
                relations:  relations,
                cache:      cache,
                lock:       world.lock(),
                components: components,
                cursor: cursor{
                        archetype: -1,
                        table:     -1,
                        index:     0,
                        maxIndex:  -1,
                },
        }
}

// Next advances the query's cursor to the next entity.
func (q *Query1[A]) Next() bool {
        q.world.checkQueryNext(&q.cursor)
        if int64(q.cursor.index) < q.cursor.maxIndex {
                q.cursor.index++
                return true
        }
        return q.nextTableOrArchetype()
}

// Entity returns the current entity.
func (q *Query1[A]) Entity() Entity {
        q.world.checkQueryGet(&q.cursor)
        return q.table.GetEntity(q.cursor.index)
}

// Get returns the queried components of the current entity.
//
// ⚠️ Do not store the obtained pointers outside of the current context (i.e. the query loop)!
func (q *Query1[A]) Get() *A {
        q.world.checkQueryGet(&q.cursor)
        return (*A)(q.columnA.Get(q.cursor.index))
}

// GetRelation returns the entity relation target of the component at the given index.
func (q *Query1[A]) GetRelation(index int) Entity {
        return q.components[index].columns[q.table.id].target
}

// Count counts the entities matching this query.
//
// Has some overhead of iterating through archetypes.
// However, this is still much faster than manual counting via iteration.
//
// Does not iterate or close the query.
func (q *Query1[A]) Count() int {
        if q.cache == nil {
                return countQuery(&q.world.storage, q.filter, q.relations)
        }
        return countQueryCache(&q.world.storage, q.cache, q.relations)
}

// Close closes the Query and unlocks the world.
//
// Automatically called when iteration completes.
// Needs to be called only if breaking out of the query iteration or not iterating at all.
func (q *Query1[A]) Close() {
        q.cursor.archetype = -2
        q.cursor.table = -2
        q.tables = nil
        q.table = nil
        q.cache = nil
        q.columnA = nil
        q.world.unlock(q.lock)
}

func (q *Query1[A]) nextTableOrArchetype() bool {
        if q.cache != nil {
                return q.nextTable(q.cache.tables)
        }
        if q.cursor.archetype >= 0 && q.nextTable(q.tables) {
                return true
        }
        return q.nextArchetype()
}

func (q *Query1[A]) nextArchetype() bool {
        maxArchIndex := len(q.world.storage.archetypes) - 1
        for q.cursor.archetype < maxArchIndex {
                q.cursor.archetype++
                archetype := &q.world.storage.archetypes[q.cursor.archetype]
                if !q.filter.matches(&archetype.mask) {
                        continue
                }

                if !archetype.HasRelations() {
                        table := &q.world.storage.tables[archetype.tables[0]]
                        if table.Len() > 0 {
                                q.setTable(0, table)
                                return true
                        }
                        continue
                }

                q.tables = archetype.GetTables(q.relations)
                q.cursor.table = -1
                if q.nextTable(q.tables) {
                        return true
                }
        }
        q.Close()
        return false
}

func (q *Query1[A]) nextTable(tables []tableID) bool {
        maxTableIndex := len(tables) - 1
        for q.cursor.table < maxTableIndex {
                q.cursor.table++
                table := &q.world.storage.tables[tables[q.cursor.table]]
                if table.Len() == 0 {
                        continue
                }
                if !table.Matches(q.relations) {
                        continue
                }
                q.setTable(q.cursor.table, table)
                return true
        }
        if q.cache != nil {
                q.Close()
        }
        return false
}

func (q *Query1[A]) setTable(index int, table *table) {
        q.cursor.table = index
        q.table = table
        q.columnA = q.components[0].columns[q.table.id]
        q.cursor.index = 0
        q.cursor.maxIndex = int64(q.table.Len() - 1)
}

type queryResult2[A any, B any] struct {
        Entity Entity
        A      *A
        B      *B
}

// Query2 is a query for 2 components.
// Use a [Filter2] to create one.
//
// Queries are one-time use iterators and must be re-created each time before iterating.
type Query2[A any, B any] struct {
        world      *World
        filter     *filter
        relations  []RelationID
        lock       uint8
        cursor     cursor
        tables     []tableID
        table      *table
        cache      *cacheEntry
        components []*componentStorage
        columnA    *column
        columnB    *column
}

func newQuery2[A any, B any](world *World, filter *filter, ids []ID, relations []RelationID,
        cacheID cacheID, components []*componentStorage) Query2[A, B] {
        var cache *cacheEntry
        if cacheID != maxCacheID {
                cache = world.storage.getRegisteredFilter(cacheID)
        }

        return Query2[A, B]{
                world:      world,
                filter:     filter,
                relations:  relations,
                cache:      cache,
                lock:       world.lock(),
                components: components,
                cursor: cursor{
                        archetype: -1,
                        table:     -1,
                        index:     0,
                        maxIndex:  -1,
                },
        }
}

// Next advances the query's cursor to the next entity.
func (q *Query2[A, B]) Next() bool {
        q.world.checkQueryNext(&q.cursor)
        if int64(q.cursor.index) < q.cursor.maxIndex {
                q.cursor.index++
                return true
        }
        return q.nextTableOrArchetype()
}

// Entity returns the current entity.
func (q *Query2[A, B]) Entity() Entity {
        q.world.checkQueryGet(&q.cursor)
        return q.table.GetEntity(q.cursor.index)
}

// Get returns the queried components of the current entity.
//
// ⚠️ Do not store the obtained pointers outside of the current context (i.e. the query loop)!
func (q *Query2[A, B]) Get() (*A, *B) {
        q.world.checkQueryGet(&q.cursor)
        return (*A)(q.columnA.Get(q.cursor.index)),
                (*B)(q.columnB.Get(q.cursor.index))
}

// GetRelation returns the entity relation target of the component at the given index.
func (q *Query2[A, B]) GetRelation(index int) Entity {
        return q.components[index].columns[q.table.id].target
}

// Count counts the entities matching this query.
//
// Has some overhead of iterating through archetypes.
// However, this is still much faster than manual counting via iteration.
//
// Does not iterate or close the query.
func (q *Query2[A, B]) Count() int {
        if q.cache == nil {
                return countQuery(&q.world.storage, q.filter, q.relations)
        }
        return countQueryCache(&q.world.storage, q.cache, q.relations)
}

// Close closes the Query and unlocks the world.
//
// Automatically called when iteration completes.
// Needs to be called only if breaking out of the query iteration or not iterating at all.
func (q *Query2[A, B]) Close() {
        q.cursor.archetype = -2
        q.cursor.table = -2
        q.tables = nil
        q.table = nil
        q.cache = nil
        q.columnA = nil
        q.columnB = nil
        q.world.unlock(q.lock)
}

func (q *Query2[A, B]) nextTableOrArchetype() bool {
        if q.cache != nil {
                return q.nextTable(q.cache.tables)
        }
        if q.cursor.archetype >= 0 && q.nextTable(q.tables) {
                return true
        }
        return q.nextArchetype()
}

func (q *Query2[A, B]) nextArchetype() bool {
        maxArchIndex := len(q.world.storage.archetypes) - 1
        for q.cursor.archetype < maxArchIndex {
                q.cursor.archetype++
                archetype := &q.world.storage.archetypes[q.cursor.archetype]
                if !q.filter.matches(&archetype.mask) {
                        continue
                }

                if !archetype.HasRelations() {
                        table := &q.world.storage.tables[archetype.tables[0]]
                        if table.Len() > 0 {
                                q.setTable(0, table)
                                return true
                        }
                        continue
                }

                q.tables = archetype.GetTables(q.relations)
                q.cursor.table = -1
                if q.nextTable(q.tables) {
                        return true
                }
        }
        q.Close()
        return false
}

func (q *Query2[A, B]) nextTable(tables []tableID) bool {
        maxTableIndex := len(tables) - 1
        for q.cursor.table < maxTableIndex {
                q.cursor.table++
                table := &q.world.storage.tables[tables[q.cursor.table]]
                if table.Len() == 0 {
                        continue
                }
                if !table.Matches(q.relations) {
                        continue
                }
                q.setTable(q.cursor.table, table)
                return true
        }
        if q.cache != nil {
                q.Close()
        }
        return false
}

func (q *Query2[A, B]) setTable(index int, table *table) {
        q.cursor.table = index
        q.table = table
        q.columnA = q.components[0].columns[q.table.id]
        q.columnB = q.components[1].columns[q.table.id]
        q.cursor.index = 0
        q.cursor.maxIndex = int64(q.table.Len() - 1)
}

type queryResult3[A any, B any, C any] struct {
        Entity Entity
        A      *A
        B      *B
        C      *C
}

// Query3 is a query for 3 components.
// Use a [Filter3] to create one.
//
// Queries are one-time use iterators and must be re-created each time before iterating.
//
// See [Query2] for a usage example.
type Query3[A any, B any, C any] struct {
        world      *World
        filter     *filter
        relations  []RelationID
        lock       uint8
        cursor     cursor
        tables     []tableID
        table      *table
        cache      *cacheEntry
        components []*componentStorage
        columnA    *column
        columnB    *column
        columnC    *column
}

func newQuery3[A any, B any, C any](world *World, filter *filter, ids []ID, relations []RelationID,
        cacheID cacheID, components []*componentStorage) Query3[A, B, C] {
        var cache *cacheEntry
        if cacheID != maxCacheID {
                cache = world.storage.getRegisteredFilter(cacheID)
        }

        return Query3[A, B, C]{
                world:      world,
                filter:     filter,
                relations:  relations,
                cache:      cache,
                lock:       world.lock(),
                components: components,
                cursor: cursor{
                        archetype: -1,
                        table:     -1,
                        index:     0,
                        maxIndex:  -1,
                },
        }
}

// Next advances the query's cursor to the next entity.
func (q *Query3[A, B, C]) Next() bool {
        q.world.checkQueryNext(&q.cursor)
        if int64(q.cursor.index) < q.cursor.maxIndex {
                q.cursor.index++
                return true
        }
        return q.nextTableOrArchetype()
}

// Entity returns the current entity.
func (q *Query3[A, B, C]) Entity() Entity {
        q.world.checkQueryGet(&q.cursor)
        return q.table.GetEntity(q.cursor.index)
}

// Get returns the queried components of the current entity.
//
// ⚠️ Do not store the obtained pointers outside of the current context (i.e. the query loop)!
func (q *Query3[A, B, C]) Get() (*A, *B, *C) {
        q.world.checkQueryGet(&q.cursor)
        return (*A)(q.columnA.Get(q.cursor.index)),
                (*B)(q.columnB.Get(q.cursor.index)),
                (*C)(q.columnC.Get(q.cursor.index))
}

// GetRelation returns the entity relation target of the component at the given index.
func (q *Query3[A, B, C]) GetRelation(index int) Entity {
        return q.components[index].columns[q.table.id].target
}

// Count counts the entities matching this query.
//
// Has some overhead of iterating through archetypes.
// However, this is still much faster than manual counting via iteration.
//
// Does not iterate or close the query.
func (q *Query3[A, B, C]) Count() int {
        if q.cache == nil {
                return countQuery(&q.world.storage, q.filter, q.relations)
        }
        return countQueryCache(&q.world.storage, q.cache, q.relations)
}

// Close closes the Query and unlocks the world.
//
// Automatically called when iteration completes.
// Needs to be called only if breaking out of the query iteration or not iterating at all.
func (q *Query3[A, B, C]) Close() {
        q.cursor.archetype = -2
        q.cursor.table = -2
        q.tables = nil
        q.table = nil
        q.cache = nil
        q.columnA = nil
        q.columnB = nil
        q.columnC = nil
        q.world.unlock(q.lock)
}

func (q *Query3[A, B, C]) nextTableOrArchetype() bool {
        if q.cache != nil {
                return q.nextTable(q.cache.tables)
        }
        if q.cursor.archetype >= 0 && q.nextTable(q.tables) {
                return true
        }
        return q.nextArchetype()
}

func (q *Query3[A, B, C]) nextArchetype() bool {
        maxArchIndex := len(q.world.storage.archetypes) - 1
        for q.cursor.archetype < maxArchIndex {
                q.cursor.archetype++
                archetype := &q.world.storage.archetypes[q.cursor.archetype]
                if !q.filter.matches(&archetype.mask) {
                        continue
                }

                if !archetype.HasRelations() {
                        table := &q.world.storage.tables[archetype.tables[0]]
                        if table.Len() > 0 {
                                q.setTable(0, table)
                                return true
                        }
                        continue
                }

                q.tables = archetype.GetTables(q.relations)
                q.cursor.table = -1
                if q.nextTable(q.tables) {
                        return true
                }
        }
        q.Close()
        return false
}

func (q *Query3[A, B, C]) nextTable(tables []tableID) bool {
        maxTableIndex := len(tables) - 1
        for q.cursor.table < maxTableIndex {
                q.cursor.table++
                table := &q.world.storage.tables[tables[q.cursor.table]]
                if table.Len() == 0 {
                        continue
                }
                if !table.Matches(q.relations) {
                        continue
                }
                q.setTable(q.cursor.table, table)
                return true
        }
        if q.cache != nil {
                q.Close()
        }
        return false
}

func (q *Query3[A, B, C]) setTable(index int, table *table) {
        q.cursor.table = index
        q.table = table
        q.columnA = q.components[0].columns[q.table.id]
        q.columnB = q.components[1].columns[q.table.id]
        q.columnC = q.components[2].columns[q.table.id]
        q.cursor.index = 0
        q.cursor.maxIndex = int64(q.table.Len() - 1)
}

type queryResult4[A any, B any, C any, D any] struct {
        Entity Entity
        A      *A
        B      *B
        C      *C
        D      *D
}

// Query4 is a query for 4 components.
// Use a [Filter4] to create one.
//
// Queries are one-time use iterators and must be re-created each time before iterating.
//
// See [Query2] for a usage example.
type Query4[A any, B any, C any, D any] struct {
        world      *World
        filter     *filter
        relations  []RelationID
        lock       uint8
        cursor     cursor
        tables     []tableID
        table      *table
        cache      *cacheEntry
        components []*componentStorage
        columnA    *column
        columnB    *column
        columnC    *column
        columnD    *column
}

func newQuery4[A any, B any, C any, D any](world *World, filter *filter, ids []ID, relations []RelationID,
        cacheID cacheID, components []*componentStorage) Query4[A, B, C, D] {
        var cache *cacheEntry
        if cacheID != maxCacheID {
                cache = world.storage.getRegisteredFilter(cacheID)
        }

        return Query4[A, B, C, D]{
                world:      world,
                filter:     filter,
                relations:  relations,
                cache:      cache,
                lock:       world.lock(),
                components: components,
                cursor: cursor{
                        archetype: -1,
                        table:     -1,
                        index:     0,
                        maxIndex:  -1,
                },
        }
}

// Next advances the query's cursor to the next entity.
func (q *Query4[A, B, C, D]) Next() bool {
        q.world.checkQueryNext(&q.cursor)
        if int64(q.cursor.index) < q.cursor.maxIndex {
                q.cursor.index++
                return true
        }
        return q.nextTableOrArchetype()
}

// Entity returns the current entity.
func (q *Query4[A, B, C, D]) Entity() Entity {
        q.world.checkQueryGet(&q.cursor)
        return q.table.GetEntity(q.cursor.index)
}

// Get returns the queried components of the current entity.
//
// ⚠️ Do not store the obtained pointers outside of the current context (i.e. the query loop)!
func (q *Query4[A, B, C, D]) Get() (*A, *B, *C, *D) {
        q.world.checkQueryGet(&q.cursor)
        return (*A)(q.columnA.Get(q.cursor.index)),
                (*B)(q.columnB.Get(q.cursor.index)),
                (*C)(q.columnC.Get(q.cursor.index)),
                (*D)(q.columnD.Get(q.cursor.index))
}

// GetRelation returns the entity relation target of the component at the given index.
func (q *Query4[A, B, C, D]) GetRelation(index int) Entity {
        return q.components[index].columns[q.table.id].target
}

// Count counts the entities matching this query.
//
// Has some overhead of iterating through archetypes.
// However, this is still much faster than manual counting via iteration.
//
// Does not iterate or close the query.
func (q *Query4[A, B, C, D]) Count() int {
        if q.cache == nil {
                return countQuery(&q.world.storage, q.filter, q.relations)
        }
        return countQueryCache(&q.world.storage, q.cache, q.relations)
}

// Close closes the Query and unlocks the world.
//
// Automatically called when iteration completes.
// Needs to be called only if breaking out of the query iteration or not iterating at all.
func (q *Query4[A, B, C, D]) Close() {
        q.cursor.archetype = -2
        q.cursor.table = -2
        q.tables = nil
        q.table = nil
        q.cache = nil
        q.columnA = nil
        q.columnB = nil
        q.columnC = nil
        q.columnD = nil
        q.world.unlock(q.lock)
}

func (q *Query4[A, B, C, D]) nextTableOrArchetype() bool {
        if q.cache != nil {
                return q.nextTable(q.cache.tables)
        }
        if q.cursor.archetype >= 0 && q.nextTable(q.tables) {
                return true
        }
        return q.nextArchetype()
}

func (q *Query4[A, B, C, D]) nextArchetype() bool {
        maxArchIndex := len(q.world.storage.archetypes) - 1
        for q.cursor.archetype < maxArchIndex {
                q.cursor.archetype++
                archetype := &q.world.storage.archetypes[q.cursor.archetype]
                if !q.filter.matches(&archetype.mask) {
                        continue
                }

                if !archetype.HasRelations() {
                        table := &q.world.storage.tables[archetype.tables[0]]
                        if table.Len() > 0 {
                                q.setTable(0, table)
                                return true
                        }
                        continue
                }

                q.tables = archetype.GetTables(q.relations)
                q.cursor.table = -1
                if q.nextTable(q.tables) {
                        return true
                }
        }
        q.Close()
        return false
}

func (q *Query4[A, B, C, D]) nextTable(tables []tableID) bool {
        maxTableIndex := len(tables) - 1
        for q.cursor.table < maxTableIndex {
                q.cursor.table++
                table := &q.world.storage.tables[tables[q.cursor.table]]
                if table.Len() == 0 {
                        continue
                }
                if !table.Matches(q.relations) {
                        continue
                }
                q.setTable(q.cursor.table, table)
                return true
        }
        if q.cache != nil {
                q.Close()
        }
        return false
}

func (q *Query4[A, B, C, D]) setTable(index int, table *table) {
        q.cursor.table = index
        q.table = table
        q.columnA = q.components[0].columns[q.table.id]
        q.columnB = q.components[1].columns[q.table.id]
        q.columnC = q.components[2].columns[q.table.id]
        q.columnD = q.components[3].columns[q.table.id]
        q.cursor.index = 0
        q.cursor.maxIndex = int64(q.table.Len() - 1)
}

type queryResult5[A any, B any, C any, D any, E any] struct {
        Entity Entity
        A      *A
        B      *B
        C      *C
        D      *D
        E      *E
}

// Query5 is a query for 5 components.
// Use a [Filter5] to create one.
//
// Queries are one-time use iterators and must be re-created each time before iterating.
//
// See [Query2] for a usage example.
type Query5[A any, B any, C any, D any, E any] struct {
        world      *World
        filter     *filter
        relations  []RelationID
        lock       uint8
        cursor     cursor
        tables     []tableID
        table      *table
        cache      *cacheEntry
        components []*componentStorage
        columnA    *column
        columnB    *column
        columnC    *column
        columnD    *column
        columnE    *column
}

func newQuery5[A any, B any, C any, D any, E any](world *World, filter *filter, ids []ID, relations []RelationID,
        cacheID cacheID, components []*componentStorage) Query5[A, B, C, D, E] {
        var cache *cacheEntry
        if cacheID != maxCacheID {
                cache = world.storage.getRegisteredFilter(cacheID)
        }

        return Query5[A, B, C, D, E]{
                world:      world,
                filter:     filter,
                relations:  relations,
                cache:      cache,
                lock:       world.lock(),
                components: components,
                cursor: cursor{
                        archetype: -1,
                        table:     -1,
                        index:     0,
                        maxIndex:  -1,
                },
        }
}

// Next advances the query's cursor to the next entity.
func (q *Query5[A, B, C, D, E]) Next() bool {
        q.world.checkQueryNext(&q.cursor)
        if int64(q.cursor.index) < q.cursor.maxIndex {
                q.cursor.index++
                return true
        }
        return q.nextTableOrArchetype()
}

// Entity returns the current entity.
func (q *Query5[A, B, C, D, E]) Entity() Entity {
        q.world.checkQueryGet(&q.cursor)
        return q.table.GetEntity(q.cursor.index)
}

// Get returns the queried components of the current entity.
//
// ⚠️ Do not store the obtained pointers outside of the current context (i.e. the query loop)!
func (q *Query5[A, B, C, D, E]) Get() (*A, *B, *C, *D, *E) {
        q.world.checkQueryGet(&q.cursor)
        return (*A)(q.columnA.Get(q.cursor.index)),
                (*B)(q.columnB.Get(q.cursor.index)),
                (*C)(q.columnC.Get(q.cursor.index)),
                (*D)(q.columnD.Get(q.cursor.index)),
                (*E)(q.columnE.Get(q.cursor.index))
}

// GetRelation returns the entity relation target of the component at the given index.
func (q *Query5[A, B, C, D, E]) GetRelation(index int) Entity {
        return q.components[index].columns[q.table.id].target
}

// Count counts the entities matching this query.
//
// Has some overhead of iterating through archetypes.
// However, this is still much faster than manual counting via iteration.
//
// Does not iterate or close the query.
func (q *Query5[A, B, C, D, E]) Count() int {
        if q.cache == nil {
                return countQuery(&q.world.storage, q.filter, q.relations)
        }
        return countQueryCache(&q.world.storage, q.cache, q.relations)
}

// Close closes the Query and unlocks the world.
//
// Automatically called when iteration completes.
// Needs to be called only if breaking out of the query iteration or not iterating at all.
func (q *Query5[A, B, C, D, E]) Close() {
        q.cursor.archetype = -2
        q.cursor.table = -2
        q.tables = nil
        q.table = nil
        q.cache = nil
        q.columnA = nil
        q.columnB = nil
        q.columnC = nil
        q.columnD = nil
        q.columnE = nil
        q.world.unlock(q.lock)
}

func (q *Query5[A, B, C, D, E]) nextTableOrArchetype() bool {
        if q.cache != nil {
                return q.nextTable(q.cache.tables)
        }
        if q.cursor.archetype >= 0 && q.nextTable(q.tables) {
                return true
        }
        return q.nextArchetype()
}

func (q *Query5[A, B, C, D, E]) nextArchetype() bool {
        maxArchIndex := len(q.world.storage.archetypes) - 1
        for q.cursor.archetype < maxArchIndex {
                q.cursor.archetype++
                archetype := &q.world.storage.archetypes[q.cursor.archetype]
                if !q.filter.matches(&archetype.mask) {
                        continue
                }

                if !archetype.HasRelations() {
                        table := &q.world.storage.tables[archetype.tables[0]]
                        if table.Len() > 0 {
                                q.setTable(0, table)
                                return true
                        }
                        continue
                }

                q.tables = archetype.GetTables(q.relations)
                q.cursor.table = -1
                if q.nextTable(q.tables) {
                        return true
                }
        }
        q.Close()
        return false
}

func (q *Query5[A, B, C, D, E]) nextTable(tables []tableID) bool {
        maxTableIndex := len(tables) - 1
        for q.cursor.table < maxTableIndex {
                q.cursor.table++
                table := &q.world.storage.tables[tables[q.cursor.table]]
                if table.Len() == 0 {
                        continue
                }
                if !table.Matches(q.relations) {
                        continue
                }
                q.setTable(q.cursor.table, table)
                return true
        }
        if q.cache != nil {
                q.Close()
        }
        return false
}

func (q *Query5[A, B, C, D, E]) setTable(index int, table *table) {
        q.cursor.table = index
        q.table = table
        q.columnA = q.components[0].columns[q.table.id]
        q.columnB = q.components[1].columns[q.table.id]
        q.columnC = q.components[2].columns[q.table.id]
        q.columnD = q.components[3].columns[q.table.id]
        q.columnE = q.components[4].columns[q.table.id]
        q.cursor.index = 0
        q.cursor.maxIndex = int64(q.table.Len() - 1)
}

type queryResult6[A any, B any, C any, D any, E any, F any] struct {
        Entity Entity
        A      *A
        B      *B
        C      *C
        D      *D
        E      *E
        F      *F
}

// Query6 is a query for 6 components.
// Use a [Filter6] to create one.
//
// Queries are one-time use iterators and must be re-created each time before iterating.
//
// See [Query2] for a usage example.
type Query6[A any, B any, C any, D any, E any, F any] struct {
        world      *World
        filter     *filter
        relations  []RelationID
        lock       uint8
        cursor     cursor
        tables     []tableID
        table      *table
        cache      *cacheEntry
        components []*componentStorage
        columnA    *column
        columnB    *column
        columnC    *column
        columnD    *column
        columnE    *column
        columnF    *column
}

func newQuery6[A any, B any, C any, D any, E any, F any](world *World, filter *filter, ids []ID, relations []RelationID,
        cacheID cacheID, components []*componentStorage) Query6[A, B, C, D, E, F] {
        var cache *cacheEntry
        if cacheID != maxCacheID {
                cache = world.storage.getRegisteredFilter(cacheID)
        }

        return Query6[A, B, C, D, E, F]{
                world:      world,
                filter:     filter,
                relations:  relations,
                cache:      cache,
                lock:       world.lock(),
                components: components,
                cursor: cursor{
                        archetype: -1,
                        table:     -1,
                        index:     0,
                        maxIndex:  -1,
                },
        }
}

// Next advances the query's cursor to the next entity.
func (q *Query6[A, B, C, D, E, F]) Next() bool {
        q.world.checkQueryNext(&q.cursor)
        if int64(q.cursor.index) < q.cursor.maxIndex {
                q.cursor.index++
                return true
        }
        return q.nextTableOrArchetype()
}

// Entity returns the current entity.
func (q *Query6[A, B, C, D, E, F]) Entity() Entity {
        q.world.checkQueryGet(&q.cursor)
        return q.table.GetEntity(q.cursor.index)
}

// Get returns the queried components of the current entity.
//
// ⚠️ Do not store the obtained pointers outside of the current context (i.e. the query loop)!
func (q *Query6[A, B, C, D, E, F]) Get() (*A, *B, *C, *D, *E, *F) {
        q.world.checkQueryGet(&q.cursor)
        return (*A)(q.columnA.Get(q.cursor.index)),
                (*B)(q.columnB.Get(q.cursor.index)),
                (*C)(q.columnC.Get(q.cursor.index)),
                (*D)(q.columnD.Get(q.cursor.index)),
                (*E)(q.columnE.Get(q.cursor.index)),
                (*F)(q.columnF.Get(q.cursor.index))
}

// GetRelation returns the entity relation target of the component at the given index.
func (q *Query6[A, B, C, D, E, F]) GetRelation(index int) Entity {
        return q.components[index].columns[q.table.id].target
}

// Count counts the entities matching this query.
//
// Has some overhead of iterating through archetypes.
// However, this is still much faster than manual counting via iteration.
//
// Does not iterate or close the query.
func (q *Query6[A, B, C, D, E, F]) Count() int {
        if q.cache == nil {
                return countQuery(&q.world.storage, q.filter, q.relations)
        }
        return countQueryCache(&q.world.storage, q.cache, q.relations)
}

// Close closes the Query and unlocks the world.
//
// Automatically called when iteration completes.
// Needs to be called only if breaking out of the query iteration or not iterating at all.
func (q *Query6[A, B, C, D, E, F]) Close() {
        q.cursor.archetype = -2
        q.cursor.table = -2
        q.tables = nil
        q.table = nil
        q.cache = nil
        q.columnA = nil
        q.columnB = nil
        q.columnC = nil
        q.columnD = nil
        q.columnE = nil
        q.columnF = nil
        q.world.unlock(q.lock)
}

func (q *Query6[A, B, C, D, E, F]) nextTableOrArchetype() bool {
        if q.cache != nil {
                return q.nextTable(q.cache.tables)
        }
        if q.cursor.archetype >= 0 && q.nextTable(q.tables) {
                return true
        }
        return q.nextArchetype()
}

func (q *Query6[A, B, C, D, E, F]) nextArchetype() bool {
        maxArchIndex := len(q.world.storage.archetypes) - 1
        for q.cursor.archetype < maxArchIndex {
                q.cursor.archetype++
                archetype := &q.world.storage.archetypes[q.cursor.archetype]
                if !q.filter.matches(&archetype.mask) {
                        continue
                }

                if !archetype.HasRelations() {
                        table := &q.world.storage.tables[archetype.tables[0]]
                        if table.Len() > 0 {
                                q.setTable(0, table)
                                return true
                        }
                        continue
                }

                q.tables = archetype.GetTables(q.relations)
                q.cursor.table = -1
                if q.nextTable(q.tables) {
                        return true
                }
        }
        q.Close()
        return false
}

func (q *Query6[A, B, C, D, E, F]) nextTable(tables []tableID) bool {
        maxTableIndex := len(tables) - 1
        for q.cursor.table < maxTableIndex {
                q.cursor.table++
                table := &q.world.storage.tables[tables[q.cursor.table]]
                if table.Len() == 0 {
                        continue
                }
                if !table.Matches(q.relations) {
                        continue
                }
                q.setTable(q.cursor.table, table)
                return true
        }
        if q.cache != nil {
                q.Close()
        }
        return false
}

func (q *Query6[A, B, C, D, E, F]) setTable(index int, table *table) {
        q.cursor.table = index
        q.table = table
        q.columnA = q.components[0].columns[q.table.id]
        q.columnB = q.components[1].columns[q.table.id]
        q.columnC = q.components[2].columns[q.table.id]
        q.columnD = q.components[3].columns[q.table.id]
        q.columnE = q.components[4].columns[q.table.id]
        q.columnF = q.components[5].columns[q.table.id]
        q.cursor.index = 0
        q.cursor.maxIndex = int64(q.table.Len() - 1)
}

type queryResult7[A any, B any, C any, D any, E any, F any, G any] struct {
        Entity Entity
        A      *A
        B      *B
        C      *C
        D      *D
        E      *E
        F      *F
        G      *G
}

// Query7 is a query for 7 components.
// Use a [Filter7] to create one.
//
// Queries are one-time use iterators and must be re-created each time before iterating.
//
// See [Query2] for a usage example.
type Query7[A any, B any, C any, D any, E any, F any, G any] struct {
        world      *World
        filter     *filter
        relations  []RelationID
        lock       uint8
        cursor     cursor
        tables     []tableID
        table      *table
        cache      *cacheEntry
        components []*componentStorage
        columnA    *column
        columnB    *column
        columnC    *column
        columnD    *column
        columnE    *column
        columnF    *column
        columnG    *column
}

func newQuery7[A any, B any, C any, D any, E any, F any, G any](world *World, filter *filter, ids []ID, relations []RelationID,
        cacheID cacheID, components []*componentStorage) Query7[A, B, C, D, E, F, G] {
        var cache *cacheEntry
        if cacheID != maxCacheID {
                cache = world.storage.getRegisteredFilter(cacheID)
        }

        return Query7[A, B, C, D, E, F, G]{
                world:      world,
                filter:     filter,
                relations:  relations,
                cache:      cache,
                lock:       world.lock(),
                components: components,
                cursor: cursor{
                        archetype: -1,
                        table:     -1,
                        index:     0,
                        maxIndex:  -1,
                },
        }
}

// Next advances the query's cursor to the next entity.
func (q *Query7[A, B, C, D, E, F, G]) Next() bool {
        q.world.checkQueryNext(&q.cursor)
        if int64(q.cursor.index) < q.cursor.maxIndex {
                q.cursor.index++
                return true
        }
        return q.nextTableOrArchetype()
}

// Entity returns the current entity.
func (q *Query7[A, B, C, D, E, F, G]) Entity() Entity {
        q.world.checkQueryGet(&q.cursor)
        return q.table.GetEntity(q.cursor.index)
}

// Get returns the queried components of the current entity.
//
// ⚠️ Do not store the obtained pointers outside of the current context (i.e. the query loop)!
func (q *Query7[A, B, C, D, E, F, G]) Get() (*A, *B, *C, *D, *E, *F, *G) {
        q.world.checkQueryGet(&q.cursor)
        return (*A)(q.columnA.Get(q.cursor.index)),
                (*B)(q.columnB.Get(q.cursor.index)),
                (*C)(q.columnC.Get(q.cursor.index)),
                (*D)(q.columnD.Get(q.cursor.index)),
                (*E)(q.columnE.Get(q.cursor.index)),
                (*F)(q.columnF.Get(q.cursor.index)),
                (*G)(q.columnG.Get(q.cursor.index))
}

// GetRelation returns the entity relation target of the component at the given index.
func (q *Query7[A, B, C, D, E, F, G]) GetRelation(index int) Entity {
        return q.components[index].columns[q.table.id].target
}

// Count counts the entities matching this query.
//
// Has some overhead of iterating through archetypes.
// However, this is still much faster than manual counting via iteration.
//
// Does not iterate or close the query.
func (q *Query7[A, B, C, D, E, F, G]) Count() int {
        if q.cache == nil {
                return countQuery(&q.world.storage, q.filter, q.relations)
        }
        return countQueryCache(&q.world.storage, q.cache, q.relations)
}

// Close closes the Query and unlocks the world.
//
// Automatically called when iteration completes.
// Needs to be called only if breaking out of the query iteration or not iterating at all.
func (q *Query7[A, B, C, D, E, F, G]) Close() {
        q.cursor.archetype = -2
        q.cursor.table = -2
        q.tables = nil
        q.table = nil
        q.cache = nil
        q.columnA = nil
        q.columnB = nil
        q.columnC = nil
        q.columnD = nil
        q.columnE = nil
        q.columnF = nil
        q.columnG = nil
        q.world.unlock(q.lock)
}

func (q *Query7[A, B, C, D, E, F, G]) nextTableOrArchetype() bool {
        if q.cache != nil {
                return q.nextTable(q.cache.tables)
        }
        if q.cursor.archetype >= 0 && q.nextTable(q.tables) {
                return true
        }
        return q.nextArchetype()
}

func (q *Query7[A, B, C, D, E, F, G]) nextArchetype() bool {
        maxArchIndex := len(q.world.storage.archetypes) - 1
        for q.cursor.archetype < maxArchIndex {
                q.cursor.archetype++
                archetype := &q.world.storage.archetypes[q.cursor.archetype]
                if !q.filter.matches(&archetype.mask) {
                        continue
                }

                if !archetype.HasRelations() {
                        table := &q.world.storage.tables[archetype.tables[0]]
                        if table.Len() > 0 {
                                q.setTable(0, table)
                                return true
                        }
                        continue
                }

                q.tables = archetype.GetTables(q.relations)
                q.cursor.table = -1
                if q.nextTable(q.tables) {
                        return true
                }
        }
        q.Close()
        return false
}

func (q *Query7[A, B, C, D, E, F, G]) nextTable(tables []tableID) bool {
        maxTableIndex := len(tables) - 1
        for q.cursor.table < maxTableIndex {
                q.cursor.table++
                table := &q.world.storage.tables[tables[q.cursor.table]]
                if table.Len() == 0 {
                        continue
                }
                if !table.Matches(q.relations) {
                        continue
                }
                q.setTable(q.cursor.table, table)
                return true
        }
        if q.cache != nil {
                q.Close()
        }
        return false
}

func (q *Query7[A, B, C, D, E, F, G]) setTable(index int, table *table) {
        q.cursor.table = index
        q.table = table
        q.columnA = q.components[0].columns[q.table.id]
        q.columnB = q.components[1].columns[q.table.id]
        q.columnC = q.components[2].columns[q.table.id]
        q.columnD = q.components[3].columns[q.table.id]
        q.columnE = q.components[4].columns[q.table.id]
        q.columnF = q.components[5].columns[q.table.id]
        q.columnG = q.components[6].columns[q.table.id]
        q.cursor.index = 0
        q.cursor.maxIndex = int64(q.table.Len() - 1)
}

type queryResult8[A any, B any, C any, D any, E any, F any, G any, H any] struct {
        Entity Entity
        A      *A
        B      *B
        C      *C
        D      *D
        E      *E
        F      *F
        G      *G
        H      *H
}

// Query8 is a query for 8 components.
// Use a [Filter8] to create one.
//
// Queries are one-time use iterators and must be re-created each time before iterating.
//
// See [Query2] for a usage example.
type Query8[A any, B any, C any, D any, E any, F any, G any, H any] struct {
        world      *World
        filter     *filter
        relations  []RelationID
        lock       uint8
        cursor     cursor
        tables     []tableID
        table      *table
        cache      *cacheEntry
        components []*componentStorage
        columnA    *column
        columnB    *column
        columnC    *column
        columnD    *column
        columnE    *column
        columnF    *column
        columnG    *column
        columnH    *column
}

func newQuery8[A any, B any, C any, D any, E any, F any, G any, H any](world *World, filter *filter, ids []ID, relations []RelationID,
        cacheID cacheID, components []*componentStorage) Query8[A, B, C, D, E, F, G, H] {
        var cache *cacheEntry
        if cacheID != maxCacheID {
                cache = world.storage.getRegisteredFilter(cacheID)
        }

        return Query8[A, B, C, D, E, F, G, H]{
                world:      world,
                filter:     filter,
                relations:  relations,
                cache:      cache,
                lock:       world.lock(),
                components: components,
                cursor: cursor{
                        archetype: -1,
                        table:     -1,
                        index:     0,
                        maxIndex:  -1,
                },
        }
}

// Next advances the query's cursor to the next entity.
func (q *Query8[A, B, C, D, E, F, G, H]) Next() bool {
        q.world.checkQueryNext(&q.cursor)
        if int64(q.cursor.index) < q.cursor.maxIndex {
                q.cursor.index++
                return true
        }
        return q.nextTableOrArchetype()
}

// Entity returns the current entity.
func (q *Query8[A, B, C, D, E, F, G, H]) Entity() Entity {
        q.world.checkQueryGet(&q.cursor)
        return q.table.GetEntity(q.cursor.index)
}

// Get returns the queried components of the current entity.
//
// ⚠️ Do not store the obtained pointers outside of the current context (i.e. the query loop)!
func (q *Query8[A, B, C, D, E, F, G, H]) Get() (*A, *B, *C, *D, *E, *F, *G, *H) {
        q.world.checkQueryGet(&q.cursor)
        return (*A)(q.columnA.Get(q.cursor.index)),
                (*B)(q.columnB.Get(q.cursor.index)),
                (*C)(q.columnC.Get(q.cursor.index)),
                (*D)(q.columnD.Get(q.cursor.index)),
                (*E)(q.columnE.Get(q.cursor.index)),
                (*F)(q.columnF.Get(q.cursor.index)),
                (*G)(q.columnG.Get(q.cursor.index)),
                (*H)(q.columnH.Get(q.cursor.index))
}

// GetRelation returns the entity relation target of the component at the given index.
func (q *Query8[A, B, C, D, E, F, G, H]) GetRelation(index int) Entity {
        return q.components[index].columns[q.table.id].target
}

// Count counts the entities matching this query.
//
// Has some overhead of iterating through archetypes.
// However, this is still much faster than manual counting via iteration.
//
// Does not iterate or close the query.
func (q *Query8[A, B, C, D, E, F, G, H]) Count() int {
        if q.cache == nil {
                return countQuery(&q.world.storage, q.filter, q.relations)
        }
        return countQueryCache(&q.world.storage, q.cache, q.relations)
}

// Close closes the Query and unlocks the world.
//
// Automatically called when iteration completes.
// Needs to be called only if breaking out of the query iteration or not iterating at all.
func (q *Query8[A, B, C, D, E, F, G, H]) Close() {
        q.cursor.archetype = -2
        q.cursor.table = -2
        q.tables = nil
        q.table = nil
        q.cache = nil
        q.columnA = nil
        q.columnB = nil
        q.columnC = nil
        q.columnD = nil
        q.columnE = nil
        q.columnF = nil
        q.columnG = nil
        q.columnH = nil
        q.world.unlock(q.lock)
}

func (q *Query8[A, B, C, D, E, F, G, H]) nextTableOrArchetype() bool {
        if q.cache != nil {
                return q.nextTable(q.cache.tables)
        }
        if q.cursor.archetype >= 0 && q.nextTable(q.tables) {
                return true
        }
        return q.nextArchetype()
}

func (q *Query8[A, B, C, D, E, F, G, H]) nextArchetype() bool {
        maxArchIndex := len(q.world.storage.archetypes) - 1
        for q.cursor.archetype < maxArchIndex {
                q.cursor.archetype++
                archetype := &q.world.storage.archetypes[q.cursor.archetype]
                if !q.filter.matches(&archetype.mask) {
                        continue
                }

                if !archetype.HasRelations() {
                        table := &q.world.storage.tables[archetype.tables[0]]
                        if table.Len() > 0 {
                                q.setTable(0, table)
                                return true
                        }
                        continue
                }

                q.tables = archetype.GetTables(q.relations)
                q.cursor.table = -1
                if q.nextTable(q.tables) {
                        return true
                }
        }
        q.Close()
        return false
}

func (q *Query8[A, B, C, D, E, F, G, H]) nextTable(tables []tableID) bool {
        maxTableIndex := len(tables) - 1
        for q.cursor.table < maxTableIndex {
                q.cursor.table++
                table := &q.world.storage.tables[tables[q.cursor.table]]
                if table.Len() == 0 {
                        continue
                }
                if !table.Matches(q.relations) {
                        continue
                }
                q.setTable(q.cursor.table, table)
                return true
        }
        if q.cache != nil {
                q.Close()
        }
        return false
}

func (q *Query8[A, B, C, D, E, F, G, H]) setTable(index int, table *table) {
        q.cursor.table = index
        q.table = table
        q.columnA = q.components[0].columns[q.table.id]
        q.columnB = q.components[1].columns[q.table.id]
        q.columnC = q.components[2].columns[q.table.id]
        q.columnD = q.components[3].columns[q.table.id]
        q.columnE = q.components[4].columns[q.table.id]
        q.columnF = q.components[5].columns[q.table.id]
        q.columnG = q.components[6].columns[q.table.id]
        q.columnH = q.components[7].columns[q.table.id]
        q.cursor.index = 0
        q.cursor.maxIndex = int64(q.table.Len() - 1)
}

mlange-42 / ark / 13754430621

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous