AER-3393 Added SubReceptorMode option for OnlyInsideNatureAreas (#311)

source/imaer-shared/src/main/java/nl/overheid/aerius/shared/domain/calculation/SubReceptorsMode.java

@@ -16,6 +16,10 @@
  */
 package nl.overheid.aerius.shared.domain.calculation;
 
+import java.util.Arrays;
+import java.util.HashSet;
+import java.util.Set;
+
 /**
  * Different modes we support for calculations with subreceptors.
  */
@@ -24,30 +28,46 @@ public enum SubReceptorsMode {
   /**
    * Never use subreceptors
    */
-  DISABLED(false, false, true),
+  DISABLED(),
+  /**
+   * Always use subreceptors, but only inside relevant habitats.
+   */
+  ENABLED(Option.ENABLED, Option.ONLY_INSIDE_HABITATS),
   /**
    * Always use subreceptors
+   * Only include subreceptors when inside nature area
    */
-  ENABLED(true, false, true),
+  ENABLED_INSIDE_NATURE_AREAS(Option.ENABLED, Option.ONLY_INSIDE_NATURE_AREAS),
   /**
    * Use subreceptors only when calculating receptors
    * Do not use subreceptors when calculating custom points
    */
-  ENABLED_RECEPTORS_ONLY(true, true, true),
+  ENABLED_RECEPTORS_ONLY(Option.ENABLED, Option.RECEPTORS_ONLY, Option.ONLY_INSIDE_HABITATS),
   /**
    * Always use subreceptors
    * Also include subreceptors outside relevant habitat areas
    */
-  ENABLED_OUTSIDE_HABITATS(true, false, false);
+  ENABLED_OUTSIDE_HABITATS(Option.ENABLED);
+
+  private enum Option {
+    ENABLED,
+    ONLY_INSIDE_NATURE_AREAS,
+    ONLY_INSIDE_HABITATS,
+    RECEPTORS_ONLY,
+  }
 
   private final boolean enabled;
-  private final boolean receptorsOnly;
   private final boolean onlyInsideHabitats;
+  private final boolean onlyInsideNatureAreas;
+  private final boolean receptorsOnly;
+
+  SubReceptorsMode(final Option ...options) {
+    final Set<Option> set = new HashSet<>(Arrays.asList(options));
 
-  SubReceptorsMode(final boolean enabled, final boolean receptorsOnly, final boolean onlyInsideHabitats) {
-    this.enabled = enabled;
-    this.receptorsOnly = receptorsOnly;
-    this.onlyInsideHabitats = onlyInsideHabitats;
+    this.enabled = set.contains(Option.ENABLED);
+    this.onlyInsideHabitats = set.contains(Option.ONLY_INSIDE_HABITATS);
+    this.onlyInsideNatureAreas = set.contains(Option.ONLY_INSIDE_NATURE_AREAS);
+    this.receptorsOnly = set.contains(Option.RECEPTORS_ONLY);
   }
 
   /**
@@ -74,6 +94,14 @@ public boolean isEnabledForPoints() {
     return enabled && !receptorsOnly;
   }
 
+  /**
+   * Whether subreceptors should only be placed inside nature areas
+   * @return true if subreceptors should only be placed inside nature areas
+   */
+  public boolean isOnlyInsideNatureAreas() {
+    return onlyInsideNatureAreas;
+  }
+
   /**
    * Whether subreceptors should only be placed inside habitats
    * @return true if subreceptors should only be placed inside habitats