def load_small_frame_building_sample_data():
    """
    Sample data for the FrameBuilding object

    :param fb:
    :return:
    """

    interstorey_heights = np.array([4.9, 3.4, 3.4, 3.4])  # m
    bay_lengths = np.array([4.5, 4.5, 3.2, 3.2])
    number_of_storeys = len(interstorey_heights)
    storey_pressure = 8.0e3  # Pa
    n_bays = len(bay_lengths)

    fb = sfsimodels.FrameBuilding2D(number_of_storeys, n_bays)
    fb.interstorey_heights = interstorey_heights
    fb.floor_length = 15.4  # m
    fb.floor_width = 5.0  # m
    fb.storey_masses = storey_pressure / 9.8 * fb.floor_length * fb.floor_width * np.ones(number_of_storeys)  # kg

    fb.bay_lengths = bay_lengths
    fb.set_beam_prop("depth", 0.6, repeat="all")
    fb.set_beam_prop("width", 0.4, repeat="all")
    fb.set_column_prop("width", 0.5, repeat="all")
    fb.set_column_prop("depth", 0.5, repeat="all")
    return fb
def load_small_frame_building_sample_data():
    """
    Sample data for the FrameBuilding object

    :param fb:
    :return:
    """
    number_of_storeys = 1
    interstorey_height = 3.4  # m
    masses = 40.0e3  # kg
    n_bays = 1

    fb = sfsimodels.FrameBuilding2D(number_of_storeys, n_bays)
    fb.interstorey_heights = interstorey_height * np.ones(number_of_storeys)
    fb.floor_length = 18.0  # m
    fb.floor_width = 16.0  # m
    fb.storey_masses = masses * np.ones(number_of_storeys)  # kg

    fb.bay_lengths = [6.]
    fb.set_beam_prop("depth", [0.5], repeat="up")
    fb.set_beam_prop("width", [0.4], repeat="up")
    fb.set_column_prop("width", [0.5, 0.5], repeat="up")
    fb.set_column_prop("depth", [0.5, 0.5], repeat="up")
    return fb