example.py

"""
an example about how to load data and generate a schedule
"""
import CourseScheduling as cs
from DataHelper.loadData import DataLoading
import time
__author__ = "Jenny Zeng"
__email__ = "jennyzengzzh@gmail.com"

if __name__ == '__main__':
    start_time = time.time()

    # config upper standing units
    upper_units = 90
    # load taken info
    startQ, applied_units, taken = DataLoading.load_taken(filename="info/taken2.txt")
    # load avoid info
    avoid = DataLoading.load_avoid(filename="info/avoid.txt")
    # load graph, config if user is upper standing
    G = DataLoading.load_courses(prereq_filename="info/fullcourses_new.txt",
                                 show_upper=cs.is_upper_standing(applied_units, upper_units))
    # load requirement sheet
    R_detail, R = DataLoading.load_requirements(
        requirements=["University", "GEI", "GEII", "GEIII", "GEIV",
                      "GEV", "GEVI", "GEVII", "GEVIII", "CS-Lower-division", "CS-Upper-division",
                   "Intelligent Systems"
                      ],
        filename="info/specializations.txt")

    # update requirement table based on the taken information
    cs.update_requirements(R_detail, R, taken)
    print(R_detail)
    print(R)
    # load max width for each quarter
    max_widths = DataLoading.load_width_func_table("info/widthFunc.txt")

    # construct CourseGraph. graph is labeled after init
    graph = cs.CourseGraph(G, r_detail=R_detail, R=R, avoid=avoid, taken=taken)

    # construct Schedule with width func requirements
    L = cs.Schedule(widths=max_widths)
    # construct the scheduling class
    generator = cs.Scheduling(start_q=startQ)
    # get the best schedule when the upper bound ranges from 0 to 10, inclusive.
    L, best_u, best_r = generator.get_best_schedule(graph, L, R, 0, 10)
    print(L)
    print(best_u)
    print(R_detail)
    print(best_r)
    print("--- %s seconds ---" % (time.time() - start_time))
    # in terminal, type:
    # python -m cProfile example.py
    # to see the time and calls for each function