theon-1/PolyCubeCounting
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

replaced plot function

Browse Source
This commit is contained in:
Maxime Vorwerk
2023-09-25 09:33:26 +02:00
parent ec5069a9fb
commit 3cc7b5d102
2 changed files with 35 additions and 128 deletions
Download Patch File Download Diff File Expand all files Collapse all files

151
julia/Plot.jl
View File

@@ -1,131 +1,34 @@
using PyCall
using Plots
using Serialization
py"""
import numpy as np
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
function plotPolycubes(nCubes::Int64, index::Int64=-1)
pyplot()
pygui(true)
T = deserialize("julia/results.bin")
i = if (index == -1); :; else index:index end
v = T[2][nCubes][i]
n = size(v)[1]
def plotShapePY(v):
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
plotSingleShape(ax, v)
colors = (:reds, :greens, :blues)
plt.show()
X = [0, 0, 1, 1, 0, 0, 1, 1]
Y = [0, 1, 1, 0, 0, 1, 1, 0]
Z = [0, 0, 0, 0, 1, 1, 1, 1]
def plotShapesPY(vs):
n = len(vs)
if n == 1:
plotShapePY(vs[0])
else:
fig = plt.figure()
n_rows = int(np.ceil(0.6*n))
n_cols = int(np.ceil(n/n_rows))
for i in range(len(vs)):
v = vs[i]
ax = fig.add_subplot(n_cols, n_rows, i+1, projection='3d')
plotSingleShape(ax, v)
plt.show()
I = [7, 0, 0, 0, 4, 4, 6, 6, 4, 0, 3, 2]
J = [3, 4, 1, 2, 5, 6, 5, 2, 0, 1, 6, 3]
K = [0, 7, 2, 3, 6, 7, 1, 1, 5, 5, 7, 6]
def plotSingleShape(ax, v):
l = len(v)
for i in range(l):
X1, Y1, Z1 = horizontalPlane(v[i][0] , v[i][1] , v[i][2] )
X2, Y2, Z2 = horizontalPlane(v[i][0] , v[i][1] , v[i][2]+1)
X3, Y3, Z3 = verticalPlaneX( v[i][0] , v[i][1] , v[i][2] )
X4, Y4, Z4 = verticalPlaneX( v[i][0] , v[i][1]+1, v[i][2] )
X5, Y5, Z5 = verticalPlaneY( v[i][0] , v[i][1] , v[i][2] )
X6, Y6, Z6 = verticalPlaneY( v[i][0]+1, v[i][1] , v[i][2] )
ax.plot_surface(X1, Y1, Z1, alpha=0.8, color='orange')
ax.plot_surface(X2, Y2, Z2, alpha=0.8, color='orange')
ax.plot_surface(X3, Y3, Z3, alpha=0.8, color='green' )
ax.plot_surface(X4, Y4, Z4, alpha=0.8, color='green' )
ax.plot_surface(X5, Y5, Z5, alpha=0.8, color='cyan' )
ax.plot_surface(X6, Y6, Z6, alpha=0.8, color='cyan' )
x_min = np.inf
x_max = -np.inf
y_min = np.inf
y_max = -np.inf
z_min = np.inf
z_max = -np.inf
for i in range(l):
x_min = np.min([x_min, v[i][0]])
x_max = np.max([x_max, v[i][0]])
y_min = np.min([y_min, v[i][1]])
y_max = np.max([y_max, v[i][1]])
z_min = np.min([z_min, v[i][2]])
z_max = np.max([z_max, v[i][2]])
biggest_length = max([x_max-x_min, y_max-y_min, z_max-z_min])
x_diff = (biggest_length - (x_max - x_min)) / 2
x_lim_min = x_min - x_diff
x_lim_max = x_max + x_diff
y_diff = (biggest_length - (y_max - y_min)) / 2
y_lim_min = y_min - y_diff
y_lim_max = y_max + y_diff
z_diff = (biggest_length - (z_max - z_min)) / 2
z_lim_min = z_min - z_diff
z_lim_max = z_max + z_diff
ax.scatter([x_lim_min, x_lim_max+1], [y_lim_min, y_lim_max+1], [z_lim_min, z_lim_max+1], alpha=0)
def getPoints(a):
x = a[0]
y = a[1]
z = a[2]
p = [
[x, y, z ],
[x+1, y, z ],
[x+1, y+1, z ],
[x, y+1, z ],
[x, y, z+1],
[x+1, y, z+1],
[x+1, y+1, z+1],
[x, y+1, z+1]
]
return p
def horizontalPlane(x, y, z):
one = np.ones(4).reshape(2, 2)
r1 = [x, x + 1]
r2 = [y, y + 1]
X, Y = np.meshgrid(r1, r2)
Z = one*z
return X, Y, Z
def verticalPlaneX(x, y, z):
one = np.ones(4).reshape(2, 2)
r1 = [x, x + 1]
r2 = [z, z + 1]
X, Z = np.meshgrid(r1, r2)
Y = one*y
return X, Y, Z
def verticalPlaneY(x, y, z):
one = np.ones(4).reshape(2, 2)
r1 = [y, y + 1]
r2 = [z, z + 1]
Y, Z = np.meshgrid(r1, r2)
X = one*x
return X, Y, Z
"""
function plotShape(t::Vector{Int64})
T = deserialize("results.bin")
v = T[2][t[1]][t[2]]
py"plotShapePY"(v)
end
function plotShapes(t::Vector{Int64})
T = deserialize("results.bin")
v = T[2][t[1]]
py"plotShapesPY"(v)
P = Vector{Plots.Plot{Plots.PyPlotBackend}}(undef, n)
@show v
for i ∈ 1:n
polyCube = v[i]
p = mesh3d()
for cube ∈ polyCube
@show cube
mesh3d!(p, X.+cube[1], Y.+cube[2], Z.+cube[3], connections=(I, J, K), legend=:none, c=colors[rand(1:3)])
end
P[i] = p
end
plot(P..., colorbar=false)
end

