Drift at different depths

from datetime import datetime, timedelta
import numpy as np
from opendrift.models.oceandrift import OceanDrift

o = OceanDrift(loglevel=20)  # Set loglevel to 0 for debug information
16:40:58 INFO    opendrift.models.basemodel:528: OpenDriftSimulation initialised (version 1.10.7 / v1.10.6-168-g8266719)

Using live data from Thredds

o.add_readers_from_list([
    'https://thredds.met.no/thredds/dodsC/sea/norkyst800m/1h/aggregate_be'])

Adding some diffusion

o.set_config('drift:horizontal_diffusivity', 10)  # m2/s

Seed 1000 elements at random depths

z = -np.random.rand(2000)*50
o.seed_elements(lon=4.8, lat=60.0, z=z, radius=0, number=2000,
                time=datetime.utcnow())

print(o)
16:40:58 INFO    opendrift.models.basemodel.environment:220: Adding a dynamical landmask with max. priority based on assumed maximum speed of 1.0 m/s. Adding a customised landmask may be faster...
16:41:04 INFO    opendrift.models.basemodel.environment:247: Fallback values will be used for the following variables which have no readers:
16:41:04 INFO    opendrift.models.basemodel.environment:250:    x_sea_water_velocity: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    y_sea_water_velocity: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    x_wind: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    y_wind: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    upward_sea_water_velocity: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    ocean_vertical_diffusivity: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    sea_surface_wave_significant_height: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    sea_surface_wave_stokes_drift_x_velocity: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    sea_surface_wave_stokes_drift_y_velocity: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    sea_surface_wave_period_at_variance_spectral_density_maximum: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    sea_surface_wave_mean_period_from_variance_spectral_density_second_frequency_moment: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    sea_surface_swell_wave_to_direction: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    sea_surface_swell_wave_peak_period_from_variance_spectral_density: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    sea_surface_swell_wave_significant_height: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    sea_surface_wind_wave_to_direction: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    sea_surface_wind_wave_mean_period: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    sea_surface_wind_wave_significant_height: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    surface_downward_x_stress: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    surface_downward_y_stress: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    turbulent_kinetic_energy: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    turbulent_generic_length_scale: 0.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    ocean_mixed_layer_thickness: 50.000000
16:41:04 INFO    opendrift.models.basemodel.environment:250:    sea_floor_depth_below_sea_level: 10000.000000
===========================
Model:  OceanDrift     (OpenDrift version 1.10.7)
        0 active Lagrangian3DArray particles  (0 deactivated, 2000 scheduled)
-------------------
Environment variables:
  -----
  land_binary_mask
     1) global_landmask
  -----
Readers not added for the following variables:
  ocean_mixed_layer_thickness
  ocean_vertical_diffusivity
  sea_floor_depth_below_sea_level
  sea_surface_swell_wave_peak_period_from_variance_spectral_density
  sea_surface_swell_wave_significant_height
  sea_surface_swell_wave_to_direction
  sea_surface_wave_mean_period_from_variance_spectral_density_second_frequency_moment
  sea_surface_wave_period_at_variance_spectral_density_maximum
  sea_surface_wave_significant_height
  sea_surface_wave_stokes_drift_x_velocity
  sea_surface_wave_stokes_drift_y_velocity
  sea_surface_wind_wave_mean_period
  sea_surface_wind_wave_significant_height
  sea_surface_wind_wave_to_direction
  surface_downward_x_stress
  surface_downward_y_stress
  turbulent_generic_length_scale
  turbulent_kinetic_energy
  upward_sea_water_velocity
  x_sea_water_velocity
  x_wind
  y_sea_water_velocity
  y_wind
---
Lazy readers:
  LazyReader: https://thredds.met.no/thredds/dodsC/sea/norkyst800m/1h/aggregate_be

Discarded readers:
===========================

Running model

