Commit 9786b3f1 authored by Robert Griesemer's avatar Robert Griesemer

gofmt-ify 4s, iterable

R=rsc
http://go/go-review/1016055
parent 2a899154
......@@ -32,7 +32,7 @@ func main() {
p = pieces5;
}
dx, dy := 500, 500;
w, err := av.Init(av.SubsystemVideo|av.SubsystemAudio, dx, dy);
w, err := av.Init(av.SubsystemVideo | av.SubsystemAudio, dx, dy);
if err != nil {
log.Exit(err);
}
......@@ -55,7 +55,7 @@ func audioServer() {
b := <-sndc;
for len(b)*2 >= n {
var a []uint16;
a, b = b[0:n/2], b[n/2:len(b)];
a, b = b[0 : n/2], b[n/2 : len(b)];
n, err = av.AudioStream(a);
if err != nil {
log.Exit(err);
......@@ -74,6 +74,6 @@ func PlaySound(b []uint16) {
sndc <- b;
}
var whoosh = []uint16 {
// Insert your favorite sound samples here.
var whoosh = []uint16{
// Insert your favorite sound samples here.
}
This diff is collapsed.
This diff is collapsed.
......@@ -12,77 +12,77 @@ import "container/vector"
type Iterable interface {
// Iter should return a fresh channel each time it is called.
Iter() <-chan interface {}
Iter() <-chan interface{};
}
func not(f func(interface {}) bool) (func(interface {}) bool) {
return func(e interface {}) bool { return !f(e) }
func not(f func(interface{}) bool) (func(interface{}) bool) {
return func(e interface{}) bool { return !f(e) };
}
// All tests whether f is true for every element of iter.
func All(iter Iterable, f func(interface {}) bool) bool {
func All(iter Iterable, f func(interface{}) bool) bool {
for e := range iter.Iter() {
if !f(e) {
return false
return false;
}
}
return true
return true;
}
// Any tests whether f is true for at least one element of iter.
func Any(iter Iterable, f func(interface {}) bool) bool {
return !All(iter, not(f))
func Any(iter Iterable, f func(interface{}) bool) bool {
return !All(iter, not(f));
}
// Data returns a slice containing the elements of iter.
func Data(iter Iterable) []interface {} {
func Data(iter Iterable) []interface{} {
vec := vector.New(0);
for e := range iter.Iter() {
vec.Push(e)
vec.Push(e);
}
return vec.Data()
return vec.Data();
}
// filteredIterable is a struct that implements Iterable with each element
// passed through a filter.
type filteredIterable struct {
it Iterable;
f func(interface {}) bool;
f func(interface{}) bool;
}
func (f *filteredIterable) iterate(out chan<- interface {}) {
func (f *filteredIterable) iterate(out chan<- interface{}) {
for e := range f.it.Iter() {
if f.f(e) {
out <- e
out <- e;
}
}
close(out)
close(out);
}
func (f *filteredIterable) Iter() <-chan interface {} {
ch := make(chan interface {});
func (f *filteredIterable) Iter() <-chan interface{} {
ch := make(chan interface{});
go f.iterate(ch);
return ch;
}
// Filter returns an Iterable that returns the elements of iter that satisfy f.
func Filter(iter Iterable, f func(interface {}) bool) Iterable {
return &filteredIterable{ iter, f }
func Filter(iter Iterable, f func(interface{}) bool) Iterable {
return &filteredIterable{iter, f};
}
// Find returns the first element of iter that satisfies f.
// Returns nil if no such element is found.
func Find(iter Iterable, f func(interface {}) bool) interface {} {
func Find(iter Iterable, f func(interface{}) bool) interface{} {
for e := range Filter(iter, f).Iter() {
return e
return e;
}
return nil
return nil;
}
// Injector is a type representing a function that takes two arguments,
// an accumulated value and an element, and returns the next accumulated value.
// See the Inject function.
type Injector func(interface {}, interface {}) interface{};
type Injector func(interface{}, interface{}) interface{}
// Inject combines the elements of iter by repeatedly calling f with an
// accumulated value and each element in order. The starting accumulated value
......@@ -92,42 +92,42 @@ type Injector func(interface {}, interface {}) interface{};
// sum := iterable.Inject(arr, 0,
// func(ax interface {}, x interface {}) interface {} {
// return ax.(int) + x.(int) }).(int)
func Inject(iter Iterable, initial interface {}, f Injector) interface {} {
func Inject(iter Iterable, initial interface{}, f Injector) interface{} {
acc := initial;
for e := range iter.Iter() {
acc = f(acc, e)
acc = f(acc, e);
}
return acc
return acc;
}
// mappedIterable is a helper struct that implements Iterable, returned by Map.
type mappedIterable struct {
it Iterable;
f func(interface {}) interface {};
f func(interface{}) interface{};
}
func (m *mappedIterable) iterate(out chan<- interface {}) {
func (m *mappedIterable) iterate(out chan<- interface{}) {
for e := range m.it.Iter() {
out <- m.f(e)
out <- m.f(e);
}
close(out)
close(out);
}
func (m *mappedIterable) Iter() <-chan interface {} {
ch := make(chan interface {});
func (m *mappedIterable) Iter() <-chan interface{} {
ch := make(chan interface{});
go m.iterate(ch);
return ch
return ch;
}
// Map returns an Iterable that returns the result of applying f to each
// element of iter.
func Map(iter Iterable, f func(interface {}) interface {}) Iterable {
return &mappedIterable{ iter, f }
func Map(iter Iterable, f func(interface{}) interface{}) Iterable {
return &mappedIterable{iter, f};
}
// Partition(iter, f) returns Filter(iter, f) and Filter(iter, !f).
func Partition(iter Iterable, f func(interface {}) bool) (Iterable, Iterable) {
return Filter(iter, f), Filter(iter, not(f))
func Partition(iter Iterable, f func(interface{}) bool) (Iterable, Iterable) {
return Filter(iter, f), Filter(iter, not(f));
}
// TODO:
......
......@@ -8,82 +8,83 @@ import (
"testing";
)
type IntArray []int;
type IntArray []int
func (arr IntArray) Iter() <-chan interface {} {
ch := make(chan interface {});
func (arr IntArray) Iter() <-chan interface{} {
ch := make(chan interface{});
go func() {
for _, x := range arr {
ch <- x
ch <- x;
}
close(ch)
close(ch);
}();
return ch
return ch;
}
var oneToFive = IntArray{ 1, 2, 3, 4, 5 };
var oneToFive = IntArray{1, 2, 3, 4, 5}
func isNegative(n interface {}) bool {
return n.(int) < 0
func isNegative(n interface{}) bool {
return n.(int) < 0;
}
func isPositive(n interface {}) bool {
return n.(int) > 0
func isPositive(n interface{}) bool {
return n.(int) > 0;
}
func isAbove3(n interface {}) bool {
return n.(int) > 3
func isAbove3(n interface{}) bool {
return n.(int) > 3;
}
func isEven(n interface {}) bool {
return n.(int) % 2 == 0
func isEven(n interface{}) bool {
return n.(int) % 2 == 0;
}
func doubler(n interface {}) interface {} {
return n.(int) * 2
func doubler(n interface{}) interface{} {
return n.(int) * 2;
}
func addOne(n interface {}) interface {} {
return n.(int) + 1
func addOne(n interface{}) interface{} {
return n.(int) + 1;
}
func adder(acc interface {}, n interface {}) interface {} {
return acc.(int) + n.(int)
func adder(acc interface{}, n interface{}) interface{} {
return acc.(int) + n.(int);
}
// A stream of the natural numbers: 0, 1, 2, 3, ...
type integerStream struct {}
func (i integerStream) Iter() <-chan interface {} {
ch := make(chan interface {});
type integerStream struct{}
func (i integerStream) Iter() <-chan interface{} {
ch := make(chan interface{});
go func() {
for i := 0; ; i++ {
ch <- i
ch <- i;
}
}();
return ch
return ch;
}
func TestAll(t *testing.T) {
if !All(oneToFive, isPositive) {
t.Error("All(oneToFive, isPositive) == false")
t.Error("All(oneToFive, isPositive) == false");
}
if All(oneToFive, isAbove3) {
t.Error("All(oneToFive, isAbove3) == true")
t.Error("All(oneToFive, isAbove3) == true");
}
}
func TestAny(t *testing.T) {
if Any(oneToFive, isNegative) {
t.Error("Any(oneToFive, isNegative) == true")
t.Error("Any(oneToFive, isNegative) == true");
}
if !Any(oneToFive, isEven) {
t.Error("Any(oneToFive, isEven) == false")
t.Error("Any(oneToFive, isEven) == false");
}
}
func assertArraysAreEqual(t *testing.T, res []interface {}, expected []int) {
func assertArraysAreEqual(t *testing.T, res []interface{}, expected []int) {
if len(res) != len(expected) {
t.Errorf("len(res) = %v, want %v", len(res), len(expected));
goto missing
goto missing;
}
for i := range res {
if v := res[i].(int); v != expected[i] {
t.Errorf("res[%v] = %v, want %v", i, v, expected[i]);
goto missing
goto missing;
}
}
return;
......@@ -94,35 +95,35 @@ missing:
func TestFilter(t *testing.T) {
ints := integerStream{};
moreInts := Filter(ints, isAbove3).Iter();
res := make([]interface {}, 3);
res := make([]interface{}, 3);
for i := 0; i < 3; i++ {
res[i] = <-moreInts;
}
assertArraysAreEqual(t, res, []int{ 4, 5, 6 })
assertArraysAreEqual(t, res, []int{4, 5, 6});
}
func TestFind(t *testing.T) {
ints := integerStream{};
first := Find(ints, isAbove3);
if first.(int) != 4 {
t.Errorf("Find(ints, isAbove3) = %v, want 4", first)
t.Errorf("Find(ints, isAbove3) = %v, want 4", first);
}
}
func TestInject(t *testing.T) {
res := Inject(oneToFive, 0, adder);
if res.(int) != 15 {
t.Errorf("Inject(oneToFive, 0, adder) = %v, want 15", res)
t.Errorf("Inject(oneToFive, 0, adder) = %v, want 15", res);
}
}
func TestMap(t *testing.T) {
res := Data(Map(Map(oneToFive, doubler), addOne));
assertArraysAreEqual(t, res, []int{ 3, 5, 7, 9, 11 })
assertArraysAreEqual(t, res, []int{3, 5, 7, 9, 11});
}
func TestPartition(t *testing.T) {
ti, fi := Partition(oneToFive, isEven);
assertArraysAreEqual(t, Data(ti), []int{ 2, 4 });
assertArraysAreEqual(t, Data(fi), []int{ 1, 3, 5 })
assertArraysAreEqual(t, Data(ti), []int{2, 4});
assertArraysAreEqual(t, Data(fi), []int{1, 3, 5});
}
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment