Restore tslist for trace info( fix #295 )

nillerusr · Oct 5, 2023 · 6447101 · 6447101
1 parent efeda62
commit 6447101
Showing 1 changed file with 19 additions and 3 deletions.
diff --git a/engine/cmodel.cpp b/engine/cmodel.cpp
@@ -50,18 +50,29 @@ csurface_t *CCollisionBSPData::GetSurfaceAtIndex( unsigned short surfaceIndex )
 	return &map_surfaces[surfaceIndex];
 }
 
+#if TEST_TRACE_POOL
+CTSPool<TraceInfo_t> g_TraceInfoPool;
+#else
 class CTraceInfoPool : public CTSList<TraceInfo_t *>
 {
 public:
 	CTraceInfoPool() = default;
 };
 
-TraceInfo_t g_TraceInfo;
+CTraceInfoPool g_TraceInfoPool;
+#endif
 
 TraceInfo_t *BeginTrace()
 {
-	TraceInfo_t * pTraceInfo = &g_TraceInfo;
-
+#if TEST_TRACE_POOL
+	TraceInfo_t *pTraceInfo = g_TraceInfoPool.GetObject();
+#else
+	TraceInfo_t *pTraceInfo;
+	if ( !g_TraceInfoPool.PopItem( &pTraceInfo ) )
+	{
+		pTraceInfo = new TraceInfo_t;
+	}
+#endif
 	if ( pTraceInfo->m_BrushCounters[0].Count() != GetCollisionBSPData()->numbrushes + 1 )
 	{
 		memset( pTraceInfo->m_Count, 0, sizeof( pTraceInfo->m_Count ) );
@@ -107,6 +118,11 @@ void EndTrace( TraceInfo_t *&pTraceInfo )
 {
 	PopTraceVisits( pTraceInfo );
 	Assert( pTraceInfo->m_nCheckDepth == -1 );
+#if TEST_TRACE_POOL
+	g_TraceInfoPool.PutObject( pTraceInfo );
+#else
+	g_TraceInfoPool.PushItem( pTraceInfo );
+#endif
 	pTraceInfo = NULL;
 }