class="kw">package com.hypixel.hytale.builtin.hytalegenerator.density;

class="kw">import javax.annotation.Nonnull;
class="kw">import javax.annotation.Nullable;

class="kw">public class SwitchStateDensity class="kw">extends Density {
   class="kw">public class="kw">static class="kw">final int DEFAULT_SWITCH_STATE = 0;
   @Nullable
   class="kw">private Density input;
   class="kw">private class="kw">final int switchState;
   @Nonnull
   class="kw">private class="kw">final Density.Context rChildContext;

   class="kw">public SwitchStateDensity(Density var1, int var2) {
      this.input = var1;
      this.switchState = var2;
      this.rChildContext = new Density.Context();
   }

   @Override
   class="kw">public double process(@Nonnull Density.Context var1) {
      if (this.input == null) {
         class="kw">return 0.0;
      }

      this.rChildContext.assign(var1);
      this.rChildContext.switchState = this.switchState;
      double var2 = this.input.process(this.rChildContext);
      this.rChildContext.clearReferences();
      class="kw">return var2;
   }

   @Override
   class="kw">public void setInputs(@Nonnull Density[] var1) {
      if (var1.length == 0) {
         this.input = null;
      }

      this.input = var1[0];
   }
}