---

EDControl.cxx

Go to the documentation of this file.

//_______________________________________________________________________
//
// EDControl class.
//
// Overall control class.  Connects last known data directory, starts the
// GUI, opens the canvas and initialises analysis histos.  Responds to
// requests made by user via the GUI.
//
// -------------------------------------
//
// Authors: N Lumb, Gloria Spandre; INFN-Pisa.
//
// Start date: Sept. 2001.
//
//-----------------------------------------------------------------------


#include "TKey.h"
#include "TSystem.h"
#include "TArrayF.h"


#include "bfemDisplay/EDControl.h"
#include "bfemDisplay/Enum.h"


ifstream   m_rawPathFile;      // ASCII file containing name of raw file.
ifstream   m_reconPathFile;    // ASCII file containing name of recon file.
ifstream   m_acdPedFile;       // ASCII file containing ACD peds.
ifstream   m_xgtPedFile;       // ASCII file containing XGT peds.

EDControl *m_evtCtrl;

ClassImp(EDControl)

EDControl::EDControl() {

  // This pointer to event control is global variable accessible to all classes.
  m_evtCtrl = this;

  m_rawFile = 0;
  m_rawTree = 0;
  m_event = 0;

  m_reconFile = 0;
  m_reconTree = 0;
  m_recon = 0;

  m_ped = 0;
  m_acdped = 0;
  m_xgtped = 0;


  m_StopDisplayLoop = kFALSE;

  m_numEntries = 0;

  // Start the graphical user interface.
  gui = new EDGui( gClient->GetRoot(), 650, 420 );

  // Get data type ( real or MC ).
  setRawDataType( gui->m_isRealData );

  // Open the drawing canvas used to display events and analysis histos.
  m_canvas = new TCanvas( "c1", "GLAST Event Display", 800, 840 );

  // Initialise the analysis histos.
  m_histos = new EDHistos( m_canvas );
  m_histos->InitHistos();

  // Get thresholds from GUI.
  gui->GetThresholds( m_threshold );

  // Read pedestal values.
  m_ped = ReadCALPed();
  m_acdped = ReadACDPed();
  //  m_xgtped = ReadXGTPed();

  analysis = new EDAnalysis();  // Set up the analysis.

  display = new DisplayCanvas();   // Set up the event drawer.

  display->setThresholds( m_threshold );

  display->Draw( m_isRealData );

}


EDControl::~EDControl() {

  if ( m_rawFile ) delete m_rawFile;
  if ( m_event ) delete m_event;
  if ( m_reconFile ) delete m_reconFile;
  if ( m_recon ) delete m_recon;
  if ( m_histos ) delete m_histos;
  if ( m_ped ) delete m_ped;
  if ( m_acdped ) delete m_acdped;
  if ( m_xgtped ) delete m_xgtped;

  delete m_canvas;
  delete gui;

}


void EDControl::ConnectNewRawFile( Char_t *fname )

{
  // Open the specified raw data file.

  if ( m_event ) m_event->Clean();
  if ( m_rawTree ) m_rawTree = 0;
  if ( m_rawFile ) delete m_rawFile;

  cout << "\nOpening raw data file " << fname << "...\n";

  m_rawFile = new TFile( fname );

  m_rawTree = ( TTree* )gDirectory->Get( "T" );
  if ( !m_event ) m_event = new DigiEvent();
  m_rawTree->SetBranchAddress( "DigiEvent", &m_event );

  m_numEntries = checkNumEntries();

  if ( m_numEntries ) gui->setSliderRange( m_numEntries );

}

void EDControl::ConnectNewReconFile( Char_t *fname )

{
  // Open the specified recon data file.

  if ( m_recon ) delete m_recon;
  if ( m_reconTree ) m_reconTree = 0;
  if ( m_reconFile ) delete m_reconFile;

  cout << "\nOpening recon data file " << fname << "...\n";

  m_reconFile = new TFile( fname );

  m_reconTree = ( TTree* )gDirectory->Get( "T" );
  m_recon = new Recon();
  m_reconTree->SetBranchAddress( "Recon", &m_recon );

  m_numEntries = checkNumEntries();

  if ( m_numEntries ) gui->setSliderRange( m_numEntries );

}


void EDControl::UpdateGuiEvents( Int_t evtNo )

{
  // Update the event number indicators in the GUI display.

  gui->setSliderPosition( evtNo );
  gui->SetEventNumber( evtNo );

}

void EDControl::UpdateCanvas()

{
  // Refresh the display canvas.

  m_canvas->Draw();
  m_canvas->Update();

}


Int_t EDControl::checkNumEntries()

{
  // Return maximum number of events that can be processed.  Checks for unequal raw and recon file
  // lengths.
  
  Int_t nentries;
  
  if ( m_reconTree == 0 ) {
    
    nentries = ( Int_t )m_rawTree->GetEntries();
    
  }
  
  else if ( m_rawTree == 0 ) {

    nentries = ( Int_t )m_reconTree->GetEntries();
    
  }
  
  else if ( m_rawTree->GetEntries() != m_reconTree->GetEntries() ) {
    
    printf ( "Two trees have unequal event counts! %i %i taking min \n", 
            ( Int_t )m_rawTree->GetEntries(), ( Int_t )m_reconTree->GetEntries() );
    
    nentries = TMath::Min( ( Int_t )m_rawTree->GetEntries(), ( Int_t )m_reconTree->GetEntries() );
    
  }

  else {
    
    nentries = ( Int_t )m_rawTree->GetEntries();
    
  }

  return nentries - 1;

}

