move StoreSnapshot to separate method

machinery/src/computervision/main.go

@@ -130,26 +130,7 @@ func ProcessMotion(motionCursor *pubsub.QueueCursor, configuration *models.Confi
 
 				// Store snapshots (jpg) for hull.
 				if config.Capture.Snapshots != "false" {
-					files, err := ioutil.ReadDir("./data/snapshots")
-					if err == nil {
-						rgbImage, err := GetRawImage(frame, pkt, decoder, decoderMutex)
-						if err == nil {
-							sort.Slice(files, func(i, j int) bool {
-								return files[i].ModTime().Before(files[j].ModTime())
-							})
-							if len(files) > 3 {
-								os.Remove("./data/snapshots/" + files[0].Name())
-							}
-
-							// Save image
-							t := strconv.FormatInt(time.Now().Unix(), 10)
-							f, err := os.Create("./data/snapshots/" + t + ".jpg")
-							if err == nil {
-								jpeg.Encode(f, &rgbImage.Image, &jpeg.Options{Quality: 15})
-								f.Close()
-							}
-						}
-					}
+					StoreSnapshot(frame, pkt, decoder, decoderMutex)
 				}
 
 				// Check if within time interval
@@ -255,3 +236,26 @@ func AbsDiffBitwiseAndThreshold(img1 *image.Gray, img2 *image.Gray, img3 *image.
 	}
 	return changes
 }
+
+func StoreSnapshot(frame *ffmpeg.VideoFrame, pkt av.Packet, decoder *ffmpeg.VideoDecoder, decoderMutex *sync.Mutex) {
+	files, err := ioutil.ReadDir("./data/snapshots")
+	if err == nil {
+		rgbImage, err := GetRawImage(frame, pkt, decoder, decoderMutex)
+		if err == nil {
+			sort.Slice(files, func(i, j int) bool {
+				return files[i].ModTime().Before(files[j].ModTime())
+			})
+			if len(files) > 3 {
+				os.Remove("./data/snapshots/" + files[0].Name())
+			}
+
+			// Save image
+			t := strconv.FormatInt(time.Now().Unix(), 10)
+			f, err := os.Create("./data/snapshots/" + t + ".jpg")
+			if err == nil {
+				jpeg.Encode(f, &rgbImage.Image, &jpeg.Options{Quality: 15})
+				f.Close()
+			}
+		}
+	}
+}