[2018-03-30 16:27:22] update BFS in treeMap script

hijiangtao · Mar 31, 2018 · eb70592 · eb70592
1 parent 68936b3
commit eb70592
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/util/tripFlow/constructTreeMap.py b/util/tripFlow/constructTreeMap.py
@@ -291,15 +291,15 @@ def BFSOneTreeMap(self, parentNode, recordNum=0, treeQueue=[], currentDis=0):
 			if treeStructID in self.keepTreeStructList[cateName]:
 				continue
 
-			self.treeNodesID += 1
-			self.currentData[cateName]['count'] += 1 
-
 			# 距离判断 START
 			dis = currentDis + getRealDistance(parentNode[0], parentNode[1], vertex[0], vertex[1])
 			if dis > self.custom_params['max_distance'] * 1000:
 				continue
 			# 距离判断 END
 
+			self.treeNodesID += 1
+			self.currentData[cateName]['count'] += 1 
+
 			subres = {
 				"root": {
 					"id": self.treeNodesID,