1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
// $NoKeywords: $
//
//=============================================================================//
#include "filememcache.h"
namespace {
unsigned long s_ulCachedFileSignature = 0xCACEF11E;
};
//
// Cached file data implementation
//
CachedFileData * CachedFileData::Create( char const *szFilename )
{
FILE *f = fopen( szFilename, "rb" );
int nSize = -1;
if ( f )
{
fseek( f, 0, SEEK_END );
nSize = ftell( f );
fseek( f, 0, SEEK_SET );
}
CachedFileData *pData = ( CachedFileData * ) malloc( eHeaderSize + max( nSize, 0 ) );
strcpy( pData->m_chFilename, szFilename );
pData->m_numRefs = 0;
pData->m_numDataBytes = nSize;
pData->m_signature = s_ulCachedFileSignature;
if ( f )
{
fread( pData->m_data, 1, nSize, f );
fclose( f );
}
return pData;
}
void CachedFileData::Free( void )
{
free( this );
}
CachedFileData *CachedFileData::GetByDataPtr( void const *pvDataPtr )
{
unsigned char const *pbBuffer = reinterpret_cast< unsigned char const * >( pvDataPtr );
// Assert( pbBuffer );
CachedFileData const *pData = reinterpret_cast< CachedFileData const * >( pbBuffer - eHeaderSize );
// Assert( pData->m_signature == s_ulCachedFileSignature );
return const_cast< CachedFileData * >( pData );
}
char const * CachedFileData::GetFileName() const
{
return m_chFilename;
}
void const * CachedFileData::GetDataPtr() const
{
return m_data;
}
int CachedFileData::GetDataLen() const
{
return max( m_numDataBytes, 0 );
}
bool CachedFileData::IsValid() const
{
return ( m_numDataBytes >= 0 );
}
//
// File cache implementation
//
FileCache::FileCache()
{
NULL;
}
CachedFileData * FileCache::Get( char const *szFilename )
{
// Search the cache first
Mapping::iterator it = m_map.find( szFilename );
if ( it != m_map.end() )
return it->second;
// Create the cached file data
CachedFileData *pData = CachedFileData::Create( szFilename );
if ( pData )
m_map.insert( Mapping::value_type( pData->GetFileName(), pData ) );
return pData;
}
void FileCache::Clear()
{
for ( Mapping::iterator it = m_map.begin(), itEnd = m_map.end(); it != itEnd; ++ it )
{
it->second->Free();
}
m_map.clear();
}
|