query.py

Go to the documentation of this file.

## @file query.py Support routines for @ref swarm-query "swarm-query.py" command-line utility.
#
#  The routines here are not documented because it solely consists of generating
#  a table output consistent with "swarm query" command.
#

from logdb import IndexedLogDB, PKey
from bsddb3.db import *
from logrecord import LogRecord
from struct import pack, unpack
from functools import partial
import math
from exceptions import *


RAD2DEG = 180.0/math.pi
class Keplerian(object):
    ASTROCENTRIC, BARYCENTRIC, JACOBI, ORIGIN = range(4)
    choices = [ "astrocentric", "barycentric", "jacobi", "origin" ]
    def __init__(self,type):
        if isinstance(type,str):
            self.type = Keplerian.choices.index(type)
        else:
            self.type = type
    def __repr__(self):
        return "Keplerian(%s)" % Keplerian.choices[self.type]



TABLE, KEYS = range(2)
def print_record(print_mode, r, body_range):
    k, l = r
    if print_mode == TABLE :
        i = 0
        for b in l.bodies:
            if(body_range.contains(i)):
                print "%10d %lg  %6d %6d  %9.2g  %10.4g %10.4g %10.4g  %10.5lg %10.5lg %10.5lg  %d" % (l.msgid, l.time, l.sys, i, b.mass, b.position[0], b.position[1], b.position[2], b.velocity[0], b.velocity[1], b.velocity[2], l.state)
            i = i + 1

    elif isinstance( print_mode , Keplerian ):
        tp = print_mode.type
        if tp == Keplerian.JACOBI:
            body_orbits = l.bodies_in_keplerian_jacobi()
        else:
            if tp == Keplerian.ASTROCENTRIC:
                center = l.star()
            elif tp == Keplerian.BARYCENTRIC:
                center = l.barycenter()
            elif tp == Keplerian.ORIGIN:
                center = l.origin()
            else:
                raise ValueError("Wrong keplerian index")
            body_orbits = l.bodies_in_keplerian(center)
        
        for i, b, orbit in body_orbits:
            if(body_range.contains(i)):
                print "%10d %lg  %6d %6d  %9.2g  %9.5lg %9.5lg %9.5lg  %9.5lg %9.5lg %9.5lg  %d" % (l.msgid, l.time, l.sys, i, b.mass, orbit.a, orbit.e , orbit.i*RAD2DEG, orbit.O*RAD2DEG, orbit.w *RAD2DEG, orbit.M*RAD2DEG, l.state)

    elif print_mode == KEYS :
        print k
    else:
        print print_mode, type(print_mode)


def truncate(number,iterator):
    for i in range(0,number):
        yield iterator.next()

def run_with_args(args):
    d = IndexedLogDB(args.database)

    if args.keplerian != None:
      print_mode  = Keplerian(args.keplerian)
    else:
      print_mode  = TABLE


    if args.initial_conditions :
      q = d.initial_conditions(args.system_range)
    elif args.final_conditions :
      q = d.final_conditions(args.system_range)
    else:
      q = d.query(args.time_range,args.system_range,args.evt_id)

    for r in truncate(args.max_records,q):
      print_record(print_mode,r, args.body_range)