m_general.cc

Go to the documentation of this file.

/*===========================================================================
  === File description 
  ===========================================================================*/
 
/*===========================================================================
  === External declarations
  ===========================================================================*/
 
#include "arts.h"
 
#include <chrono>
#include <cstdlib>
#include <ratio>
#include <stdexcept>
 
#include "array.h"
#include "arts_constants.h"
#include "check_input.h"
#include "m_general.h"
#include "messages.h"
#include "mystring.h"
 
#include "math_funcs.h"
#include "wsv_aux.h"
 
#include "auto_md.h"
#include "workspace_ng.h"
 
#include "sensor.h"
 
#include "fastem.h"
#include "tessem.h"
 
inline constexpr Numeric SPEED_OF_LIGHT=Constant::speed_of_light;
 
/*===========================================================================
  === The functions (in alphabetical order)
  ===========================================================================*/
 
/* Workspace method: Doxygen documentation will be auto-generated */
void INCLUDE(const Verbosity&) {}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void Print(Workspace& ws _U_,
           // WS Generic Input:
           const Agenda& x,
           // Keywords:
           const Index& level,
           const Verbosity& verbosity) {
  ostringstream os;
  os << "    " << x.name() << " {\n";
  x.print(os, "        ");
  os << "    "
     << "}";
  CREATE_OUTS;
  SWITCH_OUTPUT(level, os.str());
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void Print(Workspace& ws _U_,
           // WS Generic Input:
           const ArrayOfAgenda& x,
           // Keywords:
           const Index& level,
           const Verbosity& verbosity) {
  ostringstream os;
  os << "    " << x.nelem() << " agendas: {\n";
  for (Index i = 0; i < x.nelem(); i++) {
    os << "      " << x[i].name() << ": {\n";
    x[i].print(os, "          ");
    os << "      "
       << "}\n";
  }
  os << "    "
     << "}";
  CREATE_OUTS;
  SWITCH_OUTPUT(level, os.str());
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void Print(  // WS Generic Input:
    const ArrayOfGridPos& x,
    // Keywords:
    const Index& level,
    const Verbosity& verbosity) {
  ostringstream os;
  for (Index i = 0; i < x.nelem(); i++) {
    if (i) os << '\n';
    os << "  " << x[i].idx << "  " << x[i].fd[0] << "  " << x[i].fd[1];
  }
  CREATE_OUTS;
  SWITCH_OUTPUT(level, os.str());
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void Print(  // WS Generic Input:
    const ArrayOfCIARecord& cia_data,
    // Keywords:
    const Index& level,
    const Verbosity& verbosity) {
  CREATE_OUTS;
 
  ostringstream os;
  os << "  CIA tag; Spectral range [cm-1]; Temp range [K]; # of sets\n";
  for (Index i = 0; i < cia_data.nelem(); i++)
    for (Index j = 0; j < cia_data[i].DatasetCount(); j++) {
      Vector temp_grid{cia_data[i].TemperatureGrid(j)};
      Vector freq_grid{cia_data[i].FrequencyGrid(j)};
 
      os << setprecision(2) << std::fixed << "  " << cia_data[i].MoleculeName(0)
         << "-CIA-" << cia_data[i].MoleculeName(1) << "-" << j << "; "
         << freq_grid[0] / 100. / SPEED_OF_LIGHT << " - "
         << freq_grid[freq_grid.nelem() - 1] / 100. / SPEED_OF_LIGHT
         << std::fixed << "; " << temp_grid[0] << " - "
         << temp_grid[temp_grid.nelem() - 1] << "; " << temp_grid.nelem()
         << "\n";
    }
  SWITCH_OUTPUT(level, os.str());
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void Print(  // WS Generic Input:
    const ArrayOfString& x,
    // Keywords:
    const Index& level,
    const Verbosity& verbosity) {
  ostringstream os;
  for (Index i = 0; i < x.nelem(); i++) {
    if (i) os << '\n';
    os << "  " << x[i];
  }
  CREATE_OUTS;
  SWITCH_OUTPUT(level, os.str());
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void Print(  // WS Generic Input:
    const ArrayOfPpath& x,
    // Keywords:
    const Index& level,
    const Verbosity& verbosity) {
  CREATE_OUTS;
  for (Index i = 0; i < x.nelem(); i++) {
    ostringstream os;
    os << "Ppath element " << i << ": ";
    SWITCH_OUTPUT(level, os.str());
    Print(x[i], level, verbosity);
  }
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void Print(  // WS Generic Input:
    const Timer& timer,
    // Keywords:
    const Index& level,
    const Verbosity& verbosity) {
  CREATE_OUTS;
  if (!timer.finished) {
    SWITCH_OUTPUT(
        level,
        "Timer error: Nothing to output. Use timerStart/timerStop first.");
    return;
  }
 
  ostringstream os;
 
  const auto cputime =
      static_cast<double>(timer.cputime_end - timer.cputime_start) /
      CLOCKS_PER_SEC;
  const auto walltime = std::chrono::duration<double, std::ratio<1>>(
                            timer.realtime_end - timer.realtime_start)
                            .count();
  os << std::fixed << setprecision(2) << "  * Timing: CPU " << cputime << "s, "
     << "Wall " << walltime << "s, " << 100. * cputime / walltime << "%CPU\n";
 
  SWITCH_OUTPUT(level, os.str());
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void Print(  // WS Generic Input:
    const TessemNN& x,
    // Keywords:
    const Index& level,
    const Verbosity& verbosity) {
  CREATE_OUTS;
  ostringstream os;
  os << "TessemNN size: Inputs = " << x.nb_inputs
     << ", Outputs = " << x.nb_outputs << ", Cache = " << x.nb_cache;
  SWITCH_OUTPUT(level, os.str());
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void PrintWorkspace(  // Workspace reference
    Workspace& ws,
    // Keywords:
    const Index& only_allocated,
    const Index& level,
    const Verbosity& verbosity) {
  ostringstream os;
 
  if (only_allocated)
    os << "  Allocated workspace variables: \n";
  else
    os << "  Workspace variables: \n";
  for (Index i = 0; i < ws.nelem(); i++) {
    if (!only_allocated) {
      os << "    ";
      ws.PrintWsvName(os, i);
      if (ws.is_initialized(i)) os << ws.depth(i);
      os << "\n";
    } else if (ws.is_initialized(i)) {
      os << "    ";
      ws.PrintWsvName(os, i);
      os << ws.depth(i) << "\n";
    }
  }
  CREATE_OUTS;
  SWITCH_OUTPUT(level, os.str());
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void StringJoin(String& out,
                const String& in1,
                const String& in2,
                const String& in3,
                const String& in4,
                const String& in5,
                const String& in6,
                const String& in7,
                const String& in8,
                const String& in9,
                const String& in10,
                const Verbosity&) {
  out = in1 + in2 + in3 + in4 + in5 + in6 + in7 + in8 + in9 + in10;
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void timerStart(  // WS Output
    Timer& timer,
    const Verbosity&) {
  timer.cputime_start = std::clock();
  timer.realtime_start = std::chrono::high_resolution_clock::now();
 
  timer.running = true;
  timer.finished = false;
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void timerStop(  // WS Input
    Timer& timer,
    const Verbosity&) {
  ARTS_USER_ERROR_IF(!timer.running,
                     "Timer error: Unable to stop timer that's not running.");
 
  timer.realtime_end = std::chrono::high_resolution_clock::now();
  timer.cputime_end = std::clock();
 
  timer.running = false;
  timer.finished = true;
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void Error(const String& msg, const Verbosity& verbosity) {
  CREATE_OUT0;
  ARTS_USER_ERROR ( msg);
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void Exit(const Verbosity& verbosity) {
  CREATE_OUT1;
  out1 << "  Forced exit.\n";
  arts_exit(EXIT_SUCCESS);
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void TestArrayOfAgenda(Workspace& ws,
                       const ArrayOfAgenda& test_agenda_array,
                       const Index& index,
                       const Verbosity&) {
  ostringstream os;
  os << "  Local value of iy_unit, agenda #" << index << " of "
     << test_agenda_array.nelem();
  test_agenda_arrayExecute(ws, index, os.str(), test_agenda_array);
}
 
void Test(const Verbosity&) {
  Numeric za, aa, dza_new, daa_new;
  const Numeric za0 = 67, aa0 = 12, dza = 9, daa = 11;
  add_za_aa(za, aa, za0, aa0, dza, daa);
  cout << za << " " << aa << endl;
  diff_za_aa(dza_new, daa_new, za0, aa0, za, aa);
  cout << dza_new << " " << daa_new << endl;
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void verbosityInit(  // WS Output:
    Verbosity& verbosity) {
  extern Verbosity verbosity_at_launch;
 
  verbosity.set_screen_verbosity(verbosity_at_launch.get_screen_verbosity());
  verbosity.set_agenda_verbosity(verbosity_at_launch.get_agenda_verbosity());
  verbosity.set_file_verbosity(verbosity_at_launch.get_file_verbosity());
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void verbositySet(  // WS Output:
    Verbosity& verbosity,
    // WS Generic Input:
    const Index& agenda,
    const Index& screen,
    const Index& file) {
  verbosity.set_agenda_verbosity(agenda);
  verbosity.set_screen_verbosity(screen);
  verbosity.set_file_verbosity(file);
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void verbositySetAgenda(  // WS Output:
    Verbosity& verbosity,
    // WS Generic Input:
    const Index& level) {
  verbosity.set_agenda_verbosity(level);
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void verbositySetFile(  // WS Output:
    Verbosity& verbosity,
    // WS Generic Input:
    const Index& level) {
  verbosity.set_file_verbosity(level);
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void verbositySetScreen(  // WS Output:
    Verbosity& verbosity,
    // WS Generic Input:
    const Index& level) {
  verbosity.set_screen_verbosity(level);
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void GetEnvironmentVariable(  // WS Generic Output:
    String& str,
    // WS Generic Input:
    const String& envvar,
    const Verbosity& /* verbosity */) {
  char* cstr;
  cstr = std::getenv(envvar.c_str());
  ARTS_USER_ERROR_IF (cstr == NULL,
    "Environment variable ", envvar, " does not exist.")
  str = cstr != NULL ? String(cstr) : "";
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void GetNumberOfThreads(Index& nthreads, const Verbosity& /* verbosity */) {
  nthreads = arts_omp_get_max_threads();
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
void GetEnvironmentVariable(  // WS Generic Output:
    Index& i,
    // WS Generic Input:
    const String& envvar,
    const Verbosity& /* verbosity */) {
  char* cstr;
  cstr = std::getenv(envvar.c_str());
  ARTS_USER_ERROR_IF (cstr == NULL || std::strlen(cstr) == 0,
    "Environment variable ", envvar, " "
    "is empty or does not exist.")
  std::istringstream is(cstr);
  is >> i;
  ARTS_USER_ERROR_IF (!is.eof(),
      "Cannot convert environment variable ", envvar, " "
      "to Index: ", cstr)
}
 
/* Workspace method: Doxygen documentation will be auto-generated */
#ifdef _OPENMP
void SetNumberOfThreads(const Index& nthreads,
                        const Verbosity& /* verbosity */) {
  omp_set_num_threads((int)nthreads);
}
#else
void SetNumberOfThreads(const Index& /* nthreads */,
                        const Verbosity& verbosity) {
  CREATE_OUT1;
  out1 << "No OpenMP support. Can't change number of threads.\n";
}
#endif