---

EDControl.cxx

Go to the documentation of this file.

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


#include "bfemDisplay/EDControl.h"
#include "bfemDisplay/EDGui.h"
#include "bfemDisplay/EDAnalysis.h"
#include "bfemDisplay/DisplayCanvas.h"
#include "bfemDisplay/Enum.h"


EDAnalysis *analysis;
EDGui *gui;
DisplayCanvas *display;


EDControl::EDControl() {

  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(), 600, 420, this );

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

  if ( gui->m_rawFileExists ) ConnectNewRawFile( gui->m_RawFilePath );
  if ( gui->m_reconFileExists ) ConnectNewReconFile( gui->m_ReconFilePath );

  // 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();

  m_ped = ReadCALPed();
  m_acdped = ReadACDPed();
  //  m_xgtped = ReadXGTPed();

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

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

}


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 ) {

  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();

}

void EDControl::ConnectNewReconFile( Char_t *fname ) {

  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();

}


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;

}

TObjArray* EDControl::ReadCALPed()

{

  // Reading CAL pedfile

  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.", CALPEDS_AUG2001 );

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

    return 0;

  }
  
  TObjArray *ped = new TObjArray(8);

  ped->InheritsFrom("TVector");

  //TFile *pedFile = new TFile( CALPEDS_MAY2001 ); //Berrie's Peds -  May 2001
  TFile *pedFile = new TFile( 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()

{

  // Reading ACD pedfile

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

    // Warn that ACD ped file not available.

    Char_t msg[ 200 ];

    sprintf( msg, "ACD ped file \"%s\" not found.", ACDPEDS );

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

    return 0;

    }


  TObjArray *acdped = new TObjArray(2);

  acdped->InheritsFrom("TVector");

  TVector *acd_ped = new TVector(NACD);

  TVector *acd_id = new TVector(NACD);

  m_acdPedFile.open( 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()

{

  // Reading XGT pedfile

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

    // Warn that XGT ped file not available.

    Char_t msg[ 200 ];

    sprintf( msg, "XGT ped file \"%s\" not found.", XGTPEDS );

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

    return 0;

    }

  
  TObjArray *xgtped = new TObjArray(2);

  xgtped->InheritsFrom("TVector");

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

  m_xgtPedFile.open( 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 ) {


  switch( userReq ) {

  case kReqDisplay:
    // Draw event display instead of analysis histos.
    m_canvas->Clear();
    display->setDrawingPads();
    display->Draw( m_isRealData );
    break;
    

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

  case kReqRawFile:
    // Load a raw data file.

    setRawDataType( gui->m_isRealData );

    gui->SetModeDisplay();

    ConnectNewRawFile( gui->m_RawFilePath );

    /*    sprintf( msg, "File \"%s\" scheduled for addition.\nProgram will now terminate; new file will be connected at next startup.", gui->m_RawFilePath );

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

    gSystem->ExitLoop();*/

    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();

    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 );

    /*    sprintf( msg, "File \"%s\" scheduled for addition.\nProgram will now terminate; new file will be connected at next startup.", gui->m_ReconFilePath );

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

    gSystem->ExitLoop();*/

    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 );
    display->Draw( m_isRealData );
    break;

  case kReqAnalNext:
    // Analyse one more event.
    analysis->Go( 1 );
    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;
    display->Draw( m_isRealData );
    break;

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

  case kReqAnalGoto:
    // Analyse a further specified number of events.
    analysis->Go( evtNo );
    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 ) break;
      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 ) break;
      display->Draw( m_isRealData );
      gSystem->ProcessEvents();
    }

    m_StopDisplayLoop = kFALSE;

    break;

  case kReqRewind:
    // Go back to start of file for analysis.
    analysis->Rewind();
    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;

  }

}

---