PyColloid/src/InitialValues.py at master · JARA99/PyColloid · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
#-----------------------------------------------------------------------------------------#
#----------------------------------   Initial Values   -----------------------------------#
#-----------------------------------------------------------------------------------------#
#   Date: 11/2021                                                                         #
#   File: InitialValues.py                                                                #
#-----------------------------------------------------------------------------------------#


#-----------------------------------------------------------------------------------------#
#                                        Packages                                         #
#-----------------------------------------------------------------------------------------#

import csv
import random as rd     # Librería para generar números aleatorios
import math

#-----------------------------------------------------------------------------------------#
#                                    Global Variables                                     #
#-----------------------------------------------------------------------------------------#

margin = 0
dec = 40

#-----------------------------------------------------------------------------------------#
#                                          Code                                           #
#-----------------------------------------------------------------------------------------#


#-----------------------------------------------------------------------------------------#
#                                         Classes                                         #
#-----------------------------------------------------------------------------------------#

class coloide:  # Clase para las partículas
    def __init__(self, ID, masa, carga, radio, rx, ry, vx, vy, ax, ay):
        self.ID = ID
        self.masa = masa
        self.carga = carga
        self.radio = radio
        self.rx = rx
        self.ry = ry
        self.vx = vx
        self.vy = vy
        self.ax = ax
        self.ay = ay

#-----------------------------------------------------------------------------------------#
#                                        Functions                                        #
#-----------------------------------------------------------------------------------------#

def LowAPF(i,m,q,r,x,y,vx,vy,ax,ay,Lista):   # Función para asignar valores random en un rango dado
    mas= round(rd.uniform(m[0],m[1]),dec)
    car= round(rd.uniform(q[0],q[1]),dec)
    r  = round(rd.uniform(r[0],r[1]),dec)
    rx = round(rd.uniform(x[0],x[1]),dec)
    ry = round(rd.uniform(y[0],y[1]),dec)
    vx = round(rd.uniform(vx[0],vx[1]),dec)
    vy = round(rd.uniform(vy[0],vy[1]),dec)
    ax = round(rd.uniform(ax[0],ax[1]),dec)
    ay = round(rd.uniform(ay[0],ay[1]),dec)
    Lista.append(coloide(i, mas, car, r, rx, ry, vx, vy, ax, ay))

def export(Lista,nombre,N):   # Función para crear un archivo .csv con toda la información de las particulas dadas
    a = str(nombre)+'.csv'
    f = open(a, 'w', newline='')
    w = csv.writer(f)
    w.writerow(["Id", "m", "q","radius","rx","ry","vx","vy","ax","ay"])  # Encabezado
    w.writerow([])
    for ide in range(N):
        col=[Lista[ide].ID,Lista[ide].masa,Lista[ide].carga,Lista[ide].radio,
             Lista[ide].rx,Lista[ide].ry,Lista[ide].vx,Lista[ide].vy,Lista[ide].ax,Lista[ide].ay]
        w.writerow(col)
    f.close()


def HighAPF(n_particles,r_m,r_q,r_r,r_x,r_y,r_vx,r_vy,r_ax,r_ay):
    '''Creates output.csv file with initial values for high atomic packing number.'''

    radius = r_r[1] + margin
    x_max = r_x[1]
    y_max = r_y[1]


    lines = math.floor(x_max/(2*radius))
    cols = math.floor(y_max/(2*radius))

    if n_particles > lines*cols:
        print('Too many particles')
        return

    points = []

    for line in range(lines):
        for col in range(cols):
            x_coord = (line*2*radius) + radius
            y_coord = (col*2*radius) + radius
            points.append([x_coord,y_coord])

    while len(points) > n_particles:
        remove_point = rd.randint(0,len(points)-1)
        # print(remove_point)
        points.pop(remove_point)

    # print(points)

    particles = []

    for particle in range(n_particles):
        mass= round(rd.uniform(r_m[0],r_m[1]),dec)
        charge = round(rd.uniform(r_q[0],r_q[1]),dec)
        rad  = round(rd.uniform(r_r[0],r_r[1]),dec)
        rx = points[particle][0]
        ry = points[particle][1]
        vx = round(rd.uniform(r_vx[0],r_vx[1]),dec)
        vy = round(rd.uniform(r_vy[0],r_vy[1]),dec)
        ax = round(rd.uniform(r_ax[0],r_ax[1]),dec)
        ay = round(rd.uniform(r_ay[0],r_ay[1]),dec)
        particles.append([particle, mass, charge, rad, rx, ry, vx, vy, ax, ay])

    with open('output.csv','w',newline='') as file:
        data = csv.writer(file)
        data.writerow(["Id", "m", "q","radius","rx","ry","vx","vy","ax","ay"])
        data.writerow([])
        data.writerows(particles)
        data.writerow([])
        data.writerows(particles)
        file.close()

    # print(particles)