source: trunk/FACT++/src/DataWriteFits2.cc@ 18915

Last change on this file since 18915 was 18889, checked in by tbretz, 7 years ago
Propagate some variables as references where it makes sense.
File size: 11.9 KB
Line 
1#include "DataWriteFits2.h"
2
3#include <boost/filesystem.hpp>
4
5#include "HeadersFAD.h"
6#include "EventBuilder.h"
7
8#include "externals/factofits.h"
9#include "externals/DrsCalib.h"
10
11using namespace std;
12
13DataWriteFits2::DataWriteFits2(const std::string &path, uint64_t night, uint32_t runid, MessageImp &imp)
14 : DataProcessorImp(path, night, runid, imp)
15{
16 fFile = std::make_shared<ofits>();
17}
18
19DataWriteFits2::DataWriteFits2(const std::string &path, uint64_t night, uint32_t runid, const DrsCalibration &cal, MessageImp &imp)
20 : DataProcessorImp(path, night, runid, imp)
21{
22 factofits *file = new factofits;
23 file->SetDrsCalibration(cal);
24 fFile = std::shared_ptr<ofits>(file);
25}
26
27void DataWriteFits2::WriteHeader(const RUN_HEAD &h, const FAD::RunDescription &d)
28{
29 const int16_t realRoiTM = (h.NroiTM >= 2*h.Nroi && h.Nroi<=512) ? h.Nroi : 0;
30
31 fFile->AddColumnInt("EventNum", "uint32", "FAD board event counter");
32 fFile->AddColumnInt("TriggerNum", "uint32", "FTM board trigger counter");
33 fFile->AddColumnShort("TriggerType", "uint16", "FTM board trigger type");
34 fFile->AddColumnInt("NumBoards", "uint32", "Number of connected boards");
35 fFile->AddColumnInt(2, "UnixTimeUTC", "uint32", "Unix time seconds and microseconds");
36 fFile->AddColumnInt(NBOARDS, "BoardTime", "uint32", "Board internal time counter");
37 fFile->AddColumnShort(NPIX, "StartCellData", "uint16", "DRS4 start cell of readout");
38 fFile->AddColumnShort(NTMARK, "StartCellTimeMarker", "uint16", "DRS4 start cell of readout time marker");
39
40 vector<uint16_t> processing(2);
41 processing[0] = FITS::kFactSmoothing;
42 processing[1] = FITS::kFactHuffman16;
43
44 const FITS::Compression comp(processing, FITS::kOrderByRow);
45
46 fFile->AddColumnShort(comp, h.NPix*h.Nroi, "Data", "int16", "Digitized data");
47 fFile->AddColumnShort(comp, h.NTm*realRoiTM, "TimeMarker", "int16", "Digitized time marker - if available");
48
49 const size_t sz = (h.NPix*h.Nroi + h.NTm*realRoiTM)*2;
50 if (fFile->GetBytesPerRow()-sz+4!=sizeof(EVENT))
51 {
52 ostringstream str;
53 str << "The EVENT structure size (" << sizeof(EVENT) << ") doesn't match the described FITS row (";
54 str << fFile->GetBytesPerRow()-sz+4 << ")";
55 throw runtime_error(str.str());
56 }
57
58 // =============== Default keys for all files ================
59 fFile->SetDefaultKeys();
60 fFile->SetInt("NIGHT", GetNight(), "Night as int");
61
62 // ================ Header keys for raw-data =================
63 fFile->SetInt("BLDVER", h.Version, "Builder version");
64 fFile->SetInt("RUNID", GetRunId(), "Run number");
65 fFile->SetInt("NBOARD", h.NBoard, "Number of acquisition boards");
66 fFile->SetInt("NPIX", h.NPix, "Number of pixels");
67 fFile->SetInt("NTMARK", h.NTm, "Number of time marker channels");
68 fFile->SetInt("NCELLS", 1024, "Maximum number of slices per pixels");
69 fFile->SetInt("NROI", h.Nroi, "Number of slices per pixels");
70 fFile->SetInt("NROITM", realRoiTM, "Number of slices per time-marker");
71
72 const uint16_t realOffset = (h.NroiTM > h.Nroi) ? h.NroiTM - 2*h.Nroi : 0;
73 fFile->SetInt("TMSHIFT", realOffset, "Shift of marker readout w.r.t. to data");
74
75 //FIXME should we also put the start and stop time of the received data ?
76 //now the events header related variables
77 fFile->SetStr("CAMERA", "MGeomCamFACT", "MARS camera geometry class");
78 fFile->SetStr("DAQ", "DRS4", "Data acquisition type");
79 fFile->SetInt("ADCRANGE", 2000, "Dynamic range in mV");
80 fFile->SetInt("ADC", 12, "Resolution in bits");
81 fFile->SetStr("RUNTYPE", d.name, "File type according to FAD configuration");
82
83 // Write a single key for:
84 // -----------------------
85 // Start package flag
86 // package length
87 // version number
88 // status
89 // Prescaler
90
91 // Write 40 keys for (?)
92 // Phaseshift
93 // DAC
94
95 for (int i=0; i<h.NBoard; i++)
96 {
97 const PEVNT_HEADER &hh = h.FADhead[i];
98
99 ostringstream sout;
100 sout << "Board " << setw(2) << i<< ": ";
101
102 const string num = to_string(i);
103
104 // Header values whihc won't change during the run
105 fFile->SetInt("ID"+num, hh.board_id, sout.str()+"Board ID");
106 fFile->SetInt("FWVER"+num, hh.version_no, sout.str()+"Firmware Version");
107 fFile->SetHex("DNA"+num, hh.DNA, sout.str()+"Unique FPGA device identifier (DNA)");
108 }
109
110 // FIXME: Calculate average ref clock frequency
111 for (int i=0; i<h.NBoard; i++)
112 {
113 const PEVNT_HEADER &hh = h.FADhead[i];
114 if (hh.start_package_flag==0)
115 continue;
116
117 fFile->SetInt("BOARD", i, "Board number for RUN, PRESC, PHASE and DAC");
118 fFile->SetInt("PRESC", hh.trigger_generator_prescaler, "Trigger generator prescaler");
119 fFile->SetInt("PHASE", (int16_t)hh.adc_clock_phase_shift, "ADC clock phase shift");
120
121 for (int j=0; j<8; j++)
122 {
123 ostringstream dac, cmt;
124 dac << "DAC" << j;
125 cmt << "Command value for " << dac.str();
126 fFile->SetInt(dac.str(), hh.dac[j], cmt.str());
127 }
128
129 break;
130 }
131
132 double avg = 0;
133 int cnt = 0;
134 for (int i=0; i<h.NBoard; i++)
135 {
136 const PEVNT_HEADER &hh = h.FADhead[i];
137
138 if (hh.start_package_flag==0)
139 continue;
140
141 avg += hh.REFCLK_frequency;
142 cnt ++;
143 }
144
145 // FIXME: I cannot write a double! WHY?
146 fFile->SetFloat("REFCLK", avg/cnt*2.048, "Average reference clock frequency in Hz");
147
148 fFile->SetBool("DRSCALIB", GetDrsStep()>=0, "This file belongs to a DRS calibration");
149 if (GetDrsStep()>=0)
150 fFile->SetInt("DRSSTEP", GetDrsStep(), "Step of the DRS calibration");
151
152 fTstart[0] = h.RunTime;
153 fTstart[1] = h.RunUsec;
154
155 fTstop[0] = 0;
156 fTstop[1] = 0;
157
158 fTriggerCounter.fill(0);
159
160 WriteFooter();
161
162 fFile->WriteTableHeader("Events");
163};
164
165// --------------------------------------------------------------------------
166//
167//! DataWriteFits constructor. This is the one that should be used, not the default one (parameter-less)
168//! @param runid This parameter should probably be removed. I first thought it was the run number, but apparently it is not
169//! @param h a pointer to the RUN_HEAD structure that contains the informations relative to this run
170//
171bool DataWriteFits2::Open(const RUN_HEAD &h, const FAD::RunDescription &d)
172{
173 //Form filename, based on runid and run-type
174 fFileName = FormFileName(dynamic_pointer_cast<factofits>(fFile)?"fits.fz":"fits");
175
176 if (boost::filesystem::exists(fFileName))
177 {
178 Error("ofits - file '"+fFileName+"' already exists.");
179 return false;
180 }
181
182 zofits *fits = dynamic_cast<zofits*>(fFile.get());
183 if (fits)
184 {
185 const uint32_t nrpt = zofits::DefaultNumRowsPerTile();
186
187 // Maximum number of events if taken with 100Hz
188 // (If no limit requested, maxtime is 24*60*60)
189 const uint32_t ntime = d.maxtime*100/nrpt;
190
191 // Maximum number of events if taken as number
192 // (If no limit requested, maxevts is INT32_MAX)
193 const uint32_t nevts = d.maxevt/nrpt+1;
194
195 // get the minimum of all three
196 uint32_t num = zofits::DefaultMaxNumTiles();
197 if (ntime<num)
198 num = ntime;
199 if (nevts<num)
200 num = nevts;
201
202 fits->SetNumTiles(num);
203 }
204
205 try
206 {
207 fFile->open(fFileName.c_str());
208 }
209 catch (const exception &e)
210 {
211 Error("ofits::open() failed for '"+fFileName+"': "+e.what());
212 return false;
213 }
214
215 if (!(*fFile))
216 {
217 ostringstream str;
218 str << "ofstream::open() failed for '" << fFileName << "': " << strerror(errno) << " [errno=" << errno << "]";
219 Error(str);
220 return false;
221 }
222
223 try
224 {
225 WriteHeader(h, d);
226 }
227 catch (const exception &e)
228 {
229 Error("ofits - Writing header failed for '"+fFileName+"': "+e.what());
230 return false;
231 }
232
233 if (!(*fFile))
234 {
235 ostringstream str;
236 str << "ofstream::write() failed for '" << fFileName << "': " << strerror(errno) << " [errno=" << errno << "]";
237 Error(str);
238 return false;
239 }
240
241 return true;
242}
243
244// --------------------------------------------------------------------------
245//
246//! This writes one event to the file
247//! @param e the pointer to the EVENT
248//
249bool DataWriteFits2::WriteEvt(const EVT_CTRL2 &evt)
250{
251 // Remember the counter of the last written event
252 fTriggerCounter = evt.triggerCounter;
253
254 // Remember the time of the last event
255 fTstop[0] = evt.time.tv_sec;
256 fTstop[1] = evt.time.tv_usec;
257
258 const EVENT &e = *evt.fEvent;
259
260 const int realRoiTM = (e.RoiTM > e.Roi) ? e.Roi : 0;
261 const size_t sz = sizeof(EVENT) + sizeof(e.StartPix)*e.Roi+sizeof(e.StartTM)*realRoiTM; //ETIENNE from RoiTm to Roi
262
263 try
264 {
265 fFile->WriteRow(reinterpret_cast<const char*>(&e)+4, sz-4);
266 }
267 catch (const exception &ex)
268 {
269 Error("ofits::WriteRow failed for '"+fFileName+"': "+ex.what());
270 return false;
271 }
272
273 if (!(*fFile))
274 {
275 ostringstream str;
276 str << "fstream::write() failed for '" << fFileName << "': " << strerror(errno) << " [errno=" << errno << "]";
277 Error(str);
278 return false;
279 }
280
281 return true;
282}
283
284void DataWriteFits2::WriteFooter()
285{
286 //FIXME shouldn't we convert start and stop time to MjD first ?
287 //FIXME shouldn't we also add an MjD reference ?
288
289 const Time start(fTstart[0], fTstart[1]);
290 const Time stop (fTstop[0], fTstop[1]);
291
292 fFile->SetInt("TSTARTI", uint32_t(floor(start.UnixDate())),
293 "Time when first evt received (integral part)");
294 fFile->SetFloat("TSTARTF", fmod(start.UnixDate(), 1),
295 "Time when first evt received (fractional part)");
296 fFile->SetInt("TSTOPI", uint32_t(floor(stop.UnixDate())),
297 "Time when last evt received (integral part)");
298 fFile->SetFloat("TSTOPF", fmod(stop.UnixDate(), 1),
299 "Time when last evt received (fractional part)");
300 fFile->SetStr("DATE-OBS", start.Iso(),
301 "Time when first event received");
302 fFile->SetStr("DATE-END", stop.Iso(),
303 "Time when last event received");
304
305 fFile->SetInt("NTRG", fTriggerCounter[0], "No of physics triggered events");
306 fFile->SetInt("NTRGPED", fTriggerCounter[1], "No of pure pedestal triggered events");
307 fFile->SetInt("NTRGLPE", fTriggerCounter[2], "No of external light pulser triggered events");
308 fFile->SetInt("NTRGTIM", fTriggerCounter[3], "No of time calibration triggered events");
309 fFile->SetInt("NTRGLPI", fTriggerCounter[4], "No of internal light pulser triggered events");
310 fFile->SetInt("NTRGEXT1", fTriggerCounter[5], "No of triggers from ext1 triggered events");
311 fFile->SetInt("NTRGEXT2", fTriggerCounter[6], "No of triggers from ext2 triggered events");
312 fFile->SetInt("NTRGMISC", fTriggerCounter[7], "No of all other triggered events");
313}
314
315// --------------------------------------------------------------------------
316//
317//! Closes the file, and before this it write the TAIL data
318//! @param rt the pointer to the RUN_TAIL data structure
319//
320bool DataWriteFits2::Close(const EVT_CTRL2 &)
321{
322 if (!fFile->is_open())
323 {
324 Error("DataWriteFits2::Close() called but file '"+fFileName+"' not open.");
325 return false;
326 }
327
328 try
329 {
330 WriteFooter();
331 }
332 catch (const exception &e)
333 {
334 Error("ofits - Setting footer key values failed for '"+fFileName+"': "+e.what());
335 return false;
336 }
337
338 try
339 {
340 fFile->close();
341 }
342 catch (const exception &e)
343 {
344 Error("ofits::close() failed for '"+fFileName+"': "+e.what());
345 return false;
346 }
347
348 if (!(*fFile))
349 {
350 ostringstream str;
351 str << "ofstream::close() failed for '" << fFileName << "': " << strerror(errno) << " [errno=" << errno << "]";
352 Error(str);
353 return false;
354 }
355
356 return true;
357}
Note: See TracBrowser for help on using the repository browser.