Knot.cxx

Go to the documentation of this file.

#include "Knot.h"
#include "AcnetDevice.h"

#include <TTree.h>

#include <map>
#include <string>
#include <iostream>
using namespace std;

static string acnet2branchname(string n)
{
    string out = "";
    for (size_t ind=0; ind<8 && ind<n.size(); ++ind) { 
        if (n[ind] == ':') continue;
        out.push_back(tolower(n[ind]));
    }
    out += ".";
    return out;
}

class KnotImp {
public:

    TTree& fTree;
    int fCurrentEntry;
    Knot::DeviceMap fDeviceMap;

    KnotImp(TTree& tree) : fTree(tree), fCurrentEntry(-1) {}
    ~KnotImp() {
        Knot::DeviceMap::iterator it, done = fDeviceMap.end();
        for (it = fDeviceMap.begin(); it != done; ++it) {
            delete it->second;
            it->second = 0;
        }
    }

    const AcnetDevice* GetDevice(const char* devname) {
        Knot::DeviceMap::iterator it = fDeviceMap.find(devname);
        if (it != fDeviceMap.end()) return it->second;

        AcnetDevice* ad = new AcnetDevice(devname);
        fDeviceMap[devname] = ad;

        string dns = acnet2branchname(devname);
        //cerr << "Getting device " << devname << " as " << dns << endl;
        fTree.SetBranchStatus(dns.c_str(),1);
        fTree.SetBranchAddress(dns.c_str(),&fDeviceMap[devname]);
        if (fCurrentEntry >= 0)
            this->GetEntry(fCurrentEntry);
        return ad;
    }

    int GetEntry(int entry) {
        fCurrentEntry = entry;
        return fTree.GetEntry(entry);
    }
};

Knot::Knot(TTree& tree) : fImp(new KnotImp(tree)) {}
Knot::Knot() : fImp(0) {}

void Knot::SetTree(TTree& tree)
{
    if (fImp) delete fImp; fImp = 0;
    fImp = new KnotImp(tree);
}

int Knot::GetSize()
{
    if (!fImp) return -1;
    return fImp->fTree.GetEntries();
}

int Knot::GetEntry(int entry)
{
    if (!fImp) return -1;
    return fImp->GetEntry(entry);
}

int Knot::GetEntryNumber()
{
    if (!fImp) return -1;
    return fImp->fCurrentEntry;
}
const AcnetDevice* Knot::GetDevice(const char* name)
{
    if (!fImp) return 0;
    return fImp->GetDevice(name);
}
    
const Knot::DeviceMap& Knot::GetDevices() const
{
    return fImp->fDeviceMap;
}

---