mirror of https://bitbucket.org/ausocean/av.git
Initial implementation
This commit is contained in:
parent
c2df8a8fc1
commit
15efc8331d
|
@ -265,7 +265,7 @@ type Config struct {
|
|||
|
||||
HorizontalFlip bool // HorizontalFlip flips video horizontally for Raspivid input.
|
||||
VerticalFlip bool // VerticalFlip flips video vertically for Raspivid input.
|
||||
Filter int // Defines the method of filtering to be used in between lexing and encoding.
|
||||
Filters []int // Defines the methods of filtering to be used in between lexing and encoding.
|
||||
PSITime int // Sets the time between a packet being sent
|
||||
|
||||
// RTMP ring buffer parameters.
|
||||
|
|
|
@ -129,8 +129,8 @@ type Revid struct {
|
|||
// lexTo, encoder and packer handle transcoding the input stream.
|
||||
lexTo func(dest io.Writer, src io.Reader, delay time.Duration) error
|
||||
|
||||
// filter will hold the filter interface that will write to the chosen filter from the lexer.
|
||||
filter filter.Filter
|
||||
// filters will hold the filter interface that will write to the chosen filter from the lexer.
|
||||
filters []filter.Filter
|
||||
|
||||
// encoders will hold the multiWriteCloser that writes to encoders from the filter.
|
||||
encoders io.WriteCloser
|
||||
|
@ -346,19 +346,29 @@ func (r *Revid) setupPipeline(mtsEnc func(dst io.WriteCloser, rate float64) (io.
|
|||
|
||||
r.encoders = multiWriter(encoders...)
|
||||
|
||||
switch r.cfg.Filter {
|
||||
case config.FilterNoOp:
|
||||
r.filter = filter.NewNoOp(r.encoders)
|
||||
case config.FilterMOG:
|
||||
r.filter = filter.NewMOGFilter(r.encoders, mogMinArea, mogThreshold, mogHistory, r.cfg.ShowWindows)
|
||||
case config.FilterVariableFPS:
|
||||
r.filter = filter.NewVariableFPSFilter(r.encoders, minFPS, filter.NewMOGFilter(r.encoders, mogMinArea, mogThreshold, mogHistory, r.cfg.ShowWindows))
|
||||
case config.FilterKNN:
|
||||
r.filter = filter.NewKNNFilter(r.encoders, knnMinArea, knnThreshold, knnHistory, knnKernel, r.cfg.ShowWindows)
|
||||
l := len(r.cfg.Filters)
|
||||
r.filters = []filter.Filter{filter.NewNoOp(r.encoders)}
|
||||
if l != 0 {
|
||||
r.filters = make([]filter.Filter, l)
|
||||
dst := r.encoders
|
||||
|
||||
for i := l - 1; i >= 0; i-- {
|
||||
switch r.cfg.Filters[i] {
|
||||
case config.FilterNoOp:
|
||||
r.filters[i] = filter.NewNoOp(dst)
|
||||
case config.FilterMOG:
|
||||
r.filters[i] = filter.NewMOGFilter(dst, mogMinArea, mogThreshold, mogHistory, r.cfg.ShowWindows)
|
||||
case config.FilterVariableFPS:
|
||||
r.filters[i] = filter.NewVariableFPSFilter(dst, minFPS, filter.NewMOGFilter(dst, mogMinArea, mogThreshold, mogHistory, r.cfg.ShowWindows))
|
||||
case config.FilterKNN:
|
||||
r.filters[i] = filter.NewKNNFilter(dst, knnMinArea, knnThreshold, knnHistory, knnKernel, r.cfg.ShowWindows)
|
||||
default:
|
||||
panic("Undefined Filter")
|
||||
}
|
||||
dst = r.filters[i]
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
switch r.cfg.Input {
|
||||
case config.InputRaspivid:
|
||||
|
@ -474,7 +484,7 @@ func (r *Revid) Stop() {
|
|||
r.cfg.Logger.Log(logger.Error, pkg+"failed to close pipeline", "error", err.Error())
|
||||
}
|
||||
|
||||
err = r.filter.Close()
|
||||
//err = r.filter.Close()
|
||||
if err != nil {
|
||||
r.cfg.Logger.Log(logger.Error, pkg+"failed to close pipeline", "error", err.Error())
|
||||
}
|
||||
|
@ -667,13 +677,17 @@ func (r *Revid) Update(vars map[string]string) error {
|
|||
default:
|
||||
r.cfg.Logger.Log(logger.Warning, pkg+"invalid VerticalFlip param", "value", value)
|
||||
}
|
||||
case "Filter":
|
||||
case "Filters":
|
||||
filters := strings.Split(value, ",")
|
||||
m := map[string]int{"NoOp": config.FilterNoOp, "MOG": config.FilterMOG, "VariableFPS": config.FilterVariableFPS, "KNN": config.FilterKNN}
|
||||
v, ok := m[value]
|
||||
r.cfg.Filters = make([]int, len(filters))
|
||||
for i, filter := range filters {
|
||||
v, ok := m[filter]
|
||||
if !ok {
|
||||
r.cfg.Logger.Log(logger.Warning, pkg+"invalid FilterMethod param", "value", value)
|
||||
r.cfg.Logger.Log(logger.Warning, pkg+"invalid Filters param", "value", value)
|
||||
}
|
||||
r.cfg.Filters[i] = v
|
||||
}
|
||||
r.cfg.Filter = v
|
||||
case "PSITime":
|
||||
v, err := strconv.Atoi(value)
|
||||
if err != nil || v < 0 {
|
||||
|
@ -792,7 +806,7 @@ func (r *Revid) Update(vars map[string]string) error {
|
|||
// processFrom is run as a routine to read from a input data source, lex and
|
||||
// then send individual access units to revid's encoders.
|
||||
func (r *Revid) processFrom(read io.Reader, delay time.Duration) {
|
||||
r.err <- r.lexTo(r.filter, read, delay)
|
||||
r.err <- r.lexTo(r.filters[0], read, delay)
|
||||
r.cfg.Logger.Log(logger.Info, pkg+"finished lexing")
|
||||
r.wg.Done()
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue