Modified source engine (2017) developed by valve and leaked in 2020. Not for commercial purporses
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

158 lines
3.9 KiB

5 years ago
//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
// $NoKeywords: $
//
//=============================================================================//
#include "togl/rendermechanism.h"
#include "recording.h"
#include "shaderapi/ishaderutil.h"
#include "materialsystem/imaterialsystem.h"
#include "shaderapidx8_global.h"
#include "utlvector.h"
#include <stdio.h>
// memdbgon must be the last include file in a .cpp file!!!
#include "tier0/memdbgon.h"
#ifdef RECORDING
//-----------------------------------------------------------------------------
// Globals
//-----------------------------------------------------------------------------
static CUtlVector<unsigned char> g_pRecordingBuffer;
static int g_ArgsRemaining = 0;
static int g_CommandStartIdx = 0;
//-----------------------------------------------------------------------------
// Opens the recording file
//-----------------------------------------------------------------------------
static FILE* OpenRecordingFile()
{
#ifdef CRASH_RECORDING
static FILE *fp = 0;
#else
FILE* fp = 0;
#endif
static bool g_CantOpenFile = false;
static bool g_NeverOpened = true;
if (!g_CantOpenFile)
{
#ifdef CRASH_RECORDING
if( g_NeverOpened )
{
fp = fopen( "shaderdx8.rec", "wbc" );
}
#else
fp = fopen( "shaderdx8.rec", g_NeverOpened ? "wb" : "ab" );
#endif
if (!fp)
{
Warning("Unable to open recording file shaderdx8.rec!\n");
g_CantOpenFile = true;
}
g_NeverOpened = false;
}
return fp;
}
//-----------------------------------------------------------------------------
// Writes to the recording file
//-----------------------------------------------------------------------------
#define COMMAND_BUFFER_SIZE 32768
static void WriteRecordingFile()
{
// Store the command size
*(int*)&g_pRecordingBuffer[g_CommandStartIdx] =
g_pRecordingBuffer.Size() - g_CommandStartIdx;
#ifndef CRASH_RECORDING
// When not crash recording, flush when buffer gets too big,
// or when Present() is called
if ((g_pRecordingBuffer.Size() < COMMAND_BUFFER_SIZE) &&
(g_pRecordingBuffer[g_CommandStartIdx+4] != DX8_PRESENT))
return;
#endif
FILE* fp = OpenRecordingFile();
if (fp)
{
// store the command size
fwrite( g_pRecordingBuffer.Base(), 1, g_pRecordingBuffer.Size(), fp );
fflush( fp );
#ifndef CRASH_RECORDING
fclose( fp );
#endif
}
g_pRecordingBuffer.RemoveAll();
}
// Write the buffered crap out on shutdown.
void FinishRecording()
{
#ifndef CRASH_RECORDING
FILE* fp = OpenRecordingFile();
if (fp)
{
// store the command size
fwrite( g_pRecordingBuffer.Base(), 1, g_pRecordingBuffer.Size(), fp );
fflush( fp );
}
g_pRecordingBuffer.RemoveAll();
#endif
}
// set this to true in the debugger to actually record commands.
static bool g_bDoRecord = true;
//-----------------------------------------------------------------------------
// Records a command
//-----------------------------------------------------------------------------
void RecordCommand( RecordingCommands_t cmd, int numargs )
{
if( !g_bDoRecord )
{
return;
}
Assert( g_ArgsRemaining == 0 );
g_CommandStartIdx = g_pRecordingBuffer.AddMultipleToTail( 6 );
// save space for the total command size
g_pRecordingBuffer[g_CommandStartIdx+4] = cmd;
g_pRecordingBuffer[g_CommandStartIdx+5] = numargs;
g_ArgsRemaining = numargs;
if (g_ArgsRemaining == 0)
WriteRecordingFile();
}
//-----------------------------------------------------------------------------
// Records an argument for a command, flushes when the command is done
//-----------------------------------------------------------------------------
void RecordArgument( void const* pMemory, int size )
{
if( !g_bDoRecord )
{
return;
}
Assert( g_ArgsRemaining > 0 );
int tail = g_pRecordingBuffer.Size();
g_pRecordingBuffer.AddMultipleToTail( size );
memcpy( &g_pRecordingBuffer[tail], pMemory, size );
if (--g_ArgsRemaining == 0)
WriteRecordingFile();
}
#endif // RECORDING