summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--binarize/main.go35
1 files changed, 25 insertions, 10 deletions
diff --git a/binarize/main.go b/binarize/main.go
index 610effc..aa8d4ee 100644
--- a/binarize/main.go
+++ b/binarize/main.go
@@ -6,10 +6,12 @@ package main
import (
"flag"
"fmt"
+ "image"
+ "image/draw"
+ _ "image/jpeg"
+ "image/png"
"log"
"os"
-
- "github.com/Ernyoke/Imger/imgio" // TODO: get rid of this and do things myself
)
func main() {
@@ -29,16 +31,29 @@ func main() {
*wsize++
}
- img, err := imgio.ImreadGray(flag.Arg(0))
+ f, err := os.Open(flag.Arg(0))
+ defer f.Close()
if err != nil {
- log.Fatalf("Could not read image %s\n", flag.Arg(0))
+ log.Fatalf("Could not open file %s: %v\n", flag.Arg(0), err)
}
-
- // TODO: estimate an appropriate window size based on resolution
- thresh := IntegralSauvola(img, *ksize, *wsize)
-
- err = imgio.Imwrite(thresh, flag.Arg(1))
+ img, _, err := image.Decode(f)
if err != nil {
- log.Fatalf("Could not write image %s\n", flag.Arg(1))
+ log.Fatalf("Could not decode image: %v\n", err)
}
+ b := img.Bounds()
+ gray := image.NewGray(image.Rect(0, 0, b.Dx(), b.Dy()))
+ draw.Draw(gray, b, img, b.Min, draw.Src)
+
+ // TODO: estimate an appropriate window size based on resolution
+ thresh := IntegralSauvola(gray, *ksize, *wsize)
+
+ f, err = os.Create(flag.Arg(1))
+ if err != nil {
+ log.Fatalf("Could not create file %s: %v\n", flag.Arg(1), err)
+ }
+ defer f.Close()
+ err = png.Encode(f, thresh)
+ if err != nil {
+ log.Fatalf("Could not encode image: %v\n", err)
+ }
}