Add: functions to manipulate string attributes

dkoerner · Jul 20, 2020 · d283a78 · d283a78
1 parent 5684f6b
commit d283a78
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 2 deletions.
diff --git a/include/houio/HouGeo.h b/include/houio/HouGeo.h
@@ -34,6 +34,13 @@ namespace houio
 			virtual void                          getPacking( std::vector<int> &packing )const;
 			virtual int                           getNumElements()const;
 			virtual std::string                   getString( int index )const;
+			virtual std::string                   getUniqueString( int index )const;
+			virtual int                           getStringIndex( int index )const;
+			virtual std::vector<std::string>      getUniqueStrings()const;
+			virtual std::vector<int>              getStringIndices()const;
+
+
+
 			virtual RawPointer::Ptr               getRawPointer();
 
 			//int                                   addV4f(math::V4f value);

diff --git a/src/HouGeo.cpp b/src/HouGeo.cpp
@@ -285,14 +285,38 @@ namespace houio
 		return numElements++;
 	}
 
+	std::string HouGeo::HouAttribute::getUniqueString( int index )const
+	{
+		if(strings.size() < index)
+			throw std::runtime_error("HouGeo::getString Index is out of strings range.");
+		return strings[index];
+	}
+
 	std::string HouGeo::HouAttribute::getString( int index )const
 	{
 		if(stringsIdxs.size() < index)
-			throw std::runtime_error("HouGeo::getString Index is out of stringsIdx range.");
-		int string_index = stringsIdxs[index];
+			throw std::runtime_error("HouGeo::getString Index is out of strings range.");
+		const int string_index = stringsIdxs[index];
 		return strings[string_index];
 	}
 
+	int HouGeo::HouAttribute::getStringIndex( int index )const
+	{
+		if(stringsIdxs.size() < index)
+			throw std::runtime_error("HouGeo::getStringIndex Index is out of stringsIdx range.");
+		return stringsIdxs[index];
+	}
+
+	std::vector<std::string> HouGeo::HouAttribute::getUniqueStrings()const
+	{
+		return strings;
+	}
+
+	std::vector<int> HouGeo::HouAttribute::getStringIndices()const
+	{
+		return stringsIdxs;
+	}
+