M_RecordCFG.cpp 2.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102
  1. #include "stdafx.h"
  2. #include <cassert>
  3. #include <fstream>
  4. #include <iostream>
  5. #include <iostream>
  6. #include <boost/smart_ptr.hpp>
  7. #include <boost/algorithm/string.hpp>
  8. #include "tinyxml2.h"
  9. #include "M_RecordCFG.h"
  10. #include "FileEncrypt.h"
  11. std::auto_ptr<M_RecordCFG> M_RecordCFG::msSingleton(nullptr);
  12. int M_RecordCFG::GetCount()
  13. {
  14. return (int)mMapData.size();
  15. }
  16. const M_RecordCFGData* M_RecordCFG::GetData(int RecordID)
  17. {
  18. auto it = mMapData.find(RecordID);
  19. if (it != mMapData.end())
  20. {
  21. return &it->second;
  22. }
  23. return NULL;
  24. }
  25. boost::unordered_map<int, M_RecordCFGData>& M_RecordCFG::GetMapData()
  26. {
  27. return mMapData;
  28. }
  29. void M_RecordCFG::Reload()
  30. {
  31. mMapData.clear();
  32. Load();
  33. }
  34. void M_RecordCFG::Load(const std::string& path)
  35. {
  36. std::ifstream readStream(path, std::ios::binary);
  37. if (!readStream.is_open())
  38. {
  39. assert(false);
  40. return;
  41. }
  42. readStream.seekg(0, std::ios::end);
  43. int fileSize = readStream.tellg();
  44. boost::shared_array<char> buffer(new char[fileSize+1]);
  45. buffer.get()[fileSize] = '\0';
  46. readStream.seekg(0, std::ios::beg);
  47. readStream.read(buffer.get(), fileSize);
  48. readStream.close();
  49. FileEncrypt::decryptBuffer( buffer, fileSize );
  50. tinyxml2::XMLDocument xmlDoc;
  51. auto result = xmlDoc.Parse(buffer.get(), fileSize);
  52. if (result != tinyxml2::XML_SUCCESS)
  53. {
  54. assert(false);
  55. return;
  56. }
  57. auto root = xmlDoc.RootElement();
  58. if (root == NULL)
  59. {
  60. assert(false);
  61. return;
  62. }
  63. auto element = root->FirstChildElement("Data");
  64. while (element != NULL)
  65. {
  66. M_RecordCFGData data;
  67. data.mRecordID = element->IntAttribute("RecordID");
  68. data.mRecordName = element->Attribute("RecordName");
  69. {
  70. const char* RecordInfoList = element->Attribute("RecordInfoList");
  71. std::vector<std::string> vecRecordInfoList;
  72. boost::split(vecRecordInfoList, RecordInfoList, boost::is_any_of(","));
  73. for (unsigned int i = 0; i < vecRecordInfoList.size(); i++)
  74. {
  75. data.mRecordInfoList.push_back(vecRecordInfoList[i]);
  76. }
  77. }
  78. if (mMapData.find(data.mRecordID) != mMapData.end())std::cout <<"data refind:" << data.mRecordID << std::endl;
  79. assert(mMapData.find(data.mRecordID) == mMapData.end());
  80. mMapData.insert(std::make_pair(data.mRecordID, data));
  81. element = element->NextSiblingElement();
  82. }
  83. }
  84. void M_RecordCFG::Load()
  85. {
  86. Load("../Config/M_RecordCFG.xml");
  87. }
  88. M_RecordCFG* M_RecordCFG::GetSingleton()
  89. {
  90. if (msSingleton.get() == nullptr)
  91. {
  92. msSingleton.reset(new M_RecordCFG());
  93. }
  94. return msSingleton.get();
  95. }