/[PAMELA software]/quicklook/SatelliteInclination/src/SatInc.cpp
ViewVC logotype

Contents of /quicklook/SatelliteInclination/src/SatInc.cpp

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.1 - (show annotations) (download)
Thu Feb 8 00:49:36 2007 UTC (17 years, 9 months ago) by cafagna
Branch: MAIN
Branch point for: first
Initial revision

1 /**
2 * SaaaatInc
3 * author Malakhov V.
4 * version 1.0 - 05/02/2007
5 */
6
7 #include "InclinationInfo.h"
8
9 #include <mcmd/McmdEvent.h>
10 #include <mcmd/McmdRecord.h>
11 #include <EventHeader.h>
12 #include <PscuHeader.h>
13 #include "RunHeaderEvent.h"
14 #include <TTree.h>
15 #include "sgp4.h"
16 #include "TH2F.h"
17
18 #include "TFrame.h"
19 #include "TGraph.h"
20 #include "TCanvas.h"
21
22 #include <TDatime.h>
23 #include <TFile.h>
24
25 #include <TTimeStamp.h>
26 #include "TString.h"
27 #include "TObjString.h"
28 #include "TPaletteAxis.h"
29 #include "TROOT.h"
30 #include <sys/stat.h>
31 #include <fstream>
32 #include <iostream>
33
34 #include "OrbitalRate.h"
35
36 //#include "TMatrixD.h"
37
38
39 using namespace std;
40
41 int main(int argc, char* argv[]){
42
43 TString *rootFile = NULL;
44 TString outDir = "./";
45 TString mapFile = "";
46 TString tleFile = "";
47 int offDate = 20060928;
48 //int offDate = 20060614;
49 int offTime = 210000;
50
51 if (argc < 2){
52 printf("You have to insert at least the file to analyze and the mapFile \n");
53 printf("Try '--help' for more information. \n");
54 exit(1);
55 }
56
57 if (!strcmp(argv[1], "--help")){
58 //printf( "Usage: OrbitRate FILE -map mapFile [OPTION] \n");
59 //printf( "mapFile have to be a mercator map image [gif|jpg|png] \n");
60 printf( "\t --help Print this help and exit \n");
61 printf( "\t -tle[File path] Path where to find the tle infos \n");
62 printf( "\t\tUse the script retrieve_TLE.sh to create the file.\n ");
63 printf( "\t -outDir[path] Path where to put the output without last directory.\n");
64 printf( "\t -offDate Date of resetting of the Resource counter [format YYMMDD (UTC date) default 20060928] \n");
65 printf( "\t -offTime Time of resetting of the Resource counter [format HHMMSS (UTC date) default 210000] \n");
66 exit(1);
67 }
68
69 // Ok, here we should have at least one root file. We check that
70 // the filename contains ".root".
71 if(strstr(argv[1], ".root"))
72 rootFile = new TString(argv[1]);
73 else {
74 cerr << "OrbitalRate: no root file." << endl << "See --help" << endl;
75 exit(EXIT_FAILURE);
76 }
77
78 for (int i = 2; i < argc; i++){
79 if (!strcmp(argv[i], "-outDir")){
80 if (++i >= argc){
81 printf( "-outDir needs arguments. \n");
82 printf( "Try '--help' for more information. \n");
83 exit(1);
84 } else {
85 outDir = argv[i];
86 continue;
87 }
88 }
89
90 if (!strcmp(argv[i], "-tle")){
91 if (++i >= argc){
92 printf( "-tle needs arguments. \n");
93 printf( "Try '--help' for more information. \n");
94 exit(1);
95 } else {
96 tleFile = argv[i];
97 continue;
98 }
99 }
100
101 if (!strcmp(argv[i], "-offTime")){
102 if (++i >= argc){
103 printf( "-offTime needs arguments. \n");
104 printf( "Try '--help' for more information. \n");
105 exit(1);
106 }
107 else{
108 offTime = atol(argv[i]);
109 continue;
110 }
111 }
112
113 if (!strcmp(argv[i], "-offDate")){
114 if (++i >= argc){
115 printf( "-offDate needs arguments. \n");
116 printf( "Try '--help' for more information. \n");
117 exit(1);
118 }
119 else{
120 offDate = atol(argv[i]);
121 continue;
122 }
123 }
124
125 }
126
127 Rate(rootFile, outDir, mapFile, tleFile, offDate, offTime);
128
129 }
130
131
132 void Rate(TString *filename, TString outDirectory = "", TString mapFile = "", TString tleFile = "", int offDate = 20060614, int offTime = 210000)
133 {
134 // **** Offset to temporarily correct the TDatime bug ****/
135 // offTime += 10000;
136 //********************************************************/
137
138 McmdScan *mcmdReader = new McmdScan();
139
140 pamela::McmdEvent *mcmdev = 0;
141 pamela::McmdRecord *mcmdrc = 0;
142 pamela::EventHeader *eh = 0;
143 pamela::PscuHeader *ph = 0;
144 TArrayC *mcmddata;
145
146 pamela::RunHeaderEvent *reh = new pamela::RunHeaderEvent;
147 pamela::EventHeader *eH = new pamela::EventHeader;
148
149 Quaternions *L_QQ_Q_l = new Quaternions();
150 InclinationInfo *Plux = new InclinationInfo();
151
152 float timesync = 0, obt_timesync = 0;
153
154 ULong64_t nevents = 0;
155 stringstream oss;
156
157 Long64_t offsetTime = 0;
158 Long64_t timeElapsedFromTLE = 0;
159 Long64_t deltaTime = 0, oldtimeElapsedFromTLE = 0;
160 bool a_second_is_over;
161
162 // Get a char* to "file" from "/dir1/dir2/.../file.root"
163 TString basename;
164 basename = ((TObjString*) filename->Tokenize('/')->Last())->GetString(); // we get file.root
165 basename = ((TObjString*)basename.Tokenize('.')->First())->GetString(); // we get file
166
167 TFile *rootFile = new TFile(*filename);
168
169 if (rootFile->IsZombie()) printf("The %s file does not exist",basename.Data());
170 TString fileName = ((TObjString*)basename.Tokenize('/')->Last())->GetString();
171 TString filePath = basename;
172 filePath.ReplaceAll(fileName, "");
173 filePath.ReplaceAll(".root", "");
174
175 TTree *tr = (TTree*)rootFile->Get("Mcmd");
176 nevents = tr->GetEntries();
177 tr->SetBranchAddress("Mcmd",&mcmdev);
178 tr->SetBranchAddress("Header",&eh);
179
180 ULong_t firstime = 99999999;//ph->GetOrbitalTime();
181 UInt_t firstID = 65535;
182 ULong_t lastime = 0;//ph->GetOrbitalTime();
183 UInt_t lastID = 0;
184 for(int ii = 0; ii < nevents; ii++){
185 tr->GetEntry(ii);
186 ph = eh->GetPscuHeader();
187 Int_t tmpSize = mcmdev->Records->GetEntries();
188 for (int qq=0; qq < tmpSize; qq++){
189 mcmdrc = (pamela::McmdRecord*)mcmdev->Records->At(qq);
190 if (mcmdrc->SeqID <= firstID) firstID = mcmdrc->SeqID;
191 if (mcmdrc->SeqID >= lastID) lastID = mcmdrc->SeqID;
192 if ((int)mcmdrc->ID1 == 226){
193 L_QQ_Q_l->fill(mcmdrc->McmdData);
194 if (L_QQ_Q_l->time[0] >= lastime) lastime = L_QQ_Q_l->time[0];
195 if (L_QQ_Q_l->time[0] <= firstime) firstime = L_QQ_Q_l->time[0];
196 }
197 };
198 }
199
200 TTree *tp = (TTree*)rootFile->Get("RunHeader");
201 tp->SetBranchAddress("Header", &eH);
202 tp->SetBranchAddress("RunHeader", &reh);
203 tp->GetEntry(0);
204 ph = eH->GetPscuHeader();
205 ULong_t TimeSync = reh->LAST_TIME_SYNC_INFO;
206
207 //Get Orientation information
208
209 TH2F *_L1 = new TH2F("_L1",basename+";OnBoard Time, s; Value of l0",1000,firstime,lastime,1000,-1,1);
210 TH2F *_L2 = new TH2F("_L2",basename+";OnBoard Time, s; Value of l1",1000,firstime,lastime,1000,-1,1);
211 TH2F *_L3 = new TH2F("_L3",basename+";OnBoard Time, s; Value of l2",1000,firstime,lastime,1000,-1,1);
212 TH2F *_L4 = new TH2F("_L4",basename+";OnBoard Time, s; Value of l3",1000,firstime,lastime,1000,-1,1);
213
214 //TH2F *_A1 = new TH2F("_A1",basename,1000,firstime,lastime,1000,-180,180);
215 //TH2F *_A2 = new TH2F("_A2",basename,1000,firstime,lastime,1000,-180,180);
216 //TH2F *_A3 = new TH2F("_A3",basename,1000,firstime,lastime,1000,-180,180);
217
218 //TH2F *_secID = new TH2F("_secID",basename,1000,firstime,lastime,1000,firstID,lastID);
219
220 TH2F *_Tangazh = new TH2F("Pitch",basename+";OnBoard Time, s;Angel of pith, deg",1000,firstime,lastime,1000,-20,20);
221 TH2F *_Ryskanie = new TH2F("Yaw",basename+";OnBoard Time, s;Angel of yaw, deg",1000,firstime,lastime,1000,-100,-80);
222 TH2F *_Kren = new TH2F("Roll",basename+";OnBoard Time, s;Angel of roll, deg",1000,firstime,lastime,1000,-50,50);
223
224 // Open the root file.
225 rootFile = new TFile(filename->Data());
226 if (rootFile->IsZombie()) {
227 printf("The file %s does not exist\n", (filename->Data()));
228 exit(EXIT_FAILURE);
229 }
230
231 // Look for a timesync in the TFile rootFile. We also get the obt
232 // of the timesync mcmd.
233 bool err;
234 err = lookforTimesync(rootFile, &timesync, &obt_timesync);
235 if(!err) {
236 cerr << "Warning!!! No timesync info has been found in the file "
237 << filename->Data() << endl;
238 exit(EXIT_FAILURE);
239 }
240
241 //Get the Julian date of the Resours offset
242 TDatime offRes = TDatime(offDate, offTime);
243 // Add to the Resours Offset the timesync. This is now the date at
244 // the moment of the timesync.
245 ULong_t TH=offRes.Convert();
246 //cout<<"TH= "<<TH/86400<<"\n";
247 offRes.Set(offRes.Convert() + (UInt_t) timesync);
248
249 // Now I need a pointer to a cTle object. The class misses a
250 // constructor without arguments, so we have to give it a dummy TLE.
251 string str1 = "RESURS-DK 1";
252 string str2 = "1 29228U 06021A 06170.19643714 .00009962 00000-0 21000-3 0 196";
253 string str3 = "2 29228 069.9363 054.7893 0167576 127.4359 017.0674 15.31839265 604";
254 cTle *tle1 = new cTle(str1, str2, str3);
255
256 // If we have to use a TLE file, call getTle().
257 if (tleFile != "")
258 tle1 = getTle(tleFile, offRes);
259
260 cOrbit orbit(*tle1);
261 cEci eci;
262 cCoordGeo coo;
263
264 // Get the Julian date of the TLE epoch
265 string datetime = getTleDatetime(tle1);
266 TDatime tledate = TDatime(datetime.c_str());
267
268
269 tr = (TTree*)rootFile->Get("Mcmd");
270 nevents = tr->GetEntries();
271 tr->SetBranchAddress("Mcmd", &mcmdev);
272 tr->SetBranchAddress("Header", &eh);
273
274 // oss.str("");
275 // oss << basename+".txt";
276
277 // ofstream outputFile;
278 // outputFile.open(oss.str().c_str());
279
280 Int_t tmpSize;
281
282 for(UInt_t i = 0; i < nevents; i++) //Fill variables from root-ple
283 {
284 tr->GetEntry(i);
285 tmpSize = mcmdev->Records->GetEntries();
286 ph = eh->GetPscuHeader();
287
288 for (Int_t j3 = 0; j3 < tmpSize; j3++){
289 mcmdrc = (pamela::McmdRecord*)mcmdev->Records->At(j3);
290 if ((int)mcmdrc->ID1 == 226){
291 L_QQ_Q_l->fill(mcmdrc->McmdData);
292 Plux->QuaternionstoAngle(*L_QQ_Q_l);
293 for (Int_t oi = 0; oi < 6; oi++){
294 //if ((Int_t)L_QQ_Q_l->CodeErrQua[oi]==0){
295
296 _L1->Fill(L_QQ_Q_l->time[oi],L_QQ_Q_l->quat[oi][0]);
297 _L2->Fill(L_QQ_Q_l->time[oi],L_QQ_Q_l->quat[oi][1]);
298 _L3->Fill(L_QQ_Q_l->time[oi],L_QQ_Q_l->quat[oi][2]);
299 _L4->Fill(L_QQ_Q_l->time[oi],L_QQ_Q_l->quat[oi][3]);
300
301 //outputFile << " " << L_QQ_Q_l->time[oi] << " " << -L_QQ_Q_l->quat[oi][0] << " " << -L_QQ_Q_l->quat[oi][1] << " " << -L_QQ_Q_l->quat[oi][2] << " " << -L_QQ_Q_l->quat[oi][3];
302
303 //_A1->Fill(L_QQ_Q_l->time[oi],Plux->A11);
304 //_A2->Fill(L_QQ_Q_l->time[oi],Plux->A12);
305 //_A3->Fill(L_QQ_Q_l->time[oi],Plux->A13);
306
307 //outputFile << " " << Plux->A11 << " " << Plux->A12 << " " << Plux->A13;
308
309 //_secID->Fill(L_QQ_Q_l->time[oi],mcmdrc->SeqID);
310
311 timeElapsedFromTLE = TH - (Long64_t) tledate.Convert()+ L_QQ_Q_l->time[oi];
312 orbit.getPosition(((double) timeElapsedFromTLE)/60., &eci);
313 //outputFile<<" "<< eci.getPos().m_z;//eci.getPos().m_x<<" "<<eci.getPos().m_y<<" "<<eci.getPos().m_z<<" "<<eci.getVel().m_x<<" "<<eci.getVel().m_y<<" "<<eci.getVel().m_z;
314 Plux->TransAngle(eci.getPos().m_x,eci.getPos().m_y,eci.getPos().m_z,eci.getVel().m_x,eci.getVel().m_y,eci.getVel().m_z,Plux->A11,Plux->A12,Plux->A13,L_QQ_Q_l->quat[oi][0],L_QQ_Q_l->quat[oi][1],L_QQ_Q_l->quat[oi][2],L_QQ_Q_l->quat[oi][3]);
315
316 _Tangazh->Fill(L_QQ_Q_l->time[oi],Plux->Tangazh);
317 _Ryskanie->Fill(L_QQ_Q_l->time[oi],Plux->Ryskanie);
318 _Kren->Fill(L_QQ_Q_l->time[oi],Plux->Kren);
319
320 //outputFile<<" "<<Plux->Tangazh<<" "<<Plux->Kren<<" "<<Plux->Ryskanie<<"\n";
321 //}
322 }
323 }
324 }
325 }
326
327 //outputFile.close();
328
329 TCanvas *c1 = new TCanvas("c1","Quaternions",1200,1600);
330 //TCanvas *c2 = new TCanvas("c2","Anglees",1200,1600);
331 TCanvas *c4 = new TCanvas("c4","AngleesTRK",1200,1600);
332
333 c1->Divide(1,4);
334 //c2->Divide(1,3);
335 c4->Divide(1,3);
336
337 c1->cd(1);
338 _L1->SetMarkerColor(kBlack);
339 _L1->Draw("");
340
341 c1->cd(2);
342 _L2->SetMarkerColor(kBlue);
343 _L2->Draw("");
344
345 c1->cd(3);
346 _L3->SetMarkerColor(kRed);
347 _L3->Draw("");
348
349 c1->cd(4);
350 _L4->SetMarkerColor(kGreen);
351 _L4->Draw("");
352
353 // c1->cd(5);
354 // _secID->SetMarkerColor(kYellow);
355 // _secID->Draw("");
356 /*
357 c2->cd(1);
358 _A3->SetMarkerColor(kRed);
359 _A3->Draw("");
360
361 c2->cd(2);
362 _A2->SetMarkerColor(kGreen);
363 _A2->Draw("");
364
365 c2->cd(3);
366 _A1->SetMarkerColor(13);
367 _A1->Draw("");
368 */
369 c4->cd(1);
370 _Tangazh->SetMarkerColor(142);
371 _Tangazh->Draw("");
372
373 c4->cd(2);
374 _Ryskanie->SetMarkerColor(226);
375 _Ryskanie->Draw("");
376
377 c4->cd(3);
378 _Kren->SetMarkerColor(142);
379 _Kren->Draw("");
380
381 oss.str("");
382 oss << outDirectory+basename+"/"+basename << "_Inclinations_Quaternions.png";
383 c1->SaveAs(oss.str().c_str());
384 oss.str();
385
386 /* oss.str("");
387 oss << basename<< "_Angles.png";
388 c2->SaveAs(oss.str().c_str());
389 oss.str();
390 */
391 oss.str("");
392 oss << outDirectory+basename+"/"+basename << "_Inclinations_Angles.png";
393 c4->SaveAs(oss.str().c_str());
394 oss.str();
395
396
397
398 delete _L1;
399 delete _L2;
400 delete _L3;
401 delete _L4;
402
403 // delete _A1;
404 // delete _A2;
405 // delete _A3;
406
407 //delete _secID;
408
409 delete _Tangazh;
410 delete _Kren;
411 delete _Ryskanie;
412
413 rootFile->Close();
414 }
415
416 // Get the TLE from tleFile. The right TLE is that one with the
417 // closest previous date to offRes, that is the date at the time of
418 // the first timesync found in the root file.
419 //
420 // Warning: you must use a tle file obtained by space-track.org
421 // querying the database with the RESURS DK-1 id number 29228,
422 // selecting the widest timespan, including the satellite name in the
423 // results.
424 cTle *getTle(TString tleFile, TDatime offRes)
425 {
426 Float_t tledatefromfile, tledatefromroot;
427 fstream tlefile(tleFile.Data(), ios::in);
428 vector<cTle*> ctles;
429 vector<cTle*>::iterator iter;
430
431
432 // Build a vector of *cTle
433 while(1) {
434 cTle *tlef;
435 string str1, str2, str3;
436
437 getline(tlefile, str1);
438 if(tlefile.eof()) break;
439
440 getline(tlefile, str2);
441 if(tlefile.eof()) break;
442
443 getline(tlefile, str3);
444 if(tlefile.eof()) break;
445
446 // We now have three good lines for a cTle.
447 tlef = new cTle(str1, str2, str3);
448 ctles.push_back(tlef);
449 }
450
451 // Sort by date
452 sort(ctles.begin(), ctles.end(), compTLE);
453
454 tledatefromroot = (offRes.GetYear()-2000)*1e3 + (offRes.Convert() - (TDatime(offRes.GetYear(), 1, 1, 0, 0, 0)).Convert())/ (24.*3600.);
455
456 for(iter = ctles.begin(); iter != ctles.end(); iter++) {
457 cTle *tle = *iter;
458
459 tledatefromfile = getTleJulian(tle);
460
461 if(tledatefromroot > tledatefromfile) {
462 tlefile.close();
463 cTle *thisTle = tle;
464 ctles.clear();
465
466 return thisTle;
467 }
468 }
469
470 // File ended withoud founding a TLE with a date after offRes. We'll use the last aviable date.
471 cerr << "Warning: using last available TLE in " << tleFile.Data() << ". Consider updating your tle file." << endl;
472
473 tlefile.close();
474 cTle *thisTle = ctles[ctles.size()-1];
475 ctles.clear();
476
477 return thisTle;
478 }
479
480
481 // Return whether the first TLE is older than the second
482 bool compTLE (cTle *tle1, cTle *tle2)
483 {
484 return getTleJulian(tle1) > getTleJulian(tle2);
485 }
486
487
488 // Return the date of the tle using the format (year-2000)*1e3 +
489 // julian day. e.g. 6364.5 is the 31th Dec 2006 12:00:00.
490 // It does *not* return a cJulian date.
491 float getTleJulian(cTle *tle) {
492 return tle->getField(cTle::FLD_EPOCHYEAR)*1e3 + tle->getField(cTle::FLD_EPOCHDAY);
493 }
494
495
496 // Look for a timesync in the TFile rootFile. Set timesync and
497 // obt_timesync. Returns 1 if timesync is found, 0 otherwise.
498 int lookforTimesync(TFile *rootFile, Float_t *timesync, Float_t *obt_timesync) {
499 *timesync = -1; // will be != -1 if found
500
501 ULong64_t nevents = 0;
502 pamela::McmdRecord *mcmdrc = 0;
503 pamela::McmdEvent *mcmdev = 0;
504 TArrayC *mcmddata;
505 TTree *tr = (TTree*) rootFile->Get("Mcmd");
506
507 tr->SetBranchAddress("Mcmd", &mcmdev);
508
509 nevents = tr->GetEntries();
510
511 // Looking for a timesync. We stop at the first one found.
512 long int recEntries;
513
514 for(UInt_t i = 0; i < nevents; i++) {
515 tr->GetEntry(i);
516 recEntries = mcmdev->Records->GetEntries();
517
518 for(UInt_t j = 0; j < recEntries; j++) {
519 mcmdrc = (pamela::McmdRecord*)mcmdev->Records->At(j);
520
521 if ((mcmdrc != 0) && (mcmdrc->ID1 == 0xE0)) //Is it a TimeSync?
522 {
523 mcmddata = mcmdrc->McmdData;
524 *timesync = (((unsigned int)mcmddata->At(0)<<24)&0xFF000000)
525 + (((unsigned int)mcmddata->At(1)<<16)&0x00FF0000)
526 + (((unsigned int)mcmddata->At(2)<<8)&0x0000FF00)
527 + (((unsigned int)mcmddata->At(3))&0x000000FF);
528 *obt_timesync = (mcmdrc->MCMD_RECORD_OBT)*(1./1000.);
529
530 goto out; // a timesync is found
531 }
532 }
533 }
534
535 out:
536
537 if (*timesync == -1)
538 return 0;
539 else
540 return 1;
541 }
542
543
544 // Return a string like YYYY-MM-DD hh:mm:ss, a datetime format.
545 string getTleDatetime(cTle *tle)
546 {
547 int year, mon, day, hh, mm, ss;
548 double dom; // day of month (is double!)
549 stringstream date; // date in datetime format
550
551 // create a cJulian from the date in tle
552 cJulian jdate = cJulian( 2000 + (int) tle->getField(cTle::FLD_EPOCHYEAR), tle->getField(cTle::FLD_EPOCHDAY));
553
554 // get year, month, day of month
555 jdate.getComponent(&year, &mon, &dom);
556
557 // build a datetime YYYY-MM-DD hh:mm:ss
558 date.str("");
559 day = (int) floor(dom);
560 hh = (int) floor( (dom - day) * 24);
561 mm = (int) floor( ((dom - day) * 24 - hh) * 60);
562 ss = (int) floor( ((((dom - day) * 24 - hh) * 60 - mm) * 60));
563 // ms = (int) floor( (((((dom - day) * 24 - hh) * 60 - mm) * 60) - ss) * 1000);
564
565 date << year << "-" << mon << "-" << day << " " << hh << ":" << mm << ":" << ss;
566
567 return date.str();
568 }
569
570 //
571 // Solve the overflow for anticoincidence because this counter is
572 // stored in 2 bytes so counts from 0 to 65535.
573 //
574 // counter is the actual value.
575 // oldValue is meant to be the previous value of counter.
576 //
577 // Example:
578 // for(...) {
579 // ...
580 // corrected_diff = solve_ac_overflow(oldValueOfTheCounter, actualValue);
581 // ...
582 // }
583 //
584 //
585 // Returns the corrected difference between counter and oldValue and
586 // set oldValue to the value of counter.
587 // Attention: oldValue is a reference.
588 Int_t solve_ac_overflow(Int_t& oldValue, Int_t counter)
589 {
590 Int_t truediff = 0;
591
592 if (counter < oldValue) // overflow!
593 truediff = 0xFFFF - oldValue + counter;
594 else
595 truediff = counter - oldValue;
596
597 oldValue = counter;
598
599 return truediff;
600 }

  ViewVC Help
Powered by ViewVC 1.1.23