source: trunk/MagicSoft/Mars/mbase/MRunIter.cc@ 3897

/* ======================================================================== *\
!
! *
! * This file is part of MARS, the MAGIC Analysis and Reconstruction
! * Software. It is distributed to you in the hope that it can be a useful
! * and timesaving tool in analysing Data of imaging Cerenkov telescopes.
! * It is distributed WITHOUT ANY WARRANTY.
! *
! * Permission to use, copy, modify and distribute this software and its
! * documentation for any purpose is hereby granted without fee,
! * provided that the above copyright notice appear in all copies and
! * that both that copyright notice and this permission notice appear
! * in supporting documentation. It is provided "as is" without express
! * or implied warranty.
! *
!
!
!   Author(s): Thomas Bretz, 1/2004 <mailto:tbretz@astro.uni-wuerzburg.de>
!   Author(s): Javier Rico,  4/2004 <mailto:jrico@ifae.es>
!
!   Copyright: MAGIC Software Development, 2000-2004
!
!
\* ======================================================================== */

/////////////////////////////////////////////////////////////////////////////
//
//  MRunIter
//
//  Use this to iterate over run-files giving only the run-number.
//
//  You need the specify the run-file root-directory (eg /data/MAGIC).
//
/////////////////////////////////////////////////////////////////////////////
#include "MRunIter.h"

#include <TSystem.h>
#include <iostream>

ClassImp(MRunIter);

using namespace std;

Int_t MRunIter::AddRun(UInt_t run, const char *path)
{
    TString p(path);

    if (p.IsNull())
        p = fPath;

    if (p.IsNull())
        p = ".";

    MDirIter Next(p, Form("*_%05d_*_*.root", run), -1);

    const TString name(Next());
    if (name.IsNull())
        return 0;

    AddRunNumber(run);

    return AddDirectory(gSystem->DirName(name), gSystem->BaseName(name), -1);
}

// --------------------------------------------------------------------------
//
// Add runs specified in a character chain with the format:
// run1,run2-run3,run4-run5,...
// e.g  if runrange="100,105-107,110-112,115" 
// runs 100,105,106,107,110,111,112 and 115 are included in the iterator list
//
Int_t MRunIter::AddRuns(const char* runrange, const char* path)
{ 
  TString chcopy = runrange;
  Ssiz_t  last=0;
  Int_t   lowrun=-1;
  Int_t   upprun;
  UInt_t  totdir=0;


  // loop over the elements of the character chain 
  for(Int_t i=0;i<chcopy.Length();i++)
    {
      // look for a digit, a '-' or a ',' 
      char c=chcopy[i];
      if(! ((c>='0' && c<='9') || c=='-' || c==','))
        return totdir;
      
      // if '-' is found, save the previous number as initial run
      else if(c=='-' && lowrun<0 && i>last)
        {
          TSubString chrun = chcopy(last,i-last);
          lowrun=atoi(chrun.Data());              
          last=i+1;
        }
      // if ',' or the end of the string are found, save the previous run or run range
      else if(c==',' && i>last)
        {
          TSubString chrun = chcopy(last,i-last);
          upprun=atoi(chrun.Data());              
          if(lowrun>=0 && lowrun<=upprun)
            totdir+=AddRuns(lowrun,upprun,path);
          else if(lowrun<0)
            totdir+=AddRun(upprun,path);          
          
          lowrun=-1;
          last=i+1;
        }
      // if find two contiguous separators exit
      else if((c=='-' && i==last) || (c==',' && i==last))
        return totdir;
    }

  // save last run range
  TSubString chrun = chcopy(last,chcopy.Length()-last);
  upprun=atoi(chrun.Data());              
  if(lowrun>=0 && lowrun<=upprun)
    totdir+=AddRuns(lowrun,upprun,path);
  else if(lowrun<0)
    totdir+=AddRun(upprun,path);          

  return totdir;
}