Changeset 3800 for trunk/MagicSoft/Mars/mraw/MRawFileRead.cc

Timestamp:

04/22/04 19:40:02 (21 years ago)

Author:

tbretz

Message:

*** empty log message ***

File:

• trunk/MagicSoft/Mars/mraw/MRawFileRead.cc (modified) (9 diffs)

trunk/MagicSoft/Mars/mraw/MRawFileRead.cc

@@ -24 +24 @@
 
 //////////////////////////////////////////////////////////////////////////////
-//                                                                          //
-//  MRawFileRead                                                            //
-//                                                                          //
-//  This tasks reads the raw binary file like specified in the TDAS???      //
-//  and writes the data in the corresponding containers which are           //
-//  either retrieved from the parameter list or created and added.          //
-//                                                                          //
-//  Input Containers:                                                       //
-//   -/-                                                                    //
-//                                                                          //
-//  Output Containers:                                                      //
-//   MRawRunHeader, MRawEvtHeader, MRawEvtData, MRawCrateArray, MRawEvtTime //
-//                                                                          //
+//
+//  MRawFileRead
+//
+//  This tasks reads the raw binary file like specified in the TDAS???
+//  and writes the data in the corresponding containers which are
+//  either retrieved from the parameter list or created and added.
+//
+//  Use SetInterleave() if you don't want to read all events, eg
+//    SetInterleave(5) reads only each 5th event.
+//
+//  Input Containers:
+//   -/-
+//
+//  Output Containers:
+//   MRawRunHeader, MRawEvtHeader, MRawEvtData, MRawCrateArray, MRawEvtTime
+//
 //////////////////////////////////////////////////////////////////////////////
 #include "MRawFileRead.h"
@@ -43 +46 @@
 #include <fstream>
 
+#include <TSystem.h>
+
 #include "MLog.h"
 #include "MLogManip.h"
@@ -48 +53 @@
 #include "MTime.h"
 #include "MParList.h"
+
 #include "MRawRunHeader.h"
 #include "MRawEvtHeader.h"
@@ -95 +101 @@
 //
 MRawFileRead::MRawFileRead(const char *fname, const char *name, const char *title)
-    : fFileName(fname), fIn(NULL)
+    : fFileNames(NULL), fNumFile(0), fIn(NULL), fParList(NULL), fInterleave(1)
 {
     fName  = name  ? name  : "MRawFileRead";
     fTitle = title ? title : "Read task to read DAQ binary files";
 
-    fIn = new ifstream;
+    fFileNames = new TList;
+    fFileNames->SetOwner();
+
+    AddFile(fname);
 }
 
@@ -109 +118 @@
 MRawFileRead::~MRawFileRead()
 {
-    delete fIn;
+    delete fFileNames;
+    if (fIn)
+        delete fIn;
+}
+
+// --------------------------------------------------------------------------
+//
+// Add a new file to a list of files to be processed, Returns the number
+// of files added. (We can enhance this with a existance chack and
+// wildcard support)
+//
+Int_t MRawFileRead::AddFile(const char *fname)
+{
+    TNamed *name = new TNamed(fname, "");
+    fFileNames->AddLast(name);
+    return 1;
+
+}
+
+// --------------------------------------------------------------------------
+//
+// This opens the next file in the list and deletes its name from the list.
+//
+Bool_t MRawFileRead::OpenNextFile()
+{
+    //
+    // open the input stream and check if it is really open (file exists?)
+    //
+    if (fIn)
+        delete fIn;
+    fIn = NULL;
+
+    //
+    // Check for the existance of a next file to read
+    //
+    TObject *file = fFileNames->At(fNumFile);
+    if (!file)
+        return kFALSE;
+
+    //
+    // open the file which is the first one in the chain
+    //
+    const char *name = file->GetName();
+
+    const char *expname = gSystem->ExpandPathName(name);
+    fIn = new ifstream(expname);
+
+    const Bool_t noexist = !(*fIn);
+    if (noexist)
+    {
+        *fLog << err << "Cannot open file " << expname << ": ";
+        *fLog << strerror(errno) << endl;
+    }
+    else
+        *fLog << inf << "Open file: '" << name << "'" << endl;
+
+    delete [] expname;
+
+    if (noexist)
+        return kFALSE;
+
+    fNumFile++;
+
+    //
+    // Read RUN HEADER (see specification) from input stream
+    //
+    if (!fRawRunHeader->ReadEvt(*fIn))
+        return kFALSE;
+
+    if (!(*fIn))
+    {
+        *fLog << err << "Error: Accessing file '" << name << "'" << endl;
+        return kFALSE;
+    }
+
+    //
+    // Print Run Header
+    //
+    fRawRunHeader->Print();
+    *fRawEvtTime = fRawRunHeader->GetRunStart();
+
+    fNumEvents += fRawRunHeader->GetNumEvents();
+
+    //
+    // Give the run header information to the 'sub-classes'
+    // Run header must be valid!
+    //
+    fRawEvtHeader->Init(fRawRunHeader, fRawEvtTime);
+    fRawEvtData  ->Init(fRawRunHeader, fRawCrateArray);
+
+    //
+    // Search for MTaskList
+    //
+    MTask *tlist = (MTask*)fParList->FindObject("MTaskList");
+    if (!tlist)
+    {
+        *fLog << err << dbginf << "MTaskList not found... abort." << endl;
+        return kFALSE;
+    }
+
+    //
+    // A new file has been opened and new headers have been read.
+    //  --> ReInit tasklist
+    //
+    return tlist->ReInit(fParList);
+}
+
+// --------------------------------------------------------------------------
+//
+// Return file name of current file.
+//
+const TString MRawFileRead::GetFileName() const
+{
+    const TObject *file = fFileNames->At(fNumFile-1);
+    return file ? file->GetName() : "";
+}
+
+// --------------------------------------------------------------------------
+//
+// Restart with the first file
+//
+Bool_t MRawFileRead::Rewind()
+{
+    fNumFile=0;
+    fNumEvents=0;
+    return OpenNextFile();
 }
 
