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; |