GlobalUtils.py

import datetime
from math import *
from typing import TextIO
import numpy as np
import time
from tkinter import *
import matplotlib.pyplot as plt
import matplotlib.ticker as ticker
from PIL import ImageTk
from GlobalConstants import *
import numpy.random as rand

# Энергия
#kineticPlot = []
#potentialPlot = []
#summaryPlot = []
#forcePlot = []
#velocityPlot = []
#jerkPlot = []
#wallInteraction = []
#ballInteraction = []
#stepCount = [0]
#stepCountForce = []


# --------------- Функции для отображения и результатов --------------
def getName(elements, coordinatesFileName: str, ballStartFileName: str, freqX: float, freqY: float, velTheta: float):
    delimiter = '__'
    directory = './results/'
    date = datetime.datetime.now().strftime("%d-%m-%Y_%H-%M-%S")
    coordinatesFileName = coordinatesFileName.split('/')[-1].split('.')[0] + delimiter + 'frX=' + str(freqX) + 'frY=' + str(
        freqY) + 'vTh=' + str(velTheta)
    ballStartFileName = ballStartFileName.split('/')[-1].split('.')[0]
    numSec = 'sec=' + str(numOfSeconds)
    end = '.txt'
    name = directory + \
           coordinatesFileName + delimiter + \
           ballStartFileName + delimiter + \
           numSec + delimiter + date + end
    return name


def makeUtils(file: TextIO, width: float, height: float):
    width = str(width)
    height = str(height)
    deltaTimeStr = str(deltaTime)
    #   width      height     deltaTime
    file.write(utilsFlag + inLineDelimiter + width + inLineDelimiter + height + inLineDelimiter + deltaTimeStr + inLineDelimiter + '\n')


def Buttons():
    but_1 = Button(text='Start',
                   width=17, height=2,
                   bg='#5195fc', fg='white',
                   activebackground='#77DDE7',  # цвет нажатой кнопки
                   activeforeground='#FF2400',  # цвет надписи когда кнопка нажата
                   font='Hack 16')  # шрифт и размер надписи
    folderImage = ImageTk.PhotoImage(file="folder.png")
    but_2 = Button(image=folderImage)
    but_3 = Button(text='Stop',
                   width=17, height=2,
                   bg='#fc5151', fg='white',
                   activebackground='#77DDE7',
                   activeforeground='#FF2400',
                   font='Hack 16')
    buttons = [but_1, but_2, but_3]
    return buttons


def saveInteraction():
    linesEndFile = ['middle num of interaction = ' + str(midInteractionNum)]
    if len(wallInteraction) > 0:
        linesEndFile.append(
            'wall\tmin = ' + str(min(wallInteraction)) + '\tmax = ' + str(max(wallInteraction)) + '\tmiddle = ' + str(
                round(sum(wallInteraction) / len(wallInteraction), 1)) + '\tlen = ' + str(len(wallInteraction)))
    if len(ballInteraction) > 0:
        linesEndFile.append(
            'ball\tmin = ' + str(min(ballInteraction)) + '\tmax = ' + str(max(ballInteraction)) + '\tmiddle = ' + str(
                round(sum(ballInteraction) / len(ballInteraction), 1)) + '\tlen = ' + str(len(ballInteraction)))
    if len(wallInteraction) > 0 and len(ballInteraction) > 0:
        linesEndFile.append('all \tmin = ' + str(min(min(wallInteraction), min(ballInteraction))) + '\tmax = ' + str(
            max(max(wallInteraction), max(ballInteraction))) + '\tmiddle = ' + str(
            round((sum(ballInteraction) + sum(wallInteraction)) / (len(ballInteraction) + len(wallInteraction)),
                  1)) + '\tlen = ' + str(len(ballInteraction) + len(wallInteraction)))
    ballsInteractionFile = open('balls_end_interaction.txt', 'w')
    for line in linesEndFile:
        ballsInteractionFile.write(line + '\n')
    ballsInteractionFile.close()


def saveResults(elements):
    # Запись результатов в файл
    if isForce:
        saveInteraction()
    ballsEndFile = open('balls_end.txt', 'w')
    linesEndFile = ['\t\tx\t\t\t\t\ty\t\tradius\t\talpha\t\t\tvelocity\t\tacceleration']
    for ball in elements.balls:
        linesEndFile.append(
            str(ball.x) + ' ' + str(ball.y) + ' ' + str(ball.radius) + ' ' + str(ball.getAlpha()) + ' ' + str(
                ball.velocityAbsolute) + ' ' + str(ball.getAcceleration()))
    for line in linesEndFile:
        ballsEndFile.write(line + '\n')
    ballsEndFile.close()


