Changeset 19878

Timestamp:

11/30/19 15:07:23 (5 years ago)

Author:

tbretz

Message:

Branches must always be written, branch names must not start with a dot, added more possible delimiters, the default is now to read from stdin.

File:

• trunk/FACT++/src/csv2root.cc (modified) (11 diffs)

trunk/FACT++/src/csv2root.cc

@@ -26 +26 @@
     po::options_description control("Root to SQL");
     control.add_options()
-        ("file",           var<string>()->required(), "The csv input file")
+        ("file",           var<string>("-"),          "The csv input file. The default ('-') is reading from stdin.")
         ("out,o",          var<string>(""),           "Output root file name")
         ("force,f",        po_switch(),               "Force overwrite if output file already exists.")
@@ -34 +34 @@
         ("no-header,n",    po_switch(),               "Use if the first line contains no header")
         ("rename.*",       var<string>(),             "Can be used to rename a column")
+        ("delimiter",      var<string>(";:, \t"),     "List of possible delimiters")
         ("dry-run",        po_switch(),               "Do not create or manipulate any output file")
         ("verbose,v",      var<uint16_t>(1),          "Verbosity (0: quiet, 1: default, 2: more, 3, ...)")
@@ -148 +149 @@
 //    const bool dryrun            = conf.Get<bool>("dry-run");
     const bool noheader          = conf.Get<bool>("no-header");
+    const string delimiter       = conf.Get<string>("delimiter");
 
     const uint16_t verbose       = conf.Get<uint16_t>("verbose");
@@ -158 +160 @@
     if (out.empty())
     {
-        out = file;
+        out = file.empty() || file=="-" ? "csv2root" : file;
         const auto p = out.find_last_of('.');
         if (p!=string::npos)
-            out = string(out.substr(0, p))+".root";
+            out.erase(p);
+        out += ".root";
     }
 
@@ -179 +182 @@
     cout << "Reading from '" << file << "'.\n";
 
-    ifstream fin(file.c_str());
+    ifstream ifs(file.c_str());
+
+    istream &fin = file=="-" ? cin : ifs;
     if (!fin.good())
     {
@@ -195 +200 @@
 
     buf = buf.Strip(TString::kBoth);
-    TObjArray *title = buf.Tokenize(" ");
+    TObjArray *title = buf.Tokenize(delimiter);
     if (title->GetEntries()==0)
     {
@@ -203 +208 @@
 
     if (title->At(0)->GetName()[0]=='#')
+    {
         title->RemoveAt(0);
+        title->Compress();
+    }
 
     const auto numcol = title->GetEntries();
@@ -280 +288 @@
         }
     }
-
     const auto rename = conf.GetWildcardOptions("rename.*");
 
@@ -293 +300 @@
         boost::regex rexpr(":");
         col = boost::regex_replace(col, rexpr, "");
+
+        if (col[0]=='.')
+            col.erase(0, 1);
 
         if (verbose>1)
@@ -333 +343 @@
             continue;
 
-        TObjArray *arr = buf.Tokenize(" ");
+        TObjArray *arr = buf.Tokenize(delimiter);
         if (arr->GetEntries()!=numcol)
         {
@@ -369 +379 @@
     for (auto it=ttree.begin(); it!=ttree.end(); it++)
     {
-        if (update)
-        {
-            TIter NextBranch((*it)->GetListOfBranches());
-            TBranch *b=0;
-            while ((b=static_cast<TBranch*>(NextBranch())))
-                if (b->GetAddress() && b->GetEntries()>0)
-                {
-                    (*it)->SetEntries(b->GetEntries());
-                    break;
-                }
-        }
+        TIter NextBranch((*it)->GetListOfBranches());
+        TBranch *b=0;
+        while ((b=static_cast<TBranch*>(NextBranch())))
+            if (b->GetAddress() && b->GetEntries()>0)
+            {
+                (*it)->SetEntries(b->GetEntries());
+                break;
+            }
 
         (*it)->Write("", TObject::kOverwrite);