def __init__(self):
super().__init__()
#store objects into array
self.i = 0
self.listOfObjects = []
with open('local2.csv', 'r') as file:
reader = csv.reader(file)
for row in reader:
self.listOfObjects.append(row)
print(self.listOfObjects)
# Camera
self.camera().lens().setPerspectiveProjection(45, 16 / 9, 0.1, 1000)
#setting focus on selected object
if self.listOfObjects[0][0] == 's':
self.camera().setPosition(
QVector3D(int(row[7]), int(row[8]), 50 + int(row[9])))
self.camera().setViewCenter(
QVector3D(int(row[7]), int(row[8]), int(row[9])))
else:
self.camera().setPosition(
QVector3D(int(row[9]), int(row[10]), 50 + int(row[11])))
self.camera().setViewCenter(
QVector3D(int(row[9]), int(row[10]), int(row[11])))
# For camera controls
self.rootEntity = Qt3DCore.QEntity()
self.showAll()
self.camController = Qt3DExtras.QOrbitCameraController(self.rootEntity)
self.camController.setLinearSpeed(50)
self.camController.setLookSpeed(180)
self.camController.setCamera(self.camera())
self.setRootEntity(self.rootEntity)
def __init__(self):
super(Window, self).__init__()
# Scene creation
# Kuesa content must be child of a Kuesa.SceneEntity.
# These entites are all themselves standard Qt3D::QEntity
self.rootEntity = Kuesa.SceneEntity()
# GLTF2Importer will load the glTF 2.0 content and add
# it as a child in the SceneEntity.
self.gltfImporter = Kuesa.GLTF2Importer(self.rootEntity)
self.gltfImporter.setSceneEntity(self.rootEntity)
self.gltfImporter.setSource(assetsUrl() + "/models/Box.glb")
# Since Kuesa is based on a PBR pipeline, we need an environment map.
self.rootEntity.addComponent(DefaultEnvMap(self.rootEntity))
# Camera.
self.camera().setPosition(QVector3D(10, 1.5, 10))
self.camera().setViewCenter(QVector3D(0, .5, 0))
self.camera().setUpVector(QVector3D(0, 1, 0))
self.camera().setAspectRatio(16. / 9.)
# Camera controls.
self.camController = Qt3DExtras.QOrbitCameraController(self.rootEntity)
self.camController.setCamera(self.camera())
# Frame graph.
self.fg = Kuesa.ForwardRenderer()
self.fg.setCamera(self.camera())
self.fg.setClearColor("white")
self.setActiveFrameGraph(self.fg)
self.setRootEntity(self.rootEntity)
def __init__(self):
super(Window, self).__init__()
# Default set-up
self.rootEntity = Kuesa.SceneEntity()
self.rootEntity.addComponent(DefaultEnvMap(self.rootEntity))
self.camera().setPosition(QVector3D(5, 1.5, 5))
self.camera().setViewCenter(QVector3D(0, 0.5, 0))
self.camera().setUpVector(QVector3D(0, 1, 0))
self.camera().setAspectRatio(16. / 9.)
self.camController = Qt3DExtras.QOrbitCameraController(self.rootEntity)
self.camController.setCamera(self.camera())
self.fg = Kuesa.ForwardRenderer()
self.fg.setCamera(self.camera())
self.fg.setClearColor("white")
self.setActiveFrameGraph(self.fg)
# Load a glTF model
self.gltfImporter = Kuesa.GLTF2Importer(self.rootEntity)
self.gltfImporter.setSceneEntity(self.rootEntity)
self.gltfImporter.setSource(assetsUrl() + "/models/duck/Duck.glb")
self.gltfImporter.statusChanged.connect(self.importerLoaded)
# Skybox creation
envmap_root = assetsUrl() + "/envmaps/pink_sunrise"
envmap_name = "pink_sunrise" + ("_16f" if platform.system() == "Darwin"
else "") + "_radiance"
self.skybox = Kuesa.Skybox(self.rootEntity)
self.skybox.setBaseName(envmap_root + "/" + envmap_name)
self.skybox.setExtension(".dds")
self.setRootEntity(self.rootEntity)
def __init__(self):
super(Window, self).__init__()
self.animationAspect = Qt3DAnimation.QAnimationAspect(self)
self.registerAspect(self.animationAspect)
# Camera
self.camera().setPosition(QVector3D(5.5, 1.5, 5.5))
self.camera().setViewCenter(QVector3D(0, .5, 0))
self.camera().setUpVector(QVector3D(0, 1, 0))
self.camera().setAspectRatio(4. / 3.)
# For camera controls
self.createScene()
self.camController = Qt3DExtras.QOrbitCameraController(self.rootEntity)
self.camController.setLinearSpeed(50)
self.camController.setLookSpeed(180)
self.camController.setCamera(self.camera())
self.fg = Kuesa.ForwardRenderer()
self.fg.setCamera(self.camera())
self.fg.setClearColor("white")
self.setActiveFrameGraph(self.fg)
self.setRootEntity(self.rootEntity)
def __init__(self):
super(Window, self).__init__()
# Since we are going to use animations, we need to register
# the Qt3D animation aspect :
self.animationAspect = Qt3DAnimation.QAnimationAspect(self)
self.registerAspect(self.animationAspect)
# Default set-up
self.rootEntity = Kuesa.SceneEntity()
self.rootEntity.addComponent(DefaultEnvMap(self.rootEntity))
self.camera().setPosition(QVector3D(25, 1.5, 25))
self.camera().setViewCenter(QVector3D(0, 3, 0))
self.camera().setUpVector(QVector3D(0, 1, 0))
self.camera().setAspectRatio(16. / 9.)
self.camController = Qt3DExtras.QOrbitCameraController(self.rootEntity)
self.camController.setCamera(self.camera())
self.fg = Kuesa.ForwardRenderer()
self.fg.setCamera(self.camera())
self.fg.setClearColor("white")
self.setActiveFrameGraph(self.fg)
# Load a glTF model
self.gltfImporter = Kuesa.GLTF2Importer(self.rootEntity)
self.gltfImporter.setSceneEntity(self.rootEntity)
self.gltfImporter.setSource(assetsUrl() +
"/models/InterpolationTest.glb")
self.gltfImporter.statusChanged.connect(self.on_sceneLoaded)
self.setRootEntity(self.rootEntity)
def __init__(self):
super(Window, self).__init__()
# Default set-up
self.rootEntity = Kuesa.SceneEntity()
self.rootEntity.addComponent(DefaultEnvMap(self.rootEntity))
self.camera().setPosition(QVector3D(5, 1.5, 5))
self.camera().setViewCenter(QVector3D(0, 0.5, 0))
self.camera().setUpVector(QVector3D(0, 1, 0))
self.camera().setAspectRatio(16. / 9.)
self.camController = Qt3DExtras.QOrbitCameraController(self.rootEntity)
self.camController.setCamera(self.camera())
self.fg = Kuesa.ForwardRenderer()
self.fg.setCamera(self.camera())
self.fg.setClearColor("white")
self.setActiveFrameGraph(self.fg)
# Load a glTF model
self.gltfImporter = Kuesa.GLTF2Importer(self.rootEntity)
self.gltfImporter.setSceneEntity(self.rootEntity)
self.gltfImporter.setSource(assetsUrl() + "/models/duck/Duck.glb")
self.gltfImporter.statusChanged.connect(self.importerLoaded)
# Skybox creation
envmap_root = assetsUrl() + "/envmaps/pink_sunrise"
envmap_name = "pink_sunrise" + ("_16f" if platform.system() == "Darwin"
else "") + "_radiance"
self.skybox = Kuesa.Skybox(self.rootEntity)
self.skybox.setBaseName(envmap_root + "/" + envmap_name)
self.skybox.setExtension(".dds")
# Creation of a few post-processing effects
self.blurFx = Kuesa.GaussianBlurEffect()
self.blurFx.setBlurPassCount(8)
self.dofFx = Kuesa.DepthOfFieldEffect()
self.dofFx.setFocusRange(3.1)
self.dofFx.setRadius(21.)
self.dofFx.setFocusDistance(6.6)
self.threshFx = Kuesa.ThresholdEffect()
self.threshFx.setThreshold(.1)
# self.fg.addPostProcessingEffect(self.blurFx)
# self.fg.addPostProcessingEffect(self.dofFx)
self.fg.addPostProcessingEffect(self.threshFx)
self.setRootEntity(self.rootEntity)
def __init__(self):
super(Window, self).__init__()
# Camera
self.camera().lens().setPerspectiveProjection(45, 16 / 9, 0.1, 1000)
self.camera().setPosition(QVector3D(0, 0, 40))
self.camera().setViewCenter(QVector3D(0, 0, 0))
# For camera controls
self.createScene()
self.camController = Qt3DExtras.QOrbitCameraController(self.rootEntity)
self.camController.setLinearSpeed(50)
self.camController.setLookSpeed(180)
self.camController.setCamera(self.camera())
self.setRootEntity(self.rootEntity)
def __init__(self):
super(Window, self).__init__()
# Default set-up
self.rootEntity = Kuesa.SceneEntity()
self.rootEntity.addComponent(DefaultEnvMap(self.rootEntity))
self.camera().setPosition(QVector3D(5, 1.5, 5))
self.camera().setViewCenter(QVector3D(0, 0.5, 0))
self.camera().setUpVector(QVector3D(0, 1, 0))
self.camera().setAspectRatio(16. / 9.)
self.camController = Qt3DExtras.QOrbitCameraController(self.rootEntity)
self.camController.setCamera(self.camera())
self.fg = Kuesa.ForwardRenderer()
self.fg.setCamera(self.camera())
self.fg.setClearColor("white")
self.setActiveFrameGraph(self.fg)
# Load a glTF model
self.gltfImporter = Kuesa.GLTF2Importer(self.rootEntity)
self.gltfImporter.setSceneEntity(self.rootEntity)
self.gltfImporter.setSource(assetsUrl() + "/models/duck/Duck.glb")
# Skybox creation
envmap_root = assetsUrl() + "/envmaps/pink_sunrise"
envmap_name = "pink_sunrise" + ("_16f" if platform.system() == "Darwin"
else "") + "_radiance"
self.skybox = Kuesa.Skybox(self.rootEntity)
self.skybox.setBaseName(envmap_root + "/" + envmap_name)
self.skybox.setExtension(".dds")
# Tonemapping and gamma correction
self.toneMapAlgorithm = Kuesa.ToneMappingAndGammaCorrectionEffect.ToneMapping(
)
self.fg.setExposure(2)
self.fg.setGamma(2.8)
self.fg.setToneMappingAlgorithm(self.toneMapAlgorithm)
# Cycle through the various tonemapping algorithms provided by Kuesa
timer = QTimer(self)
timer.setInterval(1000)
timer.timeout.connect(self.changeTonemapping)
timer.start()
self.setRootEntity(self.rootEntity)
def __init__(self, rootEntity):
super(SceneView, self).__init__()
self.rootEntity = rootEntity
self.defaultFrameGraph().setClearColor(QColor('#333'))
# Camera
self.camera().lens().setPerspectiveProjection(45, 16 / 9, 0.1, 1000)
self.camera().setPosition(QVector3D(0, 0, 40))
self.camera().setViewCenter(QVector3D(0, 0, 0))
# For camera controls
# self.createScene()
self.camController = Qt3DExtras.QOrbitCameraController(self.rootEntity)
self.camController.setLinearSpeed(60)
self.camController.setLookSpeed(180)
self.camController.setCamera(self.camera())
self.setRootEntity(self.rootEntity)
def __init__(self):
super(Window, self).__init__()
# Default set-up
self.rootEntity = Kuesa.SceneEntity()
self.rootEntity.addComponent(DefaultEnvMap(self.rootEntity))
self.camera().setPosition(QVector3D(25, 1.5, 25))
self.camera().setViewCenter(QVector3D(0, 3, 0))
self.camera().setUpVector(QVector3D(0, 1, 0))
self.camera().setAspectRatio(16. / 9.)
self.camController = Qt3DExtras.QOrbitCameraController(self.rootEntity)
self.camController.setCamera(self.camera())
self.fg = Kuesa.ForwardRenderer()
self.fg.setCamera(self.camera())
self.fg.setClearColor("white")
self.setActiveFrameGraph(self.fg)
# Load a glTF model
self.gltfImporter = Kuesa.GLTF2Importer(self.rootEntity)
self.gltfImporter.setSceneEntity(self.rootEntity)
self.gltfImporter.setSource(assetsUrl() +
"/models/InterpolationTest.glb")
# When opening this model in gltfEditor, we can see :
# - Multiple entities :
# Cube, Cube.001, Cube.002, Light, Plane...
# - Multiple meshes :
# Cube.001_0, Cube.002_0, ...
# - Multiple materials :
# Material, Material.001, ...
# etc.
# Let's use Kuesa to change one of these entities on-the-fly.
# We need to wait for the glTF file to be fully loaded in order to do that :
self.gltfImporter.statusChanged.connect(self.on_sceneLoaded)
self.setRootEntity(self.rootEntity)
def __init__(self):
super(Window, self).__init__()
# Default set-up - by now you know the drill :-)
self.rootEntity = Kuesa.SceneEntity()
self.rootEntity.addComponent(DefaultEnvMap(self.rootEntity))
self.gltfImporter = Kuesa.GLTF2Importer(self.rootEntity)
self.gltfImporter.setSceneEntity(self.rootEntity)
self.gltfImporter.setSource(assetsUrl() + "/models/duck/Duck.glb")
self.camera().setPosition(QVector3D(10, 1.5, 10))
self.camera().setViewCenter(QVector3D(0, .5, 0))
self.camera().setUpVector(QVector3D(0, 1, 0))
self.camera().setAspectRatio(16. / 9.)
self.camController = Qt3DExtras.QOrbitCameraController(self.rootEntity)
self.camController.setCamera(self.camera())
self.fg = Kuesa.ForwardRenderer()
self.fg.setCamera(self.camera())
self.fg.setClearColor("white")
self.setActiveFrameGraph(self.fg)
envmap_root = assetsUrl() + "/envmaps/pink_sunrise"
envmap_name = "pink_sunrise" + ("_16f" if platform.system() == "Darwin"
else "") + "_radiance"
self.skybox = Kuesa.Skybox(self.rootEntity)
self.skybox.setBaseName(envmap_root + "/" + envmap_name)
self.skybox.setExtension(".dds")
# Now we create some lights.
# All lights have an intensity and a color.
# Specific light types have relevant additional properties.
# Point light
self.pointLightEntity = Qt3DCore.QEntity(self.rootEntity)
self.pointLightTransform = Qt3DCore.QTransform(self.pointLightEntity)
self.pointLightTransform.setTranslation(QVector3D(20, -10, -10))
self.pointLight = Kuesa.PointLight(self.pointLightEntity)
self.pointLight.setIntensity(1000.0)
self.pointLight.setColor("red")
self.pointLightEntity.addComponent(self.pointLightTransform)
self.pointLightEntity.addComponent(self.pointLight)
# Spot light
self.spotLightEntity = Qt3DCore.QEntity(self.rootEntity)
self.spotLightTransform = Qt3DCore.QTransform(self.spotLightEntity)
self.spotLightTransform.setTranslation(QVector3D(-10, 10, 10))
self.spotLight = Kuesa.SpotLight(self.spotLightEntity)
self.spotLight.setIntensity(1000.0)
self.spotLight.setColor("green")
self.spotLight.setInnerConeAngle(50)
self.spotLight.setOuterConeAngle(100)
self.spotLight.setRange(1000)
self.spotLightEntity.addComponent(self.spotLightTransform)
self.spotLightEntity.addComponent(self.spotLight)
# Directional light
self.directionalLightEntity = Qt3DCore.QEntity(self.rootEntity)
self.directionalLightTransform = Qt3DCore.QTransform(
self.directionalLightEntity)
self.directionalLight = Kuesa.DirectionalLight(
self.directionalLightEntity)
self.directionalLight.setIntensity(100.0)
self.directionalLight.setColor("blue")
self.directionalLight.setDirection(QVector3D(10, 10, -10))
self.directionalLightEntity.addComponent(
self.directionalLightTransform)
self.directionalLightEntity.addComponent(self.directionalLight)
self.setRootEntity(self.rootEntity)
def __init__(self, parent, fs_subject="sample"):
super(MeshDisplayWidget, self).__init__(parent)
self.config = parent.config
self.fs_subject = fs_subject
self.view = Qt3DExtras.Qt3DWindow()
self.view.defaultFrameGraph().setClearColor(QtGui.QColor("#4d4d4f"))
self.container = QtWidgets.QWidget.createWindowContainer(self.view, self)
screen_size = self.view.screen().size()
self.container.setMinimumSize(QtCore.QSize(600, 100))
self.container.setMaximumSize(screen_size)
#h_layout = QtWidgets.QHBoxLayout(self)
#h_layout.addWidget(self.container)
self.addWidget(self.container)
self.tableView = QtWidgets.QTableView(self)
self.tableView.setObjectName("tableView")
v_layout = QtWidgets.QVBoxLayout()
v_layout.addWidget(self.tableView)
self.overlay_btn = QtWidgets.QPushButton("Update and overlay digitized points")
self.overlay_btn.clicked.connect(self.update_acquisition_overlay)
v_layout.addWidget(self.overlay_btn)
subject_name_layout = QtWidgets.QHBoxLayout()
subject_name_layout.addWidget(QtWidgets.QLabel("Subject name:"))
self.subject_name_wgt = QtWidgets.QComboBox()
subject_name_layout.addWidget(self.subject_name_wgt)
self.new_subject_wgt = QtWidgets.QPushButton("New subject")
self.new_subject_wgt.clicked.connect(self.new_subject)
subject_name_layout.addWidget(self.new_subject_wgt)
self.mute_sound_wgt = QtWidgets.QCheckBox("Mute sound")
subject_name_layout.addWidget(self.mute_sound_wgt)
v_layout.addLayout(subject_name_layout)
if "subject_dir" not in self.config["DEFAULT"]:
self.config["DEFAULT"]["subject_dir"] = str(Path(__file__).parent.absolute() / "subjects_data")
subject_dir_layout = QtWidgets.QHBoxLayout()
subject_dir_layout.addWidget(QtWidgets.QLabel("Subject dir:"))
self.subject_dir_wgt = QtWidgets.QLineEdit(self.config["DEFAULT"]["subject_dir"])
self.subject_dir_wgt.setReadOnly(False)
subject_dir_layout.addWidget(self.subject_dir_wgt)
select_subject_dir_btn = QtWidgets.QPushButton("...")
subject_dir_layout.addWidget(select_subject_dir_btn)
select_subject_dir_btn.clicked.connect(self.open_subject_dir_dlg)
v_layout.addLayout(subject_dir_layout)
self.subject_dir_wgt.textChanged.connect(self.set_subject_dir)
if not Path(self.config["DEFAULT"]["subject_dir"]).exists():
msgBox = QtWidgets.QMessageBox()
msgBox.setIcon(QtWidgets.QMessageBox.Information)
msgBox.setText(f"The chosen subject directory (i.e. {self.config['DEFAULT']['subject_dir']}) does not "
"exist. Do you want to create it or to select a new directory?")
msgBox.setWindowTitle("Non-existent subject directory")
button_create = msgBox.addButton("Create it!", QtWidgets.QMessageBox.YesRole);
msgBox.addButton("Select a different directory", QtWidgets.QMessageBox.NoRole)
msgBox.exec_()
if msgBox.clickedButton() == button_create:
Path(self.config["DEFAULT"]["subject_dir"]).mkdir(exist_ok=True, parents=True)
else:
self.open_subject_dir_dlg()
#h_layout.addLayout(v_layout)
v_layout_widget = QtWidgets.QWidget(self)
v_layout_widget.setLayout(v_layout)
self.addWidget(v_layout_widget)
self.data = QtCore.QUrl.fromLocalFile(str(Path(__file__).parent.absolute() / f"{self.fs_subject}_head.obj"))
self.root_entity = Qt3DCore.QEntity()
self.material = Qt3DExtras.QPhongMaterial()
self.material.setDiffuse(QtGui.QColor(254, 254, 254))
self.camera = self.view.camera()
self.camera.lens().setPerspectiveProjection(400.0, 16.0/9.0, 100, 1200.0)
self.camera.setPosition(QtGui.QVector3D(0, 500, 0))
self.camera.setUpVector(QtGui.QVector3D(0, 0, 1))
self.lightEntities = {}
self.lights = {}
self.lightTransforms = {}
for trans_x in [200, -200]:
for trans_y in [200, -200]:
self.lightEntities[(trans_x, trans_y)] = Qt3DCore.QEntity(self.root_entity)
self.lights[(trans_x, trans_y)] = Qt3DRender.QPointLight(self.lightEntities[(trans_x, trans_y)])
self.lights[(trans_x, trans_y)].setColor("white")
self.lights[(trans_x, trans_y)].setIntensity(0.4)
self.lightEntities[(trans_x, trans_y)].addComponent(self.lights[(trans_x, trans_y)])
self.lightTransforms[(trans_x, trans_y)] = Qt3DCore.QTransform(self.lightEntities[(trans_x, trans_y)])
self.lightTransforms[(trans_x, trans_y)].setTranslation(QtGui.QVector3D(trans_x, trans_y, 100.0))
self.lightEntities[(trans_x, trans_y)].addComponent(self.lightTransforms[(trans_x, trans_y)])
# Get co-registered head, fiducials and electrodes
data_path = Path(mne.datasets.sample.data_path())
fs_subjects_dir = data_path / 'subjects'
trans_fname = data_path / 'MEG' / 'sample' / 'sample_audvis_raw-trans.fif'
trans = mne.read_trans(trans_fname)
montage = mne.channels.make_standard_montage('GSN-HydroCel-129')
info = mne.create_info(montage.ch_names, 100, ch_types='eeg')
raw = mne.io.RawArray(np.zeros((len(montage.ch_names), 100)), info)
raw.set_montage(montage)
eeg, fid, surf = get_aligned_artifacts(raw.info, subject=self.fs_subject, trans=trans,
subjects_dir=fs_subjects_dir, coord_frame='mri')
self.head_surf = surf
path_out = Path(str(Path(__file__).parent.absolute() / f"{self.fs_subject}_head.obj"))
if True: #not path_out.exists():
mesh = trimesh.Trimesh(surf["rr"] * 1000, surf["tris"])
open3d_mesh = open3d.geometry.TriangleMesh(vertices=open3d.utility.Vector3dVector(mesh.vertices),
triangles=open3d.utility.Vector3iVector(mesh.faces))
mesh = open3d_mesh.simplify_quadric_decimation(int(20000))
mesh = trimesh.Trimesh(np.asarray(mesh.vertices), np.asarray(mesh.triangles))
with path_out.open('w') as file_obj:
file_obj.write(trimesh.exchange.obj.export_obj(mesh))
# Head
self.data = QtCore.QUrl.fromLocalFile(str(path_out))
self.head_entity = Qt3DCore.QEntity(self.root_entity)
self.head_mesh = Qt3DRender.QMesh()
self.head_mesh.setMeshName(str(path_out))
self.head_mesh.setSource(self.data)
self.head_entity.addComponent(self.head_mesh)
self.head_entity.addComponent(self.material)
self.camController = Qt3DExtras.QOrbitCameraController(self.head_mesh)
self.camController.setCamera(self.camera)
self.camController.setLinearSpeed(self.camController.linearSpeed()*100)
self.mesh_center_of_mass = np.median(surf['rr'], 0)*1000
self.camera.setViewCenter(QtGui.QVector3D(*(self.mesh_center_of_mass)))
self.camera.viewCenterChanged.connect(self.cam_view_center_changed)
## EEG electrodes and Fiducials
self.electrode_material = Qt3DExtras.QPhongMaterial(diffuse=QtGui.QColor("#665423"))
self.fid_material = Qt3DExtras.QPhongMaterial(diffuse=QtGui.QColor("#0000FF"))
self.selected_material = Qt3DExtras.QPhongMaterial(diffuse=QtGui.QColor("#FF0000"))
self.edited_material = Qt3DExtras.QPhongMaterial(diffuse=QtGui.QColor("#FFFF00"))
self.acq_material = Qt3DExtras.QPhongMaterial(diffuse=QtGui.QColor("#00FF00"))
self.selected_item = None
self.coordinate_meshes = {}
self.coordinate_entities = {}
self.coordinate_transforms = {}
self.coordinate_materials = {}
for kind, var in zip(["fid", "elec"], [fid, eeg]):
for label, pos in var.iterrows():
pos = pos.values*1000
self.coordinate_meshes[label] = Qt3DExtras.QSphereMesh(rings=20, slices=20, radius=3)
self.coordinate_entities[label] = Qt3DCore.QEntity(self.root_entity)
self.coordinate_transforms[label] = Qt3DCore.QTransform(self.coordinate_meshes[label])
self.coordinate_transforms[label].setTranslation(QtGui.QVector3D(*pos))
self.coordinate_entities[label].addComponent(self.coordinate_meshes[label])
self.coordinate_entities[label].addComponent(self.coordinate_transforms[label])
self.acq_coordinate_meshes = {}
self.acq_coordinate_entities = {}
self.acq_coordinate_transforms = {}
self.df_montage = pd.concat([fid, eeg])
self.df_montage.columns = ["x", "y", "z"]
self.df_acq = pd.concat([self.df_montage.copy().rename(columns={"x": f"x{i+1}", "y": f"y{i+1}", "z": f"z{i+1}"})
for i in range(4)], axis=1)
model = DataFrameModel(self.df_acq)
self.tableView.setModel(model)
self.df_acq.loc[:, :] = np.nan
self.set_table_view_selection(0)
self.view.setRootEntity(self.root_entity)
self.subject_name_wgt.currentTextChanged.connect(self.load_subject)
self.update_subject_lst()
def __init__(self, parent=None):
super().__init__(parent)
self.rnd_lat = random.randrange(5200,5300) / 100.0
self.rnd_hdg = random.randrange(-300,300) / 100.0
self.rnd_dec = random.randrange(-50,50) / 10.0
print(self.rnd_lat, self.rnd_hdg, self.rnd_dec)
self.view = Qt3DExtras.Qt3DWindow()
self.view.defaultFrameGraph().setClearColor(QtGui.QColor("#4d4d4f"))
self.container = QtWidgets.QWidget.createWindowContainer(self.view)
self.vLayout = QtWidgets.QVBoxLayout(self)
self.vLayout.addWidget(self.container, 1)
self.input_aspect = Qt3DInput.QInputAspect()
self.view.registerAspect(self.input_aspect)
self.rootEntity = Qt3DCore.QEntity()
self.view.setRootEntity(self.rootEntity)
###
cameraEntity = self.view.camera()
cameraEntity.lens().setPerspectiveProjection(5.0, 16.0 / 9.0, 10, 100000.0)
cameraEntity.setPosition(QtGui.QVector3D(0, 1000, 50000))
cameraEntity.setUpVector(QtGui.QVector3D(0, 1, 0))
cameraEntity.setViewCenter(QtGui.QVector3D(0, 1000, 0))
self.camController = Qt3DExtras.QOrbitCameraController(self.rootEntity)
self.camController.setCamera(cameraEntity)
###
self.lightEntity = Qt3DCore.QEntity(self.rootEntity)
self.light = Qt3DRender.QPointLight(self.lightEntity)
self.light.setColor("white")
self.light.setIntensity(1)
self.lightTransform = Qt3DCore.QTransform(self.lightEntity)
self.lightTransform.setTranslation(cameraEntity.position())
self.lightEntity.addComponent(self.light)
self.lightEntity.addComponent(self.lightTransform)
### sky
# self.skyMesh = Qt3DExtras.QSphereMesh(radius=5000, rings=94, slices=48)
# self.skyMaterial = Qt3DExtras.QPhongMaterial(diffuse=QtGui.QColor("#ff0000"))
# self.skyEntity = Qt3DCore.QEntity(self.rootEntity)
# self.skyEntity.addComponent(self.skyMesh)
# self.skyEntity.addComponent(self.skyMaterial)
### hdg, x right, y up
# Cylinder cog = (0,0,0) long direction = y
self.hdgMesh = Qt3DExtras.QCylinderMesh(length=1200.0, radius=100.0, rings=100, slices=20)
rotation = QtGui.QQuaternion.fromAxisAndAngle(QtGui.QVector3D(0.0, 0.0, 1.0), 15.0)
translation = QtGui.QVector3D(0.0, 600.0, 0.0)
self.hdgTransform = Qt3DCore.QTransform(translation=translation, rotation=rotation)
self.hdgMaterial = Qt3DExtras.QPhongMaterial(diffuse=QtGui.QColor("#ff0000"))
self.hdgEntity = Qt3DCore.QEntity(self.rootEntity)
self.hdgEntity.addComponent(self.hdgMesh)
self.hdgEntity.addComponent(self.hdgMaterial)
self.hdgEntity.addComponent(self.hdgTransform)
### lat x up, y left
self.latMesh = Qt3DExtras.QCylinderMesh(length=200.0, radius=100.0, rings=100, slices=20)
rotation = QtGui.QQuaternion.fromAxisAndAngle(QtGui.QVector3D(0.0, 0.0, 1.0), 90.0)
translation = QtGui.QVector3D(0.0, 700.0, 0.0)
self.latTransform = Qt3DCore.QTransform(rotation=rotation, translation=translation)
self.latMaterial = Qt3DExtras.QPhongMaterial(diffuse=QtGui.QColor("#00ff00"))
self.latEntity = Qt3DCore.QEntity(self.hdgEntity)
self.latEntity.addComponent(self.latMesh)
self.latEntity.addComponent(self.latMaterial)
self.latEntity.addComponent(self.latTransform)
### ra x up, y left
self.raMesh = Qt3DExtras.QCylinderMesh(length=500.0, radius=100.0, rings=100, slices=20)
rotation1 = QtGui.QQuaternion.fromAxisAndAngle(QtGui.QVector3D(0.0, 0.0, -1.0), 90)
rotation2 = QtGui.QQuaternion.fromAxisAndAngle(QtGui.QVector3D(-1.0, 0.0, 0.0), 90)
rotation = rotation1 * rotation2
translation = QtGui.QVector3D(200.0, 0.0, 0.0)
self.raTransform = Qt3DCore.QTransform(rotation=rotation, translation=translation)
self.raMaterial = Qt3DExtras.QPhongMaterial(diffuse=QtGui.QColor("#beb32b"))
self.raEntity = Qt3DCore.QEntity(self.latEntity)
self.raEntity.addComponent(self.raMesh)
self.raEntity.addComponent(self.raMaterial)
self.raEntity.addComponent(self.raTransform)
### dec
self.decMesh = Qt3DExtras.QCylinderMesh(length=500.0, radius=100.0, rings=100, slices=20)
rotation = QtGui.QQuaternion.fromAxisAndAngle(QtGui.QVector3D(1.0, 0.0, 0.0), 90.0)
translation = QtGui.QVector3D(0.0, 350.0, 0.0)
self.decTransform = Qt3DCore.QTransform(rotation=rotation, translation=translation)
self.decMaterial = Qt3DExtras.QPhongMaterial(diffuse=QtGui.QColor("#0000ff"))
self.decEntity = Qt3DCore.QEntity(self.raEntity)
self.decEntity.addComponent(self.decMesh)
self.decEntity.addComponent(self.decMaterial)
self.decEntity.addComponent(self.decTransform)
### tube
self.tubeMesh = Qt3DExtras.QCylinderMesh(length=1200.0, radius=100.0, rings=100, slices=20)
rotation = QtGui.QQuaternion.fromAxisAndAngle(QtGui.QVector3D(-1.0, 0.0, 0.0), 90.0)
translation = QtGui.QVector3D(0.0, 350.0, 100.0)
self.tubeTransform = Qt3DCore.QTransform(rotation=rotation, translation=translation)
self.tubeMaterial = Qt3DExtras.QPhongMaterial(diffuse=QtGui.QColor("#ffffff"))
self.tubeEntity = Qt3DCore.QEntity(self.decEntity)
self.tubeEntity.addComponent(self.tubeMesh)
self.tubeEntity.addComponent(self.tubeMaterial)
self.tubeEntity.addComponent(self.tubeTransform)
self.ocuMesh = Qt3DExtras.QCylinderMesh(length=100.0, radius=50.0, rings=100, slices=20)
translation = QtGui.QVector3D(0.0, -650.0, 0.0)
self.ocuTransform = Qt3DCore.QTransform(translation=translation)
self.ocuMaterial = Qt3DExtras.QPhongMaterial(diffuse=QtGui.QColor("#ffffff"))
self.ocuEntity = Qt3DCore.QEntity(self.tubeEntity)
self.ocuEntity.addComponent(self.ocuMesh)
self.ocuEntity.addComponent(self.ocuMaterial)
self.ocuEntity.addComponent(self.ocuTransform)
self.fakeMesh = Qt3DExtras.QCuboidMesh(xExtent=500, yExtent=500, zExtent=500)
translation = QtGui.QVector3D(-1000.0, 0, 0.0)
self.fakeTransform = Qt3DCore.QTransform(translation=translation)
self.fakeMaterial = Qt3DExtras.QPhongMaterial(diffuse=QtGui.QColor("#ffffff"))
self.fakeEntity = Qt3DCore.QEntity(self.rootEntity)
self.fakeEntity.addComponent(self.fakeMesh)
self.fakeEntity.addComponent(self.fakeMaterial)
self.fakeEntity.addComponent(self.fakeTransform)
self._heading = 0
self._latitude = 45
self._ra = 0
self._dec = 90