Jungfraujoch/common/AzimuthalIntegrationSettings.h

// SPDX-FileCopyrightText: 2024 Filip Leonarski, Paul Scherrer Institute <filip.leonarski@psi.ch>
// SPDX-License-Identifier: GPL-3.0-only

#pragma once

#include "JFJochMath.h"
#include <optional>
#include <cstdint>

class AzimuthalIntegrationSettings {
    constexpr static float minQ_recipA = 1e-5;
    constexpr static float maxQ_recipA = 10.0;

    bool solid_angle_correction = true;
    bool polarization_correction = true;
    float high_q_recipA = 5.0;
    float low_q_recipA = 0.1;
    float bkg_estimate_high_q_recipA = 2.0f * PI / 3.0;
    float bkg_estimate_low_q_recipA = 2.0f * PI / 5.0;
    float q_spacing = 0.01;
    int32_t azim_bins = 1;
    // Compute azimuthal integration on the CPU instead of the FPGA during the FPGA
    // acquisition workflow. Lifts the FPGA bin-count limit and adds standard-deviation output.
    bool force_cpu_in_fpga_workflow = false;

    int32_t q_bins= 0;
    int32_t total_bins = 0;

    void UpdateBinCount();
public:
    AzimuthalIntegrationSettings();
    AzimuthalIntegrationSettings& SolidAngleCorrection(bool input);
    AzimuthalIntegrationSettings& PolarizationCorrection(bool input);
    AzimuthalIntegrationSettings& QRange_recipA(float low, float high);
    AzimuthalIntegrationSettings& QSpacing_recipA(float input);
    AzimuthalIntegrationSettings& BkgEstimateQRange_recipA(float low, float high);
    AzimuthalIntegrationSettings& AzimuthalBinCount(int32_t input);
    AzimuthalIntegrationSettings& ForceCPUinFPGAWorkflow(bool input);

    [[nodiscard]] bool IsSolidAngleCorrection() const;
    [[nodiscard]] bool IsPolarizationCorrection() const;
    [[nodiscard]] float GetHighQ_recipA() const;
    [[nodiscard]] float GetLowQ_recipA() const;
    [[nodiscard]] float GetQSpacing_recipA() const;

    [[nodiscard]] int32_t GetBinCount() const;
    [[nodiscard]] int32_t GetQBinCount() const;
    [[nodiscard]] int32_t GetAzimuthalBinCount() const;
    [[nodiscard]] bool IsForceCPUinFPGAWorkflow() const;

    [[nodiscard]] float GetBkgEstimateLowQ_recipA() const;
    [[nodiscard]] float GetBkgEstimateHighQ_recipA() const;
    [[nodiscard]] uint16_t QToBin(float q) const;
    [[nodiscard]] uint16_t GetBin(float q, float phi_deg) const;
};