123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203 |
- /*
- Copyright The Kubernetes Authors.
- Licensed under the Apache License, Version 2.0 (the "License");
- you may not use this file except in compliance with the License.
- You may obtain a copy of the License at
- http://www.apache.org/licenses/LICENSE-2.0
- Unless required by applicable law or agreed to in writing, software
- distributed under the License is distributed on an "AS IS" BASIS,
- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- See the License for the specific language governing permissions and
- limitations under the License.
- */
- // Code generated by set-gen. DO NOT EDIT.
- package sets
- import (
- "reflect"
- "sort"
- )
- // sets.Int is a set of ints, implemented via map[int]struct{} for minimal memory consumption.
- type Int map[int]Empty
- // NewInt creates a Int from a list of values.
- func NewInt(items ...int) Int {
- ss := Int{}
- ss.Insert(items...)
- return ss
- }
- // IntKeySet creates a Int from a keys of a map[int](? extends interface{}).
- // If the value passed in is not actually a map, this will panic.
- func IntKeySet(theMap interface{}) Int {
- v := reflect.ValueOf(theMap)
- ret := Int{}
- for _, keyValue := range v.MapKeys() {
- ret.Insert(keyValue.Interface().(int))
- }
- return ret
- }
- // Insert adds items to the set.
- func (s Int) Insert(items ...int) {
- for _, item := range items {
- s[item] = Empty{}
- }
- }
- // Delete removes all items from the set.
- func (s Int) Delete(items ...int) {
- for _, item := range items {
- delete(s, item)
- }
- }
- // Has returns true if and only if item is contained in the set.
- func (s Int) Has(item int) bool {
- _, contained := s[item]
- return contained
- }
- // HasAll returns true if and only if all items are contained in the set.
- func (s Int) HasAll(items ...int) bool {
- for _, item := range items {
- if !s.Has(item) {
- return false
- }
- }
- return true
- }
- // HasAny returns true if any items are contained in the set.
- func (s Int) HasAny(items ...int) bool {
- for _, item := range items {
- if s.Has(item) {
- return true
- }
- }
- return false
- }
- // Difference returns a set of objects that are not in s2
- // For example:
- // s1 = {a1, a2, a3}
- // s2 = {a1, a2, a4, a5}
- // s1.Difference(s2) = {a3}
- // s2.Difference(s1) = {a4, a5}
- func (s Int) Difference(s2 Int) Int {
- result := NewInt()
- for key := range s {
- if !s2.Has(key) {
- result.Insert(key)
- }
- }
- return result
- }
- // Union returns a new set which includes items in either s1 or s2.
- // For example:
- // s1 = {a1, a2}
- // s2 = {a3, a4}
- // s1.Union(s2) = {a1, a2, a3, a4}
- // s2.Union(s1) = {a1, a2, a3, a4}
- func (s1 Int) Union(s2 Int) Int {
- result := NewInt()
- for key := range s1 {
- result.Insert(key)
- }
- for key := range s2 {
- result.Insert(key)
- }
- return result
- }
- // Intersection returns a new set which includes the item in BOTH s1 and s2
- // For example:
- // s1 = {a1, a2}
- // s2 = {a2, a3}
- // s1.Intersection(s2) = {a2}
- func (s1 Int) Intersection(s2 Int) Int {
- var walk, other Int
- result := NewInt()
- if s1.Len() < s2.Len() {
- walk = s1
- other = s2
- } else {
- walk = s2
- other = s1
- }
- for key := range walk {
- if other.Has(key) {
- result.Insert(key)
- }
- }
- return result
- }
- // IsSuperset returns true if and only if s1 is a superset of s2.
- func (s1 Int) IsSuperset(s2 Int) bool {
- for item := range s2 {
- if !s1.Has(item) {
- return false
- }
- }
- return true
- }
- // Equal returns true if and only if s1 is equal (as a set) to s2.
- // Two sets are equal if their membership is identical.
- // (In practice, this means same elements, order doesn't matter)
- func (s1 Int) Equal(s2 Int) bool {
- return len(s1) == len(s2) && s1.IsSuperset(s2)
- }
- type sortableSliceOfInt []int
- func (s sortableSliceOfInt) Len() int { return len(s) }
- func (s sortableSliceOfInt) Less(i, j int) bool { return lessInt(s[i], s[j]) }
- func (s sortableSliceOfInt) Swap(i, j int) { s[i], s[j] = s[j], s[i] }
- // List returns the contents as a sorted int slice.
- func (s Int) List() []int {
- res := make(sortableSliceOfInt, 0, len(s))
- for key := range s {
- res = append(res, key)
- }
- sort.Sort(res)
- return []int(res)
- }
- // UnsortedList returns the slice with contents in random order.
- func (s Int) UnsortedList() []int {
- res := make([]int, 0, len(s))
- for key := range s {
- res = append(res, key)
- }
- return res
- }
- // Returns a single element from the set.
- func (s Int) PopAny() (int, bool) {
- for key := range s {
- s.Delete(key)
- return key, true
- }
- var zeroValue int
- return zeroValue, false
- }
- // Len returns the size of the set.
- func (s Int) Len() int {
- return len(s)
- }
- func lessInt(lhs, rhs int) bool {
- return lhs < rhs
- }
|