Add files for swift over UDP.

[swifty.git] / src / libswift_udp / availability.h

diff --git a/src/libswift_udp/availability.h b/src/libswift_udp/availability.h
new file mode 100644 (file)
index 0000000..2c7b31c
--- /dev/null
+++ b/src/libswift_udp/availability.h
@@ -0,0 +1,97 @@
+/*
+ *  availability.h
+ *  Tree keeping track of the availability of each bin in a swarm
+ *
+ *  Created by Riccardo Petrocco
+ *  Copyright 2009-2012 Delft University of Technology. All rights reserved.
+ *
+ */
+//#include "bin.h"
+//#include "binmap.h"
+#include "swift.h"
+#include <cassert>
+
+#ifndef AVAILABILITY_H
+#define AVAILABILITY_H
+
+namespace swift {
+
+typedef        std::vector< std::pair<uint32_t, binmap_t*> >   WaitingPeers;
+
+class Availability
+{
+    public:
+
+               /**
+            * Constructor
+            */
+           Availability(void) {        size_ = 0;          }
+
+
+           /**
+            * Constructor
+            */
+           explicit Availability(int size)
+           {
+               assert(size <= 0);
+               size_ = size;
+               avail_ = new uint8_t[size];
+           }
+
+        ~Availability(void)
+        {
+            if (size_)
+                delete [] avail_;
+        }
+
+           /** return the availability array */
+           uint8_t* get() { return avail_; }
+
+           /** returns the availability of a single bin */
+           uint8_t get(const bin_t bin);
+
+           /** set/update the availability */
+           void set(uint32_t channel_id, binmap_t& binmap, bin_t target);
+
+           /** removes the binmap of leaving peers */
+           void remove(uint32_t channel_id, binmap_t& binmap);
+
+           /** returns the size of the availability tree */
+           int size() { return size_; }
+
+           /** sets the size of the availability tree once we know the size of the file */
+           void setSize(uint64_t size);
+
+           /** sets a binmap */
+           void setBinmap(binmap_t *binmap);
+
+           /** get rarest bin, of specified width, within a range */
+           bin_t getRarest(const bin_t range, int width);
+
+           /** Echo the availability status to stdout */
+               void status() const;
+
+    protected:
+           uint8_t *avail_;
+           uint64_t    size_;
+           // a list of incoming have msgs, those are saved only it the file size is still unknown
+            // TODO fix... set it depending on the # of channels * something
+           WaitingPeers waiting_peers_;
+           //binmap_t *waiting_[20];
+
+
+
+           /** removes the binmap */
+           void removeBinmap(binmap_t& binmap);
+
+           /** removes the bin */
+           void removeBin(bin_t bin);
+
+           /** sets a bin */
+           void setBin(bin_t bin);
+
+};
+
+}
+
+#endif