orbital_center.hpp

Go to the documentation of this file.

#pragma once
 
#include "centers.hpp"
#include "ffi_core.hpp"
#include "orbit.hpp"
#include <cstdint>
#include <ostream>
 
namespace siderust {
 
enum class OrbitReferenceCenter : std::uint8_t {
  Barycentric = 0,
  Heliocentric = 1,
  Geocentric = 2,
};
 
struct BodycentricParams {
  KeplerianOrbit orbit;
 
  OrbitReferenceCenter orbit_center;
 
  BodycentricParams(const KeplerianOrbit &orb, OrbitReferenceCenter center)
      : orbit(orb), orbit_center(center) {}
 
  static BodycentricParams heliocentric(const KeplerianOrbit &orb) {
    return BodycentricParams(orb, OrbitReferenceCenter::Heliocentric);
  }
 
  static BodycentricParams geocentric(const KeplerianOrbit &orb) {
    return BodycentricParams(orb, OrbitReferenceCenter::Geocentric);
  }
 
  static BodycentricParams barycentric(const KeplerianOrbit &orb) {
    return BodycentricParams(orb, OrbitReferenceCenter::Barycentric);
  }
 
  BodycentricParams()
      : orbit{qtty::AstronomicalUnit(1.0), Eccentricity{0.0}, qtty::Degree(0.0),
              qtty::Degree(0.0),           qtty::Degree(0.0), qtty::Degree(0.0),
              Time<TT, JD>::J2000()},
        orbit_center(OrbitReferenceCenter::Heliocentric) {}
 
  SiderustBodycentricParams to_c() const {
    SiderustBodycentricParams c{};
    c.orbit = orbit.to_c();
    c.orbit_center = static_cast<uint8_t>(orbit_center);
    return c;
  }
};
 
// Stream operator for OrbitReferenceCenter
inline std::ostream &operator<<(std::ostream &os, OrbitReferenceCenter center) {
  switch (center) {
  case OrbitReferenceCenter::Barycentric:
    return os << "Barycentric";
  case OrbitReferenceCenter::Heliocentric:
    return os << "Heliocentric";
  case OrbitReferenceCenter::Geocentric:
    return os << "Geocentric";
  }
  return os << "Unknown";
}
 
} // namespace siderust