1HEADmaster

1 files changed, 129 insertions, 0 deletions

diff --git a/utils/classcheck/codeprocessor.h b/utils/classcheck/codeprocessor.h
new file mode 100644
index 0000000..d50b8d2
--- /dev/null
+++ b/utils/classcheck/codeprocessor.h
@@ -0,0 +1,129 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: 
+//
+// $NoKeywords: $
+//=============================================================================//
+#if !defined( CODEPROCESSOR_H )
+#define CODEPROCESSOR_H
+#ifdef _WIN32
+#pragma once
+#endif
+
+#include "class.h"
+#include "icodeprocessor.h"
+
+#include "utldict.h"
+
+class CCodeProcessor : public ICodeProcessor
+{
+public:
+				CCodeProcessor( void );
+				~CCodeProcessor( void );
+
+	void		Process( const char *baseentityclass, const char *gamespecific, const char *sourcetreebase, const char *subdir );
+	void		Process( const char *baseentityclass, const char *gamespecific, const char *sourcetreebase, const char *subdir, const char *pFileName );
+	CClass		*AddClass( const char *classname );
+
+	void		SetQuiet( bool quiet );
+	bool		GetQuiet( void ) const;
+
+	void		SetPrintHierarchy( bool print );
+	bool		GetPrintHierarchy( void ) const;
+
+	void		SetPrintMembers( bool print );
+	bool		GetPrintMembers( void ) const;
+
+	void		SetPrintTDs( bool print );
+	bool		GetPrintTDs( void ) const;
+
+	void		SetPrintPredTDs( bool print );
+	bool		GetPrintPredTDs( void ) const;
+
+	void		SetPrintCreateMissingTDs( bool print );
+	bool		GetPrintCreateMissingTDs( void ) const;
+
+	void		SetPrintCreateMissingPredTDs( bool print );
+	bool		GetPrintCreateMissingPredTDs( void ) const;
+
+	void		SetLogFile( bool log );
+	bool		GetLogFile( void ) const;
+
+	void		SetCheckHungarian( bool check );
+	bool		GetCheckHungarian() const;
+
+	CClass		*FindClass( const char *name ) const;
+	
+	void		ResolveBaseClasses( const char *baseentityclass );
+	void		Clear( void );
+	void		PrintClassList( void ) const;
+	void		PrintMissingTDFields( void ) const;
+	void		ReportHungarianNotationErrors();
+	
+	int			Count( void ) const;
+	
+	void		SortClassList( void );
+
+private:
+	char		*ParseTypeDescription( char *current, bool fIsMacroized );
+	char		*ParsePredictionTypeDescription( char *current );
+	char		*ParseReceiveTable( char *current );
+
+	void		ProcessModule( bool forcequiet, int depth, int& maxdepth, int& numheaders, int& skippedfiles, 
+		const char *srcroot, const char *baseroot, const char *root, const char *module );
+	void		ProcessModules( const char *srcroot, const char *root, const char *rootmodule );
+	void		PrintResults( const char *baseentityclass );
+	void		ConstructModuleList_R( int level, const char *baseentityclass, const char *gamespecific, const char *root, char const *srcroot );
+	
+	void		AddHeader( int depth, const char *filename, const char *rootmodule );
+
+	bool		CheckShouldSkip( bool forcequiet, int depth, char const *filename, int& numheaders, int& skippedfiles);
+	bool		LoadFile( char **buffer, char *filename, char const *module, bool forcequiet, 
+					int depth, int& filelength, int& numheaders, int& skippedfiles,
+					char const *srcroot, char const *root, char const *baseroot );
+
+	// include file path
+	void		CleanupIncludePath();
+	void		AddIncludePath( const char *pPath );
+	void		SetupIncludePath( const char *sourcetreebase, const char *subdir, const char *gamespecific );
+
+	CClass		*m_pClassList;
+
+	typedef struct
+	{
+		bool	skipped;
+	} CODE_MODULE;
+
+	CUtlDict< CODE_MODULE, int >	m_Modules;
+	CUtlDict< CODE_MODULE, int >	m_Headers;
+
+	CUtlVector< char * >	m_IncludePath;
+
+	bool		m_bQuiet;
+	bool		m_bPrintHierarchy;
+	bool		m_bPrintMembers;
+	bool		m_bPrintTypedescriptionErrors;
+	bool		m_bPrintPredictionDescErrors;
+	bool		m_bCreateMissingTDs;
+	bool		m_bCreateMissingPredTDs;
+	bool		m_bLogToFile;
+	bool		m_bCheckHungarian;
+
+	int			m_nFilesProcessed;
+	int			m_nHeadersProcessed;
+	int			m_nClassesParsed;
+	int			m_nOffset;
+	int			m_nBytesProcessed;
+	int			m_nLinesOfCode;
+
+	double		m_flStart;
+	double		m_flEnd;
+
+	char		m_szCurrentCPP[ 128 ];
+	char		m_szBaseEntityClass[ 256 ];
+
+};
+
+extern ICodeProcessor *processor;
+
+#endif // CODEPROCESSOR_H
\ No newline at end of file