Bdnt.cxx

Go to the documentation of this file.

#include "Bdnt.h"

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

#include <Validity/VldContext.h>
#include <RawData/RawRecord.h>
#include <RawData/RawBeamMonHeader.h>
#include <RawData/RawBeamMonBlock.h>
#include <RawData/RawDataBlock.h>
#include <RawData/RawBeamData.h>
#include <RawData/RawBeamSwicData.h>
#include <RawData/RawBeamPosData.h>

#include <cassert>
#include <iostream>
#include <vector>
#include <string>
#include <map>

using namespace std;

struct BdntImp {
    TFile* file;
    TTree* tree;
    typedef map<string,SwicBranch*> SwicMap;
    SwicMap swics;
    typedef map<string,BpmBranch*> BpmMap;
    BpmMap bpms;
    TriggerBranch *trig;
    string trigname;

    bool Fill(const RawBeamMonBlock* rbmb);
};

bool BdntImp::Fill(const RawBeamMonBlock* rbmb)
{
#if 0
    vector<string> namelist = rbmb->GetNames();
    for (size_t ind=0; ind<namelist.size(); ++ ind)
        cout << namelist[ind] << " ";
    cout << endl;
#endif

    // fill trigger info 
    TriggerBranch* trig_br = trig;
    memset(trig_br,0,sizeof(TriggerBranch));

    const int delay = rbmb->TclkTriggerDelay();
    if (delay < 1000)
        trig_br->in_spill = 1;
    else trig_br->in_spill = 0;

    const RawBeamData* rbd = (*rbmb)[trigname];
    if (rbd) {
        trig_br->sec = rbd->GetSeconds();
        trig_br->msec = rbd->GetMsecs();
        trig_br->ts = (1.0*trig_br->sec) + (trig_br->msec/1000.0);
        if (rbd->GetDataLength())
            trig_br->timesince = rbd->GetData()[0];
        else
            cerr << "Empty data for tigger snc " << trigname << endl;
    }
    else {
        //cerr << "No trigger snc device " << trigname << endl;
    }

    // fill swic info
    RawBeamSwicData swic;
    SwicMap::iterator sit, sdone = swics.end();
    for (sit=swics.begin(); sit != sdone; ++sit) {
        string swic_name = sit->first;
        SwicBranch* swic_br = sit->second;

        memset(swic_br,0,sizeof(SwicBranch));

        const RawBeamData* rbd = (*rbmb)[swic_name.c_str()];
        if (!rbd) {
            //cerr << "No SWIC data for " << swic_name << endl;
            return false;
        }

        swic.SetData(*rbd);
        if (!swic.IsValid()) {
            cerr << "Not valid SWIC data for " << swic_name << endl;
            return false;
        }

        int sec = rbd->GetSeconds();
        int msec = rbd->GetMsecs();
        double dst = (1.0*sec) + (msec/1000.0);
        swic_br->dae_sec = sec;
        swic_br->dae_msec = msec;
        swic_br->dae_ts = dst;

        int vsec = swic.VmeSeconds();
        int vmsec = swic.VmeNanoseconds()/1000000;
        double vme = (1.0*vsec) + (vmsec/1000.0);
        swic_br->vme_sec = vsec;
        swic_br->vme_msec = vmsec;
        swic_br->vme_ts = vme;
#if 0
        cout.precision(20);
        cout << "SWIC: " << swic_name 
             << " VME@ " << vsec << "." << vmsec << " = " << vme << ", DAE@"
             << sec << "." << msec << " = " << dst << endl;
        cout << rbd->GetData()[205] << " "
             << rbd->GetData()[206] << " "
             << rbd->GetData()[207] << " "
             << rbd->GetData()[208] << endl;
#endif
    }

    // fill bpm info
    RawBeamPosData bpm;
    BpmMap::iterator pit, pdone = bpms.end();
    for (pit=bpms.begin(); pit != pdone; ++pit) {
        string bpm_name = pit->first;
        BpmBranch* bpm_br = pit->second;

        memset(bpm_br,0,sizeof(BpmBranch));

        const RawBeamData* rbd = (*rbmb)[bpm_name.c_str()];
        if (!rbd) {
            //cerr << "No BPM data for " << bpm_name << endl;
            return false;
        }

        bpm.SetData(*rbd);
        if (!bpm.IsValid()) {
            cerr << "Not valid BPM data for " << bpm_name << endl;
            return false;
        }

        int sec = rbd->GetSeconds();
        int msec = rbd->GetMsecs();
        double dst = (1.0*sec) + (msec/1000.0);
        bpm_br->dae_sec = sec;
        bpm_br->dae_msec = msec;
        bpm_br->dae_ts = dst;

        int vsec = bpm.VmeSeconds();
        int vmsec = bpm.VmeNanoseconds()/1000000;
        double vme = (1.0*vsec) + (vmsec/1000.0);
        bpm_br->vme_sec = vsec;
        bpm_br->vme_msec = vmsec;
        bpm_br->vme_ts = vme;

#if 0
        cout.precision(20);
        cout << "BPM: " << bpm_name
             << " VME@ " << vsec << "." << vmsec << " = " << vme << ", DAE@"
             << sec << "." << msec << " = " << dst << endl;
        cout << rbd->GetData()[205] << " "
             << rbd->GetData()[206] << " "
             << rbd->GetData()[207] << " "
             << rbd->GetData()[208] << endl;
#endif
    }
    return true;
}


