source: trunk/MagicSoft/Mars/mmain/MOnlineDisplay.cc@ 2576

/* ======================================================================== *\
!
! *
! * This file is part of MARS, the MAGIC Analysis and Reconstruction
! * Software. It is distributed to you in the hope that it can be a useful
! * and timesaving tool in analysing Data of imaging Cerenkov telescopes.
! * It is distributed WITHOUT ANY WARRANTY.
! *
! * Permission to use, copy, modify and distribute this software and its
! * documentation for any purpose is hereby granted without fee,
! * provided that the above copyright notice appear in all copies and
! * that both that copyright notice and this permission notice appear
! * in supporting documentation. It is provided "as is" without express
! * or implied warranty.
! *
!
!
!   Author(s): Thomas Bretz, 11/2003 <mailto:tbretz@astro.uni-wuerzburg.de>
!
!   Copyright: MAGIC Software Development, 2000-2003
!
!
\* ======================================================================== */
#include "MOnlineDisplay.h"

//
// C-lib
//
#include <stdlib.h>              // atoi

//
// root
//
#include <TList.h>               // TList::Add
#include <TStyle.h>              // gStyle->SetOptStat
#include <TCanvas.h>             // TCanvas::cd

//
// root GUI
//
#include <TGLabel.h>             // TGLabel
#include <TGButton.h>            // TGPictureButton
#include <TGSlider.h>            // TGSlider
#include <TG3DLine.h>            // TGHorizontal3DLine
#include <TGTextEntry.h>         // TGTextEntry
#include <TGButtonGroup.h>       // TGVButtonGroup
#include <TRootEmbeddedCanvas.h> // TRootEmbeddedCanvas

//
// General
//
#include "MGList.h"              // MGList

#include "MParList.h"            // MParList::AddToList
#include "MEvtLoop.h"            // MEvtLoop::GetParList
#include "MTaskList.h"           // MTaskList::AddToList

//
// Tasks
//
#include "MReadMarsFile.h"       // MReadMarsFile
#include "MGeomApply.h"          // MGeomApply
#include "MFDataMember.h"        // MFDataMember
#include "MMcPedestalCopy.h"     // MMcPedestalCopy
#include "MMcPedestalNSBAdd.h"   // MMcPedestalNSBAdd
#include "MCerPhotCalc.h"        // MCerPhotCalc
#include "MCerPhotAnal2.h"       // MCerPhotAnal2
#include "MImgCleanStd.h"        // MImgCleanStd
#include "MHillasCalc.h"         // MHillasCalc
#include "MHillasSrcCalc.h"      // MHillasSrcCalc
#include "MBlindPixelCalc.h"     // MBlindPixelCalc
#include "MFillH.h"              // MFillH
#include "MGTask.h"              // MGTask

//
// Container
//
#include "MHEvent.h"             // MHEvent
#include "MHCamera.h"            // MHCamera
#include "MRawEvtData.h"         // MRawEvtData

ClassImp(MOnlineDisplay);

// --------------------------------------------------------------------------
//
//  Constructor.
//
MOnlineDisplay::MOnlineDisplay() : MStatusDisplay(), fTask(0)
{
    //
    // Setup Task list for hillas calculation
    //
    //SetupTaskList("Events", filename);

    //
    // Add MOnlineDisplay GUI elements to the display
    //
    AddUserFrame("");

    //
    // Show new part of the window, resize to correct aspect ratio
    //
    // FIXME: This should be done by MStatusDisplay automatically
    Resize(GetWidth(), GetHeight() + fUserFrame->GetDefaultHeight());
    //SetWindowName("Online Display");
    MapSubwindows();
    //MapWindow();

    //
    // Readin foirst event and display it
    //
    //ReadFirstEvent();
}

// --------------------------------------------------------------------------
//
//  Destructor: PostProcess eventloop, delete eventloop with all containers
//
MOnlineDisplay::~MOnlineDisplay()
{
    //fEvtLoop->PostProcess();
    //delete fEvtLoop;
}

