From db863422b8a49bbd43fc0204e36f7a2bdc51a674 Mon Sep 17 00:00:00 2001
From: Mattes D <github@xoft.cz>
Date: Mon, 8 Jun 2015 21:52:13 +0200
Subject: cBlockArea: Added CountSpecificBlocks() API function.

---
 src/BlockArea.cpp | 59 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 src/BlockArea.h   |  9 +++++++++
 2 files changed, 68 insertions(+)

(limited to 'src')

diff --git a/src/BlockArea.cpp b/src/BlockArea.cpp
index 7982afc31..938351207 100644
--- a/src/BlockArea.cpp
+++ b/src/BlockArea.cpp
@@ -1665,6 +1665,65 @@ size_t cBlockArea::CountNonAirBlocks(void) const
 
 
 
+size_t cBlockArea::CountSpecificBlocks(BLOCKTYPE a_BlockType) const
+{
+	// If blocktypes are not valid, log a warning and return zero occurences:
+	if (m_BlockTypes == nullptr)
+	{
+		LOGWARNING("%s: BlockTypes not available!", __FUNCTION__);
+		return 0;
+	}
+
+	// Count the blocks:
+	size_t num = GetBlockCount();
+	size_t res = 0;
+	for (size_t i = 0; i < num; i++)
+	{
+		if (m_BlockTypes[i] == a_BlockType)
+		{
+			res++;
+		}
+	}
+	return res;
+}
+
+
+
+
+
+size_t cBlockArea::CountSpecificBlocks(BLOCKTYPE a_BlockType, NIBBLETYPE a_BlockMeta) const
+{
+	// If blocktypes are not valid, log a warning and return zero occurences:
+	if (m_BlockTypes == nullptr)
+	{
+		LOGWARNING("%s: BlockTypes not available!", __FUNCTION__);
+		return 0;
+	}
+
+	// If blockmetas are not valid, log a warning and count only blocktypes:
+	if (m_BlockMetas == nullptr)
+	{
+		LOGWARNING("%s: BlockMetas not available, comparing blocktypes only!", __FUNCTION__);
+		return CountSpecificBlocks(a_BlockType);
+	}
+
+	// Count the blocks:
+	size_t num = GetBlockCount();
+	size_t res = 0;
+	for (size_t i = 0; i < num; i++)
+	{
+		if ((m_BlockTypes[i] == a_BlockType) && (m_BlockMetas[i] == a_BlockMeta))
+		{
+			res++;
+		}
+	}
+	return res;
+}
+
+
+
+
+
 void cBlockArea::GetNonAirCropRelCoords(int & a_MinRelX, int & a_MinRelY, int & a_MinRelZ, int & a_MaxRelX, int & a_MaxRelY, int & a_MaxRelZ, BLOCKTYPE a_IgnoreBlockType)
 {
 	// Check if blocktypes are valid:
diff --git a/src/BlockArea.h b/src/BlockArea.h
index 4b672029b..a9963b4ef 100644
--- a/src/BlockArea.h
+++ b/src/BlockArea.h
@@ -308,6 +308,15 @@ public:
 	Returns 0 if blocktypes not available. Block metas are ignored (if present, air with any meta is still considered air). */
 	size_t CountNonAirBlocks(void) const;
 
+	/** Returns how many times the specified block is contained in the area.
+	The blocks' meta values are ignored, only the blocktype is compared. */
+	size_t CountSpecificBlocks(BLOCKTYPE a_BlockType) const;
+
+	/** Returns how many times the specified block is contained in the area.
+	Both the block's type and meta must match in order to be counted in.
+	If the block metas aren't present in the area, logs a warning and ignores the meta specification. */
+	size_t CountSpecificBlocks(BLOCKTYPE a_BlockType, NIBBLETYPE a_BlockMeta) const;
+
 	// tolua_end
 
 	/** Returns the minimum and maximum coords in each direction for the first non-ignored block in each direction.
-- 
cgit v1.2.3