/
multi.go
53 lines (46 loc) · 1.14 KB
/
multi.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
package encmulti
import (
"image"
"image/gif"
"image/jpeg"
"image/png"
"io"
"strings"
"golang.org/x/image/bmp"
"golang.org/x/image/tiff"
"github.com/srlehn/termimg/internal"
"github.com/srlehn/termimg/internal/errors"
)
var _ internal.ImageEncoder = (*MultiEncoder)(nil)
type MultiEncoder struct{}
func (e *MultiEncoder) Encode(w io.Writer, img image.Image, fileExt string) error {
if img == nil {
return errors.NilParam()
}
// allow passing whole filename
fileExtParts := strings.Split(fileExt, `.`)
fileExt = fileExtParts[len(fileExtParts)-1]
fmtStr := strings.ToLower(strings.TrimPrefix(fileExt, `.`))
if len(fmtStr) == 0 {
return errors.New(`no file format specified`)
}
var err error
switch fmtStr {
case `bmp`:
err = bmp.Encode(w, img)
case `gif`:
err = gif.Encode(w, img, nil)
case `png`:
err = png.Encode(w, img)
case `tiff`:
err = tiff.Encode(w, img, &tiff.Options{Compression: tiff.LZW, Predictor: true})
case `jpg`, `jpeg`:
err = jpeg.Encode(w, img, &jpeg.Options{Quality: 90})
default:
err = errors.New(`unsupported file format: "` + fmtStr + `"`)
}
if err != nil {
return errors.New(err)
}
return nil
}