very_large_n_dims.py

# @license
# Copyright 2020 Google Inc.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#      http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import neuroglancer
import numpy as np

from neuroglancer_utils.layer_utils import add_render_panel
from neuroglancer_utils.viewer_utils import (
    launch_nglancer,
    open_browser,
    remove_axis_lines,
    set_gpu_memory,
    threedee_view,
    update_projection,
    update_title,
)


def add_image_layer(state, **kwargs):
    dimensions = neuroglancer.CoordinateSpace(
        names=["x", "y", "z", "qudfa", "energy", "distance", "time", "mass"],
        units=("nm", "um", "", "s", "Hz", "m", "kHz", "us"),
        scales=(4, 8, 2.4, 0.5, 1000, 10.5, 19.2, 0.0002),
    )
    generator = np.random.default_rng(0)
    data = generator.random((20, 55, 10, 100, 10, 5, 2, 10))
    local_volume = neuroglancer.LocalVolume(data, dimensions)
    state.layers["image"] = neuroglancer.ImageLayer(
        source=local_volume,
        volume_rendering_mode="max",
        tool_bindings={
            "A": neuroglancer.VolumeRenderingGainTool(),
        },
        panels=[add_render_panel(side="right")],
        **kwargs,
    )


def get_shader():
    return """#uicontrol invlerp normalized(range=[0,255], clamp=true)
void main() {
    float val = normalized();
    emitGrayscale(val);
}
"""


if __name__ == "__main__":
    viewer = launch_nglancer()
    with viewer.txn() as s:
        add_image_layer(s, shader=get_shader())
    threedee_view(viewer)
    remove_axis_lines(viewer)
    update_title(viewer, "Large example")
    set_gpu_memory(viewer, gpu_memory=2)
    update_projection(viewer, orientation=[0.25, 0.6, 0.65, 0.3])
    open_browser(viewer, hang=True)