TESTDATA = ['/users/oli/Dropbox/Stanford_Postdoc/CODING/cctbx_testing/PolG_test_data']

import logging
import numpy as np
import matplotlib.pyplot as plt
import logging
import brewer2mpl
# Set up logging
reload(logging)  # work-around for IPython
FORMAT = '%(message)s'
logging.basicConfig(level=logging.INFO, format=FORMAT)
# pretty colors
cols = brewer2mpl.get_map('BrBG', 'Diverging', 3).mpl_colors

from xfel.clustering.cluster import Cluster
t_clus = Cluster.from_directories(TESTDATA)

sub_clusters, _ = t_clus.ab_cluster(labels=False, write_file_lists=False)

from xfel.clustering.cluster_groups import unit_cell_info
pretty_str = unit_cell_info(sub_clusters)
print pretty_str

clu_b, clu_a = sub_clusters[-2:]
print "clu_a size: {}\nclu_b size: {}".format(len(clu_a.members), len(clu_b.members))

_ = clu_b.all_frames_intensity_stats()

clu_b.pg_composition

clu_bC = clu_b.point_group_filter('C222')
clu_bP = clu_b.point_group_filter('P222') 

_ = clu_bC.all_frames_intensity_stats()

_ = clu_bP.all_frames_intensity_stats()

print clu_bP.info

clu_bP.dump_file_list(out_file_name='temp.lst')

cat temp.lst