######################################################################
# Code for generating most of the examples computed in our abacus histories paper
# Diagrams of abacus histories are in ascii
# Greg Warrington: Aug 4, 2020

from ah import AH
from lc import LC

######################################################################
# for generating data for paper
######################################################################
def paper_table_1():
    print("======== Table 1 ========")
    lc = LC([AH.snu_abacus([8,8,8,4,4,2,2,1])])
    for i in range(10,-10,-1):
        print("S_%d: " % (i))
        AH.Sm_op(i,lc,verbose=False).print_lc_simplified()
    print()
    
def paper_figure_2():
    print("======== Figure 2 ========")
    ab = AH.snu_abacus([3,1,1,0])
    ab._apply_hc(2).print_lc()
    print()
    
def paper_figure_3():
    print("======== Figure 3 ========")
    ab = AH.snu_abacus([3,3,2])
    ab._apply_hc_skew(2).print_lc()
    print()
    
def paper_figure_4():
    print("======== Figure 4 ========")
    ab = AH.snu_abacus([4,1,1,0,0])
    ab._apply_ec(2,verbose=False).print_lc()
    print()

def paper_figure_5():
    print("======== Figure 5 ========")
    ab = AH.snu_abacus([4,4,2])
    ab._apply_ec_skew(2,verbose=False).print_lc()
    print()

def paper_figure_6():
    print("======== Figure 6 ========")
    ab = AH.snu_abacus([3,1,1,0])
    ab.Sm_op(1,LC([ab])).print_lc()
    print()

def paper_figure_9():
    print("======== Figure 9 ========")
    ab = AH.snu_abacus([3,1])
    ab.Hm_op(-2,LC([ab]),simplified=True,verbose=False).print_lc()

def paper_figure_10():
    print("======== Figure 10 ========")
    AH.H_alpha([1,2,3],simplified=True,verbose=False).print_lc()

#################################################################3

def paper_eq25():
    print("======== Eq (25) ========")
    # Sec 3.6, (25)
    ab = AH.snu_abacus([3,1])
    ans = ab.Hm_op(-2,LC([ab]),verbose=False)
    ans.print_lc_simplified()

def paper_eq26():
    print("======== Eq (26) ========")
    # Sec 3.6, (26)
    ab = AH.snu_abacus([3,1])
    ans = ab.Hm_op(1,LC([ab]),verbose=False)
    ans.print_lc_simplified()

def paper_sec36():
    print("======== Sec 3.6, first equation ========")
    # Sec 3.6, End of section
    ab = AH.snu_abacus([3,1])
    ans = ab.Cm_op(-2,LC([ab]),verbose=False)
    ans.print_lc_simplified()

def paper_sec36v2():
    print("======== Sec 3.6, second equation ========")
    # Sec 3.6, End of section
    ab = AH.snu_abacus([3,1])
    ans = ab.Cm_op(1,LC([ab]),verbose=False)
    ans.print_lc_simplified()

def paper_sec37():
    print("======== Sec 3.7, first equation ========")
    # Sec 3.7, first equation
    ab = AH.snu_abacus([2,1,1])
    ans = ab.Bm_op(-2,LC([ab]),verbose=False)
    ans.print_lc_simplified()

def paper_sec37v2():
    print("======== Sec 3.7, second equation ========")
    # Sec 3.7, second equation
    ab = AH.snu_abacus([2,1,1])
    ans = ab.Bm_op(1,LC([ab]),verbose=False)
    ans.print_lc_simplified()

def paper_example_3():
    print("======== Example 3 ========")
    # Example 3
    ab = AH.snu_abacus()
    ab.H_alpha([1,2,3],verbose=False).print_lc_simplified()
    
def generate_expansions():
    """ Generate data from examples and expansions in paper
    """
    paper_eq25()
    paper_eq26()
    paper_sec36()
    paper_sec36v2()
    paper_sec37()
    paper_sec37v2()
    paper_example_3()

def generate_figure_data():
    """ Generate data for examples 
    """
    paper_table_1()
    paper_figure_2()
    paper_figure_3()
    paper_figure_4()
    paper_figure_5()
    paper_figure_6()
    paper_figure_9()
    paper_figure_10()

# Generate the pictures of abacus histories occuring in the figures in the paper
generate_figure_data()
generate_expansions()