add bgr and yuv420 to jpeg support

zhangmeng

2023-12-26 0ead91b8788b382d0d1e7f6eda3f1d008ce39602

 capture.go

@@ -5,9 +5,8 @@
   "fmt"
   "time"

   "basic.com/valib/goffmpeg.git"

   "basic.com/valib/godraw.git"
   "basic.com/valib/goffmpeg.git"
   "github.com/disintegration/imaging"
)

@@ -50,7 +49,7 @@
      return nil, err
   }

   reterr := errors.New(fmt.Sprintf("gb28181 try %d times to capture image, is url correct?", maxTry))
   reterr := fmt.Errorf("gb28181 try %d times to capture image, is url correct?", maxTry)
   // for i := 0; i < maxTry; i++ {
   ret = goffmpeg.GetGBJpg(url, maxTry)
   if len(ret) > 0 {
@@ -106,9 +105,8 @@
   gf.BuildDecoder()

   tryTime := 0
   interval := 40 // ms
   // 保证有一个关键帧
   maxTry *= 1000
   maxTry = maxTry * 1000

   var bgrData, jpgData []byte
   var err error
@@ -117,8 +115,12 @@
      data, ow, oh, _ := gf.GetYUV()

      if ow > 0 && oh > 0 {

         bgrData = yuv2bgr(data, ow, oh)
         if ow*oh*3 != len(data) {
            // data 不是 bgr 格式，需要转换
            bgrData = yuv2bgr(data, ow, oh)
         } else {
            bgrData = data
         }
         if w > 0 && h > 0 {
            bgrData = bgresize(bgrData, ow, oh, w, h)
            ow, oh = w, h
@@ -130,11 +132,11 @@
         }
         break
      } else {
         tryTime += interval
         tryTime += 40
         if tryTime > maxTry {
            break
         }
         time.Sleep(time.Duration(interval) * time.Millisecond)
         time.Sleep(40 * time.Millisecond)
      }
   }