void makedisk(bool randspin)
{
int i;
bodyptr bp;
double m, r, v, phi;
matrix xmat, zmat;
vector tmpv;
for (i = 0; i < ndisk; i++) { // loop initializing bodies
bp = NthBody(disk, i); // set ptr to body number i
m = mdtab[NTAB-1] * ((double) i + 0.5) / ndisk;
r = gsl_interp_eval(rm_spline, mdtab, rdtab, m, NULL);
v = gsl_interp_eval(vr_spline, rdtab, vctab, r, NULL);
phi = xrandom(0.0, 2 * M_PI);
Pos(bp)[0] = r * sin(phi);
Pos(bp)[1] = r * cos(phi);
Pos(bp)[2] = 0.0;
Vel(bp)[0] = v * cos(phi);
Vel(bp)[1] = - v * sin(phi);
Vel(bp)[2] = 0.0;
pickshell(AuxVec(bp), NDIM, 1.0);
if (randspin) {
xmatrix(xmat, acos(AuxVec(bp)[2]));
zmatrix(zmat, atan2(AuxVec(bp)[0], AuxVec(bp)[1]));
MULMV(tmpv, xmat, Pos(bp));
MULMV(Pos(bp), zmat, tmpv);
MULMV(tmpv, xmat, Vel(bp));
MULMV(Vel(bp), zmat, tmpv);
}
}
}
void snaprect(bodyptr btab, int nbody)
{
matrix qmat, tmpm;
bodyptr bp;
vector frame[3], tmpv;
static vector oldframe[3] =
{ { 1.0, 0.0, 0.0, }, { 0.0, 1.0, 0.0, }, { 0.0, 0.0, 1.0, }, };
int i;
snapcenter(btab, nbody, WeightField.offset);
CLRM(qmat);
for (bp = btab; bp < NthBody(btab, nbody); bp = NextBody(bp)) {
OUTVP(tmpm, Pos(bp), Pos(bp));
MULMS(tmpm, tmpm, Weight(bp));
ADDM(qmat, qmat, tmpm);
}
eigenvect(frame[0], frame[1], frame[2], qmat);
if (dotvp(oldframe[0], frame[0]) < 0.0)
MULVS(frame[0], frame[0], -1.0);
if (dotvp(oldframe[2], frame[2]) < 0.0)
MULVS(frame[2], frame[2], -1.0);
CROSSVP(frame[1], frame[2], frame[0]);
printvect("e_x:", frame[0]);
printvect("e_y:", frame[1]);
printvect("e_z:", frame[2]);
for (bp = btab; bp < NthBody(btab, nbody); bp = NextBody(bp)) {
if (PosField.offset != BadOffset) {
for (i = 0; i < NDIM; i++)
tmpv[i] = dotvp(Pos(bp), frame[i]);
SETV(Pos(bp), tmpv);
}
if (VelField.offset != BadOffset) {
for (i = 0; i < NDIM; i++)
tmpv[i] = dotvp(Vel(bp), frame[i]);
SETV(Vel(bp), tmpv);
}
if (AccField.offset != BadOffset) {
for (i = 0; i < NDIM; i++)
tmpv[i] = dotvp(Acc(bp), frame[i]);
SETV(Acc(bp), tmpv);
}
if (AuxVecField.offset != BadOffset) {
for (i = 0; i < NDIM; i++)
tmpv[i] = dotvp(AuxVec(bp), frame[i]);
SETV(AuxVec(bp), tmpv);
}
}
for (i = 0; i < NDIM; i++)
SETV(oldframe[i], frame[i]);
}
void rotate(snapshot *snap, string *tags, string *vecs, char axis,
real thetax, real thetay, real thetaz)
{
matrix rmat;
bodyptr bp;
switch (tolower(axis)) {
case 'x':
xmatrix(rmat, thetax);
break;
case 'y':
ymatrix(rmat, thetay);
break;
case 'z':
zmatrix(rmat, thetaz);
break;
default:
error("%s: unknown axis %c\n", getargv0(), axis);
}
if (set_member(tags, PosTag) && set_member(vecs, PosTag))
for_all_bodies(bp, *snap)
rotatevec(Pos(bp), rmat);
if (set_member(tags, VelTag) && set_member(vecs, VelTag))
for_all_bodies(bp, *snap)
rotatevec(Vel(bp), rmat);
if (set_member(tags, AccTag) && set_member(vecs, AccTag))
for_all_bodies(bp, *snap)
rotatevec(Acc(bp), rmat);
if (set_member(tags, AuxVecTag) && set_member(vecs, AuxVecTag))
for_all_bodies(bp, *snap)
rotatevec(AuxVec(bp), rmat);
}
void print_data(bodyptr btab, int nbody, real tnow, string *fields,
string ifmt, string rfmt)
{
bodyptr bp;
for (bp = btab; bp < NthBody(btab, nbody); bp = NextBody(bp)) {
if (set_member(fields, TimeTag))
printf(rfmt, tnow);
if (set_member(fields, MassTag))
printf(rfmt, Mass(bp));
if (set_member(fields, PosTag)) {
printf(rfmt, Pos(bp)[0]);
printf(rfmt, Pos(bp)[1]);
printf(rfmt, Pos(bp)[2]);
}
if (set_member(fields, VelTag)) {
printf(rfmt, Vel(bp)[0]);
printf(rfmt, Vel(bp)[1]);
printf(rfmt, Vel(bp)[2]);
}
if (set_member(fields, AccTag)) {
printf(rfmt, Acc(bp)[0]);
printf(rfmt, Acc(bp)[1]);
printf(rfmt, Acc(bp)[2]);
}
if (set_member(fields, PhiTag))
printf(rfmt, Phi(bp));
if (set_member(fields, SmoothTag))
printf(rfmt, Smooth(bp));
if (set_member(fields, RhoTag))
printf(rfmt, Rho(bp));
if (set_member(fields, EntFuncTag))
printf(rfmt, EntFunc(bp));
if (set_member(fields, UinternTag))
printf(rfmt, Uintern(bp));
if (set_member(fields, UdotIntTag))
printf(rfmt, UdotInt(bp));
if (set_member(fields, UdotRadTag))
printf(rfmt, UdotRad(bp));
if (set_member(fields, UdotVisTag))
printf(rfmt, UdotVis(bp));
if (set_member(fields, TauTag))
printf(rfmt, Tau(bp));
if (set_member(fields, BirthTag))
printf(rfmt, Birth(bp));
if (set_member(fields, DeathTag))
printf(rfmt, Death(bp));
if (set_member(fields, TypeTag))
printf(ifmt, (int) Type(bp));
if (set_member(fields, KeyTag))
printf(ifmt, Key(bp));
if (set_member(fields, AuxTag))
printf(rfmt, Aux(bp));
if (set_member(fields, AuxVecTag)) {
printf(rfmt, AuxVec(bp)[0]);
printf(rfmt, AuxVec(bp)[1]);
printf(rfmt, AuxVec(bp)[2]);
}
printf("\n");
}
}