// --------------------------------------------------------------------------
//
//  Return reading task
//
/*
MReadTree *MOnlineDisplay::GetReader() const
{
    MParList  *plist  = (MParList*)fEvtLoop->GetParList();
    MTaskList *tlist  = (MTaskList*)plist->FindObject("MTaskList");
    MReadTree *reader = (MReadTree*)tlist->FindObject("MRead");
    return reader;
}

// --------------------------------------------------------------------------
//
//  Setup Task and parameter list for hillas calculation,
//  preprocess tasks and read in first event (process)
//
void MOnlineDisplay::SetupTaskList(const char *tname, const char *fname)
{
    //
    // Setup an empty job, with a reader task only.
    // All tasks and parameter containers are deleted automatically
    // (via SetOwner())
    //
    MTaskList *tlist = new MTaskList;
    tlist->SetOwner();

    MReadMarsFile *read = new MReadMarsFile(tname, fname);
    read->DisableAutoScheme();
    tlist->AddToList(read);

    MGeomApply *apl = new MGeomApply;
    tlist->AddToList(apl);

    MParList *plist = new MParList;
    plist->SetOwner();
    plist->AddToList(tlist);

    fEvtLoop = new MEvtLoop;
    fEvtLoop->SetOwner();
    fEvtLoop->SetParList(plist);

    MHEvent *evt1 = new MHEvent(MHEvent::kEvtSignal);
    MHEvent *evt2 = new MHEvent(MHEvent::kEvtSignal);
    MHEvent *evt3 = new MHEvent(MHEvent::kEvtPedestal);
    MHEvent *evt4 = new MHEvent(MHEvent::kEvtPedestalRMS);
    MHEvent *evt5 = new MHEvent(MHEvent::kEvtRelativeSignal);
    MHEvent *evt6 = new MHEvent(MHEvent::kEvtCleaningLevels);
    evt1->SetName("Signal");
    evt2->SetName("Cleaned");
    evt3->SetName("Pedestal");
    evt4->SetName("PedRMS");
    evt5->SetName("Signal/PedRMS");
    evt6->SetName("CleanLevels");
    plist->AddToList(evt1);
    plist->AddToList(evt2);
    plist->AddToList(evt3);
    plist->AddToList(evt4);
    plist->AddToList(evt5);
    plist->AddToList(evt6);

    MMcPedestalCopy   *pcopy = new MMcPedestalCopy;
    MMcPedestalNSBAdd *pdnsb = new MMcPedestalNSBAdd;
    MCerPhotCalc      *ncalc = new MCerPhotCalc;
    MCerPhotAnal2     *nanal = new MCerPhotAnal2;
    MFillH            *fill1 = new MFillH(evt1, "MCerPhotEvt",  "MFillH1");
    MImgCleanStd      *clean = new MImgCleanStd;
    MFillH            *fill2 = new MFillH(evt2, "MCerPhotEvt",  "MFillH2");
    MFillH            *fill3 = new MFillH(evt3, "MPedestalCam", "MFillH3");
    MFillH            *fill4 = new MFillH(evt4, "MPedestalCam", "MFillH4");
    MFillH            *fill5 = new MFillH(evt5, "MCameraData",  "MFillH5");
    MFillH            *fill6 = new MFillH(evt6, "MCameraData",  "MFillH6");
    MBlindPixelCalc   *blind = new MBlindPixelCalc;
    MHillasCalc       *hcalc = new MHillasCalc;
    MHillasSrcCalc    *scalc = new MHillasSrcCalc;

    MFilter *f1=new MFDataMember("MRawRunHeader.fRunType", '>', 255.5);
    MFilter *f2=new MFDataMember("MRawRunHeader.fRunType", '<', 255.5);

    ncalc->SetFilter(f1);
    nanal->SetFilter(f2);

    tlist->AddToList(f1);
    tlist->AddToList(f2);
    tlist->AddToList(pcopy);
    tlist->AddToList(pdnsb);
    tlist->AddToList(ncalc);
    tlist->AddToList(nanal);
    tlist->AddToList(fill1);
    tlist->AddToList(clean);
    tlist->AddToList(fill2);
    tlist->AddToList(fill3);
    tlist->AddToList(fill4);
    tlist->AddToList(fill5);
    tlist->AddToList(fill6);
    tlist->AddToList(blind);
    tlist->AddToList(hcalc);
    tlist->AddToList(scalc);

    //
    // Now distribute Display to all tasks
    //
    tlist->SetDisplay(this);
}
*/
// --------------------------------------------------------------------------
//
//  Add the top part of the frame: This is filename and treename display
//
void MOnlineDisplay::AddTopFramePart1(TGCompositeFrame *vf1)
{/*
    //
    //  --- the top1 part of the window ---
    //
    TGHorizontalFrame *top1 = new TGHorizontalFrame(frame, 1, 1);
    fList->Add(top1);

    //
    // create gui elements
    //
    TGLabel *file = new TGLabel(top1, new TGString(Form("%s#%s", filename, treename)));
    fList->Add(file);

    //
    // layout and add gui elements in/to frame
    //
    TGLayoutHints *laystd = new TGLayoutHints(kLHintsCenterX, 5, 5);
    fList->Add(laystd);

    top1->AddFrame(file,  laystd);

    //
    //  --- the top1 part of the window ---
    //
    TGHorizontalFrame *top2 = new TGHorizontalFrame(frame, 1, 1);
    fList->Add(top2);

    //
    // layout and add frames
    //
    TGLayoutHints *laytop1 = new TGLayoutHints(kLHintsCenterX, 5, 5, 5);
    fList->Add(laytop1);
    frame->AddFrame(top1, laytop1);
    frame->AddFrame(top2, laytop1);*/
    TGLabel *file = new TGLabel(vf1, new TGString("Magic Online Display -- MONA"));
    TGLayoutHints *laystd = new TGLayoutHints(kLHintsCenterX, 5, 5);
    fList->Add(file);
    fList->Add(laystd);
    vf1->AddFrame(file,  laystd);
}

