def bv2conc(self, bv: BitVector) -> Tuple[float, float]:
if len(bv) == 0:
return (self.lb, self.ub)
left = self.lb
left += bv2int(bv) * self.binwidth
right = left + self.binwidth
return (left, right)
def bv2conc(self, bv: BitVector) -> Tuple[float, float]:
nbits = len(bv)
if nbits == 0:
return (self.lb, self.ub)
index = bv2int(bv)
if self.periodic:
index = graytobin(index)
eps = (self.ub - self.lb) / (2**nbits)
left = self.lb + index * eps
right = self.lb + (index + 1) * eps
return (left, right)
def pixel2D(mgr, xspace, yspace, pred, title=None, fname=None, invertcolor=False, opacity=1.0):
"""
Plot a 2D set represented by a predicate as a pixel image.
Parameters
----------
mgr : dd manager
xspace : tuple (str, symbolic space)
horizontal axis name and space
yspace : tuple (str, symbolic space)
vertical axis name and space
pred : bdd
Predicate representing set over xspace and yspace
Returns
-------
matplotlib figure
matplotlib axis
"""
xname, xgrid = xspace
yname, ygrid = yspace
support = pred.support
xbits = len([bit for bit in support if bv_var_name(bit) == xname])
ybits = len([bit for bit in support if bv_var_name(bit) == yname])
xbins = 2**xbits
ybins = 2**ybits
if invertcolor:
mask = np.full((ybins, xbins), 111-(111*opacity), dtype=np.int8)
else:
mask = np.full((ybins, xbins), 111*opacity, dtype=np.int8)
xpts = []
ypts = []
config = mgr.configure() # pick_iter alters config so save config state
# Add all BDD assignments to a list of points
for pt in mgr.pick_iter(pred):
xvars = [k for k, v in pt.items() if bv_var_name(k) == xname]
yvars = [k for k, v in pt.items() if bv_var_name(k) == yname]
xvars.sort() # Sort by bit names
yvars.sort()
xbv = [pt[bit] for bit in xvars] # FIXME: Issues if a bit is skipped. Assumes no gaps in bit range.
ybv = [pt[bit] for bit in yvars]
# bv2int(xbv)
ycoord = bv2int(ybv)
xcoord = bv2int(xbv)
if ygrid.periodic:
ycoord = graytobin(ycoord)
if xgrid.periodic:
xcoord = graytobin(xcoord)
if invertcolor:
mask[ycoord, xcoord] = 111*opacity
else:
mask[ycoord, xcoord] = 111-(111*opacity)
mgr.configure(reordering=config['reordering']) # Reinstate config state
fig, ax = plt.subplots()
if invertcolor:
ax.pcolormesh(mask, edgecolors='w', cmap = plt.cm.bone, vmin=0, vmax = 111, linewidths=.01)
else:
ax.pcolormesh(mask, edgecolors='k', cmap = plt.cm.bone, vmin=0, vmax = 111, linewidths=.01)
# ax.set_xlim(xgrid.lb, xgrid.ub)
ax.set_xlabel(xname)
# ax.set_ylim(ygrid.lb, ygrid.ub)
ax.set_ylabel(yname)
if title:
ax.set_title(title)
if fname is not None:
extent = ax.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
fig.savefig(str(fname)+'.png', dpi=400, bbox_inches=extent.expanded(1.1, 1.2))
else:
plt.show()
plt.close()
return
def plot3D_QT(mgr, xspace, yspace, zspace, pred, opacity=255):
"""Somewhat buggy pyqtgraph plotting"""
from pyqtgraph.Qt import QtCore, QtGui
import pyqtgraph.opengl as gl
app = QtGui.QApplication([])
w = gl.GLViewWidget()
w.opts['distance'] = 200
w.show()
# w.setWindowTitle('pyqtgraph example: GLVolumeItem')
# print(w.width())
# print(w.height())
# w.setFixedWidth(640)
# w.setFixedHeight(480)
w.resizeGL(1600, 1200)
xname, xgrid = xspace
yname, ygrid = yspace
zname, zgrid = zspace
# Construct spaces
support = pred.support
xbits = len([bit for bit in support if bv_var_name(bit) == xname])
ybits = len([bit for bit in support if bv_var_name(bit) == yname])
zbits = len([bit for bit in support if bv_var_name(bit) == zname])
xbins = 2**xbits
ybins = 2**ybits
zbins = 2**zbits
# Construct bitmask
mask = np.full((xbins, ybins, zbins), False)
config = mgr.configure() # pick_iter alters config so save config state
for pt in mgr.pick_iter(pred):
xvars = [k for k, v in pt.items() if bv_var_name(k) == xname]
yvars = [k for k, v in pt.items() if bv_var_name(k) == yname]
zvars = [k for k, v in pt.items() if bv_var_name(k) == zname]
xvars.sort() # Sort by bit names
yvars.sort()
zvars.sort()
xbv = [pt[bit] for bit in xvars]
ybv = [pt[bit] for bit in yvars]
zbv = [pt[bit] for bit in zvars]
x_idx = bv2int(xbv) if not dynamicperiodic(xgrid) else graytobin(bv2int(xbv))
y_idx = bv2int(ybv) if not dynamicperiodic(ygrid) else graytobin(bv2int(ybv))
z_idx = bv2int(zbv) if not dynamicperiodic(zgrid) else graytobin(bv2int(zbv))
mask[x_idx, y_idx, z_idx] = True
d2 = np.empty(mask.shape + (4,), dtype=np.ubyte)
d2[..., 0] = mask.astype(np.float) * 255
d2[..., 1] = mask.astype(np.float) * 255
d2[..., 2] = mask.astype(np.float) * 255
d2[..., 3] = mask.astype(np.float) * opacity
v = gl.GLVolumeItem(d2, smooth=False)
v.translate(-xbins//2,
-ybins//2,
-zbins//2)
w.addItem(v)
g = gl.GLGridItem()
g.scale(xbins//10, ybins//10, zbins//10)
w.addItem(g)
ax = gl.GLAxisItem()
w.addItem(ax)
QtGui.QApplication.instance().exec_()
mgr.configure(reordering=config['reordering']) # Reinstate config state
def plot3D(mgr, xspace, yspace, zspace, pred,
opacity=40, view=None, title=None, fname=None, **kwargs):
"""Matplotlib based plotter with voxels"""
voxelcolors = '#7A88CC' + format(opacity, "02x")
edgecolors = '#000000' + format(opacity // 3, "02x")
xname, xgrid = xspace
yname, ygrid = yspace
zname, zgrid = zspace
# Construct spaces
support = pred.support
xbits = len([bit for bit in support if bv_var_name(bit) == xname])
ybits = len([bit for bit in support if bv_var_name(bit) == yname])
zbits = len([bit for bit in support if bv_var_name(bit) == zname])
xbins = 2**xbits
ybins = 2**ybits
zbins = 2**zbits
# Voxel corners
x, y, z = np.indices((xbins+1, ybins+1, zbins+1))
x = (x * (xgrid.ub - xgrid.lb)/xbins) + xgrid.lb
y = (y * (ygrid.ub - ygrid.lb)/ybins) + ygrid.lb
z = (z * (zgrid.ub - zgrid.lb)/zbins) + zgrid.lb
config = mgr.configure() # pick_iter alters config so save config state
# Construct bitmask
mask = np.full((xbins, ybins, zbins), False)
for pt in mgr.pick_iter(pred):
xvars = [k for k, v in pt.items() if bv_var_name(k) == xname]
yvars = [k for k, v in pt.items() if bv_var_name(k) == yname]
zvars = [k for k, v in pt.items() if bv_var_name(k) == zname]
xvars.sort() # Sort by bit names
yvars.sort()
zvars.sort()
xbv = [pt[bit] for bit in xvars]
ybv = [pt[bit] for bit in yvars]
zbv = [pt[bit] for bit in zvars]
x_idx = bv2int(xbv) if not dynamicperiodic(xgrid) else graytobin(bv2int(xbv))
y_idx = bv2int(ybv) if not dynamicperiodic(ygrid) else graytobin(bv2int(ybv))
z_idx = bv2int(zbv) if not dynamicperiodic(zgrid) else graytobin(bv2int(zbv))
mask[x_idx, y_idx, z_idx] = True
mgr.configure(reordering=config['reordering']) # Reinstate config state
fig = plt.figure()
ax = fig.gca(projection='3d')
ax.voxels(x, y, z, mask, facecolors=voxelcolors, edgecolor=edgecolors)
ax.set_xlabel(xname)
ax.set_ylabel(yname)
ax.set_zlabel(zname)
if view:
ax.view_init(view[0], view[1])
if title:
ax.set_title(title)
if fname:
extent = ax.get_window_extent().transformed(fig.dpi_scale_trans.inverted())
fig.savefig(str(fname)+'.png', dpi=400, bbox_inches=extent.expanded(1.1, 1.2))
else:
plt.show()
def bv2conc(self, bv: BitVector) -> int:
"""
Converts a bitvector into a concrete grid point
"""
return bv2int(bv)
def bv2conc(self, bv: BitVector) -> float:
"""Converts a bitvector into a concrete grid point."""
return self.pts[bv2int(bv)]