stat/trid.cxx

Go to the documentation of this file.

#include <cassert>

#include "TROOT.h"
#include "TRint.h"
#include "TString.h"
#include "TSystem.h"
#include "TStyle.h"

#include "sigc++/sigc++.h"
#include "sigc++/class_slot.h"
#include "sigc++/slot.h"

#include "Calibrator/Calibrator.h"
#include "MessageService/MsgService.h"
#include "JobControl/JobC.h"
#include "JobControl/JobCModule.h"
#include "JobControl/JobCModuleRegistry.h" // For JOBMODULE macro
#include "Algorithm/AlgFactory.h"
#include "Algorithm/AlgConfig.h"

#include "Midad/Base/Mint.h"
#include "Midad/Base/PageDisplay.h"

#include "TriD/TridControl.h"
#include "TriD/TridPageDisplay.h"
#include "TriD/TridSetup.h"
#include "TriD/stat/ReadDispatcherModule.h"

#include "Plex/PlexLoanPool.h"
#include "DatabaseInterface/DbiTableProxyRegistry.h"

extern void InitGui(); // Initializer for GUI
VoidFuncPtr_t initfuncs[] = { InitGui, 0 };
CVSID("$Id: trid.cxx,v 1.16 2007/03/01 16:59:43 rhatcher Exp $");

TROOT root("trid", "TriD Static Viewer Program",initfuncs);


int SetupInput();
int SetupAuto();
int SetupAnimation();
void setup_paths(void) ;
int main(int argc, char **argv);

void DoNext(void);
void DoPrev(void);

typedef enum EInputMode {
  kFromFile = 0,
  kFromDispatcher = 1
} InputMode_t;


// Global variables.
InputMode_t gInputMode = kFromFile;
JobC* jc;
SigC::Ptr<TridControl> tc; // Get a sigC ptr to make sure that the TC object doesn't disappear.

string hostname = "localhost";
int    port     = 9090;
float  timeout  = 1;
bool   autoNext = false;
float  autoNextDelay = 30.0;
vector<string> filenames;
bool   animate = false;
float  animateSpeed = 0.1;
int    nhitThresh = 30;
int    nhitMax = 2000;
string plexfile = "~/cache/plex_cache.dat";

int SetupInput()
{
  assert(jc);
  
  if(gInputMode==kFromDispatcher) {
    cout << "Setting up dispatcher client connected to " 
         << hostname << " :" << port << endl;

    /*
    jc->Input.Set("Format=DDS");
        jc->Input.Set(Form("DDSServer=%s",hostname.c_str())); // dispatcher server host
    jc->Input.Set(Form("DDSPort=%d",port));  // default dispatcher server port is 9090
    jc->Input.Set("Streams=DaqSnarl"); //default is DS,DM,LightInjection
    jc->Input.Set("DDSKeepUpMode=RecordKeepUp");  // default is FileKeepUp
    jc->Input.Set("DDSDataSource=Daq");  // default is Daq
    jc->Input.Set(Form("DDSTimeOut=%d",(int)timeout));      // (seconds) default is 120
    jc->Input.Set("DDSMaxSyncDelay=1");      // (seconds) default is 15    
    jc->Input.Select("DaqSnarl",
                     Form("(((RawDaqSnarlHeader*)fHeader)->GetNumRawDigits()>%d)"
                          "&&(((RawDaqSnarlHeader*)fHeader)->GetNumRawDigits()<%d)",
                          nhitThresh,nhitMax));
    
    */
    // Add a dummy input method to save on warnings.
    jc->Input.Use("DummyInput");

    // Add my own input
    jc->Path.Add("default","ReadDispatcherModule::Get ");
    JobCModule& mod = jc->Path("default").Mod("ReadDispatcherModule");
    mod.Set(Form("DDSServer=%s",hostname.c_str())); // dispatcher server host
    mod.Set(Form("DDSPort=%d",port));  // default dispatcher server port is 9090
    mod.Set("Streams=DaqSnarl"); //default is DS,DM,LightInjection
    mod.Set("DDSKeepUpMode=RecordKeepUp");  // default is FileKeepUp
    mod.Set("DDSDataSource=Daq");  // default is Daq
    mod.Set(Form("DDSTimeOut=%d",(int)timeout));      // (seconds) default is 120
    mod.Set("DDSMaxSyncDelay=1");      // (seconds) default is 15    
    mod.Set(Form("SelectRule=(((RawDaqSnarlHeader*)fHeader)->GetNumRawDigits()>%d)"
                 "&&(((RawDaqSnarlHeader*)fHeader)->GetNumRawDigits()<%d)",
                 nhitThresh,nhitMax));
    

  } else {
   jc->Input.Set("Streams=DaqSnarl,Cand");
        
    MSG("TriD",Msg::kDebug) << "Adding files..\n";
    for(UInt_t i=0;i<filenames.size();i++) {
      jc->Input.AddFile(filenames[i].c_str());
      MSG("TriD",Msg::kDebug) << "  Adding file " << filenames[i] << endl;
    }
    
  }

  return 1;
}


