added flexibility

src/motions/base.h

@@ -27,6 +27,7 @@ public:
     [[nodiscard]] double getEta() const { return m_eta; }
     void setEta(const double eta) { m_eta = eta; }
     [[nodiscard]] std::string getName() const { return m_name; }
+    [[nodiscard]] double getInitOmega() const { return m_initial_omega; };
 
     static Motion* createFromInput(const std::string& input, std::mt19937_64& rng);
 
@@ -36,6 +37,7 @@ protected:
     double m_eta{0.};
     std::mt19937_64& m_rng;
     std::uniform_real_distribution<> m_uni_dist;
+    double m_initial_omega{0.};
 };
 
 std::ostream& operator<<(std::ostream& os, const Motion& m);

src/motions/bimodalangle.cpp

@@ -11,6 +11,7 @@ BimodalAngle::BimodalAngle(std::mt19937_64 &rng) : Motion(std::string("BimodalAn
 
 void BimodalAngle::initialize() {
     m_prev_pos = draw_position();
+    m_initial_omega = omega_q(m_prev_pos);
 };
 
 double BimodalAngle::jump() {

src/motions/isosmallangle.cpp

@@ -11,6 +11,7 @@ SmallAngle::SmallAngle(std::mt19937_64 &rng) : Motion(std::string("IsotropicAngl
 
 void SmallAngle::initialize() {
     m_prev_pos = draw_position();
+    m_initial_omega = omega_q(m_prev_pos);
 };
 
 double SmallAngle::jump() {

src/motions/random.cpp

@@ -6,7 +6,9 @@ RandomJump::RandomJump(const double delta, const double eta, std::mt19937_64 &rn
 
 RandomJump::RandomJump(std::mt19937_64 &rng) : Motion(std::string("RandomJump"), rng) {}
 
-void RandomJump::initialize() {}
+void RandomJump::initialize() {
+    m_initial_omega = RandomJump::jump();
+}
 
 double RandomJump::jump() {
     return omega_q(draw_position());

src/motions/tetrahedral.cpp

@@ -19,6 +19,7 @@ void TetrahedralJump::initialize() {
         auto corner_pos = rotate(pos, m_beta, alpha + (i-1) * 2*M_PI/3.);
         m_corners[i] = omega_q(corner_pos);
     }
+    m_initial_omega = TetrahedralJump::jump();
 }
 
 double TetrahedralJump::jump() {