// --------------------------------------------------------------------------
//
//  Add the second part of the top frame: This are the event number controls
//
void MOnlineDisplay::AddTopFramePart2(TGCompositeFrame *vf1)
{
    //
    // --- the top2 part of the window ---
    //
    TGHorizontalFrame *top2 = new TGHorizontalFrame(vf1, 1, 1);
    fList->Add(top2);

    //
    // Create the gui elements
    //
    TGTextButton *freeze = new TGTextButton(top2, " Freeze ", kFreeze);
    freeze->SetUserData(freeze);
    freeze->Associate(this);
    freeze->SetToolTipText("Freeze the current Event");

    fList->Add(freeze);

    //
    // add the gui elements to the frame
    //
    TGLayoutHints *laystd = new TGLayoutHints(kLHintsLeft|kLHintsCenterY, 5, 5);
    fList->Add(laystd);

    top2->AddFrame(freeze, laystd);

    TGLayoutHints *laystd2 = new TGLayoutHints(kLHintsCenterX, 5, 5, 5, 5);
    fList->Add(laystd2);
    vf1->AddFrame(top2, laystd2);

    //
    // Add trailing line...
    //
    TGHorizontal3DLine *line = new TGHorizontal3DLine(vf1);
    TGLayoutHints *layline = new TGLayoutHints(kLHintsExpandX);
    fList->Add(line);
    fList->Add(layline);
    vf1->AddFrame(line, layline);
}

// --------------------------------------------------------------------------
//
//  Add the user frame part of the display
//
void MOnlineDisplay::AddUserFrame(const char* filename)
{
    fUserFrame->ChangeOptions(kHorizontalFrame);

    // Slider
    TGVSlider *slider = new TGVSlider(fUserFrame, 1, kSlider1, kSlider);
    slider->SetRange(10, 300);
    slider->SetScale(15);
    slider->SetPosition(10);
    slider->Associate(this);
    //slider->SetToolTipText("Change the update frequency of the event display (1-30s)");

    // frame1/2
    TGCompositeFrame *vf1 = new TGVerticalFrame(fUserFrame, 1, 1);
    TGCompositeFrame *vf2 = new TGVerticalFrame(fUserFrame, 1, 1);

    AddTopFramePart1(vf1);
    AddTopFramePart2(vf1);

    // create root embedded canvas and add it to the tab
    TRootEmbeddedCanvas *ec = new TRootEmbeddedCanvas("Slices", vf2, vf1->GetDefaultHeight()*3/2, vf1->GetDefaultHeight(), 0);
    vf2->AddFrame(ec);
    fList->Add(ec);

    // set background and border mode of the canvas
    fCanvas = ec->GetCanvas();
    fCanvas->SetBorderMode(0);
    gROOT->GetListOfCanvases()->Add(fCanvas);
    //fCanvas->SetBorderSize(1);
    //fCanvas->SetBit(kNoContextMenu);
    //fCanvas->SetFillColor(16);

    // layout
    TGLayoutHints *lays = new TGLayoutHints(kLHintsExpandY);
    TGLayoutHints *lay = new TGLayoutHints(kLHintsExpandX);
    fUserFrame->AddFrame(slider, lays);
    fUserFrame->AddFrame(vf1, lay);
    fUserFrame->AddFrame(vf2);

    fList->Add(lay);
    fList->Add(lays);
    fList->Add(vf1);
    fList->Add(vf2);
    fList->Add(slider);
}

