1 | #include "MGTPoint.h"
|
---|
2 |
|
---|
3 | #include <fstream.h> // ifstream
|
---|
4 | #include <iostream.h> // cout
|
---|
5 | #
|
---|
6 | #include <TGMenu.h>
|
---|
7 | #include <TSystem.h>
|
---|
8 | #include <TGSplitter.h> // TGHorizontal3DLine
|
---|
9 | #include <TGTextEntry.h>
|
---|
10 |
|
---|
11 | #include "MGImage.h"
|
---|
12 | #include "MGCoordinates.h"
|
---|
13 |
|
---|
14 | #include "coord.h"
|
---|
15 |
|
---|
16 | #include "StarCatalog.h"
|
---|
17 |
|
---|
18 | #include "Filter.h"
|
---|
19 | #include "Writer.h"
|
---|
20 | #include "MTime.h"
|
---|
21 |
|
---|
22 | ClassImp(MGStarguider);
|
---|
23 |
|
---|
24 | enum {
|
---|
25 | IDM_kFilter,
|
---|
26 | IDM_kCatalog,
|
---|
27 | IDM_kStart,
|
---|
28 | IDM_kStop,
|
---|
29 | IDM_kFileType,
|
---|
30 | IDM_kPPM,
|
---|
31 | IDM_kPNG,
|
---|
32 | IDM_kOnce,
|
---|
33 | IDM_kUseFileRaDec,
|
---|
34 | IDM_kContinous,
|
---|
35 | IDM_kRate25ps,
|
---|
36 | IDM_kRate5ps,
|
---|
37 | IDM_kRate1s,
|
---|
38 | IDM_kRate5s,
|
---|
39 | IDM_kRate30s,
|
---|
40 | IDM_kRate1m,
|
---|
41 | IDM_kRate5m,
|
---|
42 | IDM_kSetup,
|
---|
43 | IDM_kLimMag3,
|
---|
44 | IDM_kLimMag4,
|
---|
45 | IDM_kLimMag5,
|
---|
46 | IDM_kLimMag6,
|
---|
47 | IDM_kLimMag7,
|
---|
48 | IDM_kLimMag8,
|
---|
49 | IDM_kLimMag9,
|
---|
50 | IDM_kPixSize
|
---|
51 | };
|
---|
52 |
|
---|
53 | MGStarguider::MGStarguider()
|
---|
54 | : Camera(), TGMainFrame(gClient->GetRoot(), 768, 700)
|
---|
55 | {
|
---|
56 | fList = new MGList;
|
---|
57 |
|
---|
58 | fSao = new StarCatalog;
|
---|
59 | fRaDec = new RaDec(180, 40);
|
---|
60 |
|
---|
61 | // p = pointer to MainFrame (not owner)
|
---|
62 |
|
---|
63 | const TGWindow *p=gClient->GetRoot();
|
---|
64 |
|
---|
65 | //
|
---|
66 | // Create Menu for MGStarguider Display
|
---|
67 | //
|
---|
68 | fDisplay = new TGPopupMenu(p);
|
---|
69 | fDisplay->AddEntry("&Filter", IDM_kFilter);
|
---|
70 | fDisplay->AddEntry("Sao &Catalog", IDM_kCatalog);
|
---|
71 | fDisplay->Associate(this);
|
---|
72 | fList->Add(fDisplay);
|
---|
73 |
|
---|
74 | fFileType = new TGPopupMenu(p);
|
---|
75 | fFileType->AddEntry("PP&M", IDM_kPPM);
|
---|
76 | fFileType->AddEntry("&PNG", IDM_kPNG);
|
---|
77 | fFileType->CheckEntry(IDM_kPNG);
|
---|
78 | fFileType->Associate(this);
|
---|
79 | fList->Add(fFileType);
|
---|
80 |
|
---|
81 | fWriteType = new TGPopupMenu(p);
|
---|
82 | fWriteType->AddEntry("Once", IDM_kOnce);
|
---|
83 | fWriteType->AddEntry("Continous", IDM_kContinous);
|
---|
84 | fWriteType->CheckEntry(IDM_kOnce);
|
---|
85 | fWriteType->Associate(this);
|
---|
86 | fList->Add(fWriteType);
|
---|
87 |
|
---|
88 | fWriteRate = new TGPopupMenu(p);
|
---|
89 | fWriteRate->AddEntry("25/s", IDM_kRate25ps);
|
---|
90 | fWriteRate->AddEntry("5/s", IDM_kRate5ps);
|
---|
91 | fWriteRate->AddEntry("1s", IDM_kRate1s);
|
---|
92 | fWriteRate->AddEntry("5s", IDM_kRate5s);
|
---|
93 | fWriteRate->AddEntry("30s", IDM_kRate30s);
|
---|
94 | fWriteRate->AddEntry("1min", IDM_kRate1m);
|
---|
95 | fWriteRate->AddEntry("5min", IDM_kRate5m);
|
---|
96 | fWriteRate->CheckEntry(IDM_kRate1m);
|
---|
97 | fWriteRate->Associate(this);
|
---|
98 | fList->Add(fWriteRate);
|
---|
99 |
|
---|
100 | fWrtRate = 25*60;
|
---|
101 |
|
---|
102 | fWrite = new TGPopupMenu(p);
|
---|
103 | fWrite->AddEntry("&Start", IDM_kStart);
|
---|
104 | fWrite->AddEntry("Sto&p", IDM_kStop);
|
---|
105 | fWrite->AddSeparator();
|
---|
106 | fWrite->AddPopup("File &Type", fFileType);
|
---|
107 | fWrite->AddPopup("&Write Type", fWriteType);
|
---|
108 | fWrite->AddPopup("Write &Rate", fWriteRate);
|
---|
109 | fWrite->DisableEntry(IDM_kStop);
|
---|
110 | fWrite->Associate(this);
|
---|
111 | fList->Add(fWrite);
|
---|
112 |
|
---|
113 | fLimMag = new TGPopupMenu(p);
|
---|
114 | fLimMag->AddEntry("3", IDM_kLimMag3);
|
---|
115 | fLimMag->AddEntry("4", IDM_kLimMag4);
|
---|
116 | fLimMag->AddEntry("5", IDM_kLimMag5);
|
---|
117 | fLimMag->AddEntry("6", IDM_kLimMag6);
|
---|
118 | fLimMag->AddEntry("7", IDM_kLimMag7);
|
---|
119 | fLimMag->AddEntry("8", IDM_kLimMag8);
|
---|
120 | fLimMag->AddEntry("9", IDM_kLimMag9);
|
---|
121 | fLimMag->CheckEntry(IDM_kLimMag8);
|
---|
122 | fLimMag->Associate(this);
|
---|
123 | fList->Add(fLimMag);
|
---|
124 |
|
---|
125 | fSao->SetLimitMag(8.0);
|
---|
126 |
|
---|
127 | fSetup = new TGPopupMenu(p);
|
---|
128 | fSetup->AddPopup("Lim. &Magnitude", fLimMag);
|
---|
129 | fSetup->AddEntry("Use Ra/Dec from file", IDM_kUseFileRaDec);
|
---|
130 | fSetup->Associate(this);
|
---|
131 | fList->Add(fSetup);
|
---|
132 |
|
---|
133 | fMenu = new TGMenuBar(this, 0, 0, kHorizontalFrame);
|
---|
134 | fMenu->AddPopup("&Display", fDisplay, NULL);
|
---|
135 | fMenu->AddPopup("&Write", fWrite, NULL);
|
---|
136 | fMenu->AddPopup("&Setup", fSetup, NULL);
|
---|
137 | fMenu->Resize(fMenu->GetDefaultSize());
|
---|
138 | AddFrame(fMenu); //, new TGLayoutHints (kLHintsNormal, 0, 4, 0, 0));
|
---|
139 | fList->Add(fMenu);
|
---|
140 |
|
---|
141 | fCRaDec = new MGCoordinates(this, kETypeRaDec);
|
---|
142 | fCRaDec->Move(1, fMenu->GetDefaultHeight()+584);
|
---|
143 | AddFrame(fCRaDec);
|
---|
144 | fList->Add(fCRaDec);
|
---|
145 |
|
---|
146 | fCZdAz = new MGCoordinates(this, kETypeZdAz);
|
---|
147 | fCZdAz->Move(240+12, fMenu->GetDefaultHeight()+584);
|
---|
148 | AddFrame(fCZdAz);
|
---|
149 | fList->Add(fCZdAz);
|
---|
150 |
|
---|
151 | const Double_t pixsize = 23.4;
|
---|
152 |
|
---|
153 | fSao->SetPixSize(pixsize);
|
---|
154 |
|
---|
155 | TString txt;
|
---|
156 | txt += pixsize;
|
---|
157 |
|
---|
158 | fPixSize = new TGTextEntry(this, txt, IDM_kPixSize);
|
---|
159 | fPixSize->SetAlignment(kTextCenterX);
|
---|
160 | fPixSize->Move(600, fMenu->GetDefaultHeight()+584);
|
---|
161 | AddFrame(fPixSize);
|
---|
162 | fList->Add(fPixSize);
|
---|
163 |
|
---|
164 | // TGHorizontal3DLine *fLineSep = new TGHorizontal3DLine(this);
|
---|
165 | // AddFrame(fLineSep, new TGLayoutHints (kLHintsNormal | kLHintsExpandX));
|
---|
166 | // fList->Add(fLineSep);
|
---|
167 |
|
---|
168 | //
|
---|
169 | // Create Image Display
|
---|
170 | //
|
---|
171 | fImage = new MGImage(this, 768, 576);
|
---|
172 | fImage->Move(0, fMenu->GetDefaultHeight());
|
---|
173 | AddFrame(fImage);
|
---|
174 | fList->Add(fImage);
|
---|
175 |
|
---|
176 | //
|
---|
177 | // Make everything visible
|
---|
178 | //
|
---|
179 | SetWindowName("MGStarguider Main Window");
|
---|
180 | SetIconName("MGStarguider");
|
---|
181 |
|
---|
182 | MapSubwindows();
|
---|
183 | MapWindow();
|
---|
184 | }
|
---|
185 |
|
---|
186 | MGStarguider::~MGStarguider()
|
---|
187 | {
|
---|
188 | delete fSao;
|
---|
189 | delete fRaDec;
|
---|
190 |
|
---|
191 | delete fList;
|
---|
192 | cout << "Camera Display destroyed." << endl;
|
---|
193 | }
|
---|
194 |
|
---|
195 | void MGStarguider::Layout()
|
---|
196 | {
|
---|
197 | // Resize(GetDefaultSize());
|
---|
198 | }
|
---|
199 |
|
---|
200 | void MGStarguider::CloseWindow()
|
---|
201 | {
|
---|
202 | cout << "EventDisplay::CloseWindow: Exit Application Loop." << endl;
|
---|
203 |
|
---|
204 | ExitLoop();
|
---|
205 |
|
---|
206 | gSystem->ExitLoop();
|
---|
207 | }
|
---|
208 |
|
---|
209 | void MGStarguider::Toggle(TGPopupMenu *p, UInt_t id)
|
---|
210 | {
|
---|
211 | if (p->IsEntryChecked(id))
|
---|
212 | p->UnCheckEntry(id);
|
---|
213 | else
|
---|
214 | p->CheckEntry(id);
|
---|
215 |
|
---|
216 | }
|
---|
217 |
|
---|
218 | Bool_t MGStarguider::ProcessMessage(Long_t msg, Long_t mp1, Long_t mp2)
|
---|
219 | {
|
---|
220 | switch (GET_MSG(msg))
|
---|
221 | {
|
---|
222 | case kC_TEXTENTRY:
|
---|
223 | if (GET_SUBMSG(msg)==kTE_ENTER)
|
---|
224 | {
|
---|
225 | const Float_t pixsize = atof(fPixSize->GetText());
|
---|
226 | cout << "Pixel Size changed to " << pixsize << "\"/pix" << endl;
|
---|
227 | fSao->SetPixSize(pixsize);
|
---|
228 | }
|
---|
229 | return kTRUE;
|
---|
230 |
|
---|
231 | case kC_COMMAND:
|
---|
232 | switch (GET_SUBMSG(msg))
|
---|
233 | {
|
---|
234 | case kCM_MENU:
|
---|
235 | switch (mp1)
|
---|
236 | {
|
---|
237 | case IDM_kCatalog:
|
---|
238 | Toggle(fDisplay, IDM_kCatalog);
|
---|
239 | return kTRUE;
|
---|
240 |
|
---|
241 | case IDM_kFilter:
|
---|
242 | Toggle(fDisplay, IDM_kFilter);
|
---|
243 | return kTRUE;
|
---|
244 |
|
---|
245 | case IDM_kUseFileRaDec:
|
---|
246 | Toggle(fSetup, IDM_kUseFileRaDec);
|
---|
247 | return kTRUE;
|
---|
248 |
|
---|
249 | case IDM_kStart:
|
---|
250 | fWrite->DisableEntry(IDM_kStart);
|
---|
251 | fWrite->EnableEntry(IDM_kStop);
|
---|
252 | return kTRUE;
|
---|
253 |
|
---|
254 | case IDM_kStop:
|
---|
255 | fWrite->DisableEntry(IDM_kStop);
|
---|
256 | fWrite->EnableEntry(IDM_kStart);
|
---|
257 | return kTRUE;
|
---|
258 |
|
---|
259 | case IDM_kPNG:
|
---|
260 | fFileType->CheckEntry(IDM_kPNG);
|
---|
261 | fFileType->UnCheckEntry(IDM_kPPM);
|
---|
262 | return kTRUE;
|
---|
263 |
|
---|
264 | case IDM_kPPM:
|
---|
265 | fFileType->CheckEntry(IDM_kPPM);
|
---|
266 | fFileType->UnCheckEntry(IDM_kPNG);
|
---|
267 | return kTRUE;
|
---|
268 |
|
---|
269 | case IDM_kOnce:
|
---|
270 | fWriteType->CheckEntry(IDM_kOnce);
|
---|
271 | fWriteType->UnCheckEntry(IDM_kContinous);
|
---|
272 | return kTRUE;
|
---|
273 |
|
---|
274 | case IDM_kContinous:
|
---|
275 | fWriteType->CheckEntry(IDM_kContinous);
|
---|
276 | fWriteType->UnCheckEntry(IDM_kOnce);
|
---|
277 | return kTRUE;
|
---|
278 |
|
---|
279 | case IDM_kRate25ps:
|
---|
280 | case IDM_kRate5ps:
|
---|
281 | case IDM_kRate1s:
|
---|
282 | case IDM_kRate5s:
|
---|
283 | case IDM_kRate30s:
|
---|
284 | case IDM_kRate1m:
|
---|
285 | case IDM_kRate5m:
|
---|
286 | for (int i=IDM_kRate25ps; i<=IDM_kRate5m; i++)
|
---|
287 | if (mp1==i)
|
---|
288 | fWriteRate->CheckEntry(i);
|
---|
289 | else
|
---|
290 | fWriteRate->UnCheckEntry(i);
|
---|
291 | switch (mp1)
|
---|
292 | {
|
---|
293 | case IDM_kRate25ps:
|
---|
294 | fWrtRate = 1;
|
---|
295 | return kTRUE;
|
---|
296 | case IDM_kRate5ps:
|
---|
297 | fWrtRate = 5;
|
---|
298 | return kTRUE;
|
---|
299 | case IDM_kRate1s:
|
---|
300 | fWrtRate = 25;
|
---|
301 | return kTRUE;
|
---|
302 | case IDM_kRate5s:
|
---|
303 | fWrtRate = 5*25;
|
---|
304 | return kTRUE;
|
---|
305 | case IDM_kRate30s:
|
---|
306 | fWrtRate = 30*25;
|
---|
307 | return kTRUE;
|
---|
308 | case IDM_kRate1m:
|
---|
309 | fWrtRate = 60*25;
|
---|
310 | return kTRUE;
|
---|
311 | case IDM_kRate5m:
|
---|
312 | fWrtRate = 5*60*25;
|
---|
313 | return kTRUE;
|
---|
314 | }
|
---|
315 | return kTRUE;
|
---|
316 |
|
---|
317 | case IDM_kLimMag3:
|
---|
318 | case IDM_kLimMag4:
|
---|
319 | case IDM_kLimMag5:
|
---|
320 | case IDM_kLimMag6:
|
---|
321 | case IDM_kLimMag7:
|
---|
322 | case IDM_kLimMag8:
|
---|
323 | case IDM_kLimMag9:
|
---|
324 | for (int i=IDM_kLimMag3; i<=IDM_kLimMag9; i++)
|
---|
325 | if (mp1==i)
|
---|
326 | fLimMag->CheckEntry(i);
|
---|
327 | else
|
---|
328 | fLimMag->UnCheckEntry(i);
|
---|
329 |
|
---|
330 | fSao->SetLimitMag(mp1-IDM_kLimMag3+3);
|
---|
331 | return kTRUE;
|
---|
332 | }
|
---|
333 | break;
|
---|
334 | }
|
---|
335 | break;
|
---|
336 | }
|
---|
337 | return kTRUE;
|
---|
338 | }
|
---|
339 |
|
---|
340 | void MGStarguider::GetCoordinates()
|
---|
341 | {
|
---|
342 | XY xy = fCRaDec->GetCoordinates();
|
---|
343 |
|
---|
344 | if (fSetup->IsEntryChecked(IDM_kUseFileRaDec))
|
---|
345 | {
|
---|
346 | ifstream fin("coordinates.txt");
|
---|
347 | if (!fin)
|
---|
348 | cout << "Error: Cannot open 'coordinates.txt' using fall back solution." << endl;
|
---|
349 | else
|
---|
350 | fin >> xy;
|
---|
351 | }
|
---|
352 |
|
---|
353 | fCRaDec->SetCoordinates(xy);
|
---|
354 | fRaDec->Set(xy.X()*360/24, xy.Y());
|
---|
355 | }
|
---|
356 |
|
---|
357 | void MGStarguider::ProcessFrame(const unsigned long n, byte *img, struct timeval *tm)
|
---|
358 | {
|
---|
359 |
|
---|
360 | if (!fWrite->IsEntryEnabled(IDM_kStart) &&
|
---|
361 | (!(n%fWrtRate) || fWriteType->IsEntryChecked(IDM_kOnce)))
|
---|
362 | {
|
---|
363 | if (fFileType->IsEntryChecked(IDM_kPNG))
|
---|
364 | {
|
---|
365 | static int num = 0;
|
---|
366 |
|
---|
367 | char name[80];
|
---|
368 | sprintf(name, "pix/file%04d.png", num);
|
---|
369 | Writer::Png(name, img, tm);
|
---|
370 | }
|
---|
371 |
|
---|
372 | if (fFileType->IsEntryChecked(IDM_kPPM))
|
---|
373 | {
|
---|
374 | static int num = 0;
|
---|
375 | char name[80];
|
---|
376 | sprintf(name, "pix/file%04d.ppm", num);
|
---|
377 | Writer::Ppm(name, img);
|
---|
378 | }
|
---|
379 |
|
---|
380 | if (fWriteType->IsEntryChecked(IDM_kOnce))
|
---|
381 | ProcessMessage(MK_MSG(kC_COMMAND, kCM_MENU), IDM_kStop, 0);
|
---|
382 | }
|
---|
383 |
|
---|
384 | static unsigned char myimg[768*576];
|
---|
385 |
|
---|
386 | if (!(n%25))
|
---|
387 | {
|
---|
388 | cout << "Img: " << n << endl;
|
---|
389 |
|
---|
390 | if (fDisplay->IsEntryChecked(IDM_kFilter))
|
---|
391 | Filter::Execute(myimg);
|
---|
392 |
|
---|
393 | if (fDisplay->IsEntryChecked(IDM_kCatalog))
|
---|
394 | {
|
---|
395 | byte cimg[768*576];
|
---|
396 |
|
---|
397 | GetCoordinates();
|
---|
398 |
|
---|
399 | MTime time(*tm);
|
---|
400 | fSao->GetImg(myimg, cimg, time.CalcMjd(), *fRaDec);
|
---|
401 | fImage->DrawColImg(myimg, cimg);
|
---|
402 |
|
---|
403 | fCZdAz->SetCoordinates(fSao->GetZdAz());
|
---|
404 | }
|
---|
405 | else
|
---|
406 | fImage->DrawImg(myimg);
|
---|
407 |
|
---|
408 | memset(myimg, 0, 768*576);
|
---|
409 | }
|
---|
410 | else
|
---|
411 | {
|
---|
412 | for (int i=0; i<768*576; i++)
|
---|
413 | myimg[i] += img[i]/25;
|
---|
414 |
|
---|
415 | }
|
---|
416 |
|
---|
417 | }
|
---|