o.run(duration=timedelta(hours=24), time_step=1800)
16:41:04 INFO    opendrift.models.basemodel:891: Using existing reader for land_binary_mask
16:41:04 INFO    opendrift.models.basemodel:903: All points are in ocean
16:41:04 INFO    opendrift.models.basemodel:1989: 2023-11-30 16:40:58.609004 - step 1 of 48 - 2000 active elements (0 deactivated)
16:41:04 INFO    opendrift.readers.reader_netCDF_CF_generic:92: Opening dataset: https://thredds.met.no/thredds/dodsC/sea/norkyst800m/1h/aggregate_be
/opt/conda/envs/opendrift/lib/python3.11/site-packages/pyproj/crs/crs.py:1286: UserWarning: You will likely lose important projection information when converting to a PROJ string from another format. See: https://proj.org/faq.html#what-is-the-best-format-for-describing-coordinate-reference-systems
  proj = self._crs.to_proj4(version=version)
16:41:08 INFO    opendrift.models.basemodel:1989: 2023-11-30 17:10:58.609004 - step 2 of 48 - 2000 active elements (0 deactivated)
16:41:10 INFO    opendrift.models.basemodel:1989: 2023-11-30 17:40:58.609004 - step 3 of 48 - 2000 active elements (0 deactivated)
16:41:10 INFO    opendrift.models.basemodel:1989: 2023-11-30 18:10:58.609004 - step 4 of 48 - 2000 active elements (0 deactivated)
16:41:12 INFO    opendrift.models.basemodel:1989: 2023-11-30 18:40:58.609004 - step 5 of 48 - 2000 active elements (0 deactivated)
16:41:12 INFO    opendrift.models.basemodel:1989: 2023-11-30 19:10:58.609004 - step 6 of 48 - 2000 active elements (0 deactivated)
16:41:13 INFO    opendrift.models.basemodel:1989: 2023-11-30 19:40:58.609004 - step 7 of 48 - 2000 active elements (0 deactivated)
16:41:13 INFO    opendrift.models.basemodel:1989: 2023-11-30 20:10:58.609004 - step 8 of 48 - 2000 active elements (0 deactivated)
16:41:15 INFO    opendrift.models.basemodel:1989: 2023-11-30 20:40:58.609004 - step 9 of 48 - 2000 active elements (0 deactivated)
16:41:15 INFO    opendrift.models.basemodel:1989: 2023-11-30 21:10:58.609004 - step 10 of 48 - 2000 active elements (0 deactivated)
16:41:17 INFO    opendrift.models.basemodel:1989: 2023-11-30 21:40:58.609004 - step 11 of 48 - 2000 active elements (0 deactivated)
16:41:17 INFO    opendrift.models.basemodel:1989: 2023-11-30 22:10:58.609004 - step 12 of 48 - 2000 active elements (0 deactivated)
16:41:18 INFO    opendrift.models.basemodel:1989: 2023-11-30 22:40:58.609004 - step 13 of 48 - 2000 active elements (0 deactivated)
16:41:19 INFO    opendrift.models.basemodel:1989: 2023-11-30 23:10:58.609004 - step 14 of 48 - 2000 active elements (0 deactivated)
16:41:20 INFO    opendrift.models.basemodel:1989: 2023-11-30 23:40:58.609004 - step 15 of 48 - 2000 active elements (0 deactivated)
16:41:20 INFO    opendrift.models.basemodel:1989: 2023-12-01 00:10:58.609004 - step 16 of 48 - 2000 active elements (0 deactivated)
16:41:21 INFO    opendrift.models.basemodel:1989: 2023-12-01 00:40:58.609004 - step 17 of 48 - 2000 active elements (0 deactivated)
16:41:21 INFO    opendrift.models.basemodel:1989: 2023-12-01 01:10:58.609004 - step 18 of 48 - 2000 active elements (0 deactivated)
16:41:23 INFO    opendrift.models.basemodel:1989: 2023-12-01 01:40:58.609004 - step 19 of 48 - 2000 active elements (0 deactivated)
16:41:23 INFO    opendrift.models.basemodel:1989: 2023-12-01 02:10:58.609004 - step 20 of 48 - 2000 active elements (0 deactivated)
16:41:24 INFO    opendrift.models.basemodel:1989: 2023-12-01 02:40:58.609004 - step 21 of 48 - 2000 active elements (0 deactivated)
16:41:24 INFO    opendrift.models.basemodel:1989: 2023-12-01 03:10:58.609004 - step 22 of 48 - 2000 active elements (0 deactivated)
16:41:26 INFO    opendrift.models.basemodel:1989: 2023-12-01 03:40:58.609004 - step 23 of 48 - 2000 active elements (0 deactivated)
16:41:26 INFO    opendrift.models.basemodel:1989: 2023-12-01 04:10:58.609004 - step 24 of 48 - 2000 active elements (0 deactivated)
16:41:27 INFO    opendrift.models.basemodel:1989: 2023-12-01 04:40:58.609004 - step 25 of 48 - 2000 active elements (0 deactivated)
16:41:27 INFO    opendrift.models.basemodel:1989: 2023-12-01 05:10:58.609004 - step 26 of 48 - 2000 active elements (0 deactivated)
16:41:28 INFO    opendrift.models.basemodel:1989: 2023-12-01 05:40:58.609004 - step 27 of 48 - 2000 active elements (0 deactivated)
16:41:28 INFO    opendrift.models.basemodel:1989: 2023-12-01 06:10:58.609004 - step 28 of 48 - 2000 active elements (0 deactivated)
16:41:31 INFO    opendrift.models.basemodel:1989: 2023-12-01 06:40:58.609004 - step 29 of 48 - 2000 active elements (0 deactivated)
16:41:31 INFO    opendrift.models.basemodel:1989: 2023-12-01 07:10:58.609004 - step 30 of 48 - 2000 active elements (0 deactivated)
/root/project/opendrift/readers/interpolation/interpolators.py:17: RuntimeWarning: overflow encountered in cast
  data[mask] = np.finfo(np.float64).min
