]>
Commit | Line | Data |
---|---|---|
7c673cae FG |
1 | ---------------------------------------------------------------- |
2 | -- ZLib for Ada thick binding. -- | |
3 | -- -- | |
4 | -- Copyright (C) 2002-2003 Dmitriy Anisimkov -- | |
5 | -- -- | |
6 | -- Open source license information is in the zlib.ads file. -- | |
7 | ---------------------------------------------------------------- | |
8 | ||
9 | -- $Id: zlib-streams.adb,v 1.10 2004/05/31 10:53:40 vagul Exp $ | |
10 | ||
11 | with Ada.Unchecked_Deallocation; | |
12 | ||
13 | package body ZLib.Streams is | |
14 | ||
15 | ----------- | |
16 | -- Close -- | |
17 | ----------- | |
18 | ||
19 | procedure Close (Stream : in out Stream_Type) is | |
20 | procedure Free is new Ada.Unchecked_Deallocation | |
21 | (Stream_Element_Array, Buffer_Access); | |
22 | begin | |
23 | if Stream.Mode = Out_Stream or Stream.Mode = Duplex then | |
24 | -- We should flush the data written by the writer. | |
25 | ||
26 | Flush (Stream, Finish); | |
27 | ||
28 | Close (Stream.Writer); | |
29 | end if; | |
30 | ||
31 | if Stream.Mode = In_Stream or Stream.Mode = Duplex then | |
32 | Close (Stream.Reader); | |
33 | Free (Stream.Buffer); | |
34 | end if; | |
35 | end Close; | |
36 | ||
37 | ------------ | |
38 | -- Create -- | |
39 | ------------ | |
40 | ||
41 | procedure Create | |
42 | (Stream : out Stream_Type; | |
43 | Mode : in Stream_Mode; | |
44 | Back : in Stream_Access; | |
45 | Back_Compressed : in Boolean; | |
46 | Level : in Compression_Level := Default_Compression; | |
47 | Strategy : in Strategy_Type := Default_Strategy; | |
48 | Header : in Header_Type := Default; | |
49 | Read_Buffer_Size : in Ada.Streams.Stream_Element_Offset | |
50 | := Default_Buffer_Size; | |
51 | Write_Buffer_Size : in Ada.Streams.Stream_Element_Offset | |
52 | := Default_Buffer_Size) | |
53 | is | |
54 | ||
55 | subtype Buffer_Subtype is Stream_Element_Array (1 .. Read_Buffer_Size); | |
56 | ||
57 | procedure Init_Filter | |
58 | (Filter : in out Filter_Type; | |
59 | Compress : in Boolean); | |
60 | ||
61 | ----------------- | |
62 | -- Init_Filter -- | |
63 | ----------------- | |
64 | ||
65 | procedure Init_Filter | |
66 | (Filter : in out Filter_Type; | |
67 | Compress : in Boolean) is | |
68 | begin | |
69 | if Compress then | |
70 | Deflate_Init | |
71 | (Filter, Level, Strategy, Header => Header); | |
72 | else | |
73 | Inflate_Init (Filter, Header => Header); | |
74 | end if; | |
75 | end Init_Filter; | |
76 | ||
77 | begin | |
78 | Stream.Back := Back; | |
79 | Stream.Mode := Mode; | |
80 | ||
81 | if Mode = Out_Stream or Mode = Duplex then | |
82 | Init_Filter (Stream.Writer, Back_Compressed); | |
83 | Stream.Buffer_Size := Write_Buffer_Size; | |
84 | else | |
85 | Stream.Buffer_Size := 0; | |
86 | end if; | |
87 | ||
88 | if Mode = In_Stream or Mode = Duplex then | |
89 | Init_Filter (Stream.Reader, not Back_Compressed); | |
90 | ||
91 | Stream.Buffer := new Buffer_Subtype; | |
92 | Stream.Rest_First := Stream.Buffer'Last + 1; | |
93 | Stream.Rest_Last := Stream.Buffer'Last; | |
94 | end if; | |
95 | end Create; | |
96 | ||
97 | ----------- | |
98 | -- Flush -- | |
99 | ----------- | |
100 | ||
101 | procedure Flush | |
102 | (Stream : in out Stream_Type; | |
103 | Mode : in Flush_Mode := Sync_Flush) | |
104 | is | |
105 | Buffer : Stream_Element_Array (1 .. Stream.Buffer_Size); | |
106 | Last : Stream_Element_Offset; | |
107 | begin | |
108 | loop | |
109 | Flush (Stream.Writer, Buffer, Last, Mode); | |
110 | ||
111 | Ada.Streams.Write (Stream.Back.all, Buffer (1 .. Last)); | |
112 | ||
113 | exit when Last < Buffer'Last; | |
114 | end loop; | |
115 | end Flush; | |
116 | ||
117 | ------------- | |
118 | -- Is_Open -- | |
119 | ------------- | |
120 | ||
121 | function Is_Open (Stream : Stream_Type) return Boolean is | |
122 | begin | |
123 | return Is_Open (Stream.Reader) or else Is_Open (Stream.Writer); | |
124 | end Is_Open; | |
125 | ||
126 | ---------- | |
127 | -- Read -- | |
128 | ---------- | |
129 | ||
130 | procedure Read | |
131 | (Stream : in out Stream_Type; | |
132 | Item : out Stream_Element_Array; | |
133 | Last : out Stream_Element_Offset) | |
134 | is | |
135 | ||
136 | procedure Read | |
137 | (Item : out Stream_Element_Array; | |
138 | Last : out Stream_Element_Offset); | |
139 | ||
140 | ---------- | |
141 | -- Read -- | |
142 | ---------- | |
143 | ||
144 | procedure Read | |
145 | (Item : out Stream_Element_Array; | |
146 | Last : out Stream_Element_Offset) is | |
147 | begin | |
148 | Ada.Streams.Read (Stream.Back.all, Item, Last); | |
149 | end Read; | |
150 | ||
151 | procedure Read is new ZLib.Read | |
152 | (Read => Read, | |
153 | Buffer => Stream.Buffer.all, | |
154 | Rest_First => Stream.Rest_First, | |
155 | Rest_Last => Stream.Rest_Last); | |
156 | ||
157 | begin | |
158 | Read (Stream.Reader, Item, Last); | |
159 | end Read; | |
160 | ||
161 | ------------------- | |
162 | -- Read_Total_In -- | |
163 | ------------------- | |
164 | ||
165 | function Read_Total_In (Stream : in Stream_Type) return Count is | |
166 | begin | |
167 | return Total_In (Stream.Reader); | |
168 | end Read_Total_In; | |
169 | ||
170 | -------------------- | |
171 | -- Read_Total_Out -- | |
172 | -------------------- | |
173 | ||
174 | function Read_Total_Out (Stream : in Stream_Type) return Count is | |
175 | begin | |
176 | return Total_Out (Stream.Reader); | |
177 | end Read_Total_Out; | |
178 | ||
179 | ----------- | |
180 | -- Write -- | |
181 | ----------- | |
182 | ||
183 | procedure Write | |
184 | (Stream : in out Stream_Type; | |
185 | Item : in Stream_Element_Array) | |
186 | is | |
187 | ||
188 | procedure Write (Item : in Stream_Element_Array); | |
189 | ||
190 | ----------- | |
191 | -- Write -- | |
192 | ----------- | |
193 | ||
194 | procedure Write (Item : in Stream_Element_Array) is | |
195 | begin | |
196 | Ada.Streams.Write (Stream.Back.all, Item); | |
197 | end Write; | |
198 | ||
199 | procedure Write is new ZLib.Write | |
200 | (Write => Write, | |
201 | Buffer_Size => Stream.Buffer_Size); | |
202 | ||
203 | begin | |
204 | Write (Stream.Writer, Item, No_Flush); | |
205 | end Write; | |
206 | ||
207 | -------------------- | |
208 | -- Write_Total_In -- | |
209 | -------------------- | |
210 | ||
211 | function Write_Total_In (Stream : in Stream_Type) return Count is | |
212 | begin | |
213 | return Total_In (Stream.Writer); | |
214 | end Write_Total_In; | |
215 | ||
216 | --------------------- | |
217 | -- Write_Total_Out -- | |
218 | --------------------- | |
219 | ||
220 | function Write_Total_Out (Stream : in Stream_Type) return Count is | |
221 | begin | |
222 | return Total_Out (Stream.Writer); | |
223 | end Write_Total_Out; | |
224 | ||
225 | end ZLib.Streams; |