IntConditionBuilder.java

package com.hypixel.hytale.server.worldgen.util.condition;

import com.hypixel.hytale.procedurallib.condition.IIntCondition;
import it.unimi.dsi.fastutil.ints.IntConsumer;
import it.unimi.dsi.fastutil.ints.IntSet;
import it.unimi.dsi.fastutil.ints.IntSets;
import java.util.function.Supplier;
import javax.annotation.Nullable;

public class IntConditionBuilder implements IntConsumer {
   private final Supplier<IntSet> setSupplier;
   private final int nullValue;
   private int first;
   @Nullable
   private IntSet set = null;

   public IntConditionBuilder(Supplier<IntSet> setSupplier, int nullValue) {
      this.setSupplier = setSupplier;
      this.nullValue = nullValue;
      this.first = nullValue;
   }

   public void accept(int value) {
      this.add(value);
   }

   public boolean add(int value) {
      if (value != this.first && value != this.nullValue) {
         if (this.first == this.nullValue) {
            this.first = value;
            return true;
         } else {
            if (this.set == null) {
               this.set = (IntSet)this.setSupplier.get();
               this.set.add(this.first);
            }

            return this.set.add(value);
         }
      } else {
         return false;
      }
   }

   public IIntCondition buildOrDefault(IIntCondition defaultCondition) {
      if (this.first == this.nullValue) {
         return defaultCondition;
      } else {
         IntSet set = this.set;
         if (set == null) {
            set = IntSets.singleton(this.first);
         }

         return new HashSetIntCondition(set);
      }
   }
}