Add function to make a flat list of any value.

Author: kovacsv

Sources/NodeEngine/NE_Value.cpp

@@ -212,4 +212,13 @@ void FlatEnumerate (const ValueConstPtr& value, const std::function<void (const
 	});
 }
 
+ValueConstPtr FlattenValue (const ValueConstPtr& value)
+{
+	ListValuePtr listValue (new ListValue ());
+	FlatEnumerate (value, [&] (const ValueConstPtr& value) {
+		listValue->Push (value);
+	});
+	return listValue;
+}
+
 }

Sources/NodeEngine/NE_Value.hpp

@@ -208,6 +208,7 @@ bool IsComplexType (const ValueConstPtr& val)
 ValueConstPtr		CreateSingleValue (const ValueConstPtr& value);
 IListValueConstPtr	CreateListValue (const ValueConstPtr& value);
 void				FlatEnumerate (const ValueConstPtr& value, const std::function<void (const ValueConstPtr&)>& processor);
+ValueConstPtr		FlattenValue (const ValueConstPtr& value);
 
 }
 

Sources/NodeEngineTest/ValueTest.cpp

@@ -165,4 +165,25 @@ TEST (ConstNonConstTest)
 	ASSERT (IntValue::Get (cval) == 5);
 }
 
+TEST (FlattenTest)
+{
+	ListValuePtr theListValue (new ListValue ());
+
+	ValuePtr innerSingleValue (new IntValue (1));
+	ListValuePtr innerListValue (new ListValue ());
+	innerListValue->Push (ValuePtr (new IntValue (2)));
+	innerListValue->Push (ValuePtr (new IntValue (3)));
+
+	theListValue->Push (innerSingleValue);
+	theListValue->Push (innerListValue);
+
+	ValueConstPtr flatten = FlattenValue (theListValue);
+	ASSERT (Value::IsType<ListValue> (flatten));
+	ListValueConstPtr flattenList = Value::Cast<ListValue> (flatten);
+	ASSERT (flattenList->GetSize () == 3);
+	ASSERT (IntValue::Get (flattenList->GetValue (0)) == 1);
+	ASSERT (IntValue::Get (flattenList->GetValue (1)) == 2);
+	ASSERT (IntValue::Get (flattenList->GetValue (2)) == 3);
+}
+
 }