From c070c02f3fcb8c2c8dfa1ffcadf72a4fb827a43a Mon Sep 17 00:00:00 2001 From: liuxiaolong <736321739@qq.com> Date: 星期三, 20 十一月 2019 14:15:34 +0800 Subject: [PATCH] back --- extend/util/util.go | 103 +++++++++++++++++++++++++++++++++++++++++++++++---- 1 files changed, 95 insertions(+), 8 deletions(-) diff --git a/extend/util/util.go b/extend/util/util.go index 77cd755..542dd2c 100644 --- a/extend/util/util.go +++ b/extend/util/util.go @@ -1,13 +1,19 @@ package util import ( - "basic.com/pubsub/protomsg.git" + "bytes" "encoding/json" + "image" + "io/ioutil" + "net/http" + "strconv" + "webserver/extend/code" + + "basic.com/pubsub/protomsg.git" "github.com/gin-gonic/gin" "github.com/golang/glog" + "github.com/pierrec/lz4" "gocv.io/x/gocv" - "image" - "webserver/extend/code" "crypto/rand" "fmt" @@ -52,14 +58,14 @@ //struct杞琺ap func Struct2Map(obj interface{}) map[string]interface{} { - resultMap :=make(map[string]interface{},0) + resultMap := make(map[string]interface{}, 0) bytesData, err := json.Marshal(obj) - if err !=nil { + if err != nil { return resultMap } json.Unmarshal(bytesData, &resultMap) return resultMap - + // //t := reflect.TypeOf(obj) //v := reflect.ValueOf(obj) @@ -71,8 +77,21 @@ //return data } +func ReadImgData(url string) ([]byte, error) { + resp, err := http.Get(url) + if err != nil { + return nil, err + } + defer resp.Body.Close() + pix, err := ioutil.ReadAll(resp.Body) + if err != nil { + return nil, err + } + return pix, nil +} + // 鎸夊昂瀵稿幓鍒囧浘 -func SubImg(i protomsg.Image, x0, y0, x1, y1 int, ) []byte { +func SubImg(i protomsg.Image, x0, y0, x1, y1 int) []byte { img, _ := gocv.NewMatFromBytes(int(i.Height), int(i.Width), gocv.MatTypeCV8UC3, i.Data) rect := image.Rect(EnlargeSize(x0, y0, x1, y1, i)) region := img.Region(rect) @@ -112,4 +131,72 @@ y1_new = int(i.Height) } return -} \ No newline at end of file +} + +func StringIpToInt(ipstring string) int { + ipSegs := strings.Split(ipstring, ".") + var ipInt int = 0 + var pos uint = 24 + for _, ipSeg := range ipSegs { + tempInt, _ := strconv.Atoi(ipSeg) + tempInt = tempInt << pos + ipInt = ipInt | tempInt + pos -= 8 + } + return ipInt +} + +func IpIntToString(ipInt int) string { + ipSegs := make([]string, 4) + var len int = len(ipSegs) + buffer := bytes.NewBufferString("") + for i := 0; i < len; i++ { + tempInt := ipInt & 0xFF + ipSegs[len-i-1] = strconv.Itoa(tempInt) + + ipInt = ipInt >> 8 + } + for i := 0; i < len; i++ { + buffer.WriteString(ipSegs[i]) + if i < len-1 { + buffer.WriteString(".") + } + } + return buffer.String() +} + +func ParseScore(compareScore float32) float32 { + if compareScore < 1 { + compareScore = compareScore * 100 + } + f, _ := strconv.ParseFloat(fmt.Sprintf("%2.2f", compareScore), 32) + return float32(f) +} + +// UnCompress uncompress +func UnCompress(in []byte) ([]byte, error) { + out := make([]byte, 10*len(in)) + n, err := lz4.UncompressBlock(in, out) + if err != nil { + fmt.Println(err) + return nil, err + } + out = out[:n] // uncompressed data + return out, nil +} + +// Compress compress +func Compress(in []byte) ([]byte, error) { + out := make([]byte, len(in)) + ht := make([]int, 64<<10) // buffer for the compression table + n, err := lz4.CompressBlock(in, out, ht) + if err != nil { + fmt.Println(err) + return nil, err + } + if n >= len(in) { + fmt.Println("image is not compressible") + } + out = out[:n] // compressed data + return out, nil +} -- Gitblit v1.8.0