func main() {
// Run the CPU profile if we're instructed to.
if len(flagProfile) > 0 {
f, err := os.Create(flagProfile)
if err != nil {
log.Fatal(err)
}
pprof.StartCPUProfile(f)
defer pprof.StopCPUProfile()
}
// Whoops!
if flag.NArg() == 0 {
fmt.Fprint(os.Stderr, "\n")
log.Print("No images specified.\n\n")
usage()
}
driver.Main(func(s screen.Screen) {
// Decode all images (in parallel).
names, imgs := decodeImages(findFiles(flag.Args()))
// Die now if we don't have any images!
if len(imgs) == 0 {
log.Fatal("No images specified could be shown. Quitting...")
}
winSize := image.Point{flagWidth, flagHeight}
// Auto-size the window if appropriate.
if flagAutoResize {
log.Printf(">>> img[%s]...\n", names[0])
b := imgs[0].Bounds()
winSize = image.Point{b.Dx(), b.Dy()}
}
w, err := s.NewWindow(&screen.NewWindowOptions{
Width: winSize.X,
Height: winSize.Y,
})
if err != nil {
log.Fatal(err)
}
defer w.Release()
b, err := s.NewBuffer(winSize)
if err != nil {
log.Fatal(err)
}
defer b.Release()
w.Fill(b.Bounds(), color.White, draw.Src)
w.Publish()
var sz size.Event
var i int // index of image to display
for {
e := w.NextEvent()
switch e := e.(type) {
default:
case mouse.Event:
case key.Event:
repaint := false
switch e.Code {
case key.CodeEscape, key.CodeQ:
return
case key.CodeRightArrow:
if e.Direction == key.DirPress {
if i == len(imgs)-1 {
i = -1
}
i++
repaint = true
b.Release()
b, err = s.NewBuffer(sz.Size())
if err != nil {
log.Fatal(err)
}
}
case key.CodeLeftArrow:
if e.Direction == key.DirPress {
if i == 0 {
i = len(imgs)
}
i--
repaint = true
b, err = s.NewBuffer(sz.Size())
if err != nil {
log.Fatal(err)
}
}
case key.CodeR:
if e.Direction == key.DirPress {
// resize to current image
r := imgs[i].Bounds()
sz.HeightPx = r.Dy()
sz.WidthPx = r.Dx()
repaint = true
b, err = s.NewBuffer(sz.Size())
if err != nil {
log.Fatal(err)
}
w.Publish()
}
}
if repaint {
w.Send(paint.Event{})
}
case paint.Event:
img := imgs[i]
draw.Draw(b.RGBA(), b.Bounds(), img, image.Point{}, draw.Src)
dp := vpCenter(img, sz.WidthPx, sz.HeightPx)
zero := image.Point{}
if dp != zero {
w.Fill(sz.Bounds(), color.Black, draw.Src)
}
w.Upload(dp, b, b.Bounds())
case size.Event:
sz = e
case error:
log.Print(e)
}
}
})
}
