mirror of
https://github.com/Mauler125/r5sdk.git
synced 2025-02-09 19:15:03 +01:00
110 lines
2.3 KiB
C++
110 lines
2.3 KiB
C++
//========= Copyright <20> 1996-2006, Valve Corporation, All rights reserved. ============//
|
||
//
|
||
// Purpose: generates 4 random numbers in the range 0..1 quickly, using SIMD
|
||
//
|
||
//=====================================================================================//
|
||
|
||
#include "core/stdafx.h"
|
||
#include "tier0/dbg.h"
|
||
#include "tier0/threadtools.h"
|
||
#include "mathlib/mathlib.h"
|
||
#include "mathlib/vector.h"
|
||
#include "mathlib/ssemath.h"
|
||
|
||
// memdbgon must be the last include file in a .cpp file!!!
|
||
//#include "tier0/memdbgon.h"
|
||
|
||
// see knuth volume 3 for insight.
|
||
|
||
class SIMDRandStreamContext
|
||
{
|
||
fltx4 m_RandY[55];
|
||
|
||
fltx4* m_pRand_J, * m_pRand_K;
|
||
|
||
|
||
public:
|
||
void Seed(uint32 seed)
|
||
{
|
||
m_pRand_J = m_RandY + 23; m_pRand_K = m_RandY + 54;
|
||
for (int i = 0; i < 55; i++)
|
||
{
|
||
for (int j = 0; j < 4; j++)
|
||
{
|
||
SubFloat(m_RandY[i], j) = (seed >> 16) / 65536.0f;
|
||
seed = (seed + 1) * 3141592621u;
|
||
}
|
||
}
|
||
}
|
||
|
||
inline fltx4 RandSIMD(void)
|
||
{
|
||
// ret= rand[k]+rand[j]
|
||
fltx4 retval = AddSIMD(*m_pRand_K, *m_pRand_J);
|
||
|
||
// if ( ret>=1.0) ret-=1.0
|
||
bi32x4 overflow_mask = CmpGeSIMD(retval, Four_Ones);
|
||
retval = SubSIMD(retval, AndSIMD(Four_Ones, overflow_mask));
|
||
|
||
*m_pRand_K = retval;
|
||
|
||
// update pointers w/ wrap-around
|
||
if (--m_pRand_J < m_RandY)
|
||
m_pRand_J = m_RandY + 54;
|
||
if (--m_pRand_K < m_RandY)
|
||
m_pRand_K = m_RandY + 54;
|
||
|
||
return retval;
|
||
}
|
||
};
|
||
|
||
#define MAX_SIMULTANEOUS_RANDOM_STREAMS 32
|
||
|
||
static SIMDRandStreamContext s_SIMDRandContexts[MAX_SIMULTANEOUS_RANDOM_STREAMS];
|
||
|
||
static volatile LONG s_nRandContextsInUse[MAX_SIMULTANEOUS_RANDOM_STREAMS];
|
||
|
||
void SeedRandSIMD(uint32 seed)
|
||
{
|
||
for (int i = 0; i < MAX_SIMULTANEOUS_RANDOM_STREAMS; i++)
|
||
s_SIMDRandContexts[i].Seed(seed + i);
|
||
}
|
||
|
||
fltx4 RandSIMD(int nContextIndex)
|
||
{
|
||
return s_SIMDRandContexts[nContextIndex].RandSIMD();
|
||
}
|
||
|
||
int GetSIMDRandContext(void)
|
||
{
|
||
for (;;)
|
||
{
|
||
for (int i = 0; i < NELEMS(s_SIMDRandContexts); i++)
|
||
{
|
||
if (!s_nRandContextsInUse[i]) // available?
|
||
{
|
||
// try to take it!
|
||
if (ThreadInterlockedAssignIf(&(s_nRandContextsInUse[i]), 1, 0))
|
||
{
|
||
ThreadMemoryBarrier();
|
||
return i; // done!
|
||
}
|
||
}
|
||
}
|
||
Assert(0); // why don't we have enough buffers?
|
||
ThreadSleep(0);
|
||
}
|
||
}
|
||
|
||
void ReleaseSIMDRandContext(int nContext)
|
||
{
|
||
ThreadMemoryBarrier();
|
||
s_nRandContextsInUse[nContext] = 0;
|
||
}
|
||
|
||
|
||
fltx4 RandSIMD(void)
|
||
{
|
||
return s_SIMDRandContexts[0].RandSIMD();
|
||
}
|