solweig-gpu 0.1.0__tar.gz

solweig_gpu-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Harsh Kamath and Naveen Sudharsan
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

solweig_gpu-0.1.0/MANIFEST.in

@@ -0,0 +1,3 @@
+include README.md
+include LICENSE
+

solweig_gpu-0.1.0/PKG-INFO

@@ -0,0 +1,33 @@
+Metadata-Version: 2.4
+Name: solweig-gpu
+Version: 0.1.0
+Summary: GPU-accelerated SOLWEIG model for urban thermal comfort simulation
+Home-page: https://github.com/nvnsudharsan/utci-pipeline
+Author: Harsh Kamath, Naveen Sudharsan
+Author-email: harsh.kamath@utexas.edu, naveens@utexas.edu
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: POSIX :: Linux
+Classifier: Intended Audience :: Science/Research
+Classifier: Topic :: Scientific/Engineering :: Atmospheric Science
+Requires-Python: >=3.8
+License-File: LICENSE
+Requires-Dist: torch
+Requires-Dist: numpy
+Requires-Dist: scipy
+Requires-Dist: pandas
+Requires-Dist: netCDF4
+Requires-Dist: pytz
+Requires-Dist: shapely
+Requires-Dist: timezonefinder
+Requires-Dist: gdal
+Requires-Dist: xarray
+Requires-Dist: tqdm
+Requires-Dist: PyQt5
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

solweig_gpu-0.1.0/README.md

