103 lines
1.9 KiB
C++
103 lines
1.9 KiB
C++
#ifndef GRAVITY_H
|
|
#define GRAVITY_H
|
|
|
|
#include <thread>
|
|
#include <mutex>
|
|
#include <condition_variable>
|
|
#include "Config.h"
|
|
#include "Simulation.h"
|
|
|
|
#ifdef GRAVFFT
|
|
#include <fftw3.h>
|
|
#endif
|
|
|
|
class Simulation;
|
|
|
|
struct mask_el {
|
|
char *shape;
|
|
char shapeout;
|
|
void *next;
|
|
};
|
|
typedef struct mask_el mask_el;
|
|
|
|
class Gravity
|
|
{
|
|
private:
|
|
|
|
float *th_ogravmap;
|
|
float *th_gravmap;
|
|
float *th_gravx;
|
|
float *th_gravy;
|
|
float *th_gravp;
|
|
|
|
int th_gravchanged;
|
|
|
|
std::thread gravthread;
|
|
std::mutex gravmutex;
|
|
std::condition_variable gravcv;
|
|
int grav_ready;
|
|
int gravthread_done;
|
|
|
|
#ifdef GRAVFFT
|
|
bool grav_fft_status;
|
|
float *th_ptgravx, *th_ptgravy, *th_gravmapbig, *th_gravxbig, *th_gravybig;
|
|
fftwf_complex *th_ptgravxt, *th_ptgravyt, *th_gravmapbigt, *th_gravxbigt, *th_gravybigt;
|
|
fftwf_plan plan_gravmap, plan_gravx_inverse, plan_gravy_inverse;
|
|
#endif
|
|
|
|
//Simulation * sim;
|
|
public:
|
|
unsigned *gravmask;
|
|
float *gravmap;
|
|
float *gravp;
|
|
float *gravy;
|
|
float *gravx;
|
|
unsigned char (*bmap)[XRES/CELL];
|
|
unsigned char (*obmap)[XRES/CELL];
|
|
int ngrav_enable;
|
|
void grav_mask_r(int x, int y, char checkmap[YRES/CELL][XRES/CELL], char shape[YRES/CELL][XRES/CELL], char *shapeout);
|
|
void mask_free(mask_el *c_mask_el);
|
|
|
|
void Clear();
|
|
|
|
void gravity_init();
|
|
void gravity_cleanup();
|
|
void gravity_update_async();
|
|
|
|
void update_grav_async();
|
|
|
|
void start_grav_async();
|
|
void stop_grav_async();
|
|
void update_grav();
|
|
void gravity_mask();
|
|
|
|
void bilinear_interpolation(float *src, float *dst, int sw, int sh, int rw, int rh);
|
|
|
|
#ifdef GRAVFFT
|
|
void grav_fft_init();
|
|
void grav_fft_cleanup();
|
|
#endif
|
|
|
|
Gravity();
|
|
~Gravity();
|
|
};
|
|
|
|
/*extern int ngrav_enable; //Newtonian gravity
|
|
extern int gravwl_timeout;
|
|
extern int gravityMode;*/
|
|
|
|
/*float *gravmap;//Maps to be used by the main thread
|
|
float *gravp;
|
|
float *gravy;
|
|
float *gravx;
|
|
unsigned *gravmask;
|
|
|
|
float *th_ogravmap;// Maps to be processed by the gravity thread
|
|
float *th_gravmap;
|
|
float *th_gravx;
|
|
float *th_gravy;
|
|
float *th_gravp;*/
|
|
|
|
|
|
#endif
|