Added unique function and a workaround for non-matching inputs to rules passing

the rule input filter.

1 files changed, 47 insertions, 0 deletions

diff --git a/src/functionunique.cpp b/src/functionunique.cpp
new file mode 100644
index 0000000..ca2461f
--- /dev/null
+++ b/src/functionunique.cpp
@@ -0,0 +1,47 @@
+#include "functionunique.h"
+
+#include <bu/hash.h>
+#include <bu/plugger.h>
+PluginInterface3( pluginFunctionUnique, unique, FunctionUnique, Function,
+                "Mike Buland", 0, 1 );
+
+FunctionUnique::FunctionUnique()
+{
+}
+
+FunctionUnique::~FunctionUnique()
+{
+}
+
+Bu::String FunctionUnique::getName() const
+{
+        return "unique";
+}
+
+Variable FunctionUnique::call( Variable &input, VarList lParams )
+{
+        if( lParams.getSize() > 0 )
+                throw Bu::ExceptionBase("Unique does not take any parameters.");
+        if( input.getType() != Variable::typeList )
+                throw Bu::ExceptionBase("unique does not work on non-list types.");
+
+        Bu::Hash<Bu::String, bool> hHas;
+
+        Variable vOut( Variable::typeList );
+        for( VarList::iterator i = input.begin(); i; i++ )
+        {
+                if( (*i).getType() != Variable::typeString )
+                        continue;
+
+                Bu::String s = (*i).getString();
+                
+                if( hHas.has( s ) )
+                        continue;
+
+                hHas.insert( s, true );
+                vOut.append( *i );
+        }
+
+        return vOut;
+}
+