mirror of https://bitbucket.org/ausocean/av.git
Add 3 different options: no motion, edge detect, motion detect
This commit is contained in:
richardsonjack
parent
bb2a593dc1
commit
e9fb26d624
168
revid/revid.go
168
revid/revid.go
|
|
@ -36,9 +36,11 @@ import (
|
|||
"encoding/hex"
|
||||
"flag"
|
||||
"fmt"
|
||||
"image"
|
||||
"io"
|
||||
"io/ioutil"
|
||||
"log"
|
||||
"math"
|
||||
"math/rand"
|
||||
"net"
|
||||
"net/http"
|
||||
|
|
@ -58,21 +60,22 @@ import (
|
|||
|
||||
// defaults and networking consts
|
||||
const (
|
||||
clipDuration = 1 // s
|
||||
defaultPID = 256
|
||||
defaultFrameRate = 25
|
||||
defaultHTTPOutput = "http://localhost:8080?"
|
||||
defaultUDPOutput = "udp://0.0.0.0:16384"
|
||||
defaultRTPOutput = "rtp://0.0.0.0:16384"
|
||||
mp2tPacketSize = 188 // MPEG-TS packet size
|
||||
mp2tMaxPackets = 2016 // # first multiple of 7 and 8 greater than 2000
|
||||
mp2tMaxPackets = 2016*clipDuration // # first multiple of 7 and 8 greater than 2000
|
||||
udpPackets = 7 // # of UDP packets per ethernet frame (8 is the max)
|
||||
rtpPackets = 7 // # of RTP packets per ethernet frame (7 is the max)
|
||||
rtpHeaderSize = 12
|
||||
rtpSSRC = 1 // any value will do
|
||||
bufferSize = 1000
|
||||
bufferSize = 1000/clipDuration
|
||||
bitrateOutputDelay = 60 // s
|
||||
httpTimeOut = 5 // s
|
||||
clipDuration = 1 // s
|
||||
motionThreshold = 0
|
||||
)
|
||||
|
||||
// flag values
|
||||
|
|
@ -82,6 +85,8 @@ const (
|
|||
filterScale640 = 0x0004
|
||||
filterScale320 = 0x0008
|
||||
filterFixContinuity = 0x0010
|
||||
filterEdgeDetection = 0x0020
|
||||
filterMotionDetect = 0x0040
|
||||
dumpProgramInfo = 0x0100 // 256
|
||||
dumpPacketStats = 0x0200 // 512
|
||||
dumpPacketHeader = 0x0400 // 1024
|
||||
|
|
@ -168,7 +173,7 @@ func main() {
|
|||
case err := <-inputErrChan:
|
||||
fmt.Fprintln(os.Stderr, err)
|
||||
fmt.Fprintln(os.Stderr, "Trying again in 10s")
|
||||
time.Sleep(10 * time.Second)
|
||||
//time.Sleep(10 * time.Second)
|
||||
go input(*inputURL, *outputURL)
|
||||
case err := <-outputErrChan:
|
||||
fmt.Fprintln(os.Stderr, err)
|
||||
|
|
@ -196,6 +201,72 @@ func setUpDirs() {
|
|||
func input(input string, output string) {
|
||||
fmt.Printf("Reading video from %s\n", input)
|
||||
|
||||
// (re)initialize globals
|
||||
clipCount = 0
|
||||
expectCC = -1
|
||||
dumpCC = -1
|
||||
dumpPCRBase = 0
|
||||
rtpSequenceNum = uint16(rand.Intn(1 << 15))
|
||||
|
||||
var err error
|
||||
|
||||
// for UDP and RTP only dial once
|
||||
if strings.HasPrefix(output, "udp://") || strings.HasPrefix(output, "rtp://") {
|
||||
conn, err = net.Dial("udp", output[6:])
|
||||
if err != nil {
|
||||
inputErrChan <- err
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
var br *bufio.Reader
|
||||
|
||||
if *flags&filterEdgeDetection != 0{
|
||||
args := []string{
|
||||
"-r", "25",
|
||||
"-i", input,
|
||||
"-pix_fmt", "gray",
|
||||
"-vf", "edgedetect",
|
||||
"-f", "mpegts", "-",
|
||||
}
|
||||
|
||||
fmt.Printf("Executing: %s %s\n", ffmpegPath, strings.Join(args, " "))
|
||||
cmd := exec.Command(ffmpegPath, args...)
|
||||
stdout, _ := cmd.StdoutPipe()
|
||||
|
||||
err := cmd.Start()
|
||||
|
||||
if err != nil {
|
||||
inputErrChan <- err
|
||||
return
|
||||
}
|
||||
|
||||
br = bufio.NewReader(stdout)
|
||||
}else{
|
||||
|
||||
|
||||
|
||||
if *flags&filterMotionDetect != 0{
|
||||
args := []string{
|
||||
"-i", input,
|
||||
"-vf", "\"select=gt(scene,0.0001),setpts=N/(25*TB),fps=fps=25\"",
|
||||
"-f", "mpegts", "-",
|
||||
}
|
||||
|
||||
fmt.Printf("Executing: %s %s\n", ffmpegPath, strings.Join(args, " "))
|
||||
cmd := exec.Command(ffmpegPath, args...)
|
||||
stdout, _ := cmd.StdoutPipe()
|
||||
|
||||
err := cmd.Start()
|
||||
|
||||
if err != nil {
|
||||
inputErrChan <- err
|
||||
return
|
||||
}
|
||||
|
||||
br = bufio.NewReader(stdout)
|
||||
|
||||
}else{
|
||||
args := []string{
|
||||
"-r", strconv.Itoa(*frameRate),
|
||||
"-i", input,
|
||||
|
|
@ -238,28 +309,25 @@ func input(input string, output string) {
|
|||
return
|
||||
}
|
||||
|
||||
// (re)initialize globals
|
||||
clipCount = 0
|
||||
expectCC = -1
|
||||
dumpCC = -1
|
||||
dumpPCRBase = 0
|
||||
rtpSequenceNum = uint16(rand.Intn(1 << 15))
|
||||
|
||||
// for UDP and RTP only dial once
|
||||
if strings.HasPrefix(output, "udp://") || strings.HasPrefix(output, "rtp://") {
|
||||
conn, err = net.Dial("udp", output[6:])
|
||||
if err != nil {
|
||||
inputErrChan <- err
|
||||
return
|
||||
}
|
||||
|
||||
br = bufio.NewReader(stdout)
|
||||
}
|
||||
|
||||
br := bufio.NewReader(stdout)
|
||||
}
|
||||
|
||||
clipStore(br,input)
|
||||
|
||||
}
|
||||
|
||||
func clipStore(br *bufio.Reader, input string){
|
||||
|
||||
clipSize := 0
|
||||
packetCount := 0
|
||||
now := time.Now()
|
||||
prevTime := now
|
||||
fmt.Printf("Looping\n")
|
||||
|
||||
for {
|
||||
if clip, err := ringBuffer.Get(); err != nil {
|
||||
inputErrChan <- err
|
||||
|
|
@ -268,15 +336,18 @@ func input(input string, output string) {
|
|||
for {
|
||||
upperBound := clipSize + mp2tPacketSize
|
||||
_, err := io.ReadFull(br, clip[clipSize:upperBound])
|
||||
|
||||
if err != nil {
|
||||
inputErrChan <- err
|
||||
return
|
||||
}
|
||||
if *flags&filterFixContinuity != 0 && mp2tFixContinuity(clip[clipSize:upperBound], packetCount, uint16(*selectedPID)) {
|
||||
|
||||
if *flags&filterFixContinuity != 0 && mp2tFixContinuity(clip[clipSize:upperBound], uint16(*selectedPID)) {
|
||||
fmt.Printf("Packet #%d.%d fixed\n", clipCount, packetCount)
|
||||
}
|
||||
packetCount++
|
||||
clipSize += mp2tPacketSize
|
||||
|
||||
// send if (1) our buffer is full or (2) 1 second has elapsed and we have % packetsPerFrame
|
||||
now = time.Now()
|
||||
if (packetCount == mp2tMaxPackets) ||
|
||||
|
|
@ -296,6 +367,42 @@ func input(input string, output string) {
|
|||
}
|
||||
}
|
||||
|
||||
func printGray(img *image.Gray) {
|
||||
shade := []string{" ", "█"}
|
||||
for i, p := range img.Pix {
|
||||
x := 0
|
||||
if p > 20 {
|
||||
x = 1
|
||||
}
|
||||
fmt.Print(shade[x])
|
||||
if (i+1)%640 == 0 {
|
||||
fmt.Print("\n")
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func checkSlice(a,b *image.Gray){
|
||||
for i , _ := range a.Pix{
|
||||
fmt.Printf("%v - %v = %v\n", a.Pix[i], b.Pix[i], a.Pix[i]-b.Pix[i])
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
func FastCompare(img1, img2 *image.Gray) (float64, error) {
|
||||
if img1.Bounds() != img2.Bounds() {
|
||||
return 0, fmt.Errorf("image bounds not equal: %+v, %+v", img1.Bounds(), img2.Bounds())
|
||||
}
|
||||
|
||||
accumError := float64(0)
|
||||
|
||||
for i := 0; i < len(img1.Pix); i+= 20 {
|
||||
accumError += float64((img1.Pix[i] - img2.Pix[i]) * (img1.Pix[i] - img2.Pix[i]))
|
||||
}
|
||||
|
||||
return math.Sqrt(accumError), nil
|
||||
}
|
||||
|
||||
// output handles the writing to specified output
|
||||
func output(output string) {
|
||||
elapsedTime := time.Duration(0)
|
||||
|
|
@ -304,7 +411,9 @@ func output(output string) {
|
|||
for {
|
||||
if clip, err := ringBuffer.Read(); err == nil {
|
||||
now := time.Now()
|
||||
|
||||
err := sendClip(clip, output, conn)
|
||||
sendClipToStdout(clip, output, conn)
|
||||
for err != nil {
|
||||
outputErrChan <- err
|
||||
err = sendClip(clip, output, conn)
|
||||
|
|
@ -440,6 +549,7 @@ func sendClipToStdout(clip []byte, _ string, _ net.Conn) error {
|
|||
tcs := pkt[3] & 0xc0 >> 6
|
||||
afc := pkt[3] & 0x30 >> 4
|
||||
cc = int(pkt[3] & 0xf)
|
||||
di := pkt[5]&0x80
|
||||
|
||||
if dumpCC != -1 && cc != dumpCC {
|
||||
discontinuities++
|
||||
|
|
@ -458,14 +568,14 @@ func sendClipToStdout(clip []byte, _ string, _ net.Conn) error {
|
|||
if adaptationfield.HasPCR(pkt) {
|
||||
pcrBase, pcrExt, _ := mp2tGetPCR(pkt)
|
||||
if *flags&dumpPacketHeader != 0 {
|
||||
fmt.Printf("\t\tAFL=%d, PCRbase=%d, PCRext=%d\n", afl, pcrBase, pcrExt)
|
||||
fmt.Printf("\t\tAFL=%d, PCRbase=%d, PCRext=%d, DI=%v\n", afl, pcrBase, pcrExt, di)
|
||||
}
|
||||
if pcrBase < dumpPCRBase {
|
||||
fmt.Printf("Warning: PCRbase went backwards!\n")
|
||||
}
|
||||
dumpPCRBase = pcrBase
|
||||
} else if *flags&dumpPacketHeader != 0 {
|
||||
fmt.Printf("\t\tAFL=%d\n", afl)
|
||||
fmt.Printf("\t\tAFL=%d, DI=%v\n", afl, di)
|
||||
}
|
||||
}
|
||||
if *flags&dumpPacketPayload != 0 {
|
||||
|
|
@ -528,10 +638,11 @@ func mp2tDumpPmt(pn uint16, pmt psi.PMT) {
|
|||
}
|
||||
|
||||
// Mp2tFixContinuity fixes discontinous MPEG-TS continuity counts (CC)
|
||||
func mp2tFixContinuity(pkt []byte, packetCount int, pid uint16) bool {
|
||||
func mp2tFixContinuity(pkt []byte, pid uint16) bool {
|
||||
|
||||
hasPayload, err := packet.ContainsPayload(pkt)
|
||||
if err != nil {
|
||||
fmt.Printf("Warning: Packet #%d.%d bad.\n", clipCount, packetCount)
|
||||
fmt.Printf("Warning: Packet bad.\n")
|
||||
return false
|
||||
}
|
||||
if !hasPayload {
|
||||
|
|
@ -546,6 +657,7 @@ func mp2tFixContinuity(pkt []byte, packetCount int, pid uint16) bool {
|
|||
if expectCC == -1 {
|
||||
expectCC = cc
|
||||
} else if cc != expectCC {
|
||||
fmt.Println("Have to fix")
|
||||
pkt[3] = pkt[3]&0xf0 | byte(expectCC&0xf)
|
||||
fixed = true
|
||||
}
|
||||
|
|
@ -553,6 +665,16 @@ func mp2tFixContinuity(pkt []byte, packetCount int, pid uint16) bool {
|
|||
return fixed
|
||||
}
|
||||
|
||||
func resetContinuityCount(pkt []byte) (bool){
|
||||
|
||||
if afc := pkt[3] & 0x30 >> 4; afc == 3{
|
||||
pkt[5] = pkt[5] | byte(1 << 7)
|
||||
}
|
||||
|
||||
return true
|
||||
|
||||
}
|
||||
|
||||
// Mp2tGetPCR extracts the Program Clock Reference (PCR) from an MPEG-TS packet (if any)
|
||||
func mp2tGetPCR(pkt []byte) (uint64, uint32, bool) {
|
||||
if !adaptationfield.HasPCR(pkt) {
|
||||
|
|
|
|||
Loading…
Reference in New Issue