bool compare_ensembles( swarm::ensemble& e1, swarm::ensemble &e2 , double & pos_diff, double & vel_diff, double & time_diff ) { if (e1.nsys() != e2.nsys() || e1.nbod() != e2.nbod() ) return false; pos_diff = vel_diff = time_diff = 0; for(int i = 0; i < e1.nsys(); i++) { for(int j = 0; j < e1.nbod() ; j++){ // Distance between body position in ensemble e1 and e2 double dp = sqrt( square ( e1[i][j][0].pos() - e2[i][j][0].pos() ) + square ( e1[i][j][1].pos() - e2[i][j][1].pos() ) + square ( e1[i][j][2].pos() - e2[i][j][2].pos() ) ) ; // Average magnitude of the position in e1 and e2 double ap = sqrt( square ( e1[i][j][0].pos() + e2[i][j][0].pos() ) + square ( e1[i][j][1].pos() + e2[i][j][1].pos() ) + square ( e1[i][j][2].pos() + e2[i][j][2].pos() ) ) / 2.0 ; // Difference between body velocities in ensemble e1 and e2 double dv = sqrt( square ( e1[i][j][0].vel() - e2[i][j][0].vel() ) + square ( e1[i][j][1].vel() - e2[i][j][1].vel() ) + square ( e1[i][j][2].vel() - e2[i][j][2].vel() ) ) ; // Average magnitude of the velocity in e1 and e2 double av = sqrt( square ( e1[i][j][0].vel() + e2[i][j][0].vel() ) + square ( e1[i][j][1].vel() + e2[i][j][1].vel() ) + square ( e1[i][j][2].vel() + e2[i][j][2].vel() ) ) / 2.0 ; if ( dp > pos_diff ) pos_diff = dp; if ( dv > vel_diff ) vel_diff = dv; } // Difference between time divided by average of the two double dt = fabs(e1[i].time() - e2[i].time()) /(e1[i].time() + e2[i].time())*2; if ( dt > time_diff ) time_diff = dt; } return true; }
void print_selected_systems_for_demo(swarm::ensemble& ens, unsigned int nprint, std::ostream &os = std::cout) { if(nprint>ens.nsys()) nprint = ens.nsys(); for(unsigned int systemid = 0; systemid< nprint; ++systemid) print_system(ens,systemid,os); }
void print_system(const swarm::ensemble& ens, const int systemid, std::ostream &os = std::cout) { enum { JACOBI, BARYCENTRIC, ASTROCENTRIC } COORDINATE_SYSTEM = BARYCENTRIC; const bool use_jacobi = cfg.optional("use_jacobi", 0); if(use_jacobi) COORDINATE_SYSTEM = JACOBI; std::streamsize cout_precision_old = os.precision(); os.precision(10); os << "sys_idx= " << systemid << " sys_id= " << ens[systemid].id() << " time= " << ens.time(systemid) << "\n"; double star_mass = ens.mass(systemid,0); double mass_effective = star_mass; double bx, by, bz, bvx, bvy, bvz; switch(COORDINATE_SYSTEM) { case JACOBI: ens.get_barycenter(systemid,bx,by,bz,bvx,bvy,bvz,0); break; case BARYCENTRIC: ens.get_barycenter(systemid,bx,by,bz,bvx,bvy,bvz); break; case ASTROCENTRIC: ens.get_body(systemid,0,star_mass,bx,by,bz,bvx,bvy,bvz); break; } for(unsigned int bod=1;bod<ens.nbod();++bod) // Skip star since printing orbits { // std::cout << "pos= (" << ens.x(systemid, bod) << ", " << ens.y(systemid, bod) << ", " << ens.z(systemid, bod) << ") vel= (" << ens.vx(systemid, bod) << ", " << ens.vy(systemid, bod) << ", " << ens.vz(systemid, bod) << ").\n"; double mass = ens.mass(systemid,bod); switch(COORDINATE_SYSTEM) { case JACOBI: ens.get_barycenter(systemid,bx,by,bz,bvx,bvy,bvz,bod-1); mass_effective += mass; break; case BARYCENTRIC: mass_effective = star_mass + mass; break; case ASTROCENTRIC: mass_effective = star_mass + mass; break; } double x = ens.x(systemid,bod)-bx; double y = ens.y(systemid,bod)-by; double z = ens.z(systemid,bod)-bz; double vx = ens.vx(systemid,bod)-bvx; double vy = ens.vy(systemid,bod)-bvy; double vz = ens.vz(systemid,bod)-bvz; double a, e, i, O, w, M; calc_keplerian_for_cartesian(a,e,i,O,w,M, x,y,z,vx,vy,vz, mass_effective); i *= 180/M_PI; O *= 180/M_PI; w *= 180/M_PI; M *= 180/M_PI; // os << " b= " << bod << " m= " << mass << " a= " << a << " e= " << e << " i= " << i << " Omega= " << O << " omega= " << w << " M= " << M << "\n"; os << ens[systemid].id() << " " << bod << " " << mass << " " << a << " " << e << " " << i << " " << O << " " << w << " " << M << "\n"; } os.precision(cout_precision_old); os << std::flush; }