PedStudy.cxx

Go to the documentation of this file.

#include "PedStudy.h"

#include <BeamDataUtil/BDDevices.h>
#include <BeamDataUtil/BDSwicPeds.h>
#include <HistMan/HistMan.h>
#include <RawData/RawBeamMonBlock.h>
#include <RawData/RawBeamMonHeaderBlock.h>
#include <RawData/RawBeamSwicData.h>
#include <RawData/RawDataBlock.h>
#include <RawData/RawRecord.h>

#include <TH2I.h>
#include <TSystem.h>
#include <TFile.h>
#include <TTree.h>

#include <MessageService/MsgService.h>
CVSID("$Id: PedStudy.cxx,v 1.3 2005/05/12 17:52:36 bv Exp $");

#include <list>
#include <sstream>
#include <fstream>
using namespace std;

PedStudy::PedStudy(const char* directories[])
{
    cerr << "PedStudy\n";

    this->GetFiles(directories);
    this->Book();
    this->Fill();
}

void PedStudy::GetFiles(const char* directories[])
{
    for (int ind=0; directories[ind]; ++ind) {
        void* dir = gSystem->OpenDirectory(directories[ind]);
        if (!dir) {
            MSG("BD",Msg::kWarning)
                << "Skipping empty directory: " << directories[ind] << endl;
            continue;
        }

        const char* cptr=0;
        while ( (cptr = gSystem->GetDirEntry(dir)) ) {
            string file = cptr;
            if (string::npos == file.find(".mbeam.root")) continue;
            fFileMap[file] = directories[ind];
        }
    }
}

void PedStudy::Book()
{
    int nfiles = fFileMap.size();
    HistMan hm("pedstudy");
    vector<string> swics = BDDevices::SwicDevices();
    for (size_t ind=0; ind<swics.size(); ++ind) {
        fHist[swics[ind]].all =
            hm.Book<TH2I>(swics[ind].substr(2).c_str(),
                          swics[ind].c_str(),
                          nfiles,0,nfiles,
                          96,0,96,"all");
    }
}

void PedStudy::Fill()
{
    map<string,string>::iterator it, done = fFileMap.end();
    list<string> files;
    for (it=fFileMap.begin(); it != done; ++it) files.push_back(it->first);
    files.sort();
    list<string>::iterator lit, ldone = files.end();

    int count = 0;
    for (lit = files.begin(); lit != ldone; ++lit) {
        string file = *lit;
        string dir = fFileMap[file];

        MSG("BD",Msg::kInfo) << dir << "/" << file << endl;
        this->OneFile(count,file,dir);
        ++count;
    }
}

void PedStudy::OneFile(int count, string filename, string dirname)
{
    HistMan hm("pedstudy");

    HistMap::iterator it, done=fHist.end();
    for (it=fHist.begin(); it != done; ++it) {
        string name = it->first.substr(2);
        string hmdir = name + "/" + filename;
        TProfile* hist = hm.Book<TProfile>(name.c_str(),hmdir.c_str(),
                                           96,0,96,hmdir.c_str());
        fHist[it->first].perfile.push_back(hist);
    }

    string rootfile = dirname + "/" + filename;
    TFile file(rootfile.c_str(),"READ");
    TTree* tree = (TTree*)(file.Get("BeamMon"));
    RawRecord* record = 0;
    for ( Int_t ient = 0; ient < tree -> GetEntries(); ient++ ) {
        tree -> SetBranchAddress("RawRecord",&record);
        tree->GetEntry(ient);
        TIter itr = record->GetRawBlockIter();

        const RawBeamMonBlock* rbmb = 0;;
        const RawBeamMonHeaderBlock* rbmhb = 0;

        // check all blocks
        RawDataBlock* rdb = 0;
        while ((rdb = dynamic_cast<RawDataBlock*>(itr()))) {
            if (rdb->InheritsFrom("RawBeamMonBlock"))
                rbmb = dynamic_cast<const RawBeamMonBlock*>(rdb);
            if (rdb->InheritsFrom("RawBeamMonHeaderBlock"))
                rbmhb = dynamic_cast<const RawBeamMonHeaderBlock*>(rdb);
        }

        if (!(rbmb && rbmhb)) continue;

        if (rbmb->TclkTriggerEvent() != 0xa9) continue;
        if (! BDSwicPeds::IsPedSpill(*rbmb)) continue;

        this->OneRecord(count,*rbmb);
    }
    delete record; record = 0;
}

