1 | #include <stdio.h>
|
---|
2 | #include <stdlib.h>
|
---|
3 | #include <windows.h>
|
---|
4 |
|
---|
5 | #include "zlib.h"
|
---|
6 |
|
---|
7 |
|
---|
8 | void MyDoMinus64(LARGE_INTEGER *R,LARGE_INTEGER A,LARGE_INTEGER B)
|
---|
9 | {
|
---|
10 | R->HighPart = A.HighPart - B.HighPart;
|
---|
11 | if (A.LowPart >= B.LowPart)
|
---|
12 | R->LowPart = A.LowPart - B.LowPart;
|
---|
13 | else
|
---|
14 | {
|
---|
15 | R->LowPart = A.LowPart - B.LowPart;
|
---|
16 | R->HighPart --;
|
---|
17 | }
|
---|
18 | }
|
---|
19 |
|
---|
20 | #ifdef _M_X64
|
---|
21 | // see http://msdn2.microsoft.com/library/twchhe95(en-us,vs.80).aspx for __rdtsc
|
---|
22 | unsigned __int64 __rdtsc(void);
|
---|
23 | void BeginCountRdtsc(LARGE_INTEGER * pbeginTime64)
|
---|
24 | {
|
---|
25 | // printf("rdtsc = %I64x\n",__rdtsc());
|
---|
26 | pbeginTime64->QuadPart=__rdtsc();
|
---|
27 | }
|
---|
28 |
|
---|
29 | LARGE_INTEGER GetResRdtsc(LARGE_INTEGER beginTime64,BOOL fComputeTimeQueryPerf)
|
---|
30 | {
|
---|
31 | LARGE_INTEGER LIres;
|
---|
32 | unsigned _int64 res=__rdtsc()-((unsigned _int64)(beginTime64.QuadPart));
|
---|
33 | LIres.QuadPart=res;
|
---|
34 | // printf("rdtsc = %I64x\n",__rdtsc());
|
---|
35 | return LIres;
|
---|
36 | }
|
---|
37 | #else
|
---|
38 | #ifdef _M_IX86
|
---|
39 | void myGetRDTSC32(LARGE_INTEGER * pbeginTime64)
|
---|
40 | {
|
---|
41 | DWORD dwEdx,dwEax;
|
---|
42 | _asm
|
---|
43 | {
|
---|
44 | rdtsc
|
---|
45 | mov dwEax,eax
|
---|
46 | mov dwEdx,edx
|
---|
47 | }
|
---|
48 | pbeginTime64->LowPart=dwEax;
|
---|
49 | pbeginTime64->HighPart=dwEdx;
|
---|
50 | }
|
---|
51 |
|
---|
52 | void BeginCountRdtsc(LARGE_INTEGER * pbeginTime64)
|
---|
53 | {
|
---|
54 | myGetRDTSC32(pbeginTime64);
|
---|
55 | }
|
---|
56 |
|
---|
57 | LARGE_INTEGER GetResRdtsc(LARGE_INTEGER beginTime64,BOOL fComputeTimeQueryPerf)
|
---|
58 | {
|
---|
59 | LARGE_INTEGER LIres,endTime64;
|
---|
60 | myGetRDTSC32(&endTime64);
|
---|
61 |
|
---|
62 | LIres.LowPart=LIres.HighPart=0;
|
---|
63 | MyDoMinus64(&LIres,endTime64,beginTime64);
|
---|
64 | return LIres;
|
---|
65 | }
|
---|
66 | #else
|
---|
67 | void myGetRDTSC32(LARGE_INTEGER * pbeginTime64)
|
---|
68 | {
|
---|
69 | }
|
---|
70 |
|
---|
71 | void BeginCountRdtsc(LARGE_INTEGER * pbeginTime64)
|
---|
72 | {
|
---|
73 | }
|
---|
74 |
|
---|
75 | LARGE_INTEGER GetResRdtsc(LARGE_INTEGER beginTime64,BOOL fComputeTimeQueryPerf)
|
---|
76 | {
|
---|
77 | LARGE_INTEGER lr;
|
---|
78 | lr.QuadPart=0;
|
---|
79 | return lr;
|
---|
80 | }
|
---|
81 | #endif
|
---|
82 | #endif
|
---|
83 |
|
---|
84 | void BeginCountPerfCounter(LARGE_INTEGER * pbeginTime64,BOOL fComputeTimeQueryPerf)
|
---|
85 | {
|
---|
86 | if ((!fComputeTimeQueryPerf) || (!QueryPerformanceCounter(pbeginTime64)))
|
---|
87 | {
|
---|
88 | pbeginTime64->LowPart = GetTickCount();
|
---|
89 | pbeginTime64->HighPart = 0;
|
---|
90 | }
|
---|
91 | }
|
---|
92 |
|
---|
93 | DWORD GetMsecSincePerfCounter(LARGE_INTEGER beginTime64,BOOL fComputeTimeQueryPerf)
|
---|
94 | {
|
---|
95 | LARGE_INTEGER endTime64,ticksPerSecond,ticks;
|
---|
96 | DWORDLONG ticksShifted,tickSecShifted;
|
---|
97 | DWORD dwLog=16+0;
|
---|
98 | DWORD dwRet;
|
---|
99 | if ((!fComputeTimeQueryPerf) || (!QueryPerformanceCounter(&endTime64)))
|
---|
100 | dwRet = (GetTickCount() - beginTime64.LowPart)*1;
|
---|
101 | else
|
---|
102 | {
|
---|
103 | MyDoMinus64(&ticks,endTime64,beginTime64);
|
---|
104 | QueryPerformanceFrequency(&ticksPerSecond);
|
---|
105 |
|
---|
106 |
|
---|
107 | {
|
---|
108 | ticksShifted = Int64ShrlMod32(*(DWORDLONG*)&ticks,dwLog);
|
---|
109 | tickSecShifted = Int64ShrlMod32(*(DWORDLONG*)&ticksPerSecond,dwLog);
|
---|
110 |
|
---|
111 | }
|
---|
112 |
|
---|
113 | dwRet = (DWORD)((((DWORD)ticksShifted)*1000)/(DWORD)(tickSecShifted));
|
---|
114 | dwRet *=1;
|
---|
115 | }
|
---|
116 | return dwRet;
|
---|
117 | }
|
---|
118 |
|
---|
119 | int ReadFileMemory(const char* filename,long* plFileSize,void** pFilePtr)
|
---|
120 | {
|
---|
121 | FILE* stream;
|
---|
122 | void* ptr;
|
---|
123 | int retVal=1;
|
---|
124 | stream=fopen(filename, "rb");
|
---|
125 | if (stream==NULL)
|
---|
126 | return 0;
|
---|
127 |
|
---|
128 | fseek(stream,0,SEEK_END);
|
---|
129 |
|
---|
130 | *plFileSize=ftell(stream);
|
---|
131 | fseek(stream,0,SEEK_SET);
|
---|
132 | ptr=malloc((*plFileSize)+1);
|
---|
133 | if (ptr==NULL)
|
---|
134 | retVal=0;
|
---|
135 | else
|
---|
136 | {
|
---|
137 | if (fread(ptr, 1, *plFileSize,stream) != (*plFileSize))
|
---|
138 | retVal=0;
|
---|
139 | }
|
---|
140 | fclose(stream);
|
---|
141 | *pFilePtr=ptr;
|
---|
142 | return retVal;
|
---|
143 | }
|
---|
144 |
|
---|
145 | int main(int argc, char *argv[])
|
---|
146 | {
|
---|
147 | int BlockSizeCompress=0x8000;
|
---|
148 | int BlockSizeUncompress=0x8000;
|
---|
149 | int cprLevel=Z_DEFAULT_COMPRESSION ;
|
---|
150 | long lFileSize;
|
---|
151 | unsigned char* FilePtr;
|
---|
152 | long lBufferSizeCpr;
|
---|
153 | long lBufferSizeUncpr;
|
---|
154 | long lCompressedSize=0;
|
---|
155 | unsigned char* CprPtr;
|
---|
156 | unsigned char* UncprPtr;
|
---|
157 | long lSizeCpr,lSizeUncpr;
|
---|
158 | DWORD dwGetTick,dwMsecQP;
|
---|
159 | LARGE_INTEGER li_qp,li_rdtsc,dwResRdtsc;
|
---|
160 |
|
---|
161 | if (argc<=1)
|
---|
162 | {
|
---|
163 | printf("run TestZlib <File> [BlockSizeCompress] [BlockSizeUncompress] [compres. level]\n");
|
---|
164 | return 0;
|
---|
165 | }
|
---|
166 |
|
---|
167 | if (ReadFileMemory(argv[1],&lFileSize,&FilePtr)==0)
|
---|
168 | {
|
---|
169 | printf("error reading %s\n",argv[1]);
|
---|
170 | return 1;
|
---|
171 | }
|
---|
172 | else printf("file %s read, %u bytes\n",argv[1],lFileSize);
|
---|
173 |
|
---|
174 | if (argc>=3)
|
---|
175 | BlockSizeCompress=atol(argv[2]);
|
---|
176 |
|
---|
177 | if (argc>=4)
|
---|
178 | BlockSizeUncompress=atol(argv[3]);
|
---|
179 |
|
---|
180 | if (argc>=5)
|
---|
181 | cprLevel=(int)atol(argv[4]);
|
---|
182 |
|
---|
183 | lBufferSizeCpr = lFileSize + (lFileSize/0x10) + 0x200;
|
---|
184 | lBufferSizeUncpr = lBufferSizeCpr;
|
---|
185 |
|
---|
186 | CprPtr=(unsigned char*)malloc(lBufferSizeCpr + BlockSizeCompress);
|
---|
187 |
|
---|
188 | BeginCountPerfCounter(&li_qp,TRUE);
|
---|
189 | dwGetTick=GetTickCount();
|
---|
190 | BeginCountRdtsc(&li_rdtsc);
|
---|
191 | {
|
---|
192 | z_stream zcpr;
|
---|
193 | int ret=Z_OK;
|
---|
194 | long lOrigToDo = lFileSize;
|
---|
195 | long lOrigDone = 0;
|
---|
196 | int step=0;
|
---|
197 | memset(&zcpr,0,sizeof(z_stream));
|
---|
198 | deflateInit(&zcpr,cprLevel);
|
---|
199 |
|
---|
200 | zcpr.next_in = FilePtr;
|
---|
201 | zcpr.next_out = CprPtr;
|
---|
202 |
|
---|
203 |
|
---|
204 | do
|
---|
205 | {
|
---|
206 | long all_read_before = zcpr.total_in;
|
---|
207 | zcpr.avail_in = min(lOrigToDo,BlockSizeCompress);
|
---|
208 | zcpr.avail_out = BlockSizeCompress;
|
---|
209 | ret=deflate(&zcpr,(zcpr.avail_in==lOrigToDo) ? Z_FINISH : Z_SYNC_FLUSH);
|
---|
210 | lOrigDone += (zcpr.total_in-all_read_before);
|
---|
211 | lOrigToDo -= (zcpr.total_in-all_read_before);
|
---|
212 | step++;
|
---|
213 | } while (ret==Z_OK);
|
---|
214 |
|
---|
215 | lSizeCpr=zcpr.total_out;
|
---|
216 | deflateEnd(&zcpr);
|
---|
217 | dwGetTick=GetTickCount()-dwGetTick;
|
---|
218 | dwMsecQP=GetMsecSincePerfCounter(li_qp,TRUE);
|
---|
219 | dwResRdtsc=GetResRdtsc(li_rdtsc,TRUE);
|
---|
220 | printf("total compress size = %u, in %u step\n",lSizeCpr,step);
|
---|
221 | printf("time = %u msec = %f sec\n",dwGetTick,dwGetTick/(double)1000.);
|
---|
222 | printf("defcpr time QP = %u msec = %f sec\n",dwMsecQP,dwMsecQP/(double)1000.);
|
---|
223 | printf("defcpr result rdtsc = %I64x\n\n",dwResRdtsc.QuadPart);
|
---|
224 | }
|
---|
225 |
|
---|
226 | CprPtr=(unsigned char*)realloc(CprPtr,lSizeCpr);
|
---|
227 | UncprPtr=(unsigned char*)malloc(lBufferSizeUncpr + BlockSizeUncompress);
|
---|
228 |
|
---|
229 | BeginCountPerfCounter(&li_qp,TRUE);
|
---|
230 | dwGetTick=GetTickCount();
|
---|
231 | BeginCountRdtsc(&li_rdtsc);
|
---|
232 | {
|
---|
233 | z_stream zcpr;
|
---|
234 | int ret=Z_OK;
|
---|
235 | long lOrigToDo = lSizeCpr;
|
---|
236 | long lOrigDone = 0;
|
---|
237 | int step=0;
|
---|
238 | memset(&zcpr,0,sizeof(z_stream));
|
---|
239 | inflateInit(&zcpr);
|
---|
240 |
|
---|
241 | zcpr.next_in = CprPtr;
|
---|
242 | zcpr.next_out = UncprPtr;
|
---|
243 |
|
---|
244 |
|
---|
245 | do
|
---|
246 | {
|
---|
247 | long all_read_before = zcpr.total_in;
|
---|
248 | zcpr.avail_in = min(lOrigToDo,BlockSizeUncompress);
|
---|
249 | zcpr.avail_out = BlockSizeUncompress;
|
---|
250 | ret=inflate(&zcpr,Z_SYNC_FLUSH);
|
---|
251 | lOrigDone += (zcpr.total_in-all_read_before);
|
---|
252 | lOrigToDo -= (zcpr.total_in-all_read_before);
|
---|
253 | step++;
|
---|
254 | } while (ret==Z_OK);
|
---|
255 |
|
---|
256 | lSizeUncpr=zcpr.total_out;
|
---|
257 | inflateEnd(&zcpr);
|
---|
258 | dwGetTick=GetTickCount()-dwGetTick;
|
---|
259 | dwMsecQP=GetMsecSincePerfCounter(li_qp,TRUE);
|
---|
260 | dwResRdtsc=GetResRdtsc(li_rdtsc,TRUE);
|
---|
261 | printf("total uncompress size = %u, in %u step\n",lSizeUncpr,step);
|
---|
262 | printf("time = %u msec = %f sec\n",dwGetTick,dwGetTick/(double)1000.);
|
---|
263 | printf("uncpr time QP = %u msec = %f sec\n",dwMsecQP,dwMsecQP/(double)1000.);
|
---|
264 | printf("uncpr result rdtsc = %I64x\n\n",dwResRdtsc.QuadPart);
|
---|
265 | }
|
---|
266 |
|
---|
267 | if (lSizeUncpr==lFileSize)
|
---|
268 | {
|
---|
269 | if (memcmp(FilePtr,UncprPtr,lFileSize)==0)
|
---|
270 | printf("compare ok\n");
|
---|
271 |
|
---|
272 | }
|
---|
273 |
|
---|
274 | return 0;
|
---|
275 | }
|
---|