---测试

panlei

2019-07-01 9ee1d7d6631fbf9c292d93624cfedd7683161926

---测试

 util/105.jpg

 util/simpleCV.go

@@ -1,9 +1,19 @@
package util

import (
    "bufio"
    "bytes"
    "encoding/json"
    "errors"
    "fmt"
    "image"
    "image/color"
    "io"
    "log"
    "mime/multipart"
    "net/http"
    "os"
    "time"

    "gocv.io/x/gocv"
)
@@ -149,34 +159,104 @@
    }
}

func DrawPolygon(){
func DrawPolygon()(maps map[string]interface{}, err0 error){
    // draw the rook
    rook := gocv.NewMatWithSize(w, w, gocv.MatTypeCV8UC3)
    //rook := gocv.NewMatWithSize(w, w, gocv.MatTypeCV8UC3)
    //imgs := gocv.NewMat()
    ddd, err := RetrieveROM("./l05.jpg")
    if err != nil {
        fmt.Println("解码有误")
    }
    bbb := bytes.NewBuffer(ddd)  // 必须加一个buffer 不然没有read方法就会报错
    rook, _ := gocv.NewMatFromBytes(500, 500, gocv.MatTypeCV8UC3, bbb.Bytes())
    defer rook.Close()
    green := color.RGBA{0, 255, 0, 0}
    points := [][]image.Point{
        {
            image.Pt(w/4., 7*w/8.),
            image.Pt(3*w/4., 7*w/8.),
            image.Pt(3*w/4., 13*w/16.),
            image.Pt(11*w/16., 13*w/16.),
            image.Pt(19*w/32., 3*w/8.),
            image.Pt(3*w/4., 3*w/8.),
            image.Pt(3*w/4., w/8.),
            image.Pt(26*w/40., w/8.),
            image.Pt(26*w/40., w/4.),
            image.Pt(22*w/40., w/4.),
            image.Pt(22*w/40., w/8.),
            image.Pt(18*w/40., w/8.),
            image.Pt(18*w/40., w/4.),
            image.Pt(14*w/40., w/4.),
            image.Pt(14*w/40., w/8.),
            image.Pt(w/4., w/8.),
            image.Pt(w/4., 3*w/8.),
            image.Pt(13*w/32., 3*w/8.),
            image.Pt(5*w/16., 13*w/16.),
            image.Pt(w/4., 13*w/16.),
            image.Pt(100., 100),
            image.Pt(100., 400),
            image.Pt(400, 400),
            image.Pt(400, 100),
        },
    }
    gocv.FillPoly(&rook, points, green)
    // 上传
    fdata,_ := gocv.IMEncode(".jpg",rook)
    body := &bytes.Buffer{}
    writer := multipart.NewWriter(body)
    _, err1 := writer.CreateFormFile("file", "fasjuierf")
    if err1 != nil {
        return nil, err1
    }
    boundary := writer.Boundary()
    //close_string := fmt.Sprintf("\r\n--%s--\r\n", boundary)
    close_buf := bytes.NewBufferString(fmt.Sprintf("\r\n--%s--\r\n", boundary))
    file := bytes.NewBuffer(fdata)
    request_reader := io.MultiReader(body, file, close_buf)
    //_, err = io.Copy(part, file)
    //writer.WriteField(key, val)
    request, err := http.NewRequest("POST", "http://192.168.1.182:6333/submit", request_reader)
    request.Header.Add("Content-Type", writer.FormDataContentType())
    timeout := time.Duration(5 * time.Second) //超时时间50ms
    client := &http.Client{Timeout: timeout}
    resp, err := client.Do(request)
    if err != nil {
        log.Fatal(err)
        return nil, err
    }
    defer func() {
        if r := recover(); r != nil {
            fmt.Printf("panic的内容%v\n", r)
            msg := "上传图片服务器异常"
            if _, ok := r.(error); ok {
                msg = r.(error).Error()
                fmt.Println("panic--recover()得到的是error类型")
            }
            if _, ok := r.(string); ok {
                msg = r.(string)
                fmt.Println("panic--recover()得到的是string类型")
            }
            err0 = errors.New(msg)
        }
    }()
    defer resp.Body.Close()
    {
        body := &bytes.Buffer{}
        _, err := body.ReadFrom(resp.Body)
        if err != nil {
            log.Fatal(err)
        }
        fmt.Println(resp.StatusCode)
        //fmt.Println(resp.Header)
        fmt.Println(body)
        //decoder := json.NewDecoder(strings.NewReader(body.String()))
        decoder := make(map[string]interface{})
        if err := json.Unmarshal([]byte(body.String()), &decoder); err != nil {
            return nil, err
        }
        return decoder, nil
    }
}

// 把图片转成二进制流
func RetrieveROM(filename string) ([]byte, error) {
    file, err := os.Open(filename)

    if err != nil {
        return nil, err
    }
    defer file.Close()

    stats, statsErr := file.Stat()
    if statsErr != nil {
        return nil, statsErr
    }

    var size int64 = stats.Size()
    bytes := make([]byte, size)

    bufr := bufio.NewReader(file)
    _, err = bufr.Read(bytes)

    return bytes, err
}

 util/simpleCV_test.go

@@ -4,6 +4,10 @@
    "testing"
)

func TestCvRTSP(t *testing.T)  {
    CvRTSP()
func TestDrawPolygon(t *testing.T)  {
    resp,err := DrawPolygon()
    if err != nil {
        t.Error("画框或者上传有问题")
    }
    t.Log(resp["fileUrl"].(string))
}