TObjArray* EDControl::ReadCALPed()

{
  // Read the CAL pedestal file.

  Int_t j = 0;

  if ( gSystem->AccessPathName( CALPEDS_AUG2001, kFileExists ) ) {

    // Warn that CAL ped file not available.

    Char_t msg[ 200 ];

    sprintf( msg, "CAL ped file \"%s\" not found. Program will now terminate.", (const char *)CALPEDS_AUG2001 );

    new  TGMsgBox( gui->fClient->GetRoot(), gui, "Warning", msg, 0, kMBOk );

    exit( 1 );

  }
  
  TObjArray *ped = new TObjArray(8);

  ped->InheritsFrom("TVector");

  //TFile *pedFile = new TFile((const char *) CALPEDS_MAY2001 ); //Berrie's Peds -  May 2001
  TFile *pedFile = new TFile((const char *) CALPEDS_AUG2001 ); //Eric's Peds - Aug 2001

  TIter next(pedFile->GetListOfKeys());

  TKey *key;

  while( ( key = (TKey*)next() ) ){

    TVector *v = (TVector*)key->ReadObj();

  
    ped->AddAt(v,j);

    j++;

  }

  cout << "===> Read CAL Peds" << endl;

  return ped;

}


TObjArray* EDControl::ReadACDPed()

{
  // Read the ACD pedestal file.


  if ( gSystem->AccessPathName( (const char *)ACDPEDS, kFileExists ) ) {

    // Warn that ACD ped file not available.

    Char_t msg[ 200 ];

    sprintf( msg, "ACD ped file \"%s\" not found. Program will now terminate.",(const char *) ACDPEDS );

    new  TGMsgBox( gui->fClient->GetRoot(), gui, "Warning", msg, 0, kMBOk );

    exit( 1 );

    }


  TObjArray *acdped = new TObjArray(2);

  acdped->InheritsFrom("TVector");

  TVector *acd_ped = new TVector(NACD);

  TVector *acd_id = new TVector(NACD);

  m_acdPedFile.open( (const char *)ACDPEDS , ios::in ); //from RootAnalysis v2r2
 
  for (Int_t j=0; j<NACD; j++) {   
    
    m_acdPedFile >> acd_id->operator()( j );

    m_acdPedFile >> acd_ped->operator()( j );
    
  }

  acdped->AddAt(acd_id,0);

  acdped->AddAt(acd_ped,1);

  cout << "===> Read ACD Peds" << endl;
  

  return(acdped) ;

}


TObjArray* EDControl::ReadXGTPed()

{
  // Read the XGT pedestal file.


  if ( gSystem->AccessPathName( (const char *)XGTPEDS, kFileExists ) ) {

    // Warn that XGT ped file not available.

    Char_t msg[ 200 ];

    sprintf( msg, "XGT ped file \"%s\" not found. Program will now terminate.",(const char *) XGTPEDS );

    new  TGMsgBox( gui->fClient->GetRoot(), gui, "Warning", msg, 0, kMBOk );

    exit( 1 );

    }

  
  TObjArray *xgtped = new TObjArray(2);

  xgtped->InheritsFrom("TVector");

  TVector *xgt_ped = new TVector(NXGT);
  
  TVector *xgt_id = new TVector(NXGT);

  m_xgtPedFile.open((const char *) XGTPEDS , ios::in ); //from RootAnalysis v2r2
 
  for (Int_t j=0; j<NXGT; j++) {   
    
    m_xgtPedFile >> xgt_id->operator()( j );

    m_xgtPedFile >> xgt_ped->operator()( j );

  }

  xgtped->AddAt(xgt_id,0);

  xgtped->AddAt(xgt_ped,1);

  cout << "===> Read XGT Peds" << endl;
  
  return(xgtped) ;

}


void EDControl::processCommand( Int_t userReq, Int_t evtNo )

