Jython script that uses SNT to perform a 'brute force' ML query

BruteForceMLQuery.python

from sc.fiji.snt.io import MouseLightLoader, MouseLightQuerier
from sc.fiji.snt.analysis import NodeStatistics, TreeStatistics
from sc.fiji.snt.annotation import AllenUtils

soma_area = AllenUtils.getCompartment("MOs")
target_area = AllenUtils.getCompartment("ACA")
input_ids = MouseLightQuerier.getIDs(soma_area)
output_ids = []

for in_id in input_ids:
    print("processing %s..." % in_id)
    tree = MouseLightLoader(in_id).getTree('axon')
    tree_stats = TreeStatistics(tree)
    node_stats = tree_stats.getNodeStatistics('tips')
    tips_at_target = node_stats.get(target_area)
    if len(tips_at_target) > 9:
        output_ids.append(in_id)


print(output_ids)