/*
// --------------------------------------------------------------------------
//
//  Checks if the event number is valid, and if so reads the new event
//  and updates the display
//
void MOnlineDisplay::ReadinEvent(Int_t dir)
{
    MParList    *plist = (MParList*)fEvtLoop->GetParList();
    MTaskList   *tlist = (MTaskList*)plist->FindObject("MTaskList");
    MRawEvtData *raw = (MRawEvtData*)plist->FindObject("MRawEvtData");
    if (!raw)
        return;

    //
    // Read first event.
    //
    MReadTree *reader = GetReader();

    const Int_t num = reader->GetNumEntry();

    do
    {
        if (dir<0 && !reader->DecEventNum())
        {
            reader->SetEventNum(num);
            return;
        }
        if (dir>0 && !reader->IncEventNum())
        {
            reader->SetEventNum(num);
            return;
        }

        if (!tlist->Process())
            return;

        reader->DecEventNum();

    } while (raw->GetNumPixels()<1 && dir!=0);

    //
    // Cleare the 'FADC canvas'
    //
    fCanvas->Clear();
    fCanvas->Modified();
    fCanvas->Update();

    //
    // Print parameters
    //
    plist->FindObject("MHillas")->Print();
    plist->FindObject("MHillasExt")->Print();
    plist->FindObject("MHillasSrc")->Print();
    plist->FindObject("MNewImagePar")->Print();

    //
    // UpdateDisplay
    //
    gStyle->SetOptStat(1101);
    Update();

    TGTextEntry *entry = (TGTextEntry*)fList->FindWidget(kEvtNumber);
    entry->SetText(Form("%d", reader->GetNumEntry()+1));
}

// --------------------------------------------------------------------------
//
//  Read first event to get display booted
//
void MOnlineDisplay::ReadFirstEvent()
{
    if (!fEvtLoop->PreProcess())
        return;

    //
    // Get parlist
    //
    MParList *plist = (MParList*)fEvtLoop->GetParList();

    //
    // Add Geometry tab
    //
    MGeomCam *geom = (MGeomCam*)plist->FindObject("MGeomCam");

    TCanvas &c=AddTab("Geometry");

    c.SetFillColor(16);

    MHCamera *cam = new MHCamera(*geom);
    cam->SetBit(TH1::kNoStats|MHCamera::kNoLegend);
    cam->Draw();
    cam->DrawPixelIndices();
    fList->Add(cam);

    c.Modified();
    c.Update();

    //
    // Now read event...
    //
    ReadinEvent();

    TGString *txt = new TGString(Form("of %d", GetReader()->GetEntries()));
    fNumOfEvts->SetText(txt);

    //
    // Set name for 'FADC canvas'. The name is the anchor for MHCamera.
    // and clear the canvas
    //
    MHEvent *o = (MHEvent*)plist->FindObject("Signal");
    fCanvas->SetName(Form("%p;%p;PixelContent", o->GetHist(),
                          GetCanvas(1)->GetPad(1)));

    //
    // Draw ellipse on top of all pads
    //
    TObject *hillas = plist->FindObject("MHillas");
    for (int i=1; i<7;i++)
    {
        TCanvas *c = GetCanvas(i);
        c->GetPad(1)->cd(1);
        hillas->Draw();
    }
}
*/
// --------------------------------------------------------------------------
//
//  Adds the geometry tab 
//
void MOnlineDisplay::AddGeometryTab()
{
    /*
    MGeomCam *geom = (MGeomCam*)fEvtLoop->GetParList()->FindObject("MGeomCam");
    if (!geom)
        return;

    TCanvas &c=AddTab("Geometry");

    MHCamera *cam = new MHCamera(*geom);
    cam->SetBit(TH1::kNoStats);
    cam->Draw();
    cam->DrawPixelIndices();
    fList->Add(cam);

    c.Modified();
    c.Update();
    */
}

// --------------------------------------------------------------------------
//
//  ProcessMessage(Long_t msg, Long_t parm1, Long_t parm2)
//
// Processes information from all GUI items.
// Selecting an item usually generates an event with 4 parameters.
// The first two are packed into msg (first and second bytes).
// The other two are parm1 and parm2.
//
Bool_t MOnlineDisplay::ProcessMessage(Long_t msg, Long_t mp1, Long_t mp2)
{
    if (fTask && fTask->ProcessMessage(GET_MSG(msg), GET_SUBMSG(msg), mp1, mp2))
        return kTRUE;

    return MStatusDisplay::ProcessMessage(msg, mp1, mp2);
}