class AutoEventTimer : public TTimer
{
  Bool_t Notify() {
    cout << "Auto-advancing..." << endl;
    gMint->Next();
    Start((Long_t)(autoNextDelay*1000.),true);
    return true;
  };
};

int SetupAuto(void)
{
  AutoEventTimer* timer = new AutoEventTimer();
  timer->Start((Long_t)(autoNextDelay*1000.),true);
  timer->TurnOn();
  return 0;
}


int SetupAnimation(void)
{
  assert(tc);
  tc->GetAnimator().SetAnimateViews(true);
  tc->GetAnimator().SetAnimateModels(false);
  tc->GetAnimator().SetViewAnimSpeed( animateSpeed );
  tc->GetAnimator().StartAnimation();
  return 0;
}


void DoNext(void)
{
  // Called when the user hits the 'next' button.
  cout << "DoNext()" << endl;

  
  if(gInputMode==kFromDispatcher) {
    // Get the module.
    ReadDispatcherModule* rdm = dynamic_cast<ReadDispatcherModule*>  
      (&(jc->Path("default").Mod("ReadDispatcherModule")));

    assert(rdm);
            
    if(rdm->IsNewEventReady()) gMint->Next();
    else {
      MSG("TriD",Msg::kInfo) << "No event ready from dispatcher." << endl;
    }
  } else 
  
  {
    // Regualar old file
    gMint->Next();
  }
}

void DoPrev(void)
{
  cout << "DoPrev()" << endl;
  if(gInputMode==kFromDispatcher) {
  } else {
    gMint->Prev();
  }
}

void setup_paths(void) 
{
  // Only need this for interactive jobs.
  MSG("TriD",Msg::kDebug) << "Setting up macro paths.\n";  cout.flush();

  TString mp = gROOT->GetMacroPath();
  TString ip;
  const char* p = gSystem->Getenv("SRT_PRIVATE_CONTEXT");
  if (p) {
    mp += ":";
    mp += p;
    mp += ":";
    mp += p;
    mp += "/macros";
    ip += " -I";
    ip += p;
  }
  
  const char* p2 = gSystem->Getenv("SRT_PUBLIC_CONTEXT");
  if (p2) {
    mp += ":";
    mp += p2;
    mp += ":";
    mp += p2;
    mp += "/macros";
    ip += " -I";
    ip += p2;
  }
  
  gROOT->SetMacroPath(mp.Data());
  gSystem->SetIncludePath(ip);
  
   MSG("TriD",Msg::kDebug) << "Setting up include paths.\n";  cout.flush();  

  if(p) {
    TString dip = ".include ";
    dip += p;
    gROOT->ProcessLine(dip.Data());
  }

  if(p2) {
    TString dip = ".include ";
    dip += p2;
    gROOT->ProcessLine(dip.Data());
  }
}



