kernel.h

Go to the documentation of this file.

#ifndef HNBODY_KERNEL_H
#define HNBODY_KERNEL_H

#include <mutils/util.h>

#ifdef __cplusplus
extern "C" {
namespace HNBODY {
#endif


typedef enum hnb_integcoord_enum {
  Origin=1+2,          
  Barycentric=0,       
  Bodycentric=1,       
  Jacobi=2,            
  Regularized=4,       
  TWform=Regularized,  
  RegBarycentric=Regularized+Barycentric,
  RegJacobi=Regularized+Jacobi,
  ModBodycentric=TWform+Bodycentric,
  Coordinates=Origin+Regularized,
  Splitting=8+16,      
  Kick_Drift=0,        
  Drift_Kick=8,        
  Shift_Drift=16,      
  Order=32+64,         
  Order2=0,            
  Order4=32,           
  ModKick=64,          
  Order4mk=Order4+ModKick,
  Integrator=Coordinates+Splitting+Order,
  Corrected=128+256+512,
  Reverse=1024,        
  Uncorrected=0,       
  Corrected2=128,      
  Corrected4=256,      
  Corrected6=512,      
  Corrector=Reverse+Corrected,
  Gravity=2048,        
  Newtonian=0,         
  PostNewtonian=Gravity,
  Kernel=4096,         
  PolyKernel=0,        
  TanhKernel=Kernel,   
  SplitI=8192          
} hnb_integcoord_t;

typedef enum hnb_integ_enum {
  Symplectic,         
  BulirschStoer,      
  RungeKutta          
} hnb_integ_t;

typedef enum LWP_enum {
  LWP_NONE,    
  LWP_ZWP      
} hnb_LWP_t;

typedef double (*hnb_vector_t)[3];

/* Forward reference. */
struct hnb_data_struct;

typedef
  void (*hnb_extra_t)(hnb_vector_t, double t, hnb_vector_t, const double m[],
         int nHL, int n, double dt, const struct hnb_data_struct *sys);
typedef
  void (*hnb_drift_t)(hnb_vector_t dx, hnb_vector_t dv, double t,
         hnb_vector_t x, hnb_vector_t v, const double m[],
         int nHL, int n, double dt, const struct hnb_data_struct *sys);
typedef
  double (*hnb_energy_t)(double t, hnb_vector_t x, hnb_vector_t v,
         const double m[], int nHL, const struct hnb_data_struct *sys);
typedef
  int  (*hnb_derivs_t)(hnb_vector_t dxdt, hnb_vector_t dvdt,
         double t, hnb_vector_t x, hnb_vector_t v,
         const double m[], int nHL, int n, const struct hnb_data_struct *sys);


typedef struct hnb_data_struct {
  /* "Public" data. */
  double G, c, Msun, dm, J2, J4, J6, obRad, dzH, dzZ, eps, tinit,
      tH[2], dtH[2], tZ[2], dtZ[2];
  int nH, nHL, n, fixedH, enc, prune;
  hnb_vector_t x0, dx0, v0, dv0, x1, dx1, v1, dv1;
  double *m, *rcapt, *renc;
  hnb_integ_t integ;
  hnb_integcoord_t itH, itZ;
  hnb_LWP_t Ltype;
  int *id, *imap, *jindex, *prune_ctrl;
  FILE *log;

  hnb_extra_t extra_kick, extra_shift;
  hnb_drift_t extra_drift;
  hnb_derivs_t extra_derivs;
  hnb_energy_t extra_energy;

  /* "Private" data from here onward... */
  hnb_vector_t x00;
  int pindex;

  /* Volatile, driver-related data. */
  volatile int halt;
  const char *recover;

  /* Temporary system vector cache. */
  hnb_vector_t *tmpvec;
  int ntmp, tmpmax;

  /* Output file pointer cache. */
  FILE *body[512];
  int bopen;

  /* Data used only with certain integrations. */
  double dzODE, *tmpODE;
  double *deta, *ieta, *dsZ, *E0, *E1, *t, *tt;
  hnb_vector_t xH, vH, xt, vt, xHtab, vHtab;
  int iZ0, ntab, maxtab, itmax;
  unsigned char *ace;
  double dtdz, ttab0;

  hnb_vector_t xh0, xh1, xha, xhb, vh0, vh1, vha, vhb, xa, xb, va, vb;
} hnb_data_t;


/* User-callable functions. */
DLLSPEC extern void
  hnb_angmom(double L[], hnb_vector_t x, hnb_vector_t v, hnb_data_t *sys),
  hnb_calcCoord(hnb_vector_t xNew, hnb_vector_t vNew,
    hnb_integcoord_t newFrame, const hnb_data_t *sys),
  hnb_checkpoint(hnb_data_t *sys, const char *file, int halt),
  hnb_free_data(hnb_data_t *sys),
  hnb_write(hnb_data_t *sys, const char *file),
  hnb_util_version(FILE *f, const char *name, const char *revision,
    const char *prefix),
  hnb_print_info(FILE *f, void (*custom)(FILE *f, const char *prefix),
    const char *prefix),
  hnb_step(hnb_data_t *sys);

DLLSPEC extern const char
 *hnb_contact(void), *hnb_version(void), *hnb_website(void);

DLLSPEC extern const int
 *hnb_idtags(const hnb_data_t *sys),
 *hnb_jacmap(const hnb_data_t *sys);

DLLSPEC extern int
  hnb_N(const hnb_data_t *sys),
  hnb_hwp_N(const hnb_data_t *sys),
  hnb_lwp_N(const hnb_data_t *sys),
  hnb_zwp_N(const hnb_data_t *sys),
  hnb_save(const char *save, double key, hnb_data_t *sys),
  hnb_restore(double *key, const char *save, hnb_data_t *sys),
  hnb_argv_driver(int argc, char *argv[], hnb_extra_t kick,
    hnb_extra_t shift, hnb_drift_t drift, hnb_derivs_t derivs,
    hnb_energy_t energy, void (*custom)(FILE *f, const char *prefix));

DLLSPEC extern hnb_integcoord_t
  hnb_integcoord(const hnb_data_t *sys),
  hnb_zwp_integcoord(const hnb_data_t *sys);

DLLSPEC extern const double
 *hnb_masses(const hnb_data_t *sys),
 *hnb_enc_radii(const hnb_data_t *sys),
 *hnb_capt_radii(const hnb_data_t *sys);

DLLSPEC extern double
  hnb_G(const hnb_data_t *sys), hnb_c(const hnb_data_t *sys),
  hnb_Msun(const hnb_data_t *sys), hnb_M(const hnb_data_t *sys),
  hnb_tinit(const hnb_data_t *sys),
  hnb_stepsize(const hnb_data_t *sys),
  hnb_zwp_stepsize(const hnb_data_t *sys),
  hnb_eff_stepsize(const hnb_data_t *sys),
  hnb_output(double *t, double (**x)[3], double (**v)[3], hnb_data_t *sys);


DLLSPEC extern hnb_data_t
 *hnb_read(const char *file),
 *hnb_init_file(const char *hnbfile, FILE *logfile),
 *hnb_init_extra(const char *hnbfile, FILE *logfile, hnb_extra_t extra_kick,
     hnb_extra_t extra_shift, hnb_drift_t extra_drift,
     hnb_derivs_t extra_derivs, hnb_energy_t extra_energy),

 *hnb_ode_init(double tinit, const double xinit[][3], const double vinit[][3],
     const double m[], const int id[],
     int nH, int nL, int nZ, hnb_LWP_t ltype, int fixedH,
     int enc, const double rcapt[], double G, double c, double Msun,
     double J2, double J4, double J6, double obRadius,
     double dt, double eps,
     hnb_integ_t integ, hnb_integcoord_t icoord,
     hnb_derivs_t extra_derivs, hnb_energy_t extra_energy, int nthreads),

 *hnb_init(double tinit, const double xinit[][3], const double vinit[][3],
     const double m[], const int id[], const int jindex[],
     int nH, int nL, int nZ, hnb_LWP_t ltype, int fixedH,
     int enc, int prune,
     const double renc[], const double rcapt[], double G, double c, double Msun,
     double J2, double J4, double J6, double obRadius,
     double dm, double dzH, double dzZ,
     hnb_integcoord_t itH, hnb_integcoord_t itZ,
     hnb_extra_t extra_kick, hnb_extra_t extra_shift, hnb_drift_t extra_drift,
     hnb_energy_t extra_energy, int nthreads);


#ifdef __cplusplus
}  // namespace HNBODY
}  // extern "C"
#endif

#endif  /* HNBODY_KERNEL_H */