latPoints.py

from pyqtgraph.Qt import QtCore, QtGui
import numpy as np
import itertools as itTl
from scipy import spatial
import pyqtgraph.opengl as gl
#Iterate through and constructure all possible locations for lattice points for a single basis
#once complete, remove duplicates and order points from smallest largest in x->y->z.
#Goes through additional basis' after completing 1

def latpts(basis = np.array([[0,0,0]]), plv = np.array([[1,0,0],[0,1,0],[0,0,1]]), bounds = np.array([1,1,1]), plane = False):
	latpoints = {}
	for i in basis:
		latpoints[i[0]*100+i[1]*10+i[2]]=i
		for j in plv:
		#Note bounds should be adjusted to be the solutions to being in or outside a volume (spacial testing for non rectangular shapes)
			if i[0] + j[0] <= bounds[0] and i[1] + j[1] <= bounds[1] and i[2] + j[2] <= bounds[2]:
				a=(i[0]*100+i[1]*10+i[2])+(j[0]*100+j[1]*10+j[2])
				latpoints[a] = i + j

	return latpoints