class="kw">package com.hypixel.hytale.server.worldgen.cave.shape.distorted;

class="kw">import com.hypixel.hytale.server.worldgen.cave.shape.CaveNodeShapeEnum;
class="kw">import java.util.Map;
class="kw">import java.util.concurrent.ConcurrentHashMap;
class="kw">import java.util.function.BiConsumer;
class="kw">import java.util.function.Consumer;
class="kw">import javax.annotation.Nonnull;
class="kw">import javax.annotation.Nullable;

class="kw">public class="kw">final class DistortedShapes {
   class="kw">public class="kw">static class="kw">final DistortedShape.Factory CYLINDER = new DistortedCylinderShape.Factory();
   class="kw">public class="kw">static class="kw">final DistortedShape.Factory ELLIPSE = new DistortedEllipsoidShape.Factory();
   class="kw">public class="kw">static class="kw">final DistortedShape.Factory PIPE = new DistortedPipeShape.Factory();
   class="kw">private class="kw">static class="kw">final Map<String, DistortedShape.Factory> SHAPES = new ConcurrentHashMap<>();

   class="kw">private DistortedShapes() {
   }

   class="kw">public class="kw">static void register(String var0, DistortedShape.Factory var1) {
      SHAPES.putIfAbsent(var0, var1);
   }

   class="kw">public class="kw">static void forEach(BiConsumer<String, DistortedShape.Factory> var0) {
      SHAPES.forEach(var0);
   }

   class="kw">public class="kw">static void forEachName(Consumer<String> var0) {
      SHAPES.keySet().forEach(var0);
   }

   class="kw">public class="kw">static void forEachShape(Consumer<DistortedShape.Factory> var0) {
      SHAPES.values().forEach(var0);
   }

   @Nonnull
   class="kw">public class="kw">static DistortedShape.Factory getDefault() {
      class="kw">return PIPE;
   }

   @Nonnull
   class="kw">public class="kw">static DistortedShape.Factory getOrDefault(String var0) {
      DistortedShape.Factory var1 = SHAPES.get(var0);
      class="kw">return var1 == null ? getDefault() : var1;
   }

   @Nullable
   class="kw">public class="kw">static DistortedShape.Factory getByName(String var0) {
      class="kw">return SHAPES.get(var0);
   }

   class="kw">static {
      register(CaveNodeShapeEnum.PIPE.name(), PIPE);
      register(CaveNodeShapeEnum.CYLINDER.name(), CYLINDER);
      register(CaveNodeShapeEnum.ELLIPSOID.name(), ELLIPSE);
   }
}