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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > P2P编程 > 查看源码
CategoryFileDB.h
资源名称：GGBT.rar　[点击查看]
上传用户：lds876
上传日期：2013-05-25
资源大小：567k
文件大小：3k
源码类别：

P2P编程

开发平台：

Visual C++

CategoryFileDB.h：源码内容
  1. // CategoryFileDB.h: interface for the CCategoryFileDB class.
  2. //
  3. //////////////////////////////////////////////////////////////////////
  5. #if !defined(AFX_CATEGORYFILEDB_H__B8B8E016_0450_41EB_A7C2_424943981800__INCLUDED_)
  6. #define AFX_CATEGORYFILEDB_H__B8B8E016_0450_41EB_A7C2_424943981800__INCLUDED_
  8. #if _MSC_VER > 1000
  9. #pragma once
  10. #endif // _MSC_VER > 1000
  12. #include "ctrlext.h"
  15. class CCategoryBarWnd;
  16. class CFileDBItem;
  17. class CVal;
  18. class CSetupRegister;
  20. class CCategoryFileDB  : public CTreeCtrlEx
  21. {
  22. public:
  23. class CCategoryDBItem
  24. {
  25. public :
  26. CCategoryDBItem(string strDefaultDirectory, long lDirectoryID)
  27. {
  28. m_strDefaultDirectory = strDefaultDirectory;
  29. m_lDirectoryID = lDirectoryID;
  31. }
  33. long m_lDirectoryID;
  34. string  m_strDefaultDirectory;
  35. vector<CFileDBItem*> m_vFileDBItems;
  36. };
  38. public:
  39. CCategoryFileDB();
  40. virtual ~CCategoryFileDB();
  42. BOOL CreateEx(CWnd *pParent, CRect &rc, long lParam, CSetupRegister* pSetupRegister );
  44. CTreeCursor findCursorByDirid(long lDirectoryID);
  46. void NewDatabase();
  47. bool OpenDBFile(string strDatabasePath);
  48. void SaveAs(string strDatabasePath);
  49. void Save();
  50. bool IsModifyed();
  51. void SetModifed(bool bModified = true);
  52. string GetDatabaseName();
  53. string GetDatabaseFullPath();
  54. string GetDBSavePath() const;
  56. CSetupRegister* GetSetupRegister() const;
  58. // data
  59. CTreeCursor m_curRoot;
  60. CTreeCursor m_curDownloading;
  61. CTreeCursor m_curDownloaded;
  62. CTreeCursor m_curRecycle;
  64. protected:
  65. string _GetDBSavePath(const string strDatabaseFileName) const;
  66. void LoadDBFile();
  67. void SaveDatabase(string strDatabasePath);
  68. bool LoadDatabase(string strDatabasePath);
  69. virtual void DeleteAllData(CTreeCursor& curDel);
  70. // bool IsLinkFileDBItem(CTreeCursor curItem, CFileDBItem* pFileDBItem);
  72. // data
  73. CSetupRegister* m_pSetupRegister;
  74. bool m_bNoNotifications;
  75. long m_lDirectoryIDMax;
  76. enum ImageIDS{IID_ROOT = 0, IID_DOWNLOADED_DIRECTORY, IID_DOWNLOADING_DIRECTORY, IID_RECYCLE, IID_DIRECTORY};
  78. class CUnclosed
  79. {
  80. public:
  81. CUnclosed(CFileDBItem* pFileDBItem, HTREEITEM hTreeItem)
  82. {
  83. m_pFileDBItem = pFileDBItem;
  84. m_hTreeItem = hTreeItem;
  85. }
  86. CFileDBItem* m_pFileDBItem;
  87. HTREEITEM m_hTreeItem;
  88. };
  89. vector<CUnclosed> m_vUncloseds;
  91. private:
  92. CVal* bencode_rec(CTreeCursor& curItem);
  93. CVal* bencode_FileDBItems(vector<CFileDBItem*>& vFileDbItems);
  94. bool bdencode_FileDBItems(CVal* pVal, vector<CFileDBItem*>& vFileDbItems, HTREEITEM hTreeItem);
  95. bool bdencode_rec(CVal* pVal, CTreeCursor& curItem);
  96. CVal* bencode_UnNeededFiles(vector<CSize>& vUnNeededFiles);
  97. bool bdencode_UnNeedFiles(CVal* pVal, vector<CSize>& vFileDbItems);
  99. // bool LoadGenerateLinks();
  100. // CFileDBItem* GetLinkTrueFileDBItem(CFileDBItem* pLinkFileDBItem);
  101. static CTreeCursor findCursorByDiridRec(CTreeCursor curItem, long lDirectoryID);
  103. string m_strDatabaseFileName;
  104. static string GetDefaultDatabasePath();
  105. bool m_bModified;
  109. // string GetDBSavePath();
  110. // void SetDBSavePath(CSetupRegister* pSetupRegister);
  111. // string m_strDBSavePath;
  112. };
  114. #endif // !defined(AFX_CATEGORYFILEDB_H__B8B8E016_0450_41EB_A7C2_424943981800__INCLUDED_)