value_set_generic.go

package rel

import (
	"bytes"
	"reflect"
	"sort"

	"github.com/arr-ai/frozen"
	"github.com/arr-ai/wbnf/parser"
)

// GenericSet is a set of Values.
type GenericSet struct {
	set frozen.Set
}

// genericSet equivalents for Boolean true and false
var (
	None  = Set(GenericSet{frozen.Set{}})
	False = None
	True  = None.With(EmptyTuple)

	stringCharTupleType = reflect.TypeOf(StringCharTuple{})
	bytesByteTupleType  = reflect.TypeOf(BytesByteTuple{})
	arrayItemTupleType  = reflect.TypeOf(ArrayItemTuple{})
	dictEntryTupleType  = reflect.TypeOf(DictEntryTuple{})
)

// NewSet constructs a genericSet from a set of Values.
func NewSet(values ...Value) Set {
	set := None
	if len(values) > 0 {
		typ := reflect.TypeOf(values[0])
		for _, value := range values[1:] {
			if reflect.TypeOf(value) != typ {
				typ = nil
				break
			}
		}
		if typ != nil {
			switch typ {
			case stringCharTupleType:
				for _, value := range values {
					set = set.With(value)
				}
				s, is := AsString(set)
				if !is {
					panic("unsupported array expr")
				}
				return s
			case bytesByteTupleType:
				for _, value := range values {
					set = set.With(value)
				}
				b, is := AsBytes(set)
				if !is {
					panic("unsupported array expr")
				}
				return b
			case arrayItemTupleType:
				for _, value := range values {
					set = set.With(value)
				}
				array, is := asArray(set)
				if !is {
					panic("unsupported array expr")
				}
				return array
			case dictEntryTupleType:
				tuples := make([]DictEntryTuple, 0, len(values))
				for _, value := range values {
					tuples = append(tuples, value.(DictEntryTuple))
				}
				return NewDict(true, tuples...)
			}
		}
		for _, value := range values {
			set = set.With(value)
		}
	}
	return set
}

func CanonicalSet(s Set) Set {
	if s, ok := s.(GenericSet); ok {
		values := make([]Value, 0, s.Count())
		for e := s.Enumerator(); e.MoveNext(); {
			values = append(values, e.Current())
		}
		return NewSet(values...)
	}
	return s
}

// NewSetFrom constructs a genericSet from interfaces.
func NewSetFrom(intfs ...interface{}) (Set, error) {
	set := None
	for _, intf := range intfs {
		value, err := NewValue(intf)
		if err != nil {
			return nil, err
		}
		set = set.With(value)
	}
	return set, nil
}

func newSetFromSet(s Set) Set {
	set := None
	for e := s.Enumerator(); e.MoveNext(); {
		set = set.With(e.Current())
	}
	return set
}

// NewBool constructs a bool as a relation.
func NewBool(b bool) Set {
	if b {
		return True
	}
	return False
}

// Hash computes a hash for a genericSet.
func (s GenericSet) Hash(seed uintptr) uintptr {
	h := seed
	for e := s.Enumerator(); e.MoveNext(); {
		h ^= e.Current().Hash(0)
	}
	return h
}

// Equal tests two Sets for equality. Any other type returns false.
func (s GenericSet) Equal(v interface{}) bool {
	if t, ok := v.(GenericSet); ok {
		return s.set.Equal(t.set)
	}
	return false
}

// String returns a string representation of a genericSet.
func (s GenericSet) String() string {
	// {} == none
	if !s.IsTrue() {
		return "{}"
	}

	// {()} == true
	if s.Count() == 1 {
		e := s.Enumerator()
		e.MoveNext()
		if tuple, ok := e.Current().(Tuple); ok && !tuple.IsTrue() {
			return "true"
		}
	}

	var buf bytes.Buffer
	buf.WriteString("{")
	for i, value := range s.OrderedValues() {
		if i != 0 {
			buf.WriteString(", ")
		}
		buf.WriteString(value.String())
	}
	buf.WriteString("}")
	return buf.String()
}

// Eval returns the set.
func (s GenericSet) Eval(local Scope) (Value, error) {
	return s, nil
}

// Source returns a scanner locating the GenericSet's source code.
func (s GenericSet) Source() parser.Scanner {
	return *parser.NewScanner("")
}

var genericSetKind = registerKind(200, reflect.TypeOf(Function{}))

// Kind returns a number that is unique for each major kind of Value.
func (s GenericSet) Kind() int {
	return genericSetKind
}

// Bool returns true iff the tuple has attributes.
func (s GenericSet) IsTrue() bool {
	return s.Count() > 0
}

