git.rm.cloudns.org Git - xonotic/xonotic.git/blob

   1 package com.nexuiz.demorecorder.application.democutter;
   2 import java.io.DataInputStream;
   3 import java.io.DataOutputStream;
   4 import java.io.EOFException;
   5 import java.io.File;
   6 import java.io.FileInputStream;
   7 import java.io.FileNotFoundException;
   8 import java.io.FileOutputStream;
   9 import java.io.IOException;
  10 import java.io.UnsupportedEncodingException;
  11
  12 public class DemoCutter {
  13
  14         private static final byte CDTRACK_SEPARATOR = 0x0A;
  15
  16         private DataInputStream inStream;
  17         private DataOutputStream outStream;
  18         private File inFile;
  19         private File outFile;
  20
  21         /**
  22          * Calls the cutDemo method with reasonable default values for the second and first fast-forward stage.
  23          * @param inFile @see other cutDemo method
  24          * @param outFile @see other cutDemo method
  25          * @param startTime @see other cutDemo method
  26          * @param endTime @see other cutDemo method
  27          * @param injectAtStart @see other cutDemo method
  28          * @param injectBeforeCap @see other cutDemo method
  29          * @param injectAfterCap @see other cutDemo method
  30          */
  31         public void cutDemo(File inFile, File outFile, float startTime, float endTime, String injectAtStart, String injectBeforeCap, String injectAfterCap) {
  32                 this.cutDemo(inFile, outFile, startTime, endTime, injectAtStart, injectBeforeCap, injectAfterCap, 100, 10);
  33         }
  34
  35         /**
  36          * Cuts the demo by injecting a 2-phase fast forward command until startTime is reached, then injects the cl_capturevideo 1 command
  37          * and once endTime is reached the cl_capturevideo 0 command is injected.
  38          * @param inFile the original demo file
  39          * @param outFile the new cut demo file
  40          * @param startTime when to start capturing (use the gametime in seconds)
  41          * @param endTime when to stop capturing
  42          * @param injectAtStart a String that will be injected right at the beginning of the demo
  43          *                                              can be anything that would make sense and can be parsed by DP's console
  44          * @param injectBeforeCap a String that will be injected 5 seconds before capturing starts
  45          * @param injectAfterCap a String that will be injected shortly after capturing ended
  46          * @param ffwSpeedFirstStage fast-forward speed at first stage, when the startTime is still about a minute away (use high values, e.g. 100)
  47          * @param ffwSpeedSecondStage fast-forward speed when coming a few seconds close to startTime, use lower values e.g. 5 or 10
  48          */
  49         public void cutDemo(File inFile, File outFile, float startTime, float endTime, String injectAtStart, String injectBeforeCap, String injectAfterCap, int ffwSpeedFirstStage, int ffwSpeedSecondStage) {
  50                 this.inFile = inFile;
  51                 this.outFile = outFile;
  52                 this.prepareStreams();
  53                 this.readCDTrack();
  54                 injectAfterCap = this.checkInjectString(injectAfterCap);
  55                 injectAtStart = this.checkInjectString(injectAtStart);
  56                 injectBeforeCap = this.checkInjectString(injectBeforeCap);
  57
  58                 byte[] data;
  59                 float svctime = -1;
  60                 boolean firstLoop = true;
  61                 String injectBuffer = "";
  62                 int demoStarted = 0;
  63                 boolean endIsReached = false;
  64                 boolean finalInjectionDone = false;
  65                 boolean disconnectIssued = false;
  66                 int svcLoops = 0;
  67                 float firstSvcTime = -1;
  68                 float lastSvcTime = -1;
  69
  70                 try {
  71                         while (true) {
  72                                 DemoPacket demoPacket = new DemoPacket(this.inStream);
  73                                 if (demoPacket.isEndOfFile()) {
  74                                         break;
  75                                 }
  76
  77                                 if (demoPacket.isClientToServerPacket()) {
  78                                         try {
  79                                                 this.outStream.write(demoPacket.getOriginalLengthAsByte());
  80                                                 this.outStream.write(demoPacket.getAngles());
  81                                                 this.outStream.write(demoPacket.getOriginalData());
  82                                         } catch (IOException e) {
  83                                                 throw new DemoCutterException("Unexpected I/O Exception occurred when writing to the cut demo", e);
  84                                         }
  85
  86                                         continue;
  87                                 }
  88
  89                                 if (demoPacket.getSvcTime() != -1) {
  90                                         svctime = demoPacket.getSvcTime();
  91                                 }
  92
  93                                 if (svctime != -1) {
  94                                         if (firstSvcTime == -1) {
  95                                                 firstSvcTime = svctime;
  96                                         }
  97                                         lastSvcTime = svctime;
  98
  99                                         if (firstLoop) {
 100                                                 injectBuffer = "\011\n" + injectAtStart + ";slowmo " + ffwSpeedFirstStage + "\n\000";
 101                                                 firstLoop = false;
 102                                         } else {
 103                                                 if (demoStarted < 1 && svctime > (startTime - 50)) {
 104                                                         injectBuffer = "\011\nslowmo " + ffwSpeedSecondStage + "\n\000";
 105                                                         demoStarted = 1;
 106                                                 }
 107                                                 if (demoStarted < 2 && svctime > (startTime - 5)) {
 108                                                         injectBuffer = "\011\nslowmo 1;" + injectBeforeCap +"\n\000";
 109                                                         demoStarted = 2;
 110                                                 }
 111                                                 if (demoStarted < 3 && svctime > startTime) {
 112                                                         injectBuffer = "\011\ncl_capturevideo 1\n\000";
 113                                                         demoStarted = 3;
 114                                                 }
 115                                                 if (!endIsReached && svctime > endTime) {
 116                                                         injectBuffer = "\011\ncl_capturevideo 0\n\000";
 117                                                         endIsReached = true;
 118                                                 }
 119                                                 if (endIsReached && !finalInjectionDone && svctime > (endTime + 1)) {
 120                                                         injectBuffer = "\011\n" + injectAfterCap + "\n\000";
 121                                                         finalInjectionDone = true;
 122                                                 }
 123                                                 if (finalInjectionDone && !disconnectIssued && svctime > (endTime + 2)) {
 124                                                         injectBuffer = "\011\ndisconnect\n\000";
 125                                                         disconnectIssued = true;
 126                                                 }
 127                                         }
 128                                         svcLoops++;
 129                                 }
 130
 131                                 byte[] injectBufferAsBytes = null;
 132                                 try {
 133                                         injectBufferAsBytes = injectBuffer.getBytes("US-ASCII");
 134                                 } catch (UnsupportedEncodingException e) {
 135                                         throw new DemoCutterException("Could not convert String to bytes using US-ASCII charset!", e);
 136                                 }
 137
 138                                 data = demoPacket.getOriginalData();
 139                                 if ((injectBufferAsBytes.length + data.length) < 65536) {
 140                                         data = DemoCutterUtils.mergeByteArrays(injectBufferAsBytes, data);
 141                                         injectBuffer = "";
 142                                 }
 143
 144                                 byte[] newLengthLittleEndian = DemoCutterUtils.convertLittleEndian(data.length);
 145                                 try {
 146                                         this.outStream.write(newLengthLittleEndian);
 147                                         this.outStream.write(demoPacket.getAngles());
 148                                         this.outStream.write(data);
 149                                 } catch (IOException e) {
 150                                         throw new DemoCutterException("Unexpected I/O Exception occurred when writing to the cut demo", e);
 151                                 }
 152
 153                         }
 154
 155                         if (startTime < firstSvcTime) {
 156                                 throw new DemoCutterException("Start time for the demo is " + startTime + ", but demo doesn't start before " + firstSvcTime);
 157                         }
 158                         if (endTime > lastSvcTime) {
 159                                 throw new DemoCutterException("End time for the demo is " + endTime + ", but demo already stops at " + lastSvcTime);
 160                         }
 161                 } catch (DemoCutterException e) {
 162                         throw e;
 163                 } catch (Throwable e) {
 164                         throw new DemoCutterException("Internal error in demo cutter sub-route (invalid demo file?)", e);
 165                 } finally {
 166                         try {
 167                                 this.outStream.close();
 168                                 this.inStream.close();
 169                         } catch (IOException e) {}
 170                 }
 171         }
 172
 173
 174
 175         /**
 176          * Seeks forward in the inStream until CDTRACK_SEPARATOR byte was reached.
 177          * All the content is copied to the outStream.
 178          */
 179         private void readCDTrack() {
 180                 byte lastByte;
 181                 try {
 182                         while ((lastByte = inStream.readByte()) != CDTRACK_SEPARATOR) {
 183                                 this.outStream.write(lastByte);
 184                         }
 185                         this.outStream.write(CDTRACK_SEPARATOR);
 186                 } catch (EOFException e) {
 187                         throw new DemoCutterException("Unexpected EOF occurred when reading CD track of demo " + inFile.getPath(), e);
 188                 }
 189                 catch (IOException e) {
 190                         throw new DemoCutterException("Unexpected I/O Exception occurred when reading CD track of demo " + inFile.getPath(), e);
 191                 }
 192         }
 193
 194         private void prepareStreams() {
 195                 try {
 196                         this.inStream = new DataInputStream(new FileInputStream(this.inFile));
 197                 } catch (FileNotFoundException e) {
 198                         throw new DemoCutterException("Could not open demo file " + inFile.getPath(), e);
 199                 }
 200
 201                 try {
 202                         this.outStream = new DataOutputStream(new FileOutputStream(this.outFile));
 203                 } catch (FileNotFoundException e) {
 204                         throw new DemoCutterException("Could not open demo file " + outFile.getPath(), e);
 205                 }
 206         }
 207
 208         private String checkInjectString(String injectionString) {
 209                 while (injectionString.endsWith(";") || injectionString.endsWith("\n")) {
 210                         injectionString = injectionString.substring(0, injectionString.length()-1);
 211                 }
 212                 return injectionString;
 213         }
 214 }