euler_rozamiento.py

import os
import pandas as pd

NP = 103958
g = 9.81
L0 = (((0.1 / 10000) * (NP - 100000)) + 0.25) * 150
m = (40 / 10000) * (NP - 100000) + 50
k1 = (10 / 10000) * (NP - 100000) + 40
k2 = 0.66
c1 = (2 / 10000) * (NP - 100000) + 3
c2 = 1.5


def write_to_excel(rows):
    new_sheet = {}
    for i in range(len(rows[0])):
        column_name = rows[0][i]
        data = []
        for row in rows:
            data.append(row[i])

        new_sheet[column_name] = data

    new_df = pd.DataFrame(new_sheet)
    new_df = new_df.iloc[1:, :]

    filename = 'datos.xlsx'
    filePath = os.path.join(filename, filename)
    writer = pd.ExcelWriter(filePath, engine='xlsxwriter')

    new_df.to_excel(writer)

    writer.save()


def u_n1(un, vn, h):
    un1 = un + (h * vn)
    return un1


def v_n1(un, vn, h):
    if un >= L0:

        if vn >= 0:
            vn1 = vn + h * (g - (k1 * (((un - L0) ** k2) / m)) - ((c1 * (abs(vn)) ** c2) / m))

        else:

            vn1 = vn + h * (g - (k1 * (((un - L0) ** k2) / m)) + ((c1 * (abs(vn)) ** c2) / m))

    else:

        if vn >= 0:
            vn1 = vn + h * (g - ((c1 * (abs(vn)) ** c2) / m))

        else:

            vn1 = vn + h * (g + ((c1 * (abs(vn)) ** c2) / m))

    return vn1


def w_n1(un, vn):
    if un >= L0:
        if vn >= 0:
            wn1 = g - (k1 * (((un - L0) ** k2) / m)) - ((c1 * (abs(vn)) ** c2) / m)

        else:
            wn1 = g - (k1 * (((un - L0) ** k2) / m)) + ((c1 * (abs(vn)) ** c2) / m)

    else:

        if vn >= 0:
            wn1 = g - ((c1 * (abs(vn)) ** c2) / m)

        else:

            wn1 = g + ((c1 * (abs(vn)) ** c2) / m)

    return wn1


def main():
    rows = [["N", "h", "un", "vn", "wn"]]
    un = 0
    vn = 0
    wn = 0

    h = 0.1
    i = 1000
    n = 0

    limit = 150 * 0.9
    cumple_1 = False
    cumple_2 = True
    cumple_3 = False

    u_max = 0
    a_max = 0

    rows.append([n, h, un, vn, wn])
    print("n:", n, "| un:", un, "| vn:", vn, "| wn:", wn)

    while n < i:
        n += 1

        un1 = u_n1(un, vn, h)
        vn1 = v_n1(un1, vn, h)
        wn1 = w_n1(un, vn)

        un = un1
        vn = vn1
        wn = wn1

        rows.append([n, h, un, vn, wn])

        if limit < un < 150:
            cumple_1 = True
        elif un > 150:
            cumple_2 = False

        if un > u_max:
            u_max = un
            a_max = wn

        print("n:", n, "| un:", un, "| vn:", vn, "| wn:", wn)

    if abs(a_max) <= (2.5 * g):
        cumple_3 = True

    if cumple_1 and cumple_2 and cumple_3:
        print("Condiciones cumplidas! | k1:", k1, "| k2:", k2)
        print("U MAX:", u_max)
        print("A on MAX:", a_max)
    else:
        print("No cumple las condiciones | k1:", k1, "| k2:", k2)
        print("U MAX:", u_max)
        print("A on MAX:", a_max)

    write_to_excel(rows)


main()