12
julia/PolycubeCounting.jl
View File

@@ -6,10 +6,10 @@ using Serialization
using Combinatorics
function options()
println("scanForPolycubes(Int::n): scans for Polycubes of size <=n")
println("scanForPolycubes(MaxSize::Int64): scans for Polycubes of size <=MaxSize")
println("countPolycubes(): opens Polycube storage and displays the amount of Polycubes for the generated sizes")
println("listPolycubes(): lists all Polycubes from the Polycube storage")
println("plot(Vector{Int}::v): plots Polycubes of size v[1], or just v[2] from the list")
println("plotPolycubes(nCubes::Int64, index::Int64=-1): plots Polycubes of size v[1], or just v[2] from the list")
end
function scanForPolycubes(MaxSize::Int64)
@@ -18,6 +18,7 @@ function scanForPolycubes(MaxSize::Int64)
immutableCube = getImmutableOrientedPolycube(singletonCube)
D[immutableCube.hash] = immutableCube
evaluatePolycube(singletonCube, D, MaxSize)
serialize("julia/results.bin", sanitize(D, MaxSize))
end
function evaluatePolycube(polycube::Polycube, D::Dict{UInt, ImmutableOrientedPolycube}, MaxSize::Int64)
@@ -40,7 +41,7 @@ function evaluatePolycube(polycube::Polycube, D::Dict{UInt, ImmutableOrientedPol
end
function countPolycubes()
T = deserialize("results.bin")
T = deserialize("julia/results.bin")
n = T[1]
for i ∈ 1:n
print("n = ")
@@ -51,7 +52,7 @@ function countPolycubes()
end
function listPolycubes()
T = deserialize("results.bin")
T = deserialize("julia/results.bin")
print("max size: ")
println(T[1])
for V ∈ T[2]
@@ -90,3 +91,6 @@ function sanitize(D::Dict{UInt, ImmutableOrientedPolycube}, size::Int64)
end
return (size, data)
end
Base.:+(t1 = Tuple{Int64, Int64, Int64}, t2 = Tuple{Int64, Int64, Int64}) = (t1[1] + t2[1], t1[2] + t2[2], t1[3] + t2[3])
Base.:-(t1 = Tuple{Int64, Int64, Int64}, t2 = Tuple{Int64, Int64, Int64}) = t1 + t2.*-1