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