int main(int argc, char **argv)
{
  //
  // MAIN PROGRAM.
  //
  MsgService::Instance()->GetStream("TriD")->SetLogLevel(Msg::kInfo);

  // Configuration: read the command line.
  
  int iarg = 0;
  while(++iarg<argc) {
    // Switches.
    if(argv[iarg][0] == '-') {
      
      if((strcmp(argv[iarg],"-d")==0) || (strncmp(argv[iarg],"--disp",6)==0) ) {
        gInputMode = kFromDispatcher;   
        
        if( (iarg+1<argc) && (argv[iarg+1][0]!='-'))   hostname = argv[++iarg];
        if( (iarg+1<argc) && (argv[iarg+1][0]!='-')) {
          int tmp = 0;
          if(sscanf(argv[++iarg],"%d",&tmp)>0) port = tmp;
        }
        if( (iarg+1<argc) && (argv[iarg+1][0]!='-')) {
          float tmp = 0;
          if(sscanf(argv[++iarg],"%f",&tmp)>0) timeout = tmp;
        }
        cout << "Config: Using dispatcher: " << hostname << ":" << port 
             << " timeout: " << timeout << endl;
        continue;
      }

      
      if( (strcmp(argv[iarg],"-a")==0) || (strncmp(argv[iarg],"--auto",6)==0)) {
        autoNext = true;
        if( (iarg+1<argc) && (argv[iarg+1][0]!='-'))  {
          float tmp;
          if(sscanf(argv[++iarg],"%f",&tmp)>0) autoNextDelay = tmp;
        }
        cout << "Config: Using Auto Timer: " << autoNextDelay << " s" << endl;
        continue;
      }

      if( (strcmp(argv[iarg],"-n")==0) || (strncmp(argv[iarg],"--nhits",6)==0)) {

        if( (iarg+1<argc) && (argv[iarg+1][0]!='-'))  
          sscanf(argv[++iarg],"%d",&nhitThresh);

        if( (iarg+1<argc) && (argv[iarg+1][0]!='-'))  
          sscanf(argv[++iarg],"%d",&nhitMax);
        cout << "Using NHIT threshold of : " << nhitThresh 
             << " maximum NHIT " << nhitMax << endl;
        continue;
      }

      if( (strcmp(argv[iarg],"-r")==0) || (strncmp(argv[iarg],"--rot",5)==0)) {
        animate = true;
        if( (iarg+1<argc) && (argv[iarg+1][0]!='-'))  {
          float tmp;
          if(sscanf(argv[++iarg],"%f",&tmp)>0) animateSpeed = tmp;
        }
        cout << "Config: Turning on animation " << animateSpeed << " s" << endl;
        continue;
      }

      if( (strcmp(argv[iarg],"-D")==0) || (strncmp(argv[iarg],"--debug",7)==0)) {
        cout << "Config: Turning on debugging output." << endl;
        MsgService::Instance()->GetStream("TriD")->SetLogLevel(Msg::kDebug);
        continue;
      }

      if( (strcmp(argv[iarg],"-h")==0) || (strncmp(argv[iarg],"--help",6)==0)) {
        cout << "--- TriD Help ----- " << endl;
        cout << "Options: " << endl;
        cout << "  -d <host> <port>    Open dispatcher for input from host:port" << endl;
        cout << "  -n <nhits>          NHit threshold for dispatcher." << endl;
        cout << "  -a <time>           Auto-next-event every <time> seconds" << endl;
        cout << "  -r <speed>          Rotate view at <speed> rotations/second" << endl;
        cout << "  -D --debug          Turn on debugging output." << endl;

      }

    }
    // No switch, it's a filename.
    filenames.push_back(argv[iarg]);
  }



  int dummy =1;
  //TRint *theApp = new TRint("trid", &dummy, argv, NULL, 0);
  TApplication* theApp = new TApplication("PMT App",&dummy,argv);

  setup_paths(); // Needed to find the stupid Algorithm macros.


  MSG("TriD",Msg::kDebug) << "Creating job path.\n";
 
  jc = new JobC;
 
  MSG("TriD",Msg::kDebug) << "Loading plex cache from filename: " << plexfile << "\n";
  PlexLoanPool::ReadFromFile(plexfile.c_str());

  MSG("TriD",Msg::kDebug) << "Setting up DBI cache:\n";
  CfgConfigurable& dbiCfg = DbiTableProxyRegistry::Instance();
  dbiCfg.Set("Level2Cache = '~/cache/'");
  dbiCfg.Update();


  MSG("TriD",Msg::kDebug) << "Setting up job path\n";
 
  // Create the job paths.
  // Three paths, set up so demux gets done only if neccessary,
  // and digits get made only if neccessary.
  //
  // default -> getdigits -> demux -> <display>
  //     |                     |
  //     +---------------------+
  jc->Path.Create("default","");  
 // Find out our input source.
  SetupInput();
  jc->Path.Add("default","FilterEvent::Ana ");

  jc->Path.Create("getdigits",
                 "TestDigitsModule::Ana "
                 "DigitListModule::Get "
                 "DigitListModule::Reco "
                  "DumpMomModule::Ana "
                 );
  jc->Path("getdigits").Node("TestDigitsModule::Ana ").ReverseFilter(); // pass events with no digits, so digits get built.


  jc->Path.Create("demux",
                 "TestDemuxModule::Ana "
                 "DeMuxDigitListModule::Reco "
                  //"DemuxFast::Reco " 
                  "DumpMomModule::Ana "
                 );
  jc->Path("demux").Node("TestDemuxModule::Ana ").ReverseFilter(); // pass events with no digits, so digits get built.
  
  jc->Msg.SetLevel("AltDeMuxModule","Error");
  jc->Msg.SetLevel("AlgAltDeMux","Error");
  jc->Msg.SetLevel("AlgAltDeMuxDevel","Error");

  // Configuration:
  // Turn off calibration.
  Calibrator::Instance().Set("TimeCalibrator=SimpleCalScheme "
                             "PeCalibrator=SimpleCalScheme "
                             "DriftCalibrator=SimpleCalScheme "
                             "LinCalibrator=SimpleCalScheme "
                             "StripCalibrator=SimpleCalScheme "
                             "AttenCalibrator=SimpleCalScheme "
                             "MIPCalibrator=SimpleCalScheme "
                             "Thermometer=SimpleCalScheme " );


  jc->Path.Attach("default","getdigits");
  jc->Path.Attach("getdigits","demux");
   

  // Make it look pretty.
  gStyle->SetPalette(1,0);

  TridSetup(*jc);
  
  // Insert custom hooks
  // This allows the next and back buttons to be 'smart' and
  // not attempt to do things the dispatcher can't do.
  dynamic_cast<TridPageDisplay*>(gPageDisplay)->fNextSignal.clear();
  dynamic_cast<TridPageDisplay*>(gPageDisplay)->fPrevSignal.clear();
  dynamic_cast<TridPageDisplay*>(gPageDisplay)->fNextSignal.connect(SigC::slot(&DoNext));
  dynamic_cast<TridPageDisplay*>(gPageDisplay)->fPrevSignal.connect(SigC::slot(&DoPrev));

  // Get a sigC ptr to make sure that the TC object doesn't disappear.
  tc = TridControl::Instance(gPageDisplay);

  MSG("TriD",Msg::kDebug) << "Running job.\n";

 
  jc->Path("default").Report();

  // Set up auto display timer
  if(autoNext) {
    MSG("TriD",Msg::kDebug) << "Starting Auto-timer\n";
    SetupAuto();
  }

  if(animate) {  
    MSG("TriD",Msg::kDebug) << "Starting Animation\n";
    SetupAnimation();
  }
   gMint->Next();

  // Manually do an autozoom to make it look right.
  gPageDisplay->AutoZoom();

  // Run interactive interface
  theApp->Run();

  return 0;

}

---