Home | History | Annotate | Download | only in SevenZip
      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 }
    254