// Eccentricity damping
// This one is more complicated as it needs orbital elements to compute the forces.
void problem_edot(){
for(int i=1;i<N;i++){
if (tau_e[i]!=0){
double d = dt/tau_e[i];
struct particle* p = &(particles[i]);
struct orbit o = tools_p2orbit(*p,particles[0].m);
double rdot = o.h/o.a/( 1. - o.e*o.e ) * o.e * sin(o.f);
double rfdote = o.h/o.a/( 1. - o.e*o.e ) * ( 1. + o.e*cos(o.f) ) * (o.e + cos(o.f)) / (1.-o.e*o.e) / (1.+o.e*cos(o.f));
//position
double tmpfac = d * ( o.r/(o.a*(1.-o.e*o.e)) - (1.+o.e*o.e)/(1.-o.e*o.e));
p->x -= tmpfac * p->x;
p->y -= tmpfac * p->y;
p->z -= tmpfac * p->z;
//vx
tmpfac = rdot/(o.e*(1.-o.e*o.e));
p->vx -= d * o.e * ( cos(o.Omega) * (tmpfac * cos(o.omega+o.f) - rfdote*sin(o.omega+o.f) )
-cos(o.inc) * sin(o.Omega) * (tmpfac * sin(o.omega+o.f) + rfdote*cos(o.omega+o.f) ));
//vy
p->vy -= d * o.e * ( sin(o.Omega) * (tmpfac * cos(o.omega+o.f) - rfdote*sin(o.omega+o.f) )
+cos(o.inc) * cos(o.Omega) * (tmpfac * sin(o.omega+o.f) + rfdote*cos(o.omega+o.f) ));
//vz
p->vz -= d * o.e * ( sin(o.inc) * (tmpfac * sin(o.omega+o.f) + rfdote*cos(o.omega+o.f) ));
}
}
}
void problem_output(){
if(output_check(4000.*dt)){ // output something to screen
output_timing();
}
if(output_check(M_PI*2.*0.01)){ // output semimajor axis to file
FILE* f = fopen("a.txt","a");
const struct particle planet = particles[0];
for (int i=1;i<N;i++){
struct orbit o = tools_p2orbit(particles[i],planet);
fprintf(f,"%.15e\t%.15e\t%.15e\t%.15e\n",t,o.a,o.e,o.omega);
}
fclose(f);
}
}
void display(){
if (display_pause) return;
#ifdef TREE
if (display_tree){
tree_update();
#ifdef GRAVITY_TREE
tree_update_gravity_data();
#endif
}
#endif
if (display_clear){
glClear(GL_DEPTH_BUFFER_BIT | GL_COLOR_BUFFER_BIT | GL_STENCIL_BUFFER_BIT);
}
if (!display_wire) {
if (display_spheres){
glDisable(GL_BLEND);
glDepthMask(GL_TRUE);
glEnable(GL_DEPTH_TEST);
glEnable(GL_LIGHTING);
glEnable(GL_LIGHT0);
GLfloat lightpos[] = {0, boxsize_max, boxsize_max, 0.f};
glLightfv(GL_LIGHT0, GL_POSITION, lightpos);
}else{
glEnable(GL_BLEND);
glDepthMask(GL_FALSE);
glDisable(GL_DEPTH_TEST);
glDisable(GL_LIGHTING);
glDisable(GL_LIGHT0);
}
}
glEnable(GL_POINT_SMOOTH);
glVertexPointer(3, GL_DOUBLE, sizeof(struct particle), particles);
int _N_active = (N_active==-1)?N:N_active;
if (display_reference>=0){
glTranslatef(-particles[display_reference].x,-particles[display_reference].y,-particles[display_reference].z);
}
glRotatef(display_rotate_x,1,0,0);
glRotatef(display_rotate_z,0,0,1);
for (int i=-display_ghostboxes*nghostx;i<=display_ghostboxes*nghostx;i++){
for (int j=-display_ghostboxes*nghosty;j<=display_ghostboxes*nghosty;j++){
for (int k=-display_ghostboxes*nghostz;k<=display_ghostboxes*nghostz;k++){
struct ghostbox gb = boundaries_get_ghostbox(i,j,k);
glTranslatef(gb.shiftx,gb.shifty,gb.shiftz);
if (!(!display_clear&&display_wire)){
if (display_spheres){
// Drawing Spheres
glColor4f(1.0,1.0,1.0,1.0);
#ifndef COLLISIONS_NONE
for (int i=0;i<N;i++){
struct particle p = particles[i];
glTranslatef(p.x,p.y,p.z);
glScalef(p.r,p.r,p.r);
#ifdef _APPLE
glCallList(display_dlist_sphere);
#else //_APPLE
glutSolidSphere(1,40,10);
#endif //_APPLE
glScalef(1./p.r,1./p.r,1./p.r);
glTranslatef(-p.x,-p.y,-p.z);
}
#endif // COLLISIONS_NONE
}else{
// Drawing Points
glEnableClientState(GL_VERTEX_ARRAY);
glPointSize(3.);
glColor4f(1.0,1.0,1.0,0.5);
glDrawArrays(GL_POINTS, _N_active, N-_N_active);
glColor4f(1.0,1.0,0.0,0.9);
glPointSize(5.);
glDrawArrays(GL_POINTS, 0, _N_active);
glDisableClientState(GL_VERTEX_ARRAY);
}
}
// Drawing wires
if (display_wire){
#ifndef INTEGRATOR_SEI
double radius = 0;
struct particle com = particles[0];
for (int i=1;i<N;i++){
struct particle p = particles[i];
if (N_active>0){
// Different colors for active/test particles
if (i>=N_active){
glColor4f(0.9,1.0,0.9,0.9);
}else{
glColor4f(1.0,0.9,0.0,0.9);
}
}else{
// Alternating colors
if (i%2 == 1){
glColor4f(0.0,1.0,0.0,0.9);
}else{
glColor4f(0.0,0.0,1.0,0.9);
}
}
struct orbit o = tools_p2orbit(p,com);
glPushMatrix();
glTranslatef(com.x,com.y,com.z);
glRotatef(o.Omega/DEG2RAD,0,0,1);
glRotatef(o.inc/DEG2RAD,1,0,0);
glRotatef(o.omega/DEG2RAD,0,0,1);
glBegin(GL_LINE_LOOP);
for (double trueAnom=0; trueAnom < 2.*M_PI; trueAnom+=M_PI/100.) {
//convert degrees into radians
radius = o.a * (1. - o.e*o.e) / (1. + o.e*cos(trueAnom));
glVertex3f(radius*cos(trueAnom),radius*sin(trueAnom),0);
}
glEnd();
glPopMatrix();
com = tools_get_center_of_mass(p,com);
}
#else // INTEGRATOR_SEI
for (int i=1;i<N;i++){
struct particle p = particles[i];
glBegin(GL_LINE_LOOP);
for (double _t=-100.*dt;_t<=100.*dt;_t+=20.*dt){
double frac = 1.-fabs(_t/(120.*dt));
glColor4f(1.0,(_t+100.*dt)/(200.*dt),0.0,frac);
glVertex3f(p.x+p.vx*_t, p.y+p.vy*_t, p.z+p.vz*_t);
}
glEnd();
}
#endif // INTEGRATOR_SEI
}
// Drawing Tree
glColor4f(1.0,0.0,0.0,0.4);
#ifdef TREE
if (display_tree){
glColor4f(1.0,0.0,0.0,0.4);
display_entire_tree();
}
#endif // TREE
glTranslatef(-gb.shiftx,-gb.shifty,-gb.shiftz);
}
}
}
glColor4f(1.0,0.0,0.0,0.4);
glScalef(boxsize_x,boxsize_y,boxsize_z);
glutWireCube(1);
glScalef(1./boxsize_x,1./boxsize_y,1./boxsize_z);
glRotatef(-display_rotate_z,0,0,1);
glRotatef(-display_rotate_x,1,0,0);
if (display_reference>=0){
glTranslatef(particles[display_reference].x,particles[display_reference].y,particles[display_reference].z);
}
glutSwapBuffers();
}