void PedStudy::OneRecord(int count, const RawBeamMonBlock& block)
{
    map<string,Hists>::iterator it, done = fHist.end();
    RawBeamSwicData swic;
    for (it = fHist.begin(); it != done; ++it) {
        const RawBeamData* rbd = block[it->first];
        if (!rbd) continue;
        swic.SetData(*rbd);

        vector<int> wire;
        swic.UnscaledWireData(wire);
        for (size_t ind=0; ind<wire.size(); ++ind) {
            it->second.all->Fill(count,ind,wire[ind]);
            it->second.perfile.back()->Fill(ind,wire[ind]);

        }
    }
}



void PedStudy::WriteOut(const char* filename)
{
    HistMan hm("");
    hm.WriteOut(filename);
}

PedViewer* PedStudy::View()
{
    return new PedViewer(*this);
}


#include <TCanvas.h>

PedViewer::PedViewer(PedStudy& pedstudy)
    : ps(pedstudy)
{
    colz = new TCanvas("colz","",600,500);
    oned = new TCanvas("oned","",600,500);
    colz->AddExec("updater",Form("pedviewer_updater((void*)%d)",this));
    oned->AddExec("clicker",Form("pedviewer_clicker((void*)%d)",this));
    cur = ps.fHist.begin();

    this->Update();
        
}

void PedViewer::Next()
{
    PedStudy::HistMap::iterator next = cur;
    ++next;
    if (next == ps.fHist.end()) {
        cerr << "At end, looping to begin\n";
        cur = ps.fHist.begin();
    }
    else cur = next;
    this->Update();
}
void PedViewer::Prev()
{
    if (cur == ps.fHist.begin()) {
        cerr << "At begining, looping to end\n";
        cur = ps.fHist.end();
    }
    --cur;
    this->Update();
}

void PedViewer::Update()
{
    PedStudy::Hists& h = cur->second;
    if (!h.all) {
        cerr << "No histogram to update!\n";
        return;
    }

    colz->cd();
    h.all->Draw("COLZ");
}



void pedviewer_updater(void* ptr)
{
    int event = gPad->GetEvent();

    if (event != 51) return;    // movement

    // from root/tutorials/DynamicSlice
    TObject *select = gPad->GetSelected();
    if(!select) return;
    if (!select->InheritsFrom("TH2")) {gPad->SetUniqueID(0); return;}
    gPad->GetCanvas()->FeedbackMode(kTRUE);

    // Get hist X coords
    int px = gPad->GetEventX();
    Float_t upx = gPad->AbsPixeltoX(px);
    Float_t x = gPad->PadtoX(upx);

    PedViewer* viewer = (PedViewer*)ptr;
    PedStudy::Hists& hists = viewer->cur->second;

    static int lastbinx = 0;
    int binx = hists.all->GetXaxis()->FindBin(x);
    if (!binx) return;
    if (binx>(int)hists.perfile.size()) return;
    if (binx == lastbinx) return;
    lastbinx = binx;

    viewer->cur_file_num = binx-1;
    TProfile* h1 = hists.perfile[binx-1];
    viewer->oned->cd();
    cerr << binx << ": " << h1->GetName() << " " << h1->GetTitle() << endl;
    h1->Draw();
    viewer->oned->Update();
    //viewer->colz->cd();
}

void pedviewer_clicker(void* ptr)
{
    int event = gPad->GetEvent();

    if (event != 1) return;     // click

    // from root/tutorials/DynamicSlice
    TObject *select = gPad->GetSelected();
    if(!select) return;
    if (!select->InheritsFrom("TProfile")) {gPad->SetUniqueID(0); return;}
    gPad->GetCanvas()->FeedbackMode(kTRUE);

    // Get hist X coords
    int px = gPad->GetEventX();
    Float_t upx = gPad->AbsPixeltoX(px);
    Float_t x = gPad->PadtoX(upx);

    PedViewer* viewer = (PedViewer*)ptr;
    PedStudy::Hists& hists = viewer->cur->second;

    TProfile* h1 = hists.perfile[viewer->cur_file_num];
    int binx = h1->GetXaxis()->FindBin(x);
    if (!binx) return;
    if (binx>96) return;

    int channel = binx - 1;

    stringstream ss;

    ss << "E:" <<  h1->GetName()
       << " E:" << h1->GetTitle()
       << " " << channel
       << " " << h1->GetBinContent(binx)
       << ends;

    ofstream o("pedstudy.log",ios_base::app);
    o << ss.str() << endl;
    cerr << ss.str() << endl;
}

---