@@ -0,0 +1,147 @@
+# SOLWEIG-GPU: GPU-Accelerated Thermal Comfort Modeling Framework
+
+**SOLWEIG-GPU** is a Python package and command-line interface for running the SOLWEIG (Solar and LongWave Environmental Irradiance Geometry) model on CPU or GPU (if available). It enables high-resolution urban microclimate modeling by computing key variables such as Sky View Factor (SVF), Mean Radiant Temperature (Tmrt), and the Universal Thermal Climate Index (UTCI).
+
+**SOLWEIG** was originally developed by Dr. Fredrik Lindberg's group. Journal reference: Lindberg, F., Holmer, B. & Thorsson, S. SOLWEIG 1.0 – Modelling spatial variations of 3D radiant fluxes and mean radiant temperature in complex urban settings. Int J Biometeorol 52, 697–713 (2008). https://doi.org/10.1007/s00484-008-0162-7
+
+**SOLWEIG GPU** code is an extention of the original **SOLWEIG** python model that is the part of the Urban Multi-scale Environmental Predictor (UMEP) (GitHub code reference: https://github.com/UMEP-dev/UMEP). UMEP journal reference: Lindberg, F., Grimmond, C.S.B., Gabey, A., Huang, B., Kent, C.W., Sun, T., Theeuwes, N.E., Järvi, L., Ward, H.C., Capel-Timms, I. and Chang, Y., 2018. Urban Multi-scale Environmental Predictor (UMEP): An integrated tool for city-based climate services. Environmental modelling & software, 99, pp.70-87. https://doi.org/10.1016/j.envsoft.2017.09.020
+
+---
+
+## Features
+
+- CPU and GPU support (automatically uses GPU if available)
+- Divides larger areas into tiles based on the selected tile size
+- CPU-based computations of wall height and aspect parallelized across multiple CPUs
+- GPU-based computation of SVF, shortwave/longwave radiation, shadows, Tmrt, and UTCI
+- Compatible with meteorological data from UMEP, ERA5, and WRF (`wrfout`)
+
+![SOLWEIG-GPU workflow ](/solweig_diagram.png)
+*Flowchart of the SOLWEIG-GPU modeling framework*
+
+---
+
+## Required Input Data
+
+- `Building DSM`: Includes both buildings and terrain elevation (e.g., `Building_DSM.tif`)
+- `DEM`: Digital Elevation Model excluding buildings (e.g., `DEM.tif`)
+- `Tree DSM`: Vegetation height data only (e.g., `Trees.tif`)
+- Meteorological forcing:
+  - Custom `.txt` file (from UMEP)
+  - ERA5 (both instantaneous and accumulated)
+  - WRF output NetCDF (`wrfout`)
+
+> Please refer to the sample dataset to familiarize yourself with the expected inputs. Sample data can be found at https://utexas.box.com/s/8fctqicidr5cup8kj3tk53jd444pow6z
+
+### ERA5 Variables Required
+- 2-meter air temperature
+- 2-meter dew point temperature
+- Surface pressure
+- 10-meter U and V wind components
+- Downwelling shortwave radiation (accumulated)
+- Downwelling longwave radiation (accumulated)
+
+---
+
+## Output Details
+
+- Output directory: `Outputs/`
+- Structure: One folder per tile (e.g., `tile_0_0/`, `tile_0_600/`)
+- SVF: Single-band raster
+- Other outputs: Multi-band raster (e.g., 24 bands for hourly results)
+
+![UTCI for New Delhi](/UTCI_New_Delhi.jpeg)
+*UTCI for New Delhi, India, generated using SOLWEIG-GPU and visualized with ArcGIS Online.*
+
+---
+
+## Installation
+
+```bash
+conda create -n solweig python=3.10
+conda activate solweig
+conda install -c conda-forge gdal pytorch
+git clone https://github.com/nvnsudharsan/solweig-gpu.git
+cd solweig-gpu
+pip install .
+```
+
+---
+
+## Python Usage
+
+```python
+from solweig_gpu import thermal_comfort
+
+thermal_comfort(
+    base_path='/path/to/input',
+    selected_date_str='2020-08-13',
+    building_dsm_filename='Building_DSM.tif',
+    dem_filename='DEM.tif',
+    trees_filename='Trees.tif',
+    tile_size=3600,
+    use_own_met=True,
+    own_met_file='/path/to/met.txt',
+    start_time='2020-08-13 00:00:00',
+    end_time='2020-08-13 23:00:00',
+    data_source_type='ERA5',  # or 'WRFOUT'
+    data_folder='/path/to/era5_or_wrfout',
+    save_tmrt=True,
+    save_svf=False,
+    save_kup=False,
+    save_kdown=False,
+    save_lup=False,
+    save_ldown=False,
+    save_shadow=False
+)
+```
+
+---
+
+## Command-Line Interface (CLI)
+
+```bash
+conda activate solweig
+thermal_comfort --base_path /path/to/input \
+                --selected_date_str 2020-08-13 \
+                --building_dsm_filename Building_DSM.tif \
+                --dem_filename DEM.tif \
+                --trees_filename Trees.tif \
+                --tile_size 3600 \
+                --use_own_met True \
+                --own_met_file /path/to/met.txt \
+                --start_time "2020-08-13 00:00:00" \
+                --end_time "2020-08-13 23:00:00" \
+                --data_source_type ERA5 \
+                --data_folder /path/to/era5_or_wrfout \
+                --save_tmrt True \
+                --save_svf False
+```
+
+> Tip: Use `--help` to list all CLI options.
+
+---
+
+## GUI Usage
+
+To launch the GUI:
+```bash
+conda activate solweig
+solweig_gpu
+```
+
+![GUI](/GUI.png)
+
+### GUI Workflow
+1. Select the **base path** containing input datasets.
+2. Choose the **Building DSM**, **DEM**, and **Tree DSM** raster files.
+3. Set the **tile size** (e.g., 600 or 1200 pixels).
+4. Select a **meteorological source** (`metfile`, `ERA5`, or `wrfout`):
+   - If `metfile`: Provide a `.txt` file.
+   - If `ERA5`: Provide a folder with both instantaneous and accumulated files.
+   - If `wrfout`: Provide a folder with WRF output NetCDF files.
+5. Set the **start** and **end times** in UTC (`YYYY-MM-DD HH:MM:SS`).
+6. Choose which outputs to generate (e.g., Tmrt, UTCI, radiation fluxes).
+7. Output will be saved in `Outputs/`, with subfolders for each tile.
+
+---

solweig_gpu-0.1.0/pyproject.toml

@@ -0,0 +1,3 @@
+[build-system]
+requires = ["setuptools>=42", "wheel"]
+build-backend = "setuptools.build_meta"

solweig_gpu-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

solweig_gpu-0.1.0/setup.py

@@ -0,0 +1,38 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="solweig-gpu",
+    version="0.1.0",
+    description="GPU-accelerated SOLWEIG model for urban thermal comfort simulation",
+    author="Harsh Kamath, Naveen Sudharsan",
+    author_email="harsh.kamath@utexas.edu, naveens@utexas.edu",
+    url="https://github.com/nvnsudharsan/utci-pipeline",
+    packages=find_packages(),  # Automatically finds `solweig_gpu`
+    install_requires=[
+        "torch",
+        "numpy",
+        "scipy",
+        "pandas",
+        "netCDF4",
+        "pytz",
+        "shapely",
+        "timezonefinder",
+        "gdal",
+        "xarray",
+        "tqdm",
+        "PyQt5",
+    ],
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "Operating System :: POSIX :: Linux",
+        "Intended Audience :: Science/Research",
+        "Topic :: Scientific/Engineering :: Atmospheric Science",
+    ],
+    python_requires=">=3.8",
+    entry_points={
+        'console_scripts': [
+            'thermal_comfort=solweig_gpu.cli:main',
+            'solweig_gpu=solweig_gpu.solweig_gpu_gui:main',
+        ],
+    },
+)

