import sys
auto_directory = !echo $AUTO_DIR
if auto_directory == ['']:
    home = !echo $HOME
    auto_directory = home[0]+'/auto/07p/'
    sys.path.append(auto_directory+'/python')
else:
    sys.path.append(auto_directory[0] + '/python')

import AUTOCommands as ac
import AUTOclui as acl
import interactiveBindings as ib
import runAUTO as ra

runner = ra.runAUTO()

lpa = ac.load('lpa', runner=runner);

lpa.run()

!ls

import pandas as pd
from matplotlib import pylab as pl

content = None
with open('b.lpa', 'r') as f:
    content = f.readlines()

content[15:20]

content_csv = [[el for el in content[15].split(' ') if len(el) > 0 and el != '\n']]
content_csv[0][0] = 'branch'
column_names = content_csv[0]

column_names

for line in content:
    dummy = line.split(' ')
    dummy = [el for el in dummy if len(el) > 0 and el != '\n']
    if dummy[0] == '0':
        continue
        
    for el_i, el in enumerate(dummy):
        if el_i < 4:
            dummy[el_i] = int(el)
        else:
            dummy[el_i] = float(el)
            
    if len(dummy) > 1:
        content_csv.append(dummy)

df = pd.DataFrame(content_csv, columns=column_names)

df

scatter(df[df['branch'] == 1].tot,df[df['branch'] == 1].aL)
scatter(df[df['branch'] == 2].tot,df[df['branch'] == 2].aL)

import parseB

pb_obj = parseB.parseBMixin()

pb_obj.readFilename('b.lpa')

pb_obj = parseB.parseB()

b_lpa = open('b.lpa', 'r')
ab_obj.read(b_lpa)

pb_obj.read(b_lpa)

pb_obj.branches[:2]

pb_obj.branches[0].name

pb_obj.branches[0].keys()

pb_obj.branches[0]['tot']

pb_obj.branches[0].todict()

pb_obj.branches[0].labels

pb_obj.branches[1].labels

pb_obj.branches[1]['tot'][37]

import parseD as pd

pd_obj = pd.parseD()
pd_obj.read(open('d.lpa', 'r'))

pd_obj[0]

for i in range(60,100):
    print 'Branch number',pd_obj[i]['Branch number'],'eigenvalues',pd_obj[i]['Eigenvalues']

pd_obj[71]

import parseS as ps

ps_obj = ps.parseS()

ps_obj.read(open('s.lpa', 'r'))

ps_obj[0]

ps_obj[0].__str__()

ps_obj[0].data

dlpa = None
with open('d.lpa', 'r') as f:
    dlpa = f.readlines()

dlpa[:20]