repo sync | baa3858 | 2013-07-26 17:53:31 -0700 | [diff] [blame] | 1 | package SevenZip;
|
| 2 |
|
| 3 | public class LzmaAlone
|
| 4 | {
|
| 5 | static public class CommandLine
|
| 6 | {
|
| 7 | public static final int kEncode = 0;
|
| 8 | public static final int kDecode = 1;
|
| 9 | public static final int kBenchmak = 2;
|
| 10 |
|
| 11 | public int Command = -1;
|
| 12 | public int NumBenchmarkPasses = 10;
|
| 13 |
|
| 14 | public int DictionarySize = 1 << 23;
|
| 15 | public boolean DictionarySizeIsDefined = false;
|
| 16 |
|
| 17 | public int Lc = 3;
|
| 18 | public int Lp = 0;
|
| 19 | public int Pb = 2;
|
| 20 |
|
| 21 | public int Fb = 128;
|
| 22 | public boolean FbIsDefined = false;
|
| 23 |
|
| 24 | public boolean Eos = false;
|
| 25 |
|
| 26 | public int Algorithm = 2;
|
| 27 | public int MatchFinder = 1;
|
| 28 |
|
| 29 | public String InFile;
|
| 30 | public String OutFile;
|
| 31 |
|
| 32 | boolean ParseSwitch(String s)
|
| 33 | {
|
| 34 | if (s.startsWith("d"))
|
| 35 | {
|
| 36 | DictionarySize = 1 << Integer.parseInt(s.substring(1));
|
| 37 | DictionarySizeIsDefined = true;
|
| 38 | }
|
| 39 | else if (s.startsWith("fb"))
|
| 40 | {
|
| 41 | Fb = Integer.parseInt(s.substring(2));
|
| 42 | FbIsDefined = true;
|
| 43 | }
|
| 44 | else if (s.startsWith("a"))
|
| 45 | Algorithm = Integer.parseInt(s.substring(1));
|
| 46 | else if (s.startsWith("lc"))
|
| 47 | Lc = Integer.parseInt(s.substring(2));
|
| 48 | else if (s.startsWith("lp"))
|
| 49 | Lp = Integer.parseInt(s.substring(2));
|
| 50 | else if (s.startsWith("pb"))
|
| 51 | Pb = Integer.parseInt(s.substring(2));
|
| 52 | else if (s.startsWith("eos"))
|
| 53 | Eos = true;
|
| 54 | else if (s.startsWith("mf"))
|
| 55 | {
|
| 56 | String mfs = s.substring(2);
|
| 57 | if (mfs.equals("bt2"))
|
| 58 | MatchFinder = 0;
|
| 59 | else if (mfs.equals("bt4"))
|
| 60 | MatchFinder = 1;
|
| 61 | else if (mfs.equals("bt4b"))
|
| 62 | MatchFinder = 2;
|
| 63 | else
|
| 64 | return false;
|
| 65 | }
|
| 66 | else
|
| 67 | return false;
|
| 68 | return true;
|
| 69 | }
|
| 70 |
|
| 71 | public boolean Parse(String[] args) throws Exception
|
| 72 | {
|
| 73 | int pos = 0;
|
| 74 | boolean switchMode = true;
|
| 75 | for (int i = 0; i < args.length; i++)
|
| 76 | {
|
| 77 | String s = args[i];
|
| 78 | if (s.length() == 0)
|
| 79 | return false;
|
| 80 | if (switchMode)
|
| 81 | {
|
| 82 | if (s.compareTo("--") == 0)
|
| 83 | {
|
| 84 | switchMode = false;
|
| 85 | continue;
|
| 86 | }
|
| 87 | if (s.charAt(0) == '-')
|
| 88 | {
|
| 89 | String sw = s.substring(1).toLowerCase();
|
| 90 | if (sw.length() == 0)
|
| 91 | return false;
|
| 92 | try
|
| 93 | {
|
| 94 | if (!ParseSwitch(sw))
|
| 95 | return false;
|
| 96 | }
|
| 97 | catch (NumberFormatException e)
|
| 98 | {
|
| 99 | return false;
|
| 100 | }
|
| 101 | continue;
|
| 102 | }
|
| 103 | }
|
| 104 | if (pos == 0)
|
| 105 | {
|
| 106 | if (s.equalsIgnoreCase("e"))
|
| 107 | Command = kEncode;
|
| 108 | else if (s.equalsIgnoreCase("d"))
|
| 109 | Command = kDecode;
|
| 110 | else if (s.equalsIgnoreCase("b"))
|
| 111 | Command = kBenchmak;
|
| 112 | else
|
| 113 | return false;
|
| 114 | }
|
| 115 | else if(pos == 1)
|
| 116 | {
|
| 117 | if (Command == kBenchmak)
|
| 118 | {
|
| 119 | try
|
| 120 | {
|
| 121 | NumBenchmarkPasses = Integer.parseInt(s);
|
| 122 | if (NumBenchmarkPasses < 1)
|
| 123 | return false;
|
| 124 | }
|
| 125 | catch (NumberFormatException e)
|
| 126 | {
|
| 127 | return false;
|
| 128 | }
|
| 129 | }
|
| 130 | else
|
| 131 | InFile = s;
|
| 132 | }
|
| 133 | else if(pos == 2)
|
| 134 | OutFile = s;
|
| 135 | else
|
| 136 | return false;
|
| 137 | pos++;
|
| 138 | continue;
|
| 139 | }
|
| 140 | return true;
|
| 141 | }
|
| 142 | }
|
| 143 |
|
| 144 |
|
| 145 | static void PrintHelp()
|
| 146 | {
|
| 147 | System.out.println(
|
| 148 | "\nUsage: LZMA <e|d> [<switches>...] inputFile outputFile\n" +
|
| 149 | " e: encode file\n" +
|
| 150 | " d: decode file\n" +
|
| 151 | " b: Benchmark\n" +
|
| 152 | "<Switches>\n" +
|
| 153 | // " -a{N}: set compression mode - [0, 1], default: 1 (max)\n" +
|
| 154 | " -d{N}: set dictionary - [0,28], default: 23 (8MB)\n" +
|
| 155 | " -fb{N}: set number of fast bytes - [5, 273], default: 128\n" +
|
| 156 | " -lc{N}: set number of literal context bits - [0, 8], default: 3\n" +
|
| 157 | " -lp{N}: set number of literal pos bits - [0, 4], default: 0\n" +
|
| 158 | " -pb{N}: set number of pos bits - [0, 4], default: 2\n" +
|
| 159 | " -mf{MF_ID}: set Match Finder: [bt2, bt4], default: bt4\n" +
|
| 160 | " -eos: write End Of Stream marker\n"
|
| 161 | );
|
| 162 | }
|
| 163 |
|
| 164 | public static void main(String[] args) throws Exception
|
| 165 | {
|
| 166 | System.out.println("\nLZMA (Java) 4.61 2008-11-23\n");
|
| 167 |
|
| 168 | if (args.length < 1)
|
| 169 | {
|
| 170 | PrintHelp();
|
| 171 | return;
|
| 172 | }
|
| 173 |
|
| 174 | CommandLine params = new CommandLine();
|
| 175 | if (!params.Parse(args))
|
| 176 | {
|
| 177 | System.out.println("\nIncorrect command");
|
| 178 | return;
|
| 179 | }
|
| 180 |
|
| 181 | if (params.Command == CommandLine.kBenchmak)
|
| 182 | {
|
| 183 | int dictionary = (1 << 21);
|
| 184 | if (params.DictionarySizeIsDefined)
|
| 185 | dictionary = params.DictionarySize;
|
| 186 | if (params.MatchFinder > 1)
|
| 187 | throw new Exception("Unsupported match finder");
|
| 188 | SevenZip.LzmaBench.LzmaBenchmark(params.NumBenchmarkPasses, dictionary);
|
| 189 | }
|
| 190 | else if (params.Command == CommandLine.kEncode || params.Command == CommandLine.kDecode)
|
| 191 | {
|
| 192 | java.io.File inFile = new java.io.File(params.InFile);
|
| 193 | java.io.File outFile = new java.io.File(params.OutFile);
|
| 194 |
|
| 195 | java.io.BufferedInputStream inStream = new java.io.BufferedInputStream(new java.io.FileInputStream(inFile));
|
| 196 | java.io.BufferedOutputStream outStream = new java.io.BufferedOutputStream(new java.io.FileOutputStream(outFile));
|
| 197 |
|
| 198 | boolean eos = false;
|
| 199 | if (params.Eos)
|
| 200 | eos = true;
|
| 201 | if (params.Command == CommandLine.kEncode)
|
| 202 | {
|
| 203 | SevenZip.Compression.LZMA.Encoder encoder = new SevenZip.Compression.LZMA.Encoder();
|
| 204 | if (!encoder.SetAlgorithm(params.Algorithm))
|
| 205 | throw new Exception("Incorrect compression mode");
|
| 206 | if (!encoder.SetDictionarySize(params.DictionarySize))
|
| 207 | throw new Exception("Incorrect dictionary size");
|
| 208 | if (!encoder.SetNumFastBytes(params.Fb))
|
| 209 | throw new Exception("Incorrect -fb value");
|
| 210 | if (!encoder.SetMatchFinder(params.MatchFinder))
|
| 211 | throw new Exception("Incorrect -mf value");
|
| 212 | if (!encoder.SetLcLpPb(params.Lc, params.Lp, params.Pb))
|
| 213 | throw new Exception("Incorrect -lc or -lp or -pb value");
|
| 214 | encoder.SetEndMarkerMode(eos);
|
| 215 | encoder.WriteCoderProperties(outStream);
|
| 216 | long fileSize;
|
| 217 | if (eos)
|
| 218 | fileSize = -1;
|
| 219 | else
|
| 220 | fileSize = inFile.length();
|
| 221 | for (int i = 0; i < 8; i++)
|
| 222 | outStream.write((int)(fileSize >>> (8 * i)) & 0xFF);
|
| 223 | encoder.Code(inStream, outStream, -1, -1, null);
|
| 224 | }
|
| 225 | else
|
| 226 | {
|
| 227 | int propertiesSize = 5;
|
| 228 | byte[] properties = new byte[propertiesSize];
|
| 229 | if (inStream.read(properties, 0, propertiesSize) != propertiesSize)
|
| 230 | throw new Exception("input .lzma file is too short");
|
| 231 | SevenZip.Compression.LZMA.Decoder decoder = new SevenZip.Compression.LZMA.Decoder();
|
| 232 | if (!decoder.SetDecoderProperties(properties))
|
| 233 | throw new Exception("Incorrect stream properties");
|
| 234 | long outSize = 0;
|
| 235 | for (int i = 0; i < 8; i++)
|
| 236 | {
|
| 237 | int v = inStream.read();
|
| 238 | if (v < 0)
|
| 239 | throw new Exception("Can't read stream size");
|
| 240 | outSize |= ((long)v) << (8 * i);
|
| 241 | }
|
| 242 | if (!decoder.Code(inStream, outStream, outSize))
|
| 243 | throw new Exception("Error in data stream");
|
| 244 | }
|
| 245 | outStream.flush();
|
| 246 | outStream.close();
|
| 247 | inStream.close();
|
| 248 | }
|
| 249 | else
|
| 250 | throw new Exception("Incorrect command");
|
| 251 | return;
|
| 252 | }
|
| 253 | }
|