remove unnecessary variable
[xonotic/xonotic.git] / misc / tools / NexuizDemoRecorder / main / src / main / java / com / nexuiz / demorecorder / application / democutter / DemoCutter.java
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                 float firstSvcTime = -1;
67                 float lastSvcTime = -1;
68                 
69                 try {
70                         while (true) {
71                                 DemoPacket demoPacket = new DemoPacket(this.inStream);
72                                 if (demoPacket.isEndOfFile()) {
73                                         break;
74                                 }
75                                 
76                                 if (demoPacket.isClientToServerPacket()) {
77                                         try {
78                                                 this.outStream.write(demoPacket.getOriginalLengthAsByte());
79                                                 this.outStream.write(demoPacket.getAngles());
80                                                 this.outStream.write(demoPacket.getOriginalData());
81                                         } catch (IOException e) {
82                                                 throw new DemoCutterException("Unexpected I/O Exception occurred when writing to the cut demo", e);
83                                         }
84                                         
85                                         continue;
86                                 }
87
88                                 if (demoPacket.getSvcTime() != -1) {
89                                         svctime = demoPacket.getSvcTime();
90                                 }
91
92                                 if (svctime != -1) {
93                                         if (firstSvcTime == -1) {
94                                                 firstSvcTime = svctime;
95                                         }
96                                         lastSvcTime = svctime;
97                                         
98                                         if (firstLoop) {
99                                                 injectBuffer = "\011\n" + injectAtStart + ";slowmo " + ffwSpeedFirstStage + "\n\000";
100                                                 firstLoop = false;
101                                         } else {
102                                                 if (demoStarted < 1 && svctime > (startTime - 50)) {
103                                                         injectBuffer = "\011\nslowmo " + ffwSpeedSecondStage + "\n\000";
104                                                         demoStarted = 1;
105                                                 }
106                                                 if (demoStarted < 2 && svctime > (startTime - 5)) {
107                                                         injectBuffer = "\011\nslowmo 1;" + injectBeforeCap +"\n\000";
108                                                         demoStarted = 2;
109                                                 }
110                                                 if (demoStarted < 3 && svctime > startTime) {
111                                                         injectBuffer = "\011\ncl_capturevideo 1\n\000";
112                                                         demoStarted = 3;
113                                                 }
114                                                 if (!endIsReached && svctime > endTime) {
115                                                         injectBuffer = "\011\ncl_capturevideo 0\n\000";
116                                                         endIsReached = true;
117                                                 }
118                                                 if (endIsReached && !finalInjectionDone && svctime > (endTime + 1)) {
119                                                         injectBuffer = "\011\n" + injectAfterCap + "\n\000";
120                                                         finalInjectionDone = true;
121                                                 }
122                                                 if (finalInjectionDone && !disconnectIssued && svctime > (endTime + 2)) {
123                                                         injectBuffer = "\011\ndisconnect\n\000";
124                                                         disconnectIssued = true;
125                                                 }
126                                         }
127                                 }
128
129                                 byte[] injectBufferAsBytes = null;
130                                 try {
131                                         injectBufferAsBytes = injectBuffer.getBytes("US-ASCII");
132                                 } catch (UnsupportedEncodingException e) {
133                                         throw new DemoCutterException("Could not convert String to bytes using US-ASCII charset!", e);
134                                 }
135
136                                 data = demoPacket.getOriginalData();
137                                 if ((injectBufferAsBytes.length + data.length) < 65536) {
138                                         data = DemoCutterUtils.mergeByteArrays(injectBufferAsBytes, data);
139                                         injectBuffer = "";
140                                 }
141                                 
142                                 byte[] newLengthLittleEndian = DemoCutterUtils.convertLittleEndian(data.length);
143                                 try {
144                                         this.outStream.write(newLengthLittleEndian);
145                                         this.outStream.write(demoPacket.getAngles());
146                                         this.outStream.write(data);
147                                 } catch (IOException e) {
148                                         throw new DemoCutterException("Unexpected I/O Exception occurred when writing to the cut demo", e);
149                                 }
150
151                         }
152                         
153                         if (startTime < firstSvcTime) {
154                                 throw new DemoCutterException("Start time for the demo is " + startTime + ", but demo doesn't start before " + firstSvcTime);
155                         }
156                         if (endTime > lastSvcTime) {
157                                 throw new DemoCutterException("End time for the demo is " + endTime + ", but demo already stops at " + lastSvcTime);
158                         }
159                 } catch (DemoCutterException e) {
160                         throw e;
161                 } catch (Throwable e) {
162                         throw new DemoCutterException("Internal error in demo cutter sub-route (invalid demo file?)", e);
163                 } finally {
164                         try {
165                                 this.outStream.close();
166                                 this.inStream.close();
167                         } catch (IOException e) {}
168                 }
169         }
170
171         
172
173         /**
174          * Seeks forward in the inStream until CDTRACK_SEPARATOR byte was reached.
175          * All the content is copied to the outStream.
176          */
177         private void readCDTrack() {
178                 byte lastByte;
179                 try {
180                         while ((lastByte = inStream.readByte()) != CDTRACK_SEPARATOR) {
181                                 this.outStream.write(lastByte);
182                         }
183                         this.outStream.write(CDTRACK_SEPARATOR);
184                 } catch (EOFException e) {
185                         throw new DemoCutterException("Unexpected EOF occurred when reading CD track of demo " + inFile.getPath(), e);
186                 }
187                 catch (IOException e) {
188                         throw new DemoCutterException("Unexpected I/O Exception occurred when reading CD track of demo " + inFile.getPath(), e);
189                 }
190         }
191
192         private void prepareStreams() {
193                 try {
194                         this.inStream = new DataInputStream(new FileInputStream(this.inFile));
195                 } catch (FileNotFoundException e) {
196                         throw new DemoCutterException("Could not open demo file " + inFile.getPath(), e);
197                 }
198                 
199                 try {
200                         this.outStream = new DataOutputStream(new FileOutputStream(this.outFile));
201                 } catch (FileNotFoundException e) {
202                         throw new DemoCutterException("Could not open demo file " + outFile.getPath(), e);
203                 }
204         }
205         
206         private String checkInjectString(String injectionString) {
207                 while (injectionString.endsWith(";") || injectionString.endsWith("\n")) {
208                         injectionString = injectionString.substring(0, injectionString.length()-1);
209                 }
210                 return injectionString;
211         }
212 }