Bdnt::Bdnt(const char* trig_name, const char* outfile)  : fImp(new BdntImp)
{
    fImp->trigname = trig_name;
    fImp->file = new TFile(outfile,"recreate");
    fImp->tree = new TTree("bd", Form("Beam Data with trigger %s",trig_name));
    fImp->trig = new TriggerBranch;
    fImp->tree->Branch("trig.","TriggerBranch",&fImp->trig);
}
static string swic_name2branch_name(string n)
{
    string out = "";
    for (size_t ind=2; ind<6; ++ind) { // keep Xs: Z:XXXXDS
        out.push_back(tolower(n[ind]));
    }
    return out;
}
static string bpm_name2branch_name(string n)
{
    string out = "";
    for (size_t ind=2; ind<5; ++ind) { // keep Xs: R:XXXCOV
        out.push_back(tolower(n[ind]));
    }
    return out;
}

void Bdnt::AddSwic(const char* swicname)
{
    fImp->swics[swicname] = new SwicBranch;
    string branchname = swic_name2branch_name(swicname);
    branchname += ".";
    fImp->tree->Branch(branchname.c_str(),"SwicBranch",
                       &(fImp->swics[swicname]));
}
void Bdnt::AddBpm(const char* bpmname)
{
    fImp->bpms[bpmname] = new BpmBranch;
    string branchname = bpm_name2branch_name(bpmname);
    branchname += ".";
    fImp->tree->Branch(branchname.c_str(),"BpmBranch",
                       &(fImp->bpms[bpmname]));
}

void Bdnt::ProcessFile(const char* filename)
{
    TFile file(filename,"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);

        if (!ient) {
            const VldContext* vld = record->GetVldContext();
            cout << "Reading " << filename << ":\n"
                 << *vld << endl;
        }

        TIter itr = record->GetRawBlockIter();
        const RawDataBlock* rdb = 0;

        // loop over blocks in record
        while ((rdb = dynamic_cast<RawDataBlock*>(itr()))) {
            if (! rdb->InheritsFrom("RawBeamMonBlock")) {
                //cerr << "Doesn't inherit from RawBeamMonBlock" << endl;
                continue;
            }
            const RawBeamMonBlock* rbmb = 
                dynamic_cast<const RawBeamMonBlock*>(rdb);
            assert(rbmb);

            fImp->Fill(rbmb);
            fImp->tree->Fill();
        }

        if (tree->GetEntries()-ient == 1) {
            const VldContext* vld = record->GetVldContext();
            cout << "finishing " << filename << ":\n"
                 << *vld << endl;
        }
        delete record; record = 0;
    }
    
}
TTree* Bdnt::GetTree() { return fImp->tree; }

void Bdnt::Write()
{
    fImp->file->cd();
    fImp->tree->Write();
    fImp->file->Close();
}


ClassImp(TriggerBranch)
ClassImp(SwicBranch)
ClassImp(BpmBranch)

---