solweig_gpu-0.1.0/solweig_gpu/__init__.py

@@ -0,0 +1 @@
+from .solweig_gpu import thermal_comfort

solweig_gpu-0.1.0/solweig_gpu/calculate_utci.py

@@ -0,0 +1,257 @@
+import torch
+
+# Function to computed UTCI using 6th order polynomial approximation. 
+# Inputs:
+#    - Mean radiant temperature
+#    - Air temperature
+#    - Wind speed
+#    - Air vapor pressure
+
+
+def utci_polynomial(D_Tmrt, Ta, va, Pa):
+    UTCI_approx = Ta + \
+    (6.07562052E-01) + \
+    (-2.27712343E-02) * Ta + \
+    (8.06470249E-04) * Ta**2 + \
+    (-1.54271372E-04) * Ta**3 + \
+    (-3.24651735E-06) * Ta**4 + \
+    (7.32602852E-08) * Ta**5 + \
+    (1.35959073E-09) * Ta**6 + \
+    (-2.25836520E+00) * va + \
+    (8.80326035E-02) * Ta * va + \
+    (2.16844454E-03) * Ta**2 * va + \
+    (-1.53347087E-05) * Ta**3 * va + \
+    (-5.72983704E-07) * Ta**4 * va + \
+    (-2.55090145E-09) * Ta**5 * va + \
+    (-7.51269505E-01) * va**2 + \
+    (-4.08350271E-03) * Ta * va**2 + \
+    (-5.21670675E-05) * Ta**2 * va**2 + \
+    (1.94544667E-06) * Ta**3 * va**2 + \
+    (1.14099531E-08) * Ta**4 * va**2 + \
+    (1.58137256E-01) * va**3 + \
+    (-6.57263143E-05) * Ta * va**3 + \
+    (2.22697524E-07) * Ta**2 * va**3 + \
+    (-4.16117031E-08) * Ta**3 * va**3 + \
+    (-1.27762753E-02) * va**4 + \
+    (9.66891875E-06) * Ta * va**4 + \
+    (2.52785852E-09) * Ta**2 * va**4 + \
+    (4.56306672E-04) * va**5 + \
+    (-1.74202546E-07) * Ta * va**5 + \
+    (-5.91491269E-06) * va**6 + \
+    (3.98374029E-01) * D_Tmrt + \
+    (1.83945314E-04) * Ta * D_Tmrt + \
+    (-1.73754510E-04) * Ta**2 * D_Tmrt + \
+    (-7.60781159E-07) * Ta**3 * D_Tmrt + \
+    (3.77830287E-08) * Ta**4 * D_Tmrt + \
+    (5.43079673E-10) * Ta**5 * D_Tmrt + \
+    (-2.00518269E-02) * va * D_Tmrt + \
+    (8.92859837E-04) * Ta * va * D_Tmrt + \
+    (3.45433048E-06) * Ta**2 * va * D_Tmrt + \
+    (-3.77925774E-07) * Ta**3 * va * D_Tmrt + \
+    (-1.69699377E-09) * Ta**4 * va * D_Tmrt + \
+    (1.69992415E-04) * va**2 * D_Tmrt + \
+    (-4.99204314E-05) * Ta * va**2 * D_Tmrt + \
+    (2.47417178E-07) * Ta**2 * va**2 * D_Tmrt + \
+    (1.07596466E-08) * Ta**3 * va**2 * D_Tmrt + \
+    (8.49242932E-05) * va**3 * D_Tmrt + \
+    (1.35191328E-06) * Ta * va**3 * D_Tmrt + \
+    (-6.21531254E-09) * Ta**2 * va**3 * D_Tmrt + \
+    (-4.99410301E-06) * va**4 * D_Tmrt + \
+    (-1.89489258E-08) * Ta * va**4 * D_Tmrt + \
+    (8.15300114E-08) * va**5 * D_Tmrt + \
+    (7.55043090E-04) * D_Tmrt**2 + \
+    (-5.65095215E-05) * Ta * D_Tmrt**2 + \
+    (-4.52166564E-07) * Ta**2 * D_Tmrt**2 + \
+    (2.46688878E-08) * Ta**3 * D_Tmrt**2 + \
+    (2.42674348E-10) * Ta**4 * D_Tmrt**2 + \
+    (1.54547250E-04) * va * D_Tmrt**2 + \
+    (5.24110970E-06) * Ta * va * D_Tmrt**2 + \
+    (-8.75874982E-08) * Ta**2 * va * D_Tmrt**2 + \
+    (-1.50743064E-09) * Ta**3 * va * D_Tmrt**2 + \
+    (-1.56236307E-05) * va**2 * D_Tmrt**2 + \
+    (-1.33895614E-07) * Ta * va**2 * D_Tmrt**2 + \
+    (2.49709824E-09) * Ta**2 * va**2 * D_Tmrt**2 + \
+    (6.51711721E-07) * va**3 * D_Tmrt**2 + \
+    (1.94960053E-09) * Ta * va**3 * D_Tmrt**2 + \
+    (-1.00361113E-08) * va**4 * D_Tmrt**2 + \
+    (-1.21206673E-05) * D_Tmrt**3 + \
+    (-2.18203660E-07) * Ta * D_Tmrt**3 + \
+    (7.51269482E-09) * Ta**2 * D_Tmrt**3 + \
+    (9.79063848E-11) * Ta**3 * D_Tmrt**3 + \
+    (1.25006734E-06) * va * D_Tmrt**3 + \
+    (-1.81584736E-09) * Ta * va * D_Tmrt**3 + \
+    (-3.52197671E-10) * Ta**2 * va * D_Tmrt**3 + \
+    (-3.36514630E-08) * va**2 * D_Tmrt**3 + \
+    (1.35908359E-10) * Ta * va**2 * D_Tmrt**3 + \
+    (4.17032620E-10) * va**3 * D_Tmrt**3 + \
+    (-1.30369025E-09) * D_Tmrt**4 + \
+    (4.13908461E-10) * Ta * D_Tmrt**4 + \
+    (9.22652254E-12) * Ta**2 * D_Tmrt**4 + \
+    (-5.08220384E-09) * va * D_Tmrt**4 + \
+    (-2.24730961E-11) * Ta * va * D_Tmrt**4 + \
+    (1.17139133E-10) * va**2 * D_Tmrt**4 + \
+    (6.62154879E-10) * D_Tmrt**5 + \
+    (4.03863260E-13) * Ta * D_Tmrt**5 + \
+    (1.95087203E-12) * va * D_Tmrt**5 + \
+    (-4.73602469E-12) * D_Tmrt**6 + \
+    (5.12733497E+00) * Pa + \
+    (-3.12788561E-01) * Ta * Pa + \
+    (-1.96701861E-02) * Ta**2 * Pa + \
+    (9.99690870E-04) * Ta**3 * Pa + \
+    (9.51738512E-06) * Ta**4 * Pa + \
+    (-4.66426341E-07) * Ta**5 * Pa + \
+    (5.48050612E-01) * va * Pa + \
+    (-3.30552823E-03) * Ta * va * Pa + \
+    (-1.64119440E-03) * Ta**2 * va * Pa + \
+    (-5.16670694E-06) * Ta**3 * va * Pa + \
+    (9.52692432E-07) * Ta**4 * va * Pa + \
+    (-4.29223622E-02) * va**2 * Pa + \
+    (5.00845667E-03) * Ta * va**2 * Pa + \
+    (1.00601257E-06) * Ta**2 * va**2 * Pa + \
+    (-1.81748644E-06) * Ta**3 * va**2 * Pa + \
+    (-1.25813502E-03) * va**3 * Pa + \
+    (-1.79330391E-04) * Ta * va**3 * Pa + \
+    (2.34994441E-06) * Ta**2 * va**3 * Pa + \
+    (1.29735808E-04) * va**4 * Pa + \
+    (1.29064870E-06) * Ta * va**4 * Pa + \
+    (-2.28558686E-06) * va**5 * Pa + \
+    (-3.69476348E-02) * D_Tmrt * Pa + \
+    (1.62325322E-03) * Ta * D_Tmrt * Pa + \
+    (-3.14279680E-05) * Ta**2 * D_Tmrt * Pa + \
+    (2.59835559E-06) * Ta**3 * D_Tmrt * Pa + \
+    (-4.77136523E-08) * Ta**4 * D_Tmrt * Pa + \
+    (8.64203390E-03) * va * D_Tmrt * Pa + \
+    (-6.87405181E-04) * Ta * va * D_Tmrt * Pa + \
+    (-9.13863872E-06) * Ta**2 * va * D_Tmrt * Pa + \
+    (5.15916806E-07) * Ta**3 * va * D_Tmrt * Pa + \
+    (-3.59217476E-05) * va**2 * D_Tmrt * Pa + \
+    (3.28696511E-05) * Ta * va**2 * D_Tmrt * Pa + \
+    (-7.10542454E-07) * Ta**2 * va**2 * D_Tmrt * Pa + \
+    (-1.24382300E-05) * va**3 * D_Tmrt * Pa + \
+    (-7.38584400E-09) * Ta * va**3 * D_Tmrt * Pa + \
+    (2.20609296E-07) * va**4 * D_Tmrt * Pa + \
+    (-7.32469180E-04) * D_Tmrt**2 * Pa + \
+    (-1.87381964E-05) * Ta * D_Tmrt**2 * Pa + \
+    (4.80925239E-06) * Ta**2 * D_Tmrt**2 * Pa + \
+    (-8.75492040E-08) * Ta**3 * D_Tmrt**2 * Pa + \
+    (2.77862930E-05) * va * D_Tmrt**2 * Pa + \
+    (-5.06004592E-06) * Ta * va * D_Tmrt**2 * Pa + \
+    (1.14325367E-07) * Ta**2 * va * D_Tmrt**2 * Pa + \
+    (2.53016723E-06) * va**2 * D_Tmrt**2 * Pa + \
+    (-1.72857035E-08) * Ta * va**2 * D_Tmrt**2 * Pa + \
+    (-3.95079398E-08) * va**3 * D_Tmrt**2 * Pa + \
+    (-3.59413173E-07) * D_Tmrt**3 * Pa + \
+    (7.04388046E-07) * Ta * D_Tmrt**3 * Pa + \
+    (-1.89309167E-08) * Ta**2 * D_Tmrt**3 * Pa + \
+    (-4.79768731E-07) * va * D_Tmrt**3 * Pa + \
+    (7.96079978E-09) * Ta * va * D_Tmrt**3 * Pa + \
+    (1.62897058E-09) * va**2 * D_Tmrt**3 * Pa + \
+    (3.94367674E-08) * D_Tmrt**4 * Pa + \
+    (-1.18566247E-09) * Ta * D_Tmrt**4 * Pa + \
+    (3.34678041E-10) * va * D_Tmrt**4 * Pa + \
+    (-1.15606447E-10) * D_Tmrt**5 * Pa + \
+    (-2.80626406E+00) * Pa**2 + \
+    (5.48712484E-01) * Ta * Pa**2 + \
+    (-3.99428410E-03) * Ta**2 * Pa + \
+    (-9.54009191E-04) * Ta**3 * Pa**2 + \
+    (1.93090978E-05) * Ta**4 * Pa**2 + \
+    (-3.08806365E-01) * va * Pa**2 + \
+    (1.16952364E-02) * Ta * va * Pa**2 + \
+    (4.95271903E-04) * Ta**2 * va * Pa**2 + \
+    (-1.90710882E-05) * Ta**3 * va * Pa**2 + \
+    (2.10787756E-03) * va**2 * Pa + \
+    (-6.98445738E-04) * Ta * va**2 * Pa + \
+    (2.30109073E-05) * Ta**2 * va**2 * Pa + \
+    (4.17856590E-04) * va**3 * Pa + \
+    (-1.27043871E-05) * Ta * va**3 * Pa + \
+    (-3.04620472E-06) * va**4 * Pa + \
+    (5.14507424E-02) * D_Tmrt * Pa**2 + \
+    (-4.32510997E-03) * Ta * D_Tmrt * Pa**2 + \
+    (8.99281156E-05) * Ta**2 * D_Tmrt * Pa**2 + \
+    (-7.14663943E-07) * Ta**3 * D_Tmrt * Pa**2 + \
+    (-2.66016305E-04) * va * D_Tmrt * Pa**2 + \
+    (2.63789586E-04) * Ta * va * D_Tmrt * Pa**2 + \
+    (-7.01199003E-06) * Ta**2 * va * D_Tmrt * Pa**2 + \
+    (-1.06823306E-04) * va**2 * D_Tmrt * Pa**2 + \
+    (3.61341136E-06) * Ta * va**2 * D_Tmrt * Pa**2 + \
+    (2.29748967E-07) * va**3 * D_Tmrt * Pa**2 + \
+    (3.04788893E-04) * D_Tmrt**2 * Pa**2 + \
+    (-6.42070836E-05) * Ta * D_Tmrt**2 * Pa**2 + \
+    (1.16257971E-06) * Ta**2 * D_Tmrt**2 * Pa**2 + \
+    (7.68023384E-06) * va * D_Tmrt**2 * Pa**2 + \
+    (-5.47446896E-07) * Ta * va * D_Tmrt**2 * Pa**2 + \
+    (-3.59937910E-08) * va**2 * D_Tmrt**2 * Pa**2 + \
+    (-4.36497725E-06) * D_Tmrt**3 * Pa**2 + \
+    (1.68737969E-07) * Ta * D_Tmrt**3 * Pa**2 + \
+    (2.67489271E-08) * va * D_Tmrt**3 * Pa**2 + \
+    (3.23926897E-09) * D_Tmrt**4 * Pa**2 + \
+    (-3.53874123E-02) * Pa**3 + \
+    (-2.21201190E-01) * Ta * Pa**3 + \
+    (1.55126038E-02) * Ta**2 * Pa**3 + \
+    (-2.63917279E-04) * Ta**3 * Pa**3 + \
+    (4.53433455E-02) * va * Pa**3 + \
+    (-4.32943862E-03) * Ta * va * Pa**3 + \
+    (1.45389826E-04) * Ta**2 * va * Pa**3 + \
+    (2.17508610E-04) * va**2 * Pa**3 + \
+    (-6.66724702E-05) * Ta * va**2 * Pa**3 + \
+    (3.33217140E-05) * va**3 * Pa**3 + \
+    (-2.26921615E-03) * D_Tmrt * Pa**3 + \
+    (3.80261982E-04) * Ta * D_Tmrt * Pa**3 + \
+    (-5.45314314E-09) * Ta**2 * D_Tmrt * Pa**3 + \
+    (-7.96355448E-04) * va * D_Tmrt * Pa**3 + \
+    (2.53458034E-05) * Ta * va * D_Tmrt * Pa**3 + \
+    (-6.31223658E-06) * va**2 * D_Tmrt * Pa**3 + \
+    (3.02122035E-04) * D_Tmrt**2 * Pa**3 + \
+    (-4.77403547E-06) * Ta * D_Tmrt**2 * Pa**3 + \
+    (1.73825715E-06) * va * D_Tmrt**2 * Pa**3 + \
+    (-4.09087898E-07) * D_Tmrt**3 * Pa**3 + \
+    (6.14155345E-01) * Pa**4 + \
+    (-6.16755931E-02) * Ta * Pa**4 + \
+    (1.33374846E-03) * Ta**2 * Pa**4 + \
+    (3.55375387E-03) * va * Pa**4 + \
+    (-5.13027851E-04) * Ta * va * Pa**4 + \
+    (1.02449757E-04) * va**2 * Pa**4 + \
+    (-1.48526421E-03) * D_Tmrt * Pa**4 + \
+    (-4.11469183E-05) * Ta * D_Tmrt * Pa**4 + \
+    (-6.80434415E-06) * va * D_Tmrt * Pa**4 + \
+    (-9.77675906E-06) * D_Tmrt**2 * Pa**4 + \
+    (8.82773108E-02) * Pa**5 + \
+    (-3.01859306E-03) * Ta * Pa**5 + \
+    (1.04452989E-03) * va * Pa**5 + \
+    (2.47090539E-04) * D_Tmrt * Pa**5 + \
+    (1.48348065E-03) * Pa**6
+
+    return UTCI_approx
+
+def utci_calculator(Ta, RH, Tmrt, va10m):
+    # Ta = torch.tensor(Ta, dtype=torch.float32)
+    # RH = torch.tensor(RH, dtype=torch.float32)
+    # Tmrt = torch.tensor(Tmrt, dtype=torch.float32)
+    # va10m = torch.tensor(va10m, dtype=torch.float32)
+
+    invalid_mask = (Ta <= -999) | (RH <= -999) | (va10m <= -999) | (Tmrt <= -999)
+    valid_mask = ~invalid_mask
+
+    tk = Ta[valid_mask] + 273.15  # air temp in K
+
+    # saturation vapour pressure (es)
+    g = torch.tensor([-2.8365744E3, -6.028076559E3, 1.954263612E1, -2.737830188E-2,
+                      1.6261698E-5, 7.0229056E-10, -1.8680009E-13, 2.7150305], dtype=torch.float32)
+
+    es = g[7] * torch.log(tk)
+    for i in range(0, 7):
+        es = es + g[i] * tk ** (i + 1 - 3.)
+
+    es = torch.exp(es) * 0.01
+
+    ehPa = es * RH[valid_mask] / 100.
+
+    D_Tmrt = Tmrt[valid_mask] - Ta[valid_mask]
+    Pa = ehPa / 10.0  # use vapour pressure in kPa
+    va = va10m[valid_mask]
+
+    # Calculate 6th order polynomial as approximation
+    UTCI_approx = torch.full_like(Ta, -999, dtype=torch.float32)
+    UTCI_approx[valid_mask] = utci_polynomial(D_Tmrt, Ta[valid_mask], va, Pa)
+
+    return UTCI_approx

