353 lines
15 KiB
Java
353 lines
15 KiB
Java
package com.replaymod.render.gui;
|
|
|
|
import com.google.common.collect.Iterables;
|
|
import com.google.common.util.concurrent.FutureCallback;
|
|
import com.google.common.util.concurrent.Futures;
|
|
import com.replaymod.core.ReplayMod;
|
|
import com.replaymod.core.utils.Utils;
|
|
import com.replaymod.core.versions.MCVer;
|
|
import com.replaymod.render.ReplayModRender;
|
|
import com.replaymod.render.VideoWriter;
|
|
import com.replaymod.render.rendering.VideoRenderer;
|
|
import com.replaymod.render.utils.RenderJob;
|
|
import com.replaymod.replay.ReplayHandler;
|
|
import com.replaymod.replaystudio.pathing.path.Timeline;
|
|
import de.johni0702.minecraft.gui.GuiRenderer;
|
|
import de.johni0702.minecraft.gui.RenderInfo;
|
|
import de.johni0702.minecraft.gui.container.AbstractGuiClickableContainer;
|
|
import de.johni0702.minecraft.gui.container.AbstractGuiScreen;
|
|
import de.johni0702.minecraft.gui.container.GuiContainer;
|
|
import de.johni0702.minecraft.gui.container.GuiPanel;
|
|
import de.johni0702.minecraft.gui.container.GuiVerticalList;
|
|
import de.johni0702.minecraft.gui.element.GuiButton;
|
|
import de.johni0702.minecraft.gui.element.GuiElement;
|
|
import de.johni0702.minecraft.gui.element.GuiLabel;
|
|
import de.johni0702.minecraft.gui.element.GuiTextField;
|
|
import de.johni0702.minecraft.gui.function.Typeable;
|
|
import de.johni0702.minecraft.gui.layout.CustomLayout;
|
|
import de.johni0702.minecraft.gui.layout.GridLayout;
|
|
import de.johni0702.minecraft.gui.layout.HorizontalLayout;
|
|
import de.johni0702.minecraft.gui.popup.AbstractGuiPopup;
|
|
import de.johni0702.minecraft.gui.popup.GuiYesNoPopup;
|
|
import de.johni0702.minecraft.gui.utils.Colors;
|
|
import de.johni0702.minecraft.gui.utils.lwjgl.Dimension;
|
|
import de.johni0702.minecraft.gui.utils.lwjgl.ReadableDimension;
|
|
import de.johni0702.minecraft.gui.utils.lwjgl.ReadablePoint;
|
|
import net.minecraft.client.gui.screen.NoticeScreen;
|
|
import net.minecraft.util.crash.CrashReport;
|
|
|
|
import javax.annotation.Nullable;
|
|
import java.util.ArrayList;
|
|
import java.util.HashSet;
|
|
import java.util.List;
|
|
import java.util.Set;
|
|
|
|
import static com.replaymod.render.ReplayModRender.LOGGER;
|
|
|
|
//#if MC>=11400
|
|
import net.minecraft.text.TranslatableText;
|
|
//#else
|
|
//$$ import com.replaymod.replaystudio.util.I18n;
|
|
//#endif
|
|
|
|
public class GuiRenderQueue extends AbstractGuiPopup<GuiRenderQueue> implements Typeable {
|
|
private final GuiLabel title = new GuiLabel().setI18nText("replaymod.gui.renderqueue.title").setColor(Colors.BLACK);
|
|
private final GuiVerticalList list = new GuiVerticalList().setDrawShadow(true).setDrawSlider(true);
|
|
private final GuiButton addButton = new GuiButton().setI18nLabel("replaymod.gui.renderqueue.add").setSize(150, 20);
|
|
private final GuiButton renameButton = new GuiButton().setI18nLabel("replaymod.gui.rename").setSize(73, 20);
|
|
private final GuiButton removeButton = new GuiButton().setI18nLabel("replaymod.gui.remove").setSize(73, 20);
|
|
private final GuiButton renderButton = new GuiButton().setSize(150, 20);
|
|
private final GuiButton closeButton = new GuiButton().setI18nLabel("replaymod.gui.close").setSize(150, 20).onClick(this::close);
|
|
|
|
/*
|
|
|
|
|---------------------------------|
|
|
| Add | Render |
|
|
|---------------------------------|
|
|
| Rename | Remove | Close |
|
|
|---------------------------------|
|
|
|
|
*/
|
|
private final GuiPanel buttonPanel = new GuiPanel()
|
|
.setLayout(new GridLayout().setSpacingX(5).setSpacingY(5).setColumns(2))
|
|
.addElements(null,
|
|
addButton,
|
|
renderButton,
|
|
new GuiPanel().setLayout(new HorizontalLayout().setSpacing(4)).addElements(null,
|
|
renameButton, removeButton),
|
|
closeButton);
|
|
|
|
private final AbstractGuiScreen container;
|
|
private final ReplayHandler replayHandler;
|
|
private final Set<Entry> selectedEntries = new HashSet<>();
|
|
|
|
{
|
|
popup.setLayout(new CustomLayout<GuiPanel>() {
|
|
@Override
|
|
protected void layout(GuiPanel container, int width, int height) {
|
|
pos(title, width / 2 - width(title) / 2, 0);
|
|
pos(list, 0, y(title) + height(title) + 5);
|
|
pos(buttonPanel, width / 2 - width(buttonPanel) / 2, height - height(buttonPanel));
|
|
size(list, width, y(buttonPanel) - y(list) - 10);
|
|
}
|
|
|
|
@Override
|
|
public ReadableDimension calcMinSize(GuiContainer<?> container) {
|
|
ReadableDimension screenSize = GuiRenderQueue.this.container.getMinSize();
|
|
return new Dimension(screenSize.getWidth() - 40,
|
|
screenSize.getHeight() - 20 - buttonPanel.getMinSize().getHeight() - title.getMinSize().getHeight());
|
|
}
|
|
}).addElements(null, title, list, buttonPanel);
|
|
}
|
|
|
|
public GuiRenderQueue(AbstractGuiScreen container, GuiRenderSettings guiRenderSettings, ReplayHandler replayHandler, Timeline timeline) {
|
|
super(container);
|
|
this.container = container;
|
|
this.replayHandler = replayHandler;
|
|
LOGGER.trace("Opening render queue popup");
|
|
|
|
setBackgroundColor(Colors.DARK_TRANSPARENT);
|
|
|
|
List<RenderJob> queue = ReplayModRender.instance.getRenderQueue();
|
|
|
|
for (RenderJob renderJob : queue) {
|
|
LOGGER.trace("Adding {} to job queue list", renderJob);
|
|
list.getListPanel().addElements(null, new Entry(renderJob));
|
|
}
|
|
|
|
addButton.onClick(() -> {
|
|
LOGGER.trace("Add button clicked");
|
|
// Open popup
|
|
GuiYesNoPopup popup = GuiYesNoPopup.open(container)
|
|
.setYesI18nLabel("replaymod.gui.add").setNoI18nLabel("replaymod.gui.cancel");
|
|
popup.getInfo().setLayout(new HorizontalLayout(HorizontalLayout.Alignment.CENTER).setSpacing(5));
|
|
// Add content
|
|
GuiLabel label = new GuiLabel().setI18nText("replaymod.gui.renderqueue.jobname").setColor(Colors.BLACK);
|
|
GuiTextField nameField = new GuiTextField().setSize(150, 20).setFocused(true);
|
|
nameField.setText(guiRenderSettings.getOutputFile().getName());
|
|
popup.getInfo().addElements(new HorizontalLayout.Data(0.5), label, nameField);
|
|
// Disable "Yes" button while name is empty
|
|
nameField.onTextChanged(old -> popup.getYesButton().setEnabled(!nameField.getText().isEmpty())).onEnter(() -> {
|
|
if (popup.getYesButton().isEnabled()) {
|
|
popup.getYesButton().onClick();
|
|
}
|
|
});
|
|
// Register callback
|
|
Futures.addCallback(popup.getFuture(), new FutureCallback<Boolean>() {
|
|
@Override
|
|
public void onSuccess(@Nullable Boolean result) {
|
|
if (result == Boolean.TRUE) {
|
|
RenderJob newJob = new RenderJob();
|
|
newJob.setName(nameField.getText());
|
|
newJob.setSettings(guiRenderSettings.save(false));
|
|
newJob.setTimeline(timeline);
|
|
LOGGER.trace("Adding new job: {}", newJob);
|
|
queue.add(newJob);
|
|
list.getListPanel().addElements(null, new Entry(newJob));
|
|
} else {
|
|
LOGGER.trace("Adding cancelled");
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void onFailure(Throwable t) {
|
|
LOGGER.error("Add Job popup:", t);
|
|
}
|
|
});
|
|
});
|
|
if (guiRenderSettings != null) {
|
|
addButton.setEnabled(guiRenderSettings.renderButton.isEnabled());
|
|
}
|
|
|
|
renameButton.onClick(() -> {
|
|
Entry selectedEntry = selectedEntries.iterator().next();
|
|
LOGGER.trace("Rename button clicked for {}", selectedEntry.job);
|
|
// Open popup
|
|
GuiYesNoPopup popup = GuiYesNoPopup.open(container)
|
|
.setYesI18nLabel("replaymod.gui.rename").setNoI18nLabel("replaymod.gui.cancel");
|
|
popup.getInfo().setLayout(new HorizontalLayout(HorizontalLayout.Alignment.CENTER).setSpacing(5));
|
|
// Add content
|
|
GuiLabel label = new GuiLabel().setI18nText("replaymod.gui.renderqueue.jobname").setColor(Colors.BLACK);
|
|
GuiTextField nameField = new GuiTextField().setSize(150, 20).setFocused(true)
|
|
.setText(selectedEntry.job.getName());
|
|
popup.getInfo().addElements(new HorizontalLayout.Data(0.5), label, nameField);
|
|
// Disable "Yes" button while name is empty
|
|
nameField.onTextChanged(old -> popup.getYesButton().setEnabled(!nameField.getText().isEmpty())).onEnter(() -> {
|
|
if (popup.getYesButton().isEnabled()) {
|
|
popup.getYesButton().onClick();
|
|
}
|
|
});
|
|
// Register callback
|
|
Futures.addCallback(popup.getFuture(), new FutureCallback<Boolean>() {
|
|
@Override
|
|
public void onSuccess(@Nullable Boolean result) {
|
|
if (result == Boolean.TRUE) {
|
|
LOGGER.trace("Renaming {} to \"{}\"", selectedEntry.job, nameField.getText());
|
|
selectedEntry.setName(nameField.getText());
|
|
} else {
|
|
LOGGER.trace("Renaming cancelled");
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void onFailure(Throwable t) {
|
|
LOGGER.error("Rename Job popup:", t);
|
|
}
|
|
});
|
|
});
|
|
|
|
removeButton.onClick(() -> {
|
|
for (Entry entry : selectedEntries) {
|
|
LOGGER.trace("Remove button clicked for {}", entry.job);
|
|
list.getListPanel().removeElement(entry);
|
|
queue.remove(entry.job);
|
|
}
|
|
selectedEntries.clear();
|
|
updateButtons();
|
|
});
|
|
|
|
renderButton.onClick(() -> {
|
|
LOGGER.trace("Render button clicked");
|
|
List<RenderJob> renderQueue = new ArrayList<>();
|
|
for (Entry entry : selectedEntries) {
|
|
renderQueue.add(entry.job);
|
|
}
|
|
ReplayMod.instance.runLaterWithoutLock(() -> processQueue(renderQueue));
|
|
});
|
|
|
|
updateButtons();
|
|
}
|
|
|
|
private void processQueue(Iterable<RenderJob> queue) {
|
|
// Close all GUIs (so settings in GuiRenderSettings are saved)
|
|
getMinecraft().openScreen(null);
|
|
// Start rendering
|
|
int jobsDone = 0;
|
|
for (RenderJob renderJob : queue) {
|
|
LOGGER.info("Starting render job {}", renderJob);
|
|
try {
|
|
VideoRenderer videoRenderer = new VideoRenderer(renderJob.getSettings(), replayHandler, renderJob.getTimeline());
|
|
videoRenderer.renderVideo();
|
|
} catch (VideoWriter.NoFFmpegException e) {
|
|
LOGGER.error("Rendering video:", e);
|
|
NoticeScreen errorScreen = new NoticeScreen(
|
|
//#if MC>=11400
|
|
container::display,
|
|
new TranslatableText("replaymod.gui.rendering.error.title"),
|
|
new TranslatableText("replaymod.gui.rendering.error.message")
|
|
//#else
|
|
//$$ I18n.format("replaymod.gui.rendering.error.title"),
|
|
//$$ I18n.format("replaymod.gui.rendering.error.message")
|
|
//#endif
|
|
);
|
|
getMinecraft().openScreen(errorScreen);
|
|
return;
|
|
} catch (VideoWriter.FFmpegStartupException e) {
|
|
int jobsToSkip = jobsDone;
|
|
GuiExportFailed.tryToRecover(e, newSettings -> {
|
|
// Update current job with fixed ffmpeg arguments
|
|
renderJob.setSettings(newSettings);
|
|
// Restart queue, skipping the already completed jobs
|
|
ReplayMod.instance.runLaterWithoutLock(() -> processQueue(Iterables.skip(queue, jobsToSkip)));
|
|
});
|
|
return;
|
|
} catch (Throwable t) {
|
|
Utils.error(LOGGER, this, CrashReport.create(t, "Rendering video"), () -> {});
|
|
container.display(); // Re-show the queue popup and the new error popup
|
|
return;
|
|
}
|
|
jobsDone++;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void open() {
|
|
super.open();
|
|
}
|
|
|
|
@Override
|
|
protected GuiRenderQueue getThis() {
|
|
return this;
|
|
}
|
|
|
|
public void updateButtons() {
|
|
int selected = selectedEntries.size();
|
|
renameButton.setEnabled(selected == 1);
|
|
removeButton.setEnabled(selected >= 1);
|
|
renderButton.setEnabled(selected > 0);
|
|
renderButton.setI18nLabel("replaymod.gui.renderqueue.render" + (selected > 0 ? "selected" : "all"));
|
|
}
|
|
|
|
@Override
|
|
public boolean typeKey(ReadablePoint mousePosition, int keyCode, char keyChar, boolean ctrlDown, boolean shiftDown) {
|
|
if (MCVer.Keyboard.hasControlDown() && keyCode == MCVer.Keyboard.KEY_A) {
|
|
if (selectedEntries.size() < list.getListPanel().getChildren().size()) {
|
|
for (GuiElement<?> child : list.getListPanel().getChildren()) {
|
|
if (child instanceof Entry) {
|
|
selectedEntries.add((Entry) child);
|
|
}
|
|
}
|
|
} else {
|
|
selectedEntries.clear();
|
|
}
|
|
updateButtons();
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
public class Entry extends AbstractGuiClickableContainer<Entry> {
|
|
public final GuiLabel label = new GuiLabel(this);
|
|
public final RenderJob job;
|
|
|
|
public Entry(RenderJob job) {
|
|
this.job = job;
|
|
|
|
setLayout(new CustomLayout<Entry>() {
|
|
@Override
|
|
protected void layout(Entry container, int width, int height) {
|
|
pos(label, 5, height / 2 - height(label) / 2);
|
|
}
|
|
|
|
@Override
|
|
public ReadableDimension calcMinSize(GuiContainer<?> container) {
|
|
return new Dimension(buttonPanel.calcMinSize().getWidth(), 16);
|
|
}
|
|
});
|
|
label.setText(job.getName());
|
|
}
|
|
|
|
@Override
|
|
protected void onClick() {
|
|
if (!MCVer.Keyboard.hasControlDown()) {
|
|
selectedEntries.clear();
|
|
}
|
|
if (selectedEntries.contains(this)) {
|
|
selectedEntries.remove(this);
|
|
} else {
|
|
selectedEntries.add(this);
|
|
}
|
|
updateButtons();
|
|
}
|
|
|
|
@Override
|
|
public void draw(GuiRenderer renderer, ReadableDimension size, RenderInfo renderInfo) {
|
|
if (selectedEntries.contains(this)) {
|
|
renderer.drawRect(0, 0, size.getWidth(), size.getHeight(), Colors.BLACK);
|
|
renderer.drawRect(0, 0, 2, size.getHeight(), Colors.WHITE);
|
|
}
|
|
super.draw(renderer, size, renderInfo);
|
|
}
|
|
|
|
public void setName(String name) {
|
|
job.setName(name);
|
|
label.setText(name);
|
|
}
|
|
|
|
@Override
|
|
protected Entry getThis() {
|
|
return this;
|
|
}
|
|
}
|
|
}
|