85 lines
2.5 KiB
Go
85 lines
2.5 KiB
Go
// Copyright 2013 <me@cwchang.me>. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package opencv
|
|
|
|
//#include "opencv.h"
|
|
//#cgo linux pkg-config: opencv
|
|
//#cgo darwin pkg-config: opencv
|
|
//#cgo freebsd pkg-config: opencv
|
|
//#cgo windows LDFLAGS: -lopencv_core242.dll -lopencv_imgproc242.dll -lopencv_photo242.dll -lopencv_highgui242.dll -lstdc++
|
|
import "C"
|
|
import (
|
|
//"errors"
|
|
//"log"
|
|
"unsafe"
|
|
)
|
|
|
|
const (
|
|
CV_INTER_NN = int(C.CV_INTER_NN)
|
|
CV_INTER_LINEAR = int(C.CV_INTER_LINEAR)
|
|
CV_INTER_CUBIC = int(C.CV_INTER_CUBIC)
|
|
CV_INTER_AREA = int(C.CV_INTER_AREA)
|
|
CV_INTER_LANCZOS4 = int(C.CV_INTER_LANCZOS4)
|
|
)
|
|
|
|
func Resize(src *IplImage, width, height, interpolation int) *IplImage {
|
|
if width == 0 && height == 0 {
|
|
panic("Width and Height cannot be 0 at the same time")
|
|
}
|
|
if width == 0 {
|
|
ratio := float64(height) / float64(src.Height())
|
|
width = int(float64(src.Width()) * ratio)
|
|
} else if height == 0 {
|
|
ratio := float64(width) / float64(src.Width())
|
|
height = int(float64(src.Height()) * ratio)
|
|
}
|
|
|
|
dst := CreateImage(width, height, src.Depth(), src.Channels())
|
|
C.cvResize(unsafe.Pointer(src), unsafe.Pointer(dst), C.int(interpolation))
|
|
return dst
|
|
}
|
|
|
|
func Crop(src *IplImage, x, y, width, height int) *IplImage {
|
|
r := C.cvRect(C.int(x), C.int(y), C.int(width), C.int(height))
|
|
rect := Rect(r)
|
|
|
|
src.SetROI(rect)
|
|
dest := CreateImage(width, height, src.Depth(), src.Channels())
|
|
Copy(src, dest, nil)
|
|
src.ResetROI()
|
|
|
|
return dest
|
|
}
|
|
|
|
/* Returns a Seq of countours in an image, detected according to the parameters.
|
|
Caller must Release() the Seq returned */
|
|
func (image *IplImage) FindContours(mode, method int, offset Point) *Seq {
|
|
storage := C.cvCreateMemStorage(0)
|
|
header_size := (C.size_t)(unsafe.Sizeof(C.CvContour{}))
|
|
var seq *C.CvSeq
|
|
C.cvFindContours(
|
|
unsafe.Pointer(image),
|
|
storage,
|
|
&seq,
|
|
C.int(header_size),
|
|
C.int(mode),
|
|
C.int(method),
|
|
C.cvPoint(C.int(offset.X), C.int(offset.Y)))
|
|
|
|
return (*Seq)(seq)
|
|
}
|
|
|
|
//cvDrawContours(CvArr* img, CvSeq* contour, CvScalar externalColor, CvScalar holeColor, int maxLevel, int thickness=1, int lineType=8
|
|
func DrawContours(image *IplImage, contours *Seq, externalColor, holeColor Scalar, maxLevel, thickness, lineType int, offset Point) {
|
|
C.cvDrawContours(
|
|
unsafe.Pointer(image),
|
|
(*C.CvSeq)(contours),
|
|
(C.CvScalar)(externalColor),
|
|
(C.CvScalar)(holeColor),
|
|
C.int(maxLevel),
|
|
C.int(thickness),
|
|
C.int(lineType),
|
|
C.cvPoint(C.int(offset.X), C.int(offset.Y)))
|
|
}
|