Category

Development Platform
Home > SourceCode/Document > Compress-Decompress algrithms > View Code
main.cpp
Package: solitude.rar　[view]
Upload User: szqxhk
Upload Date: 2010-02-15
Package Size: 54k
Code Size: 3k
Category:

Compress-Decompress algrithms

Development Platform:

Visual C++

main.cpp：Code Content
  1. #include <windows.h>
  2. #include <stdio.h>
  3. #include <memory.h>
  5. #include "lz77.h"
  7. //////////////////////////////////////////////////////////////////
  8. // out file format:
  9. // 0;flag2;buffer;0;flag2;buffer;...flag1;flag2;bufferlast
  10. // flag1 - 2 bytes, source buffer block length
  11. // if block size is 65536, be zero
  12. // flag2 - 2 bytes, compressed buffer length
  13. // if can not compress, be same with flag1
  14. //////////////////////////////////////////////////////////////////
  17. void main(int argc, char* argv[])
  18. {
  19. if (argc != 4)
  20. {
  21. puts("Usage: ");
  22. printf("    Compress : %s c sourcefile destfilen", argv[0]);
  23. printf("  Decompress : %s d sourcefile destfilen", argv[0]);
  24. return;
  25. }
  27. BYTE soubuf[65536];
  28. BYTE destbuf[65536 + 16];
  30. FILE* in;
  31. FILE* out;
  32. in = fopen(argv[2], "rb");
  33. if (in == NULL)
  34. {
  35. puts("Can't open source file");
  36. return;
  37. }
  38. out = fopen(argv[3], "wb");
  39. if (out == NULL)
  40. {
  41. puts("Can't open dest file");
  42. fclose(in);
  43. return;
  44. }
  45. fseek(in, 0, SEEK_END);
  46. long soulen = ftell(in);
  47. fseek(in, 0, SEEK_SET);
  49. CCompressLZ77 cc;
  50. WORD flag1, flag2;
  52. if (argv[1][0] == 'c') // compress
  53. {
  54. int last = soulen, act;
  55. while ( last > 0 )
  56. {
  57. act = min(65536, last);
  58. fread(soubuf, act, 1, in);
  59. last -= act;
  60. if (act == 65536) // out 65536 bytes
  61. flag1 = 0;
  62. else // out last blocks
  63. flag1 = act;
  64. fwrite(&flag1, sizeof(WORD), 1, out);
  66. int destlen = cc.Compress((BYTE*)soubuf, act, (BYTE*)destbuf);
  67. if (destlen == 0) // can't compress the block
  68. {
  69. flag2 = flag1;
  70. fwrite(&flag2, sizeof(WORD), 1, out);
  71. fwrite(soubuf, act, 1, out);
  72. }
  73. else
  74. {
  75. flag2 = (WORD)destlen;
  76. fwrite(&flag2, sizeof(WORD), 1, out);
  77. fwrite(destbuf, destlen, 1, out);
  78. }
  79. }
  80. }
  81. else if (argv[1][0] == 'd') // decompress
  82. {
  83. int last = soulen, act;
  84. while (last > 0)
  85. {
  86. fread(&flag1, sizeof(WORD), 1, in);
  87. fread(&flag2, sizeof(WORD), 1, in);
  88. last -= 2 * sizeof(WORD);
  89. if (flag1 == 0)
  90. act = 65536;
  91. else
  92. act = flag1;
  93. last-= flag2 ? (flag2) : act;
  95. if (flag2 == flag1)
  96. {
  97. fread(soubuf, act, 1, in);
  98. }
  99. else
  100. {
  101. fread(destbuf, flag2, 1, in);
  102. if (!cc.Decompress((BYTE*)soubuf, act, (BYTE*)destbuf))
  103. {
  104. puts("Decompress error");
  105. fclose(in);
  106. fclose(out);
  107. return;
  108. }
  109. }
  110. fwrite((BYTE*)soubuf, act, 1, out);
  111. }
  112. }
  113. else
  114. {
  115. puts("Usage: ");
  116. printf("    Compress : %s c sourcefile destfilen", argv[0]);
  117. printf("  Decompress : %s d sourcefile destfilen", argv[0]);
  118. }
  120. fclose(in);
  121. fclose(out);
  122. }