Environment.h

Go to the documentation of this file.

//-----------------------------------------------------------------------bl-
//--------------------------------------------------------------------------
//
// QUESO - a library to support the Quantification of Uncertainty
// for Estimation, Simulation and Optimization
//
// Copyright (C) 2008-2015 The PECOS Development Team
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the Version 2.1 GNU Lesser General
// Public License as published by the Free Software Foundation.
//
// This library 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
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc. 51 Franklin Street, Fifth Floor,
// Boston, MA  02110-1301  USA
//
//-----------------------------------------------------------------------el-

#ifndef UQ_ENVIRONMENT_H
#define UQ_ENVIRONMENT_H

#include <queso/Defines.h>
namespace QUESO {
  class EnvironmentOptions;
}  // End namespace QUESO

#undef UQ_USES_COMMAND_LINE_OPTIONS

#include <queso/MpiComm.h>
#ifdef QUESO_HAS_HDF5
#include <hdf5.h>
#endif
#include <boost/program_options.hpp>
namespace po = boost::program_options;
#include <iostream>
#include <fstream>

#include <queso/RngBase.h>
#include <queso/BasicPdfsBase.h>

namespace QUESO {


  void queso_terminate_handler();

struct FilePtrSetStruct {

  FilePtrSetStruct();

  ~FilePtrSetStruct();

  std::ofstream* ofsVar;

  std::ifstream* ifsVar;
#ifdef QUESO_HAS_HDF5
  hid_t  h5Var;
#endif
};

//------------------------------------------------------------------------
// Library versioning routines: we include them in a QUESO namespace
// here so that multiple classes can use them as required (and so we
// can have a standalone versioning binary which does not require a full
// QUESO MPI environment.
//------------------------------------------------------------------------

  void QUESO_version_print       (std::ostream &os);
  int  QUESO_get_numeric_version ();

//*****************************************************
// Base class
//*****************************************************
class BaseEnvironment {
public:

  BaseEnvironment(const char* passedOptionsInputFileName, const EnvOptionsValues* alternativeOptionsValues);

  BaseEnvironment(const BaseEnvironment& obj);


  virtual ~BaseEnvironment();


  BaseEnvironment& operator= (const BaseEnvironment& rhs);


  bool    fullEnvIsReady() const;

  int     worldRank     () const;

  int     fullRank      () const;

  const MpiComm&   fullComm      () const;

  RawType_MPI_Group     subGroup      () const;

  int     subRank       () const;

  const MpiComm&   subComm       () const;

  const MpiComm&   selfComm      () const;

  int     inter0Rank    () const;

  const MpiComm&   inter0Comm    () const;

  std::ofstream*  subDisplayFile() const;

  std::string     subDisplayFileName    () const;

  unsigned int    numSubEnvironments    () const;

  unsigned int    subId () const;

  const std::string&      subIdString   () const;

  //TODO Not implemented?
  void    checkTheParallelEnvironment   () const;

  std::string     optionsInputFileName  () const;

  void    setOptionsInputFileAccessState(bool newState) const; // Yes, 'const'


#ifdef UQ_USES_COMMAND_LINE_OPTIONS
  const po::options_description& allOptionsDesc () const;
#endif

  po::variables_map&      allOptionsMap () const;



  void    scanInputFileForMyOptions(const po::options_description& optionsDesc) const;

  unsigned int    displayVerbosity () const;

  unsigned int    syncVerbosity    () const;

  unsigned int    checkingLevel    () const;

  const RngBase* rngObject  () const;

  void                  resetSeed  (int newSeedOption);

  int                   seed       () const;

  const BasicPdfsBase* basicPdfs() const;

  std::string platformName      () const;

  std::string identifyingString () const;

  void    resetIdentifyingString(const std::string& newString) const; // Yes, const

  bool    isThereInputFile      () const;

  struct timeval  timevalBegin  () const;



  bool    openOutputFile(const std::string& fileName, const std::string& fileType,
                         const std::set<unsigned int>& allowedSubEnvIds, bool writeOver,
                         FilePtrSetStruct& filePtrSet) const;

  bool    openUnifiedOutputFile (const std::string& fileName, const std::string& fileType,
                                 bool writeOver, FilePtrSetStruct& filePtrSet) const;

  bool    openInputFile (const std::string& fileName, const std::string& fileType,
                         const std::set<unsigned int>& allowedSubEnvIds,
                         FilePtrSetStruct& filePtrSet) const;

  bool    openUnifiedInputFile  (const std::string& fileName, const std::string& fileType,
                                 FilePtrSetStruct& filePtrSet) const;

  void    closeFile     (FilePtrSetStruct& filePtrSet, const std::string& fileType) const;

  void    setExceptionalCircumstance    (bool value) const;

  bool    exceptionalCircumstance       () const;


  virtual void    print (std::ostream& os) const = 0;

protected:
  bool                       m_fullEnvIsReady;
  int                        m_worldRank;

  MpiComm*           m_fullComm;
  int                        m_fullRank;
  int                        m_fullCommSize;
  RawType_MPI_Group        m_fullGroup;

  std::string                m_optionsInputFileName;
  mutable bool               m_optionsInputFileAccessState; // Yes, 'mutable'
  po::options_description*   m_allOptionsDesc;
  po::variables_map*         m_allOptionsMap;

  unsigned int               m_subId;
  std::string                m_subIdString;
  RawType_MPI_Group        m_subGroup;
  MpiComm*            m_subComm;
  int                        m_subRank;
  int                        m_subCommSize;

  MpiComm*            m_selfComm;

  RawType_MPI_Group        m_inter0Group;
  MpiComm*            m_inter0Comm;
  int                        m_inter0Rank;
  int                        m_inter0CommSize;

  mutable std::ofstream*     m_subDisplayFile;
  RngBase*           m_rngObject;
  BasicPdfsBase*      m_basicPdfs;
  struct timeval             m_timevalBegin;
  mutable bool               m_exceptionalCircumstance;

  EnvOptionsValues    m_alternativeOptionsValues;
  EnvironmentOptions* m_optionsObj;
};

//*****************************************************
// Empty Environment
//*****************************************************
class EmptyEnvironment : public BaseEnvironment {
public:


  EmptyEnvironment();

 ~EmptyEnvironment();

void print(std::ostream& os) const;
};

//*****************************************************
// Full Environment
//*****************************************************
class FullEnvironment : public BaseEnvironment {
public:


  FullEnvironment(RawType_MPI_Comm inputComm, const char* passedOptionsInputFileName, const char* prefix, const EnvOptionsValues* alternativeOptionsValues);

 ~FullEnvironment();


  void  print       (std::ostream& os) const;

private:
  void  readOptionsInputFile();
  //void        queso_terminate_handler();

};

std::ostream& operator<<(std::ostream& os, const BaseEnvironment& obj);

}  // End namespace QUESO

#endif // UQ_ENVIRONMENT_H