From f006d54995e8b28c842263a834a94fad16549159 Mon Sep 17 00:00:00 2001
From: sjunges <sebastian.junges@rwth-aachen.de>
Date: Tue, 25 Aug 2015 17:09:29 +0200
Subject: [PATCH] vector min_if, max_if

Former-commit-id: afff48d2e5bfe5bce4b7119a73b9bd1cd0f181a5
---
 src/utility/vector.h                   | 41 ++++++++++++++++++++++++++
 test/functional/utility/VectorTest.cpp | 19 ++++++++++++
 2 files changed, 60 insertions(+)

diff --git a/src/utility/vector.h b/src/utility/vector.h
index bd21028f9..dd0ac7cf3 100644
--- a/src/utility/vector.h
+++ b/src/utility/vector.h
@@ -366,6 +366,47 @@ namespace storm {
                 return sum;
             }
             
+            /**
+             * Computes the maximum of the entries from the values that are set to one in the filter vector
+             * @param values
+             * @param filter
+             * @param smallestPossibleValue A value which is not larger than any value in values. If the filter is empty, this value is returned.
+             * @return  The maximum over the subset of the values and the smallestPossibleValue.
+             */
+            template<typename VT>
+            VT max_if(std::vector<VT> const& values, storm::storage::BitVector const& filter, VT const& smallestPossibleValue) {
+                assert(values.size() >= filter.size());
+                
+                VT max = smallestPossibleValue;
+                for(uint_fast64_t i : filter) {
+                    if(values[i] > max) { 
+                        max = values[i];
+                    }
+                }    
+                return max;
+            }
+            
+            /**
+             * Computes the minimum of the entries from the values that are set to one in the filter vector
+             * @param values
+             * @param filter
+             * @param largestPossibleValue A value which is not smaller than any value in values. If the filter is empty, this value is returned.
+             * @return  The minimum over the subset of the values and the largestPossibleValue.
+             */
+            template<typename VT>
+            VT min_if(std::vector<VT> const& values, storm::storage::BitVector const& filter, VT const& largestPossibleValue) {
+                assert(values.size() >= filter.size());
+                VT min = largestPossibleValue;
+                for(uint_fast64_t i : filter) {
+                    if(values[i] < min) { 
+                        min = values[i];
+                    }
+                }    
+                return min;
+            }
+            
+            
+            
             /*!
              * Reduces the given source vector by selecting an element according to the given filter out of each row group.
              *
diff --git a/test/functional/utility/VectorTest.cpp b/test/functional/utility/VectorTest.cpp
index 33f3c5cfe..9494d6146 100644
--- a/test/functional/utility/VectorTest.cpp
+++ b/test/functional/utility/VectorTest.cpp
@@ -12,3 +12,22 @@ TEST(VectorTest, sum_if) {
     ASSERT_EQ(24.0, storm::utility::vector::sum_if(a, f2));
     
 }
+
+TEST(VectorTest, max_if) {
+    std::vector<double> a = {1.0, 2.0, 34.0, 8.0, 16.0};
+    storm::storage::BitVector f1(5, {2,4});
+    storm::storage::BitVector f2(5, {3,4});
+    
+    ASSERT_EQ(34.0, storm::utility::vector::max_if(a,f1,0.0));
+    ASSERT_EQ(16.0, storm::utility::vector::max_if(a,f2,0.0));
+}
+
+
+TEST(VectorTest, min_if) {
+    std::vector<double> a = {1.0, 2.0, 34.0, 8.0, 16.0};
+    storm::storage::BitVector f1(5, {2,4});
+    storm::storage::BitVector f2(5, {3,4});
+    
+    ASSERT_EQ(16.0, storm::utility::vector::min_if(a,f1,100.0));
+    ASSERT_EQ(8.0, storm::utility::vector::min_if(a,f2,100.0));
+}