Changeset 17699 for trunk/Mars/fact/analysis/mc/star.C

Timestamp:

04/26/14 15:35:48 (11 years ago)

Author:

tbretz

Message:

Improved the calling convention. There is really no need to give so many arguments. Pipe some Monte Carlo values through. I wonder what special MC macros are for which omit all the Monte Carlo data in their output.

File:

• trunk/Mars/fact/analysis/mc/star.C (modified) (8 diffs)

trunk/Mars/fact/analysis/mc/star.C

@@ -1 +1 @@
 #include <sstream>
 #include <iostream>
+
+#include "MLog.h"
+#include "MLogManip.h"
 
 #include "TH1F.h"
@@ -8 +11 @@
 #include "TLine.h"
 
+#include "../mcore/DrsCalib.h"
 #include "MDrsCalibration.h"
 //#include "MLogManip.h"
@@ -53 +57 @@
 #include "MImgCleanStd.h"
 
-
 using namespace std;
 
-int star(
-    const char *mars_data_file_path = "/fhgfs/groups/app/callisto_star_test/callisto_for_mc_test_output/callisto.root",
-    const char *root_output_file_path = "/fhgfs/groups/app/callisto_star_test/callisto_for_mc_test_output/star.root",
-    const char *status_display_output_path = "/fhgfs/groups/app/callisto_star_test/callisto_for_mc_test_output/star_status_display.root", 
-    const char *status_display_title = "star_status_display_title",
-    const char *hillas_txt_file_path = "/fhgfs/groups/app/callisto_star_test/star_hillas.txt",
-    Double_t lvl1=4.0, 
-    Double_t lvl2=2.5,
-    Double_t deltat = 17.5)
+int star(const TString callistofile, TString outfile,
+         TString displayfile, TString displaytitle,
+         const TString txtfile,
+         Double_t lvl1=4.0, Double_t lvl2=2.5, Double_t deltat = 17.5);
+
+
+int star(const TString infile = "callisto.root", TString outfile = "",
+         Double_t lvl1=4.0,  Double_t lvl2=2.5, Double_t deltat = 17.5)
 {
+    return star(infile, outfile, "", "", "", lvl1, lvl2, deltat);
+}
+
+int star(const TString callistofile, TString outfile,
+         TString displayfile,  TString displaytitle,
+         const TString txtfile,
+         Double_t lvl1, Double_t lvl2, Double_t deltat)
+{
+
     //Double_t lvl1=7.8, 
     //Double_t lvl2=3.9,
@@ -72 +83 @@
     // ------------------------------------------------------
 
+    if (displaytitle.IsNull())
+        displaytitle = gSystem->BaseName(callistofile);
+
+    FileStat_t fstat;
+    int rc = gSystem->GetPathInfo(outfile, fstat);
+    bool isdir = !rc || R_ISDIR(fstat.fMode);
+
+    char *buf = gSystem->ConcatFileName(outfile, "star.root");
+    outfile = buf;
+    delete [] buf;
+
+    if (displayfile.IsNull())
+    {
+        displayfile = outfile;
+        displayfile.Insert(displayfile.Last('.'), "-display");
+    }
+    else
+    {
+        if (isdir && gSystem->DirName(displayfile)==TString("."))
+        {
+            buf = gSystem->ConcatFileName(outfile, displayfile);
+            displayfile = buf;
+            delete [] buf;
+        }
+    }
+
+    // ------------------------------------------------------
+
     gLog.Separator("Star");
     gLog << all << "Calculate image parameters of sequence ";
     gLog << endl;
-    gLog << "mars_data_file_path: " << mars_data_file_path << endl;
-    gLog << "root_output_file_path: " << root_output_file_path << endl;
-    gLog << "status_display_output_path: " << status_display_output_path << endl;
-    gLog << "status_display_title: " << status_display_title << endl;
+    gLog << "Callisto file: " << callistofile << endl;
+    gLog << "Output file:   " << outfile << endl;
+    gLog << "Display file:  " << displayfile << endl;
+    gLog << "Display title: " << displaytitle << endl;
     gLog << endl;
-    gLog << "lvl1: " << lvl1 <<endl;
-    gLog << "lvl2: " << lvl2 <<endl;
+    gLog << "lvl1: "   << lvl1 <<endl;
+    gLog << "lvl2: "   << lvl2 <<endl;
+    gLog << "deltat: " << deltat <<endl;
     
     
@@ -152 +192 @@
     MReadMarsFile read("Events");
     read.DisableAutoScheme();
-    read.AddFile(mars_data_file_path);
+    read.AddFile(callistofile);
 
     MContinue cont("MRawEvtHeader.GetTriggerID!=4", "SelectData");
@@ -276 +316 @@
     // ---------------------------------------------------------------
 
-    MWriteAsciiFile write_ascii_hillas(hillas_txt_file_path, "MHillas");
-    write_ascii_hillas.AddColumns("MHillasExt");
-    write_ascii_hillas.AddColumns("MHillasSrc");
-
-
-    MWriteRootFile write(root_output_file_path, "RECREATE", "Image parameters", 2);
-    write.AddContainer("MTime",           "Events");
-    write.AddContainer("MHillas",         "Events");
-    write.AddContainer("MHillasExt",      "Events");
-    write.AddContainer("MHillasSrc",      "Events");
-    write.AddContainer("MImagePar",       "Events");
-    write.AddContainer("MNewImagePar",    "Events");
-    write.AddContainer("MRawEvtHeader",   "Events");
-    write.AddContainer("ThetaSquared",   "Events");
-    write.AddContainer("Disp",           "Events");
-    write.AddContainer("MRawRunHeader",   "RunHeaders");
-    write.AddContainer("MGeomCam",        "RunHeaders");
+    MWriteAsciiFile writeascii(txtfile, "MHillas");
+    writeascii.AddColumns("MHillasExt");
+    writeascii.AddColumns("MHillasSrc");
+
+
+    MWriteRootFile write(outfile, "RECREATE", "Image parameters", 2);
+    write.AddContainer("MTime",               "Events");
+    write.AddContainer("MHillas",             "Events");
+    write.AddContainer("MHillasExt",          "Events");
+    write.AddContainer("MHillasSrc",          "Events");
+    write.AddContainer("MImagePar",           "Events");
+    write.AddContainer("MNewImagePar",        "Events");
+    write.AddContainer("MRawEvtHeader",       "Events");
+    write.AddContainer("ThetaSquared",        "Events");
+    write.AddContainer("Disp",                "Events");
+    write.AddContainer("MCorsikaEvtHeader",   "Events", kFALSE);
+    write.AddContainer("MMcEvt",              "Events", kFALSE);
+    write.AddContainer("IncidentAngle",       "Events", kFALSE);
+    write.AddContainer("MPointingPos",        "Events", kFALSE);
+
+    write.AddContainer("MRawRunHeader",       "RunHeaders");
+    write.AddContainer("MGeomCam",            "RunHeaders");
+    write.AddContainer("MMcCorsikaRunHeader", "RunHeaders", kFALSE);
+    write.AddContainer("MCorsikaRunHeader",   "RunHeaders", kFALSE);
+    write.AddContainer("MMcRunHeader",        "RunHeaders", kFALSE);
+
     MFDataPhrase fmuonhn("MMuonCalibPar.fRelTimeSigma>=0",
                          "MuonHistCut");
@@ -351 +400 @@
     //tlist.AddToList(&writem);
     tlist.AddToList(&write);
-    tlist.AddToList(&write_ascii_hillas);
+    if (!txtfile.IsNull())
+        tlist.AddToList(&writeascii);
 
     if (!loop.Eventloop())
@@ -359 +409 @@
         return 4;
 
-    d->SetTitle(status_display_title, kFALSE);
-    d->SaveAs(status_display_output_path);
+    d->SetTitle(displaytitle, kFALSE);
+    d->SaveAs(displayfile);
 
     return 0;
 }
+