16:41:32 INFO    opendrift.models.basemodel:1989: 2023-12-01 07:40:58.609004 - step 31 of 48 - 1998 active elements (2 deactivated)
16:41:33 INFO    opendrift.models.basemodel:1989: 2023-12-01 08:10:58.609004 - step 32 of 48 - 1992 active elements (8 deactivated)
16:41:34 INFO    opendrift.models.basemodel:1989: 2023-12-01 08:40:58.609004 - step 33 of 48 - 1987 active elements (13 deactivated)
16:41:34 INFO    opendrift.models.basemodel:1989: 2023-12-01 09:10:58.609004 - step 34 of 48 - 1981 active elements (19 deactivated)
16:41:36 INFO    opendrift.models.basemodel:1989: 2023-12-01 09:40:58.609004 - step 35 of 48 - 1974 active elements (26 deactivated)
16:41:36 INFO    opendrift.models.basemodel:1989: 2023-12-01 10:10:58.609004 - step 36 of 48 - 1966 active elements (34 deactivated)
16:41:38 INFO    opendrift.models.basemodel:1989: 2023-12-01 10:40:58.609004 - step 37 of 48 - 1963 active elements (37 deactivated)
16:41:38 INFO    opendrift.models.basemodel:1989: 2023-12-01 11:10:58.609004 - step 38 of 48 - 1955 active elements (45 deactivated)
16:41:40 INFO    opendrift.models.basemodel:1989: 2023-12-01 11:40:58.609004 - step 39 of 48 - 1943 active elements (57 deactivated)
16:41:40 INFO    opendrift.models.basemodel:1989: 2023-12-01 12:10:58.609004 - step 40 of 48 - 1929 active elements (71 deactivated)
16:41:42 INFO    opendrift.models.basemodel:1989: 2023-12-01 12:40:58.609004 - step 41 of 48 - 1919 active elements (81 deactivated)
16:41:42 INFO    opendrift.models.basemodel:1989: 2023-12-01 13:10:58.609004 - step 42 of 48 - 1901 active elements (99 deactivated)
16:41:43 INFO    opendrift.models.basemodel:1989: 2023-12-01 13:40:58.609004 - step 43 of 48 - 1878 active elements (122 deactivated)
16:41:43 INFO    opendrift.models.basemodel:1989: 2023-12-01 14:10:58.609004 - step 44 of 48 - 1863 active elements (137 deactivated)
16:41:45 INFO    opendrift.models.basemodel:1989: 2023-12-01 14:40:58.609004 - step 45 of 48 - 1841 active elements (159 deactivated)
16:41:45 INFO    opendrift.models.basemodel:1989: 2023-12-01 15:10:58.609004 - step 46 of 48 - 1827 active elements (173 deactivated)
16:41:47 INFO    opendrift.models.basemodel:1989: 2023-12-01 15:40:58.609004 - step 47 of 48 - 1810 active elements (190 deactivated)
16:41:47 INFO    opendrift.models.basemodel:1989: 2023-12-01 16:10:58.609004 - step 48 of 48 - 1792 active elements (208 deactivated)

