DumpMom.cxx

Go to the documentation of this file.

#include "DumpMom.h"

#include <MinosObjectMap/MomNavigator.h>
#include <Candidate/CandHandle.h>
#include <CandData/CandRecord.h>
#include <CandData/CandHeader.h>
#include <Record/RecMinos.h>
#include <Record/RecMinosHdr.h>
#include <RawData/RawRecord.h>
#include <RawData/RawHeader.h>
#include <RawData/RawDataBlock.h>
#include "Record/SimSnarlRecord.h"
#include "Record/SimSnarlHeader.h"

#include <TClass.h>
#include <TNamed.h>

#include <string>
#include <iostream>
using namespace std;

bool DataUtil::dump_cand(const CandHandle* ch, 
                         std::ostream& os,
                         std::string prefix,
                         int recurse_depth,
                         int max_daughters)
{
  // Add name and title if it's a CandHandle.
  if(ch) {
    os << prefix 
       << Form("%s \t\"%s\" %s",ch->ClassName(),ch->GetName(), ch->GetTitle()) 
       << std::endl;
    int n = ch->GetNDaughters();
    if( max_daughters < n ) n = max_daughters;
    if( recurse_depth <=0) n = 0;
    if(n > 0) {
      TIter iter = ch->GetDaughterIterator();
      for(int i=0;i<n;i++) {
        const CandHandle* next = dynamic_cast<const CandHandle*>(iter.Next());
        if(next)
          if(! next->InheritsFrom("CandDigitHandle") )
            dump_cand(next,
                      os,
                      prefix+"  ",
                      recurse_depth-1,
                      max_daughters);
        
      } 
    }
    if(ch->GetNDaughters() > 0) {
      os << prefix << ".. Total daughters:" <<  ch->GetNDaughters() << endl;
    }
    
    return true;
  }
  return false;
}


bool DataUtil::dump_mom(const MomNavigator* mom, std::ostream& os)
{
    if (!mom) {
        return false;
    }

    os << "Mom holds:\n";
    TIter next = mom->FragmentIter();
    TObject* obj;
    while ( (obj=next()) ) {
        RecMinos*  rm  = dynamic_cast<RecMinos*>(obj);
        RecRecord* rdr = dynamic_cast<RecRecord*>(obj);
        const TObjArray* compArray[2] = {0,0};

        if (!rm && !rdr) {
          // It's not a RecMinos/RecRecord, so just dump the class title:
          os << obj->ClassName() << endl;
          os << "\tNo Components." << endl;
          continue;
        }

        // If it comes from RecMinos or RecRecord, it's got a name and title.
        
        if (rm) {
          os <<  obj->ClassName()
             << "\t\"" << rm->GetName() << "\""
             << "\t" << rm->GetTitle() << endl;

          compArray[0] = &(rm->GetComponents());
          compArray[1] = &(rm->GetTemporaries());

          // If it's a RawRecord, print the header.
          RawRecord* rawrec = dynamic_cast<RawRecord*>(rm);
          if(rawrec) {
            if(rawrec->GetRawHeader()) {
              rawrec->GetRawHeader()->FormatToOStream(os);
              os << endl;
            }
          }

          // If it's a CandRecord, print the header
          CandRecord* candrec = dynamic_cast<CandRecord*>(rm);
          if (candrec) {
            const CandHeader* candhead = candrec->GetCandHeader();
            if (candhead) {
              candhead->Print();   // currently can't handle other than cout
            }
          }

        }

        if (rdr) {
          os <<  obj->ClassName()
             << "\t\"" << rdr->GetName() << "\""
             << "\t" << rdr->GetTitle() << endl;

          // If it's a SimSnarlRecord, print the header
          SimSnarlRecord* simrec = dynamic_cast<SimSnarlRecord*>(rdr);
          if (simrec) {
            compArray[0] = &(simrec->GetComponents());
            compArray[1] = &(simrec->GetTemporaries());

            const SimSnarlHeader* simhead = simrec->GetSimSnarlHeader();
            if (simhead) {
              simhead->Print(os);
              os << endl;
            }
          }
        }
        

        const char* arrayName[2] = {"Components","Temporaries"};
        for (int iArray=0; iArray<2; ++iArray) {
          if ( compArray[iArray] == 0 || 
               compArray[iArray]->GetEntriesFast() <= 0 ) continue;
          os << "---" << arrayName[iArray] << ":" << endl;
          const TObjArray& oa = *(compArray[iArray]);
          TIter next_comp(oa.MakeIterator());
          TObject* comp;
          while ( (comp=next_comp()) ) {
                       
            // Try to see what kind o' thing it is
            TNamed* named = dynamic_cast<TNamed*>(comp);
            CandHandle* ch = dynamic_cast<CandHandle*>(comp);
            TCollection* col = dynamic_cast<TCollection*>(comp);
            
            if(col) {
              // Add name and title if it's a collection
              os << "      " << comp->ClassName()
                 << "\t\"" << col->GetName() << "\"\t with " << col->GetSize() << " elements" << endl;

            } else if(named) {
              // Add name and title if it's a named object.
              os << "      " << comp->ClassName()
                 << "\t\"" << named->GetName() << "\"\t" << named->GetTitle() << endl;

            } else if(ch) {
              // It's a candidate. Do something clever.
              dump_cand(ch,os,"      ");

            } else {
              // Dunno. Print it's name or something.
              os << "      " << comp->ClassName() << endl;
            };        

            //const RawDataBlock* rdb = dynamic_cast<const RawDataBlock*>(comp);
            //if(rdb) rdb->Print(); // keep from printing all the rawdigits in an RDDB
          
          }
        }  // loop over Components/Temporaries
        
        os << endl;

    } // loop over objects in mom's basket

#if 0
    CandRecord* crec = dynamic_cast<CandRecord*>
        (mom->GetFragment("CandRecord", "PrimaryCandidateRecord"));
    if (!crec) {
        cerr << "dump_mom: no CandRecord\n";
        return false;
    }

    const TObjArray oa = crec->GetCandHandleList();
    int last = oa.GetLast();
    cerr << "Contents of PrimaryCandidateRecord:\n";
    for (int i = 0; i <= last; ++i) {
        TObject* o = oa.At(i);
        if (!o) continue;
        cerr << "\t" << o->ClassName() << "::" << o->GetName()<< endl;
    }
    
#endif
    return true;
}

---