1 /**
2  * Copyright 2022 Google Inc.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #include "logging.hpp"
18 
19 #include "../tuning.hpp"
20 #include "pid.hpp"
21 
22 #include <chrono>
23 #include <fstream>
24 #include <iostream>
25 #include <map>
26 #include <string>
27 
28 namespace pid_control
29 {
30 namespace ec
31 {
32 
33 // Redundant log entries only once every 60 seconds
34 static constexpr int logThrottle = 60 * 1000;
35 
36 static std::map<std::string, PidCoreLog> nameToLog;
37 
CharValid(const std::string::value_type & ch)38 static bool CharValid(const std::string::value_type& ch)
39 {
40     // Intentionally avoiding invoking locale support here
41     if ((ch >= 'A') && (ch <= 'Z'))
42     {
43         return true;
44     }
45     if ((ch >= 'a') && (ch <= 'z'))
46     {
47         return true;
48     }
49     if ((ch >= '0') && (ch <= '9'))
50     {
51         return true;
52     }
53     return false;
54 }
55 
StrClean(const std::string & str)56 static std::string StrClean(const std::string& str)
57 {
58     std::string res;
59     size_t len = str.size();
60     for (size_t i = 0; i < len; ++i)
61     {
62         const auto& c = str[i];
63         if (!(CharValid(c)))
64         {
65             continue;
66         }
67         res += c;
68     }
69     return res;
70 }
71 
DumpContextHeader(std::ofstream & file)72 static void DumpContextHeader(std::ofstream& file)
73 {
74     file << "epoch_ms,input,setpoint,error";
75     file << ",proportionalTerm";
76     file << ",integralTerm1,integralTerm2";
77     file << ",derivativeTerm";
78     file << ",feedFwdTerm,output1,output2";
79     file << ",minOut,maxOut";
80     file << ",integralTerm3,output3";
81     file << ",integralTerm,output";
82     file << "\n" << std::flush;
83 }
84 
DumpContextData(std::ofstream & file,const std::chrono::milliseconds & msNow,const PidCoreContext & pc)85 static void DumpContextData(std::ofstream& file,
86                             const std::chrono::milliseconds& msNow,
87                             const PidCoreContext& pc)
88 {
89     file << msNow.count();
90     file << "," << pc.input << "," << pc.setpoint << "," << pc.error;
91     file << "," << pc.proportionalTerm;
92     file << "," << pc.integralTerm1 << "," << pc.integralTerm2;
93     file << "," << pc.derivativeTerm;
94     file << "," << pc.feedFwdTerm << "," << pc.output1 << "," << pc.output2;
95     file << "," << pc.minOut << "," << pc.maxOut;
96     file << "," << pc.integralTerm3 << "," << pc.output3;
97     file << "," << pc.integralTerm << "," << pc.output;
98     file << "\n" << std::flush;
99 }
100 
DumpCoeffsHeader(std::ofstream & file)101 static void DumpCoeffsHeader(std::ofstream& file)
102 {
103     file << "epoch_ms,ts,integral,lastOutput";
104     file << ",proportionalCoeff,integralCoeff";
105     file << ",derivativeCoeff";
106     file << ",feedFwdOffset,feedFwdGain";
107     file << ",integralLimit.min,integralLimit.max";
108     file << ",outLim.min,outLim.max";
109     file << ",slewNeg,slewPos";
110     file << ",positiveHysteresis,negativeHysteresis";
111     file << "\n" << std::flush;
112 }
113 
DumpCoeffsData(std::ofstream & file,const std::chrono::milliseconds & msNow,pid_info_t * pidinfoptr)114 static void DumpCoeffsData(std::ofstream& file,
115                            const std::chrono::milliseconds& msNow,
116                            pid_info_t* pidinfoptr)
117 {
118     // Save some typing
119     const auto& p = *pidinfoptr;
120 
121     file << msNow.count();
122     file << "," << p.ts << "," << p.integral << "," << p.lastOutput;
123     file << "," << p.proportionalCoeff << "," << p.integralCoeff;
124     file << "," << p.derivativeCoeff;
125     file << "," << p.feedFwdOffset << "," << p.feedFwdGain;
126     file << "," << p.integralLimit.min << "," << p.integralLimit.max;
127     file << "," << p.outLim.min << "," << p.outLim.max;
128     file << "," << p.slewNeg << "," << p.slewPos;
129     file << "," << p.positiveHysteresis << "," << p.negativeHysteresis;
130     file << "\n" << std::flush;
131 }
132 
LogInit(const std::string & name,pid_info_t * pidinfoptr)133 void LogInit(const std::string& name, pid_info_t* pidinfoptr)
134 {
135     if (!coreLoggingEnabled)
136     {
137         // PID logging not enabled by configuration, silently do nothing
138         return;
139     }
140 
141     if (name.empty())
142     {
143         std::cerr << "PID logging disabled because PID does not have a name\n";
144         return;
145     }
146 
147     std::string cleanName = StrClean(name);
148     if (cleanName.empty())
149     {
150         std::cerr << "PID logging disabled because PID name is unusable: "
151                   << name << "\n";
152         return;
153     }
154 
155     auto iterExisting = nameToLog.find(name);
156 
157     if (iterExisting != nameToLog.end())
158     {
159         std::cerr << "PID logging reusing existing file: " << name << "\n";
160     }
161     else
162     {
163         // Multiple names could collide to the same clean name
164         // Make sure clean name is not already used
165         for (const auto& iter : nameToLog)
166         {
167             if (iter.second.nameClean == cleanName)
168             {
169                 std::cerr << "PID logging disabled because of name collision: "
170                           << name << "\n";
171                 return;
172             }
173         }
174 
175         std::string filec = loggingPath + "/pidcore." + cleanName;
176         std::string filef = loggingPath + "/pidcoeffs." + cleanName;
177 
178         std::ofstream outc;
179         std::ofstream outf;
180 
181         outc.open(filec);
182         if (!(outc.good()))
183         {
184             std::cerr << "PID logging disabled because unable to open file: "
185                       << filec << "\n";
186             return;
187         }
188 
189         outf.open(filef);
190         if (!(outf.good()))
191         {
192             // Be sure to clean up all previous initialization
193             outf.close();
194 
195             std::cerr << "PID logging disabled because unable to open file: "
196                       << filef << "\n";
197             return;
198         }
199 
200         PidCoreLog newLog;
201 
202         // All good, commit to doing logging by moving into the map
203         newLog.nameOriginal = name;
204         newLog.nameClean = cleanName;
205         newLog.fileContext = std::move(outc);
206         newLog.fileCoeffs = std::move(outf);
207 
208         // The streams within this object are not copyable, must move them
209         nameToLog[name] = std::move(newLog);
210 
211         // This must now succeed, as it must be in the map
212         iterExisting = nameToLog.find(name);
213 
214         // Write headers only when creating files for the first time
215         DumpContextHeader(iterExisting->second.fileContext);
216         DumpCoeffsHeader(iterExisting->second.fileCoeffs);
217 
218         std::cerr << "PID logging initialized: " << name << "\n";
219     }
220 
221     auto msNow = LogTimestamp();
222 
223     // Write the coefficients only once per PID loop initialization
224     // If they change, caller will reinitialize the PID loops
225     DumpCoeffsData(iterExisting->second.fileCoeffs, msNow, pidinfoptr);
226 
227     // Force the next logging line to be logged
228     iterExisting->second.lastLog = iterExisting->second.lastLog.zero();
229     iterExisting->second.lastContext = PidCoreContext();
230 }
231 
LogPeek(const std::string & name)232 PidCoreLog* LogPeek(const std::string& name)
233 {
234     auto iter = nameToLog.find(name);
235     if (iter != nameToLog.end())
236     {
237         return &(iter->second);
238     }
239 
240     return nullptr;
241 }
242 
LogContext(PidCoreLog & pidLog,const std::chrono::milliseconds & msNow,const PidCoreContext & coreContext)243 void LogContext(PidCoreLog& pidLog, const std::chrono::milliseconds& msNow,
244                 const PidCoreContext& coreContext)
245 {
246     bool shouldLog = false;
247 
248     if (pidLog.lastLog == pidLog.lastLog.zero())
249     {
250         // It is the first time
251         shouldLog = true;
252     }
253     else
254     {
255         auto since = msNow - pidLog.lastLog;
256         if (since.count() >= logThrottle)
257         {
258             // It has been long enough since the last time
259             shouldLog = true;
260         }
261     }
262 
263     if (pidLog.lastContext != coreContext)
264     {
265         // The content is different
266         shouldLog = true;
267     }
268 
269     if (!shouldLog)
270     {
271         return;
272     }
273 
274     pidLog.lastLog = msNow;
275     pidLog.lastContext = coreContext;
276 
277     DumpContextData(pidLog.fileContext, msNow, coreContext);
278 }
279 
LogTimestamp(void)280 std::chrono::milliseconds LogTimestamp(void)
281 {
282     auto clockNow = std::chrono::high_resolution_clock::now();
283     auto msNow = std::chrono::duration_cast<std::chrono::milliseconds>(
284         clockNow.time_since_epoch());
285     return msNow;
286 }
287 
288 } // namespace ec
289 } // namespace pid_control
290