PRunDataHandler.h

Go to the documentation of this file.

/***************************************************************************

  PRunDataHandler.h

  Author: Andreas Suter
  e-mail: andreas.suter@psi.ch

***************************************************************************/

/***************************************************************************
 *   Copyright (C) 2007-2023 by Andreas Suter                              *
 *   andreas.suter@psi.ch                                                  *
 *                                                                         *
 *   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 2 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, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/

#ifndef _PRUNDATAHANDLER_H_
#define _PRUNDATAHANDLER_H_

#include <TString.h>

#include "PMusr.h"
#include "PMsrHandler.h"

class PRunDataHandler
{
  public:
    PRunDataHandler();
    PRunDataHandler(TString fileName, const TString fileFormat);
    PRunDataHandler(TString fileName, const TString fileFormat, const PStringVector dataPath);
    PRunDataHandler(TString fileName, const TString fileFormat, const TString dataPath, PRawRunData &runData);
    PRunDataHandler(PAny2ManyInfo *any2ManyInfo);
    PRunDataHandler(PAny2ManyInfo *any2ManyInfo, const PStringVector dataPath);
    PRunDataHandler(PMsrHandler *msrInfo);
    PRunDataHandler(PMsrHandler *msrInfo, const PStringVector dataPath);
    virtual ~PRunDataHandler();

    virtual void ReadData();
    virtual void ConvertData();
    virtual Bool_t WriteData(TString fileName="");

    virtual Bool_t IsAllDataAvailable() const { return fAllDataAvailable; }
    virtual PRawRunData* GetRunData(const TString &runName);
    virtual PRawRunData* GetRunData(const UInt_t idx=0);
    virtual Int_t GetNoOfRunData() {return fData.size(); }

    virtual Bool_t SetRunData(PRawRunData *data, UInt_t idx=0);

  private:
    PMsrHandler   *fMsrInfo; 
    PAny2ManyInfo *fAny2ManyInfo; 
    PStringVector fDataPath; 

    Bool_t fAllDataAvailable; 
    TString fFileFormat;      
    TString fRunName;         
    TString fRunPathName;     
    PRawRunDataList fData;    

    virtual void Init(const Int_t tag=0);
    virtual Bool_t ReadFilesMsr();
    virtual Bool_t ReadWriteFilesList();
    virtual Bool_t FileAlreadyRead(TString runName);
    virtual void TestFileName(TString &runName, const TString &ext);
    virtual Bool_t FileExistsCheck(PMsrRunBlock &runInfo, const UInt_t idx);
    virtual Bool_t FileExistsCheck(const Bool_t fileName, const Int_t idx);
    virtual Bool_t FileExistsCheck(const TString fileName);
    virtual Bool_t ReadRootFile();
    virtual Bool_t ReadNexusFile();
    virtual Bool_t ReadWkmFile();
    virtual Bool_t ReadPsiBinFile();
    virtual Bool_t ReadMudFile();
    virtual Bool_t ReadMduAsciiFile();
    virtual Bool_t ReadAsciiFile();
    virtual Bool_t ReadDBFile();
    virtual Bool_t ReadDatFile();

    virtual Bool_t WriteMusrRootFile(TString fln="");
    virtual Bool_t WriteRootFile(TString fln="");
    virtual Bool_t WriteNexusFile(TString fln="");
    virtual Bool_t WriteWkmFile(TString fln="");
    virtual Bool_t WritePsiBinFile(TString fln="");
    virtual Bool_t WriteMudFile(TString fln="");
    virtual Bool_t WriteAsciiFile(TString fln="");

    virtual Bool_t StripWhitespace(TString &str);
    virtual Bool_t IsWhitespace(const Char_t *str);
    virtual Double_t ToDouble(TString &str, Bool_t &ok);
    virtual Int_t ToInt(TString &str, Bool_t &ok);
    virtual Int_t GetDataTagIndex(TString &str, const PStringVector* fLabels);

    virtual TString GenerateOutputFileName(const TString fileName, const TString extension, Bool_t &ok);
    virtual TString GetFileName(const TString extension, Bool_t &ok);
    virtual TString FileNameFromTemplate(TString &fileNameTemplate, Int_t run, TString &year, Bool_t &ok);
    virtual bool DateToISO8601(std::string inDate, std::string &iso8601Date);
    virtual void SplitTimeDate(TString timeDate, TString &time, TString &date, Bool_t &ok);
    virtual TString GetMonth(Int_t month);
    virtual TString GetYear(Int_t month);
};

#endif //  _PRUNDATAHANDLER_H_