src/main/java/kubatech/api/helpers/ProgressBarWrapper.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

package kubatech.api.helpers;

import static kubatech.api.utils.ModUtils.isClientSided;

import java.io.IOException;

import kubatech.Tags;
import kubatech.api.LoaderReference;
import alexiil.mods.load.ProgressDisplayer;
import cpw.mods.fml.common.FMLCommonHandler;
import cpw.mods.fml.common.ProgressManager;

@SuppressWarnings("deprecation")
public class ProgressBarWrapper {

    ProgressManager.ProgressBar internalFMLBar;
    boolean isFMLBar;
    String name;
    int maxSteps;
    int steps = 0;

    public ProgressBarWrapper(String name, int steps) {
        if (!isClientSided) return;
        maxSteps = steps;
        this.name = name;
        if (!LoaderReference.BetterLoadingScreen) {
            internalFMLBar = ProgressManager.push(name, steps);
            isFMLBar = true;
            return;
        }
        isFMLBar = false;
    }

    public void step(String message) {
        if (!isClientSided) return;
        if (isFMLBar) internalFMLBar.step(message);
        else {
            steps++;
            try {
                ProgressDisplayer.displayProgress(
                        Tags.MODNAME + ": " + name + " -> " + message,
                        (float) steps / (float) maxSteps);
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
            // Prevent game freeze
            FMLCommonHandler.instance().processWindowMessages();
        }
    }

    public void end() {
        if (!isClientSided) return;
        if (isFMLBar) ProgressManager.pop(internalFMLBar);
    }
}