format and quality option

Author: Manas140

help.go

@@ -13,10 +13,12 @@ func help() {
     -i input_directory
     -o output_directory (defaults to ./out)
     -r rough_height (defaults to 5000)
-
+    -f output_format [jpeg|jpg|png] (defaults to jpeg)
+    -q jpeg_quality (defaults to 100)
+    
   %s
     -a: use absolute height (defaults to false)
-  
+
   %s
     -n neighbor_count (defaults to 5)
     -s skip_step (defaults to 5)
@@ -26,7 +28,7 @@ func help() {
     https://github.com/manas140/seam/issues
   %s`, col("Seam",
 		34),
-		col("v0.0.1", 2),
+		col("v0.0.2", 2),
 		col("General:", 32),
 		col("Toggle:", 36),
 		col("Advanced:", 35),

main.go

@@ -43,6 +43,7 @@ func main() {
 		format         string = "jpeg"
 		skipStep       int    = 5
 		neighborCount  int    = 5
+		quality        int    = 100
 	)
 
 	// get arguments
@@ -71,6 +72,8 @@ func main() {
 			}
 			i++
 			format = args[i]
+		case "-q":
+			i = checkArgs("quality", args, &quality, i)
 		case "-s":
 			i = checkArgs("skip step", args, &skipStep, i)
 		case "-n":
@@ -165,7 +168,7 @@ func main() {
 	var wg sync.WaitGroup
 	wg.Add(len(slicedImg))
 	for i, img := range slicedImg {
-		go saveImage(img, fmt.Sprintf("%s/%d.%s", outputDir, i, format), format, &wg)
+		go saveImage(img, fmt.Sprintf("%s/%d.%s", outputDir, i, format), format, quality, &wg)
 	}
 	wg.Wait()
 	out.Debug("Took %.2fs", time.Since(start).Seconds())

utils.go

@@ -40,7 +40,7 @@ func readImage(path string) (image.Image, error) {
 	return img, nil
 }
 
-func saveImage(img image.Image, path string, ext string, wg *sync.WaitGroup) {
+func saveImage(img image.Image, path string, ext string, quality int, wg *sync.WaitGroup) {
 	file, err := os.Create(path)
 	if err != nil {
 		out.Error("Unable to create file '%s'", path)
@@ -51,7 +51,7 @@ func saveImage(img image.Image, path string, ext string, wg *sync.WaitGroup) {
 	case "png":
 		png.Encode(file, img)
 	default:
-		jpeg.Encode(file, img, nil)
+		jpeg.Encode(file, img, &jpeg.Options{Quality: quality})
 	}
 	wg.Done()
 }