@@ -130 +264 @@
 Int_t MRawFileRead::PreProcess(MParList *pList)
 {
+    fParList = pList;
+
     //
     // open the input stream
@@ -135 +271 @@
     // successfull
     //
-    fIn->open(fFileName);
-    if (!(*fIn))
-    {
-        *fLog << err << "Cannot open file " << fFileName << ": ";
-        *fLog << strerror(errno) << endl;
-        return kFALSE;
-    }
-
     if (!MRawRead::PreProcess(pList))
         return kFALSE;
 
     //
-    // Read RUN HEADER (see specification) from input stream
-    //
-    if (!fRawRunHeader->ReadEvt(*fIn))
-        return kFALSE;
-
-    if (!(*fIn))
-    {
-        *fLog << err << "Error: Accessing file '" << fFileName << "'" << endl;
-        return kFALSE;
-    }
-    //if (fRawRunHeader->GetMagicNumber()!=kMagicNumber)
-    //    return kFALSE;
-
-    fRawRunHeader->Print();
-
-    *fRawEvtTime = fRawRunHeader->GetRunStart();
-
-    //
-    // Give the run header information to the 'sub-classes'
-    // Run header must be valid!
-    //
-    fRawEvtHeader->Init(fRawRunHeader, fRawEvtTime);
-    fRawEvtData  ->Init(fRawRunHeader, fRawCrateArray);
+    // Now open next (first) file
+    //
+    if (!Rewind())
+        return kFALSE;
 
     return kTRUE;
@@ -184 +293 @@
 Int_t MRawFileRead::Process()
 {
-    return ReadEvent(*fIn);
+    while (1)
+    {
+        //
+        // skip events if requested
+        //
+        if (fInterleave>1 && GetNumExecutions()%fInterleave>0 && fIn->peek()!=EOF)
+        {
+            SkipEvent(*fIn);
+            return kCONTINUE;
+        }
+
+        //
+        // Read a single event from file
+        //
+        const Bool_t rc = ReadEvent(*fIn);
+        if (rc!=kFALSE)
+            return rc;
+
+        //
+        // If an event could not be read from file try to open new file
+        //
+        if (!OpenNextFile())
+            return kFALSE;
+    }
+    return kTRUE;
 }
 
@@ -195 +328 @@
 Int_t MRawFileRead::PostProcess()
 {
-    fIn->close();
-
     //
     // Sanity check for the number of events
     //
-    if (fRawRunHeader->GetNumEvents()==GetNumExecutions()-1 || GetNumExecutions()==0)
+    if (fNumEvents==GetNumExecutions()-1 || GetNumExecutions()==0)
         return kTRUE;
 
     *fLog << warn << dec;
     *fLog << "Warning - number of read events (" << GetNumExecutions()-1;
-    *fLog << ") doesn't match number in run header (";
-    *fLog << fRawRunHeader->GetNumEvents() << ")." << endl;
+    *fLog << ") doesn't match number in run header(s) (";
+    *fLog << fNumEvents << ")." << endl;
 
     return kTRUE;