BDTestDataModule.cxx

Go to the documentation of this file.

#include "BDTestDataModule.h"
#include "BDTestData.h"

#include <BeamDataUtil/BDUniquify.h>
#include <RawData/RawBeamMonBlock.h>
#include <RawData/RawBeamMonHeaderBlock.h>

#include <MinosObjectMap/MomNavigator.h>

#include <MessageService/MsgService.h>
CVSID("$Id: BDTestDataModule.cxx,v 1.1 2005/12/15 22:06:32 bv Exp $");

#include <JobControl/JobCModuleRegistry.h>
JOBMODULE(BDTestDataModule, "BDTestData", "Fill the bdtest tree");

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

#include <string>
using namespace std;

class BDTestDataModuleImp {
    TFile* outfile;
    TTree* outtree;
    BDTestData* bdtd;
    string output_file;
public:

    BDTestDataModuleImp(string output) 
        : outfile(0)
        , outtree(0)
        , bdtd(0)
        , output_file(output) 
    {
        cerr << "Starting output file: " << output_file << endl;
        outfile = new TFile(output_file.c_str(),"RECREATE");
        outtree = new TTree("bdtest", "Beam Monitoring Data");
        bdtd = new BDTestData;
        outtree->Branch("pot","BDTestData",&bdtd);
        cerr << "Bdtd at " << (void*)bdtd << endl;
    }
    ~BDTestDataModuleImp() {
        cerr << "~BDTestDataModuleImp() \n";
        outfile->cd();
        outtree->Write();
        outfile->Close();
        delete outfile;
        //delete outtree;
        delete bdtd;
        outfile = 0;
        outtree = 0;
        bdtd = 0;
        cerr << output_file << " written\n";
    }

    void Fill(const RawBeamMonBlock& block, const RawBeamMonHeaderBlock& head)
    {
        if (fill_bdtest(*bdtd,block,head))
            outtree->Fill();
    }
};

BDTestDataModule::BDTestDataModule()
{
    fImp = 0;
}
BDTestDataModule::~BDTestDataModule()
{
    if (fImp) delete fImp;
}

void BDTestDataModule::BeginFile()
{
    if (fImp) {
        cerr << "BDTestDataModule::BeginFile: deleting non-zero fImp\n";
        delete fImp; fImp = 0;
    }

    const char* output_dir = gSystem->Getenv("BEAM_DATA_TESTDATA_DIR");
    if (!output_dir || output_dir[0] == '\0') output_dir = ".";
    TString file = output_dir;
    file += "/";

    const char* input_file = this->GetCurrentFile();
    const char* base_name = gSystem->BaseName(input_file);

    file += base_name;
    file.Remove(file.Index(".mbeam.root"));
    file += ".bdtd.root";
    fImp = new BDTestDataModuleImp(file.Data());
}

void BDTestDataModule::EndFile()
{
    cerr << "BDTestDataModule::EndFile with fImp at " << (void*)fImp << endl;
    if (fImp) delete fImp;
    fImp = 0;
}

JobCResult BDTestDataModule::Reco(MomNavigator* mom)
{
    if (!fImp) {
        cerr << "Error: BDTestDataModule::Reco called w/out an implementation\n";
        return JobCResult::kError;
    }

    static BDUniquify uniqifier;
    vector<BDUniquify::BlockPair> blocks = uniqifier.GetUniqueBlocks(*mom);
    size_t nblocks = blocks.size();

    if (nblocks == 0) {
        MSG("BD",Msg::kWarning)
            << "No good beam monitoring data in MOM\n";
        return JobCResult::kFailed;
    }
    for (size_t iblock=0; iblock<nblocks; ++iblock) {
        const RawBeamMonHeaderBlock& rbmhb = *blocks[iblock].first;
        const RawBeamMonBlock& rbmb = *blocks[iblock].second;

        fImp->Fill(rbmb,rbmhb);
    }
    return JobCResult::kAOK;

}

---