class="kw">package com.hypixel.hytale.builtin.portals.utils.posqueries.predicates.generic;

class="kw">import com.hypixel.hytale.builtin.portals.utils.posqueries.SpatialQuery;
class="kw">import com.hypixel.hytale.builtin.portals.utils.posqueries.SpatialQueryDebug;
class="kw">import com.hypixel.hytale.server.core.universe.world.World;
class="kw">import java.util.stream.Stream;
class="kw">import javax.annotation.Nonnull;
class="kw">import javax.annotation.Nullable;
class="kw">import org.joml.Vector3d;

class="kw">public class FlatMapQuery class="kw">implements SpatialQuery {
   @Nonnull
   class="kw">private class="kw">final SpatialQuery generator;
   @Nonnull
   class="kw">private class="kw">final SpatialQuery expand;

   class="kw">public FlatMapQuery(@Nonnull SpatialQuery var1, @Nonnull SpatialQuery var2) {
      this.generator = var1;
      this.expand = var2;
   }

   @Nonnull
   @Override
   class="kw">public Stream<Vector3d> createCandidates(@Nonnull World var1, @Nonnull Vector3d var2, @Nullable SpatialQueryDebug var3) {
      class="kw">return this.generator.createCandidates(var1, var2, var3).flatMap(var3x -> {
         Stream var4 = this.expand.createCandidates(var1, var3x, var3);
         if (var3 != null) {
            var3.indent("Flat-map expand from " + SpatialQueryDebug.fmt(var3x) + ":");
            class="kw">return Stream.concat(var4, Stream.of((Vector3d)null).peek(var1x -> var3.unindent()).flatMap(var0 -> Stream.empty()));
         } else {
            class="kw">return var4;
         }
      });
   }
}