src/main/java/gregtech/api/util/OutputHatchWrapper.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70

package gregtech.api.util;

import java.util.function.Predicate;

import net.minecraftforge.fluids.FluidStack;
import net.minecraftforge.fluids.FluidTankInfo;

import org.jetbrains.annotations.NotNull;

import gregtech.api.interfaces.fluid.IFluidStore;
import gregtech.api.metatileentity.implementations.MTEHatchOutput;

/**
 * Wrapper for output hatch to allow multiblocks to apply specific filter.
 */
public class OutputHatchWrapper implements IFluidStore {

    private final MTEHatchOutput outputHatch;
    private final Predicate<FluidStack> filter;

    public OutputHatchWrapper(MTEHatchOutput outputHatch, Predicate<FluidStack> filter) {
        this.outputHatch = outputHatch;
        this.filter = filter;
    }

    @Override
    public FluidStack getFluid() {
        return outputHatch.getFluid();
    }

    @Override
    public int getFluidAmount() {
        return outputHatch.getFluidAmount();
    }

    @Override
    public int getCapacity() {
        return outputHatch.getCapacity();
    }

    @Override
    public FluidTankInfo getInfo() {
        return outputHatch.getInfo();
    }

    @Override
    public int fill(FluidStack resource, boolean doFill) {
        return outputHatch.fill(resource, doFill);
    }

    @Override
    public FluidStack drain(int maxDrain, boolean doDrain) {
        return outputHatch.drain(maxDrain, doDrain);
    }

    @Override
    public boolean isEmptyAndAcceptsAnyFluid() {
        return outputHatch.isEmptyAndAcceptsAnyFluid();
    }

    @Override
    public boolean canStoreFluid(@NotNull FluidStack fluidStack) {
        return outputHatch.canStoreFluid(fluidStack) && filter.test(fluidStack);
    }

    @Override
    public int getAvailableSpace() {
        return outputHatch.getAvailableSpace();
    }
}