mirror of
https://github.com/jdegenstein/jmwright-CQ-Editor.git
synced 2025-12-20 22:47:06 +01:00
add rand_color() helper function
and seed(371353) global
This commit is contained in:
@@ -1,6 +1,6 @@
|
|||||||
from PyQt5.QtWidgets import QTreeWidget, QTreeWidgetItem, QAction, QMenu, QWidget, QAbstractItemView
|
from PyQt5.QtWidgets import QTreeWidget, QTreeWidgetItem, QAction, QMenu, QWidget, QAbstractItemView
|
||||||
from PyQt5.QtCore import Qt, pyqtSlot, pyqtSignal
|
from PyQt5.QtCore import Qt, pyqtSlot, pyqtSignal
|
||||||
|
from random import randrange as rrr, seed
|
||||||
from pyqtgraph.parametertree import Parameter, ParameterTree
|
from pyqtgraph.parametertree import Parameter, ParameterTree
|
||||||
|
|
||||||
from OCP.AIS import AIS_Line
|
from OCP.AIS import AIS_Line
|
||||||
@@ -13,6 +13,8 @@ from ..cq_utils import make_AIS, export, to_occ_color, is_obj_empty, get_occ_col
|
|||||||
from .viewer import DEFAULT_FACE_COLOR
|
from .viewer import DEFAULT_FACE_COLOR
|
||||||
from ..utils import splitter, layout, get_save_filename
|
from ..utils import splitter, layout, get_save_filename
|
||||||
|
|
||||||
|
seed(371353) #preserves colors run to run, needs to be run once globally
|
||||||
|
|
||||||
class TopTreeItem(QTreeWidgetItem):
|
class TopTreeItem(QTreeWidgetItem):
|
||||||
|
|
||||||
def __init__(self,*args,**kwargs):
|
def __init__(self,*args,**kwargs):
|
||||||
@@ -391,5 +393,22 @@ class ObjectTree(QWidget,ComponentMixin):
|
|||||||
else:
|
else:
|
||||||
item.properties['Visible'] = False
|
item.properties['Visible'] = False
|
||||||
|
|
||||||
|
def rand_color(alpha = 0., cfloat=False):
|
||||||
|
#helper function to generate a random color dict
|
||||||
|
#for CQ-editor's show_object function
|
||||||
|
lower = 10
|
||||||
|
upper = 100 #not too high to keep color brightness in check
|
||||||
|
if cfloat: #for two output types depending on need
|
||||||
|
return (
|
||||||
|
(rrr(lower,upper)/255),
|
||||||
|
(rrr(lower,upper)/255),
|
||||||
|
(rrr(lower,upper)/255),
|
||||||
|
alpha,
|
||||||
|
)
|
||||||
|
return {"alpha": alpha,
|
||||||
|
"color": (
|
||||||
|
rrr(lower,upper),
|
||||||
|
rrr(lower,upper),
|
||||||
|
rrr(lower,upper),
|
||||||
|
)}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user