summaryrefslogtreecommitdiff
path: root/utils/classcheck/class.h
blob: ff1935cc60982c249cd11cf19a25356b75d83bbb (plain) (blame)
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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose: 
//
// $NoKeywords: $
//=============================================================================//
#if !defined( CLASS_H )
#define CLASS_H
#ifdef _WIN32
#pragma once
#endif


class CTypeDescriptionField
{
public:
	CTypeDescriptionField()
	{
		m_szVariableName[ 0 ] = 0;
		m_szType[ 0 ] = 0;
		m_szDefineType[ 0 ] = 0;
		m_bCommentedOut = false;
		m_bRepresentedInRecvTable = false;
	}

	char			m_szVariableName[ 128 ];
	char			m_szType[ 128 ];
	char			m_szDefineType[ 128 ];
	bool			m_bCommentedOut;
	bool			m_bRepresentedInRecvTable;
};

class CClassVariable
{
public:
	CClassVariable()
	{
		m_szName[ 0 ] = 0;
		m_szType[ 0 ] = 0;

		m_Type = TPUBLIC;
		
		m_bKnownType = false;
		m_nTypeSize = 0;

		m_bIsArray = false;
		m_szArraySize[ 0 ] = 0;

		m_bInRecvTable = false;

		m_TypeSize = 0;
	}

	typedef enum 
	{
		TPUBLIC = 0,
		TPROTECTED,
		TPRIVATE
	} VARTYPE;

	char			m_szName[ 128 ];
	char			m_szType[ 128 ];

	VARTYPE			m_Type;
	
	bool			m_bKnownType;
	int				m_nTypeSize;

	bool			m_bIsArray;
	char			m_szArraySize[ 128 ];

	bool			m_bInRecvTable;

	int				m_TypeSize;
};

class CClassMemberFunction
{
public:
	typedef enum 
	{
		TPUBLIC = 0,
		TPROTECTED,
		TPRIVATE
	} MEMBERTYPE;

	char			m_szName[ 128 ];
	// Return type
	char			m_szType[ 128 ];

	MEMBERTYPE		m_Type;

};

class CClassTypedef
{
public:
	char			m_szTypeName[ 128 ];
	char			m_szAlias[ 128 ];

//	bool			m_bIsTypedefForBaseClass;
};

class CClass
{
public:
	enum
	{
		MAX_VARIABLES = 1024,
		MAX_MEMBERS   = 1024,
		MAX_TDFIELDS  = 1024,
	};

					CClass( const char *name );
					~CClass( void );
	
	char			*ParseClassDeclaration( char *input );


	void			SetBaseClass( const char *name );
	void			CheckChildOfBaseEntity( const char *baseentityclass );
	bool			CheckForMissingTypeDescriptionFields( int& missingcount, bool createtds = false );
	bool			CheckForMissingPredictionFields( int& missingcount, bool createtds = false );
	bool			CheckForPredictionFieldsInRecvTableNotMarkedAsSuchCorrectly( int& missingcount );
	void			AddVariable( int protection, char *type, char *name, bool array, char *arraysize = 0 );

	// Parsing helper methods
	bool			ParseProtection( char *&input, int &protection );
	bool			ParseNestedClass( char *&input );
	bool			ParseBaseClass( char *&input );
	bool			ParseClassMember( char *&input, int protection );
	bool			ParseNetworkVar( char *&input, int protection );
	void			ReportTypeMismatches( CClassVariable *var, CTypeDescriptionField *td );

	void			CheckForHungarianErrors( int& warnings );

	char			m_szName[ 128 ];
	char			m_szBaseClass[ 128 ];
	char			m_szTypedefBaseClass[ 128 ];

	CClassVariable	*FindVar( const char *name, bool checkbaseclasses = false );
	CClassVariable	*AddVar( const char *name );
	int				m_nVarCount;
	CClassVariable			*m_Variables[ MAX_VARIABLES ];

	CClassMemberFunction *FindMember( const char *name );
	CClassMemberFunction *AddMember( const char *name );
	int				m_nMemberCount;
	CClassMemberFunction	*m_Members[ MAX_MEMBERS ];

	CTypeDescriptionField	*FindTD( const char *name );
	CTypeDescriptionField   *AddTD(  const char *name, const char *type, const char *definetype, bool incomments );
	int						m_nTDCount;
	CTypeDescriptionField	*m_TDFields[ MAX_TDFIELDS ];

	CTypeDescriptionField	*FindPredTD( const char *name );
	CTypeDescriptionField   *AddPredTD(  const char *name, const char *type, const char *definetype, bool incomments, bool inrecvtable );
	int						m_nPredTDCount;
	CTypeDescriptionField	*m_PredTDFields[ MAX_TDFIELDS ];


	CClass			*m_pBaseClass;

	CClass			*m_pNext;

	bool			m_bDerivedFromCBaseEntity;
	bool			m_bHasSaveRestoreData;
	bool			m_bHasPredictionData;
	bool			m_bHasRecvTableData;
	bool			m_bConstructPredictableCalled;

	int				m_nClassDataSize;

private:
	struct MemberVarParse_t
	{
		char m_pType[256];
		char m_pTypeModifier[256];
		char m_pName[256];
		char m_pArraySize[ 128 ];
		bool m_bArray;

		MemberVarParse_t() { Reset(); }

		void Reset()
		{
			m_pType[0] = 0;
			m_pTypeModifier[0] = 0;
			m_pName[0] = 0;
			m_pArraySize[0] = 0;
			m_bArray = false;
		}
	};
};

#endif // CLASS_H