Added FloraPlacer FlatRegionFunction and //flora.

src/main/java/com/sk89q/worldedit/generator/FloraPlacer.java

@@ -0,0 +1,95 @@
+/*
+ * WorldEdit, a Minecraft world manipulation toolkit
+ * Copyright (C) sk89q <http://www.sk89q.com>
+ * Copyright (C) WorldEdit team and contributors
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package com.sk89q.worldedit.generator;
+
+import com.sk89q.worldedit.EditSession;
+import com.sk89q.worldedit.Vector;
+import com.sk89q.worldedit.WorldEditException;
+import com.sk89q.worldedit.blocks.BaseBlock;
+import com.sk89q.worldedit.blocks.BlockID;
+import com.sk89q.worldedit.patterns.BlockChance;
+import com.sk89q.worldedit.patterns.Pattern;
+import com.sk89q.worldedit.patterns.RandomFillPattern;
+
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * Generates flora over an applied area.
+ */
+public class FloraPlacer extends GroundGenerator {
+
+    private final EditSession editSession;
+    private final Pattern desertPattern = getDesertPattern();
+    private final Pattern temperatePattern = getTemperatePattern();
+
+    /**
+     * Create a new instance.
+     *
+     * @param editSession the edit session
+     * @param lowerY the lower Y
+     * @param upperY the upper Y (lowerY <= upperY)
+     */
+    public FloraPlacer(EditSession editSession, int lowerY, int upperY) {
+        super(editSession, lowerY, upperY);
+        this.editSession = editSession;
+    }
+
+    /**
+     * Get a pattern for plants to place inside a desert environment.
+     *
+     * @return a pattern that places flora
+     */
+    public static Pattern getDesertPattern() {
+        List<BlockChance> chance = new ArrayList<BlockChance>();
+        chance.add(new BlockChance(new BaseBlock(BlockID.DEAD_BUSH), 30));
+        chance.add(new BlockChance(new BaseBlock(BlockID.CACTUS), 20));
+        chance.add(new BlockChance(new BaseBlock(BlockID.AIR), 300));
+        return new RandomFillPattern(chance);
+    }
+
+    /**
+     * Get a pattern for plants to place inside a temperate environment.
+     *
+     * @return a pattern that places flora
+     */
+    public static Pattern getTemperatePattern() {
+        List<BlockChance> chance = new ArrayList<BlockChance>();
+        chance.add(new BlockChance(new BaseBlock(BlockID.LONG_GRASS, 1), 300));
+        chance.add(new BlockChance(new BaseBlock(BlockID.RED_FLOWER), 5));
+        chance.add(new BlockChance(new BaseBlock(BlockID.RED_FLOWER, 1), 1));
+        chance.add(new BlockChance(new BaseBlock(BlockID.YELLOW_FLOWER), 5));
+        return new RandomFillPattern(chance);
+    }
+
+    @Override
+    protected boolean apply(Vector pt, BaseBlock block) throws WorldEditException {
+        if (block.getType() == BlockID.GRASS) {
+            editSession.setBlock(pt.add(0, 1, 0), temperatePattern.next(pt));
+            return true;
+        } else if (block.getType() == BlockID.SAND) {
+            editSession.setBlock(pt.add(0, 1, 0), desertPattern.next(pt));
+            return true;
+        }
+
+        return false;
+    }
+
+}

src/main/java/com/sk89q/worldedit/generator/ForestGenerator.java

@@ -22,21 +22,18 @@ package com.sk89q.worldedit.generator;
 import com.sk89q.worldedit.EditSession;
 import com.sk89q.worldedit.MaxChangedBlocksException;
 import com.sk89q.worldedit.Vector;
-import com.sk89q.worldedit.Vector2D;
 import com.sk89q.worldedit.blocks.BaseBlock;
 import com.sk89q.worldedit.blocks.BlockID;
-import com.sk89q.worldedit.operation.GroundFindingFunction;
 import com.sk89q.worldedit.util.TreeGenerator;
 
 /**
  * Generates forests by searching for the ground starting from the given upper Y
  * coordinate for every column given.
  */
-public class ForestGenerator extends GroundFindingFunction {
+public class ForestGenerator extends GroundGenerator {
 
     private final TreeGenerator treeGenerator;
     private final EditSession editSession;
-    private double density;
 
     /**
      * Create a new instance.
@@ -52,31 +49,6 @@ public class ForestGenerator extends GroundFindingFunction {
         this.treeGenerator = treeGenerator;
     }
 
-    /**
-     * Set the density (0 <= density <= 1) which indicates the percentage chance
-     * that a tree will spawn in each column.
-     *
-     * @return the density
-     */
-    public double getDensity() {
-        return density;
-    }
-
-    /**
-     * Get the density (0 <= density <= 1) which indicates the percentage chance
-     * that a tree will spawn in each column.
-     *
-     * @param density the density
-     */
-    public void setDensity(double density) {
-        this.density = density;
-    }
-
-    @Override
-    protected boolean shouldContinue(Vector2D pt) {
-        return Math.random() < density;
-    }
-
     @Override
     protected boolean apply(Vector pt, BaseBlock block) throws MaxChangedBlocksException {
         int t = block.getType();

src/main/java/com/sk89q/worldedit/generator/GroundGenerator.java

@@ -0,0 +1,69 @@
+/*
+ * WorldEdit, a Minecraft world manipulation toolkit
+ * Copyright (C) sk89q <http://www.sk89q.com>
+ * Copyright (C) WorldEdit team and contributors
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package com.sk89q.worldedit.generator;
+
+import com.sk89q.worldedit.EditSession;
+import com.sk89q.worldedit.Vector2D;
+import com.sk89q.worldedit.operation.GroundFindingFunction;
+
+/**
+ * An abstract implementation for generators.
+ */
+public abstract class GroundGenerator extends GroundFindingFunction {
+
+    private double density;
+
+    /**
+     * Create a new instance.
+     *
+     * @param editSession the edit session
+     * @param lowerY the lower Y
+     * @param upperY the upper Y (lowerY <= upperY)
+     */
+    protected GroundGenerator(EditSession editSession, int lowerY, int upperY) {
+        super(editSession, lowerY, upperY);
+    }
+
+    /**
+     * Set the density (0 <= density <= 1) which indicates the percentage chance
+     * that an object will spawn in each column.
+     *
+     * @return the density
+     */
+    public double getDensity() {
+        return density;
+    }
+
+    /**
+     * Get the density (0 <= density <= 1) which indicates the percentage chance
+     * that an object will spawn in each column.
+     *
+     * @param density the density
+     */
+    public void setDensity(double density) {
+        this.density = density;
+    }
+
+    @Override
+    protected boolean shouldContinue(Vector2D pt) {
+        return Math.random() < density;
+    }
+
+}