44
55import os
66import argparse
7- from lightdock .constants import DEFAULT_CLUSTER_FOLDER , GSO_OUTPUT_FILE , EVALUATION_FILE , SCORING_FILE , \
7+ from lightdock .constants import DEFAULT_SWARM_FOLDER , GSO_OUTPUT_FILE , EVALUATION_FILE , SCORING_FILE , \
88 LIGHTDOCK_PDB_FILE , CLUSTER_REPRESENTATIVES_FILE
99from lightdock .util .logger import LoggingManager
1010from lightdock .util .analysis import read_rmsd_and_contacts_data , read_lightdock_output , write_ranking_to_file , \
1111 read_cluster_representatives_file
1212
1313
14- log = LoggingManager .get_logger ('lightdock_rank ' )
14+ log = LoggingManager .get_logger ('lgd_rank ' )
1515
1616
1717def parse_command_line ():
18- parser = argparse .ArgumentParser (prog = 'lightdock_rank ' )
18+ parser = argparse .ArgumentParser (prog = 'lgd_rank ' )
1919 parser .add_argument ("num_clusters" , help = "number of clusters to consider" , type = int , metavar = "num_clusters" )
2020 parser .add_argument ("steps" , help = "steps to consider" , type = int , metavar = "steps" )
2121 parser .add_argument ("-c" , "--clashes_cutoff" , help = "clashes cutoff" , dest = "clashes_cutoff" , type = float )
@@ -39,18 +39,18 @@ def parse_command_line():
3939 num_clusters_found = 0
4040 for cluster_id in range (args .num_clusters ):
4141 if args .result_file :
42- result_file_name = os .path .join (DEFAULT_CLUSTER_FOLDER + str (cluster_id ), args .result_file )
42+ result_file_name = os .path .join (DEFAULT_SWARM_FOLDER + str (cluster_id ), args .result_file )
4343 else :
44- result_file_name = os .path .join (DEFAULT_CLUSTER_FOLDER + str (cluster_id ),
44+ result_file_name = os .path .join (DEFAULT_SWARM_FOLDER + str (cluster_id ),
4545 (GSO_OUTPUT_FILE % args .steps ))
4646
47- cluster_representatives_file = os .path .join (DEFAULT_CLUSTER_FOLDER + str (cluster_id ),
47+ cluster_representatives_file = os .path .join (DEFAULT_SWARM_FOLDER + str (cluster_id ),
4848 CLUSTER_REPRESENTATIVES_FILE )
4949 clusters = []
5050 if os .path .isfile (cluster_representatives_file ) and not args .ignore_clusters :
5151 clusters = read_cluster_representatives_file (cluster_representatives_file )
5252
53- scoring_file_name = os .path .join (DEFAULT_CLUSTER_FOLDER + str (cluster_id ), SCORING_FILE )
53+ scoring_file_name = os .path .join (DEFAULT_SWARM_FOLDER + str (cluster_id ), SCORING_FILE )
5454 try :
5555 results = read_lightdock_output (result_file_name )
5656 num_clusters_found += 1
0 commit comments