gerg_plotting 0.0.1__tar.gz

gerg_plotting-0.0.1/PKG-INFO

@@ -0,0 +1,214 @@
+Metadata-Version: 2.1
+Name: gerg_plotting
+Version: 0.0.1
+Summary: 
+Author: Alec Krueger
+Author-email: alecmkrueger@tamu.edu
+Requires-Python: >=3.12,<4.0
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: attrs (>=23.2.0,<24.0.0)
+Requires-Dist: cmocean (>=4.0.3,<5.0.0)
+Requires-Dist: matplotlib (>=3.9.1,<4.0.0)
+Requires-Dist: mayavi (>=4.8.2,<5.0.0)
+Requires-Dist: netcdf4 (>=1.7.1.post1,<2.0.0)
+Requires-Dist: numpy (>=2.0.0,<3.0.0)
+Requires-Dist: pandas (>=2.2.2,<3.0.0)
+Requires-Dist: scipy (>=1.14.0,<2.0.0)
+Requires-Dist: xarray (>=2024.6.0,<2025.0.0)
+Description-Content-Type: text/markdown
+
+<a id="readme-top"></a>
+
+<!-- PROJECT LOGO -->
+<br />
+<div align="center">
+  <a href="https://github.com/alecmkrueger/gerg_plotting">
+    <img src="https://raw.githubusercontent.com/alecmkrueger/project_images/9af2f8f58c93e183ac5473a4474619407aee08d3/gerg_logo.svg" alt="Logo" width="500" height="272">
+  </a>
+
+<h3 align="center">GERG Plotting</h3>
+
+  <p align="center">
+    Data plotting package for GERG
+    <br />
+    <a href="https://github.com/alecmkrueger/gerg_plotting"><strong>Explore the docs »</strong></a>
+    <br />
+    <br />
+    <a href="https://github.com/alecmkrueger/gerg_plotting/issues/new?labels=bug&template=bug-report---.md">Report Bug</a>
+    ·
+    <a href="https://github.com/alecmkrueger/gerg_plotting/issues/new?labels=enhancement&template=feature-request---.md">Request Feature</a>
+  </p>
+</div>
+
+
+
+<!-- TABLE OF CONTENTS -->
+<details>
+  <summary>Table of Contents</summary>
+  <ol>
+    <li>
+      <a href="#about-the-project">About The Project</a>
+      <ul>
+        <li><a href="#built-with">Built With</a></li>
+      </ul>
+    </li>
+    <li>
+      <a href="#getting-started">Getting Started</a>
+      <ul>
+        <li><a href="#dependencies">Dependencies</a></li>
+        <li><a href="#installation">Installation</a></li>
+      </ul>
+    </li>
+    <li><a href="#usage">Usage</a></li>
+    <li><a href="#contributing">Contributing</a></li>
+    <li><a href="#license">License</a></li>
+    <li><a href="#contact">Contact</a></li>
+    <li><a href="#acknowledgments">Acknowledgments</a></li>
+  </ol>
+</details>
+
+
+
+<!-- ABOUT THE PROJECT -->
+## About The Project
+
+This project was created to streamline and standardize the process of generating plots at GERG.
+
+
+
+### Built With
+
+[![Python][Python]][Python-url]
+
+
+
+<!-- GETTING STARTED -->
+## Getting Started
+
+There are three ways to get started
+1. Create a fresh virtual environment using your favorite method and install the dependencies
+2. Use an already established virtual environment and install the dependencies
+
+
+
+### Dependencies
+I have provided some commands to install the dependencies using conda but you can use any package manager
+
+List of dependencies:
+* python = 3.12
+* numpy = 2.0.0
+* pandas = 2.2.2
+* matplotlib = 3.9.1
+* xarray = 2024.6.0
+* attrs = 23.2.0
+* netcdf4 = 1.7.1.post1
+* cmocean = 4.0.3
+* scipy = 1.14.0
+* mayavi = 4.8.2
+
+1. #### Creating your own virtual environment then installing dependencies
+    You can change "gerg_plotting" to your desired environment name 
+
+    ```sh
+    conda create -n gerg_plotting python=3.12
+    ```
+    
+    ```sh
+    conda activate gerg_plotting
+    ```
+
+    ```sh
+    pip install numpy pandas xarray matplotlib attrs notebook
+    ```
+
+2. #### Using an already established virtual environment
+
+    ```sh
+    conda activate your_env
+    ```
+
+    ```sh
+    pip install numpy pandas xarray gsw attrs
+    ```
+
+### Installation
+
+1. Activate your virtual environment
+1. Verify/Install Dependencies
+1. Clone the repo
+   ```sh
+   git clone https://github.com/alecmkrueger/gerg_plotting.git
+   ```
+
+
+
+
+
+<!-- USAGE EXAMPLES -->
+## Usage
+
+Plot data at GERG using Python.
+
+
+
+
+<!-- CONTRIBUTING -->
+## Contributing
+
+Contributions are what make the open source community such an amazing place to learn, inspire, and create. Any contributions you make are **greatly appreciated**.
+
+If you have a suggestion that would make this better, please fork the repo and create a pull request. You can also simply open an issue with the tag "enhancement".
+
+1. Fork the Project
+2. Create your Feature Branch (`git checkout -b feature/AmazingFeature`)
+3. Commit your Changes (`git commit -m 'Add some AmazingFeature'`)
+4. Push to the Branch (`git push origin feature/AmazingFeature`)
+5. Open a Pull Request
+
+
+
+
+
+<!-- LICENSE -->
+## License
+
+Distributed under the MIT License. See `LICENSE` for more information.
+
+
+
+<!-- CONTACT -->
+## Contact
+
+Alec Krueger - alecmkrueger@tamu.edu
+
+Project Link: [https://github.com/alecmkrueger/gerg_plotting](https://github.com/alecmkrueger/gerg_plotting)
+
+
+
+<!-- ACKNOWLEDGMENTS -->
+## Acknowledgments
+
+* Alec Krueger, Texas A&M University, Geochemical and Environmental Research Group, alecmkrueger@tamu.edu
+
+<p align="right">(<a href="#readme-top">back to top</a>)</p>
+
+
+
+<!-- MARKDOWN LINKS & IMAGES -->
+<!-- https://www.markdownguide.org/basic-syntax/#reference-style-links -->
+[contributors-shield]: https://img.shields.io/github/contributors/alecmkrueger/gerg_plotting.svg?style=for-the-badge
+[contributors-url]: https://github.com/alecmkrueger/gerg_plotting/graphs/contributors
+[forks-shield]: https://img.shields.io/github/forks/alecmkrueger/gerg_plotting.svg?style=for-the-badge
+[forks-url]: https://github.com/alecmkrueger/gerg_plotting/network/members
+[stars-shield]: https://img.shields.io/github/stars/alecmkrueger/gerg_plotting.svg?style=for-the-badge
+[stars-url]: https://github.com/alecmkrueger/gerg_plotting/stargazers
+[issues-shield]: https://img.shields.io/github/issues/alecmkrueger/gerg_plotting.svg?style=for-the-badge
+[issues-url]: https://github.com/alecmkrueger/gerg_plotting/issues
+[license-shield]: https://img.shields.io/github/license/alecmkrueger/gerg_plotting.svg?style=for-the-badge
+[license-url]: https://github.com/alecmkrueger/gerg_plotting/LICENSE
+[linkedin-shield]: https://img.shields.io/badge/-LinkedIn-black.svg?style=for-the-badge&logo=linkedin&colorB=555
+[linkedin-url]: https://linkedin.com/in/aleckrueger
+[product-screenshot]: images/screenshot.png
+[Python]: https://img.shields.io/badge/python-000000?&logo=python
+[Python-url]: https://www.python.org/

gerg_plotting-0.0.1/README.md

@@ -0,0 +1,194 @@
+<a id="readme-top"></a>
+
+<!-- PROJECT LOGO -->
+<br />
+<div align="center">
+  <a href="https://github.com/alecmkrueger/gerg_plotting">
+    <img src="https://raw.githubusercontent.com/alecmkrueger/project_images/9af2f8f58c93e183ac5473a4474619407aee08d3/gerg_logo.svg" alt="Logo" width="500" height="272">
+  </a>
+
+<h3 align="center">GERG Plotting</h3>
+
+  <p align="center">
+    Data plotting package for GERG
+    <br />
+    <a href="https://github.com/alecmkrueger/gerg_plotting"><strong>Explore the docs »</strong></a>
+    <br />
+    <br />
+    <a href="https://github.com/alecmkrueger/gerg_plotting/issues/new?labels=bug&template=bug-report---.md">Report Bug</a>
+    ·
+    <a href="https://github.com/alecmkrueger/gerg_plotting/issues/new?labels=enhancement&template=feature-request---.md">Request Feature</a>
+  </p>
+</div>
+
+
+
+<!-- TABLE OF CONTENTS -->
+<details>
+  <summary>Table of Contents</summary>
+  <ol>
+    <li>
+      <a href="#about-the-project">About The Project</a>
+      <ul>
+        <li><a href="#built-with">Built With</a></li>
+      </ul>
+    </li>
+    <li>
+      <a href="#getting-started">Getting Started</a>
+      <ul>
+        <li><a href="#dependencies">Dependencies</a></li>
+        <li><a href="#installation">Installation</a></li>
+      </ul>
+    </li>
+    <li><a href="#usage">Usage</a></li>
+    <li><a href="#contributing">Contributing</a></li>
+    <li><a href="#license">License</a></li>
+    <li><a href="#contact">Contact</a></li>
+    <li><a href="#acknowledgments">Acknowledgments</a></li>
+  </ol>
+</details>
+
+
+
+<!-- ABOUT THE PROJECT -->
+## About The Project
+
+This project was created to streamline and standardize the process of generating plots at GERG.
+
+
+
+### Built With
+
+[![Python][Python]][Python-url]
+
+
+
+<!-- GETTING STARTED -->
+## Getting Started
+
+There are three ways to get started
+1. Create a fresh virtual environment using your favorite method and install the dependencies
+2. Use an already established virtual environment and install the dependencies
+
+
+
+### Dependencies
+I have provided some commands to install the dependencies using conda but you can use any package manager
+
+List of dependencies:
+* python = 3.12
+* numpy = 2.0.0
+* pandas = 2.2.2
+* matplotlib = 3.9.1
+* xarray = 2024.6.0
+* attrs = 23.2.0
+* netcdf4 = 1.7.1.post1
+* cmocean = 4.0.3
+* scipy = 1.14.0
+* mayavi = 4.8.2
+
+1. #### Creating your own virtual environment then installing dependencies
+    You can change "gerg_plotting" to your desired environment name 
+
+    ```sh
+    conda create -n gerg_plotting python=3.12
+    ```
+    
+    ```sh
+    conda activate gerg_plotting
+    ```
+
+    ```sh
+    pip install numpy pandas xarray matplotlib attrs notebook
+    ```
+
+2. #### Using an already established virtual environment
+
+    ```sh
+    conda activate your_env
+    ```
+
+    ```sh
+    pip install numpy pandas xarray gsw attrs
+    ```
+
+### Installation
+
+1. Activate your virtual environment
+1. Verify/Install Dependencies
+1. Clone the repo
+   ```sh
+   git clone https://github.com/alecmkrueger/gerg_plotting.git
+   ```
+
+
+
+
+
+<!-- USAGE EXAMPLES -->
+## Usage
+
+Plot data at GERG using Python.
+
+
+
+
+<!-- CONTRIBUTING -->
+## Contributing
+
+Contributions are what make the open source community such an amazing place to learn, inspire, and create. Any contributions you make are **greatly appreciated**.
+
+If you have a suggestion that would make this better, please fork the repo and create a pull request. You can also simply open an issue with the tag "enhancement".
+
+1. Fork the Project
+2. Create your Feature Branch (`git checkout -b feature/AmazingFeature`)
+3. Commit your Changes (`git commit -m 'Add some AmazingFeature'`)
+4. Push to the Branch (`git push origin feature/AmazingFeature`)
+5. Open a Pull Request
+
+
+
+
+
+<!-- LICENSE -->
+## License
+
+Distributed under the MIT License. See `LICENSE` for more information.
+
+
+
+<!-- CONTACT -->
+## Contact
+
+Alec Krueger - alecmkrueger@tamu.edu
+
+Project Link: [https://github.com/alecmkrueger/gerg_plotting](https://github.com/alecmkrueger/gerg_plotting)
+
+
+
+<!-- ACKNOWLEDGMENTS -->
+## Acknowledgments
+
+* Alec Krueger, Texas A&M University, Geochemical and Environmental Research Group, alecmkrueger@tamu.edu
+
+<p align="right">(<a href="#readme-top">back to top</a>)</p>
+
+
+
+<!-- MARKDOWN LINKS & IMAGES -->
+<!-- https://www.markdownguide.org/basic-syntax/#reference-style-links -->
+[contributors-shield]: https://img.shields.io/github/contributors/alecmkrueger/gerg_plotting.svg?style=for-the-badge
+[contributors-url]: https://github.com/alecmkrueger/gerg_plotting/graphs/contributors
+[forks-shield]: https://img.shields.io/github/forks/alecmkrueger/gerg_plotting.svg?style=for-the-badge
+[forks-url]: https://github.com/alecmkrueger/gerg_plotting/network/members
+[stars-shield]: https://img.shields.io/github/stars/alecmkrueger/gerg_plotting.svg?style=for-the-badge
+[stars-url]: https://github.com/alecmkrueger/gerg_plotting/stargazers
+[issues-shield]: https://img.shields.io/github/issues/alecmkrueger/gerg_plotting.svg?style=for-the-badge
+[issues-url]: https://github.com/alecmkrueger/gerg_plotting/issues
+[license-shield]: https://img.shields.io/github/license/alecmkrueger/gerg_plotting.svg?style=for-the-badge
+[license-url]: https://github.com/alecmkrueger/gerg_plotting/LICENSE
+[linkedin-shield]: https://img.shields.io/badge/-LinkedIn-black.svg?style=for-the-badge&logo=linkedin&colorB=555
+[linkedin-url]: https://linkedin.com/in/aleckrueger
+[product-screenshot]: images/screenshot.png
+[Python]: https://img.shields.io/badge/python-000000?&logo=python
+[Python-url]: https://www.python.org/

gerg_plotting-0.0.1/pyproject.toml

@@ -0,0 +1,37 @@
+[tool.poetry]
+name = "gerg_plotting"
+version = "0.0.01"
+description = ""
+authors = ["Alec Krueger <alecmkrueger@tamu.edu>"]
+readme = "README.md"
+packages = [{include = "gerg_plotting", from = "src"}]
+include = ["seafloor_data/gebco_2023_n31.0_s7.0_w-100.0_e-66.5.nc"]
+
+[tool.poetry.dependencies]
+python = "^3.12"
+numpy = "^2.0.0"
+pandas = "^2.2.2"
+matplotlib = "^3.9.1"
+xarray = "^2024.6.0"
+attrs = "^23.2.0"
+netcdf4 = "^1.7.1.post1"
+cmocean = "^4.0.3"
+scipy = "^1.14.0"
+mayavi = "^4.8.2"
+
+
+[tool.poetry.group.dev.dependencies]
+black = "^24.4.2"
+flake8 = "^7.1.0"
+isort = "^5.13.2"
+mypy = "^1.10.1"
+pylint = "^3.2.5"
+
+
+[tool.poetry.group.test.dependencies]
+pytest = "^8.2.2"
+faker = "^26.0.0"
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"

gerg_plotting-0.0.1/src/gerg_plotting/classes_data.py

@@ -0,0 +1,136 @@
+import numpy as np
+from attrs import define,field,asdict,validators
+import matplotlib
+from matplotlib.colors import Colormap
+import cmocean
+from pprint import pformat
+import xarray as xr
+from pathlib import Path
+
+from gerg_plotting.classes_utils import get_center_of_mass,lat_min_smaller_than_max,lon_min_smaller_than_max
+
+@define
+class NonSpatialData:
+    def __getitem__(self, key:str):
+        return asdict(self)[key]
+    def __repr__(self):
+        '''Pretty printing'''
+        return pformat(asdict(self), indent=1,width=2,compact=True,depth=1)
+
+class Lab(NonSpatialData):
+    def __init__(self,vars):
+        for key,value in vars.items():
+            setattr(self,key,value)
+    def __getitem__(self, key:str):
+        return getattr(self,key)
+    def __setitem__(self,key,value):
+        return setattr(self,key,value)
+    def __repr__(self):
+        '''Pretty printing'''
+        return pformat(self.__dict__, indent=1,width=2,compact=True,depth=1)
+
+@define
+class Bounds(NonSpatialData):
+    lat_min:float|int|None = field(default=None,validator=[validators.instance_of(float|int|None),lat_min_smaller_than_max])
+    lat_max:float|int = field(default=None)
+    
+    lon_min:float|int|None = field(default=None,validator=[validators.instance_of(float|int|None),lon_min_smaller_than_max])
+    lon_max:float|int|None = field(default=None)
+
+    depth_bottom:float|int|None = field(default=None)
+    depth_top:float|int|None = field(default=None)
+
+
+@define
+class SpatialData:
+    # Dims
+    lat:np.ndarray = field(default=None)
+    lon:np.ndarray = field(default=None)
+    depth:np.ndarray = field(default=None)
+    time:np.ndarray = field(default=None)
+    # Cmaps
+    temperature_cmap:Colormap = field(default=cmocean.cm.thermal)
+    salinity_cmap:Colormap = field(default=cmocean.cm.haline)
+    u_current_cmap:Colormap = field(default=cmocean.cm.delta)
+    v_current_cmap:Colormap = field(default=cmocean.cm.delta)
+
+    def __getitem__(self, key:str):
+        return asdict(self)[key]
+    def __repr__(self):
+        '''Pretty printing'''
+        return pformat(asdict(self), indent=1,width=2,compact=True,depth=1)
+
+@define
+class Bathy(SpatialData):
+    # Vars
+    bounds:Bounds = field(default=None)
+    resolution_level:float|int|None = field(default=5)
+    cmap:Colormap = field(default=matplotlib.cm.get_cmap('Blues'))
+    vertical_scaler:int|float = field(default=None)
+    vertical_units:str = field(default='')
+    center_of_mass:tuple = field(init=False)
+
+    def __attrs_post_init__(self):
+        self.get_bathy()
+        if self.vertical_scaler is not None:
+            self.depth = self.depth*self.vertical_scaler
+        self.center_of_mass = get_center_of_mass(self.lon,self.lat,self.depth)
+
+    def get_bathy(self):
+        '''
+        bounds (Bounds): contains attributes of lat_min,lon_min,lat_max,lon_max,depth_max,depth_min
+        resolution_level (float|int): how much to coarsen the dataset by in units of degrees
+        '''
+
+        # seafloor_path = Path('seafloor_data/gebco_2023_n31.0_s7.0_w-100.0_e-66.5.nc')
+        self_path = Path(__file__)
+        seafloor_path = self_path.parent.parent.joinpath('seafloor_data/gebco_2023_n31.0_s7.0_w-100.0_e-66.5.nc')
+        ds = xr.open_dataset(seafloor_path) #read in seafloor data
+
+        if self.resolution_level is not None:
+            ds = ds.coarsen(lat=self.resolution_level,boundary='trim').mean().coarsen(lon=self.resolution_level,boundary='trim').mean() #coarsen the seafloor data (speed up figure drawing) #type:ignore
+
+        ds = ds.sel(lat=slice(self.bounds["lat_min"],self.bounds["lat_max"])).sel(lon=slice(self.bounds["lon_min"],self.bounds["lon_max"])) #slice to the focus area
+
+        self.depth = ds['elevation'].values*-1 #extract the depth values and flip them
+
+        if self.bounds["depth_top"] is not None:
+            self.depth = np.where(self.depth>self.bounds["depth_top"],self.depth,self.bounds["depth_top"]) #set all depth values less than the depth_top to the same value as depth_top for visuals
+        if self.bounds["depth_bottom"] is not None:
+            self.depth = np.where(self.depth<self.bounds["depth_bottom"],self.depth,self.bounds["depth_bottom"]) #set all depth values less than the depth_bottom to the same value as depth_bottom for visuals
+
+        self.lon = ds.coords['lat'].values #extract the latitude values
+        self.lat = ds.coords['lon'].values #extract the longitude values
+        self.lon, self.lat = np.meshgrid(self.lat, self.lon) #create meshgrid for plotting
+
+
+@define
+class Glider(SpatialData):
+    # Vars
+    temperature:np.ndarray = field(default=None)
+    salinity:np.ndarray = field(default=None)
+
+@define
+class Buoy(SpatialData):
+    # Vars
+    u_current:np.ndarray = field(default=None)
+    v_current:np.ndarray = field(default=None)
+
+@define
+class CTD(SpatialData):
+    # Dim
+    stations:np.ndarray = field(default=None)
+    # Vars
+    temperature:np.ndarray = field(default=None)
+    salinity:np.ndarray = field(default=None)
+
+@define
+class WaveGlider(SpatialData):
+    # Vars
+    temperature:np.ndarray = field(default=None)
+    salinity:np.ndarray  = field(default=None)
+    
+@define
+class Radar(SpatialData):
+    u_current:np.ndarray = field(default=None)
+    v_current:np.ndarray = field(default=None)

gerg_plotting-0.0.1/src/gerg_plotting/classes_plotter2d.py

@@ -0,0 +1,120 @@
+import matplotlib
+import matplotlib.axes
+import matplotlib.cm
+import matplotlib.figure
+import matplotlib.pyplot
+import matplotlib.dates as mdates
+import numpy as np
+from attrs import define, field, asdict
+from pprint import pformat
+import cmocean
+
+from gerg_plotting.classes_data import Bathy,NonSpatialData,SpatialData,Bounds
+from gerg_plotting.utils import calculate_range
+
+@define
+class Plotter:
+    instrument:SpatialData|NonSpatialData
+    bounds:Bounds|None = field(default= None)
+
+    fig:matplotlib.figure.Figure = field(default=None)
+    ax:matplotlib.axes.Axes = field(default=None)
+
+    def init_figure(self,fig=None,ax=None,three_d=False):
+        if fig is None and ax is None:
+            self.fig,self.ax = matplotlib.pyplot.subplots(subplot_kw={'projection':('3d' if three_d else None)})
+        elif fig is not None and ax is not None:
+            self.fig = fig
+            self.ax = ax
+            if three_d:
+                index = [idx for idx,ax in enumerate(self.fig.axes) if ax is self.ax][0]+1
+                self.ax.remove()
+                gs = self.ax.get_gridspec()
+                self.ax = fig.add_subplot(gs.nrows,gs.ncols,index, projection='3d')
+
+    def __getitem__(self, key:str):
+        return asdict(self)[key]
+    def __repr__(self):
+        '''Pretty printing'''
+        return pformat(asdict(self), indent=1,width=2,compact=True,depth=1)
+
+@define
+class SurfacePlot(Plotter):
+    bathy:Bathy = field(init=False)
+
+    def init_bathy(self):
+        self.bathy = Bathy(bounds=self.bounds,resolution_level=5)
+
+    def map(self,var:str|None=None,surface_values:bool=False,fig=None,ax=None,seafloor=True):
+        self.init_figure(fig,ax)
+        if var is None:
+            color = 'k'
+            cmap = None
+        else:
+            color_var_values = self.instrument[var].copy()
+            if surface_values:
+                color_var_values[self.instrument.depth>2] = np.nan
+            color = color_var_values
+            cmap = self.instrument[f'{var}_cmap']
+        if self.bounds is not None:
+            self.ax.set_ylim(self.bounds.lat_min,self.bounds.lat_max)
+            self.ax.set_xlim(self.bounds.lon_min,self.bounds.lon_max)
+        if seafloor:
+            self.init_bathy()
+            # Remove the white most but of the colormap
+            self.bathy.cmap = cmocean.tools.crop_by_percent(self.bathy.cmap,20,'min')
+            # Add land color to the colormap
+            land_color = [231/255,194/255,139/255,1]
+            self.bathy.cmap.set_under(land_color)
+            self.ax.contourf(self.bathy.lon,self.bathy.lat,self.bathy.depth,levels=50,cmap=self.bathy.cmap,vmin=0)
+
+        self.ax.scatter(self.instrument.lon,self.instrument.lat,c=color,cmap=cmap,s=2)
+    def quiver(self):
+        self.init_figure()
+        raise NotImplementedError('Need to add Quiver')
+
+
+@define
+class DepthPlot(Plotter):
+
+    def time_series(self,var:str,fig=None,ax=None):
+        self.init_figure(fig,ax)
+        self.ax.scatter(self.instrument.time,self.instrument.depth,c=self.instrument[var],cmap=self.instrument[f'{var}_cmap'])
+        self.ax.invert_yaxis()
+        locator = mdates.AutoDateLocator()
+        formatter = mdates.AutoDateFormatter(locator)
+
+        self.ax.xaxis.set_major_locator(locator)
+        self.ax.xaxis.set_major_formatter(formatter)
+        matplotlib.pyplot.xticks(rotation=60, fontsize='small')
+
+    def var_var(self,x:str,y:str,color_var:str|None=None,fig=None,ax=None):
+        self.init_figure(fig,ax)
+        if color_var is not None:
+            self.ax.scatter(self.instrument[x],self.instrument[y],c=self.instrument[color_var])
+        elif color_var is None:
+            self.ax.scatter(self.instrument[x],self.instrument[y])
+
+    def cross_section(self,longitude,latitude):
+        raise NotImplementedError('Need to add method to plot cross sections')
+
+@define
+class Histogram(Plotter):
+
+    def plot(self,var:str,fig=None,ax=None):
+        self.init_figure(fig,ax)
+        self.ax.hist(self.instrument[var])
+
+    def plot2d(self,x:str,y:str,fig=None,ax=None,**kwargs):
+        self.init_figure(fig,ax)
+        range = [calculate_range(self.instrument[x]),calculate_range(self.instrument[y])]
+        self.ax.hist2d(self.instrument[x],self.instrument[y],range=range,**kwargs)
+
+    def plot3d(self,x:str,y:str,fig=None,ax=None,**kwargs):
+        from matplotlib import cm
+        self.init_figure(fig,ax,three_d=True)
+        range = [calculate_range(self.instrument[x]),calculate_range(self.instrument[y])]
+        h,xedges,yedges = np.histogram2d(self.instrument[x],self.instrument[y],range=range,**kwargs)
+        X,Y = np.meshgrid(xedges[1:],yedges[1:])
+        self.ax.plot_surface(X,Y,h, rstride=1, cstride=1, cmap=cm.coolwarm,
+                       linewidth=0, antialiased=False)

gerg_plotting-0.0.1/src/gerg_plotting/classes_plotter3d_.py

@@ -0,0 +1,8 @@
+from attrs import define
+import numpy as np
+from gerg_plotting.classes_data import SpatialData
+
+
+@define
+class Plotter3D:
+    instrument: SpatialData

gerg_plotting-0.0.1/src/gerg_plotting/classes_utils.py

@@ -0,0 +1,22 @@
+import numpy as np
+
+def lat_min_smaller_than_max(instance, attribute, value):
+    if value is not None:
+        if value >= instance.lat_max:
+            raise ValueError("'lat_min' must to be smaller than 'lat_max'")
+    
+def lon_min_smaller_than_max(instance, attribute, value):
+    if value is not None:
+        if value >= instance.lon_max:
+            raise ValueError("'lon_min' must to be smaller than 'lon_max'")
+        
+def validate_array_lengths(instance,attribute,value):
+    lengths = {attr.name:len(getattr(instance,attr.name)) for attr in instance.__attrs_attrs__ if isinstance(getattr(instance,attr.name),np.ndarray)}
+    if len(set(lengths.values()))>1:
+        raise ValueError(f'All Dims and Vars must be the same length, got lengths of {lengths}')
+
+        
+def get_center_of_mass(lon,lat,pressure) -> tuple:
+    centroid = tuple([np.nanmean(lon), np.nanmean(lat), np.nanmean(pressure)])
+    return centroid
+

gerg_plotting-0.0.1/src/gerg_plotting/testing.py

@@ -0,0 +1,62 @@
+import numpy as np
+import pandas as pd
+import datetime
+import xarray as xr
+import matplotlib.pyplot as plt
+from gerg_plotting.classes_plotter2d import SurfacePlot,DepthPlot,Histogram
+from gerg_plotting.classes_data import Radar,Bounds
+from gerg_plotting.utils import interp_data,filter_var,calculate_range
+
+df = pd.read_csv('../../test_data/radar.csv',
+                 parse_dates=['time'],date_format='%Y-%m-%dT%H:%M:%SZ',skiprows=[1])
+
+times = list(set(df.time))
+
+df = df.loc[df.time==times[2]]
+
+radar = Radar(lat=df['latitude'],
+              lon=df['longitude'],
+              depth=np.array([0]),
+              time=df['time'],
+              u_current=df['u'],
+              v_current=df['v'])
+
+lat_min,lat_max = calculate_range(radar.lat)
+lon_min,lon_max = calculate_range(radar.lon)
+
+
+bounds = Bounds(lat_min=lat_min,
+                lat_max=lat_max,
+                lon_max=lon_max,
+                lon_min=lon_min,
+                depth_bottom=1000,
+                depth_top=None)
+
+
+fig,axes = plt.subplots(nrows=3,figsize = (10,10))
+surfaces = SurfacePlot(instrument=radar,bounds=bounds)
+surfaces.map(fig=fig,ax=axes[0])
+surfaces.map(fig=fig,ax=axes[1],var='u_current',surface_values=False)
+surfaces.map(fig=fig,ax=axes[2],var='v_current',surface_values=False)
+plt.show()
+
+# depth_plot = DepthPlot(instrument=radar,bounds=bounds)
+
+# depth_plot.time_series(var='u_current')
+# plt.show()
+# depth_plot.time_series(var='v_current')
+# plt.show()
+# depth_plot.var_var(x='v_current',y='u_current',color_var='time')
+# plt.show()
+
+fig,axes = plt.subplots(nrows=4,figsize = (5,20))
+hist = Histogram(instrument=radar,bounds=bounds)
+hist.plot(fig=fig,ax=axes[0],var='u_current')
+# plt.show()
+hist.plot(fig=fig,ax=axes[1],var='v_current')
+# plt.show()
+hist.plot2d(fig=fig,ax=axes[2],x='u_current',y='v_current',bins=150,norm='log')
+hist.ax.invert_yaxis()
+# plt.show()
+hist.plot3d(fig=fig,ax=axes[3],x='u_current',y='v_current',bins=150)
+plt.show()

gerg_plotting-0.0.1/src/gerg_plotting/utils.py

@@ -0,0 +1,31 @@
+import matplotlib.pyplot as plt
+import numpy as np
+import pandas as pd
+import xarray as xr
+
+def interp_data(ds:xr.Dataset) -> pd.DataFrame:
+    new_time_values = ds['time'].values.astype('datetime64[s]').astype('float64')
+    new_mtime_values = ds['m_time'].values.astype('datetime64[s]').astype('float64')
+
+    # Create a mask of non-NaN values in the 'longitude' variable
+    valid_longitude = ~np.isnan(ds['longitude'])
+    # Now ds_filtered contains the data where NaN values have been dropped based on the longitude variable
+
+    ds['latitude'] = xr.DataArray(np.interp(new_time_values, new_mtime_values[valid_longitude], ds['latitude'].values[valid_longitude]),[('time',ds.time.values)])
+    ds['longitude'] = xr.DataArray(np.interp(new_time_values, new_mtime_values[valid_longitude], ds['longitude'].values[valid_longitude]),[('time',ds.time.values)])
+
+    df = ds[['latitude','longitude','pressure','salinity','temperature']].to_dataframe().reset_index()
+    df['time'] = df['time'].astype('datetime64[s]')
+    # df = df.set_index(['time'])
+    df = df.dropna()
+
+    return df
+
+
+def filter_var(var:pd.Series,min_value,max_value):
+    var = var.where(var>min_value)
+    var = var.where(var<max_value)
+    return var
+
+def calculate_range(var:np.ndarray):
+    return [np.nanmin(var),np.nanmax(var)]