#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
/home/magnus/rna-evo-malibu/ade # run here not up
"""
from __future__ import print_function

import argparse
import os
import glob


def get_parser():
    parser = argparse.ArgumentParser(
        description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter)

    parser.add_argument('-d', "--dryrun",
                        action="store_true", help="dry run", default=False)

    parser.add_argument('-p', '--path', help="", default='')
    parser.add_argument('-c', '--case', help="only one case, for test")
    parser.add_argument('-e', '--exe', help="only one case, for test")
    parser.add_argument("-v", "--verbose",
                        action="store_true", help="be verbose")
    return parser

def exe(cmd, dryrun=False):
    print(cmd)
    if not dryrun: os.system(cmd)

if __name__ == '__main__':
    parser = get_parser()
    args = parser.parse_args()

    if args.path:
        os.chdir(args.path)
    root = os.getcwd()
    cases = glob.glob('*')
    for c in cases:
        # print('Case: ', c)
        # mode only for a specific case
        os.chdir(root)  # change to run and go to another case
        if args.case: # only if this is used
            if c != args.case:
                if args.verbose: print('!!! skip ' + c + '!!!')
                continue

        # if not break ed, use this
        if os.path.isdir(c):
            print ('Inside %s' % c)
            os.chdir(c) # go inside a folder
            if args.exe:
                exe(args.exe)
            #subcases = glob.glob('*.fa')
            #for sc in subcases:
            ## for i in [1000]: #
            ##     exe('rm *min.out.*.pdb', dryrun)
            ##     exe('mkdir %s_top%i' % (c, i), dryrun)
            ##     exe('extract_lowscore_decoys.py *min.out %i' % (i), dryrun)
            ##     exe('mv -v *min.out.*.pdb %s_top%i' % (c, i), dryrun)
            ## os.chdir(root)
            malibu_folder = c.replace('_cst', 'cst').replace('farna_', '')
            print(c, malibu_folder)
            # exe('trash top100')
            exe('mkdir top100')
            exe('pwd')
            os.chdir('top100')  # cd top100 didn't work ?!
            exe('rsync -v malibu:"~/rna-evo-malibu/_rp13_Stanford/_reg_vs_cst/%s/*.pdb" . ' %  malibu_folder)
            exe('rm c.*pdb')
            exe('rm *out.1[0-9][1-9]*.pdb')
            exe('rm *out.1[1-9]0*.pdb')
            exe('rm *out.200.pdb')
            exe('rm p[0-9].pdb')
            exe('rna_pdb_toolsx.py --rpr --renumber_residues --inplace  *')
            os.chdir('..')
            exe('evox.py -c rp13')