package/TrafficSpeed
1
0
Fork 0
mirror of https://github.com/jehiah/TrafficSpeed.git synced 2025-12-24 12:47:54 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

step 4: crop image

Browse Source
This commit is contained in:
Jehiah Czebotar
2017-05-06 13:51:10 -04:00
parent 74db535c00
commit 9b63aa1d66
2 changed files with 34 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/cmd/open_speed_data/points.go
View File

@@ -36,9 +36,16 @@ func (b *BBox) IsZero() bool {
}
return false
}
func (b *BBox) String() string {
func (b BBox) String() string {
return fmt.Sprintf("%s %s", b.A, b.B)
}
func (b BBox) Height() float64 {
return b.B.Y - b.A.Y
}
func (b BBox) Width() float64 {
return b.B.X - b.A.X
}
type Point struct {
X float64 `json:"x"`

37
src/cmd/open_speed_data/project.go
View File

@@ -54,9 +54,9 @@ type Response struct {
CroppedResolution string `json:"cropped_resolution,omitempty"`
OverviewGif template.URL `json:"overview_gif,omitempty"`
OverviewImg template.URL `json:"overview_img,omitempty"`
Step2Img template.URL `json:"step_2_img,omitempty"`
Step3Img template.URL `json:"step_3_img,omitempty"`
Step4Img template.URL `json:"step_4_img,omitempty"`
Step2Img template.URL `json:"step_2_img,omitempty"` // rotation
Step3Img template.URL `json:"step_3_img,omitempty"` // crop
Step4Img template.URL `json:"step_4_img,omitempty"` // mask
Step4MaskImg template.URL `json:"step_4_mask_img,omitempty"`
BackgroundImg template.URL `json:"background_img,omitempty"`
FrameAnalysis []FrameAnalysis `json:"frame_analysis,omitempty"`
@@ -140,19 +140,34 @@ func (p *Project) Run() error {
if p.Step == 2 {
p.Response.Step2Img, err = dataImg(&vf.Image)
}
if p.Step == 3 {
mw := imagick.NewMagickWand()
background := imagick.NewPixelWand()
background.SetColor("#000000")
// load image
out := new(bytes.Buffer)
png.Encode(out, &vf.Image)
mw.ReadImageBlob(out.Bytes())
if p.Step >= 3 {
// apply rotation
mw := imagick.NewMagickWand()
background := imagick.NewPixelWand()
background.SetColor("#000000")
out := new(bytes.Buffer)
png.Encode(out, &vf.Image)
mw.ReadImageBlob(out.Bytes())
mw.RotateImage(background, RadiansToDegrees(p.Rotate))
err = mw.RotateImage(background, RadiansToDegrees(p.Rotate))
if err != nil {
return err
}
mw.SetImageFormat("PNG")
imgBytes := mw.GetImageBlob()
p.Response.Step3Img = dataImgFromBytes(imgBytes)
}
if p.Step >= 4 {
// rotate & crop
err = mw.CropImage(uint(p.BBox.Width()), uint(p.BBox.Height()), int(p.BBox.A.X), int(p.BBox.A.Y))
if err != nil {
return err
}
imgBytes := mw.GetImageBlob()
p.Response.Step4Img = dataImgFromBytes(imgBytes)
}
}
p.Duration = pkt.Time