1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117
|
/* Copyright (c) 2008-2022 the MRtrix3 contributors.
*
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
*
* Covered Software is provided under this License on an "as is"
* basis, without warranty of any kind, either expressed, implied, or
* statutory, including, without limitation, warranties that the
* Covered Software is free of defects, merchantable, fit for a
* particular purpose or non-infringing.
* See the Mozilla Public License v. 2.0 for more details.
*
* For more details, see http://www.mrtrix.org/.
*/
#include <fstream>
#include "file/key_value.h"
#include "file/ofstream.h"
#include "app.h"
namespace MR
{
namespace File
{
namespace KeyValue
{
void Reader::open (const std::string& file, const char* first_line)
{
filename.clear();
DEBUG ("reading key/value file \"" + file + "\"...");
in.open (file.c_str(), std::ios::in | std::ios::binary);
if (!in)
throw Exception ("failed to open key/value file \"" + file + "\": " + strerror (errno));
if (first_line) {
std::string sbuf;
getline (in, sbuf);
if (sbuf.compare (0, strlen (first_line), first_line)) {
in.close();
throw Exception ("invalid first line for key/value file \"" + file + "\" (expected \"" + first_line + "\")");
}
}
filename = file;
}
bool Reader::next ()
{
while (in.good()) {
std::string sbuf;
getline (in, sbuf);
if (in.bad())
throw Exception ("error reading key/value file \"" + filename + "\": " + strerror (errno));
sbuf = strip (sbuf.substr (0, sbuf.find_first_of ('#')));
if (sbuf == "END") {
in.setstate (std::ios::eofbit);
return false;
}
if (sbuf.size()) {
size_t colon = sbuf.find_first_of (':');
if (colon == std::string::npos) {
INFO ("malformed key/value entry (\"" + sbuf + "\") in file \"" + filename + "\" - ignored");
}
else {
K = strip (sbuf.substr (0, colon));
V = strip (sbuf.substr (colon+1));
if (K.empty()) {
INFO ("malformed key/value entry (\"" + sbuf + "\") in file \"" + filename + "\" - ignored");
}
else
return true;
}
}
}
return false;
}
void write (File::OFStream& out,
const KeyValues& keyvals,
const std::string& prefix,
const bool add_to_command_history)
{
bool command_history_appended = false;
for (const auto& keyval : keyvals) {
const auto lines = split_lines(keyval.second);
for (const auto& line : lines)
out << prefix << keyval.first << ": " << line << "\n";
if (add_to_command_history && keyval.first == "command_history") {
out << prefix << "command_history: " << App::command_history_string << "\n";
command_history_appended = true;
}
}
if (add_to_command_history && !command_history_appended)
out << prefix << "command_history: " << App::command_history_string << "\n";
}
}
}
}
|