{
  // Called from the GUI after a user request.  Parameter userReq specifies the
  // requested action.  Second parameter evtNo is the event number, required
  // for some requests.


  if ( gui->m_ModeIsDisplay ) {
    // Update threshold values for any action in display mode.
    gui->GetThresholds( m_threshold );
    display->setThresholds( m_threshold );
  }

  switch( userReq ) {

  case kReqDisplay:
    // Draw event display instead of analysis histos.
    m_canvas->Clear();
    // Display event incremented at every draw.  Must reset here to avoid drawing successive
    // events whenever switch between Display and Analysis modes.
    display->setCurrentEvent( display->getCurrentEvent() - 1 );
    UpdateGuiEvents( display->getCurrentEvent() );
    display->setDrawingPads();
    display->Draw( m_isRealData );
    break;
    

  case kReqAnalysis:
    // Switch to display of analysis histos.
    UpdateGuiEvents( analysis->getCurrentEvent() );
    m_canvas->Clear();
    m_histos->DrawCALHistos();
    break;

  case kReqRawFile:
    // Load a raw data file.

    setRawDataType( gui->m_isRealData );

    gui->SetModeDisplay();
    UpdateGuiEvents( 0 );

    ConnectNewRawFile( gui->m_RawFilePath );

    display->setCurrentEvent( 0 );
    display->setScanDirection( kForwards );
    display->Draw( m_isRealData );

    analysis->Rewind();   // Go to start of file.
    m_histos->HClr();     // Clear histos.

    break;

  case kReqReconFile:
    // Load a recon data file.

    gui->SetModeDisplay();
    UpdateGuiEvents( 0 );

    ConnectNewReconFile( gui->m_ReconFilePath );

    // Re-connect raw file.  Otherwise get recon tracks drawn without raw points for first event
    // ( number of tkr layers, nTKR, returned as zero - don't know why! )
    if ( gui->m_rawFileExists ) ConnectNewRawFile( gui->m_RawFilePath );

    display->setCurrentEvent( 0 );
    display->setScanDirection( kForwards );
    display->Draw( m_isRealData );

    analysis->Rewind();

    break;

  case kReqExit:
    gSystem->ExitLoop();
    break;
 
  case kReqDisplayNext:
    // Step to next event to be displayed.
    if ( display->getScanDirection() == kBackwards ) display->setCurrentEvent( display->getCurrentEvent() + 2 );
    display->setScanDirection( kForwards );
    if ( display->getCurrentEvent() > m_numEntries ) break;
    UpdateGuiEvents( display->getCurrentEvent() );
    display->Draw( m_isRealData );
    break;

  case kReqPrev:
    // Step back one event for display.
    if ( display->getScanDirection() == kForwards ) display->setCurrentEvent( display->getCurrentEvent() - 2 );
    display->setScanDirection( kBackwards );
    if ( display->getCurrentEvent() < 0 ) break;
    UpdateGuiEvents( display->getCurrentEvent() );
    display->Draw( m_isRealData );
    break;

  case kReqDisplayGoto:
    // Display specified event.
    display->setScanDirection( kForwards );
    display->Draw( evtNo, m_isRealData );
    UpdateGuiEvents( evtNo );
    break;

  case kReqAnalGoto:
    // Analyse a further specified number of events.
    if ( evtNo <= analysis->getCurrentEvent() || evtNo >= m_numEntries ) {
      UpdateGuiEvents( analysis->getCurrentEvent() );
      break;
    }
    analysis->Go( evtNo - analysis->getCurrentEvent() );
    break;

  case kReqCue:
    // Step through events and display them until told to stop.
    if ( display->getScanDirection() == kBackwards ) display->setCurrentEvent( display->getCurrentEvent() + 2 );
    display->setScanDirection( kForwards );

    while ( !m_StopDisplayLoop ) {
      if ( display->getCurrentEvent() > m_numEntries ) {
        gui->setInLoop( kFALSE );
        break;
      }
      UpdateGuiEvents( display->getCurrentEvent() );
      display->Draw( m_isRealData );
      gSystem->ProcessEvents();
    }

    m_StopDisplayLoop = kFALSE;

    break;

  case kReqAll:
    // Analyse all events ( user can stop at any intermediate point ).
    analysis->Go();
    break;

  case kReqReview:
    // Step backwards through events and display until told to stop.
    if ( display->getScanDirection() == kForwards ) display->setCurrentEvent( display->getCurrentEvent() - 2 );
    display->setScanDirection( kBackwards );

    while ( !m_StopDisplayLoop ) {
      if ( display->getCurrentEvent() < 0 ) {
        gui->setInLoop( kFALSE );
        break;
      }
      UpdateGuiEvents( display->getCurrentEvent() );
      display->Draw( m_isRealData );
      gSystem->ProcessEvents();
    }

    m_StopDisplayLoop = kFALSE;

    break;

  case kReqRewind:
    // Go back to start of file for analysis.
    analysis->Rewind();
    UpdateGuiEvents( 0 );
    m_histos->HClr();
    cout << "\nRewind... done.\n";
    break;

  case kReqDisplayStop:
    m_StopDisplayLoop = kTRUE;
    break;

  case kReqAnalStop:
    analysis->m_Stop = kTRUE;
    break;

  case kReqHistTKR:
    // Display tracker analysis histos.
    m_histos->DrawTKRHistos();
    break;

  case kReqHistTOTl:
    // Display tracker TOT left histos.
    m_histos->DrawTKRtotLHistos();
    break;

  case kReqHistTOTr:
    // Display tracker TOT right histos.
    m_histos->DrawTKRtotRHistos();
    break;

  case kReqHistCAL:
    // Display CAL analysis histos.
    m_histos->DrawCALHistos();
    break;

  case kReqHistACD:
    // Display ACD analysis histos.
    m_histos->DrawACDHistos();
    break;

  case kReqHistXGT:
    // Display XGT analysis histos.
    m_histos->DrawXGTHistos();
    break;

  default: break;

  }

}

---