add ini repo

    for col in range(count):

        y_data = ws.readY(col)

        array[:,col] = y_data

    array.tofile("heatmap_gpsans.np", sep='', format='%s')

    return array

from .main import main

__all__ = [

    "main",

]

import os

import logging

from trame.app import get_server

from trame.decorators import TrameApp, change

from trame.ui.vuetify import SinglePageWithDrawerLayout

from trame.widgets import vuetify, vtk

from .ui import (

    DeviceSelector,

    Directory,

    FileSelector,

    FigureControl,

    Scattering,

    Transmission,

)

from .instrument.d11_plus import D11_Plus

from .visualization import Visualization

from .utilities import file_search as fs

logger = logging.getLogger(__name__)

logger.setLevel(logging.INFO)

PANELS = [

    ("device", "mdi-microscope"),

    ("directory", "mdi-folder-eye"),

    ("file", "mdi-application"),

    ("viz", "mdi-image-multiple"),

    ("scattering", "mdi-scatter-plot"),

    ("transmission", "mdi-transfer"),

]

# ---------------------------------------------------------

# Engine class

# ---------------------------------------------------------

@TrameApp()

class AirSans:

    def __init__(self, server=None, data=None):

        self.server = get_server(server, client_type="vue2")

        self._viz = Visualization(self.server)

        self._selected_device = None

        self._active_directory = None

        # CLI

        self.server.cli.add_argument(

            "--data",

            help="Path to browse",

            dest="data",

            default=".",

        )

        args, _ = self.server.cli.parse_known_args()

        if data is None:

            data = args.data

        # search directory contents

        self.state.dirs = fs.get_directory_structure(data)

        self.state.files = []

        self.state.file = None

        # Initialize data

        self.state.data = None

        self.state.center_data = None

        self.state.pixel_ratio = 2.0

        self.ui = self._build_ui()

    @property

    def state(self):

        return self.server.state

    @property

    def ctrl(self):

        return self.server.controller

    def _build_ui(self):

        self.state.trame__title = "air-sans"

        with SinglePageWithDrawerLayout(self.server) as layout:

            with layout.icon:

                vuetify.VIcon("mdi-bullseye")

            with layout.title as title:

                title.set_text("AIR-SANS")

                title.classes = "font-weight-medium ml-0 pl-0"

            with layout.toolbar as tb:

                tb.dense = True

                with vuetify.VBtnToggle(

                    v_model=("panel_visible", ["device", "directory", "file", "viz"]),

                    multiple=True,

                    dense=True,

                    classes="ml-6",

                ):

                    for key, icon in PANELS:

                        with vuetify.VBtn(value=key, small=True):

                            vuetify.VIcon(icon, small=True)

                vuetify.VSpacer()

                with vuetify.VBtn(icon=True, click=self._viz.reset_camera):

                    vuetify.VIcon("mdi-crop-free")

            with layout.drawer as drawer:

                drawer.width = 350

                DeviceSelector()

                Directory(select_directory_fn=self.select_directory)

                FileSelector(self.selected_file)

                FigureControl()

                Scattering()

                Transmission()

            with layout.content:

                with vuetify.VContainer(

                    fluid=True,

                    classes="pa-0 fill-height",

                ):

                    with vuetify.VRow(classes="fill-height pa-0"):

                        with vuetify.VCol(cols=3, classes="pa-0"):

                            view = vtk.VtkRemoteView(

                                self._viz.left_render_window, interactive_ratio=1

                            )

                            self._viz.set_left_view(view)

                        with vuetify.VCol(cols=6, classes="pa-0"):

                            view = vtk.VtkRemoteView(

                                self._viz.center_render_window, interactive_ratio=1

                            )

                            self._viz.set_center_view(view)

                        with vuetify.VCol(cols=3, classes="pa-0"):

                            view = vtk.VtkRemoteView(

                                self._viz.right_render_window, interactive_ratio=1

                            )

                            self._viz.set_right_view(view)

                    # self.ctrl.update_d11 = plotly.Figure(

                    #     display_mode_bar=("false",),

                    #     v_show=("figure_ready", False),

                    # ).update

            return layout

    @change("selectedDevice")

    def on_device_change(self, selectedDevice, **kwargs):

        if selectedDevice == "D11+":

            self._selected_device = D11_Plus()

    def select_directory(self, active_nodes):

        self.server.state.directory_label = None

        self._active_directory = None

        if len(active_nodes):

            node_id = active_nodes[0]

            if isinstance(node_id, str):

                self.server.state.directory_label = os.path.basename(node_id)

                self._active_directory = node_id

                self.server.state.files = fs.get_file_list(node_id)

    def selected_file(self, file):

        state = self.server.state

        state.file = file

        self._selected_device.read_file(self._active_directory, state.file)

        pixel_x = self._selected_device.pixel1_x

        pixel_y = self._selected_device.pixel1_y

        state.pixel_ratio = pixel_y / pixel_x

        state.center_ny = self._selected_device.ny1

        state.center_nx = self._selected_device.nx1

        self._viz.set_center_data(

            self._selected_device.det1_data,

            self._selected_device.pixel1_x,

            self._selected_device.pixel1_y,

        )

        state.left_ny = self._selected_device.ny2

        state.left_nx = self._selected_device.nx2

        self._viz.set_left_data(

            self._selected_device.det2_data,

            self._selected_device.pixel2_x,

            self._selected_device.pixel2_y,

        )

        state.right_ny = self._selected_device.ny3

        state.right_nx = self._selected_device.nx3

        self._viz.set_right_data(

            self._selected_device.det3_data,

            self._selected_device.pixel3_x,

            self._selected_device.pixel3_y,

        )

        self._viz.create_d11_fig()

        state.figure_ready = True

    @change("device_active_data")

    def show_imask(self, device_active_data, **kwargs):

        self.server.state.figure_ready = False

        if device_active_data == "mask":

            self._viz.set_center_data(self._selected_device.detector1_imask_data)

            self._viz.set_left_data(self._selected_device.detector2_imask_data)

            self._viz.set_right_data(self._selected_device.detector3_imask_data)

            self.server.state.figure_ready = True

            self._viz.create_d11_fig()

        if device_active_data == "efficiency":

            self._viz.set_center_data(self._selected_device.detector1_efficiency_data)

            self._viz.set_left_data(self._selected_device.detector2_efficiency_data)

            self._viz.set_right_data(self._selected_device.detector3_efficiency_data)

            self.server.state.figure_ready = True

            self._viz.create_d11_fig()

        if device_active_data == "error":

            self._viz.set_center_data(self._selected_device.detector1_efficiency_error)

            self._viz.set_left_data(self._selected_device.detector2_efficiency_error)

            self._viz.set_right_data(self._selected_device.detector3_efficiency_error)

            self.server.state.figure_ready = True

            self._viz.create_d11_fig()

        if device_active_data == "" and self.server.state.file:

            self.selected_file(self.server.state.file)