solweig_gpu-0.1.0/solweig_gpu/cli.py

@@ -0,0 +1,58 @@
+import argparse
+from .solweig_gpu import thermal_comfort
+
+def main():
+    parser = argparse.ArgumentParser(description="Run SOLWEIG with GPU acceleration.")
+    parser.add_argument('--base_path', required=True, help='Base directory containing input data')
+    parser.add_argument('--date', required=True, help='Date for which thermal comfort is computed (e.g., 2021-07-01)')
+    parser.add_argument('--building_dsm', default='Building_DSM.tif', help='Filename of the building DSM raster')
+    parser.add_argument('--dem', default='DEM.tif', help='Filename of the DEM raster')
+    parser.add_argument('--trees', default='Trees.tif', help='Filename of the trees raster')
+    parser.add_argument('--tile_size', type=int, default=3600, help='Tile size for GPU processing (e.g., 100 to 4000)')
+    parser.add_argument('--use_own_met', type=bool, default=True, help='Set to True if using your own meteorological file')
+    parser.add_argument('--own_metfile', default = None, help='Path to your own meteorological file')
+    parser.add_argument('--data_source_type', default=None, help='Specify source of meteorological data (e.g., ERA5 or WRF), if not providing met file')
+    parser.add_argument('--data_folder', default = None,help='folder for meteorological data source, if it is ERA5 to WRF')
+    parser.add_argument(
+    '--start',
+    default=None,
+    help="""Start time of the input file ('2020-08-12 00:00:00')""")
+    parser.add_argument(
+    '--end',
+    default=None,
+    help="""Start time of the input file ('2020-08-12 00:00:00')""")
+    parser.add_argument('--save_tmrt', action='store_true', help='Flag to save mean radiant temperature')
+    parser.add_argument('--save_svf', action='store_true', help='Flag to save sky view factor')
+    parser.add_argument('--save_kup', action='store_true', help='Flag to save upward shortwave radiation')
+    parser.add_argument('--save_kdown', action='store_true', help='Flag to save downward shortwave radiation')
+    parser.add_argument('--save_lup', action='store_true', help='Flag to save upward longwave radiation')
+    parser.add_argument('--save_ldown', action='store_true', help='Flag to save downward longwave radiation')
+    parser.add_argument('--save_shadow', action='store_true', help='Flag to save shadow map')
+
+    args = parser.parse_args()
+
+    thermal_comfort(
+        base_path=args.base_path,
+        selected_date_str=args.date,
+        building_dsm_filename=args.building_dsm,
+        dem_filename=args.dem,
+        trees_filename=args.trees,
+        tile_size=args.tile_size,
+        use_own_met=args.use_own_met,
+        own_met_file=args.own_metfile,
+        start_time=args.start,
+        end_time=args.end,
+        data_source_type=args.data_source_type,
+        data_folder=args.data_folder,
+        save_tmrt=args.save_tmrt,
+        save_svf=args.save_svf,
+        save_kup=args.save_kup,
+        save_kdown=args.save_kdown,
+        save_lup=args.save_lup,
+        save_ldown=args.save_ldown,
+        save_shadow=args.save_shadow
+    )
+
+if __name__ == '__main__':
+    main()
+