random_root_locus_saver.py

import os
import txt_mixin
import numpy as np
import importlib

saved_path = '/Users/kraussry/git/teaching/random_root_locus_saved.py'


def get_N():
    if not os.path.exists(saved_path):
        N = 0
    else:
        import random_root_locus_saved
        importlib.reload(random_root_locus_saved)
        N = len(random_root_locus_saved.G_list)
    return N


def get_list(N):
    if N == 0:
        mylist = ['import control','G_list = []', '']
    else:
        myfile = txt_mixin.txt_file_with_list(saved_path)
        mylist = myfile.list
    return mylist

    
def G_to_text(G, Nstr):
    if type(Nstr) != str:
        Nstr = str(Nstr)
    ns = np.squeeze(G.num)
    ds = np.squeeze(G.den)
    numstr = np.array2string(ns,separator=',')
    denstr = np.array2string(ds,separator=',')
    num2 = 'num%s = %s' % (Nstr, numstr)
    den2 = 'den%s = %s' % (Nstr, denstr)
    Gstr = 'G%s = control.TransferFunction(num%s,den%s)' % \
               (Nstr, Nstr, Nstr)
    append_line = 'G_list.append(G%s)' % Nstr
    lines_out = [num2,den2,Gstr, append_line]
    return lines_out


def print_G_to_text(G, N):
    lines_out = G_to_text(G,N)
    if 'append' in lines_out[-1]:
        lines_out.pop(-1)
    for line in lines_out:
        print(line)
        

def save_G(G,N=None):
    if N is None:
        N = get_N()
    mylist = get_list(N)
    new_lines = G_to_text(G, N)
    mylist.append('')
    mylist.extend(new_lines)
    txt_mixin.dump(saved_path, mylist)
    return N