1 //
2 // � Copyright Henrik Ravn 2004
3 //
4 // Use, modification and distribution are subject to the Boost Software License, Version 1.0.
5 // (See accompanying file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
6 //
7 
8 using System;
9 using System.Diagnostics;
10 using System.Runtime.InteropServices;
11 
12 namespace DotZLib
13 {
14 
15     /// <summary>
16     /// Implements a data compressor, using the deflate algorithm in the ZLib dll
17     /// </summary>
18 	public sealed class Deflater : CodecBase
19 	{
20         #region Dll imports
21         [DllImport("ZLIB1.dll", CallingConvention=CallingConvention.Cdecl, CharSet=CharSet.Ansi)]
deflateInit_(ref ZStream sz, int level, string vs, int size)22         private static extern int deflateInit_(ref ZStream sz, int level, string vs, int size);
23 
24         [DllImport("ZLIB1.dll", CallingConvention=CallingConvention.Cdecl)]
deflate(ref ZStream sz, int flush)25         private static extern int deflate(ref ZStream sz, int flush);
26 
27         [DllImport("ZLIB1.dll", CallingConvention=CallingConvention.Cdecl)]
deflateReset(ref ZStream sz)28         private static extern int deflateReset(ref ZStream sz);
29 
30         [DllImport("ZLIB1.dll", CallingConvention=CallingConvention.Cdecl)]
deflateEnd(ref ZStream sz)31         private static extern int deflateEnd(ref ZStream sz);
32         #endregion
33 
34         /// <summary>
35         /// Constructs an new instance of the <c>Deflater</c>
36         /// </summary>
37         /// <param name="level">The compression level to use for this <c>Deflater</c></param>
Deflater(CompressLevel level)38 		public Deflater(CompressLevel level) : base()
39 		{
40             int retval = deflateInit_(ref _ztream, (int)level, Info.Version, Marshal.SizeOf(_ztream));
41             if (retval != 0)
42                 throw new ZLibException(retval, "Could not initialize deflater");
43 
44             resetOutput();
45 		}
46 
47         /// <summary>
48         /// Adds more data to the codec to be processed.
49         /// </summary>
50         /// <param name="data">Byte array containing the data to be added to the codec</param>
51         /// <param name="offset">The index of the first byte to add from <c>data</c></param>
52         /// <param name="count">The number of bytes to add</param>
53         /// <remarks>Adding data may, or may not, raise the <c>DataAvailable</c> event</remarks>
Add(byte[] data, int offset, int count)54         public override void Add(byte[] data, int offset, int count)
55         {
56             if (data == null) throw new ArgumentNullException();
57             if (offset < 0 || count < 0) throw new ArgumentOutOfRangeException();
58             if ((offset+count) > data.Length) throw new ArgumentException();
59 
60             int total = count;
61             int inputIndex = offset;
62             int err = 0;
63 
64             while (err >= 0 && inputIndex < total)
65             {
66                 copyInput(data, inputIndex, Math.Min(total - inputIndex, kBufferSize));
67                 while (err >= 0 && _ztream.avail_in > 0)
68                 {
69                     err = deflate(ref _ztream, (int)FlushTypes.None);
70                     if (err == 0)
71                         while (_ztream.avail_out == 0)
72                         {
73                             OnDataAvailable();
74                             err = deflate(ref _ztream, (int)FlushTypes.None);
75                         }
76                     inputIndex += (int)_ztream.total_in;
77                 }
78             }
79             setChecksum( _ztream.adler );
80         }
81 
82 
83         /// <summary>
84         /// Finishes up any pending data that needs to be processed and handled.
85         /// </summary>
Finish()86         public override void Finish()
87         {
88             int err;
89             do
90             {
91                 err = deflate(ref _ztream, (int)FlushTypes.Finish);
92                 OnDataAvailable();
93             }
94             while (err == 0);
95             setChecksum( _ztream.adler );
96             deflateReset(ref _ztream);
97             resetOutput();
98         }
99 
100         /// <summary>
101         /// Closes the internal zlib deflate stream
102         /// </summary>
CleanUp()103         protected override void CleanUp() { deflateEnd(ref _ztream); }
104 
105     }
106 }
107