GafferModule : Windows verifyAllocator()

Changes.md

@@ -5,7 +5,7 @@ Improvements
 ------------
 
 - Light Editor : Added `is_sphere` column for Cycles lights.
-
+- Windows : Gaffer now uses the TBB memory allocator for significantly better performance.
 
 
 1.5.0.0a3 (relative to 1.5.0.0a2)

bin/__gaffer.py

@@ -56,6 +56,9 @@
 
 import IECore
 
+if os.name == "nt" :
+	Gaffer._Gaffer._verifyAllocator()
+
 # Increase the soft limit for file handles as high as we can - we need everything we can get for
 # opening models, textures etc.
 if os.name != "nt" :

src/GafferModule/GafferModule.cpp

@@ -93,6 +93,12 @@
 #include <sys/prctl.h>
 #endif
 
+#ifdef _MSC_VER
+#include "IECore/MessageHandler.h"
+
+#include "tbb/tbbmalloc_proxy.h"
+#endif
+
 using namespace boost::python;
 using namespace Gaffer;
 using namespace GafferModule;
@@ -187,6 +193,25 @@ void nameProcess()
 #endif
 }
 
+#ifdef _MSC_VER
+void verifyAllocator()
+{
+
+	char **replacementLog;
+	int replacementStatus = TBB_malloc_replacement_log( &replacementLog );
+
+	if( replacementStatus != 0 )
+	{
+		IECore::msg( IECore::Msg::Warning, "Gaffer", "Failed to install TBB memory allocator. Performance may be degraded." );
+		for( char **logEntry = replacementLog; *logEntry != 0; logEntry++ )
+		{
+			IECore::msg( IECore::Msg::Warning, "Gaffer", *logEntry );
+		}
+	}
+
+}
+#endif
+
 } // namespace
 
 // Arrange for `storeArgcArgv()` to be called when our module loads,
@@ -257,5 +282,8 @@ BOOST_PYTHON_MODULE( _Gaffer )
 	def( "isDebug", &isDebug );
 
 	def( "_nameProcess", &nameProcess );
+#ifdef _MSC_VER
+	def( "_verifyAllocator", &verifyAllocator );
+#endif
 
 }