vispy/vispy

import numpy as np
import vispy.scene
from vispy.scene import visuals
from vispy.visuals import text

class CustomCanvas(vispy.scene.SceneCanvas):

    def __init__(self):
        super(CustomCanvas, self).__init__(keys='interactive', show=True)

        self.unfreeze()

        self.view = self.central_widget.add_view()

        self.xyz_top = None
        self.textLabelMax = None
        self.backgroundLabelMax = None
        self.bgScale = [1, 1, 1]
        self.events.mouse_move.connect(self.on_mouse_move)
        self.events.mouse_wheel.connect(self.on_mouse_wheel)

        self.freeze()

    def on_mouse_move(self, event):

        if(event.button == 1 and event.is_dragging):
            self.backgroundLabelMax.transform.reset()
            self.backgroundLabelMax.transform.matrix = self.view.camera.transform.matrix
            self.backgroundLabelMax.transform.translate(self.xyz_top)


    def on_mouse_wheel(self, event):
        pass
    """
        self.backgroundLabelMax.transform.reset()
        delta = event.delta[1]

        if(delta < 0):
            print(delta)
            if self.view.camera.scale_factor >= 30:
                self.view.camera.scale_factor = 30

        else:

            if self.view.camera.scale_factor <= 10:
                self.view.camera.scale_factor = 10

        self.backgroundLabelMax.transform.matrix = self.view.camera.transform.matrix
        self.backgroundLabelMax.transform.scale(self.bgScale)    
        self.backgroundLabelMax.transform.translate(self.xyz_top)
    """

    def draw_cloud_points(self):

        pos = np.random.normal(size=(100000, 3), scale=0.2)
        # one could stop here for the data generation, the rest is just to make the
        # data look more interesting. Copied over from magnify.py
        centers = np.random.normal(size=(50, 3))
        indexes = np.random.normal(size=100000, loc=centers.shape[0]/2.,
                                scale=centers.shape[0]/3.)
        indexes = np.clip(indexes, 0, centers.shape[0]-1).astype(int)
        scales = 10**(np.linspace(-2, 0.5, centers.shape[0]))[indexes][:, np.newaxis]
        pos *= scales
        pos += centers[indexes]

        # create scatter object and fill in the data
        scatter = visuals.Markers()
        scatter.set_data(pos, edge_color=None, face_color=(1, 1, 1, .5), size=5)
        self.view.add(scatter)

        # Adding turntable with ortographic view
        self.view.camera = vispy.scene.TurntableCamera(fov=0)

        # Adding plane as background for texts
        self.backgroundLabelMax = visuals.Rectangle(center = (0, 0, 0), width=3.5, height = 1, color=(1, 0, 0, 0.4), radius=0.3, parent=self.view.scene)
        # Adding text label at center / top (Z axis)

        self.xyz_top = (pos[:,0].max(), pos[:,1].max(), pos[:,2].max())
        self.textLabelMax = visuals.Text(pos=(0, 0, 0), text="Test_Label_Max", font_size=18, color="white", parent=self.backgroundLabelMax)

        self.backgroundLabelMax.transform = vispy.scene.transforms.STTransform(
            scale=(1, 1, 1),
            translate=(0, 0, 0)
        ).as_matrix()
        self.backgroundLabelMax.transform.matrix = self.view.camera.transform.matrix
        self.backgroundLabelMax.transform.translate(self.xyz_top)
        # Auto focus
        self.view.camera.set_range()

def main():
    custom_canvas = CustomCanvas()
    custom_canvas.draw_cloud_points()

if __name__ == '__main__':
    import sys
    if sys.flags.interactive != 1:
        main()
        vispy.app.run()