sander/go-opencv
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge https://github.com/hybridgroup/go-opencv into hybridgroup-master

Browse source
This commit is contained in:
Bert Chang 2014-05-02 20:19:36 +08:00
commit 82af0cc56b
16 changed files with 27182 additions and 988 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
opencv/cv.go
View file

@ -23,7 +23,6 @@ const (
CV_BLUR = C.CV_BLUR
)
/* Smoothes array (removes noise) */
func Smooth(src, dst *IplImage, smoothtype,
param1, param2 int, param3, param4 float64) {
@ -31,6 +30,7 @@ func Smooth(src, dst *IplImage, smoothtype,
C.int(param1), C.int(param2), C.double(param3), C.double(param4),
)
}
//CVAPI(void) cvSmooth( const CvArr* src, CvArr* dst,
// int smoothtype CV_DEFAULT(CV_GAUSSIAN),
// int param1 CV_DEFAULT(3),
@ -42,15 +42,9 @@ func Smooth(src, dst *IplImage, smoothtype,
func CvtColor(src, dst *IplImage, code int) {
C.cvCvtColor(unsafe.Pointer(src), unsafe.Pointer(dst), C.int(code))
}
//CVAPI(void) cvCvtColor( const CvArr* src, CvArr* dst, int code );
/* Runs canny edge detector */
func Canny(image, edges *IplImage, threshold1, threshold2 float64, aperture_size int) {
C.cvCanny(unsafe.Pointer(image), unsafe.Pointer(edges),
@ -58,10 +52,10 @@ func Canny(image, edges *IplImage, threshold1, threshold2 float64, aperture_size
C.int(aperture_size),
)
}
//CVAPI(void) cvCanny( const CvArr* image, CvArr* edges, double threshold1,
// double threshold2, int aperture_size CV_DEFAULT(3) );
const (
CV_INPAINT_NS = C.CV_INPAINT_NS
CV_INPAINT_TELEA = C.CV_INPAINT_TELEA
@ -77,8 +71,6 @@ func Inpaint(src, inpaint_mask, dst *IplImage, inpaintRange float64, flags int)
C.int(flags),
)
}
//CVAPI(void) cvInpaint( const CvArr* src, const CvArr* inpaint_mask,
// CvArr* dst, double inpaintRange, int flags );

44
opencv/cvaux.go
View file

@ -9,87 +9,85 @@ package opencv
//#cgo darwin pkg-config: opencv
//#cgo windows LDFLAGS: -lopencv_core242.dll -lopencv_imgproc242.dll -lopencv_photo242.dll -lopencv_highgui242.dll -lstdc++
import "C"
import (
"unsafe"
)
/****************************************************************************************\
* Eigen objects *
\****************************************************************************************/
/****************************************************************************************\
* 1D/2D HMM *
\****************************************************************************************/
/*********************************** Embedded HMMs *************************************/
/****************************************************************************************\
* A few functions from old stereo gesture recognition demosions *
\****************************************************************************************/
/****************************************************************************************\
* Additional operations on Subdivisions *
\****************************************************************************************/
/****************************************************************************************\
* More operations on sequences *
\****************************************************************************************/
/*******************************Stereo correspondence*************************************/
/*****************************************************************************************/
/************ Epiline functions *******************/
/****************************************************************************************\
* Contour Morphing *
\****************************************************************************************/
/****************************************************************************************\
* Texture Descriptors *
\****************************************************************************************/
/****************************************************************************************\
* Face eyes&mouth tracking *
\****************************************************************************************/
type HaarCascade struct {
cascade *C.CvHaarClassifierCascade
}
func LoadHaarClassifierCascade(haar string) *HaarCascade {
haarCascade := new(HaarCascade)
haarCascade.cascade = C.cvLoadHaarClassifierCascade(C.CString(haar), C.cvSize(1, 1))
return haarCascade
}
func (this *HaarCascade) DetectObjects(image *IplImage) []*Rect {
storage := C.cvCreateMemStorage(0)
seq := C.cvHaarDetectObjects(unsafe.Pointer(image), this.cascade, storage, 1.1, 3, C.CV_HAAR_DO_CANNY_PRUNING, C.cvSize(0, 0), C.cvSize(0, 0))
var faces []*Rect
for i := 0; i < (int)(seq.total); i++ {
rect := (*Rect)((*_Ctype_CvRect)(unsafe.Pointer(C.cvGetSeqElem(seq, C.int(i)))))
faces = append(faces, rect)
}
return faces
}
/****************************************************************************************\
* 3D Tracker *
\****************************************************************************************/
/****************************************************************************************\
* Skeletons and Linear-Contour Models *
\****************************************************************************************/
/****************************************************************************************\
* Background/foreground segmentation *
\****************************************************************************************/
/****************************************************************************************\
* Calibration engine *
\****************************************************************************************/
/*****************************************************************************\
* --- END --- *
\*****************************************************************************/

57
opencv/cxcore.go
View file

@ -72,7 +72,7 @@ func (img *IplImage)Release() {
/* Creates a copy of IPL image (widthStep may differ) */
func (img *IplImage) Clone() *IplImage {
p := C.cvCloneImage((*C.IplImage)(img))
return (*IplImage)(p);
return (*IplImage)(p)
}
/* Sets a Channel Of Interest (only a few functions support COI) -
@ -80,6 +80,7 @@ func (img *IplImage)Clone() *IplImage {
func (img *IplImage) SetCOI(coi int) {
C.cvSetImageCOI((*C.IplImage)(img), C.int(coi))
}
/* Retrieves image Channel Of Interest */
func (img *IplImage) GetCOI() int {
coi := C.cvGetImageCOI((*C.IplImage)(img))
@ -90,10 +91,12 @@ func (img *IplImage)GetCOI() int {
func (img *IplImage) SetROI(rect Rect) {
C.cvSetImageROI((*C.IplImage)(img), C.CvRect(rect))
}
/* Resets image ROI and COI */
func (img *IplImage) ResetROI() {
C.cvResetImageROI((*C.IplImage)(img))
}
/* Retrieves image ROI */
func (img *IplImage) GetROI() Rect {
r := C.cvGetImageROI((*C.IplImage)(img))
@ -112,6 +115,7 @@ func CreateMatHeader(rows, cols, type_ int) *Mat {
)
return (*Mat)(mat)
}
/* Allocates and initializes CvMat header and allocates data */
func CreateMat(rows, cols, type_ int) *Mat {
mat := C.cvCreateMat(
@ -144,6 +148,7 @@ func (mat *Mat)Release() {
func DecRefData(arr Arr) {
C.cvDecRefData(unsafe.Pointer(arr))
}
/* Increments CvMat data reference counter */
func IncRefData(arr Arr) {
C.cvIncRefData(unsafe.Pointer(arr))
@ -165,6 +170,7 @@ func GetSubRect(arr Arr, submat *Mat, rect Rect) *Mat {
)
return (*Mat)(mat_new)
}
//#define cvGetSubArr cvGetSubRect
/* Selects row span of the input array: arr(start_row:delta_row:end_row,:)
@ -249,7 +255,7 @@ func CreateMatNDHeader(sizes []int, type_ int) *MatND {
mat := C.cvCreateMatNDHeader(
dims, (*C.int)(&sizes_c[0]), C.int(type_),
)
return (*MatND)(mat);
return (*MatND)(mat)
}
/* Allocates and initializes CvMatND header and allocates data */
@ -263,7 +269,7 @@ func CreateMatND(sizes []int, type_ int) *MatND {
mat := C.cvCreateMatND(
dims, (*C.int)(&sizes_c[0]), C.int(type_),
)
return (*MatND)(mat);
return (*MatND)(mat)
}
/* Initializes preallocated CvMatND header */
@ -305,7 +311,7 @@ func CreateSparseMat(sizes []int, type_ int) *SparseMat {
mat := C.cvCreateSparseMat(
dims, (*C.int)(&sizes_c[0]), C.int(type_),
)
return (*SparseMat)(mat);
return (*SparseMat)(mat)
}
/* Releases CvSparseMat */
@ -360,6 +366,7 @@ func GetSize(img *IplImage) Size {
func Copy(src, dst, mask *IplImage) {
C.cvCopy(unsafe.Pointer(src), unsafe.Pointer(dst), unsafe.Pointer(mask))
}
//CVAPI(void) cvCopy( const CvArr* src, CvArr* dst,
// const CvArr* mask CV_DEFAULT(NULL) );
@ -367,51 +374,41 @@ func Copy(src, dst, mask *IplImage) {
func Zero(img *IplImage) {
C.cvSetZero(unsafe.Pointer(img))
}
//CVAPI(void) cvSetZero( CvArr* arr );
//#define cvZero cvSetZero
/****************************************************************************************\
* Arithmetic, logic and comparison operations *
\****************************************************************************************/
/* dst(idx) = ~src(idx) */
func Not(src, dst *IplImage) {
C.cvNot(unsafe.Pointer(src), unsafe.Pointer(dst))
}
//CVAPI(void) cvNot( const CvArr* src, CvArr* dst );
/****************************************************************************************\
* Math operations *
\****************************************************************************************/
/****************************************************************************************\
* Matrix operations *
\****************************************************************************************/
/****************************************************************************************\
* Array Statistics *
\****************************************************************************************/
/****************************************************************************************\
* Discrete Linear Transforms and Related Functions *
\****************************************************************************************/
/****************************************************************************************\
* Dynamic data structures *
\****************************************************************************************/
/****************************************************************************************\
* Drawing *
\****************************************************************************************/
@ -426,26 +423,34 @@ func Line(image *IplImage, pt1, pt2 Point, color Scalar, thickness, line_type, s
(C.CvScalar)(color),
C.int(thickness), C.int(line_type), C.int(shift),
)
//Scalar
}
func Rectangle(image *IplImage, pt1, pt2 Point, color Scalar, thickness, line_type, shift int) {
C.cvRectangle(
unsafe.Pointer(image),
C.cvPoint(C.int(pt1.X), C.int(pt1.Y)),
C.cvPoint(C.int(pt2.X), C.int(pt2.Y)),
(C.CvScalar)(color),
C.int(thickness), C.int(line_type), C.int(shift),
)
}
func Circle(image *IplImage, pt1 Point, radius int, color Scalar, thickness, line_type, shift int) {
C.cvCircle(
unsafe.Pointer(image),
C.cvPoint(C.int(pt1.X), C.int(pt1.Y)),
C.int(radius),
(C.CvScalar)(color),
C.int(thickness), C.int(line_type), C.int(shift),
)
}
//CVAPI(void) cvLine( CvArr* img, CvPoint pt1, CvPoint pt2,
// CvScalar color, int thickness CV_DEFAULT(1),
// int line_type CV_DEFAULT(8), int shift CV_DEFAULT(0) );
/****************************************************************************************\
* System functions *
\****************************************************************************************/
/****************************************************************************************\
* Data Persistence *
\****************************************************************************************/

1
opencv/cxcore_test.go
View file

@ -7,4 +7,3 @@ import (
func TestLoadImage2(t *testing.T) {
// t.Errorf("aaa")
}

8
opencv/cxtype.go
View file

@ -59,7 +59,6 @@ import (
//-----------------------------------------------------------------------------
// cvver.h
//-----------------------------------------------------------------------------
@ -561,7 +560,6 @@ type Point3D32f struct {
Z float32
}
type Point2D64f struct {
X float64
Y float64
@ -604,6 +602,11 @@ const (
type Scalar C.CvScalar
func NewScalar(b, g, r int) Scalar {
rv := C.cvScalar(C.double(b), C.double(g), C.double(r), C.double(0))
return (Scalar)(rv)
}
func ScalarAll(val0 float64) Scalar {
rv := C.cvScalarAll(C.double(val0))
return (Scalar)(rv)
@ -677,4 +680,3 @@ type AttrList C.CvAttrList
/*****************************************************************************\
* --- END --- *
\*****************************************************************************/

1
opencv/doc.go
View file

@ -4,4 +4,3 @@
// Bindings for Intel's OpenCV computer vision library.
package opencv

34
opencv/highgui.go
View file

@ -10,9 +10,9 @@ package opencv
//#cgo windows LDFLAGS: -lopencv_core242.dll -lopencv_imgproc242.dll -lopencv_photo242.dll -lopencv_highgui242.dll -lstdc++
import "C"
import (
_ "fmt"
"runtime"
"unsafe"
_ "fmt"
)
func init() {
@ -44,7 +44,6 @@ func WaitKey(delay int) int {
return int(key)
}
//-----------------------------------------------------------------------------
// Window wrapper for go
//-----------------------------------------------------------------------------
@ -121,6 +120,7 @@ const (
CV_WINDOW_NORMAL = int(C.CV_WINDOW_NORMAL)
CV_WINDOW_FULLSCREEN = int(C.CV_WINDOW_FULLSCREEN)
)
/* Set and Get Property of the window */
func (win *Window) SetProperty(prop_id int, value float64) {
C.cvSetWindowProperty(win.name_c, C.int(prop_id), C.double(value))
@ -155,7 +155,6 @@ func (win *Window)GetWindowName() string {
return win.name
}
//-----------------------------------------------------------------------------
// window: track bar
//-----------------------------------------------------------------------------
@ -196,7 +195,7 @@ func (win *Window)CreateTrackbar(name string,
return bool(rv != 0)
}
func destroyTrackbar(barName_, winName_ *C.char) {
C.GoOpenCV_DestroyTrackbar(barName_, winName_);
C.GoOpenCV_DestroyTrackbar(barName_, winName_)
}
//export goTrackbarCallback
@ -208,11 +207,17 @@ func goTrackbarCallback(barName_, winName_ *C.char, pos C.int) {
barName := C.GoString(barName_)
win, ok := allWindows[winName]
if !ok { return }
if !ok {
return
}
trackbarHandle, ok := win.trackbarHandle[barName]
if !ok { return }
if trackbarHandle == nil { return }
if !ok {
return
}
if trackbarHandle == nil {
return
}
if fa, ok := trackbarHandle.(func(pos int)); ok {
fa(int(pos))
@ -235,7 +240,6 @@ func (win *Window)SetTrackbarPos(name string, pos int) {
C.cvSetTrackbarPos(win.trackbarName[name], win.name_c, C.int(pos))
}
//-----------------------------------------------------------------------------
// window: mouse callback
//-----------------------------------------------------------------------------
@ -282,6 +286,7 @@ func (win *Window)SetMouseCallback(on_mouse MouseFunc, param ...interface{}) {
win.param = nil
}
}
//export goMouseCallback
func goMouseCallback(name *C.char, event, x, y, flags C.int) {
runtime.LockOSThread()
@ -290,8 +295,12 @@ func goMouseCallback(name *C.char, event, x, y, flags C.int) {
winName := C.GoString(name)
win, ok := allWindows[winName]
if !ok { return }
if win.mouseHandle == nil { return }
if !ok {
return
}
if win.mouseHandle == nil {
return
}
if fa, ok := win.mouseHandle.(func(event, x, y, flags int)); ok {
fa(int(event), int(x), int(y), int(flags))
@ -308,7 +317,6 @@ func goMouseCallback(name *C.char, event, x, y, flags C.int) {
}
}
//-----------------------------------------------------------------------------
// window: destroy
//-----------------------------------------------------------------------------
@ -349,6 +357,7 @@ const (
/* ?, any color */
CV_LOAD_IMAGE_ANYCOLOR = int(C.CV_LOAD_IMAGE_ANYCOLOR)
)
/* load image from file
iscolor can be a combination of above flags where CV_LOAD_IMAGE_UNCHANGED
overrides the other flags
@ -555,6 +564,7 @@ func FOURCC(c1, c2, c3, c4 int8) uint32 {
rv := C.GoOpenCV_FOURCC_(C.int(c1), C.int(c2), C.int(c3), C.int(c4))
return uint32(rv)
}
const (
/* Open Codec Selection Dialog (Windows only) */
CV_FOURCC_PROMPT = int(C.CV_FOURCC_PROMPT)
@ -593,5 +603,3 @@ func (writer *VideoWriter)Release() {
/*****************************************************************************\
* --- END --- *
\*****************************************************************************/

1
opencv/opencv.go
View file

@ -11,4 +11,3 @@ import(
//-----------------------------------------------------------------------------
// End
//-----------------------------------------------------------------------------

5
opencv/opencv.h
View file

@ -12,6 +12,11 @@
#include <opencv/highgui.h>
#include <opencv2/photo/photo_c.h>
#include <opencv2/imgproc/imgproc_c.h>
#include <opencv2/core/core_c.h>
#include <opencv2/core/core.hpp>
#include <opencv2/objdetect/objdetect.hpp>
#include <opencv2/legacy/compat.hpp>
#include <opencv2/legacy/legacy.hpp>
// Trackbar
int GoOpenCV_CreateTrackbar(

4
samples/edge.go
View file

@ -10,8 +10,8 @@ import (
"path"
"runtime"
//"github.com/lazywei/go-opencv/opencv"
"../opencv" // can be used in forks, comment in real application
"github.com/hybridgroup/go-opencv/opencv"
//"../opencv" // can be used in forks, comment in real application
)
func main() {

26
samples/face_detect.go Normal file
View file

@ -0,0 +1,26 @@
package main
import (
"github.com/hybridgroup/go-opencv/opencv"
"path"
"runtime"
)
func main() {
_, currentfile, _, _ := runtime.Caller(0)
image := opencv.LoadImage(path.Join(path.Dir(currentfile), "../images/lena.jpg"))
cascade := opencv.LoadHaarClassifierCascade(path.Join(path.Dir(currentfile), "haarcascade_frontalface_alt.xml"))
faces := cascade.DetectObjects(image)
for _, value := range faces {
opencv.Rectangle(image,
opencv.Point{value.X() + value.Width(), value.Y()},
opencv.Point{value.X(), value.Y() + value.Height()},
opencv.ScalarAll(255.0), 1, 1, 0)
}
win := opencv.NewWindow("Face Detection")
win.ShowImage(image)
opencv.WaitKey(0)
}

26161
samples/haarcascade_frontalface_alt.xml Normal file
View file

File diff suppressed because it is too large Load diff

4
samples/hellocv.go
View file

@ -10,8 +10,8 @@ import (
"path"
"runtime"
//"github.com/lazywei/go-opencv/opencv"
"../opencv" // can be used in forks, comment in real application
"github.com/hybridgroup/go-opencv/opencv"
//"../opencv" // can be used in forks, comment in real application
)
func main() {

4
samples/inpaint.go
View file

@ -10,8 +10,8 @@ import (
"path"
"runtime"
//"github.com/lazywei/go-opencv/opencv"
"../opencv" // can be used in forks, comment in real application
"github.com/hybridgroup/go-opencv/opencv"
//"../opencv" // can be used in forks, comment in real application
)
func main() {

4
samples/player.go
View file

@ -8,8 +8,8 @@ import (
"fmt"
"os"
//"github.com/lazywei/go-opencv/opencv"
"../opencv" // can be used in forks, comment in real application
"github.com/hybridgroup/go-opencv/opencv"
//"../opencv" // can be used in forks, comment in real application
)
func main() {

4
samples/webcam.go
View file

@ -6,8 +6,8 @@ import (
//"path"
//"runtime"
//"github.com/lazywei/go-opencv/opencv"
"../opencv" // can be used in forks, comment in real application
"github.com/hybridgroup/go-opencv/opencv"
//"../opencv" // can be used in forks, comment in real application
)
func main() {