--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/servlib/prophet/Stats.cc	Thu Apr 21 14:57:45 2011 +0100
@@ -0,0 +1,109 @@
+/*
+ *    Copyright 2007 Baylor University
+ *
+ *    Licensed under the Apache License, Version 2.0 (the "License");
+ *    you may not use this file except in compliance with the License.
+ *    You may obtain a copy of the License at
+ *
+ *        http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *    Unless required by applicable law or agreed to in writing, software
+ *    distributed under the License is distributed on an "AS IS" BASIS,
+ *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *    See the License for the specific language governing permissions and
+ *    limitations under the License.
+ */
+
+#include "Stats.h"
+
+namespace prophet
+{
+
+Stats::~Stats()
+{
+    for (pstats::iterator i = pstats_.begin(); i != pstats_.end(); i++)
+    {
+        delete (*i).second;
+    }
+    pstats_.clear();
+}
+
+void
+Stats::update_stats(const Bundle* b, double p)
+{
+    StatsEntry* se = find(b);
+
+    // weed out the oddball
+    if (se == NULL) return;
+
+    // update p_max
+    if (se->p_max_ < p) se->p_max_ = p;
+
+    // update MOPR, Section 3.7, eq. 7
+    se->mopr_ += (1 - se->mopr_) * p;
+
+    // update LMOPR, Section 3.7, eq. 8
+    se->lmopr_ += p;
+
+    // this shouldn't be necessary, since the pointer is already stored
+    // pstats_[b] = se;
+}
+
+double
+Stats::get_p_max(const Bundle* b) const
+{
+    if (b == NULL) return 0.0;
+    return const_cast<Stats*>(this)->find(b)->p_max_;
+}
+
+double
+Stats::get_mopr(const Bundle* b) const
+{
+    if (b == NULL) return 0.0;
+    return const_cast<Stats*>(this)->find(b)->mopr_;
+}
+
+double
+Stats::get_lmopr(const Bundle* b) const
+{
+    if (b == NULL) return 0.0;
+    return const_cast<Stats*>(this)->find(b)->lmopr_;
+}
+
+StatsEntry*
+Stats::find(const Bundle* b)
+{
+    if (b == NULL) return NULL;
+
+    // search for Stats for Bundle
+    iterator i = pstats_.lower_bound(b->sequence_num());
+
+    // return it if we find it
+    if (i != pstats_.end() && !pstats_.key_comp()(b->sequence_num(),i->first))
+        return (*i).second;
+
+    // otherwise create new zero'd StatsEntry
+    StatsEntry* se = new StatsEntry();
+
+    // insert into container
+    pstats_.insert(i,pstats::value_type(b->sequence_num(),se));
+
+    // return new entry
+    return se;
+}
+
+void
+Stats::drop_bundle(const Bundle* b)
+{
+    if (b == NULL) return;
+
+    iterator i = pstats_.find(b->sequence_num());
+    if (i != pstats_.end())
+    {
+        delete (*i).second;
+        pstats_.erase(i);
+        dropped_++;
+    }
+}
+
+}; // namespace prophet