| 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 |
| 172 | else printf("file %s read, %u bytes\n",argv[1],lFileSize);\r |
| 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 |