vlachos_et_al.h

#ifndef _VLACHOS_ET_AL_H_
#define _VLACHOS_ET_AL_H_

#include <memory>
#include <string>
#include <vector>
#include <Eigen/Dense>
#include "distribution.h"
#include "json_object.h"
#include "numeric_utils.h"
#include "stochastic_model.h"

namespace stochastic {
class VlachosEtAl : public StochasticModel {
 public:
  VlachosEtAl() = default;

  VlachosEtAl(double moment_magnitude, double rupture_distance, double vs30,
              double orientation, unsigned int num_spectra,
              unsigned int num_sims);

  VlachosEtAl(double moment_magnitude, double rupture_distance, double vs30,
              double orientation, unsigned int num_spectra,
              unsigned int num_sims, int seed_value);  

  virtual ~VlachosEtAl() {};

  VlachosEtAl(const VlachosEtAl&) = delete;

  VlachosEtAl& operator=(const VlachosEtAl&) = delete;

  utilities::JsonObject generate(const std::string& event_name,
                                 bool g_units = false) override;

  bool generate(const std::string& event_name,
                const std::string& output_location,
                bool g_units = false) override;

  bool time_history_family(std::vector<std::vector<double>>& time_histories,
                           const Eigen::VectorXd& parameters) const;

  void simulate_time_history(std::vector<double>& time_history,
                             const Eigen::MatrixXd& power_spectrum) const;

  bool post_process(std::vector<double>& time_history,
                    const std::vector<double>& filter_imp_resp) const;

  Eigen::VectorXd identify_parameters(const Eigen::VectorXd& initial_params) const;

  std::vector<double> modal_frequencies(
      const std::vector<double>& parameters,
      const std::vector<double>& energy) const;

  std::vector<double> energy_accumulation(
      const std::vector<double>& parameters,
      const std::vector<double>& times) const;

  std::vector<double> modal_participation_factor(
      const std::vector<double>& parameters,
      const std::vector<double>& energy) const;

  std::vector<double> amplitude_modulating_function(
      double duration, double total_energy,
      const std::vector<double>& parameters,
      const std::vector<double>& times) const;

  Eigen::VectorXd kt_2(const std::vector<double>& parameters,
                       const std::vector<double>& frequencies,
                       const std::vector<double>& highpass_butter) const;

  void rotate_acceleration(const std::vector<double>& acceleration,
                           std::vector<double>& x_accels,
                           std::vector<double>& y_accels, bool g_units) const;

 private:
  double moment_magnitude_; 
  double rupture_dist_; 
  double vs30_; 
  double orientation_; 
  double time_step_; 
  double freq_step_; 
  double cutoff_freq_; 
  unsigned int num_spectra_; 
  unsigned int num_sims_; 
  int seed_value_; 
  Eigen::VectorXd means_; 
  Eigen::MatrixXd covariance_; 
  std::vector<std::shared_ptr<stochastic::Distribution>>
      model_parameters_; 
  Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic>
      parameter_realizations_; 
  Eigen::Matrix<double, Eigen::Dynamic, Eigen::Dynamic>
      physical_parameters_; 
  std::shared_ptr<numeric_utils::RandomGenerator>
      sample_generator_; 
};
}  // namespace stochastic

#endif  // _VLACHOS_ET_AL_H_