1 | /* ======================================================================== *\
|
---|
2 | !
|
---|
3 | ! *
|
---|
4 | ! * This file is part of MARS, the MAGIC Analysis and Reconstruction
|
---|
5 | ! * Software. It is distributed to you in the hope that it can be a useful
|
---|
6 | ! * and timesaving tool in analysing Data of imaging Cerenkov telescopes.
|
---|
7 | ! * It is distributed WITHOUT ANY WARRANTY.
|
---|
8 | ! *
|
---|
9 | ! * Permission to use, copy, modify and distribute this software and its
|
---|
10 | ! * documentation for any purpose is hereby granted without fee,
|
---|
11 | ! * provided that the above copyright notice appear in all copies and
|
---|
12 | ! * that both that copyright notice and this permission notice appear
|
---|
13 | ! * in supporting documentation. It is provided "as is" without express
|
---|
14 | ! * or implied warranty.
|
---|
15 | ! *
|
---|
16 | !
|
---|
17 | !
|
---|
18 | ! Author(s): Thomas Bretz 4/2007 <mailto:tbretz@astro.uni-wuerzburg.de>
|
---|
19 | !
|
---|
20 | ! Copyright: MAGIC Software Development, 2000-2007
|
---|
21 | !
|
---|
22 | !
|
---|
23 | \* ======================================================================== */
|
---|
24 |
|
---|
25 | /////////////////////////////////////////////////////////////////////////////
|
---|
26 | //
|
---|
27 | // MMovieWrite
|
---|
28 | //
|
---|
29 | // The intention of this class is to encode movies prepard by the
|
---|
30 | // MMoviePrepare task.
|
---|
31 | //
|
---|
32 | // For writing the movies the images are converted to ppm and piped through
|
---|
33 | // ppm2y4m to mpeg2enc. The output format is a mpeg2 movie and should
|
---|
34 | // be within the specifications of mpeg2 for DVD. Its size is 720x480,
|
---|
35 | // which is a good compromise between resolution and file size. The frame
|
---|
36 | // rate is fixed to 24fps.
|
---|
37 | //
|
---|
38 | // By changing the setup you can control the output:
|
---|
39 | //
|
---|
40 | // NumEvents: To make sure the file size doesn't get too large
|
---|
41 | // (300 evts are roughly 80MB with the default seetings) you can set
|
---|
42 | // a maximum number of events. If this number of events has been encoded
|
---|
43 | // the eventloop is stopped.
|
---|
44 | //
|
---|
45 | // TargetLength: The length (in seconds) each even will be encoded to.
|
---|
46 | // For example with the default Target Length of 5s and the fixed frame
|
---|
47 | // rate of 24fps each event will be encoded into 24f/s*5s +1f = 121frames
|
---|
48 | // equally distributed between the beginning of the first and the end of
|
---|
49 | // the last frame.
|
---|
50 | //
|
---|
51 | // Threshold: The default threshold is 2. At 2 times median pedestal rms
|
---|
52 | // of the pixles with area index 0 (for MAGIC: inner pixels) the color
|
---|
53 | // palette will change from yellow to red and isolated pixels between
|
---|
54 | // the median rms and 2 times the median of the rms will be removed from
|
---|
55 | // the image. To switch off this behaviour you can set a threshold
|
---|
56 | // below one.
|
---|
57 | //
|
---|
58 | // Filename: The output filename of the movie. If it doesn't end with ".mpg"
|
---|
59 | // the suffix is added.
|
---|
60 | //
|
---|
61 | // The interpolation of the frames is done using a TSpline3. If the spline
|
---|
62 | // would extrapolate due to the shift by the relative time calibration the
|
---|
63 | // contents is set to zero. Unsuitable pixels are interpolated frame by
|
---|
64 | // frame using the surrounding suitable pixels.
|
---|
65 | //
|
---|
66 | // A few words about file size: MPEG is a motion compensation compression,
|
---|
67 | // which means that if a region of a past frame is shown again at the same
|
---|
68 | // place or somewhere else this region is referenced instead of encoded again.
|
---|
69 | // This means that in our case (almost all frames are identical!) the
|
---|
70 | // increase of file size is far from linear with the number of encoded events!
|
---|
71 | //
|
---|
72 | //
|
---|
73 | // Input:
|
---|
74 | // MGeomCam
|
---|
75 | // MRawRunHeader
|
---|
76 | // MRawEvtHeader
|
---|
77 | // MSignalCam
|
---|
78 | // MBadPixelsCam
|
---|
79 | // MMovieData
|
---|
80 | // [MMcEvt]
|
---|
81 | //
|
---|
82 | /////////////////////////////////////////////////////////////////////////////
|
---|
83 | #include "MMovieWrite.h"
|
---|
84 |
|
---|
85 | #include <errno.h>
|
---|
86 |
|
---|
87 | #include <TF1.h>
|
---|
88 | #include <TStyle.h>
|
---|
89 | #include <TColor.h>
|
---|
90 | #include <TCanvas.h>
|
---|
91 | #include <TSystem.h>
|
---|
92 | #include <TASImage.h>
|
---|
93 | #include <TStopwatch.h>
|
---|
94 |
|
---|
95 | #include "MString.h"
|
---|
96 |
|
---|
97 | #include "MParList.h"
|
---|
98 | #include "MTaskList.h"
|
---|
99 |
|
---|
100 | #include "MGeomCam.h"
|
---|
101 | #include "MGeomPix.h"
|
---|
102 |
|
---|
103 | #include "MMcEvt.hxx"
|
---|
104 |
|
---|
105 | #include "MH.h"
|
---|
106 | #include "MHCamera.h"
|
---|
107 | #include "MMovieData.h"
|
---|
108 | #include "MSignalCam.h"
|
---|
109 | #include "MRawEvtHeader.h"
|
---|
110 | #include "MRawRunHeader.h"
|
---|
111 | #include "MBadPixelsCam.h"
|
---|
112 | #include "MBadPixelsPix.h"
|
---|
113 | #include "MCalibrateData.h"
|
---|
114 |
|
---|
115 | #include "MLog.h"
|
---|
116 | #include "MLogManip.h"
|
---|
117 |
|
---|
118 | ClassImp(MMovieWrite);
|
---|
119 |
|
---|
120 | using namespace std;
|
---|
121 |
|
---|
122 | // --------------------------------------------------------------------------
|
---|
123 | //
|
---|
124 | // Default constructor.
|
---|
125 | //
|
---|
126 | MMovieWrite::MMovieWrite(const char *name, const char *title)
|
---|
127 | : fPipe(0), fTargetLength(5), fThreshold(2), fNumEvents(25000), fFilename("movie.mpg")
|
---|
128 | {
|
---|
129 | fName = name ? name : "MMovieWrite";
|
---|
130 | fTitle = title ? title : "Task to encode a movie";
|
---|
131 | }
|
---|
132 |
|
---|
133 | // --------------------------------------------------------------------------
|
---|
134 | //
|
---|
135 | // Close pipe if still open
|
---|
136 | //
|
---|
137 | MMovieWrite::~MMovieWrite()
|
---|
138 | {
|
---|
139 | if (fPipe)
|
---|
140 | gSystem->ClosePipe(fPipe);
|
---|
141 | }
|
---|
142 |
|
---|
143 | // --------------------------------------------------------------------------
|
---|
144 | //
|
---|
145 | // Check the pipe for errors. In case of error print an error message.
|
---|
146 | // return kFALSE in case of error, kTRUE in case of success.
|
---|
147 | //
|
---|
148 | Bool_t MMovieWrite::CheckPipe()
|
---|
149 | {
|
---|
150 | if (!ferror(fPipe))
|
---|
151 | return kTRUE;
|
---|
152 |
|
---|
153 | *fLog << err << "Error in pipe: " << strerror(errno) << endl;
|
---|
154 | return kFALSE;
|
---|
155 | }
|
---|
156 |
|
---|
157 | // --------------------------------------------------------------------------
|
---|
158 | //
|
---|
159 | // Open pipe for encoding the movie
|
---|
160 | //
|
---|
161 | Bool_t MMovieWrite::OpenPipe()
|
---|
162 | {
|
---|
163 | // name = "ppmtoy4m -B -S 420mpeg2 -v 0 | yuvplay";
|
---|
164 | // name = Form("ppmtoy4m -B -S 420jpeg -v 0 -F %d:%d | yuv2lav -v 0 -o output%03d.avi", TMath::Nint() TMath::Nint(fTargetLength*1000))
|
---|
165 | // name = "ppmtoy4m -B -F 3:1 -S 420jpeg -v 0 | yuv2lav -v 0 -o output.avi";
|
---|
166 |
|
---|
167 | TString name;
|
---|
168 | name = "ppmtoy4m -B -F 24:1 -S 420jpeg -v 0 | ";
|
---|
169 | name += "mpeg2enc -v 0 -F 2 -I 0 -M 2 -o ";
|
---|
170 | name += fFilename;
|
---|
171 | if (!fFilename.EndsWith(".mpg"))
|
---|
172 | name += ".mpg";
|
---|
173 |
|
---|
174 | const Int_t n = TMath::Nint(fTargetLength*24)+1;
|
---|
175 |
|
---|
176 | name += " -f 9 -E 40 -r 0 -K kvcd ";
|
---|
177 | name += MString::Format("-g %d -G %d", n, n);
|
---|
178 |
|
---|
179 | // For higher resolution add "--no-constraints"
|
---|
180 |
|
---|
181 | fPipe = gSystem->OpenPipe(name, "w");
|
---|
182 | if (!fPipe)
|
---|
183 | {
|
---|
184 | *fLog << err;
|
---|
185 | *fLog << "Pipe: " << name << endl;
|
---|
186 | *fLog << "Couldn't open pipe... aborting." << endl;
|
---|
187 | CheckPipe();
|
---|
188 | return kFALSE;
|
---|
189 | }
|
---|
190 |
|
---|
191 | *fLog << inf << "Setup pipe to ppmtoy4m and mpeg2enc to encode " << fFilename << "." << endl;
|
---|
192 |
|
---|
193 | return kTRUE;
|
---|
194 |
|
---|
195 | // 1: 37M name += "-f 9 -E 40 -H -4 1 -2 1 --dualprime-mpeg2";
|
---|
196 | // 2: 42M name += "-f 9";
|
---|
197 | // 3: 37M name += "-f 9 -E 40 -4 1 -2 1 --dualprime-mpeg2";
|
---|
198 | // 4: 37M name += "-f 9 -E 40 -4 1 -2 1";
|
---|
199 | // 5: 37M name += "-f 9 -E 40 -4 4 -2 4"; // 640x400 3 frames/slice
|
---|
200 | // 6: 11M name += "-f 3 -E 40 -b 750"; // 640x400 3 frames/slice
|
---|
201 |
|
---|
202 | // 7: 28M name += "-f 9 -E 40 -G 50"; // 640x400 3 frames/slice
|
---|
203 | // 8: 24M name += "-f 9 -E 40 -G 500"; // 640x400 3 frames/slice
|
---|
204 |
|
---|
205 | // 9: 17M name += "-f 9 -E 40 -G 2400"; // 640x400 24 frames/slice
|
---|
206 | // 10: 19M name += "-f 9 -E 40 -G 1120"; // 720x480 3 frames/slice
|
---|
207 | // 20: 33M name += "-f 9 -E 40 -g 28 -G 28"; // 720x480 3 frames/slice
|
---|
208 | // 57M name += "-f 9 -E 40 -g 28 -G 28 -q 4"; // 720x480 3 frames/slice
|
---|
209 |
|
---|
210 | // 30M name += "-f 9 -E 40 -g 84 -G 84 -r 0"; // 720x480 3 frames/slice
|
---|
211 | // 31M name += "-f 9 -E 40 -g 56 -G 56 -r 0"; // 720x480 3 frames/slice
|
---|
212 | // 34M name += "-f 9 -E 40 -g 28 -G 28 -r 0"; // 720x480 3 frames/slice
|
---|
213 | // 24: 24M name += "-f 9 -E 40 -g 28 -G 28 -r 0 -K kvcd"; // 720x480 3 frames/slice
|
---|
214 | // 25: 24M name += "-f 9 -E -40 -g 28 -G 28 -r 0 -K kvcd"; // 720x480 3 frames/slice
|
---|
215 | // 26: 26M name += "-f 9 -E 0 -g 28 -G 28 -r 0 -K kvcd"; // 720x480 3 frames/slice
|
---|
216 | // 34M name += "-f 9 -E 40 -g 28 -G 28 -r 2"; // 720x480 3 frames/slice
|
---|
217 | // 33M name += "-f 9 -E 40 -g 28 -G 28 -r 32"; // 720x480 3 frames/slice
|
---|
218 |
|
---|
219 | // name += "-f 9 -E 40 -g 121 -G 121 -r 0 -K kvcd"; // 720x480 5s 24 frames/slice
|
---|
220 |
|
---|
221 | // 11: 56M name += "-f 9 -E 40 -g 217 -G 217"; // 720x480 24 frames/slice
|
---|
222 | // 18: 59M name += "-f 9 -E 40 -G 250"; // 720x480 24 frames/slice
|
---|
223 | // 62M name += "-f 9 -E 40 -G 184"; // 720x480 24 frames/slice
|
---|
224 |
|
---|
225 | // 12: --- name += "-f 9 -E 40 -G 500 -q 31"; // 720x480 3frames/slice
|
---|
226 | // 13: 49M name += "-f 9 -E 40 -G 500 -q 4"; // 720x480 3frames/slice
|
---|
227 | // 14: 21M name += "-f 9 -E 40 -G 500 -q 4 -b 1500"; // 720x480 3frames/slice
|
---|
228 |
|
---|
229 | // 15: 57M name += "-f 9 -E 40 -G 500 --no-constraints"; // 1280 864 3frames/slice
|
---|
230 |
|
---|
231 | // 16: >80 name += "-f 9 -E 40 -G 217 --no-constraints -b 3000"; // 1280 864 24frames/slice
|
---|
232 | // 17: >50 name += "-f 9 -E 40 -G 682 -b 3000 --no-constraints"; // 1280 864 24frames/slice
|
---|
233 | }
|
---|
234 |
|
---|
235 | // --------------------------------------------------------------------------
|
---|
236 | //
|
---|
237 | // Search for:
|
---|
238 | // - MGeomCam
|
---|
239 | // - MRawRunHeader
|
---|
240 | // - MRawEvtHeader
|
---|
241 | // - MSignalCam
|
---|
242 | // - MBadPixelsCam
|
---|
243 | // - MMovieData
|
---|
244 | //
|
---|
245 | // Open a pipe to write the images to. Can be either a player or
|
---|
246 | // an encoder.
|
---|
247 | //
|
---|
248 | Int_t MMovieWrite::PreProcess(MParList *plist)
|
---|
249 | {
|
---|
250 | fCam = (MGeomCam*)plist->FindObject("MGeomCam");
|
---|
251 | if (!fCam)
|
---|
252 | {
|
---|
253 | *fLog << err << "MGeomCam not found ... aborting." << endl;
|
---|
254 | return kFALSE;
|
---|
255 | }
|
---|
256 | fRun = (MRawRunHeader*)plist->FindObject("MRawRunHeader");
|
---|
257 | if (!fRun)
|
---|
258 | {
|
---|
259 | *fLog << err << "MRawRunHeader not found ... aborting." << endl;
|
---|
260 | return kFALSE;
|
---|
261 | }
|
---|
262 | fHead = (MRawEvtHeader*)plist->FindObject("MRawEvtHeader");
|
---|
263 | if (!fHead)
|
---|
264 | {
|
---|
265 | *fLog << err << "MRawEvtHeader not found ... aborting." << endl;
|
---|
266 | return kFALSE;
|
---|
267 | }
|
---|
268 | fSig = (MSignalCam*)plist->FindObject("MSignalCam");
|
---|
269 | if (!fSig)
|
---|
270 | {
|
---|
271 | *fLog << err << "MSignalCam not found ... aborting." << endl;
|
---|
272 | return kFALSE;
|
---|
273 | }
|
---|
274 | fBad = (MBadPixelsCam*)plist->FindObject("MBadPixelsCam");
|
---|
275 | if (!fBad)
|
---|
276 | {
|
---|
277 | *fLog << err << "MBadPixelsCam not found ... aborting." << endl;
|
---|
278 | return kFALSE;
|
---|
279 | }
|
---|
280 | fIn = (MMovieData*)plist->FindObject("MMovieData");
|
---|
281 | if (!fIn)
|
---|
282 | {
|
---|
283 | *fLog << err << "MMovieData not found... aborting." << endl;
|
---|
284 | return kFALSE;
|
---|
285 | }
|
---|
286 |
|
---|
287 | fMC = (MMcEvt*)plist->FindObject("MMcEvt");
|
---|
288 |
|
---|
289 | return OpenPipe();
|
---|
290 | }
|
---|
291 |
|
---|
292 | #ifdef USE_TIMING
|
---|
293 | TStopwatch clockT, clock1, clock2, clock3;
|
---|
294 | #endif
|
---|
295 |
|
---|
296 | // --------------------------------------------------------------------------
|
---|
297 | //
|
---|
298 | // Close pipe if still open
|
---|
299 | //
|
---|
300 | Int_t MMovieWrite::PostProcess()
|
---|
301 | {
|
---|
302 | if (fPipe)
|
---|
303 | {
|
---|
304 | gSystem->ClosePipe(fPipe);
|
---|
305 | fPipe=0;
|
---|
306 | }
|
---|
307 |
|
---|
308 | #ifdef USE_TIMING
|
---|
309 | *fLog << all << endl;
|
---|
310 | *fLog << "Snap: " << flush;
|
---|
311 | clock1.Print();
|
---|
312 | *fLog << "Writ: " << flush;
|
---|
313 | clock2.Print();
|
---|
314 | *fLog << "Prep: " << flush;
|
---|
315 | clock3.Print();
|
---|
316 | *fLog << "Totl: " << flush;
|
---|
317 | clockT.Print();
|
---|
318 | *fLog << endl;
|
---|
319 | #endif
|
---|
320 |
|
---|
321 | return kTRUE;
|
---|
322 | }
|
---|
323 |
|
---|
324 | // --------------------------------------------------------------------------
|
---|
325 | //
|
---|
326 | // Produce a 99 color palette made such, that everything below one
|
---|
327 | // pedestal rms is white, everything up to two pedestal rms is yellow
|
---|
328 | // and everything above gets colors.
|
---|
329 | //
|
---|
330 | Int_t MMovieWrite::SetPalette(Double_t rms, const TH1 &h) const
|
---|
331 | {
|
---|
332 | const Double_t min = h.GetMinimum();
|
---|
333 | const Double_t max = h.GetMaximum();
|
---|
334 |
|
---|
335 | const Double_t f = (fThreshold*rms-min)/(max-min);
|
---|
336 | const Double_t w = 1-f;
|
---|
337 |
|
---|
338 | // --- Produce the nice colored palette ---
|
---|
339 |
|
---|
340 | // min th*rms max
|
---|
341 | double s[6] = {0.0, f/2, f, f+w/4, f+3*w/5, 1.0 };
|
---|
342 |
|
---|
343 | double r[6] = {1.0, 1.0, 1.0, 0.85, 0.1, 0.0 };
|
---|
344 | double g[6] = {1.0, 1.0, 1.0, 0.15, 0.1, 0.0 };
|
---|
345 | double b[6] = {0.9, 0.55, 0.4, 0.15, 0.7, 0.1 };
|
---|
346 |
|
---|
347 | TArrayI col(99);
|
---|
348 |
|
---|
349 | #if ROOT_VERSION_CODE < ROOT_VERSION(5,18,00)
|
---|
350 | const Int_t rc = gStyle->CreateGradientColorTable(6, s, r, g, b, col.GetSize());
|
---|
351 | #else
|
---|
352 | const Int_t rc = TColor::CreateGradientColorTable(6, s, r, g, b, col.GetSize());
|
---|
353 | #endif
|
---|
354 |
|
---|
355 | // --- Overwrite the 'underflow' bin with white ---
|
---|
356 |
|
---|
357 | for (int i=0; i<col.GetSize(); i++)
|
---|
358 | col[i] = gStyle->GetColorPalette(i);
|
---|
359 |
|
---|
360 | col[0] = TColor::GetColor(0xff, 0xff, 0xff);
|
---|
361 |
|
---|
362 | // --- Set Plette ---
|
---|
363 |
|
---|
364 | gStyle->SetPalette(col.GetSize(), col.GetArray());
|
---|
365 |
|
---|
366 | return rc;
|
---|
367 | }
|
---|
368 |
|
---|
369 | // --------------------------------------------------------------------------
|
---|
370 | //
|
---|
371 | // The created colors are not overwritten and must be deleted manually
|
---|
372 | // because having more than 32768 color in a palette will crash
|
---|
373 | // gPad->PaintBox
|
---|
374 | //
|
---|
375 | void MMovieWrite::DeletePalette(Int_t colidx) const
|
---|
376 | {
|
---|
377 | for (int i=0; i<99; i++)
|
---|
378 | {
|
---|
379 | TColor *col = gROOT->GetColor(colidx+i);
|
---|
380 | if (col)
|
---|
381 | delete col;
|
---|
382 | }
|
---|
383 | }
|
---|
384 |
|
---|
385 | /*
|
---|
386 | // --------------------------------------------------------------------------
|
---|
387 | //
|
---|
388 | // Do a snapshot from the pad via TASImage::FromPad and write the
|
---|
389 | // image to the pipe.
|
---|
390 | // return kFALSE in case of error, kTRUE in case of success.
|
---|
391 | //
|
---|
392 | Bool_t MMovieWrite::WriteImage(TVirtualPad &pad)
|
---|
393 | {
|
---|
394 | clock1.Start(kFALSE);
|
---|
395 | TASImage img;
|
---|
396 | img.FromPad(&pad);
|
---|
397 | clock1.Stop();
|
---|
398 |
|
---|
399 | clock2.Start(kFALSE);
|
---|
400 | const Bool_t rc = WriteImage(img);
|
---|
401 | clock2.Stop();
|
---|
402 |
|
---|
403 | return rc;
|
---|
404 | }
|
---|
405 |
|
---|
406 | #include <TVirtualPS.h>
|
---|
407 | Bool_t MMovieWrite::WriteImage(TVirtualPad &pad)
|
---|
408 | {
|
---|
409 | TVirtualPS *psave = gVirtualPS;
|
---|
410 |
|
---|
411 | clock1.Start(kFALSE);
|
---|
412 | TImage dump("", 114);
|
---|
413 | dump.SetBit(BIT(11));
|
---|
414 | pad.Paint();
|
---|
415 | TASImage *itmp = (TASImage*)dump.GetStream();
|
---|
416 | clock1.Stop();
|
---|
417 |
|
---|
418 | clock2.Start(kFALSE);
|
---|
419 | const Bool_t rc = WriteImage(*itmp);
|
---|
420 | clock2.Stop();
|
---|
421 |
|
---|
422 | gVirtualPS = psave;
|
---|
423 |
|
---|
424 | return rc;
|
---|
425 | }
|
---|
426 | */
|
---|
427 |
|
---|
428 | // --------------------------------------------------------------------------
|
---|
429 | //
|
---|
430 | // Update the part of the idst image with the contents of the pad.
|
---|
431 | //
|
---|
432 | // It is a lot faster not to rerender the parts of the image which don't
|
---|
433 | // change anyhow, because rerendering the camera is by far the slowest.
|
---|
434 | //
|
---|
435 | void MMovieWrite::UpdateImage(TASImage &idst, TVirtualPad &pad)
|
---|
436 | {
|
---|
437 | // Get image from pad
|
---|
438 | TASImage isrc;
|
---|
439 | isrc.FromPad(&pad);
|
---|
440 |
|
---|
441 | // Get position and width of destination- and source-image
|
---|
442 | const UInt_t wsrc = isrc.GetWidth(); // width of image
|
---|
443 | const UInt_t hsrc = isrc.GetHeight(); // height of image
|
---|
444 |
|
---|
445 | const UInt_t usrc = pad.UtoPixel(1)*4; // width of pad (argb)
|
---|
446 | //const UInt_t vsrc = pad.VtoPixel(0); // height of pad
|
---|
447 |
|
---|
448 | const UInt_t xsrc = pad.UtoAbsPixel(0); // offset of pad in canvas
|
---|
449 | const UInt_t ysrc = pad.VtoAbsPixel(1); // offset of pad in canvas
|
---|
450 |
|
---|
451 | const UInt_t wdst = idst.GetWidth();
|
---|
452 | //const UInt_t hdst = idst.GetHeight();
|
---|
453 |
|
---|
454 | // Update destination image with source image
|
---|
455 | const UInt_t size = wsrc*hsrc;
|
---|
456 |
|
---|
457 | UInt_t *psrc = isrc.GetArgbArray();
|
---|
458 | UInt_t *pdst = idst.GetArgbArray();
|
---|
459 |
|
---|
460 | UInt_t *src = psrc + ysrc*wsrc+xsrc;
|
---|
461 | UInt_t *dst = pdst + ysrc*wdst+xsrc;
|
---|
462 |
|
---|
463 | while (src<psrc+size)
|
---|
464 | {
|
---|
465 | memcpy(dst, src, usrc);
|
---|
466 |
|
---|
467 | src += wsrc;
|
---|
468 | dst += wdst;
|
---|
469 | }
|
---|
470 | }
|
---|
471 |
|
---|
472 | // --------------------------------------------------------------------------
|
---|
473 | //
|
---|
474 | // Write the image as ppm (raw/P6) to the pipe.
|
---|
475 | // return kFALSE in case of error, kTRUE in case of success.
|
---|
476 | //
|
---|
477 | Bool_t MMovieWrite::WriteImage(TASImage &img)
|
---|
478 | {
|
---|
479 | // Write image header
|
---|
480 | fprintf(fPipe, "P6 %d %d 255\n", img.GetWidth(), img.GetHeight());
|
---|
481 | if (!CheckPipe())
|
---|
482 | return kFALSE;
|
---|
483 |
|
---|
484 | // Write image data (remove alpha channel from argb data)
|
---|
485 | UInt_t *argb = img.GetArgbArray();
|
---|
486 | for (UInt_t *ptr=argb; ptr<argb+img.GetWidth()*img.GetHeight(); ptr++)
|
---|
487 | if (fwrite(ptr, 3, 1, fPipe)!=1)
|
---|
488 | break;
|
---|
489 |
|
---|
490 | return CheckPipe();
|
---|
491 | }
|
---|
492 |
|
---|
493 | // --------------------------------------------------------------------------
|
---|
494 | //
|
---|
495 | // Update TASImage with changing parts of the image and write image to pipe.
|
---|
496 | // return kFALSE in case of error, kTRUE in case of success.
|
---|
497 | //
|
---|
498 | Bool_t MMovieWrite::WriteImage(TASImage &img, TVirtualPad &pad)
|
---|
499 | {
|
---|
500 | #ifdef USE_TIMING
|
---|
501 | clock1.Start(kFALSE);
|
---|
502 | UpdateImage(img, pad);
|
---|
503 | clock1.Stop();
|
---|
504 |
|
---|
505 | clock2.Start(kFALSE);
|
---|
506 | const Bool_t rc = WriteImage(img);
|
---|
507 | clock2.Stop();
|
---|
508 |
|
---|
509 | return rc;
|
---|
510 | #else
|
---|
511 | UpdateImage(img, pad);
|
---|
512 | return WriteImage(img);
|
---|
513 | #endif
|
---|
514 | }
|
---|
515 |
|
---|
516 | // --------------------------------------------------------------------------
|
---|
517 | //
|
---|
518 | // Do a simple interpolation of the surrounding suitable pixels for all
|
---|
519 | // unsuitable pixels.
|
---|
520 | //
|
---|
521 | void MMovieWrite::TreatBadPixels(TH1 &h) const
|
---|
522 | {
|
---|
523 | const UShort_t entries = fCam->GetNumPixels();
|
---|
524 |
|
---|
525 | //
|
---|
526 | // Loop over all pixels
|
---|
527 | //
|
---|
528 | for (UShort_t i=0; i<entries; i++)
|
---|
529 | {
|
---|
530 | //
|
---|
531 | // Check whether pixel with idx i is blind
|
---|
532 | //
|
---|
533 | if (!(*fBad)[i].IsUnsuitable())
|
---|
534 | continue;
|
---|
535 |
|
---|
536 | const MGeom &gpix = (*fCam)[i];
|
---|
537 |
|
---|
538 | Int_t num = 0;
|
---|
539 | Double_t sum = 0;
|
---|
540 |
|
---|
541 | //
|
---|
542 | // Loop over all its neighbors
|
---|
543 | //
|
---|
544 | Int_t n = gpix.GetNumNeighbors();
|
---|
545 | while (n--)
|
---|
546 | {
|
---|
547 | const UShort_t nidx = gpix.GetNeighbor(n);
|
---|
548 |
|
---|
549 | //
|
---|
550 | // Do not use blind neighbors
|
---|
551 | //
|
---|
552 | if ((*fBad)[nidx].IsUnsuitable())
|
---|
553 | continue;
|
---|
554 |
|
---|
555 | sum += h.GetBinContent(nidx+1);
|
---|
556 | num++;
|
---|
557 | }
|
---|
558 |
|
---|
559 | h.SetBinContent(i+1, sum/num);
|
---|
560 | }
|
---|
561 | }
|
---|
562 |
|
---|
563 | // --------------------------------------------------------------------------
|
---|
564 | //
|
---|
565 | // Do a simple interpolation of the surrounding suitable pixels for all
|
---|
566 | // unsuitable pixels.
|
---|
567 | //
|
---|
568 | void MMovieWrite::Clean(TH1 &h, Double_t rms) const
|
---|
569 | {
|
---|
570 | if (fThreshold<1)
|
---|
571 | return;
|
---|
572 |
|
---|
573 | const UShort_t entries = fCam->GetNumPixels();
|
---|
574 |
|
---|
575 | //
|
---|
576 | // Loop over all pixels
|
---|
577 | //
|
---|
578 | for (UShort_t i=0; i<entries; i++)
|
---|
579 | {
|
---|
580 | Double_t val = h.GetBinContent(i+1);
|
---|
581 | if (val<rms || val>fThreshold*rms)
|
---|
582 | continue;
|
---|
583 |
|
---|
584 | const MGeom &gpix = (*fCam)[i];
|
---|
585 |
|
---|
586 | //
|
---|
587 | // Loop over all its neighbors
|
---|
588 | //
|
---|
589 | Int_t n = gpix.GetNumNeighbors();
|
---|
590 | while (n)
|
---|
591 | {
|
---|
592 | const UShort_t nidx = gpix.GetNeighbor(n-1);
|
---|
593 | if (h.GetBinContent(nidx+1)>=rms)
|
---|
594 | break;
|
---|
595 | n--;
|
---|
596 | }
|
---|
597 |
|
---|
598 | if (n==0)
|
---|
599 | h.SetBinContent(i+1, 0);
|
---|
600 | }
|
---|
601 | }
|
---|
602 |
|
---|
603 | // --------------------------------------------------------------------------
|
---|
604 | //
|
---|
605 | Bool_t MMovieWrite::Process(TH1 &h, TVirtualPad &c)
|
---|
606 | {
|
---|
607 | // ---------------- Setup ------------------
|
---|
608 |
|
---|
609 | const Float_t freq = fRun->GetFreqSampling()/1000.; // [GHz] Sampling frequency
|
---|
610 | const UInt_t slices = fIn->GetNumSlices();
|
---|
611 | const Float_t len = slices/freq; // [ns] length of data stream in data-time
|
---|
612 | //const Float_t len = (slices-2)/freq; // [ns] length of data stream in data-time
|
---|
613 |
|
---|
614 | const Double_t rms = fIn->GetMedianPedestalRms();
|
---|
615 | const Double_t max = fIn->GetMax(); // scale the lover limit such
|
---|
616 | const Double_t dif = (max-rms)*99/98; // that everything below rms is
|
---|
617 | const Double_t min = max-dif; // displayed as white
|
---|
618 |
|
---|
619 | // If the maximum is equal or less the
|
---|
620 | // pedestal rms something must be wrong
|
---|
621 | if (dif<=0)
|
---|
622 | return kFALSE;
|
---|
623 |
|
---|
624 | h.SetMinimum(min);
|
---|
625 | h.SetMaximum(max);
|
---|
626 |
|
---|
627 | // -----------------------------------------
|
---|
628 |
|
---|
629 | // Produce starting image from canvas
|
---|
630 | TASImage img;
|
---|
631 | img.FromPad(&c);
|
---|
632 |
|
---|
633 | // Set new adapted palette for further rendering
|
---|
634 | const Int_t colidx = SetPalette(rms, h);
|
---|
635 |
|
---|
636 | // Get the pad containing the camera with the movie
|
---|
637 | TVirtualPad &pad = *c.GetPad(1)->GetPad(1);
|
---|
638 |
|
---|
639 | // Calculate number of frames
|
---|
640 | const Int_t numframes = TMath::Nint(fTargetLength*24);
|
---|
641 |
|
---|
642 | // Get number of pixels in camera
|
---|
643 | const Int_t npix = fCam->GetNumPixels();
|
---|
644 |
|
---|
645 | // Loop over all frames+1 (upper edge)
|
---|
646 | for (Int_t i=0; i<=numframes; i++)
|
---|
647 | {
|
---|
648 | // Calculate corresponding time
|
---|
649 | const Float_t t = len*i/numframes;// + 0.5/freq; // Process from slice beg+0.5 to end-1.5
|
---|
650 |
|
---|
651 | // Calculate histogram contents by spline interpolation
|
---|
652 | for (UShort_t p=0; p<npix; p++)
|
---|
653 | {
|
---|
654 | const Double_t y = (*fBad)[p].IsUnsuitable() ? 0 : fIn->CheckedEval(p, t);
|
---|
655 | h.SetBinContent(p+1, y);
|
---|
656 | }
|
---|
657 |
|
---|
658 | // Interpolate unsuitable pixels
|
---|
659 | TreatBadPixels(h);
|
---|
660 |
|
---|
661 | // Clean single pixels
|
---|
662 | Clean(h, rms);
|
---|
663 |
|
---|
664 | // Set new name to be displayed
|
---|
665 | h.SetName(MString::Format("%d: %.2f/%.1fns", i+1, t*freq, t));
|
---|
666 |
|
---|
667 | // Update existing image with new data and encode into pipe
|
---|
668 | if (!WriteImage(img, pad))
|
---|
669 | return kFALSE;
|
---|
670 | }
|
---|
671 |
|
---|
672 | DeletePalette(colidx);
|
---|
673 |
|
---|
674 | cout << setw(3) << GetNumExecutions() << ": " << MString::Format("%6.2f", (float)numframes/(slices-2)) << " f/sl " << slices << " " << numframes+1 << endl;
|
---|
675 |
|
---|
676 | return kTRUE;
|
---|
677 | }
|
---|
678 |
|
---|
679 | // --------------------------------------------------------------------------
|
---|
680 | //
|
---|
681 | Int_t MMovieWrite::Process()
|
---|
682 | {
|
---|
683 | #ifdef USE_TIMING
|
---|
684 | clockT.Start(kFALSE);
|
---|
685 |
|
---|
686 | clock3.Start(kFALSE);
|
---|
687 | #endif
|
---|
688 | // ---------------- Prepare display ------------------
|
---|
689 |
|
---|
690 | Bool_t batch = gROOT->IsBatch();
|
---|
691 | gROOT->SetBatch();
|
---|
692 |
|
---|
693 | TCanvas c;
|
---|
694 | //c.Iconify();
|
---|
695 | c.SetBorderMode(0);
|
---|
696 | c.SetFrameBorderMode(0);
|
---|
697 | c.SetFillColor(kWhite);
|
---|
698 | //c.SetCanvasSize(640, 400);
|
---|
699 | c.SetCanvasSize(720, 480);
|
---|
700 | //c.SetCanvasSize(960, 640);
|
---|
701 | //c.SetCanvasSize(1024, 688);
|
---|
702 | //c.SetCanvasSize(1152, 768);
|
---|
703 | //c.SetCanvasSize(1280, 864);
|
---|
704 |
|
---|
705 | MH::SetPalette("pretty");
|
---|
706 |
|
---|
707 | c.cd();
|
---|
708 | TPad p1("Pad2", "", 0.7, 0.66, 0.99, 0.99);
|
---|
709 | p1.SetNumber(2);
|
---|
710 | p1.SetBorderMode(0);
|
---|
711 | p1.SetFrameBorderMode(0);
|
---|
712 | p1.SetFillColor(kWhite);
|
---|
713 | p1.Draw();
|
---|
714 | p1.cd();
|
---|
715 |
|
---|
716 | MHCamera hsig(*fCam, "Signal", "Calibrated Signal");
|
---|
717 | hsig.SetYTitle("S [phe]");
|
---|
718 | hsig.SetCamContent(*fSig, 99);
|
---|
719 | hsig.SetMinimum(0);
|
---|
720 | //hsig.SetContour(99);
|
---|
721 | hsig.Draw("nopal");
|
---|
722 |
|
---|
723 | c.cd();
|
---|
724 | TPad p2("Pad3", "", 0.7, 0.33, 0.99, 0.65);
|
---|
725 | p2.SetNumber(3);
|
---|
726 | p2.SetBorderMode(0);
|
---|
727 | p2.SetFrameBorderMode(0);
|
---|
728 | p2.SetFillColor(kWhite);
|
---|
729 | p2.Draw();
|
---|
730 | p2.cd();
|
---|
731 |
|
---|
732 | MHCamera htime(*fCam, "ArrivalTime", "Calibrated Arrival Time");
|
---|
733 | htime.SetYTitle("T [au]");
|
---|
734 | htime.SetCamContent(*fSig, 8);
|
---|
735 | htime.Draw("nopal");
|
---|
736 |
|
---|
737 | c.cd();
|
---|
738 | TPad p3("Pad4", "", 0.7, 0.00, 0.99, 0.32);
|
---|
739 | p3.SetNumber(4);
|
---|
740 | p3.SetBorderMode(0);
|
---|
741 | p3.SetFrameBorderMode(0);
|
---|
742 | p3.SetFillColor(kWhite);
|
---|
743 | p3.Draw();
|
---|
744 | p3.cd();
|
---|
745 | /*
|
---|
746 | TH1F htpro("TimeProj", "", slices, 0, len);
|
---|
747 | for (UInt_t i=0; i<htime.GetNumPixels(); i++)
|
---|
748 | if(htime.IsUsed(i))
|
---|
749 | htpro.Fill((htime.GetBinContent(i+1)-first)/freq, hsig.GetBinContent(i+1));
|
---|
750 | htpro.SetMinimum(0);
|
---|
751 | htpro.SetMaximum(100);
|
---|
752 | htpro.SetLineColor(kBlue);
|
---|
753 | htpro.Draw();
|
---|
754 |
|
---|
755 | TF1 fgaus("f1", "gaus");
|
---|
756 | const Double_t m = (htpro.GetMaximumBin()-0.5)*len/slices;
|
---|
757 | fgaus.SetParameter(0, htpro.GetMaximum());
|
---|
758 | fgaus.SetParameter(1, m);
|
---|
759 | fgaus.SetParameter(2, 1.0);
|
---|
760 | fgaus.SetParLimits(1, m-3, m+3);
|
---|
761 | fgaus.SetParLimits(2, 0, 3);
|
---|
762 | fgaus.SetLineWidth(1);
|
---|
763 | fgaus.SetLineColor(kMagenta);
|
---|
764 | htpro.Fit(&fgaus, "NI", "", m-3, m+3);
|
---|
765 | fgaus.Draw("same");
|
---|
766 |
|
---|
767 | g.SetMarkerStyle(kFullDotMedium);
|
---|
768 | g.Draw("PL");
|
---|
769 | //g.SetMinimum(0);
|
---|
770 | //g.SetMaximum(100);
|
---|
771 | //g.Draw("APL");
|
---|
772 |
|
---|
773 | p3.Update();
|
---|
774 | p3.cd(1);
|
---|
775 | gPad->Update();
|
---|
776 | */
|
---|
777 | c.cd();
|
---|
778 | TPad p0("MainPad", "", 0.01, 0.01, 0.69, 0.99);
|
---|
779 | p0.SetNumber(1);
|
---|
780 | p0.SetBorderMode(0);
|
---|
781 | p0.SetFrameBorderMode(0);
|
---|
782 | p0.SetFillColor(kWhite);
|
---|
783 | p0.Draw();
|
---|
784 | p0.cd();
|
---|
785 | /*
|
---|
786 | cout << "Max=" << hsig.GetMaximum() << "/" << fIn->GetMax() << " ";
|
---|
787 | cout << hsig.GetMaximum()/fIn->GetMax() << endl;
|
---|
788 | Float_t rms0 = fPed->GetAveragedRmsPerArea(*fCam, 0, fBad)[0];
|
---|
789 | Float_t rms1 = fPed->GetAveragedRmsPerArea(*fCam, 1, fBad)[0];
|
---|
790 | cout << "RMS="<<rms0<<"/"<<rms1<<endl;
|
---|
791 |
|
---|
792 | rms0 = GetMedianPedestalRms();
|
---|
793 |
|
---|
794 | cout << "MED=" << rms0 << endl;
|
---|
795 | */
|
---|
796 |
|
---|
797 | TString s = MString::Format("%d: Evt #", GetNumExecutions()+1);
|
---|
798 | s += fHead->GetDAQEvtNumber();
|
---|
799 | s += " of ";
|
---|
800 | s += "Run #";
|
---|
801 | s += fRun->GetRunNumber();
|
---|
802 | if (fMC)
|
---|
803 | s = fMC->GetDescription(s);
|
---|
804 |
|
---|
805 | MHCamera h(*fCam);
|
---|
806 | h.SetTitle(s);
|
---|
807 | h.SetAllUsed();
|
---|
808 | h.SetYTitle("V [au]");
|
---|
809 | h.SetContour(99);
|
---|
810 |
|
---|
811 | h.Draw("nopal");
|
---|
812 |
|
---|
813 | // ---------------- Show data ------------------
|
---|
814 | gStyle->SetOptStat(1000000001);
|
---|
815 | /*
|
---|
816 | p0.Modified();
|
---|
817 | p1.Modified();
|
---|
818 | p2.Modified();
|
---|
819 |
|
---|
820 | p0.GetPad(1)->Modified();
|
---|
821 | p1.GetPad(1)->Modified();
|
---|
822 | p2.GetPad(1)->Modified();
|
---|
823 |
|
---|
824 | c.Update();
|
---|
825 | */
|
---|
826 |
|
---|
827 | // ---------------- Show data ------------------
|
---|
828 | #ifdef USE_TIMING
|
---|
829 | clock3.Stop();
|
---|
830 | #endif
|
---|
831 |
|
---|
832 | // Switch off automatical adding to directory (SetName would do)
|
---|
833 | const Bool_t add = TH1::AddDirectoryStatus();
|
---|
834 | TH1::AddDirectory(kFALSE);
|
---|
835 |
|
---|
836 | const Bool_t rc = Process(h, c);
|
---|
837 |
|
---|
838 | // restore previous state
|
---|
839 | TH1::AddDirectory(add);
|
---|
840 |
|
---|
841 | #ifdef USE_TIMING
|
---|
842 | clockT.Stop();
|
---|
843 | #endif
|
---|
844 |
|
---|
845 | gROOT->SetBatch(batch);
|
---|
846 |
|
---|
847 | if (!rc)
|
---|
848 | return kERROR;
|
---|
849 |
|
---|
850 | return fNumEvents<=0 || GetNumExecutions()<fNumEvents;
|
---|
851 | }
|
---|
852 |
|
---|
853 | // --------------------------------------------------------------------------
|
---|
854 | //
|
---|
855 | // Check for corresponding entries in resource file and setup
|
---|
856 | //
|
---|
857 | // Example:
|
---|
858 | // MMovieWrite.TargetLength: 5 <seconds>
|
---|
859 | // MMovieWrite.NumEvents: 500
|
---|
860 | // MMovieWrite.Threshold: 2 <rms>
|
---|
861 | // MMovieWrite.FileName: movie.mpg
|
---|
862 | //
|
---|
863 | Int_t MMovieWrite::ReadEnv(const TEnv &env, TString prefix, Bool_t print)
|
---|
864 | {
|
---|
865 | Bool_t rc = kFALSE;
|
---|
866 | if (IsEnvDefined(env, prefix, "NumEvents", print))
|
---|
867 | {
|
---|
868 | fNumEvents = GetEnvValue(env, prefix, "NumEvents", (Int_t)fNumEvents);
|
---|
869 | rc = kTRUE;
|
---|
870 | }
|
---|
871 | if (IsEnvDefined(env, prefix, "TargetLength", print))
|
---|
872 | {
|
---|
873 | fTargetLength = GetEnvValue(env, prefix, "TargetLength", fTargetLength);
|
---|
874 | rc = kTRUE;
|
---|
875 | }
|
---|
876 | if (IsEnvDefined(env, prefix, "Threshold", print))
|
---|
877 | {
|
---|
878 | fThreshold = GetEnvValue(env, prefix, "Threshold", fThreshold);
|
---|
879 | rc = kTRUE;
|
---|
880 | }
|
---|
881 | if (IsEnvDefined(env, prefix, "FileName", print))
|
---|
882 | {
|
---|
883 | fFilename = GetEnvValue(env, prefix, "FileName", fFilename);
|
---|
884 | rc = kTRUE;
|
---|
885 | }
|
---|
886 | return rc;
|
---|
887 | }
|
---|