def plotter():
    plt.style.use('fivethirtyeight')
    fig = plt.figure(figsize=(8, 8))
    ax = plt.subplot(111)
    ax.plot(stepCount, kineticPlot, label='Кинетическая')
    ax.plot(stepCount, potentialPlot, label='Потенциальная')
    ax.plot(stepCount, summaryPlot, label='Суммарная')
    ax.set_title('')
    chartBox = ax.get_position()
    ax.set_position([chartBox.x0, chartBox.y0, chartBox.width * 0.7, chartBox.height])
    ax.legend(loc='upper right', bbox_to_anchor=(0.9, 0.8))
    ax.xaxis.set_major_locator(ticker.MultipleLocator((stepCount[-1] // 100) * 10))
    ax.xaxis.set_minor_locator(ticker.MultipleLocator((stepCount[-1] // 100) * 2))
    ax.yaxis.set_major_locator(ticker.MultipleLocator((summaryPlot[-1]) * (stepCount[-1] // 10 * 10)))
    ax.yaxis.set_minor_locator(ticker.MultipleLocator((summaryPlot[-1]) * (stepCount[-1] // 10 * 20)))
    ax.tick_params(axis='both',
                   which='major',
                   direction='inout',
                   length=20,
                   width=2,
                   color='#e54747',
                   pad=10,
                   labelsize=10,
                   labelcolor='#000',
                   bottom=True,
                   top=True,
                   left=True,
                   right=True,
                   labelbottom=True,
                   labeltop=True,
                   labelleft=True,
                   labelright=True,
                   labelrotation=70)

    ax.tick_params(axis='both',
                   which='minor',
                   direction='out',
                   length=10,
                   width=1,
                   color='#e54747',
                   pad=10,
                   labelsize=15,
                   labelcolor='#000',
                   bottom=True,
                   top=True,
                   left=True,
                   right=True)
    ax.grid(which='major',
            color='k')
    ax.minorticks_on()
    ax.grid(which='minor',
            color='gray',
            linestyle=':')
    ax.set_ylabel('Energy, J')
    ax.set_xlabel('Steps')
    # ax.set_xlim(xmin=nrg[0], xmax=nrg[-1])
    fig.tight_layout()

    plt.show()


def plotterForce():
    plt.style.use('fivethirtyeight')
    fig = plt.figure(figsize=(8, 8))
    ax = plt.subplot(111)
    ax.plot(stepCountForce, forcePlot, label='Сила')
    ax.plot(stepCountForce, velocityPlot, label='Скорость')
    ax.plot(stepCountForce, jerkPlot, label='Рывок')
    ax.set_title('')
    chartBox = ax.get_position()
    ax.set_position([chartBox.x0, chartBox.y0, chartBox.width * 0.7, chartBox.height])
    ax.legend(loc='upper right', bbox_to_anchor=(0.9, 0.8))
    # ax.xaxis.set_major_locator(ticker.MultipleLocator((stepCountForce[-1] // 100) * 10))
    # ax.xaxis.set_minor_locator(ticker.MultipleLocator((stepCountForce[-1] // 100) * 2))
    # ax.yaxis.set_major_locator(ticker.MultipleLocator((stepCountForce[-1] // 10 * 10)))
    # ax.yaxis.set_minor_locator(ticker.MultipleLocator((stepCountForce[-1] // 10 * 20)))
    ax.tick_params(axis='both',
                   which='major',
                   direction='inout',
                   length=20,
                   width=2,
                   color='#e54747',
                   pad=10,
                   labelsize=10,
                   labelcolor='#000',
                   bottom=True,
                   top=True,
                   left=True,
                   right=True,
                   labelbottom=True,
                   labeltop=True,
                   labelleft=True,
                   labelright=True,
                   labelrotation=70)

    ax.tick_params(axis='both',
                   which='minor',
                   direction='out',
                   length=10,
                   width=1,
                   color='#e54747',
                   pad=10,
                   labelsize=15,
                   labelcolor='#000',
                   bottom=True,
                   top=True,
                   left=True,
                   right=True)
    ax.grid(which='major',
            color='k')
    ax.minorticks_on()
    ax.grid(which='minor',
            color='gray',
            linestyle=':')
    ax.set_ylabel('Force, N; Velocity, 10^(-5) * m/s ')
    ax.set_xlabel('Steps')
    # ax.set_xlim(xmin=nrg[0], xmax=nrg[-1])
    fig.tight_layout()

    plt.show()


# --------------- Функции для численного рассчета --------------


def dampeningVelocity(dampening, velocity):
    if abs(velocity) - abs(dampening) > 0 and abs(velocity) > eps:
        return velocity - dampening
    else:
        return 0


def zeroToZero(number):
    if abs(number) < eps:
        return 0
    else:
        return number


def zeroToZeroBig(number):
    if abs(number) < 1e-4:
        return 0
    else:
        return number


def customSign(number):
    if abs(number) < eps:
        return 0
    elif number > 0:
        return 1
    else:
        return -1


def isCoDirected(num1, num2):
    if (customSign(num1) > 0 and customSign(num2) < 0) or (customSign(num2) > 0 and customSign(num1) < 0):
        return False
    return True


def distanceNow(i, j):
    # Расстояние между двумя шарами в данный момент времени
    return sqrt((i.x - j.x) ** 2 + (i.y - j.y) ** 2)


def getStiffness(radius, entry, E_eff):
    return (4 / 3) * E_eff * sqrt(radius * entry)


def getDampingNormal(radius, entry, mass, c, E_eff):
    return 2 * sqrt(mass * 2 * E_eff * entry * sqrt(radius)) * c


def getDampingTangent(radius, entry, mass, c, G_eff):
    return 4 * sqrt(mass * 2 * G_eff * entry * sqrt(radius)) * c


def getBallsNum():
    num = round(rand.normal(4, 2))
    if num <= 2:
        return 2
    return num