epics_metric.cc

// epics_metric.cc: Epics Metric Plugin
// Author: Eric Flumerfelt
// Last Modified: 09/25/2015
//
// An implementation of the MetricPlugin interface for Epics/ChannelAccess

#ifndef __EPICS_METRIC__
#define __EPICS_METRIC__ 1

#include <unordered_map>
#include <utility>
#include "artdaq-utilities/Plugins/MetricMacros.hh"
#include "messagefacility/MessageLogger/MessageLogger.h"
#undef STATIC_ASSERT

#ifdef __clang__
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wunused-parameter"
#endif
#include <cadef.h>
#ifdef __clang__
#pragma clang diagnostic pop
#endif

#include "TRACE/tracemf.h"
#define TRACE_NAME "EPICSMetric"

namespace artdaq {
class EpicsMetric : public MetricPlugin
{
private:
    std::string prefix_;
    std::unordered_map<std::string, chid> channels_;

    bool checkChannel_(const std::string& name)
    {
        if (channels_.count(name) == 0u)
        {
            chid channel;
            ca_search(name.c_str(), &channel);
            auto sts = ca_pend_io(5.0);
            if (sts != ECA_NORMAL)
            {
                SEVCHK(ca_clear_channel(channel), NULL);
                TLOG(TLVL_WARNING) << "Channel \"" << name << "\" not found!";
                channels_[name] = nullptr;
                return false;
            }
            channels_[name] = channel;
            return true;
        }
        return channels_[name] != nullptr;
    }

    std::string parseChannelName_(std::string prefix_, std::string name)
    {
        std::string caName = std::move(name);
        const std::string& caPrefix_ = std::move(prefix_);

        while (caName.find(' ') != std::string::npos)
        {
            caName = caName.replace(caName.find(' '), 1, "_");
        }
        while (caName.find('.') != std::string::npos)
        {
            caName = caName.replace(caName.find('.'), 1, ":");
        }
        while (caName.find('/') != std::string::npos)
        {
            caName = caName.replace(caName.find('/'), 1, "_");
        }
        while (caName.find("_%") != std::string::npos)
        {
            caName = caName.replace(caName.find("_%"), 2, "");
        }
        caName = caPrefix_ + ":" + caName;

        TLOG(TLVL_DEBUG) << "Channel name is: \"" << caName << "\"";
        return caName;
    }

    EpicsMetric(EpicsMetric const&) = delete;
    EpicsMetric(EpicsMetric&&) = delete;
    EpicsMetric& operator=(EpicsMetric const&) = delete;
    EpicsMetric& operator=(EpicsMetric&&) = delete;

public:
    explicit EpicsMetric(fhicl::ParameterSet const& pset, std::string const& app_name)
        : MetricPlugin(pset, app_name), prefix_(pset.get<std::string>("channel_name_prefix", "artdaq")) {}

    ~EpicsMetric() override { MetricPlugin::stopMetrics(); }

    std::string getLibName() const override { return "epics"; }

    void stopMetrics_() override
    {
        for (const auto& channel : channels_)
        {
            if (channel.second != 0)
            {
                SEVCHK(ca_clear_channel(channel.second), NULL);
            }
        }
        channels_.clear();
    }

    void startMetrics_() override {}

    void sendMetric_(const std::string& name, const std::string& value, const std::string& unit) override
    {
        //std::string caName = prefix_ + ":" + name;
        std::string caName = parseChannelName_(prefix_, name);

        std::string tmpValue = value + " " + unit;

        if (checkChannel_(caName))
        {
            // DBR_STRING, 40 characters
            if (tmpValue.size() > 40)
            {
                tmpValue = tmpValue.erase(40);
            }
            SEVCHK(ca_put(DBR_STRING, channels_[caName], tmpValue.c_str()), NULL);
            SEVCHK(ca_flush_io(), NULL);
        }
    }

    void sendMetric_(const std::string& name, const int& value, const std::string& unit) override
    {
        // DBR_LONG
        //std::string caName = prefix_ + ":" + name;
        std::string caName = parseChannelName_(prefix_, name);

        if (!unit.empty())
        {
            TLOG(TLVL_DEBUG) << "Not sure if I can send ChannelAccess Units...configure in db instead.";
        }

        if (checkChannel_(caName))
        {
            auto val = static_cast<dbr_long_t>(value);
            SEVCHK(ca_put(DBR_LONG, channels_[caName], &val), NULL);
            SEVCHK(ca_flush_io(), NULL);
        }
    }

    void sendMetric_(const std::string& name, const double& value, const std::string& unit) override
    {
        // DBR_DOUBLE
        //std::string caName = prefix_ + ":" + name;
        std::string caName = parseChannelName_(prefix_, name);

        if (!unit.empty())
        {
            TLOG(TLVL_DEBUG) << "Not sure if I can send ChannelAccess Units...configure in db instead.";
        }

        if (checkChannel_(caName))
        {
            auto val = static_cast<dbr_double_t>(value);
            SEVCHK(ca_put(DBR_DOUBLE, channels_[caName], &val), NULL);
            SEVCHK(ca_flush_io(), NULL);
        }
    }

    void sendMetric_(const std::string& name, const float& value, const std::string& unit) override
    {
        // DBR_FLOAT
        //std::string caName = prefix_ + ":" + name;
        std::string caName = parseChannelName_(prefix_, name);

        if (!unit.empty())
        {
            TLOG(TLVL_DEBUG) << "Not sure if I can send ChannelAccess Units...configure in db instead.";
        }

        if (checkChannel_(caName))
        {
            auto val = static_cast<dbr_float_t>(value);
            SEVCHK(ca_put(DBR_FLOAT, channels_[caName], &val), NULL);
            SEVCHK(ca_flush_io(), NULL);
        }
    }

    void sendMetric_(const std::string& name, const uint64_t& value, const std::string& unit) override
    {
        // DBR_LONG, only unsigned type is only 16 bits, use widest integral field
        //std::string caName = prefix_ + ":" + name;
        std::string caName = parseChannelName_(prefix_, name);

        if (!unit.empty())
        {
            TLOG(TLVL_DEBUG) << "Not sure if I can send ChannelAccess Units...configure in db instead.";
        }

        if (checkChannel_(caName))
        {
            auto val = static_cast<dbr_ulong_t>(value);
            SEVCHK(ca_put(DBR_LONG, channels_[caName], &val), NULL);
            SEVCHK(ca_flush_io(), NULL);
        }
    }
};
}  // End namespace artdaq

DEFINE_ARTDAQ_METRIC(artdaq::EpicsMetric)

#endif  // End ifndef __EPICS_METRIC__