summaryrefslogtreecommitdiffstats
path: root/Tools/AnvilStats/Processor.h
blob: 426e67d9158a99669015110858366b18b6d273e4 (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

// Processor.h

// Interfaces to the cProcessor class representing the overall processor engine that manages threads, calls callbacks
// etc.




#pragma once





// fwd:
class cCallback;
class cCallbackFactory;
class cParsedNBT;





class cProcessor
{
	class cThread : public cIsThread
	{
		typedef cIsThread super;

		cCallback & m_Callback;
		cProcessor & m_ParentProcessor;
		cEvent m_HasStarted;

		// cIsThread override:
		virtual void Execute(void) override;

		void ProcessFile(const AString & a_FileName);
		void ProcessFileData(const char * a_FileData, size_t a_Size, int a_ChunkBaseX, int a_ChunkBaseZ);
		void ProcessChunk(
			const char * a_FileData,
			int a_ChunkX,
			int a_ChunkZ,
			unsigned a_SectorStart,
			unsigned a_SectorSize,
			unsigned a_TimeStamp
		);
		void ProcessCompressedChunkData(
			int a_ChunkX,
			int a_ChunkZ,
			const char * a_CompressedData,
			int a_CompressedSize
		);
		void ProcessParsedChunkData(int a_ChunkX, int a_ChunkZ, cParsedNBT & a_NBT);

		// The following processing parts return true if they were interrupted by the callback, causing the processing
		// of current chunk to abort
		bool ProcessChunkSections(int a_ChunkX, int a_ChunkZ, cParsedNBT & a_NBT, int a_LevelTag);
		bool ProcessChunkEntities(int a_ChunkX, int a_ChunkZ, cParsedNBT & a_NBT, int a_LevelTag);
		bool ProcessChunkTileEntities(int a_ChunkX, int a_ChunkZ, cParsedNBT & a_NBT, int a_LevelTag);
		bool ProcessChunkTileTicks(int a_ChunkX, int a_ChunkZ, cParsedNBT & a_NBT, int a_LevelTag);

	  public:
		cThread(cCallback & a_Callback, cProcessor & a_ParentProcessor);

		/** Waits until the thread starts processing the callback code. */
		void WaitForStart(void);
	};

	typedef std::vector<cThread *> cThreads;

  public:
	cProcessor(void);
	~cProcessor();

	void ProcessWorld(const AString & a_WorldFolder, cCallbackFactory & a_CallbackFactory);

  protected:
	bool m_IsShuttingDown;  // If true, the threads should stop ASAP

	cCriticalSection m_CS;
	AStringList m_FileQueue;

	cThreads m_Threads;


	/** Populates m_FileQueue with Anvil files from the specified folder. */
	void PopulateFileQueue(const AString & a_WorldFolder);

	/** Returns one filename from m_FileQueue, and removes the name from the queue. */
	AString GetOneFileName(void);
};