Add module-dependent overlays

src/main/java/org/terasology/world/viewer/overlay/GraphNodeOverlay.java

@@ -0,0 +1,181 @@
+/*
+ * Copyright 2015 MovingBlocks
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *  http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.terasology.world.viewer.overlay;
+
+import java.awt.BasicStroke;
+import java.awt.Color;
+import java.awt.Graphics2D;
+import java.awt.Shape;
+import java.awt.geom.Area;
+import java.awt.geom.Ellipse2D;
+import java.awt.geom.Path2D;
+import java.util.Iterator;
+import java.util.function.Function;
+
+import org.terasology.math.geom.BaseVector2f;
+import org.terasology.math.geom.ImmutableVector2f;
+import org.terasology.math.geom.ImmutableVector2i;
+import org.terasology.math.geom.Rect2i;
+import org.terasology.math.geom.Vector2f;
+import org.terasology.polyworld.graph.Corner;
+import org.terasology.polyworld.graph.GraphFacet;
+import org.terasology.polyworld.graph.Triangle;
+import org.terasology.world.generation.Region;
+import org.terasology.world.viewer.picker.CirclePickerClosest;
+
+/**
+ * Shows the triangle under the mouse cursor.
+ */
+public class GraphNodeOverlay extends AbstractOverlay implements WorldOverlay {
+
+    private final Function<ImmutableVector2i, Region> regions;
+
+    private Color fillColor = new Color(192, 32, 32, 128);
+    private Color frameColor = new Color(192, 32, 32, 192);
+    private BasicStroke stroke = new BasicStroke(0.2f, BasicStroke.CAP_ROUND, BasicStroke.JOIN_ROUND);
+
+    public enum Mode {
+        CORNER,
+        TRIANGLE,
+        REGION
+    }
+
+    private Mode mode = Mode.CORNER;
+
+    public GraphNodeOverlay(Function<ImmutableVector2i, Region> regionFunc) {
+        regions = regionFunc;
+    }
+
+    @Override
+    public void render(Graphics2D g, Rect2i area, ImmutableVector2i cursor) {
+        if (cursor == null) {
+            return;
+        }
+
+        Region r = regions.apply(cursor);
+        GraphFacet graphs = r.getFacet(GraphFacet.class);
+        g.setStroke(stroke);
+        if (graphs != null) {
+            Triangle t = graphs.getWorldTriangle(cursor.getX(), cursor.getY());
+            Shape path = null;
+
+            switch (mode) {
+                case CORNER:
+                    Vector2f cursorf = new Vector2f(cursor.getX(), cursor.getY());
+                    CirclePickerClosest<Corner> picker = new CirclePickerClosest<>(cursorf, c -> 3);
+
+                    picker.offer(t.getCorner1().getLocation(), t.getCorner1());
+                    picker.offer(t.getCorner2().getLocation(), t.getCorner2());
+
+                    if (picker.getClosest() != null) {
+                        Corner c = picker.getClosest();
+                        path = createCornerPath(c);
+                    } else {
+                        path = new Path2D.Float();
+                    }
+                    break;
+
+                case TRIANGLE:
+                    path = createTrianglePath(t);
+                    break;
+
+                case REGION:
+                    path = createRegionPath(t.getRegion());
+                    break;
+            }
+
+            g.setColor(fillColor);
+            g.fill(path);
+            g.setColor(frameColor);
+            g.draw(path);
+        }
+    }
+
+    public Mode getMode() {
+        return mode;
+    }
+
+    public void setMode(Mode mode) {
+        this.mode = mode;
+    }
+
+    private Shape createCornerPath(Corner c) {
+        Path2D path = new Path2D.Float();
+
+        float radCenter = 3;
+        float radAdj = 2;
+
+        float cx = c.getLocation().getX();
+        float cy = c.getLocation().getY();
+
+        for (Corner adj : c.getAdjacent()) {
+            float adx = adj.getLocation().getX();
+            float ady = adj.getLocation().getY();
+            path.moveTo(cx, cy);
+            path.lineTo(adx, ady);
+        }
+
+        Area result = new Area(new BasicStroke(1f).createStrokedShape(path));
+        result.add(new Area(new Ellipse2D.Float(cx - radCenter, cy - radCenter, radCenter * 2, radCenter * 2)));
+
+        for (Corner adj : c.getAdjacent()) {
+            float adx = adj.getLocation().getX();
+            float ady = adj.getLocation().getY();
+            result.add(new Area(new Ellipse2D.Float(adx - radAdj, ady - radAdj, radAdj * 2, radAdj * 2)));
+        }
+
+        return result;
+    }
+
+    private Path2D createTrianglePath(Triangle t) {
+        BaseVector2f p0 = t.getRegion().getCenter();
+        BaseVector2f p1 = t.getCorner1().getLocation();
+        BaseVector2f p2 = t.getCorner2().getLocation();
+
+        Path2D path = new Path2D.Float();
+
+        path.moveTo(p0.getX(), p0.getY());
+        path.lineTo(p1.getX(), p1.getY());
+        path.lineTo(p2.getX(), p2.getY());
+        path.closePath();
+
+        return path;
+    }
+
+    private Path2D createRegionPath(org.terasology.polyworld.graph.Region region) {
+
+        Path2D path = new Path2D.Float();
+
+        Iterator<Corner> it = region.getCorners().iterator();
+
+        if (!it.hasNext()) {
+            return path;
+        }
+
+        ImmutableVector2f p = it.next().getLocation();
+        path.moveTo(p.getX(), p.getY());
+
+        while (it.hasNext()) {
+            p = it.next().getLocation();
+            path.lineTo(p.getX(), p.getY());
+        }
+
+        path.closePath();
+        return path;
+    }
+
+}

src/main/java/org/terasology/world/viewer/overlay/LakeOverlay.java

@@ -0,0 +1,63 @@
+/*
+ * Copyright 2015 MovingBlocks
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *  http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.terasology.world.viewer.overlay;
+
+import java.awt.BasicStroke;
+import java.awt.Color;
+import java.awt.Graphics2D;
+import java.awt.Polygon;
+import java.util.function.Function;
+
+import org.terasology.cities.lakes.Lake;
+import org.terasology.cities.lakes.LakeFacet;
+import org.terasology.math.geom.ImmutableVector2i;
+import org.terasology.math.geom.Rect2i;
+import org.terasology.world.generation.Region;
+
+public class LakeOverlay extends AbstractOverlay implements WorldOverlay {
+
+    private Function<ImmutableVector2i, Region> regions;
+
+    public LakeOverlay(Function<ImmutableVector2i, Region> regionFunc) {
+        regions = regionFunc;
+    }
+
+    @Override
+    public void render(Graphics2D g, Rect2i area, ImmutableVector2i cursor) {
+        if (cursor == null) {
+            return;
+        }
+
+        Color fillColor = new Color(64, 64, 255, 128);
+        Color frameColor = new Color(64, 64, 255, 224);
+
+        g.setStroke(new BasicStroke(1f, BasicStroke.CAP_SQUARE, BasicStroke.JOIN_ROUND));
+        Region r = regions.apply(cursor);
+        LakeFacet lakeFacet = r.getFacet(LakeFacet.class);
+        for (Lake lake : lakeFacet.getLakes()) {
+            if (lake.getContour().getPolygon().contains(cursor.getX(), cursor.getY())) {
+                Polygon poly = lake.getContour().getPolygon();
+                g.translate(0.5, 0.5);
+                g.setColor(fillColor);
+                g.fill(poly);
+                g.setColor(frameColor);
+                g.draw(poly);
+                g.translate(-0.5, -0.5);
+            }
+        }
+    }
+}