annotate src/IMGmanip.go @ 1:3dafd57af3b1

bug-fix.
author pyon@macmini
date Thu, 06 Oct 2016 00:55:26 +0900
parents de451fa0c9cd
children 9072560fc0f2
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
0
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
1 /* Image Manipulator - ver 0.2 ( Last Change: 2016-08-21 Sun 09:51:52.) */
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
2 package main
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
3
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
4 import(
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
5 "errors"
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
6 "flag"
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
7 "fmt"
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
8 "image"
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
9 "image/jpeg"
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
10 "image/png"
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
11 "io"
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
12 "os"
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
13 "strings"
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
14 "strconv"
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
15 )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
16
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
17 func main() {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
18 ver := "0.2"
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
19
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
20 // arguments
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
21 kind := flag.String( "t", "", "file type : jpeg / png" )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
22 resize := flag.String( "s", "", "new size : 600x800 / 320x0 / ..." )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
23 crop := flag.String( "c", "", "crop size : 300x200+10+35" )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
24 angle := flag.Uint( "r", 0, "rotate : 90 / 180 / 270" )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
25
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
26 flag.Parse()
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
27 if flag.NFlag() == 0 {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
28 fmt.Fprintf( os.Stderr, "\n=== Image-Manipulator - version %s ===", ver )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
29 fmt.Fprintf( os.Stderr, "\n$ imgmanip [option parameter] < infile > outfile\n" )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
30 flag.PrintDefaults()
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
31 os.Exit( 1 )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
32 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
33
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
34 // convert
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
35 if *kind != "" {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
36 err := Convert( os.Stdin, os.Stdout, *kind ); if err != nil {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
37 fmt.Fprintf( os.Stderr, "%s: %v\n", *kind, err )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
38 os.Exit( 1 )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
39 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
40 return
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
41 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
42
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
43 // resize
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
44 if *resize != "" {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
45 err := Resize( os.Stdin, os.Stdout, *resize ); if err != nil {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
46 fmt.Fprintf( os.Stderr, "resize: %v\n", err )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
47 os.Exit( 1 )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
48 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
49 return
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
50 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
51
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
52 // crop
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
53 if *crop != "" {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
54 err := Crop( os.Stdin, os.Stdout, *crop ); if err != nil {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
55 fmt.Fprintf( os.Stderr, "crop: %v\n", err )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
56 os.Exit( 1 )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
57 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
58 return
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
59 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
60
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
61 // rotate
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
62 if *angle != 0 {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
63 err := Rotate( os.Stdin, os.Stdout, *angle ); if err != nil {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
64 fmt.Fprintf( os.Stderr, "rotate: %v\n", err )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
65 os.Exit( 1 )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
66 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
67 return
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
68 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
69 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
70
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
71 func Convert( in io.Reader, out io.Writer, kind string ) error {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
72
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
73 img, _, err := image.Decode( in ); if err != nil {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
74 return err
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
75 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
76
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
77 switch kind {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
78 case "jpeg":
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
79 return jpeg.Encode( out, img, &jpeg.Options{ Quality: 100 } )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
80 case "png":
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
81 return png.Encode( out, img )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
82 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
83
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
84 return errors.New( "image-type is not supported")
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
85 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
86
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
87 func Resize( in io.Reader, out io.Writer, size string ) error {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
88
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
89 img, kind, err := image.Decode( in ); if err != nil {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
90 return err
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
91 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
92 srcW := img.Bounds().Max.X
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
93 srcH := img.Bounds().Max.Y
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
94
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
95 wh := strings.Split( size, "x" )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
96 w, err := strconv.Atoi( wh[0] ); if err != nil || w < 0 {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
97 return errors.New( "invalid width" )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
98 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
99 h, err := strconv.Atoi( wh[1] ); if err != nil || h < 0 {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
100 return errors.New( "invalid height" )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
101 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
102 if w == 0 && h == 0 {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
103 return errors.New( "invalid size" )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
104 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
105 if w == 0 {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
106 w = h * srcW / srcH
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
107 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
108 if h == 0 {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
109 h = w * srcH / srcW
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
110 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
111
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
112 dst := image.NewNRGBA( image.Rect( 0, 0, w, h ) )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
113 for x := 0; x < w; x++ {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
114 for y := 0; y < h; y++ {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
115 dst.Set( x, y, img.At( x * srcW / w, y * srcH / h ) )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
116 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
117 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
118
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
119 switch kind {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
120 case "jpeg":
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
121 return jpeg.Encode( out, dst, &jpeg.Options{ Quality: 100 } )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
122 case "png":
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
123 return png.Encode( out, dst )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
124 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
125
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
126 return errors.New( "image-type is not supported")
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
127 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
128
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
129 func Crop( in io.Reader, out io.Writer, geo string ) error {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
130
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
131 img, kind, err := image.Decode( in ); if err != nil {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
132 return err
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
133 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
134 srcW := img.Bounds().Max.X
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
135 srcH := img.Bounds().Max.Y
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
136
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
137 buf := strings.Split( geo, "x" )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
138 w, err := strconv.Atoi( buf[0] ); if err != nil || w < 0 || w > srcW {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
139 return errors.New( "invalid width" )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
140 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
141 buf = strings.Split( buf[1], "+" )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
142 h, err := strconv.Atoi( buf[0] ); if err != nil || h < 0 || h > srcH {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
143 return errors.New( "invalid height" )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
144 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
145 x0, err := strconv.Atoi( buf[1] ); if err != nil || x0 < 0 {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
146 return errors.New( "invalid x" )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
147 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
148 y0, err := strconv.Atoi( buf[2] ); if err != nil || y0 < 0 {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
149 return errors.New( "invalid y" )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
150 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
151
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
152 var dst *image.NRGBA
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
153
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
154 dst = image.NewNRGBA( image.Rect( 0, 0, w, h ) )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
155 for x := 0; x < x0 + w; x++ {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
156 for y := 0; y < y0 + h; y++ {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
157 dst.Set( x, y, img.At( x0 + x, y0 + y ) )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
158 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
159 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
160
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
161 switch kind {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
162 case "jpeg":
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
163 return jpeg.Encode( out, dst, &jpeg.Options{ Quality: 100 } )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
164 case "png":
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
165 return png.Encode( out, dst )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
166 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
167
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
168 return errors.New( "image-type is not supported")
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
169 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
170
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
171 func Rotate( in io.Reader, out io.Writer, angle uint ) error {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
172
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
173 img, kind, err := image.Decode( in ); if err != nil {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
174 return err
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
175 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
176 srcW := img.Bounds().Max.X
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
177 srcH := img.Bounds().Max.Y
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
178
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
179 var dst *image.NRGBA
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
180 switch angle {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
181 case 90:
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
182 dst = image.NewNRGBA( image.Rect( 0, 0, srcH, srcW ) )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
183 for x := 0; x < srcH; x++ {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
184 for y := 0; y < srcW; y++ {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
185 dst.Set( x, y, img.At( y, srcH - x ) )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
186 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
187 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
188 case 180:
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
189 dst = image.NewNRGBA( image.Rect( 0, 0, srcW, srcH ) )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
190 for x := 0; x < srcW; x++ {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
191 for y := 0; y < srcH; y++ {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
192 dst.Set( x, y, img.At( srcW - x, srcH - y ) )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
193 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
194 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
195 case 270:
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
196 dst = image.NewNRGBA( image.Rect( 0, 0, srcH, srcW ) )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
197 for x := 0; x < srcH; x++ {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
198 for y := 0; y < srcW; y++ {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
199 dst.Set( x, y, img.At( y, x ) )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
200 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
201 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
202 default:
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
203 return errors.New( "bad angle" )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
204 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
205
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
206 switch kind {
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
207 case "jpeg":
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
208 return jpeg.Encode( out, dst, &jpeg.Options{ Quality: 100 } )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
209 case "png":
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
210 return png.Encode( out, dst )
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
211 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
212
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
213 return errors.New( "image-type is not supported")
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
214 }
de451fa0c9cd golang repository.
pyon@macmini
parents:
diff changeset
215