18j-iamp/figs/atoms.fig/crystal.py

25 lines
530 B
Python

#!/usr/bin/env python3
from math import *
import random
import sys
# size of lattice
N=5
# configuration
config=[]
for i in range(N):
for j in range(N):
for k in range(N):
config.append([2*i+((j+k)%2),sqrt(3)*(j+(k%2)/3),2*sqrt(6)/3*k])
for i in range(len(config)):
print(str(config[i][0])+"+cos(u)*sin(v)", end=",")
print(str(config[i][1])+"+sin(u)*sin(v)", end=",")
print(str(config[i][2])+"+cos(v)", end=" ")
print("with pm3d", end="")
if i<len(config)-1:
print(", \\")