MergeSegments.C

Go to the documentation of this file.

/*
# Copyright (C) 2019 Gabriele Vedovato
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <https://www.gnu.org/licenses/>.
*/
 
 
//
// Merge Segments
// this macro read a list of start stop segments from a file
// and merge the overlapping segments
// Example : 
// root -b -l 'MergeSegments.C("ifile.txt","ofile.txt")'
//
// Author : Gabriele Vedovato
 
 
void MergeSegments(TString ifile, TString ofile, bool bexit=true) {
 
  // --------------------------------------------------------------
  // Open file segment list
  // --------------------------------------------------------------
  ifstream in;
  in.open(ifile.Data(),ios::in);
  if (!in.good()) {cout << "Error Opening File : " << ifile << endl;gSystem->Exit(1);}
  cout << "input file list : " << ifile << endl;
 
  char str[1024];
  int fpos=0;
  int index=0;
  double start;
  double stop;
  waveSegment seg;
  vector<waveSegment> iseg; 
  while (1) {
    fpos=in.tellg();
    in.getline(str,1024);
    if(str[0] == '#') continue;
    in.seekg(fpos, ios::beg);
    fpos=in.tellg();
 
    in >> start >> stop;
    if (!in.good()) break;
 
    seg.index=index++; seg.start=start; seg.stop=stop; iseg.push_back(seg);
  }
 
  in.close();
 
//  cout << "List of input segments" << endl;
//  for(int n=0;n<iseg.size();n++) cout << n << " " << iseg[n].start << " " << iseg[n].stop << endl;
 
  // --------------------------------------------------------------
  // merge segments
  // --------------------------------------------------------------
  vector<waveSegment> oseg = CWB::Toolbox::unionSegments(iseg);
 
//  cout << "List of output segments" << endl;
//  for(int n=0;n<oseg.size();n++) cout << n << " " << oseg[n].start << " " << oseg[n].stop << endl;
 
  // --------------------------------------------------------------
  // Write output file segment list
  // --------------------------------------------------------------
  ofstream out;
  out.open(ofile.Data(),ios::out);
  out.precision(16);
  for(int n=0;n<oseg.size();n++) {
    out << oseg[n].start << " " << oseg[n].stop << endl;
  }
  out.close();
 
  cout << "output file list : " << ofile << endl;
 
  if(bexit) exit(0);
}