Plot results with lines and particles colored by depth

print(o)
o.plot(linecolor='z', buffer=.1, show_elements=False, fast=False)
o.animation(color='z', buffer=.1, fast=True)
OpenDrift - OceanDrift 2023-11-30 16:40 to 2023-12-01 16:40 UTC (49 steps)
===========================
--------------------
Reader performance:
--------------------
global_landmask
 0:00:00.0  total
 0:00:00.0  preparing
 0:00:00.0  reading
 0:00:00.0  masking
--------------------
https://thredds.met.no/thredds/dodsC/sea/norkyst800m/1h/aggregate_be
 0:00:41.9  total
 0:00:00.0  preparing
 0:00:40.9  reading
 0:00:00.5  interpolation
 0:00:00.0  interpolation_time
 0:00:00.9  rotating vectors
 0:00:00.0  masking
--------------------
Performance:
   50.9 total time
    5.4 configuration
    0.0 preparing main loop
      0.0 moving elements to ocean
   45.3 main loop
      0.1 updating elements
    0.0 cleaning up
--------------------
===========================
Model:  OceanDrift     (OpenDrift version 1.10.7)
        1758 active Lagrangian3DArray particles  (242 deactivated, 0 scheduled)
-------------------
Environment variables:
  -----
  land_binary_mask
     1) global_landmask
  -----
  sea_floor_depth_below_sea_level
  upward_sea_water_velocity
  x_sea_water_velocity
  x_wind
  y_sea_water_velocity
  y_wind
     1) https://thredds.met.no/thredds/dodsC/sea/norkyst800m/1h/aggregate_be
  -----
Readers not added for the following variables:
  ocean_mixed_layer_thickness
  ocean_vertical_diffusivity
  sea_surface_swell_wave_peak_period_from_variance_spectral_density
  sea_surface_swell_wave_significant_height
  sea_surface_swell_wave_to_direction
  sea_surface_wave_mean_period_from_variance_spectral_density_second_frequency_moment
  sea_surface_wave_period_at_variance_spectral_density_maximum
  sea_surface_wave_significant_height
  sea_surface_wave_stokes_drift_x_velocity
  sea_surface_wave_stokes_drift_y_velocity
  sea_surface_wind_wave_mean_period
  sea_surface_wind_wave_significant_height
  sea_surface_wind_wave_to_direction
  surface_downward_x_stress
  surface_downward_y_stress
  turbulent_generic_length_scale
  turbulent_kinetic_energy

Discarded readers:

Time:
        Start: 2023-11-30 16:40:58.609004 UTC
        Present: 2023-12-01 16:40:58.609004 UTC
        Calculation steps: 48 * 0:30:00 - total time: 1 day, 0:00:00
        Output steps: 49 * 0:30:00
===========================

16:42:59 WARNING opendrift.models.basemodel:2357: Plotting fast. This will make your plots less accurate.
16:43:01 INFO    opendrift.models.basemodel:4514: Saving animation to /root/project/docs/source/gallery/animations/example_depth_0.gif...
16:43:10 INFO    opendrift.models.basemodel:2986: Time to make animation: 0:00:11.176547
../_images/example_depth_0.gif

Total running time of the script: (2 minutes 17.488 seconds)

Gallery generated by Sphinx-Gallery