R3BNeulandOnlineSpectra2.h

Go to the documentation of this file.

/******************************************************************************
 *   Copyright (C) 2019 GSI Helmholtzzentrum für Schwerionenforschung GmbH    *
 *   Copyright (C) 2019-2023 Members of R3B Collaboration                     *
 *                                                                            *
 *             This software is distributed under the terms of the            *
 *                 GNU General Public Licence (GPL) version 3,                *
 *                    copied verbatim in the file "LICENSE".                  *
 *                                                                            *
 * In applying this license GSI does not waive the privileges and immunities  *
 * granted to it by virtue of its status as an Intergovernmental Organization *
 * or submit itself to any jurisdiction.                                      *
 ******************************************************************************/
 
#pragma once
 
#include "R3BDataMonitor.h"
#include "R3BLogger.h"
#include <FairRootManager.h>
#include <FairRuntimeDb.h>
#include <R3BNeulandCalibrationTask.h>
#include <R3BNeulandCountRateCanvas.h>
#include <R3BNeulandOnlineCanvas.h>
#include <R3BNeulandTriggerTypes.h>
#include <Rtypes.h>
#include <RtypesCore.h>
#include <TRandom.h>
#include <atomic>
#include <memory>
#include <string_view>
#include <type_traits>
#include <vector>
 
class FairRunOnline;
 
namespace R3B
{
    class UcesbSource;
}
 
namespace R3B::Neuland
{
    class OnlineSpectra : public CalibrationTask
    {
      public:
        explicit OnlineSpectra(std::string_view name = "R3BNeulandOnlineSpectra", int iVerbose = 1);
        OnlineSpectra(const OnlineSpectra&) = delete;
        OnlineSpectra(OnlineSpectra&&) = delete;
        auto operator=(const OnlineSpectra&) -> OnlineSpectra& = delete;
        auto operator=(OnlineSpectra&&) -> OnlineSpectra& = delete;
        ~OnlineSpectra() override;
 
        // Getters:
        auto GetRandomGenerator() -> TRandom* { return random_gen_; }
        [[nodiscard]] auto GetDistanceToTarget() const { return distance_to_target_; }
        [[nodiscard]] auto GetGraphViewMode() const { return graph_view_mode_.load(); }
 
        // Setters:
        void SetRandomGenerator(TRandom* rand)
        {
            if (rand != nullptr)
            {
                random_gen_ = rand;
            }
        }
        void SetDistanceToTarget(double distance) { distance_to_target_ = distance; }
        void SetUcesbSource(UcesbSource* source) { ucesb_source_ = source; }
 
        template <typename SpecType, typename = std::enable_if_t<std::is_base_of_v<OnlineCanvas, SpecType>>>
        auto AddCanvas(std::string_view name, CalTrigger trigger = CalTrigger::all) -> SpecType&
        {
            auto& spectrum = spectra_.emplace_back(std::make_unique<SpecType>(name));
            spectrum->SetOnlineSpectra(this);
            spectrum->SetTrigger(trigger);
            return *static_cast<SpecType*>(spectrum.get());
        }
 
        template <typename UnaryOp>
        void ActionToSpectra(UnaryOp optn)
        {
            for (auto& spectrum : spectra_)
            {
                optn(spectrum.get());
            }
        }
 
        // Methods for https server:
        void ResetHistos();
        void ResetHistosMapped();
        void SaveAll();
        void RestartUcesbServer();
        void SetCountRateFullView()
        {
            R3BLOG(info, "Graph is changed to full view!");
            graph_view_mode_.store(CountRateCanvas::RangeViewMode::full);
        }
        void SetCountRateViewTwoHours()
        {
            R3BLOG(info, "Graph is changed to 2 hour view!");
            graph_view_mode_.store(CountRateCanvas::RangeViewMode::two_hours);
        }
        void SetCountRateViewfourHours()
        {
            R3BLOG(info, "Graph is changed to 4 hour view!");
            graph_view_mode_.store(CountRateCanvas::RangeViewMode::four_hours);
        }
 
        // for backward compability
        [[deprecated("Please use SetTrigger()")]] void SetCosmicTpat(UInt_t CosmicTpat = 0) {}
 
      private:
        double distance_to_target_ = 0.;
        std::atomic<CountRateCanvas::RangeViewMode> graph_view_mode_ = CountRateCanvas::RangeViewMode::full;
        std::vector<std::unique_ptr<OnlineCanvas>> spectra_;
        UcesbSource* ucesb_source_ = nullptr;
        TRandom* random_gen_ = gRandom; // non-owning
 
        // virtual functions:
        void HistogramInit(DataMonitor& histograms) override;
        void ExtraInit(FairRootManager* rootMan) override;
        void SetExtraPar(FairRuntimeDb* rtdb) override;
        void TriggeredExec() override;
        // void EndOfTask() override;
        // [[nodiscard]] auto CheckConditions() const -> bool override;
 
        // non-virtual member functions:
        void init_http_server(FairRunOnline* run);
 
      public:
        ClassDefOverride(OnlineSpectra, 1);
    };
 
} // namespace R3B::Neuland
 
using R3BNeulandOnlineSpectra2 = R3B::Neuland::OnlineSpectra;