def __init__(self) -> None: super().__init__() self.setMinimumSize(640, 480) self._build_menus() self.view = Qt3DExtras.Qt3DWindow() self.view.defaultFrameGraph().setClearColor(QColor(0, 0, 0)) self.widget = QWidget.createWindowContainer(self.view) self.sphere_vc = SphereVC(self.view) self.setCentralWidget(self.widget) self._textures = AlbedoTextureMapper()
def __init__(self, args, parent=None): super().__init__(parent) ############################################################################## # UI初期化処理 self.ui = MainWindowUi.loader(os.path.join(os.getcwd(), 'ui/main.ui')) self.setWindowTitle('self.viewer test') self.setCentralWidget(self.ui.widgets) self.view = Qt3DExtras.Qt3DWindow() self.view.defaultFrameGraph().setClearColor(QtGui.QColor(0, 0, 0)) self.container = QWidget.createWindowContainer(self.view) screen_size = self.view.screen().size() self.ui.viewer.addWidget(self.container) self.container.setMinimumSize(QtCore.QSize(400, 600)) self.container.setMaximumSize(screen_size) # QSize screenSize = self.view -> screen() -> size(); # container -> setMinimumSize(QSize(200, 100)); # container -> setMaximumSize(screenSize); # vLayout -> setAlignment(Qt: : AlignTop); # hLayout -> addWidget(container, 1); # hLayout -> addLayout(vLayout); # input_aspect = Qt3DInput.QInputAspect() # self.view.registerAspect(input_aspect) # root entity self.root_entity: Qt3DCore.QEntity = Qt3DCore.QEntity() # draw grid and axis """ self.x_axis: Qt3DRender.QGeometry = Qt3DRender.QGeometry(self.root_entity) x_axis_pos: QtCore.QByteArray = QtCore.QByteArray() x_axis_pos.append(0) x_axis_pos.append(0) x_axis_pos.append(0) x_axis_pos.append(10) x_axis_pos.append(0) x_axis_pos.append(0) x_axis_buf: Qt3DRender.QBuffer = Qt3DRender.QBuffer(self.x_axis) x_axis_buf.setData(x_axis_pos) x_axis_attr: Qt3DRender.QAttribute = Qt3DRender.QAttribute(self.x_axis) x_axis_attr.setVertexBaseType(Qt3DRender.QAttribute.Float) x_axis_attr.setVertexSize(3) x_axis_attr.setAttributeType(Qt3DRender.QAttribute.VertexAttribute) x_axis_attr.setBuffer(x_axis_buf) x_axis_attr.setByteStride(3) x_axis_attr.setCount(2) self.x_axis.addAttribute(x_axis_attr) """ test_mtl = Qt3DExtras.QTextureMaterial(self.root_entity) self.test = Qt3DCore.QEntity(self.root_entity) self.test_mesh: Qt3DExtras.QTorusMesh = Qt3DExtras.QTorusMesh() self.test_mesh.setRadius(5) self.test_mesh.setMinorRadius(1) self.test_mesh.setRings(100) self.test_mesh.setSlices(20) self.test_tr = Qt3DCore.QTransform() self.test_tr.setTranslation(QtGui.QVector3D(0, 0, 0)) # test_tr.setScale3D() self.test.addComponent(self.test_mesh) self.test.addComponent(self.test_tr) self.test.addComponent(self.test_mtl) # camera entity camera_entity: Qt3DRender.QCamera = self.view.camera() camera_entity.lens().setPerspectiveProjection(45.0, 16.0 / 9.0, 0.1, 1000.0) camera_entity.setPosition(QtGui.QVector3D(0, 0, 20.0)) camera_entity.setUpVector(QtGui.QVector3D(0, 1, 0)) camera_entity.setViewCenter(QtGui.QVector3D(0, 0, 0)) light_entity = Qt3DCore.QEntity(self.root_entity) light = Qt3DRender.QPointLight(light_entity) light.setColor("white") light.setIntensity(1) light_entity.addComponent(light) light_transform = Qt3DCore.QTransform(light_entity) light_transform.setTranslation(camera_entity.position()) light_entity.addComponent(light_transform) # for camera controls cam_controller = Qt3DExtras.QFirstPersonCameraController(self.root_entity) cam_controller.setCamera(camera_entity) # set root object of the scene self.view.setRootEntity(self.root_entity)
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
def __init__(self) -> None: self.view = Qt3DExtras.Qt3DWindow() self._configure_view() self.widget = QWidget.createWindowContainer(self.view) self.sphere_mgr = SphereVC(self.view) self.albedo_mapper = AlbedoTextureMapper()