// Less returns true iff v.Kind() < genericSet.Kind() or v is a
// genericSet and t precedes v in a lexicographical comparison of their
// sorted values.
func (s GenericSet) Less(v Value) bool {
	if s.Kind() != v.Kind() {
		return s.Kind() < v.Kind()
	}

	x := v.(GenericSet)
	a := s.OrderedValues()
	b := x.OrderedValues()
	n := len(a)
	if n > len(b) {
		n = len(b)
	}
	for i := 0; i < n; i++ {
		if a[i].Less(b[i]) {
			return true
		}
		if b[i].Less(a[i]) {
			return false
		}
	}
	return len(a) < len(b)
}

// Negate returns {(negateTag): s}.
func (s GenericSet) Negate() Value {
	if !s.IsTrue() {
		return s
	}
	return NewTuple(NewAttr(negateTag, s))
}

// Export exports a genericSet as an array of exported Values.
func (s GenericSet) Export() interface{} {
	if s.set.IsEmpty() {
		return []interface{}{}
	}
	if s, is := AsString(s); is {
		return s.Export()
	}
	result := make([]interface{}, 0, s.set.Count())
	for e := s.Enumerator(); e.MoveNext(); {
		result = append(result, e.Current().Export())
	}
	return result
}

// Count returns the number of elements in the genericSet.
func (s GenericSet) Count() int {
	return s.set.Count()
}

// Has returns true iff the given Value is in the genericSet.
func (s GenericSet) Has(value Value) bool {
	return s.set.Has(value)
}

// With returns the original genericSet with given value added. Iff the value was
// already present, the original genericSet is returned.
func (s GenericSet) With(value Value) Set {
	return GenericSet{s.set.With(value)}
}

// Without returns the original genericSet without the given value. Iff the value was
// already absent, the original genericSet is returned.
func (s GenericSet) Without(value Value) Set {
	set := s.set.Without(value)
	if set == s.set {
		return s
	}
	return GenericSet{set}
}

// Map maps values per f.
func (s GenericSet) Map(f func(v Value) Value) Set {
	result := NewSet()
	for e := s.Enumerator(); e.MoveNext(); {
		result = result.With(f(e.Current()))
	}
	return result
}

// Where returns a new genericSet with all the Values satisfying predicate p.
func (s GenericSet) Where(p func(v Value) (bool, error)) (_ Set, err error) {
	s.set = s.set.Where(func(elem interface{}) bool {
		if err != nil {
			return false
		}
		match, err2 := p(elem.(Value))
		if err2 != nil {
			err = err2
			return false
		}
		return match
	})
	return s, err
}

func (s GenericSet) CallAll(arg Value) (Set, error) {
	var t Tuple
	var at Value
	tm := NewTupleMatcher(map[string]Matcher{"@": Bind(&at)}, Bind(&t))
	set := None
	for e := s.Enumerator(); e.MoveNext(); {
		if tm.Match(e.Current()) && at.Equal(arg) {
			if t.Count() != 1 {
				panic("GenericSet.CallAll: only works on binary tuple with one '@' attribute")
			}
			for attr := t.Enumerator(); attr.MoveNext(); {
				_, value := attr.Current()
				set = set.With(value)
			}
		}
	}
	return set, nil
}

// Enumerator returns an enumerator over the Values in the genericSet.
func (s GenericSet) Enumerator() ValueEnumerator {
	return &genericSetEnumerator{s.set.Range()}
}

// Any return any value from the set.
func (s GenericSet) Any() Value {
	for e := s.Enumerator(); e.MoveNext(); {
		return e.Current()
	}
	panic("Any(): empty set")
}

func (s GenericSet) ArrayEnumerator() (OffsetValueEnumerator, bool) {
	return &arrayEnumerator{
		i: s.set.OrderedRange(func(a, b interface{}) bool {
			return a.(Tuple).MustGet("@").(Number) < b.(Tuple).MustGet("@").(Number)
		}),
	}, true
}

// genericSetEnumerator represents an enumerator over a genericSet.
type genericSetEnumerator struct {
	i frozen.Iterator
}

// MoveNext moves the enumerator to the next Value.
func (e *genericSetEnumerator) MoveNext() bool {
	return e.i.Next()
}

// Current returns the enumerator's current Value.
func (e *genericSetEnumerator) Current() Value {
	return e.i.Value().(Value)
}

// ValueList represents a []Value for use in sort.Sort().
type ValueList []Value

func (vl ValueList) Len() int {
	return len(vl)
}

func (vl ValueList) Less(i, j int) bool {
	return vl[i].Less(vl[j])
}

func (vl ValueList) Swap(i, j int) {
	vl[i], vl[j] = vl[j], vl[i]
}

// OrderedValues returns Values in a canonical ordering.
func (s GenericSet) OrderedValues() []Value {
	a := make([]Value, s.Count())
	i := 0
	for e := s.Enumerator(); e.MoveNext(); {
		a[i] = e.Current()
		i++
	}
	sort.Sort(ValueList(a))
	return a
}