From 1be1d81fc1ad7553f74faeaf4646f2cbe1e7d8f3 Mon Sep 17 00:00:00 2001
From: David Vierra <codewarrior@hawaii.rr.com>
Date: Tue, 12 May 2015 23:44:29 -1000
Subject: [PATCH] Add TreeGen back in so GenerateTool doesn't crash on zero
 generator classes.

xxx allow zero generator classes
---
 src/mcedit2/editortools/generate.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/src/mcedit2/editortools/generate.py b/src/mcedit2/editortools/generate.py
index 05a1454..1cfc477 100644
--- a/src/mcedit2/editortools/generate.py
+++ b/src/mcedit2/editortools/generate.py
@@ -128,9 +128,9 @@ class TreeGen(GeneratePlugin):
         self.optionsWidget = widget
         return widget
 
-    def generate(self, bounds, blocktypes):
-        schematic = createSchematic(bounds.size, blocktypes)
-        dim = schematic.getDimension()
+    def generateInSchematic(self, dim):
+        bounds = dim.bounds
+        blocktypes = dim.blocktypes
 
         trunkX = int(bounds.width / 2)
         trunkZ = int(bounds.length / 2)
@@ -143,14 +143,13 @@ class TreeGen(GeneratePlugin):
         dim.setBlocks(range(0, bounds.width), [range(crownBottom, bounds.height)], [[range(0, bounds.length)]], leaves)
         dim.setBlocks(trunkX, range(0, bounds.height), trunkZ, wood)
 
-        return schematic
 
 _pluginClasses = []
 
 def registerGeneratePlugin(cls):
     _pluginClasses.append(cls)
 
-#_pluginClasses.append(TreeGen)
+_pluginClasses.append(TreeGen)
 
 class GenerateTool(EditorTool):
     name = "Generate"