geosPythonPackages/geos-trame/src/geos/trame/app/ui/plotting.py at e5dd9728ace12c7918a5b14d653b38088b514b37 · GEOS-DEV/geosPythonPackages · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
# SPDX-License-Identifier: Apache-2.0
# SPDX-FileCopyrightText: Copyright 2023-2024 TotalEnergies.
# SPDX-FileContributor: Lionel Untereiner, Jacques Franc
from typing import Any

import matplotlib.pyplot as plt
import numpy as np
from matplotlib.figure import Figure
from trame.widgets import matplotlib
from trame.widgets import vuetify3 as vuetify

from geos.trame.app.deck.tree import DeckTree


class DeckPlotting( vuetify.VCard ):

    def __init__( self, source: DeckTree, **kwargs: Any ) -> None:
        """Constructor."""
        super().__init__( **kwargs )

        self._source = source
        if source.input_file is None:
            return

        self._filepath = ( source.input_file.path, )

        self.ctrl.permeability = self._permeability
        self.ctrl.figure_size = self._figure_size

        with self:
            vuetify.VCardTitle( "2D View" )
            html_viewX = matplotlib.Figure( figure=self.ctrl.permeability( **self.ctrl.figure_size() ),
                                            # style="position: absolute;left: 20%;top: 30%;",
                                           )
            self.ctrl.update_figure = html_viewX.update

    @property
    def source( self ) -> DeckTree:
        """Getter for source."""
        return self._source

    def _update_view( self ) -> None:
        self.ctrl.view_update( figure=self.ctrl.permeability( **self.ctrl.figure_size() ) )

    def _figure_size( self ) -> dict:

        if self.state.figure_size is None:
            return {}

        dpi = self.state.figure_size.get( "dpi" )
        rect = self.state.figure_size.get( "size" )
        w_inch = rect.get( "width" ) / dpi / 2
        h_inch = rect.get( "height" ) / dpi / 2

        return {
            "figsize": ( w_inch, h_inch ),
            "dpi": dpi,
        }

    def _permeability( self, **kwargs: Any ) -> Figure:
        # read data
        assert self.source.input_file is not None
        if self.source.input_file is None:
            return Figure
        water_x = np.array( [] )
        water_y = np.array( [] )
        gaz_x = np.array( [] )
        gaz_y = np.array( [] )
        for f in self.source.plots():
            for t in f.table_function:
                if t.name == "waterRelativePermeabilityTable":
                    fileX = t.coordinate_files.strip( "{(.+)}" ).strip()
                    assert fileX is not None and t.voxel_file is not None
                    water_x = np.loadtxt( self.source.input_file.path + "/" + fileX )
                    water_y = np.loadtxt( self.source.input_file.path + "/" + t.voxel_file )

                if t.name == "gasRelativePermeabilityTable":
                    fileX = t.coordinate_files.strip( "{(.+)}" ).strip()
                    assert fileX is not None and t.voxel_file is not None
                    gaz_x = np.loadtxt( self.source.input_file.path + "/" + fileX )
                    gaz_x = 1. - gaz_x
                    gaz_y = np.loadtxt( self.source.input_file.path + "/" + t.voxel_file )

        # make drawing
        plt.close( "all" )
        fig, ax = plt.subplots( **kwargs )

        if all( a.size > 0 for a in [ gaz_x, gaz_y, water_x, water_y ] ):
            ax.plot( water_x, water_y, '+-', label="water" )
            ax.plot( gaz_x, gaz_y, '+-', label="gaz" )

            ax.set_xlabel( "Water saturation" )
            ax.set_ylabel( "Relative permeability" )
            ax.set_title( "Relative Permeabilities", size=14 )
            ax.grid( color="lightgray", alpha=0.7 )
            plt.xlim( [ 0, 1 ] )
            plt.ylim( [ 0, 1 ] )
            plt.legend()

        return fig