From 21135fbddb0ab6abb98ea1ca03fff8e6dd265499 Mon Sep 17 00:00:00 2001
From: yvxiang <yangce@baidu.com>
Date: Wed, 18 May 2016 11:13:21 +0800
Subject: [PATCH] Support resize sliding window

---
 include/sliding_window.h | 125 ++++++++++++++++++++++++++++++++++++++-
 1 file changed, 122 insertions(+), 3 deletions(-)

diff --git a/include/sliding_window.h b/include/sliding_window.h
index fe04c0d..36cca51 100644
--- a/include/sliding_window.h
+++ b/include/sliding_window.h
@@ -22,13 +22,13 @@ class SlidingWindow {
 public:
     typedef boost::function<void (int32_t, Item)> SlidingCallback;
     SlidingWindow(int32_t size, SlidingCallback callback)
-      : bitmap_(NULL), items_(NULL), item_count_(0), 
+      : bitmap_(NULL), items_(NULL), item_count_(0),
         callback_(callback), size_(size),
-        base_offset_(0), ready_(0), notifying_(false) {
+        base_offset_(0), ready_(0), notifying_(false),
+        to_be_shrink_(false), expect_size_(size) {
         bitmap_ = new char[size];
         memset(bitmap_, 0, size);
         items_ = new Item[size];
-        size_ = size;
     }
     ~SlidingWindow() {
         delete[] bitmap_;
@@ -61,6 +61,15 @@ class SlidingWindow {
             --item_count_;
             if (ready_ >= size_) {
                 ready_ = 0;
+                if (to_be_shrink_) {
+                    to_be_shrink_ = false;
+                    delete[] bitmap_;
+                    delete[] items_;
+                    bitmap_ = new char[expect_size_];
+                    memset(bitmap_, 0, expect_size_);
+                    items_ = new Item[expect_size_];
+                    size_ = expect_size_;
+                }
             }
         }
         notifying_ = false;
@@ -95,6 +104,114 @@ class SlidingWindow {
         if (!notifying_) Notify();
         return 0;
     }
+    void Resize(int new_size) {
+        MutexLock lock(&mu_);
+        if (new_size == size_ || new_size < 0) {
+            return;
+        }
+        if (new_size > size_) {
+            EnlargeSize(new_size);
+        } else {
+            ShrinkSize(new_size);
+        }
+    }
+private:
+    int GetLastItemPos() {
+        mu_.AssertHeld();
+        for (int i = size_ - 1; i >= 0; i--) {
+            if (bitmap_[i]) {
+                return i;
+            }
+        }
+        return -1;
+    }
+    int GetFirstItemPos() {
+        mu_.AssertHeld();
+        for (int i = 0; i < size_; i++) {
+            if (bitmap_[i]) {
+                return i;
+            }
+        }
+        return size_ + 1;
+    }
+    void EnlargeSize(int new_size) {
+        mu_.AssertHeld();
+        char* new_bitmap = NULL;
+        Item* new_items = NULL;
+        int last_non_empty_pos = GetLastItemPos();
+        int first_non_empty_pos = GetFirstItemPos();
+        // maybe shrink is unfinished
+        if (to_be_shrink_)  {
+            to_be_shrink_ = false;
+        }
+        if (last_non_empty_pos == -1) {
+            //current window is empty
+            new_bitmap = new char[new_size];
+            memset(new_bitmap, 0, new_size);
+            new_items = new Item[new_size];
+        } else {
+            assert(last_non_empty_pos >= first_non_empty_pos);
+            new_bitmap = new char[new_size];
+            new_items = new Item[new_size];
+            memset(new_bitmap, 0, new_size);
+            //don't use memcpy for item, because item maybe not a POD type
+            if (last_non_empty_pos >= ready_) {
+                for (int i = ready_, j = 0; i <= last_non_empty_pos; i++, j++) {
+                    new_items[j] = items_[i];
+                }
+                memcpy(new_bitmap, bitmap_ + ready_, (last_non_empty_pos - ready_ + 1) * sizeof(char));
+            }
+            //TODO: improve here
+            if (first_non_empty_pos < ready_) {
+                for (int i = first_non_empty_pos; i < ready_; i++) {
+                    new_items[size_ - ready_ + i] = items_[i];
+                }
+                memcpy(new_bitmap + size_ - ready_ + first_non_empty_pos,
+                        bitmap_ + first_non_empty_pos,
+                        (ready_ - first_non_empty_pos) * sizeof(char));
+            }
+        }
+        delete[] bitmap_;
+        delete[] items_;
+        bitmap_ = new_bitmap;
+        items_ = new_items;
+        ready_ = 0;
+        size_ = new_size;
+    }
+    void ShrinkSize(int new_size) {
+        //TODO: deal with multiple shrink
+        mu_.AssertHeld();
+        int last_non_empty_pos = GetLastItemPos();
+        int first_non_empty_pos = GetFirstItemPos();
+        char* new_bitmap = NULL;
+        Item* new_items = NULL;
+        if (last_non_empty_pos == -1) {
+            // current window is empty
+            new_bitmap = new char[new_size];
+            memset(new_bitmap, 0, new_size);
+            new_items = new Item[new_size];
+            memset(new_items, 0, new_size);
+            size_ = new_size;
+        } else {
+            assert(last_non_empty_pos >= first_non_empty_pos);
+            int tmp_size = last_non_empty_pos - first_non_empty_pos + 1;
+            new_bitmap = new char[tmp_size];
+            //no need to memset
+            memcpy(new_bitmap, bitmap_ + first_non_empty_pos, (tmp_size) * sizeof(char));
+            new_items = new Item[tmp_size];
+            for (int i = 0; i < tmp_size; i++) {
+                new_items[i] = items_[i + first_non_empty_pos];
+            }
+            size_ = tmp_size;
+            expect_size_ = new_size;
+            to_be_shrink_ = true;
+        }
+        delete[] bitmap_;
+        delete[] items_;
+        bitmap_ = new_bitmap;
+        items_ = new_items;
+        ready_ = 0;
+    }
 private:
     char* bitmap_;
     Item* items_;
@@ -104,6 +221,8 @@ class SlidingWindow {
     int32_t base_offset_;
     int32_t ready_;
     bool notifying_;
+    bool to_be_shrink_;
+    int32_t expect_size_;
     mutable Mutex mu_;
 };