Exemplo n.º 1
0
  def analyze_prior_results(self, analysis_source):
    """ Runs analysis of previous grid search / integration results, used in an
        analyze-only mode """

    from prime.iota.iota_analysis import Analyzer
    from libtbx import easy_pickle as ep

    if os.path.isdir(analysis_source):
      int_folder = os.path.abspath(analysis_source)
    else:
      try:
        int_folder = os.path.abspath(os.path.join(os.curdir,
                     'integration/{}/image_objects'.format(analysis_source)))
      except ValueError:
        print 'Run #{} not found'.format(analysis_source)

    if os.path.isdir(int_folder):
      int_list = [os.path.join(int_folder, i) for i in os.listdir(int_folder)]
      img_objects = [ep.load(i) for i in int_list]

      analysis = Analyzer(img_objects, None, self.iver, self.now)
      analysis.print_results()
      analysis.unit_cell_analysis(self.params.analysis.cluster_threshold,
                                  int_folder, False)
      analysis.print_summary(None)
      analysis.show_heatmap()
    else:
      print 'No results found in {}'.format(int_folder)
Exemplo n.º 2
0
      init = ep.load('iota.ini')
    else:
      print 'This run not initialized. Use "--mpi init" option'
      misc.iota_exit(iota_version, True)

  if 'ana' in args.mpi:
    img_objects = [ep.load(os.path.join(init.obj_base, i)) for i in os.listdir(init.obj_base)]
    int_objects = [i for i in img_objects if i.fail == None]
    if len(int_objects) != 0:

      # Analysis of integration results
      analysis = Analyzer(int_objects, init.logfile, iota_version, init.now)
      analysis.print_results()
      analysis.unit_cell_analysis(init.params.analysis.cluster_threshold,
                                  init.int_base)
      analysis.print_summary(init.int_base)
      analysis.make_prime_input(init.int_base)
    else:
      print "No images integrated"

  else:

    if 'imp' in args.mpi:
      inp_list = init.input_list
      msg = "Importing {} images".format(len(inp_list))

    elif 'gri' in args.mpi:
      obj_list = [ep.load(os.path.join(init.obj_base, i)) for i in os.listdir(init.obj_base)]
      inp_list = [[obj, item] for obj in obj_list for item in range(len(obj.grid))]

    elif 'sel' in args.mpi or 'fin' in args.mpi: