开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > Shell编程 > 查看源码
controller.cpp
资源名称：treesize.zip　[点击查看]
上传用户：dfhlxjd
上传日期：2007-01-07
资源大小：12k
文件大小：4k
源码类别：

Shell编程

开发平台：

Visual C++

controller.cpp：源码内容
  1. //----------------------------------------
  2. // (c) Reliable Software 1997
  3. //----------------------------------------
  4. #include <windows.h>
  5. #include <iomanip.h>
  6. #include "controls.h"
  7. #include "controller.h"
  8. #include "main.h"
  9. #include "shell95.h"
  10. #include "visitor.h"
  12. Controller::Controller (HWND hwnd)
  13. : _hwnd (hwnd),
  14. _pathField (hwnd, IDC_PATHNAME),
  15. _browse (hwnd, IDC_BROWSE),
  16. _calculate (hwnd, IDC_CALCULATE),
  17. _resultField (hwnd, IDC_RESULT),
  18. _exit (hwnd, IDC_EXIT),
  19. _clear (hwnd, IDC_NEW)
  20. {
  21.     _pathField.SetFocus();
  22. _resultField.SetString("-- No directory selected --");
  23. }
  25. class PathSelector
  26. {
  27. public:
  28. PathSelector(HWND hwnd)
  29. {
  30. _path [0] = '';
  31. Desktop desktop;
  32. ShPath browseRoot (desktop);
  33. if (browseRoot.IsOK ())
  34. {
  35.     FolderBrowser browser (hwnd,
  36.                           browseRoot,
  37.                           BIF_RETURNONLYFSDIRS,
  38.                           "Select folder to analyze");
  39.     if (browser.IsOK ())
  40.     {
  41.         strcpy (_path, browser.GetPath ());
  42.     }
  43. }
  44. }
  45. bool IsValid() const {return _path[0] != '';}
  46. char const * GetPath() const {return _path;}
  47. private:
  48. char _path[MAX_PATH + 1];
  49. };
  51. void Controller::Command (HWND hwnd, int controlID, int command)
  52. {
  53.     switch (controlID)
  54.     {
  55.         case IDC_PATHNAME:
  56.             if (_pathField.IsChanged(command))
  57.             {
  58.                 if (_pathField.GetLength())
  59. {
  60.                     _calculate.Enable();
  61. _clear.Enable();
  62.                 }
  63.                 else
  64. {
  65.                     _calculate.Disable();
  66. _clear.Disable();
  67. }
  68.             }
  69.             break;
  70. case IDC_BROWSE:
  71.             if (command == BN_CLICKED)
  72. {
  73. PathSelector userPath(hwnd);
  74. if (userPath.IsValid())
  75.                 {
  76.                     _pathField.SetString(userPath.GetPath());
  77.                     _resultField.Clear();
  78. }
  79. }
  80. break;
  81. case IDC_CALCULATE:
  82.             if (command == BN_CLICKED)
  83. {
  84.                 try
  85.                 {
  86.     BeginNewCalculation();
  88.     // do the real work
  89.                     char path[MAX_PATH + 1];
  90.                     _pathField.GetString(path, MAX_PATH);
  91. if (path[0] == '.')
  92. {
  93. _calculate.Disable();
  94. MessageBox(hwnd, "TreeSizer does not support relative paths in this version.", "Our problem ...", MB_OK|MB_ICONEXCLAMATION);
  95. }
  96. else
  97. {
  98.                     ULONG size = CalculateFolderSize(path);
  100.     // and present the result
  101.     ostrstream result;
  102.     CreateResultString(size, result);
  103.     _resultField.SetString(result.str());
  104. }
  105.                 }
  106.                 catch (WinException & e)
  107.                 {
  108.                     MessageBox (0, e.GetMessage (), "Exception", MB_ICONEXCLAMATION | MB_OK);
  109.                 }
  110.                 catch (...)
  111.                 {}
  112.                 _clear.Enable();
  113. _calculate.Enable();
  114.             }
  115. break;
  116. case IDC_EXIT:
  117.             if (command == BN_CLICKED)
  118.         SendMessage (hwnd, WM_CLOSE, 0, 0L);
  119. break;
  120. case IDC_NEW:
  121. _pathField.Clear();
  122. BeginNewCalculation();
  123. break;
  124.     }
  125. }
  127. class MouseWait
  128. {
  129. public:
  130. MouseWait()
  131. {
  132. _previous = SetCursor(LoadCursor(0, IDC_WAIT));
  133. }
  134. ~MouseWait()
  135. {
  136. if (_previous)
  137. SetCursor(_previous);
  138. }
  139. private:
  140. HCURSOR _previous;
  141. };
  143. ULONG Controller::CalculateFolderSize(char const * folderName)
  144. {
  145. MouseWait mw;
  146. Sizer sizer;
  147. Traversal traversal(folderName, sizer);
  148. return sizer.GetTotalSize();
  149. }
  151. void Controller::BeginNewCalculation()
  152. {
  153. _calculate.Disable();
  154. _resultField.Clear();
  155.     _clear.Disable();
  156. }
  158. void Controller::CreateResultString(ULONG size, ostrstream & ostr)
  159. {
  160. ostr << size << " bytes";
  162. if (size > 1024)
  163. {
  164.     char *suffixes[] = {"", " KB", " MB", " GB", " TB"};
  165.     int numSuffixes = 5;
  166.     int suffix;
  167.     double realSize = static_cast<double> (size);
  168.     for (suffix = 0; suffix < numSuffixes; suffix++)
  169.     {
  170.             if (realSize <= 1024)
  171.                 break;
  172.         realSize /= 1024;
  173.     }
  175.   ostr << "  ";
  176.         ostr.precision(3);
  177.         ostr << realSize << suffixes[suffix];
  178.   }
  179. ostr << '';
  180. }