62 lines
2.2 KiB
Java
Executable File
62 lines
2.2 KiB
Java
Executable File
package eu.crushedpixel.replaymod.studio;
|
|
|
|
import java.io.File;
|
|
import java.io.FileInputStream;
|
|
import java.io.FileOutputStream;
|
|
import java.io.IOException;
|
|
|
|
import com.google.gson.JsonObject;
|
|
|
|
import net.minecraft.network.play.server.S3EPacketTeams;
|
|
import de.johni0702.replaystudio.PacketData;
|
|
import de.johni0702.replaystudio.filter.ChangeTimestampFilter;
|
|
import de.johni0702.replaystudio.filter.RemoveFilter;
|
|
import de.johni0702.replaystudio.filter.SquashFilter;
|
|
import de.johni0702.replaystudio.io.ReplayOutputStream;
|
|
import de.johni0702.replaystudio.stream.PacketStream;
|
|
import de.johni0702.replaystudio.studio.ReplayStudio;
|
|
import eu.crushedpixel.replaymod.recording.ReplayMetaData;
|
|
import eu.crushedpixel.replaymod.utils.ReplayFileIO;
|
|
|
|
public class StudioImplementation {
|
|
|
|
public static void trimReplay(File replayFile, boolean isTmcpr, int beginning, int ending, File outputFile) throws IOException {
|
|
ReplayStudio studio = new ReplayStudio();
|
|
studio.setWrappingEnabled(false);
|
|
PacketStream stream = studio.createReplayStream(new FileInputStream(replayFile), isTmcpr);
|
|
stream.addFilter(new SquashFilter(), -1, beginning);
|
|
stream.addFilter(new RemoveFilter(), ending, -1);
|
|
|
|
ChangeTimestampFilter ctf = new ChangeTimestampFilter();
|
|
JsonObject cfg = new JsonObject();
|
|
cfg.addProperty("offset", -beginning);
|
|
ctf.init(studio, cfg);
|
|
|
|
stream.addFilter(ctf, beginning, ending);
|
|
|
|
File temp = File.createTempFile("trimmed", "tmcpr");
|
|
temp.deleteOnExit();
|
|
|
|
FileOutputStream fos = new FileOutputStream(temp);
|
|
ReplayOutputStream out = new ReplayOutputStream(studio, fos);
|
|
PacketData packetData;
|
|
|
|
while((packetData = stream.next()) != null) {
|
|
out.write(packetData);
|
|
}
|
|
for(PacketData data : stream.end()) {
|
|
out.write(data);
|
|
}
|
|
|
|
out.close();
|
|
|
|
ReplayMetaData metaData = ReplayFileIO.getMetaData(replayFile);
|
|
ending = Math.min(metaData.getDuration(), ending);
|
|
metaData.setDuration(ending-beginning);
|
|
|
|
outputFile.createNewFile();
|
|
|
|
ReplayFileIO.writeReplayFile(outputFile, temp, metaData);
|
|
}
|
|
}
|