X-Git-Url: http://git.iain.cx/?a=blobdiff_plain;f=screen%2Fscreen.go;h=251ca8778824cda9ca3f7c045f913002eb62b517;hb=e5c137e9e63fff4093a152ade0eb105ba43237b2;hp=c29c6bad01080a9464b40ba2c36c8f651df0a262;hpb=c72057ae6bd4a976a079773c0c42bedb69589541;p=pstop.git

diff --git a/screen/screen.go b/screen/screen.go
index c29c6ba..251ca87 100644
--- a/screen/screen.go
+++ b/screen/screen.go
@@ -33,7 +33,7 @@ func (s *TermboxScreen) BoldPrintAt(x int, y int, text string) {
 
 // clear the screen
 func (s *TermboxScreen) Clear() {
-	termbox.Clear(termbox.ColorWhite, termbox.ColorBlack)
+	termbox.Clear(termbox.ColorDefault, termbox.ColorDefault)
 }
 
 // close the screen
@@ -83,7 +83,7 @@ func (s *TermboxScreen) Initialise() {
 	s.fg = termbox.ColorDefault
 	s.bg = termbox.ColorDefault
 
-	s.SetSize( termbox.Size() )
+	s.SetSize(termbox.Size())
 }
 
 // print the characters but don't print them outside the screen
@@ -114,3 +114,15 @@ func (s *TermboxScreen) SetSize(width, height int) {
 func (s *TermboxScreen) Size() (int, int) {
 	return s.width, s.height
 }
+
+// create a channel for termbox.Events and run a poller to send
+// these events to the channel.  Return the channel.
+func (s TermboxScreen) TermBoxChan() chan termbox.Event {
+	termboxChan := make(chan termbox.Event)
+	go func() {
+		for {
+			termboxChan <- termbox.PollEvent()
+		}
+	}()
+	return termboxChan
+}