File: SCF_statistics.cc

package info (click to toggle)
ergo 3.8-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm, bullseye
  • size: 17,396 kB
  • sloc: cpp: 94,740; ansic: 17,015; sh: 7,559; makefile: 1,402; yacc: 127; lex: 110; awk: 23
file content (146 lines) | stat: -rw-r--r-- 5,103 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
/* Ergo, version 3.8, a program for linear scaling electronic structure
 * calculations.
 * Copyright (C) 2019 Elias Rudberg, Emanuel H. Rubensson, Pawel Salek,
 * and Anastasia Kruchinina.
 * 
 * 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 3 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, see <http://www.gnu.org/licenses/>.
 * 
 * Primary academic reference:
 * Ergo: An open-source program for linear-scaling electronic structure
 * calculations,
 * Elias Rudberg, Emanuel H. Rubensson, Pawel Salek, and Anastasia
 * Kruchinina,
 * SoftwareX 7, 107 (2018),
 * <http://dx.doi.org/10.1016/j.softx.2018.03.005>
 * 
 * For further information about Ergo, see <http://www.ergoscf.org>.
 */

/** @file SCF_statistics.cc

    @brief Class for keeping timings and other statistics related to
    self-consistent field (SCF) procedure.

    @author: Emanuel Rubensson <em>responsible</em>.
*/

#include <stdexcept>
#include <fstream>
#include "SCF_statistics.h"

SCF_timer::SCF_timer()
  : stopped_already(false) {
  startTimeWall = Util::TimeMeter::get_wall_seconds();
  Util::TimeMeter::get_current_cpu_times(startTimeCPU_usr, startTimeCPU_sys);
}

void SCF_timer::stop() {
  if (stopped_already)
    throw std::runtime_error("Attempt to stop timer already stopped.");
  elapsedTimeWall = Util::TimeMeter::get_wall_seconds() - startTimeWall;
  double stopTimeCPU_sys, stopTimeCPU_usr;
  Util::TimeMeter::get_current_cpu_times(stopTimeCPU_usr, stopTimeCPU_sys);
  elapsedTimeCPU_sys = stopTimeCPU_sys - startTimeCPU_sys;
  elapsedTimeCPU_usr = stopTimeCPU_usr - startTimeCPU_usr;    
  stopped_already = true;
}

void SCF_statistics::start_timer(std::string identifier) {
  timers[identifier] = SCF_timer();
}
void SCF_statistics::stop_timer(std::string identifier) {
  if ( timers.find(identifier) == timers.end() )
    throw std::runtime_error("Attempt to stop timer not in timer map.");
  timers[identifier].stop();
}

void SCF_statistics::add_value(std::string identifier, double value) {
  if ( values.find(identifier) != values.end() )
    throw std::runtime_error("Attempt to add value already in value map.");
  values[identifier] = value;
}

void  SCF_statistics::add_values( ValueMap & values_to_add) {
  ValueMap::const_iterator it;
  for ( it=values_to_add.begin() ; it != values_to_add.end(); it++ ) 
    values[it->first] = it->second;
}


void SCF_statistics::output_mfile(std::string name) {
  std::string m_name = name + ".m";
  std::ofstream os(m_name.c_str());
  
  // First output the names of all variables as one big comment
  os << "%% SCF_statistics, list of all variables " << std::endl;
  {
    os << "%% Timers: " << std::endl;
    TimerMap::const_iterator it;
    for ( it=timers.begin() ; it != timers.end(); it++ ) {
      std::string s = (*it).first;
      std::string s_wall = s + "_walltime"; 
      std::string s_cpu_sys = s + "_cpu_sys"; 
      std::string s_cpu_usr = s + "_cpu_usr"; 
      os << "% " << s_wall    << std::endl;
      os << "% " << s_cpu_sys << std::endl;
      os << "% " << s_cpu_usr << std::endl;
    }
  }
  {
    os << "%% Other values: " << std::endl;
    ValueMap::const_iterator it;
    for ( it=values.begin() ; it != values.end(); it++ ) {
      std::string s = (*it).first;
      os << "% " << s << std::endl;
    }
  }
  os << "%" << std::endl << std::endl;
  
  // Now output the values
  os << "%% SCF_statistics timers " << std::endl;
  {
    TimerMap::const_iterator it;
    for ( it=timers.begin() ; it != timers.end(); it++ ) {
      std::string s = (*it).first;
      std::string s_wall = s + "_walltime"; 
      std::string s_cpu_sys = s + "_cpu_sys"; 
      std::string s_cpu_usr = s + "_cpu_usr"; 
      double time_cpu_sys = (*it).second.elapsedTimeCPU_sys;
      double time_cpu_usr = (*it).second.elapsedTimeCPU_usr;
      double time_wall = (*it).second.elapsedTimeWall;
      output_value( os, s_wall   , time_wall    );
      output_value( os, s_cpu_sys, time_cpu_sys );
      output_value( os, s_cpu_usr, time_cpu_usr );
    }
  }
  os << "%% SCF_statistics other values " << std::endl;
  {
    ValueMap::const_iterator it;
    for ( it=values.begin() ; it != values.end(); it++ ) {
      std::string s = (*it).first;
      double value = (*it).second;
      output_value( os, s, value );
    }
  }
}

void SCF_statistics::output_value( std::ofstream & os, 
				   std::string id, 
				   double value ) {
  os << "if ( ~exist( '" << id << "' ) )" << std::endl;
  os << "  " << id << " = [];" << std::endl;
  os << "end" << std::endl;
  os << id << " = [" << id << " " << value << "];" 
     << std::endl;  
}