无法在Rx中实现滑动窗口

时间:2013-03-06 19:43:47

标签: system.reactive

我为反应式扩展创建了一个SlidingWindow()运算符,因为我希望能够轻松监控滚动平均值等等。作为一个简单的例子,我想订阅听鼠标事件,但每次都有我希望收到的事件最后三个(而不是等待每三个事件接收最后三个)。这就是为什么我发现Window重载似乎没有给我开箱即用的东西。

这就是我想出的。鉴于其频繁的List操作,我担心它可能不是最高性能的解决方案:

public static IObservable<List<T>> SlidingWindow<T>(this IObservable<T> seq, int length)
{
    var seed = new List<T>();

    Func<List<T>, T, List<T>> accumulator = (list, arg2) =>
    {
        list.Add(arg2);

        if (list.Count > length)
            list.RemoveRange(0, (list.Count - length));

        return list;
    };

    return seq.Scan(seed, accumulator)
                .Where(list => list.Count == length);
}

可以这样称呼:

var rollingSequence = Observable.Range(1, 5).SlidingWindow().ToEnumerable();

然而,令我惊讶的是,而不是收到预期的结果

1,2,3
2,3,4
3,4,5

我收到了结果

2,3,4
3,4,5
3,4,5

非常感谢任何见解!

4 个答案:

答案 0 :(得分:11)

使用原始测试,参数为3表示计数,这会得到所需的结果:

public static IObservable<IList<T>> SlidingWindow<T>(
    this IObservable<T> source, int count)
{
    return source.Buffer(count, 1)
                 .Where(list => list.Count == count);
}

像这样测试:

var source = Observable.Range(1, 5);
var query = source.SlidingWindow(3);
using (query.Subscribe(i => Console.WriteLine(string.Join(",", i))))
{

}

输出:

1,2,3
2,3,4
3,4,5

答案 1 :(得分:6)

source.Window(count, 1) - 或source.Buffer(count, 1) 它是“计数”项目的窗口/缓冲区,由一个滑动。

答案 2 :(得分:4)

试试这个 - 我必须坐下来考虑一下它的相对表现,但是至少 可能一样好,并且更容易阅读:

public static IObservable<IList<T>> SlidingWindow<T>(
       this IObservable<T> src, 
       int windowSize)
{
    var feed = src.Publish().RefCount();    
    // (skip 0) + (skip 1) + (skip 2) + ... + (skip nth) => return as list  
    return Observable.Zip(
       Enumerable.Range(0, windowSize)
           .Select(skip => feed.Skip(skip))
           .ToArray());
}

试验台:

var source = Observable.Range(0, 10);
var query = source.SlidingWindow(3);
using(query.Subscribe(Console.WriteLine))
{               
    Console.ReadLine();
}

输出:

ListOf(0,1,2)
ListOf(1,2,3)
ListOf(2,3,4)
ListOf(3,4,5)
ListOf(4,5,6)
...

编辑:顺便说一句,我发现自己被强迫.Publish().RefCount()因为没有这样做而被烧了一次......我不认为这是严格要求的,所以。

编辑yzorg:

如果你像这样扩充方法,你会更清楚地看到运行时行为:

public static IObservable<IList<T>> SlidingWindow<T>(
    this IObservable<T> src, 
    int windowSize)
{
    var feed = src.Publish().RefCount();    
    // (skip 0) + (skip 1) + (skip 2) + ... + (skip nth) => return as list  
    return Observable.Zip(
    Enumerable.Range(0, windowSize)
        .Select(skip => 
        {
            Console.WriteLine("Skipping {0} els", skip);
            return feed.Skip(skip);
        })
        .ToArray());
}

答案 3 :(得分:4)

这里的滑动窗口实现不足以实现我对滑动窗口的想法。最接近的是使用Buffer(N, 1),但是这是一个问题,因为它在发出第一个结果之前会等待前N个项目,然后滑出序列的末尾。我一次最多要发射N个物品。

我最终完成了该实现:

public static IObservable<IList<T>> SlidingWindow<T>(this IObservable<T> obs, int windowSize) =>
    Observable.Create<IList<T>>(observer =>
    {
        var buffer = new CircularBuffer<T>(windowSize);
        return obs.Subscribe(
            value =>
            {
                buffer.Add(value);
                observer.OnNext(buffer.ToList());
            },
            ex => observer.OnError(ex),
            () => observer.OnCompleted()
        );
    });

最初,我使用缓冲区队列,但想使用一些更轻量的东西。

public class CircularBuffer<T> : IReadOnlyList<T>
{
    private readonly T[] buffer;
    private int offset;
    private int count;
    public CircularBuffer(int bufferSize) => this.buffer = new T[bufferSize];
    public int Capacity => buffer.Length;
    public int Count => count;
    public T this[int index] => index < 0 || index >= count
        ? throw new ArgumentOutOfRangeException(nameof(index))
        : buffer[(offset + index) % buffer.Length];
    public void Add(T value)
    {
        buffer[(offset + count) % buffer.Length] = value;
        if (count < buffer.Length) count++;
        else offset = (offset + 1) % buffer.Length;
    }
    public IEnumerator<T> GetEnumerator()
    {
        for (var i = 0; i < count; i++)
            yield return buffer[(offset + i) % buffer.Length];
    }
    IEnumerator IEnumerable.GetEnumerator() => GetEnumerator();
}

它将产生Observable.Range(0, 10).SlidingWindow(3)的序列:

 0,1,2,3,4,5,6,7,8,9
[0]
[0,1]
[0,1,2]
  [1,2,3]
    [2,3,4]
      [3,4,5]
        [4,5,6]
          [5,6,7]
            [6,7,8]
              [7,8,9]