def run(results, cmdenv, tdb): from commands.commandenv import ResultRow cmdenv = results.cmdenv tdb = cmdenv.tdb srcSystem = cmdenv.nearSystem results.summary = ResultRow() results.limit = cmdenv.limit fields = [ "si.station_id", "JULIANDAY('NOW') - JULIANDAY(MAX(si.modified))", "stn.ls_from_star", ] joins = [] wheres = [] havings = [] if cmdenv.minAge: wheres.append( "(JULIANDAY('NOW') - JULIANDAY(si.modified) >= {})".format( cmdenv.minAge)) nearSys = cmdenv.nearSystem if nearSys: maxLy = cmdenv.maxLyPer or tdb.maxSystemLinkLy maxLy2 = maxLy**2 fields.append("dist2(" "sys.pos_x, sys.pos_y, sys.pos_z," "{}, {}, {}" ") AS d2".format( nearSys.posX, nearSys.posY, nearSys.posZ, )) joins.append("INNER JOIN System sys USING (system_id)") wheres.append("""( sys.pos_x BETWEEN {} and {} AND sys.pos_y BETWEEN {} and {} AND sys.pos_z BETWEEN {} and {} )""".format( nearSys.posX - maxLy, nearSys.posX + maxLy, nearSys.posY - maxLy, nearSys.posY + maxLy, nearSys.posZ - maxLy, nearSys.posZ + maxLy, )) havings.append("d2 <= {}".format(maxLy2)) else: fields.append("0") fieldStr = ','.join(fields) if joins: joinStr = ' '.join(joins) else: joinStr = '' if wheres: whereStr = 'WHERE ' + ' AND '.join(wheres) else: whereStr = '' if havings: haveStr = 'HAVING ' + ' AND '.join(havings) else: haveStr = '' stmt = """ SELECT {fields} FROM StationItem as si INNER JOIN Station stn USING (station_id) {joins} {wheres} GROUP BY 1 {having} ORDER BY 2 DESC """.format( fields=fieldStr, joins=joinStr, wheres=whereStr, having=haveStr, ) cmdenv.DEBUG1(stmt) for (stnID, age, ls, dist2) in tdb.query(stmt): cmdenv.DEBUG2("{}:{}:{}", stnID, age, ls) row = ResultRow() row.station = tdb.stationByID[stnID] row.age = age if ls: row.ls = "{:n}".format(ls) else: row.ls = "?" row.dist = dist2**0.5 results.rows.append(row) if cmdenv.route and len(results.rows) > 1: def walk(startNode, dist): rows = results.rows startNode = rows[startNode] openList = set(rows) path = [startNode] openList.remove(startNode) while len(path) < len(rows): lastNode = path[-1] distFn = lastNode.station.system.distanceTo nearest = min(openList, key=lambda row: distFn(row.station.system)) openList.remove(nearest) path.append(nearest) dist += distFn(nearest.station.system) return (path, dist) if cmdenv.near: bestPath = walk(0, results.rows[0].dist) else: bestPath = (results.rows, float("inf")) for i in range(len(results.rows)): path = walk(i, 0) if path[1] < bestPath[1]: bestPath = path results.rows[:] = bestPath[0] if cmdenv.limit: results.rows[:] = results.rows[:cmdenv.limit] return results
def run(results, cmdenv, tdb): from commands.commandenv import ResultRow cmdenv = results.cmdenv tdb = cmdenv.tdb srcSystem = cmdenv.nearSystem results.summary = ResultRow() results.limit = cmdenv.limit fields = [ "si.station_id", "JULIANDAY('NOW') - JULIANDAY(MAX(si.modified))", "stn.ls_from_star", ] joins = [] wheres = [] havings = [] if cmdenv.minAge: wheres.append( "(JULIANDAY('NOW') - JULIANDAY(si.modified) >= {})" .format(cmdenv.minAge) ) nearSys = cmdenv.nearSystem if nearSys: maxLy = cmdenv.maxLyPer or tdb.maxSystemLinkLy maxLy2 = maxLy ** 2 fields.append( "dist2(" "sys.pos_x, sys.pos_y, sys.pos_z," "{}, {}, {}" ") AS d2".format( nearSys.posX, nearSys.posY, nearSys.posZ, )) joins.append("INNER JOIN System sys USING (system_id)") wheres.append("""( sys.pos_x BETWEEN {} and {} AND sys.pos_y BETWEEN {} and {} AND sys.pos_z BETWEEN {} and {} )""".format( nearSys.posX - maxLy, nearSys.posX + maxLy, nearSys.posY - maxLy, nearSys.posY + maxLy, nearSys.posZ - maxLy, nearSys.posZ + maxLy, )) havings.append("d2 <= {}".format(maxLy2)) else: fields.append("0") fieldStr = ','.join(fields) if joins: joinStr = ' '.join(joins) else: joinStr = '' if wheres: whereStr = 'WHERE ' + ' AND '.join(wheres) else: whereStr = '' if havings: haveStr = 'HAVING ' + ' AND '.join(havings) else: haveStr = '' stmt = """ SELECT {fields} FROM StationItem as si INNER JOIN Station stn USING (station_id) {joins} {wheres} GROUP BY 1 {having} ORDER BY 2 DESC """.format( fields=fieldStr, joins=joinStr, wheres=whereStr, having=haveStr, ) cmdenv.DEBUG1(stmt) for (stnID, age, ls, dist2) in tdb.query(stmt): cmdenv.DEBUG2("{}:{}:{}", stnID, age, ls) row = ResultRow() row.station = tdb.stationByID[stnID] row.age = age if ls: row.ls = "{:n}".format(ls) else: row.ls = "?" row.dist = dist2 ** 0.5 results.rows.append(row) if cmdenv.route and len(results.rows) > 1: def walk(startNode, dist): rows = results.rows startNode = rows[startNode] openList = set(rows) path = [startNode] openList.remove(startNode) while len(path) < len(rows): lastNode = path[-1] distFn = lastNode.station.system.distanceTo nearest = min(openList, key=lambda row: distFn(row.station.system)) openList.remove(nearest) path.append(nearest) dist += distFn(nearest.station.system) return path, dist if cmdenv.near: bestPath = walk(0, results.rows[0].dist) else: bestPath = (results.rows, float("inf")) for i in range(len(results.rows)): path = walk(i, 0) if path[1] < bestPath[1]: bestPath = path results.rows[:] = bestPath[0] if cmdenv.limit: results.rows[:] = results.rows[:cmdenv.limit] return results