CheezyDisplayModule.cxx

Go to the documentation of this file.

// $Id: CheezyDisplayModule.cxx,v 1.21 2002/12/13 23:18:44 rhatcher Exp $
//
// A JobControl Module for using the CheezyDisplay
//
// rhatcher@fnal.gov

#include "RerootExodus/CheezyDisplayModule.h"
// Used for drawing
#include "RerootExodus/CheezyDisplay.h"

#include "MinosObjectMap/MomNavigator.h"

#include "MessageService/MsgService.h"
#include "MessageService/MsgFormat.h"
#include "JobControl/JobCModuleRegistry.h"
#include "JobControl/JobCommand.h"

#include "Plex/PlexStripEndId.h"
#include "CandData/CandRecord.h"
#include "CandDigit/CandDigitHandle.h"
#include "CandDigit/CandDigitListHandle.h"

#include "Record/RecMinos.h"
#include "Record/RecMinosHdr.h"

#include "TMath.h"

ClassImp(CheezyDisplayModule)

//......................................................................

CVSID("$Id: CheezyDisplayModule.cxx,v 1.21 2002/12/13 23:18:44 rhatcher Exp $");
JOBMODULE(CheezyDisplayModule, "CheezyDisplayModule",
         "Draw an event using the CheezyDisplay");

//......................................................................

CheezyDisplayModule::CheezyDisplayModule() 
   : fListPlex("canddigitlist"), fListTruth("candmcdigitlist"), 
     fDrawEmpty(false), fDisplay(0)
{
   // construct a new "CheezyDisplayModule" JobControl module
   // by default:
   //    no specific choice of CandDigitList to use
   //    draw all possibilities in PlexSEIdAltL (w/ no miminum wgt)

   // Create a display
   // do it here in the ctor so that configuration state info can be
   // passed to it before any events are drawn
   // (eg. settings of DrawTruth, DrawOnlyBest, WgtThresh)

  if ( ! fDisplay ) {
     fDisplay = new CheezyDisplay("CheezyDisplayModule",
                                  "CheezyDisplayModule display");
     MSG("Cheezy",Msg::kDebug)
        << "create new CheezyDisplay within ctor" << endl;
  }

}

//......................................................................

CheezyDisplayModule::~CheezyDisplayModule() 
{
  MSG("Exodus", Msg::kVerbose) << "CheezyDisplayModule::Destructor\n";

  SafeDelete(fDisplay);
}

//......................................................................

JobCResult CheezyDisplayModule::Ana(const MomNavigator *mom)
{
  // Find the RawRecord for Header info (detector) and TRUTH (if available)
  // Find the CandDigitList
  // draw digits on CheezyDisplay

  // Get either RawRecord or CandRecord from Mom 
  RecMinos *rec = dynamic_cast<RecMinos *>(mom->GetFragment("RecMinos"));
                                        
  if ( ! rec ) {
     MSG("Exodus",Msg::kFatal) << "CheezyDisplayModule::Ana" <<
        " failed to fin a \"RecMinos\"" << endl;
     return JobCResult::kError;
  }

  VldContext vldc = rec->GetHeader()->GetVldContext();

  // Create (if necessary) a display
  if ( ! fDisplay ) {
     fDisplay = new CheezyDisplay("CheezyDisplayModule",
                                  "CheezyDisplayModule display");
     MSG("Cheezy",Msg::kWarning)
        << "create new CheezyDisplay within Ana" << endl;
  }

  fDisplay->ClearLists();
  fDisplay->SetVldContext(vldc);

  // Draw TRUTH first so other digits won't be obscured
  // Find MC version of CandDigitList fragment in MOM.

// Find (Primary)CandidateRecord fragment in MOM.
  CandRecord *candrec = dynamic_cast<CandRecord *>
    (mom->GetFragment("CandRecord")); //, "PrimaryCandidateRecord"));
  if (candrec == 0) {
    MSG("Cheezy", Msg::kWarning) 
      << "No " // << "Primary"
      << "CandRecord in MOM." 
      << endl;
    return JobCResult::kError;
  }

  int ndigits = 0;

  CandDigitListHandle *cmcdlh = dynamic_cast<CandDigitListHandle *>
     (candrec->FindCandHandle("CandDigitListHandle",fListTruth.Data()));

  if (cmcdlh) {
     // do iteration only if there is a Truth list!
     CandDigitHandleItr cmcdhItr(cmcdlh->GetDaughterIterator());

     // loop over digits
     while ( CandDigitHandle *cmcdh = cmcdhItr() ) {
        // for each digit
        PlexSEIdAltL altlist = cmcdh->GetPlexSEIdAltL();
        fDisplay->AddSEIdAltL(altlist,kTRUE);
        ndigits++;
     }
  }  

  // Find requested CandDigitList fragment in MOM.
  CandDigitListHandle *cdlh = dynamic_cast<CandDigitListHandle *>
      (candrec->FindCandHandle("CandDigitListHandle",fListPlex.Data()));

  CandDigitHandleItr cdhItr(cdlh->GetDaughterIterator());
  
  while ( CandDigitHandle *cdh = cdhItr() ) {
     // for each digit
     PlexSEIdAltL altlist = cdh->GetPlexSEIdAltL();
     fDisplay->AddSEIdAltL(altlist,kFALSE);
     ndigits++;
  }

  // Now actually draw these things
  if (!fDisplay) MSG("Cheezy",Msg::kFatal) << " no display?? " << endl;
  if (ndigits || fDrawEmpty) fDisplay->Draw();

  return JobCResult::kAOK;

}

//......................................................................

void CheezyDisplayModule::HandleCommand(JobCommand *command) 
{
//
// Process configuration commands
//

//  MSG("Exodus", Msg::kInfo) << "CheezyDisplayModule::HandleCommand\n";

  TString cmd = command->PopCmd();
  if (cmd == "Set") {
     TString opt = command->PopOpt();
     if      (opt == "ListPlex")  fListPlex  = command->PopOpt();
     else if (opt == "ListTruth") fListTruth = command->PopOpt();
     else if (opt == "DrawEmpty") fDrawEmpty = command->PopIntOpt();
     else if (opt == "DrawTruth")
     { // damn braces needed to get scoping of "if" statement right
        if (fDisplay) fDisplay->SetDrawTruth(command->PopIntOpt() != 0);
     }
     else if (opt == "OnlyBest")
     { // damn braces needed to get scoping of "if" statement right
        if (fDisplay) fDisplay->SetDrawOnlyBest(command->PopIntOpt() != 0);
     }
     else if (opt == "WgtThresh")
     { // damn braces needed to get scoping of "if" statement right
        if (fDisplay) fDisplay->SetWgtThresh(command->PopFloatOpt());
     }
     else {
        MSG("Exodus", Msg::kWarning)
           << "CheezyDisplayModule: Unrecognized option " << opt << endl;
     }
  } else if (cmd == "Redraw") {
     if (fDisplay) fDisplay->Draw();
  } else {
     MSG("Exodus", Msg::kWarning)
        << "CheezyDisplayModule: Unrecognized command " << cmd << endl;
  }
}

---