PMsr2Data.h

Go to the documentation of this file.

/***************************************************************************
 
  PMsr2Data.h
 
  Author: Bastian M. Wojek / Andreas Suter
  e-mail: andreas.suter@psi.ch
 
***************************************************************************/
 
/***************************************************************************
 *   Copyright (C) 2009-2026 by Bastian M. Wojek / 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.             *
 ***************************************************************************/
 
// note: msr2data is on purpose implemented in a way that shows string handling can be done solely
//       using std::string, boost and related standard C++ features
//       This implies, however, occasionally strange constructs when interoperating with PMusr-classes
//       which mostly rely on ROOT's TString.
 
#ifndef _PMSR2DATA_H_
#define _PMSR2DATA_H_
 
#include <string>
#include <sstream>
 
#include "PRunDataHandler.h"
#include "PStartupHandler.h"
 
#include <TSAXParser.h>
 
//-------------------------------------------------------------
class PMsr2Data
{
  public:
    //-----------------------------------------------------------------------
    PMsr2Data(const std::string& fileExtension);
 
    //-----------------------------------------------------------------------
    ~PMsr2Data();
 
    //-----------------------------------------------------------------------
    int SetRunNumbers(unsigned int runNumber);
 
    //-----------------------------------------------------------------------
    int SetRunNumbers(unsigned int firstRun, unsigned int lastRun);
 
    //-----------------------------------------------------------------------
    int SetRunNumbers(const std::string& runListFileName);
 
    //-----------------------------------------------------------------------
    int SetRunNumbers(const std::vector<unsigned int>& runVector);
 
    //-----------------------------------------------------------------------
    unsigned int GetPresentRun() const;
 
    //-----------------------------------------------------------------------
    int DetermineRunNumberDigits(unsigned int maxRunNumber, bool templateFile) const;
 
    //-----------------------------------------------------------------------
    int CheckRunNumbersInRange() const;
 
    //-----------------------------------------------------------------------
    int ParseXmlStartupFile();
 
    //-----------------------------------------------------------------------
    int ReadMsrFile(const std::string& msrFileName) const;
 
    //-----------------------------------------------------------------------
    int ReadRunDataFile();
 
    //-----------------------------------------------------------------------
    bool PrepareNewInputFile(unsigned int templateNumber, bool sorted) const;
 
    //-----------------------------------------------------------------------
    bool PrepareGlobalInputFile(unsigned int templateNumber, const std::string& globalOutputFile, unsigned int runListSize, const std::string& musrfitOptions = "") const;
 
    //-----------------------------------------------------------------------
    int WriteOutput(const std::string& outputFileName, const std::vector<unsigned int>& parameters, bool dbFormat, unsigned int precision, bool global = false, unsigned int counter = 0) const;
 
  private:
    //-----------------------------------------------------------------------
    bool PrepareNewSortedInputFile(unsigned int templateNumber) const;
 
    //-----------------------------------------------------------------------
    PMsrHandler* GetSingleRunMsrFile() const;
 
    //-----------------------------------------------------------------------
    void WriteValue(std::fstream &outFile, const double &value, const unsigned int &width) const;
 
    //-----------------------------------------------------------------------
    void WriteValue(std::fstream &outFile, const double &value, const double &errValue, const unsigned int &width, const bool &db) const;
 
    //-----------------------------------------------------------------------
    int GetFirstSignificantDigit(const double &value) const;
 
    //-----------------------------------------------------------------------
    bool InParameterList(const unsigned int &paramValue, const std::vector<unsigned int>& paramList) const;
 
    //-----------------------------------------------------------------------
    std::string StripRunNumberSuffix(const std::string &name, const std::string &runNumberStr) const;
 
    std::string fFileExtension;                                 
    std::vector<unsigned int> fRunVector;                       
    mutable std::vector<unsigned int>::const_iterator fRunVectorIter; 
    bool fRunListFile;                                          
    std::vector<std::string> fIndVar;                           
    std::unique_ptr<std::ifstream> fRunListFileStream;          
    std::unique_ptr<TSAXParser> fSaxParser;                     
    std::unique_ptr<PStartupHandler> fStartupHandler;           
    mutable std::unique_ptr<PRunDataHandler> fDataHandler;      
    mutable std::unique_ptr<PMsrHandler> fMsrHandler;           
    mutable unsigned int fNumGlobalParam;                       
    mutable unsigned int fNumSpecParam;                         
    mutable unsigned int fNumTempRunBlocks;                     
    mutable unsigned int fRunNumberDigits;                      
    mutable bool fHeaderWritten;                                
};
 
 
#endif // _PMSR2DATA_H_