X-Git-Url: https://ruin.nu/git/?p=germs.git;a=blobdiff_plain;f=src%2Fcomponenttree.h;fp=src%2Fcomponenttree.h;h=816059c29b7fe3c14463bbaf023c905b2907f834;hp=aad4ddbdd076cc4c6c5cbb42437f1c61199696fd;hb=a16f3a90b540c9adf29050f8041baf0c0781c20a;hpb=e3b0df677271c6cf14e00e2344802bfd0a66ebd5

diff --git a/src/componenttree.h b/src/componenttree.h
index aad4ddb..816059c 100644
--- a/src/componenttree.h
+++ b/src/componenttree.h
@@ -22,6 +22,7 @@
 #define __COMPONENTTREE_H__
 
 #include <vector>
+#include <map>
 #include "misc.h"
 
 class ComponentTree {
@@ -50,6 +51,8 @@ class ComponentTree {
 
 		size_t countLeaves();
 
+		size_t shortBranches();
+
 	private:
 		//Disable these, at least for now.
 		void operator=(const ComponentTree&){};
@@ -57,6 +60,7 @@ class ComponentTree {
 
 		void removeOriented(Node* n);
 		size_t countLeaves(Node* n);
+		void branches (Node* n, std::map<Node*,size_t> & b);
 
 		Node* _root;