optimised the tree transclude directive

Author: Matt Evans

angular-tree-control.js

@@ -295,13 +295,22 @@
             return {
                 link: function(scope, element, attrs, controller) {
                     if (!scope.options.isLeaf(scope.node)) {
-                        angular.forEach(scope.expandedNodesMap, function (node, id) {
-                            if (scope.options.equality(node, scope.node)) {
-                                scope.expandedNodesMap[scope.$id] = scope.node;
-                                scope.expandedNodesMap['x'+id] = undefined;
+                        var exchangeNodesInMap = function (node, id) {
+                            scope.expandedNodesMap[scope.$id] = scope.node;
+                            scope.expandedNodesMap["x"+id] = undefined;
+                        };
+
+                        for(var key in scope.expandedNodesMap) {
+                            if (scope.expandedNodesMap.hasOwnProperty(key)) {
+                                var node = scope.expandedNodesMap[key];
+                                if (scope.options.equality(node, scope.node)) {
+                                    exchangeNodesInMap(node, key);
+                                    break;
+                                }
                             }
-                        });
-                    }
+                        }
+					}
+
                     if (!scope.options.multiSelection && scope.options.equality(scope.node, scope.selectedNode)) {
                         scope.selectedNode = scope.node;
                     } else if (scope.options.multiSelection) {