Ryan OShea | de36e4a | 2020-03-13 16:26:19 +0000 | [diff] [blame] | 1 | <!-- Copyright (c) 2020 ARM Limited. --> |
| 2 | <!-- --> |
| 3 | <!-- SPDX-License-Identifier: MIT --> |
| 4 | <!-- --> |
| 5 | <!-- HTML header for doxygen 1.8.13--> |
| 6 | <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"> |
| 7 | <html xmlns="http://www.w3.org/1999/xhtml"> |
| 8 | <head> |
| 9 | <meta http-equiv="Content-Type" content="text/xhtml;charset=UTF-8"/> |
| 10 | <meta http-equiv="X-UA-Compatible" content="IE=9"/> |
| 11 | <meta name="generator" content="Doxygen 1.8.13"/> |
| 12 | <meta name="robots" content="NOINDEX, NOFOLLOW" /> |
| 13 | <meta name="viewport" content="width=device-width, initial-scale=1"/> |
| 14 | <title>ArmNN: tests/ImageCSVFileGenerator/ImageCSVFileGenerator.cpp File Reference</title> |
| 15 | <link href="tabs.css" rel="stylesheet" type="text/css"/> |
| 16 | <script type="text/javascript" src="jquery.js"></script> |
| 17 | <script type="text/javascript" src="dynsections.js"></script> |
| 18 | <link href="navtree.css" rel="stylesheet" type="text/css"/> |
| 19 | <script type="text/javascript" src="resize.js"></script> |
| 20 | <script type="text/javascript" src="navtreedata.js"></script> |
| 21 | <script type="text/javascript" src="navtree.js"></script> |
| 22 | <script type="text/javascript"> |
| 23 | $(document).ready(initResizable); |
| 24 | </script> |
| 25 | <link href="search/search.css" rel="stylesheet" type="text/css"/> |
| 26 | <script type="text/javascript" src="search/searchdata.js"></script> |
| 27 | <script type="text/javascript" src="search/search.js"></script> |
| 28 | <script type="text/x-mathjax-config"> |
| 29 | MathJax.Hub.Config({ |
| 30 | extensions: ["tex2jax.js"], |
| 31 | jax: ["input/TeX","output/HTML-CSS"], |
| 32 | }); |
| 33 | </script><script type="text/javascript" src="http://cdn.mathjax.org/mathjax/latest/MathJax.js"></script> |
| 34 | <link href="doxygen.css" rel="stylesheet" type="text/css" /> |
| 35 | <link href="stylesheet.css" rel="stylesheet" type="text/css"/> |
| 36 | </head> |
| 37 | <body> |
| 38 | <div id="top"><!-- do not remove this div, it is closed by doxygen! --> |
| 39 | <div id="titlearea"> |
| 40 | <table cellspacing="0" cellpadding="0"> |
| 41 | <tbody> |
| 42 | <tr style="height: 56px;"> |
| 43 | <img alt="ArmNN" src="Arm_NN_horizontal_blue.png" style="max-width: 10rem; margin-top: .5rem; margin-left 10px"/> |
| 44 | <td style="padding-left: 0.5em;"> |
| 45 | <div id="projectname"> |
| 46 |  <span id="projectnumber">20.02</span> |
| 47 | </div> |
| 48 | </td> |
| 49 | </tr> |
| 50 | </tbody> |
| 51 | </table> |
| 52 | </div> |
| 53 | <!-- end header part --> |
| 54 | <!-- Generated by Doxygen 1.8.13 --> |
| 55 | <script type="text/javascript"> |
| 56 | var searchBox = new SearchBox("searchBox", "search",false,'Search'); |
| 57 | </script> |
| 58 | <script type="text/javascript" src="menudata.js"></script> |
| 59 | <script type="text/javascript" src="menu.js"></script> |
| 60 | <script type="text/javascript"> |
| 61 | $(function() { |
| 62 | initMenu('',true,false,'search.php','Search'); |
| 63 | $(document).ready(function() { init_search(); }); |
| 64 | }); |
| 65 | </script> |
| 66 | <div id="main-nav"></div> |
| 67 | </div><!-- top --> |
| 68 | <div id="side-nav" class="ui-resizable side-nav-resizable"> |
| 69 | <div id="nav-tree"> |
| 70 | <div id="nav-tree-contents"> |
| 71 | <div id="nav-sync" class="sync"></div> |
| 72 | </div> |
| 73 | </div> |
| 74 | <div id="splitbar" style="-moz-user-select:none;" |
| 75 | class="ui-resizable-handle"> |
| 76 | </div> |
| 77 | </div> |
| 78 | <script type="text/javascript"> |
| 79 | $(document).ready(function(){initNavTree('_image_c_s_v_file_generator_8cpp.xhtml','');}); |
| 80 | </script> |
| 81 | <div id="doc-content"> |
| 82 | <!-- window showing the filter options --> |
| 83 | <div id="MSearchSelectWindow" |
| 84 | onmouseover="return searchBox.OnSearchSelectShow()" |
| 85 | onmouseout="return searchBox.OnSearchSelectHide()" |
| 86 | onkeydown="return searchBox.OnSearchSelectKey(event)"> |
| 87 | </div> |
| 88 | |
| 89 | <!-- iframe showing the search results (closed by default) --> |
| 90 | <div id="MSearchResultsWindow"> |
| 91 | <iframe src="javascript:void(0)" frameborder="0" |
| 92 | name="MSearchResults" id="MSearchResults"> |
| 93 | </iframe> |
| 94 | </div> |
| 95 | |
| 96 | <div class="header"> |
| 97 | <div class="summary"> |
| 98 | <a href="#func-members">Functions</a> </div> |
| 99 | <div class="headertitle"> |
| 100 | <div class="title">ImageCSVFileGenerator.cpp File Reference</div> </div> |
| 101 | </div><!--header--> |
| 102 | <div class="contents"> |
| 103 | <div class="textblock"><code>#include <boost/filesystem.hpp></code><br /> |
| 104 | <code>#include <boost/filesystem/operations.hpp></code><br /> |
| 105 | <code>#include <boost/filesystem/path.hpp></code><br /> |
| 106 | <code>#include <boost/program_options.hpp></code><br /> |
| 107 | <code>#include <algorithm></code><br /> |
| 108 | <code>#include <fstream></code><br /> |
| 109 | <code>#include <iostream></code><br /> |
| 110 | <code>#include <string></code><br /> |
| 111 | </div> |
| 112 | <p><a href="_image_c_s_v_file_generator_8cpp_source.xhtml">Go to the source code of this file.</a></p> |
| 113 | <table class="memberdecls"> |
| 114 | <tr class="heading"><td colspan="2"><h2 class="groupheader"><a name="func-members"></a> |
| 115 | Functions</h2></td></tr> |
| 116 | <tr class="memitem:a0ddf1224851353fc92bfbff6f499fa97"><td class="memItemLeft" align="right" valign="top">int </td><td class="memItemRight" valign="bottom"><a class="el" href="_image_c_s_v_file_generator_8cpp.xhtml#a0ddf1224851353fc92bfbff6f499fa97">main</a> (int argc, char *argv[])</td></tr> |
| 117 | <tr class="separator:a0ddf1224851353fc92bfbff6f499fa97"><td class="memSeparator" colspan="2"> </td></tr> |
| 118 | </table> |
| 119 | <h2 class="groupheader">Function Documentation</h2> |
| 120 | <a id="a0ddf1224851353fc92bfbff6f499fa97"></a> |
| 121 | <h2 class="memtitle"><span class="permalink"><a href="#a0ddf1224851353fc92bfbff6f499fa97">◆ </a></span>main()</h2> |
| 122 | |
| 123 | <div class="memitem"> |
| 124 | <div class="memproto"> |
| 125 | <table class="memname"> |
| 126 | <tr> |
| 127 | <td class="memname">int main </td> |
| 128 | <td>(</td> |
| 129 | <td class="paramtype">int </td> |
| 130 | <td class="paramname"><em>argc</em>, </td> |
| 131 | </tr> |
| 132 | <tr> |
| 133 | <td class="paramkey"></td> |
| 134 | <td></td> |
| 135 | <td class="paramtype">char * </td> |
| 136 | <td class="paramname"><em>argv</em>[] </td> |
| 137 | </tr> |
| 138 | <tr> |
| 139 | <td></td> |
| 140 | <td>)</td> |
| 141 | <td></td><td></td> |
| 142 | </tr> |
| 143 | </table> |
| 144 | </div><div class="memdoc"> |
| 145 | |
| 146 | <p class="definition">Definition at line <a class="el" href="_image_c_s_v_file_generator_8cpp_source.xhtml#l00167">167</a> of file <a class="el" href="_image_c_s_v_file_generator_8cpp_source.xhtml">ImageCSVFileGenerator.cpp</a>.</p> |
| 147 | <div class="fragment"><div class="line"><a name="l00168"></a><span class="lineno"> 168</span> {</div><div class="line"><a name="l00169"></a><span class="lineno"> 169</span>  CommandLineProcessor cmdline;</div><div class="line"><a name="l00170"></a><span class="lineno"> 170</span>  <span class="keywordflow">if</span> (!cmdline.ProcessCommandLine(argc, argv))</div><div class="line"><a name="l00171"></a><span class="lineno"> 171</span>  {</div><div class="line"><a name="l00172"></a><span class="lineno"> 172</span>  <span class="keywordflow">return</span> -1;</div><div class="line"><a name="l00173"></a><span class="lineno"> 173</span>  }</div><div class="line"><a name="l00174"></a><span class="lineno"> 174</span> </div><div class="line"><a name="l00175"></a><span class="lineno"> 175</span>  <span class="keyword">namespace </span>fs = boost::filesystem;</div><div class="line"><a name="l00176"></a><span class="lineno"> 176</span> </div><div class="line"><a name="l00177"></a><span class="lineno"> 177</span>  <span class="keyword">const</span> std::string fileFormat(<span class="stringliteral">".raw"</span>);</div><div class="line"><a name="l00178"></a><span class="lineno"> 178</span> </div><div class="line"><a name="l00179"></a><span class="lineno"> 179</span>  <span class="keyword">const</span> std::string rawDirectory(cmdline.GetInputDirectory());</div><div class="line"><a name="l00180"></a><span class="lineno"> 180</span>  <span class="keyword">const</span> std::string outputPath(cmdline.GetOutputFileName());</div><div class="line"><a name="l00181"></a><span class="lineno"> 181</span>  <span class="keyword">const</span> std::string bindingId(cmdline.GetInputBindingId());</div><div class="line"><a name="l00182"></a><span class="lineno"> 182</span> </div><div class="line"><a name="l00183"></a><span class="lineno"> 183</span>  std::vector<fs::path> rawFiles;</div><div class="line"><a name="l00184"></a><span class="lineno"> 184</span>  <span class="keywordflow">for</span> (<span class="keyword">auto</span>& entry : boost::make_iterator_range(fs::directory_iterator(rawDirectory), {}))</div><div class="line"><a name="l00185"></a><span class="lineno"> 185</span>  {</div><div class="line"><a name="l00186"></a><span class="lineno"> 186</span>  <span class="keywordflow">if</span> (entry.path().extension().c_str() == fileFormat)</div><div class="line"><a name="l00187"></a><span class="lineno"> 187</span>  {</div><div class="line"><a name="l00188"></a><span class="lineno"> 188</span>  rawFiles.push_back(entry.path());</div><div class="line"><a name="l00189"></a><span class="lineno"> 189</span>  }</div><div class="line"><a name="l00190"></a><span class="lineno"> 190</span>  }</div><div class="line"><a name="l00191"></a><span class="lineno"> 191</span> </div><div class="line"><a name="l00192"></a><span class="lineno"> 192</span>  <span class="keywordflow">if</span> (!rawFiles.empty())</div><div class="line"><a name="l00193"></a><span class="lineno"> 193</span>  {</div><div class="line"><a name="l00194"></a><span class="lineno"> 194</span>  <span class="keywordtype">unsigned</span> <span class="keywordtype">int</span> pass = 0;</div><div class="line"><a name="l00195"></a><span class="lineno"> 195</span>  std::ofstream refinementData;</div><div class="line"><a name="l00196"></a><span class="lineno"> 196</span>  refinementData.open(outputPath, std::ofstream::out);</div><div class="line"><a name="l00197"></a><span class="lineno"> 197</span>  <span class="keywordflow">if</span> (refinementData.is_open())</div><div class="line"><a name="l00198"></a><span class="lineno"> 198</span>  {</div><div class="line"><a name="l00199"></a><span class="lineno"> 199</span>  <span class="keywordflow">for</span> (<span class="keyword">auto</span> <span class="keyword">const</span>& raw : rawFiles)</div><div class="line"><a name="l00200"></a><span class="lineno"> 200</span>  {</div><div class="line"><a name="l00201"></a><span class="lineno"> 201</span>  refinementData << pass << <span class="stringliteral">", "</span> << bindingId << <span class="stringliteral">", "</span> << raw.c_str() << <span class="stringliteral">"\n"</span>;</div><div class="line"><a name="l00202"></a><span class="lineno"> 202</span>  <span class="keywordflow">if</span> (!refinementData)</div><div class="line"><a name="l00203"></a><span class="lineno"> 203</span>  {</div><div class="line"><a name="l00204"></a><span class="lineno"> 204</span>  std::cerr << <span class="stringliteral">"Failed to write to output file: "</span> << outputPath << std::endl;</div><div class="line"><a name="l00205"></a><span class="lineno"> 205</span>  <span class="keywordflow">continue</span>;</div><div class="line"><a name="l00206"></a><span class="lineno"> 206</span>  }</div><div class="line"><a name="l00207"></a><span class="lineno"> 207</span>  ++pass;</div><div class="line"><a name="l00208"></a><span class="lineno"> 208</span>  }</div><div class="line"><a name="l00209"></a><span class="lineno"> 209</span>  refinementData.close();</div><div class="line"><a name="l00210"></a><span class="lineno"> 210</span>  }</div><div class="line"><a name="l00211"></a><span class="lineno"> 211</span>  <span class="keywordflow">else</span></div><div class="line"><a name="l00212"></a><span class="lineno"> 212</span>  {</div><div class="line"><a name="l00213"></a><span class="lineno"> 213</span>  std::cerr << <span class="stringliteral">"Failed to open output file: "</span> << outputPath << std::endl;</div><div class="line"><a name="l00214"></a><span class="lineno"> 214</span>  <span class="keywordflow">return</span> -1;</div><div class="line"><a name="l00215"></a><span class="lineno"> 215</span>  }</div><div class="line"><a name="l00216"></a><span class="lineno"> 216</span>  }</div><div class="line"><a name="l00217"></a><span class="lineno"> 217</span>  <span class="keywordflow">else</span></div><div class="line"><a name="l00218"></a><span class="lineno"> 218</span>  {</div><div class="line"><a name="l00219"></a><span class="lineno"> 219</span>  std::cerr << <span class="stringliteral">"No matching files with the \".raw\" extension found in the directory: "</span></div><div class="line"><a name="l00220"></a><span class="lineno"> 220</span>  << rawDirectory << std::endl;</div><div class="line"><a name="l00221"></a><span class="lineno"> 221</span>  <span class="keywordflow">return</span> -1;</div><div class="line"><a name="l00222"></a><span class="lineno"> 222</span>  }</div><div class="line"><a name="l00223"></a><span class="lineno"> 223</span> </div><div class="line"><a name="l00224"></a><span class="lineno"> 224</span>  <span class="keywordflow">return</span> 0;</div><div class="line"><a name="l00225"></a><span class="lineno"> 225</span> }</div></div><!-- fragment --> |
| 148 | </div> |
| 149 | </div> |
| 150 | </div><!-- contents --> |
| 151 | </div><!-- doc-content --> |
| 152 | <!-- start footer part --> |
| 153 | <div id="nav-path" class="navpath"><!-- id is needed for treeview function! --> |
| 154 | <ul> |
| 155 | <li class="navelem"><a class="el" href="dir_59425e443f801f1f2fd8bbe4959a3ccf.xhtml">tests</a></li><li class="navelem"><a class="el" href="dir_ed4050ecbb48c03aa3407c24a3e522b1.xhtml">ImageCSVFileGenerator</a></li><li class="navelem"><a class="el" href="_image_c_s_v_file_generator_8cpp.xhtml">ImageCSVFileGenerator.cpp</a></li> |
| 156 | <li class="footer">Generated on Fri Mar 13 2020 16:09:16 for ArmNN by |
| 157 | <a href="http://www.doxygen.org/index.html"> |
| 158 | <img class="footer" src="doxygen.png" alt="doxygen"/></a> 1.8.13 </li> |
| 159 | </ul> |
| 160 | </div> |
| 161 | </body> |
| 162 | </html> |