serialize_xml.cpp 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831
  1. //
  2. // Copyright (C) 2014 Christian Schüller <schuellchr@gmail.com>
  3. //
  4. // This Source Code Form is subject to the terms of the Mozilla Public License
  5. // v. 2.0. If a copy of the MPL was not distributed with this file, You can
  6. // obtain one at http://mozilla.org/MPL/2.0/.
  7. #include "serialize_xml.h"
  8. namespace igl
  9. {
  10. template <typename T>
  11. void serialize_xml(const T& obj,const std::string& filename)
  12. {
  13. serialize_xml(obj,"object",filename,0);
  14. }
  15. template <typename T>
  16. void serialize_xml(const T& obj,const std::string& objectName,const std::string& filename,bool binary,bool overwrite)
  17. {
  18. tinyxml2::XMLDocument* doc = new tinyxml2::XMLDocument();
  19. if(overwrite)
  20. {
  21. // Check if file exists
  22. tinyxml2::XMLError error = doc->LoadFile(filename.c_str());
  23. if(error != tinyxml2::XML_NO_ERROR)
  24. {
  25. doc->Clear();
  26. }
  27. }
  28. tinyxml2::XMLElement* element = doc->FirstChildElement("serialization");
  29. if(element == NULL)
  30. {
  31. element = doc->NewElement("serialization");
  32. doc->InsertEndChild(element);
  33. }
  34. serialize_xml(obj,objectName,doc,element,binary);
  35. // Save
  36. tinyxml2::XMLError error = doc->SaveFile(filename.c_str());
  37. if(error != tinyxml2::XML_NO_ERROR)
  38. {
  39. doc->PrintError();
  40. }
  41. delete doc;
  42. }
  43. template <typename T>
  44. void serialize_xml(const T& obj,const std::string& objectName,tinyxml2::XMLDocument* doc,tinyxml2::XMLElement* element,bool binary)
  45. {
  46. static_assert(detail_xml::is_serializable<T>::value,"'igl::serialize_xml': type is not serializable");
  47. std::string name(objectName);
  48. detail_xml::encodeXMLElementName(name);
  49. if(binary)
  50. {
  51. std::vector<char> buffer;
  52. serialize(obj,name,buffer);
  53. std::string data = detail_xml::base64_encode(reinterpret_cast<const unsigned char*>(buffer.data()),buffer.size());
  54. tinyxml2::XMLElement* child = detail_xml::getElement(doc,element,name);
  55. child->SetAttribute("binary",true);
  56. detail_xml::serialize(data,doc,element,name);
  57. }
  58. else
  59. {
  60. detail_xml::serialize(obj,doc,element,name);
  61. }
  62. }
  63. template <typename T>
  64. void deserialize_xml(T& obj,const std::string& filename)
  65. {
  66. deserialize_xml(obj,"object",filename);
  67. }
  68. template <typename T>
  69. void deserialize_xml(T& obj,const std::string& objectName,const std::string& filename)
  70. {
  71. tinyxml2::XMLDocument* doc = new tinyxml2::XMLDocument();
  72. tinyxml2::XMLError error = doc->LoadFile(filename.c_str());
  73. if(error != tinyxml2::XML_NO_ERROR)
  74. {
  75. std::cerr << "File not found!" << std::endl;
  76. doc->PrintError();
  77. doc = NULL;
  78. }
  79. else
  80. {
  81. tinyxml2::XMLElement* element = doc->FirstChildElement("serialization");
  82. if(element == NULL)
  83. {
  84. std::cerr << "Name of object not found! Initialized with default value." << std::endl;
  85. obj = T();
  86. }
  87. else
  88. {
  89. deserialize_xml(obj,objectName,doc,element);
  90. }
  91. delete doc;
  92. }
  93. }
  94. template <typename T>
  95. inline void deserialize_xml(T& obj,const std::string& objectName,const tinyxml2::XMLDocument* doc,const tinyxml2::XMLElement* element)
  96. {
  97. static_assert(detail::is_serializable<T>::value,"'igl::deserialize_xml': type is not deserializable");
  98. std::string name(objectName);
  99. detail_xml::encodeXMLElementName(name);
  100. const tinyxml2::XMLElement* child = element->FirstChildElement(name.c_str());
  101. if(child != NULL)
  102. {
  103. bool isBinary = false;
  104. const tinyxml2::XMLAttribute* attr = child->FindAttribute("binary");
  105. if(attr != NULL)
  106. {
  107. std::string code;
  108. detail_xml::deserialize(code,doc,element,name);
  109. std::string decoded = detail_xml::base64_decode(code);
  110. std::vector<char> buffer;
  111. std::copy(decoded.c_str(),decoded.c_str()+decoded.length(),std::back_inserter(buffer));
  112. deserialize(obj,name,buffer);
  113. }
  114. else
  115. {
  116. detail_xml::deserialize(obj,doc,element,name);
  117. }
  118. }
  119. }
  120. namespace detail_xml
  121. {
  122. // fundamental types
  123. template <typename T>
  124. std::enable_if_t<std::is_fundamental<T>::value> serialize(const T& obj,tinyxml2::XMLDocument* doc,tinyxml2::XMLElement* element,const std::string& name)
  125. {
  126. tinyxml2::XMLElement* child = getElement(doc,element,name.c_str());
  127. child->SetAttribute("val",obj);
  128. }
  129. template <typename T>
  130. std::enable_if_t<std::is_fundamental<T>::value> deserialize(T& obj,const tinyxml2::XMLDocument* doc,const tinyxml2::XMLElement* element,const std::string& name)
  131. {
  132. const tinyxml2::XMLElement* child = element->FirstChildElement(name.c_str());
  133. if(child != NULL)
  134. {
  135. getAttribute(child->Attribute("val"),obj);
  136. }
  137. else
  138. {
  139. obj = T();
  140. }
  141. }
  142. // std::string
  143. void serialize(const std::string& obj,tinyxml2::XMLDocument* doc,tinyxml2::XMLElement* element,const std::string& name)
  144. {
  145. tinyxml2::XMLElement* child = getElement(doc,element,name.c_str());
  146. child->SetAttribute("val",obj.c_str());
  147. }
  148. void deserialize(std::string& obj,const tinyxml2::XMLDocument* doc,const tinyxml2::XMLElement* element,const std::string& name)
  149. {
  150. const tinyxml2::XMLElement* child = element->FirstChildElement(name.c_str());
  151. if(child != NULL)
  152. {
  153. getAttribute(child->Attribute("val"),obj);
  154. }
  155. else
  156. {
  157. obj = std::string("");
  158. }
  159. }
  160. // Serializable
  161. template <typename T>
  162. std::enable_if_t<std::is_base_of<XMLSerializable,T>::value> serialize(const T& obj,tinyxml2::XMLDocument* doc,tinyxml2::XMLElement* element,const std::string& name)
  163. {
  164. // Serialize object implementing Serializable interface
  165. const XMLSerializable& object = dynamic_cast<const XMLSerializable&>(obj);
  166. tinyxml2::XMLElement* child = getElement(doc,element,name.c_str());
  167. object.Serialize(doc,child);
  168. }
  169. template <typename T>
  170. std::enable_if_t<std::is_base_of<XMLSerializable,T>::value> deserialize(T& obj,const tinyxml2::XMLDocument* doc,const tinyxml2::XMLElement* element,const std::string& name)
  171. {
  172. const tinyxml2::XMLElement* child = element->FirstChildElement(name.c_str());
  173. if(child != NULL)
  174. {
  175. obj.Deserialize(doc,child);
  176. }
  177. else
  178. {
  179. obj = T();
  180. }
  181. }
  182. // STL containers
  183. template <typename T1,typename T2>
  184. void serialize(const std::pair<T1,T2>& obj,tinyxml2::XMLDocument* doc,tinyxml2::XMLElement* element,const std::string& name)
  185. {
  186. tinyxml2::XMLElement* pair = getElement(doc,element,name.c_str());
  187. serialize(obj.first,doc,pair,"first");
  188. serialize(obj.second,doc,pair,"second");
  189. }
  190. template <typename T1,typename T2>
  191. void deserialize(std::pair<T1,T2>& obj,const tinyxml2::XMLDocument* doc,const tinyxml2::XMLElement* element,const std::string& name)
  192. {
  193. const tinyxml2::XMLElement* child = element->FirstChildElement(name.c_str());
  194. if(child != NULL)
  195. {
  196. deserialize(obj.first,doc,child,"first");
  197. deserialize(obj.second,doc,child,"second");
  198. }
  199. else
  200. {
  201. obj.first = T1();
  202. obj.second = T2();
  203. }
  204. }
  205. template <typename T1,typename T2>
  206. void serialize(const std::vector<T1,T2>& obj,tinyxml2::XMLDocument* doc,tinyxml2::XMLElement* element,const std::string& name)
  207. {
  208. tinyxml2::XMLElement* vector = getElement(doc,element,name.c_str());
  209. vector->SetAttribute("size",(unsigned int)obj.size());
  210. std::stringstream num;
  211. for(unsigned int i=0;i<obj.size();i++)
  212. {
  213. num.str("");
  214. num << "value" << i;
  215. serialize(obj[i],doc,vector,num.str());
  216. }
  217. }
  218. template <typename T1,typename T2>
  219. void deserialize(std::vector<T1,T2>& obj,const tinyxml2::XMLDocument* doc,const tinyxml2::XMLElement* element,const std::string& name)
  220. {
  221. obj.clear();
  222. const tinyxml2::XMLElement* child = element->FirstChildElement(name.c_str());
  223. if(child != NULL)
  224. {
  225. unsigned int size = child->UnsignedAttribute("size");
  226. obj.resize(size);
  227. std::stringstream num;
  228. for(unsigned int i=0;i<size;i++)
  229. {
  230. num.str("");
  231. num << "value" << i;
  232. deserialize(obj[i],doc,child,num.str());
  233. }
  234. }
  235. else
  236. {
  237. obj.clear();
  238. }
  239. }
  240. template <typename T>
  241. void serialize(const std::set<T>& obj,tinyxml2::XMLDocument* doc,tinyxml2::XMLElement* element,const std::string& name)
  242. {
  243. tinyxml2::XMLElement* set = getElement(doc,element,name.c_str());
  244. set->SetAttribute("size",(unsigned int)obj.size());
  245. std::stringstream num;
  246. typename std::set<T>::iterator iter = obj.begin();
  247. for(int i=0;iter!=obj.end();iter++,i++)
  248. {
  249. num.str("");
  250. num << "value" << i;
  251. serialize((T)*iter,doc,set,num.str());
  252. }
  253. }
  254. template <typename T>
  255. void deserialize(std::set<T>& obj,const tinyxml2::XMLDocument* doc,const tinyxml2::XMLElement* element,const std::string& name)
  256. {
  257. obj.clear();
  258. const tinyxml2::XMLElement* child = element->FirstChildElement(name.c_str());
  259. if(child != NULL)
  260. {
  261. unsigned int size = child->UnsignedAttribute("size");
  262. std::stringstream num;
  263. typename std::set<T>::iterator iter = obj.begin();
  264. for(int i=0;i<size;i++)
  265. {
  266. num.str("");
  267. num << "value" << i;
  268. T val;
  269. deserialize(val,doc,child,num.str());
  270. obj.insert(val);
  271. }
  272. }
  273. else
  274. {
  275. obj.clear();
  276. }
  277. }
  278. template <typename T1,typename T2>
  279. void serialize(const std::map<T1,T2>& obj,tinyxml2::XMLDocument* doc,tinyxml2::XMLElement* element,const std::string& name)
  280. {
  281. tinyxml2::XMLElement* map = getElement(doc,element,name.c_str());
  282. map->SetAttribute("size",(unsigned int)obj.size());
  283. std::stringstream num;
  284. typename std::map<T1,T2>::const_iterator iter = obj.cbegin();
  285. for(int i=0;iter!=obj.end();iter++,i++)
  286. {
  287. num.str("");
  288. num << "value" << i;
  289. serialize((std::pair<T1,T2>)*iter,doc,map,num.str());
  290. }
  291. }
  292. template <typename T1,typename T2>
  293. void deserialize(std::map<T1,T2>& obj,const tinyxml2::XMLDocument* doc,const tinyxml2::XMLElement* element,const std::string& name)
  294. {
  295. obj.clear();
  296. const tinyxml2::XMLElement* child = element->FirstChildElement(name.c_str());
  297. if(child != NULL)
  298. {
  299. unsigned int size = child->UnsignedAttribute("size");
  300. std::stringstream num;
  301. typename std::map<T1,T2>::iterator iter = obj.begin();
  302. for(int i=0;i<size;i++)
  303. {
  304. num.str("");
  305. num << "value" << i;
  306. std::pair<T1,T2> pair;
  307. deserialize(pair,doc,child,num.str());
  308. obj.insert(pair);
  309. }
  310. }
  311. else
  312. {
  313. obj.clear();
  314. }
  315. }
  316. // Eigen types
  317. template<typename T,int R,int C,int P,int MR,int MC>
  318. void serialize(const Eigen::Matrix<T,R,C,P,MR,MC>& obj,tinyxml2::XMLDocument* doc,tinyxml2::XMLElement* element,const std::string& name)
  319. {
  320. tinyxml2::XMLElement* matrix = getElement(doc,element,name.c_str());
  321. const unsigned int rows = obj.rows();
  322. const unsigned int cols = obj.cols();
  323. matrix->SetAttribute("rows",rows);
  324. matrix->SetAttribute("cols",cols);
  325. char buffer[200];
  326. std::stringstream ms;
  327. ms << "\n";
  328. for(unsigned int r=0;r<rows;r++)
  329. {
  330. for(unsigned int c=0;c<cols;c++)
  331. {
  332. tinyxml2::XMLUtil::ToStr(obj(r,c),buffer,200);
  333. ms << buffer << ",";
  334. }
  335. ms << "\n";
  336. }
  337. std::string mString = ms.str();
  338. if(mString.size() > 1)
  339. mString[mString.size()-2] = '\0';
  340. matrix->SetAttribute("matrix",mString.c_str());
  341. }
  342. template<typename T,int R,int C,int P,int MR,int MC>
  343. void deserialize(Eigen::Matrix<T,R,C,P,MR,MC>& obj,const tinyxml2::XMLDocument* doc,const tinyxml2::XMLElement* element,const std::string& name)
  344. {
  345. const tinyxml2::XMLElement* child = element->FirstChildElement(name.c_str());
  346. bool initialized = false;
  347. if(child != NULL)
  348. {
  349. const unsigned int rows = child->UnsignedAttribute("rows");
  350. const unsigned int cols = child->UnsignedAttribute("cols");
  351. if(rows > 0 && cols > 0)
  352. {
  353. obj.resize(rows,cols);
  354. const tinyxml2::XMLAttribute* attribute = child->FindAttribute("matrix");
  355. if(attribute != NULL)
  356. {
  357. std::string matTemp;
  358. getAttribute(attribute->Value(),matTemp);
  359. std::string line,srows,scols;
  360. std::stringstream mats;
  361. mats << matTemp;
  362. int r=0;
  363. std::string val;
  364. // for each line
  365. getline(mats,line);
  366. while(getline(mats,line))
  367. {
  368. // get current line
  369. std::stringstream liness(line);
  370. for(unsigned int c=0;c<cols-1;c++)
  371. {
  372. // split line
  373. getline(liness,val,',');
  374. // push pack the data if any
  375. if(!val.empty())
  376. getAttribute(val.c_str(),obj.coeffRef(r,c));
  377. }
  378. getline(liness,val);
  379. getAttribute(val.c_str(),obj.coeffRef(r,cols-1));
  380. r++;
  381. }
  382. initialized = true;
  383. }
  384. }
  385. }
  386. if(!initialized)
  387. {
  388. obj = Eigen::Matrix<T,R,C,P,MR,MC>();
  389. }
  390. }
  391. template<typename T,int P,typename I>
  392. void serialize(const Eigen::SparseMatrix<T,P,I>& obj,tinyxml2::XMLDocument* doc,tinyxml2::XMLElement* element,const std::string& name)
  393. {
  394. tinyxml2::XMLElement* matrix = getElement(doc,element,name.c_str());
  395. const unsigned int rows = obj.rows();
  396. const unsigned int cols = obj.cols();
  397. matrix->SetAttribute("rows",rows);
  398. matrix->SetAttribute("cols",cols);
  399. char buffer[200];
  400. std::stringstream ms;
  401. ms << "\n";
  402. for(int k=0;k<obj.outerSize();++k)
  403. {
  404. for(typename Eigen::SparseMatrix<T,P,I>::InnerIterator it(obj,k);it;++it)
  405. {
  406. tinyxml2::XMLUtil::ToStr(it.value(),buffer,200);
  407. ms << it.row() << "," << it.col() << "," << buffer << "\n";
  408. }
  409. }
  410. std::string mString = ms.str();
  411. if(mString.size() > 0)
  412. mString[mString.size()-1] = '\0';
  413. matrix->SetAttribute("matrix",mString.c_str());
  414. }
  415. template<typename T,int P,typename I>
  416. void deserialize(Eigen::SparseMatrix<T,P,I>& obj,const tinyxml2::XMLDocument* doc,const tinyxml2::XMLElement* element,const std::string& name)
  417. {
  418. const tinyxml2::XMLElement* child = element->FirstChildElement(name.c_str());
  419. bool initialized = false;
  420. if(child != NULL)
  421. {
  422. const unsigned int rows = child->UnsignedAttribute("rows");
  423. const unsigned int cols = child->UnsignedAttribute("cols");
  424. if(rows > 0 && cols > 0)
  425. {
  426. obj.resize(rows,cols);
  427. obj.setZero();
  428. const tinyxml2::XMLAttribute* attribute = child->FindAttribute("matrix");
  429. if(attribute != NULL)
  430. {
  431. std::string matTemp;
  432. getAttribute(attribute->Value(),matTemp);
  433. std::string line,srows,scols;
  434. std::stringstream mats;
  435. mats << matTemp;
  436. std::vector<Eigen::Triplet<T,I> > triplets;
  437. int r=0;
  438. std::string val;
  439. // for each line
  440. getline(mats,line);
  441. while(getline(mats,line))
  442. {
  443. // get current line
  444. std::stringstream liness(line);
  445. // row
  446. getline(liness,val,',');
  447. int row = atoi(val.c_str());
  448. // col
  449. getline(liness,val,',');
  450. int col = atoi(val.c_str());
  451. // val
  452. getline(liness,val);
  453. T value;
  454. getAttribute(val.c_str(),value);
  455. triplets.push_back(Eigen::Triplet<T,I>(row,col,value));
  456. r++;
  457. }
  458. obj.setFromTriplets(triplets.begin(),triplets.end());
  459. initialized = true;
  460. }
  461. }
  462. }
  463. if(!initialized)
  464. {
  465. obj = Eigen::SparseMatrix<T,P,I>();
  466. }
  467. }
  468. // pointers
  469. template <typename T>
  470. std::enable_if_t<std::is_pointer<T>::value> serialize(const T& obj,tinyxml2::XMLDocument* doc,tinyxml2::XMLElement* element,const std::string& name)
  471. {
  472. tinyxml2::XMLElement* pointer = getElement(doc,element,name.c_str());
  473. bool isNullPtr = (obj == NULL);
  474. pointer->SetAttribute("isNullPtr",isNullPtr);
  475. if(isNullPtr == false)
  476. detail_xml::serialize(*obj,doc,element,name);
  477. }
  478. template <typename T>
  479. std::enable_if_t<std::is_pointer<T>::value> deserialize(T& obj,const tinyxml2::XMLDocument* doc,const tinyxml2::XMLElement* element,const std::string& name)
  480. {
  481. const tinyxml2::XMLElement* child = element->FirstChildElement(name.c_str());
  482. if(child != NULL)
  483. {
  484. bool isNullPtr = child->BoolAttribute("isNullPtr");
  485. if(isNullPtr)
  486. {
  487. if(obj != NULL)
  488. {
  489. std::cout << "deserialization: possible memory leak for '" << typeid(obj).name() << "'" << std::endl;
  490. obj = NULL;
  491. }
  492. }
  493. else
  494. {
  495. if(obj != NULL)
  496. std::cout << "deserialization: possible memory leak for '" << typeid(obj).name() << "'" << std::endl;
  497. obj = new std::remove_pointer<T>::type();
  498. detail_xml::deserialize(*obj,doc,element,name);
  499. }
  500. }
  501. }
  502. // helper functions
  503. tinyxml2::XMLElement* getElement(tinyxml2::XMLDocument* doc,tinyxml2::XMLElement* element,const std::string& name)
  504. {
  505. tinyxml2::XMLElement* child = element->FirstChildElement(name.c_str());
  506. if(child == NULL)
  507. {
  508. child = doc->NewElement(name.c_str());
  509. element->InsertEndChild(child);
  510. }
  511. return child;
  512. }
  513. void getAttribute(const char* src,bool& dest)
  514. {
  515. tinyxml2::XMLUtil::ToBool(src,&dest);
  516. }
  517. void getAttribute(const char* src,char& dest)
  518. {
  519. dest = (char)atoi(src);
  520. }
  521. void getAttribute(const char* src,std::string& dest)
  522. {
  523. dest = src;
  524. }
  525. void getAttribute(const char* src,float& dest)
  526. {
  527. tinyxml2::XMLUtil::ToFloat(src,&dest);
  528. }
  529. void getAttribute(const char* src,double& dest)
  530. {
  531. tinyxml2::XMLUtil::ToDouble(src,&dest);
  532. }
  533. template<typename T>
  534. std::enable_if_t<std::is_integral<T>::value && std::is_unsigned<T>::value> getAttribute(const char* src,T& dest)
  535. {
  536. unsigned int val;
  537. tinyxml2::XMLUtil::ToUnsigned(src,&val);
  538. dest = (T)val;
  539. }
  540. template<typename T>
  541. std::enable_if_t<std::is_integral<T>::value && !std::is_unsigned<T>::value> getAttribute(const char* src,T& dest)
  542. {
  543. int val;
  544. tinyxml2::XMLUtil::ToInt(src,&val);
  545. dest = (T)val;
  546. }
  547. // tinyXML2 related stuff
  548. static const int numForbiddenChars = 8;
  549. static const char forbiddenChars[] ={' ','/','~','#','&','>','<','='};
  550. void replaceSubString(std::string& str,const std::string& search,const std::string& replace)
  551. {
  552. size_t pos = 0;
  553. while((pos = str.find(search,pos)) != std::string::npos)
  554. {
  555. str.replace(pos,search.length(),replace);
  556. pos += replace.length();
  557. }
  558. }
  559. void encodeXMLElementName(std::string& name)
  560. {
  561. // must not start with a digit
  562. if(isdigit(*name.begin()))
  563. {
  564. name = ":::" + name;
  565. }
  566. std::stringstream stream;
  567. for(int i=0;i<numForbiddenChars;i++)
  568. {
  569. std::string search;
  570. search = forbiddenChars[i];
  571. std::stringstream replaces;
  572. replaces << ":" << (int)forbiddenChars[i];
  573. std::string replace = replaces.str();
  574. replaceSubString(name,search,replace);
  575. }
  576. }
  577. void decodeXMLElementName(std::string& name)
  578. {
  579. if(name.find("::",0) == 0)
  580. name.replace(0,3,"");
  581. for(auto chr : forbiddenChars)
  582. {
  583. std::stringstream searchs;
  584. searchs << ":" << (int)chr;
  585. std::string search = searchs.str();
  586. std::string replace;
  587. replace = chr;
  588. replaceSubString(name,search,replace);
  589. }
  590. }
  591. /* Copyright(C) 2004-2008 René Nyffenegger
  592. This source code is provided 'as-is',without any express or implied
  593. warranty.In no event will the author be held liable for any damages
  594. arising from the use of this software.
  595. Permission is granted to anyone to use this software for any purpose,
  596. including commercial applications,and to alter it and redistribute it
  597. freely,subject to the following restrictions:
  598. 1. The origin of this source code must not be misrepresented; you must not
  599. claim that you wrote the original source code.If you use this source code
  600. in a product,an acknowledgment in the product documentation would be
  601. appreciated but is not required.
  602. 2. Altered source versions must be plainly marked as such,and must not be
  603. misrepresented as being the original source code.
  604. 3. This notice may not be removed or altered from any source distribution.
  605. René Nyffenegger rene.nyffenegger@adp-gmbh.ch
  606. */
  607. static const std::string base64_chars =
  608. "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
  609. "abcdefghijklmnopqrstuvwxyz"
  610. "0123456789+/";
  611. static inline bool is_base64(unsigned char c) {
  612. return (isalnum(c) || (c == '+') || (c == '/'));
  613. }
  614. std::string base64_encode(unsigned char const* bytes_to_encode,unsigned int in_len)
  615. {
  616. std::string ret;
  617. int i = 0;
  618. int j = 0;
  619. unsigned char char_array_3[3];
  620. unsigned char char_array_4[4];
  621. while(in_len--) {
  622. char_array_3[i++] = *(bytes_to_encode++);
  623. if(i == 3) {
  624. char_array_4[0] = (char_array_3[0] & 0xfc) >> 2;
  625. char_array_4[1] = ((char_array_3[0] & 0x03) << 4) + ((char_array_3[1] & 0xf0) >> 4);
  626. char_array_4[2] = ((char_array_3[1] & 0x0f) << 2) + ((char_array_3[2] & 0xc0) >> 6);
  627. char_array_4[3] = char_array_3[2] & 0x3f;
  628. for(i = 0; (i <4) ; i++)
  629. ret += base64_chars[char_array_4[i]];
  630. i = 0;
  631. }
  632. }
  633. if(i)
  634. {
  635. for(j = i; j < 3; j++)
  636. char_array_3[j] = '\0';
  637. char_array_4[0] = (char_array_3[0] & 0xfc) >> 2;
  638. char_array_4[1] = ((char_array_3[0] & 0x03) << 4) + ((char_array_3[1] & 0xf0) >> 4);
  639. char_array_4[2] = ((char_array_3[1] & 0x0f) << 2) + ((char_array_3[2] & 0xc0) >> 6);
  640. char_array_4[3] = char_array_3[2] & 0x3f;
  641. for(j = 0; (j < i + 1); j++)
  642. ret += base64_chars[char_array_4[j]];
  643. while((i++ < 3))
  644. ret += '=';
  645. }
  646. return ret;
  647. }
  648. std::string base64_decode(std::string const& encoded_string)
  649. {
  650. int in_len = encoded_string.size();
  651. int i = 0;
  652. int j = 0;
  653. int in_ = 0;
  654. unsigned char char_array_4[4],char_array_3[3];
  655. std::string ret;
  656. // construct fast lookup table
  657. // added by Christian Schüller (schuellc@inf.ethz.ch)
  658. int charLookup[200];
  659. for(int i=0;i<base64_chars.length();i++)
  660. charLookup[(int)base64_chars[i]] = i;
  661. while(in_len-- && (encoded_string[in_] != '=') && is_base64(encoded_string[in_])) {
  662. char_array_4[i++] = encoded_string[in_]; in_++;
  663. if(i ==4) {
  664. for(i = 0; i <4; i++)
  665. char_array_4[i] = charLookup[char_array_4[i]]; // new fast lookup
  666. //char_array_4[i] = base64_chars.find(char_array_4[i]); // original version
  667. char_array_3[0] = (char_array_4[0] << 2) + ((char_array_4[1] & 0x30) >> 4);
  668. char_array_3[1] = ((char_array_4[1] & 0xf) << 4) + ((char_array_4[2] & 0x3c) >> 2);
  669. char_array_3[2] = ((char_array_4[2] & 0x3) << 6) + char_array_4[3];
  670. for(i = 0; (i < 3); i++)
  671. ret += char_array_3[i];
  672. i = 0;
  673. }
  674. }
  675. if(i) {
  676. for(j = i; j <4; j++)
  677. char_array_4[j] = 0;
  678. for(j = 0; j <4; j++)
  679. char_array_4[j] = base64_chars.find(char_array_4[j]);
  680. char_array_3[0] = (char_array_4[0] << 2) + ((char_array_4[1] & 0x30) >> 4);
  681. char_array_3[1] = ((char_array_4[1] & 0xf) << 4) + ((char_array_4[2] & 0x3c) >> 2);
  682. char_array_3[2] = ((char_array_4[2] & 0x3) << 6) + char_array_4[3];
  683. for(j = 0; (j < i - 1);
  684. j++) ret += char_array_3[j];
  685. }
  686. return ret;
  687. }
  688. }
  689. }