Spectrum.h

Go to the documentation of this file.

#ifndef SPARROW_SPECTRUM_H
#define SPARROW_SPECTRUM_H

#include <Eigen/Core>

namespace Scine {
namespace Sparrow {
namespace RealTimeSpectroscopy {

class IncorrectSpectrumSizeException : public std::exception {
  const char* what() const noexcept final {
    return "Incorrect number of data points in spectrum assignment.";
  }
};

class Spectrum {
 public:
  explicit Spectrum(int size) {
    xData_.resize(size);
    yData_.resize(size);
  }

  Spectrum(const Eigen::VectorXd& xData, const Eigen::VectorXd& yData) {
    if (xData.size() != yData.size())
      throw IncorrectSpectrumSizeException();
    xData_ = xData;
    yData_ = yData;
  }
  Spectrum(const Eigen::VectorXd& xData, const Eigen::VectorXd& yData, std::pair<std::string, std::string> labels)
    : Spectrum(xData, yData) {
    labels_ = std::move(labels);
  }

  int size() const {
    return xData_.size();
  }

  const Eigen::VectorXd& getXData() const {
    return xData_;
  }

  const Eigen::VectorXd& getYData() const {
    return yData_;
  }

  double getXData(int point) const {
    return xData_(point);
  }

  double getYData(int point) const {
    return yData_(point);
  }

  const std::pair<std::string, std::string>& getLabels() const {
    return labels_;
  }
  void setXData(const Eigen::VectorXd& xData) {
    if (xData.size() != xData_.size())
      throw IncorrectSpectrumSizeException();
    xData_ = xData;
  }

  void setYData(const Eigen::VectorXd& yData) {
    if (yData.size() != yData_.size())
      throw IncorrectSpectrumSizeException();
    yData_ = yData;
  }

  void setXLabel(std::string xLabel) {
    labels_.first = std::move(xLabel);
  }
  void setYLabel(std::string yLabel) {
    labels_.second = std::move(yLabel);
  }

 private:
  Eigen::VectorXd xData_;
  Eigen::VectorXd yData_;
  std::pair<std::string, std::string> labels_;
};

} // namespace RealTimeSpectroscopy
} // namespace Sparrow
} // namespace Scine

#endif // SPARROW_SPECTRUM_H