mirror of
https://gitlab.com/pamhyr/pamhyr2.git
synced 2026-10-30 03:20:41 +01:00
Compare commits
325
Commits
test_sound
...
v2.0.0rc0
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
64a26e987e | ||
|
|
eb233847de | ||
|
|
7bb810d427 | ||
|
|
e49724072b | ||
|
|
3664ba93e9 | ||
|
|
a9903e987b | ||
|
|
f9a88a211c | ||
|
|
32535af3e6 | ||
|
|
fa8948c1b6 | ||
|
|
2d5455c09d | ||
|
|
83680bbe57 | ||
|
|
1d1a00a0ff | ||
|
|
ffd95b9fdf | ||
|
|
32cbc6bc3e | ||
|
|
7d7a05c04c | ||
|
|
6ac268b668 | ||
|
|
19c72892a8 | ||
|
|
9717968e08 | ||
|
|
e19c3229bd | ||
|
|
b3c6f1031b | ||
|
|
c2d93b4a4a | ||
|
|
76430322dd | ||
|
|
31ae0ab8e0 | ||
|
|
8182d434ec | ||
|
|
da78d0cdb1 | ||
|
|
850db144ac | ||
|
|
0f6709075f | ||
|
|
e86b84b0b9 | ||
|
|
e08d212986 | ||
|
|
7b88d955e5 | ||
|
|
d1c34fb437 | ||
|
|
611f9cdfa3 | ||
|
|
bd2fc22e96 | ||
|
|
efa16837d8 | ||
|
|
db7aad26fd | ||
|
|
229b3da2d1 | ||
|
|
48d6728e1e | ||
|
|
c51c06e6ec | ||
|
|
a425c9a6ac | ||
|
|
e79ae8ebf5 | ||
|
|
e3020d15ba | ||
|
|
83e0ecbdc7 | ||
|
|
3087023caa | ||
|
|
7e080767da | ||
|
|
44c3a91d72 | ||
|
|
26dc7d8761 | ||
|
|
e573fa4197 | ||
|
|
7aa0cf4361 | ||
|
|
c3faf45f75 | ||
|
|
fb8b5aa77f | ||
|
|
8183a284af | ||
|
|
09997a0074 | ||
|
|
c44597dfbd | ||
|
|
b1632eaea5 | ||
|
|
a2461af084 | ||
|
|
5c366839f3 | ||
|
|
311df46a2c | ||
|
|
a235f61044 | ||
|
|
0b80b86d31 | ||
|
|
0c7ede5f43 | ||
|
|
445c1ac7ba | ||
|
|
c687e05d8f | ||
|
|
5fb6fecc24 | ||
|
|
3dfbdc328e | ||
|
|
97d87f2455 | ||
|
|
a78be47e2b | ||
|
|
eed2ce4243 | ||
|
|
78ba7b70b4 | ||
|
|
f86cd4602c | ||
|
|
ee28afac86 | ||
|
|
81ff6372e9 | ||
|
|
61156f1c3e | ||
|
|
e26b170b05 | ||
|
|
ddbaa0fb94 | ||
|
|
64d2b2678a | ||
|
|
3a3b5ea2dd | ||
|
|
7cb3ac8a80 | ||
|
|
190ca5d171 | ||
|
|
5d8ec51417 | ||
|
|
d922436ada | ||
|
|
4537a829af | ||
|
|
363b9c7132 | ||
|
|
f9d83ac68c | ||
|
|
63b5cbaba7 | ||
|
|
1989914aef | ||
|
|
2c0b155baf | ||
|
|
4d20c2457a | ||
|
|
2dd9003721 | ||
|
|
94ceb44fd4 | ||
|
|
ab50d2adf4 | ||
|
|
0eb979256f | ||
|
|
5ce94a63d3 | ||
|
|
42ea50d62e | ||
|
|
0df4cf0fd5 | ||
|
|
ccf3da114b | ||
|
|
348417977d | ||
|
|
914a1f8866 | ||
|
|
cbbbeeac0b | ||
|
|
c9ea726bfb | ||
|
|
fa88e05d91 | ||
|
|
a6915020cb | ||
|
|
ea2287215e | ||
|
|
e940dead70 | ||
|
|
b6a6d54a22 | ||
|
|
11a95cb391 | ||
|
|
084c22836b | ||
|
|
726b81bf12 | ||
|
|
7c3de520d9 | ||
|
|
b450796df5 | ||
|
|
b1daadf9f2 | ||
|
|
c38e3ee446 | ||
|
|
606b9919e3 | ||
|
|
e3de7e7100 | ||
|
|
92fae0e13f | ||
|
|
a8b260ad38 | ||
|
|
faff877c80 | ||
|
|
0a0eeb3c3f | ||
|
|
cdcbbc2cf4 | ||
|
|
7565d8ee95 | ||
|
|
11e1302e05 | ||
|
|
99cb2883a1 | ||
|
|
e61dede7f8 | ||
|
|
cb4dfab482 | ||
|
|
377c508038 | ||
|
|
30e34b6847 | ||
|
|
102702e1df | ||
|
|
c014f0ef34 | ||
|
|
e615a6b9d4 | ||
|
|
f1dca727c6 | ||
|
|
2db1f4c7f7 | ||
|
|
b3d631943c | ||
|
|
b6895ae99c | ||
|
|
9f26a7323c | ||
|
|
2a7e322ee5 | ||
|
|
13a44264ac | ||
|
|
6165719829 | ||
|
|
51dd6d27c5 | ||
|
|
cea4dcd9cb | ||
|
|
e1c1f411c3 | ||
|
|
4550fc95d7 | ||
|
|
1f97af9c03 | ||
|
|
df9e1ec42d | ||
|
|
ac79fae1fb | ||
|
|
fe39f07981 | ||
|
|
b578e0c50c | ||
|
|
4408a8f368 | ||
|
|
d3e922af6e | ||
|
|
8adf878584 | ||
|
|
aa6973813f | ||
|
|
e21a64fcad | ||
|
|
8b6308d3c9 | ||
|
|
bc3ea9c4b9 | ||
|
|
04ea55e21a | ||
|
|
a68270ff67 | ||
|
|
c498ed2d96 | ||
|
|
e6d059bc19 | ||
|
|
cc500ceccf | ||
|
|
7a91969f41 | ||
|
|
6bd3605d10 | ||
|
|
6c6da90952 | ||
|
|
539b63f765 | ||
|
|
e755459004 | ||
|
|
a8f95e2386 | ||
|
|
55b200a48e | ||
|
|
6b8d7c2df2 | ||
|
|
eb7c4895ca | ||
|
|
a060dbd461 | ||
|
|
9379f7a27d | ||
|
|
19fedc2278 | ||
|
|
260efc17a9 | ||
|
|
a4026d6386 | ||
|
|
d6f7755d12 | ||
|
|
00b89635f3 | ||
|
|
f54dc07432 | ||
|
|
3bbc55b4be | ||
|
|
b58d3e005d | ||
|
|
870bb41ff7 | ||
|
|
0701531d37 | ||
|
|
43eb486700 | ||
|
|
82eeb9a478 | ||
|
|
432c263cdf | ||
|
|
c2d28b0110 | ||
|
|
23361d142e | ||
|
|
63836527c7 | ||
|
|
4f6e8249e0 | ||
|
|
ab2f775b95 | ||
|
|
bb0e176956 | ||
|
|
c0d4aced5e | ||
|
|
4ed865759a | ||
|
|
70e09d3b3a | ||
|
|
2bcfa4d121 | ||
|
|
5b497d61f0 | ||
|
|
242df7f648 | ||
|
|
421e6797f6 | ||
|
|
0886956ec8 | ||
|
|
9670bde56f | ||
|
|
3277cf551b | ||
|
|
ad5cc831a3 | ||
|
|
b00cca953e | ||
|
|
1f35020e3f | ||
|
|
f0994dd801 | ||
|
|
d002eebd49 | ||
|
|
e3ffac9ce0 | ||
|
|
e8a4d5eeed | ||
|
|
fc090dadce | ||
|
|
25e4786484 | ||
|
|
81e85ab488 | ||
|
|
201b1099bd | ||
|
|
b6054dd633 | ||
|
|
912843be25 | ||
|
|
bf49a7fc87 | ||
|
|
d77be00bcc | ||
|
|
ff0fa8ae8e | ||
|
|
92f4793181 | ||
|
|
5eaac204d5 | ||
|
|
7ef41ad886 | ||
|
|
ada396f26d | ||
|
|
674247db09 | ||
|
|
5fecd29666 | ||
|
|
71e1324a0f | ||
|
|
07cae542b0 | ||
|
|
b693f67b99 | ||
|
|
18a3d51d75 | ||
|
|
118a39bd5c | ||
|
|
5357eb9735 | ||
|
|
0946212c4b | ||
|
|
d6ab44d18e | ||
|
|
1db1a543c7 | ||
|
|
2009e8223b | ||
|
|
bdaa38ea5c | ||
|
|
713732b1dd | ||
|
|
f053539cdd | ||
|
|
fc6916bb3e | ||
|
|
97c26cb4fb | ||
|
|
10c98bb96e | ||
|
|
3e92a0c40d | ||
|
|
7c2d9891d0 | ||
|
|
3c3ac1f1df | ||
|
|
80af278a0f | ||
|
|
f3c461cbd9 | ||
|
|
f254eaee25 | ||
|
|
012408d19e | ||
|
|
f7e563e71a | ||
|
|
81c6d39c1a | ||
|
|
5adb0bdee5 | ||
|
|
d682443e1d | ||
|
|
64c99d46d4 | ||
|
|
0f0defb36f | ||
|
|
1e1ee67d4b | ||
|
|
5bc514c565 | ||
|
|
c065b19ef8 | ||
|
|
1da3461dfe | ||
|
|
b20a4c3c3d | ||
|
|
d66e75c183 | ||
|
|
31b4c47379 | ||
|
|
71bba499e0 | ||
|
|
7a08fa42d9 | ||
|
|
18a008462a | ||
|
|
d81da59398 | ||
|
|
3ab1eabe75 | ||
|
|
cf4f860db8 | ||
|
|
c68e3337ef | ||
|
|
dd5eecb1e2 | ||
|
|
db1ae613d6 | ||
|
|
a800816cda | ||
|
|
d4ef06c7c1 | ||
|
|
95d1d231c9 | ||
|
|
10b278e26f | ||
|
|
0550eef2eb | ||
|
|
73eb9919eb | ||
|
|
9116009b0b | ||
|
|
4e4ceb5074 | ||
|
|
d79e2ff187 | ||
|
|
d4a2594239 | ||
|
|
b666162bdf | ||
|
|
d705accaed | ||
|
|
d93917b081 | ||
|
|
ceff5f3083 | ||
|
|
97762756f8 | ||
|
|
2c49e991b4 | ||
|
|
c2a8f38ef5 | ||
|
|
f8a41fce08 | ||
|
|
624ae826eb | ||
|
|
d47dc0687e | ||
|
|
117e5222e4 | ||
|
|
b1c7a77f37 | ||
|
|
7b833390f1 | ||
|
|
ae08642116 | ||
|
|
c1905aa7e1 | ||
|
|
97ece018aa | ||
|
|
7f0102a881 | ||
|
|
786923bdbf | ||
|
|
deb9b2069f | ||
|
|
9308a73e8e | ||
|
|
869e116ad0 | ||
|
|
a2f3d22001 | ||
|
|
2e360943b2 | ||
|
|
8976f054c7 | ||
|
|
7bce725c63 | ||
|
|
5c83d67865 | ||
|
|
6e52b1681e | ||
|
|
f27b2cc586 | ||
|
|
1ed5d69bf4 | ||
|
|
5bb6cc40fe | ||
|
|
4c0a12dcf9 | ||
|
|
a308af41e0 | ||
|
|
4cf4015579 | ||
|
|
291b97ac9b | ||
|
|
b04e367e72 | ||
|
|
81d58122d6 | ||
|
|
874f592cf4 | ||
|
|
9b0bdd1e63 | ||
|
|
14549330b9 | ||
|
|
2487bec6d6 | ||
|
|
4e1acfecdc | ||
|
|
c63c776989 | ||
|
|
44a8be6be3 | ||
|
|
a5a64f2080 | ||
|
|
16ee5a90e4 | ||
|
|
7ca6e69526 | ||
|
|
a31e2127e7 | ||
|
|
a29df8f904 | ||
|
|
97ebf9324a | ||
|
|
0112ad63d9 | ||
|
|
bcbb440ad4 |
No files matched your search
+46
-8
@@ -38,6 +38,7 @@ variables:
|
|||||||
variables:
|
variables:
|
||||||
MAGE_8_VERSION: "permalink/latest"
|
MAGE_8_VERSION: "permalink/latest"
|
||||||
ADISTS_VERSION: "permalink/latest"
|
ADISTS_VERSION: "permalink/latest"
|
||||||
|
RUBAR_VERSION: "permalink/latest"
|
||||||
|
|
||||||
dl-mage8-doc:
|
dl-mage8-doc:
|
||||||
stage: downloads
|
stage: downloads
|
||||||
@@ -111,6 +112,36 @@ dl-adists-windows:
|
|||||||
paths:
|
paths:
|
||||||
- adists-windows/adists.exe
|
- adists-windows/adists.exe
|
||||||
|
|
||||||
|
dl-rubar-linux:
|
||||||
|
stage: downloads
|
||||||
|
tags:
|
||||||
|
- linux
|
||||||
|
rules:
|
||||||
|
- if: $CI_COMMIT_BRANCH == 'ci-test' || $CI_COMMIT_BRANCH == 'master' || $CI_COMMIT_TAG
|
||||||
|
script:
|
||||||
|
- mkdir -p rubar-linux
|
||||||
|
- cd rubar-linux
|
||||||
|
- curl -L -o rubar.tgz https://forge.inrae.fr/river-hydraulics/rubarbe/-/releases/$RUBAR_VERSION/downloads/packages/rubar_linux.tgz
|
||||||
|
- tar xf rubar.tgz
|
||||||
|
artifacts:
|
||||||
|
paths:
|
||||||
|
- rubar-linux/rubarbe
|
||||||
|
|
||||||
|
dl-rubar-windows:
|
||||||
|
stage: downloads
|
||||||
|
tags:
|
||||||
|
- linux
|
||||||
|
rules:
|
||||||
|
- if: $CI_COMMIT_BRANCH == 'ci-test' || $CI_COMMIT_BRANCH == 'master' || $CI_COMMIT_TAG
|
||||||
|
script:
|
||||||
|
- mkdir -p rubar-windows
|
||||||
|
- cd rubar-windows
|
||||||
|
- curl -L -o rubar.tgz https://forge.inrae.fr/river-hydraulics/rubarbe/-/releases/$RUBAR_VERSION/downloads/packages/rubar_windows.tgz
|
||||||
|
- tar xf rubar.tgz
|
||||||
|
artifacts:
|
||||||
|
paths:
|
||||||
|
- rubar-windows/rubarbe.exe
|
||||||
|
|
||||||
#############
|
#############
|
||||||
# CONFIGURE #
|
# CONFIGURE #
|
||||||
#############
|
#############
|
||||||
@@ -149,13 +180,13 @@ unittest:
|
|||||||
- job: set-version
|
- job: set-version
|
||||||
artifacts: true
|
artifacts: true
|
||||||
script:
|
script:
|
||||||
- python3 -m venv venv
|
- python3.12 -m venv venv
|
||||||
- . venv/bin/activate
|
- . venv/bin/activate
|
||||||
- pip3 install -U pip
|
- pip3 install -U pip
|
||||||
- pip3 install -r ./full-requirements.txt
|
- pip3 install -r ./full-requirements.txt
|
||||||
- pip3 install -U -r ./full-requirements.txt
|
- pip3 install -U -r ./full-requirements.txt
|
||||||
- cd src
|
- cd src
|
||||||
- python3 -m unittest discover -t .
|
- python3.12 -m unittest discover -t .
|
||||||
|
|
||||||
test-pep8:
|
test-pep8:
|
||||||
stage: test
|
stage: test
|
||||||
@@ -167,7 +198,7 @@ test-pep8:
|
|||||||
- job: unittest
|
- job: unittest
|
||||||
script:
|
script:
|
||||||
# Setup virtual env
|
# Setup virtual env
|
||||||
- python3 -m venv venv
|
- python3.12 -m venv venv
|
||||||
- . venv/bin/activate
|
- . venv/bin/activate
|
||||||
- pip3 install -U pip
|
- pip3 install -U pip
|
||||||
- pip3 install -r ./requirements.txt
|
- pip3 install -r ./requirements.txt
|
||||||
@@ -245,7 +276,7 @@ build-developers-doc:
|
|||||||
- ./build.sh
|
- ./build.sh
|
||||||
# Create html documentation
|
# Create html documentation
|
||||||
- cd ../../src
|
- cd ../../src
|
||||||
- python3 -m venv doc-venv
|
- python3.12 -m venv doc-venv
|
||||||
- . doc-venv/bin/activate
|
- . doc-venv/bin/activate
|
||||||
- pip3 install -U pip
|
- pip3 install -U pip
|
||||||
- pip3 install -r ../full-requirements.txt
|
- pip3 install -r ../full-requirements.txt
|
||||||
@@ -273,6 +304,8 @@ build-linux:
|
|||||||
needs:
|
needs:
|
||||||
- job: dl-adists-linux
|
- job: dl-adists-linux
|
||||||
artifacts: true
|
artifacts: true
|
||||||
|
- job: dl-rubar-linux
|
||||||
|
artifacts: true
|
||||||
- job: dl-mage8-linux
|
- job: dl-mage8-linux
|
||||||
artifacts: true
|
artifacts: true
|
||||||
- job: dl-mage8-doc
|
- job: dl-mage8-doc
|
||||||
@@ -293,13 +326,13 @@ build-linux:
|
|||||||
- mkdir -p linux
|
- mkdir -p linux
|
||||||
- cd linux
|
- cd linux
|
||||||
# Setup virtual env
|
# Setup virtual env
|
||||||
- python3 -m venv linux-venv
|
- python3.12 -m venv linux-venv
|
||||||
- . linux-venv/bin/activate
|
- . linux-venv/bin/activate
|
||||||
- pip3 install -U pip
|
- pip3 install -U pip
|
||||||
- pip3 install -r ../requirements.txt
|
- pip3 install -r ../requirements.txt
|
||||||
- pip3 install -U -r ../requirements.txt
|
- pip3 install -U -r ../requirements.txt
|
||||||
# Run Pyinstaller
|
# Run Pyinstaller
|
||||||
- pyinstaller -y $HIDDEN_IMPORTS --paths linux-venv/lib/python3.8/site-packages ../src/pamhyr.py
|
- pyinstaller -y $HIDDEN_IMPORTS --paths linux-venv/lib/python3.12/site-packages ../src/pamhyr.py
|
||||||
# Create directory
|
# Create directory
|
||||||
- mkdir -p pamhyr
|
- mkdir -p pamhyr
|
||||||
- mkdir -p pamhyr/_internal
|
- mkdir -p pamhyr/_internal
|
||||||
@@ -314,6 +347,9 @@ build-linux:
|
|||||||
# Copy adists
|
# Copy adists
|
||||||
- mkdir -p pamhyr/adists
|
- mkdir -p pamhyr/adists
|
||||||
- cp -v ../adists-linux/* pamhyr/adists/
|
- cp -v ../adists-linux/* pamhyr/adists/
|
||||||
|
# Copy rubar
|
||||||
|
- mkdir -p pamhyr/rubar
|
||||||
|
- cp -v ../rubar-linux/* pamhyr/rubar/
|
||||||
# Copy Pamhyr
|
# Copy Pamhyr
|
||||||
- cp -r dist/pamhyr/* pamhyr/
|
- cp -r dist/pamhyr/* pamhyr/
|
||||||
# Pamhyr script to force x11
|
# Pamhyr script to force x11
|
||||||
@@ -379,6 +415,8 @@ build-windows:
|
|||||||
artifacts: true
|
artifacts: true
|
||||||
- job: dl-adists-windows
|
- job: dl-adists-windows
|
||||||
artifacts: true
|
artifacts: true
|
||||||
|
- job: dl-rubar-windows
|
||||||
|
artifacts: true
|
||||||
- job: dl-mage8-doc
|
- job: dl-mage8-doc
|
||||||
artifacts: true
|
artifacts: true
|
||||||
- job: set-version
|
- job: set-version
|
||||||
@@ -600,7 +638,7 @@ shadow-release:
|
|||||||
- job: pkg-hash
|
- job: pkg-hash
|
||||||
artifacts: true
|
artifacts: true
|
||||||
rules:
|
rules:
|
||||||
- if: $CI_COMMIT_BRANCH != 'master' && $CI_COMMIT_TAG
|
- if: $CI_COMMIT_TAG && $CI_COMMIT_TAG !~ /^v/
|
||||||
artifacts:
|
artifacts:
|
||||||
paths:
|
paths:
|
||||||
- linux/pamhyr-gnulinux.tar.xz
|
- linux/pamhyr-gnulinux.tar.xz
|
||||||
@@ -634,7 +672,7 @@ tag-release:
|
|||||||
- job: pkg-hash
|
- job: pkg-hash
|
||||||
artifacts: true
|
artifacts: true
|
||||||
rules:
|
rules:
|
||||||
- if: $CI_COMMIT_BRANCH == 'master' && $CI_COMMIT_TAG
|
- if: $CI_COMMIT_TAG =~ /^v/
|
||||||
artifacts:
|
artifacts:
|
||||||
paths:
|
paths:
|
||||||
- linux/pamhyr-gnulinux.tar.xz
|
- linux/pamhyr-gnulinux.tar.xz
|
||||||
|
|||||||
@@ -1,3 +1,4 @@
|
|||||||
Sylvain COULIBALY, INRAE, 2020 - 2022
|
Sylvain COULIBALY, INRAE, 2020 - 2022
|
||||||
Théophile TERRAZ, INRAE, 2022 - 2024
|
Théophile TERRAZ, INRAE, 2022 - 2024
|
||||||
Pierre-Antoine ROUBY, INRAE, 2023 - 2025
|
Pierre-Antoine ROUBY, INRAE, 2023 - 2025
|
||||||
|
Dylan JEANNIN, INRAE, 2026
|
||||||
@@ -3,7 +3,7 @@
|
|||||||
Pamhyr is a free and open source graphical user interface for 1D hydro-sedimentary
|
Pamhyr is a free and open source graphical user interface for 1D hydro-sedimentary
|
||||||
modelling of rivers.
|
modelling of rivers.
|
||||||
|
|
||||||

|

|
||||||
|
|
||||||
## Features
|
## Features
|
||||||
|
|
||||||
@@ -21,9 +21,9 @@ modelling of rivers.
|
|||||||
+ Create and edit study alternative scenario tree
|
+ Create and edit study alternative scenario tree
|
||||||
|
|
||||||
Let see the
|
Let see the
|
||||||
[documentation](https://gitlab.irstea.fr/theophile.terraz/pamhyr/-/wikis/home)
|
[documentation](https://gitlab.com/pamhyr/pamhyr2/-/wikis/home)
|
||||||
([:fr:](https://gitlab.irstea.fr/theophile.terraz/pamhyr/-/wikis/home-fr),
|
([:fr:](https://gitlab.com/pamhyr/pamhyr2/-/wikis/home-fr),
|
||||||
[:gb:](https://gitlab.irstea.fr/theophile.terraz/pamhyr/-/wikis/home-en))
|
[:gb:](https://gitlab.com/pamhyr/pamhyr2/-/wikis/home-en))
|
||||||
for more details.
|
for more details.
|
||||||
|
|
||||||
## Install
|
## Install
|
||||||
@@ -31,11 +31,11 @@ for more details.
|
|||||||
### GNU/Linux
|
### GNU/Linux
|
||||||
|
|
||||||
See documentation:
|
See documentation:
|
||||||
- [French :fr:](https://gitlab.irstea.fr/theophile.terraz/pamhyr/-/wikis/home/fr/Install%20on%20GNULinux)
|
- [French :fr:](https://gitlab.com/pamhyr/pamhyr2/-/wikis/home/fr/Install%20on%20GNULinux)
|
||||||
- [English :gb:](https://gitlab.irstea.fr/theophile.terraz/pamhyr/-/wikis/home/en/Install%20on%20GNULinux)
|
- [English :gb:](https://gitlab.com/pamhyr/pamhyr2/-/wikis/home/en/Install%20on%20GNULinux)
|
||||||
|
|
||||||
### Windows
|
### Windows
|
||||||
|
|
||||||
See documentation:
|
See documentation:
|
||||||
- [French :fr:](https://gitlab.irstea.fr/theophile.terraz/pamhyr/-/wikis/home/fr/Install%20on%20Windows)
|
- [French :fr:](https://gitlab.com/pamhyr/pamhyr2/-/wikis/home/fr/Install%20on%20Windows)
|
||||||
- [English :gb:](https://gitlab.irstea.fr/theophile.terraz/pamhyr/-/wikis/home/en/Install%20on%20Windows)
|
- [English :gb:](https://gitlab.com/pamhyr/pamhyr2/-/wikis/home/en/Install%20on%20Windows)
|
||||||
Binary file not shown.
|
After Width: | Height: | Size: 70 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 72 KiB |
Binary file not shown.
|
After Width: | Height: | Size: 42 KiB |
@@ -157,12 +157,12 @@
|
|||||||
%%
|
%%
|
||||||
|
|
||||||
%% FIXME: Delete me for the first realease
|
%% FIXME: Delete me for the first realease
|
||||||
\usepackage{draftwatermark}
|
%% \usepackage{draftwatermark}
|
||||||
\SetWatermarkLightness{0.8}
|
%% \SetWatermarkLightness{0.8}
|
||||||
\SetWatermarkAngle{25}
|
%% \SetWatermarkAngle{25}
|
||||||
\SetWatermarkScale{3}
|
%% \SetWatermarkScale{3}
|
||||||
\SetWatermarkFontSize{1cm}
|
%% \SetWatermarkFontSize{1cm}
|
||||||
\SetWatermarkText{Work in progress}
|
%% \SetWatermarkText{Work in progress}
|
||||||
|
|
||||||
%% Icons
|
%% Icons
|
||||||
|
|
||||||
|
|||||||
Binary file not shown.
@@ -1,10 +1,10 @@
|
|||||||
PyQt5==5.15.4
|
PyQt5==5.15.11
|
||||||
PyQt5-Qt5==5.15.2
|
PyQt5-Qt5==5.15.2
|
||||||
PyQt5-sip==12.12.2
|
PyQt5-sip==12.18.0
|
||||||
PyQtWebEngine==5.15.6
|
PyQtWebEngine==5.15.6
|
||||||
QScintilla>=2.14.1
|
QScintilla>=2.14.1
|
||||||
pyqtgraph>=0.12.1
|
pyqtgraph>=0.12.1
|
||||||
matplotlib>=3.7.1
|
matplotlib==3.8.3
|
||||||
numpy>=1.24.2
|
numpy>=1.24.2
|
||||||
colorama>=0.4.3
|
colorama>=0.4.3
|
||||||
pyinstaller>=5.11.0
|
pyinstaller>=5.11.0
|
||||||
|
|||||||
@@ -46,6 +46,10 @@ copy /y ..\mage8-windows\mage_extraire.exe pamhyr\mage8\
|
|||||||
copy /y ..\mage8-windows\mailleurTT.exe pamhyr\mage8\
|
copy /y ..\mage8-windows\mailleurTT.exe pamhyr\mage8\
|
||||||
copy /y ..\mage8-windows\libbief.dll pamhyr\mage8\
|
copy /y ..\mage8-windows\libbief.dll pamhyr\mage8\
|
||||||
|
|
||||||
|
rem rubar
|
||||||
|
mkdir pamhyr\rubar
|
||||||
|
copy /y ..\rubar-windows\rubarbe.exe pamhyr\rubar\
|
||||||
|
|
||||||
rem adists
|
rem adists
|
||||||
mkdir pamhyr\adists
|
mkdir pamhyr\adists
|
||||||
copy /y ..\adists-windows\adists.exe pamhyr\adists\
|
copy /y ..\adists-windows\adists.exe pamhyr\adists\
|
||||||
|
|||||||
+4
-3
@@ -1,10 +1,10 @@
|
|||||||
PyQt5==5.15.4
|
PyQt5==5.15.11
|
||||||
PyQt5-Qt5==5.15.2
|
PyQt5-Qt5==5.15.2
|
||||||
PyQt5-sip==12.12.2
|
PyQt5-sip==12.18.0
|
||||||
#PyQtWebEngine==5.15.6
|
#PyQtWebEngine==5.15.6
|
||||||
QScintilla>=2.14.1
|
QScintilla>=2.14.1
|
||||||
pyqtgraph>=0.12.1
|
pyqtgraph>=0.12.1
|
||||||
matplotlib>=3.7.1
|
matplotlib==3.8.3
|
||||||
numpy>=1.24.2
|
numpy>=1.24.2
|
||||||
colorama>=0.4.3
|
colorama>=0.4.3
|
||||||
pyinstaller>=5.11.0
|
pyinstaller>=5.11.0
|
||||||
@@ -14,3 +14,4 @@ platformdirs>=4.2.0
|
|||||||
pyshp>=2.3.1
|
pyshp>=2.3.1
|
||||||
rasterio==1.3.11
|
rasterio==1.3.11
|
||||||
#fortranformat==2.0.3
|
#fortranformat==2.0.3
|
||||||
|
|
||||||
+83
-65
@@ -40,20 +40,24 @@ class InternalMeshing(AMeshingTool):
|
|||||||
step: float = 50,
|
step: float = 50,
|
||||||
limites=[-1, -1],
|
limites=[-1, -1],
|
||||||
linear: bool = True):
|
linear: bool = True):
|
||||||
if reach is None or len(reach.profiles) == 0:
|
|
||||||
return reach
|
|
||||||
|
|
||||||
if limites[0] > limites[1]:
|
# pre process
|
||||||
lim = limites[1]
|
|
||||||
limites[1] = limites[0]
|
|
||||||
limites[0] = lim
|
|
||||||
elif limites[0] == limites[1]:
|
|
||||||
return reach
|
|
||||||
|
|
||||||
# we never modify original profiles, we work on copies
|
profiles = self.get_profiles(reach, limites)
|
||||||
m_profiles = self.st_to_m(reach, limites)
|
if profiles is None:
|
||||||
new_profiles = self.interpolate_transversal_step(m_profiles,
|
return []
|
||||||
step)
|
|
||||||
|
for p in profiles:
|
||||||
|
if len(p) < 1:
|
||||||
|
return []
|
||||||
|
|
||||||
|
guide_list = self.get_guide_list(reach, profiles)
|
||||||
|
if guide_list is None:
|
||||||
|
return []
|
||||||
|
|
||||||
|
# we never modify original profiles, we work on the copies
|
||||||
|
m_profiles = self.st_to_m(profiles, guide_list)
|
||||||
|
new_profiles = self.interpolate_transversal_step(m_profiles, step)
|
||||||
|
|
||||||
for new_profiles2 in new_profiles:
|
for new_profiles2 in new_profiles:
|
||||||
for p in new_profiles2:
|
for p in new_profiles2:
|
||||||
@@ -61,52 +65,32 @@ class InternalMeshing(AMeshingTool):
|
|||||||
|
|
||||||
return new_profiles
|
return new_profiles
|
||||||
|
|
||||||
def st_to_m(self, reach, limites):
|
def st_to_m(self, profiles, guide_list):
|
||||||
|
guide_list = ["un"] + guide_list + ["np"]
|
||||||
gl = reach.compute_guidelines()
|
max_values = [0] * (len(guide_list) - 1)
|
||||||
profiles = [reach.profile(i).copy()
|
max_values_index = [0] * (len(guide_list) - 1)
|
||||||
for i in range(limites[0], limites[1]+1)
|
|
||||||
]
|
|
||||||
|
|
||||||
# we make sure that the lines are in the left-to-right order
|
|
||||||
guide_list = [
|
|
||||||
x.name
|
|
||||||
for x in profiles[0].named_points()
|
|
||||||
if x.name in gl[0]
|
|
||||||
]
|
|
||||||
|
|
||||||
gl1 = [""] + guide_list
|
|
||||||
gl2 = guide_list + [""]
|
|
||||||
max_values = [0] * (len(guide_list) + 1)
|
|
||||||
max_values_index = [0] * (len(guide_list) + 1)
|
|
||||||
# we work between each couple of guidelines
|
# we work between each couple of guidelines
|
||||||
for j, p in enumerate(profiles):
|
for j, p in enumerate(profiles):
|
||||||
index1 = 0
|
|
||||||
index2 = p.named_point_index(guide_list[0])
|
|
||||||
if index2 - index1 > max_values[0]:
|
|
||||||
max_values[0] = index2 - index1
|
|
||||||
max_values_index[0] = j
|
|
||||||
for i in range(len(guide_list) - 1):
|
for i in range(len(guide_list) - 1):
|
||||||
index1 = p.named_point_index(guide_list[i])
|
index1 = p.named_point_index(guide_list[i])
|
||||||
index2 = p.named_point_index(guide_list[i+1])
|
index2 = p.named_point_index(guide_list[i+1])
|
||||||
if index2 - index1 > max_values[i+1]:
|
if index2 - index1 > max_values[i]:
|
||||||
max_values[i+1] = index2 - index1
|
max_values[i] = index2 - index1
|
||||||
max_values_index[i+1] = j
|
max_values_index[i] = j
|
||||||
index1 = p.named_point_index(guide_list[-1])
|
|
||||||
index2 = len(p) - 1
|
|
||||||
if index2 - index1 > max_values[-1]:
|
|
||||||
max_values[-1] = index2 - index1
|
|
||||||
max_values_index[-1] = j
|
|
||||||
|
|
||||||
for i in range(len(max_values)):
|
for i in range(len(max_values)):
|
||||||
for isect in range(max_values_index[i], len(profiles)-1):
|
for isect in range(max_values_index[i], len(profiles)-1):
|
||||||
self.compl_sect(profiles[isect],
|
self.compl_sect(profiles[isect],
|
||||||
profiles[isect+1],
|
profiles[isect+1],
|
||||||
gl1[i], gl2[i])
|
guide_list[i],
|
||||||
|
guide_list[i+1])
|
||||||
|
|
||||||
for isect in reversed(range(0, max_values_index[i])):
|
for isect in reversed(range(0, max_values_index[i])):
|
||||||
self.compl_sect(profiles[isect+1],
|
self.compl_sect(profiles[isect+1],
|
||||||
profiles[isect],
|
profiles[isect],
|
||||||
gl1[i], gl2[i])
|
guide_list[i],
|
||||||
|
guide_list[i+1])
|
||||||
|
|
||||||
return profiles
|
return profiles
|
||||||
|
|
||||||
def interpolate_transversal_step(self,
|
def interpolate_transversal_step(self,
|
||||||
@@ -146,6 +130,7 @@ class InternalMeshing(AMeshingTool):
|
|||||||
p.name = f'interpol{p.rk}'
|
p.name = f'interpol{p.rk}'
|
||||||
new_profiles2.append(p)
|
new_profiles2.append(p)
|
||||||
new_profiles.append(new_profiles2)
|
new_profiles.append(new_profiles2)
|
||||||
|
|
||||||
return new_profiles
|
return new_profiles
|
||||||
|
|
||||||
def compl_sect(self, sect1, sect2, tag1, tag2):
|
def compl_sect(self, sect1, sect2, tag1, tag2):
|
||||||
@@ -156,17 +141,9 @@ class InternalMeshing(AMeshingTool):
|
|||||||
# len1 and 2: number of intervals
|
# len1 and 2: number of intervals
|
||||||
# len1: target len
|
# len1: target len
|
||||||
|
|
||||||
if tag1 == '': # left end point
|
|
||||||
start1 = 0
|
|
||||||
start2 = 0
|
|
||||||
else:
|
|
||||||
start1 = sect1.named_point_index(tag1)
|
start1 = sect1.named_point_index(tag1)
|
||||||
start2 = sect2.named_point_index(tag1)
|
start2 = sect2.named_point_index(tag1)
|
||||||
|
|
||||||
if tag2 == '': # right end point
|
|
||||||
end1 = sect1.nb_points-1
|
|
||||||
end2 = sect2.nb_points-1
|
|
||||||
else:
|
|
||||||
end1 = sect1.named_point_index(tag2)
|
end1 = sect1.named_point_index(tag2)
|
||||||
end2 = sect2.named_point_index(tag2)
|
end2 = sect2.named_point_index(tag2)
|
||||||
|
|
||||||
@@ -174,8 +151,7 @@ class InternalMeshing(AMeshingTool):
|
|||||||
len2 = end2 - start2
|
len2 = end2 - start2
|
||||||
|
|
||||||
if len1 < len2:
|
if len1 < len2:
|
||||||
self.compl_sect(sect2, sect1, start2, start1,
|
self.compl_sect(sect2, sect1, tag1, tag2)
|
||||||
len2, len1, tag1, tag2)
|
|
||||||
return
|
return
|
||||||
elif len1 == len2:
|
elif len1 == len2:
|
||||||
return
|
return
|
||||||
@@ -190,27 +166,25 @@ class InternalMeshing(AMeshingTool):
|
|||||||
for i in range(len1):
|
for i in range(len1):
|
||||||
ltot1 += sect1.point(start1+i).dist(sect1.point(start1+i+1))
|
ltot1 += sect1.point(start1+i).dist(sect1.point(start1+i+1))
|
||||||
alpha.append(ltot1)
|
alpha.append(ltot1)
|
||||||
alpha = list(map(lambda x: x/ltot1, alpha)) # target ratios
|
|
||||||
for i in range(len2):
|
for i in range(len2):
|
||||||
ltot2 += sect2.point(start2+i).dist(sect2.point(start2+i+1))
|
ltot2 += sect2.point(start2+i).dist(sect2.point(start2+i+1))
|
||||||
beta.append(ltot2)
|
beta.append(ltot2)
|
||||||
beta = list(map(lambda x: x/ltot2, beta)) # current ratios
|
|
||||||
for i in range(len1 - len2):
|
|
||||||
beta.append(1.0)
|
|
||||||
beta.append(0.0) # beta[-1]
|
|
||||||
if len2 < 1 or ltot2 < 0.0001:
|
if len2 < 1 or ltot2 < 0.0001:
|
||||||
# unique point (copy len1 times)
|
# unique point (copy len1 times)
|
||||||
# we are at an edge of the profile
|
# we are at an edge of the profile
|
||||||
for i in range(len1-len2):
|
for i in range(len1-len2):
|
||||||
p = sect2.point(start2).copy()
|
p = sect2.point(start2).copy()
|
||||||
sect2.insert_point(start2, p)
|
sect2.insert_point(start2, p)
|
||||||
if tag1 == '': # left end point
|
len2 += 1
|
||||||
sect2.point(start2).name = ''
|
|
||||||
if tag2 == '': # left end point
|
|
||||||
sect2.point(start2+1).name = ''
|
|
||||||
elif ltot1 < 0.0001:
|
elif ltot1 < 0.0001:
|
||||||
sect2.add_npoints(len1-len2)
|
sect2.add_npoints(len1-len2)
|
||||||
|
len2 = len1
|
||||||
else: # regular case
|
else: # regular case
|
||||||
|
alpha = list(map(lambda x: x/ltot1, alpha)) # target ratios
|
||||||
|
beta = list(map(lambda x: x/ltot2, beta)) # current ratios
|
||||||
|
for i in range(len1 - len2):
|
||||||
|
beta.append(1.0)
|
||||||
|
beta.append(0.0) # beta[-1]
|
||||||
# loop on the points to insert
|
# loop on the points to insert
|
||||||
for k in range(len1-len2):
|
for k in range(len1-len2):
|
||||||
trouve = False
|
trouve = False
|
||||||
@@ -242,7 +216,7 @@ class InternalMeshing(AMeshingTool):
|
|||||||
ratio = (alpha[j0] - beta[j0-1]) \
|
ratio = (alpha[j0] - beta[j0-1]) \
|
||||||
/ (beta[j0] - beta[j0-1])
|
/ (beta[j0] - beta[j0-1])
|
||||||
if ratio < 0.0:
|
if ratio < 0.0:
|
||||||
print(f"ratio négatif {ratio}")
|
logger.warning(f"ratio négatif {ratio}")
|
||||||
# on double le point a gauche
|
# on double le point a gauche
|
||||||
p = sect2.point(start2+j0-1).copy()
|
p = sect2.point(start2+j0-1).copy()
|
||||||
sect2.insert_point(start2+j0-1, p)
|
sect2.insert_point(start2+j0-1, p)
|
||||||
@@ -269,8 +243,11 @@ class InternalMeshing(AMeshingTool):
|
|||||||
p = sect2.point(start2+len2).copy()
|
p = sect2.point(start2+len2).copy()
|
||||||
sect2.insert_point(start2+len2, p)
|
sect2.insert_point(start2+len2, p)
|
||||||
sect2.point(start2+len2).name = ''
|
sect2.point(start2+len2).name = ''
|
||||||
|
if tag1 != "un":
|
||||||
sect2.point(start2).name = tag1
|
sect2.point(start2).name = tag1
|
||||||
|
if tag2 != "np":
|
||||||
sect2.point(start2+len2).name = tag2
|
sect2.point(start2+len2).name = tag2
|
||||||
|
|
||||||
sect2.modified()
|
sect2.modified()
|
||||||
|
|
||||||
def update_rk(self, reach, begin_rk, end_rk,
|
def update_rk(self, reach, begin_rk, end_rk,
|
||||||
@@ -304,6 +281,8 @@ class InternalMeshing(AMeshingTool):
|
|||||||
sgn = -1.0
|
sgn = -1.0
|
||||||
else:
|
else:
|
||||||
sgn = old_orientation
|
sgn = old_orientation
|
||||||
|
if abs(sgn) < 0.5:
|
||||||
|
sgn = 1.0
|
||||||
|
|
||||||
new_rk = [0.0]*nprof
|
new_rk = [0.0]*nprof
|
||||||
new_rk[origin] = origin_value
|
new_rk[origin] = origin_value
|
||||||
@@ -344,3 +323,42 @@ class InternalMeshing(AMeshingTool):
|
|||||||
return new_rk, new_end_rk, new_begin_rk
|
return new_rk, new_end_rk, new_begin_rk
|
||||||
else:
|
else:
|
||||||
return new_rk, new_begin_rk, new_end_rk
|
return new_rk, new_begin_rk, new_end_rk
|
||||||
|
|
||||||
|
def get_guide_list(self, reach, profiles):
|
||||||
|
|
||||||
|
gl = reach.compute_guidelines()
|
||||||
|
|
||||||
|
# we make sure that the lines are in the left-to-right order
|
||||||
|
guide_list = [
|
||||||
|
[
|
||||||
|
x.name
|
||||||
|
for x in p.named_points()
|
||||||
|
if x.name in gl[0]
|
||||||
|
]
|
||||||
|
for p in profiles]
|
||||||
|
|
||||||
|
# we make sure the lines never cross
|
||||||
|
for gl in guide_list:
|
||||||
|
for i in range(len(guide_list[0])):
|
||||||
|
if gl[i] != guide_list[0][i]:
|
||||||
|
return None
|
||||||
|
|
||||||
|
return guide_list[0]
|
||||||
|
|
||||||
|
def get_profiles(self, reach, limites):
|
||||||
|
|
||||||
|
if reach is None or len(reach.profiles) == 0:
|
||||||
|
return None
|
||||||
|
|
||||||
|
if limites[0] > limites[1]:
|
||||||
|
lim = limites[1]
|
||||||
|
limites[1] = limites[0]
|
||||||
|
limites[0] = lim
|
||||||
|
elif limites[0] == limites[1]:
|
||||||
|
return None
|
||||||
|
|
||||||
|
profiles = [reach.profile(i).copy()
|
||||||
|
for i in range(limites[0], limites[1]+1)
|
||||||
|
]
|
||||||
|
|
||||||
|
return profiles
|
||||||
@@ -232,6 +232,8 @@ class BoundaryCondition(SQLSubModel):
|
|||||||
type TEXT NOT NULL,
|
type TEXT NOT NULL,
|
||||||
tab TEXT NOT NULL,
|
tab TEXT NOT NULL,
|
||||||
node INTEGER,
|
node INTEGER,
|
||||||
|
d50 REAL DEFAULT 0.002,
|
||||||
|
sigma REAL DEFAULT 1,
|
||||||
{Scenario.create_db_add_scenario()},
|
{Scenario.create_db_add_scenario()},
|
||||||
{Scenario.create_db_add_scenario_fk()},
|
{Scenario.create_db_add_scenario_fk()},
|
||||||
FOREIGN KEY(node) REFERENCES river_node(pamhyr_id),
|
FOREIGN KEY(node) REFERENCES river_node(pamhyr_id),
|
||||||
@@ -258,6 +260,17 @@ class BoundaryCondition(SQLSubModel):
|
|||||||
"ADD COLUMN deleted BOOLEAN NOT NULL DEFAULT FALSE"
|
"ADD COLUMN deleted BOOLEAN NOT NULL DEFAULT FALSE"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
if major == "0" and int(minor) <= 2:
|
||||||
|
if int(release) < 4:
|
||||||
|
execute(
|
||||||
|
"ALTER TABLE boundary_condition " +
|
||||||
|
"ADD COLUMN d50 REAL DEFAULT 0.002"
|
||||||
|
)
|
||||||
|
execute(
|
||||||
|
"ALTER TABLE boundary_condition " +
|
||||||
|
"ADD COLUMN sigma REAL DEFAULT 1"
|
||||||
|
)
|
||||||
|
|
||||||
return cls._update_submodel(execute, version, data)
|
return cls._update_submodel(execute, version, data)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
@@ -316,7 +329,8 @@ class BoundaryCondition(SQLSubModel):
|
|||||||
return new
|
return new
|
||||||
|
|
||||||
table = execute(
|
table = execute(
|
||||||
"SELECT pamhyr_id, deleted, name, type, node, scenario " +
|
"SELECT pamhyr_id, deleted, name, type, node, d50, sigma, " +
|
||||||
|
"scenario " +
|
||||||
"FROM boundary_condition " +
|
"FROM boundary_condition " +
|
||||||
f"WHERE tab = '{tab}' " +
|
f"WHERE tab = '{tab}' " +
|
||||||
f"AND scenario = {scenario.id} " +
|
f"AND scenario = {scenario.id} " +
|
||||||
@@ -331,6 +345,8 @@ class BoundaryCondition(SQLSubModel):
|
|||||||
name = next(it)
|
name = next(it)
|
||||||
t = next(it)
|
t = next(it)
|
||||||
node = next(it)
|
node = next(it)
|
||||||
|
d50 = next(it)
|
||||||
|
sigma = next(it)
|
||||||
owner_scenario = next(it)
|
owner_scenario = next(it)
|
||||||
|
|
||||||
ctor = cls._get_ctor_from_type(t)
|
ctor = cls._get_ctor_from_type(t)
|
||||||
@@ -340,9 +356,14 @@ class BoundaryCondition(SQLSubModel):
|
|||||||
status=data['status'],
|
status=data['status'],
|
||||||
owner_scenario=owner_scenario
|
owner_scenario=owner_scenario
|
||||||
)
|
)
|
||||||
|
|
||||||
if deleted:
|
if deleted:
|
||||||
bc.set_as_deleted()
|
bc.set_as_deleted()
|
||||||
|
|
||||||
|
if t == "SL":
|
||||||
|
bc.d50 = d50
|
||||||
|
bc.sigma = sigma
|
||||||
|
|
||||||
bc.node = None
|
bc.node = None
|
||||||
if node != -1:
|
if node != -1:
|
||||||
bc.node = next(filter(lambda n: n.id == node, nodes), None)
|
bc.node = next(filter(lambda n: n.id == node, nodes), None)
|
||||||
@@ -375,15 +396,22 @@ class BoundaryCondition(SQLSubModel):
|
|||||||
if self._node is not None:
|
if self._node is not None:
|
||||||
node = self._node.id
|
node = self._node.id
|
||||||
|
|
||||||
|
d50 = 0.002
|
||||||
|
sigma = 1
|
||||||
|
if self._type == "SL":
|
||||||
|
d50 = self._d50
|
||||||
|
sigma = self._sigma
|
||||||
|
|
||||||
execute(
|
execute(
|
||||||
"INSERT INTO " +
|
"INSERT INTO " +
|
||||||
"boundary_condition(" +
|
"boundary_condition(" +
|
||||||
"pamhyr_id, deleted, name, type, tab, node, scenario" +
|
"pamhyr_id, deleted, name, type, tab, node, d50, sigma, scenario" +
|
||||||
") " +
|
") " +
|
||||||
"VALUES (" +
|
"VALUES (" +
|
||||||
f"{self._pamhyr_id}, {self._db_format(self.is_deleted())}, " +
|
f"{self._pamhyr_id}, {self._db_format(self.is_deleted())}, " +
|
||||||
f"'{self._db_format(self._name)}', " +
|
f"'{self._db_format(self._name)}', " +
|
||||||
f"'{self._db_format(self._type)}', '{tab}', {node}, " +
|
f"'{self._db_format(self._type)}', '{tab}', {node}, " +
|
||||||
|
f"{d50}, {sigma}, " +
|
||||||
f"{self._status.scenario_id}" +
|
f"{self._status.scenario_id}" +
|
||||||
")"
|
")"
|
||||||
)
|
)
|
||||||
@@ -468,6 +496,24 @@ class BoundaryCondition(SQLSubModel):
|
|||||||
def has_node(self):
|
def has_node(self):
|
||||||
return self._node is not None
|
return self._node is not None
|
||||||
|
|
||||||
|
@property
|
||||||
|
def d50(self):
|
||||||
|
return self._d50
|
||||||
|
|
||||||
|
@d50.setter
|
||||||
|
def d50(self, value):
|
||||||
|
self._d50 = float(value)
|
||||||
|
self.modified()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def sigma(self):
|
||||||
|
return self._sigma
|
||||||
|
|
||||||
|
@sigma.setter
|
||||||
|
def sigma(self, value):
|
||||||
|
self._sigma = float(value)
|
||||||
|
self.modified()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def header(self):
|
def header(self):
|
||||||
return self._header.copy()
|
return self._header.copy()
|
||||||
|
|||||||
@@ -70,9 +70,8 @@ class Data(SQLSubModel):
|
|||||||
def _db_update(cls, execute, version, data=None):
|
def _db_update(cls, execute, version, data=None):
|
||||||
major, minor, release = version.strip().split(".")
|
major, minor, release = version.strip().split(".")
|
||||||
created = False
|
created = False
|
||||||
|
if major == "0" and minor == "0":
|
||||||
if major == "0" and int(minor) <= 1:
|
if int(release) < 11:
|
||||||
if int(release) < 7:
|
|
||||||
cls._db_create(execute)
|
cls._db_create(execute)
|
||||||
created = True
|
created = True
|
||||||
|
|
||||||
@@ -143,8 +142,9 @@ class Data(SQLSubModel):
|
|||||||
table = execute(
|
table = execute(
|
||||||
"SELECT pamhyr_id, deleted, data0, data1, scenario FROM " +
|
"SELECT pamhyr_id, deleted, data0, data1, scenario FROM " +
|
||||||
"boundary_condition_data_adists " +
|
"boundary_condition_data_adists " +
|
||||||
f"WHERE bca = {bca.id} " +
|
f"WHERE scenario = {scenario.id} " +
|
||||||
f"AND scenario = {scenario.id}"
|
# f"AND pamhyr_id NOT IN ({', '.join(map(str, loaded))}) " +
|
||||||
|
f"AND bca = {bca.id}"
|
||||||
)
|
)
|
||||||
|
|
||||||
if table is not None:
|
if table is not None:
|
||||||
@@ -187,8 +187,9 @@ class Data(SQLSubModel):
|
|||||||
execute(
|
execute(
|
||||||
"INSERT INTO " +
|
"INSERT INTO " +
|
||||||
"boundary_condition_data_adists(" +
|
"boundary_condition_data_adists(" +
|
||||||
"pamhyr_id, data0, data1, bca, scenario) " +
|
"pamhyr_id, deleted, data0, data1, bca, scenario) " +
|
||||||
f"VALUES ({self.id}, '{data0}', {data1}, {bca.id}, " +
|
f"VALUES ({self.id}, {self._db_format(self.is_deleted())}, " +
|
||||||
|
f"'{data0}', {data1}, {bca.id}, " +
|
||||||
f"{self._status.scenario_id})"
|
f"{self._status.scenario_id})"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -247,15 +248,13 @@ class BoundaryConditionAdisTS(SQLSubModel):
|
|||||||
major, minor, release = version.strip().split(".")
|
major, minor, release = version.strip().split(".")
|
||||||
created = False
|
created = False
|
||||||
|
|
||||||
if major == "0" and int(minor) <= 1:
|
if major == "0" and int(minor) < 2:
|
||||||
if int(release) < 7:
|
if cls.is_table_exists(execute, "boundary_condition_adists"):
|
||||||
|
cls._db_update_to_0_2_0(execute, data)
|
||||||
|
else:
|
||||||
cls._db_create(execute)
|
cls._db_create(execute)
|
||||||
created = True
|
created = True
|
||||||
|
|
||||||
if major == "0" and int(minor) < 2:
|
|
||||||
if not created:
|
|
||||||
cls._db_update_to_0_2_0(execute, data)
|
|
||||||
|
|
||||||
if not created:
|
if not created:
|
||||||
return cls._update_submodel(execute, version, data)
|
return cls._update_submodel(execute, version, data)
|
||||||
|
|
||||||
@@ -300,6 +299,14 @@ class BoundaryConditionAdisTS(SQLSubModel):
|
|||||||
if pol_id == -1:
|
if pol_id == -1:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
|
if pol_id not in pid_pol:
|
||||||
|
# ⚠️ cas important : probablement déjà migré
|
||||||
|
logger.warning(
|
||||||
|
f"pol_id {pol_id} not in pid_pol " +
|
||||||
|
"→ probably already migrated"
|
||||||
|
)
|
||||||
|
continue
|
||||||
|
|
||||||
execute(
|
execute(
|
||||||
f"UPDATE boundary_condition_adists " +
|
f"UPDATE boundary_condition_adists " +
|
||||||
f"SET pollutant = {pid_pol[pol_id]} " +
|
f"SET pollutant = {pid_pol[pol_id]} " +
|
||||||
@@ -312,17 +319,21 @@ class BoundaryConditionAdisTS(SQLSubModel):
|
|||||||
status = data['status']
|
status = data['status']
|
||||||
nodes = data['nodes']
|
nodes = data['nodes']
|
||||||
scenario = data["scenario"]
|
scenario = data["scenario"]
|
||||||
|
pollutant = data.get("pollutant")
|
||||||
loaded = data['loaded_pid']
|
loaded = data['loaded_pid']
|
||||||
|
|
||||||
if scenario is None:
|
if scenario is None:
|
||||||
return new
|
return new
|
||||||
|
|
||||||
table = execute(
|
sql = (
|
||||||
"SELECT pamhyr_id, deleted, pollutant, type, node, scenario " +
|
"SELECT pamhyr_id, deleted, pollutant, type, node, scenario "
|
||||||
"FROM boundary_condition_adists " +
|
"FROM boundary_condition_adists " +
|
||||||
f"WHERE scenario = {scenario.id} " +
|
f"WHERE scenario = {scenario.id} "
|
||||||
f"AND pamhyr_id NOT IN ({', '.join(map(str, loaded))}) "
|
|
||||||
)
|
)
|
||||||
|
if pollutant is not None:
|
||||||
|
sql += f"AND pollutant = {pollutant.id} "
|
||||||
|
|
||||||
|
table = execute(sql)
|
||||||
|
|
||||||
if table is not None:
|
if table is not None:
|
||||||
for row in table:
|
for row in table:
|
||||||
@@ -347,7 +358,7 @@ class BoundaryConditionAdisTS(SQLSubModel):
|
|||||||
bc.type = bc_type
|
bc.type = bc_type
|
||||||
|
|
||||||
bc.node = None
|
bc.node = None
|
||||||
if row[3] != -1:
|
if node != -1:
|
||||||
tmp = next(
|
tmp = next(
|
||||||
filter(
|
filter(
|
||||||
lambda n: n.id == node, nodes
|
lambda n: n.id == node, nodes
|
||||||
@@ -426,7 +437,14 @@ class BoundaryConditionAdisTS(SQLSubModel):
|
|||||||
d._data_traversal(predicate, modifier, data)
|
d._data_traversal(predicate, modifier, data)
|
||||||
|
|
||||||
def __len__(self):
|
def __len__(self):
|
||||||
return len(self._data)
|
return len(
|
||||||
|
list(
|
||||||
|
filter(
|
||||||
|
lambda el: el is not None and not el.is_deleted(),
|
||||||
|
self._data
|
||||||
|
)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def time_convert(cls, data):
|
def time_convert(cls, data):
|
||||||
@@ -451,7 +469,7 @@ class BoundaryConditionAdisTS(SQLSubModel):
|
|||||||
@node.setter
|
@node.setter
|
||||||
def node(self, node):
|
def node(self, node):
|
||||||
self._node = node
|
self._node = node
|
||||||
self._status.modified()
|
self.modified()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def header(self):
|
def header(self):
|
||||||
@@ -460,7 +478,7 @@ class BoundaryConditionAdisTS(SQLSubModel):
|
|||||||
@header.setter
|
@header.setter
|
||||||
def header(self, header):
|
def header(self, header):
|
||||||
self._header = header
|
self._header = header
|
||||||
self._status.modified()
|
self.modified()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def pollutant(self):
|
def pollutant(self):
|
||||||
@@ -469,7 +487,7 @@ class BoundaryConditionAdisTS(SQLSubModel):
|
|||||||
@pollutant.setter
|
@pollutant.setter
|
||||||
def pollutant(self, pollutant):
|
def pollutant(self, pollutant):
|
||||||
self._pollutant = pollutant
|
self._pollutant = pollutant
|
||||||
self._status.modified()
|
self.modified()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def type(self):
|
def type(self):
|
||||||
@@ -478,11 +496,16 @@ class BoundaryConditionAdisTS(SQLSubModel):
|
|||||||
@type.setter
|
@type.setter
|
||||||
def type(self, type):
|
def type(self, type):
|
||||||
self._type = type
|
self._type = type
|
||||||
self._status.modified()
|
self.modified()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def data(self):
|
def data(self):
|
||||||
return self._data.copy()
|
return list(
|
||||||
|
filter(
|
||||||
|
lambda el: el is not None and not el.is_deleted(),
|
||||||
|
self._data
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def _default_0(self):
|
def _default_0(self):
|
||||||
@@ -511,33 +534,36 @@ class BoundaryConditionAdisTS(SQLSubModel):
|
|||||||
return (new_0, new_1)
|
return (new_0, new_1)
|
||||||
|
|
||||||
def add(self, index: int):
|
def add(self, index: int):
|
||||||
value = (self._default_0, self._default_1)
|
value = Data(
|
||||||
|
self._default_0, self._default_1,
|
||||||
|
status=self._status
|
||||||
|
)
|
||||||
self._data.insert(index, value)
|
self._data.insert(index, value)
|
||||||
self._status.modified()
|
self.modified()
|
||||||
return value
|
return value
|
||||||
|
|
||||||
def insert(self, index: int, value):
|
def insert(self, index: int, value):
|
||||||
self._data.insert(index, value)
|
self._data.insert(index, value)
|
||||||
self._status.modified()
|
self.modified()
|
||||||
|
|
||||||
def delete_i(self, indexes):
|
def delete_i(self, indexes):
|
||||||
self._data = list(
|
list(
|
||||||
map(
|
map(
|
||||||
lambda e: e[1],
|
lambda e: e[1].set_as_deleted(),
|
||||||
filter(
|
filter(
|
||||||
lambda e: e[0] not in indexes,
|
lambda e: e[0] in indexes,
|
||||||
enumerate(self.data)
|
enumerate(self.data)
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self._status.modified()
|
self.modified()
|
||||||
|
|
||||||
def sort(self, _reverse=False, key=None):
|
def sort(self, _reverse=False, key=None):
|
||||||
if key is None:
|
if key is None:
|
||||||
self._data.sort(reverse=_reverse)
|
self._data.sort(reverse=_reverse)
|
||||||
else:
|
else:
|
||||||
self._data.sort(reverse=_reverse, key=key)
|
self._data.sort(reverse=_reverse, key=key)
|
||||||
self._status.modified()
|
self.modified()
|
||||||
|
|
||||||
def index(self, bc):
|
def index(self, bc):
|
||||||
self._data.index(bc)
|
self._data.index(bc)
|
||||||
@@ -552,10 +578,10 @@ class BoundaryConditionAdisTS(SQLSubModel):
|
|||||||
return lst
|
return lst
|
||||||
|
|
||||||
def _set_i_c_v(self, index, column, value):
|
def _set_i_c_v(self, index, column, value):
|
||||||
v = list(self._data[index])
|
v = self._data[index]
|
||||||
v[column] = self._types[column](value)
|
v[column] = self._types[column](value)
|
||||||
self._data[index] = tuple(v)
|
self._data[index] = v
|
||||||
self._status.modified()
|
self.modified()
|
||||||
|
|
||||||
def set_i_0(self, index: int, value):
|
def set_i_0(self, index: int, value):
|
||||||
self._set_i_c_v(index, 0, value)
|
self._set_i_c_v(index, 0, value)
|
||||||
|
|||||||
@@ -38,9 +38,13 @@ class BoundaryConditionsAdisTSList(PamhyrModelList):
|
|||||||
if data is None:
|
if data is None:
|
||||||
data = {}
|
data = {}
|
||||||
|
|
||||||
new._lst = BoundaryConditionAdisTS._db_load(
|
previous_pollutant = data.get("pollutant")
|
||||||
execute, data
|
data.pop("pollutant", None)
|
||||||
)
|
|
||||||
|
new._lst = BoundaryConditionAdisTS._db_load(execute, data)
|
||||||
|
|
||||||
|
if previous_pollutant is not None:
|
||||||
|
data["pollutant"] = previous_pollutant
|
||||||
|
|
||||||
return new
|
return new
|
||||||
|
|
||||||
|
|||||||
@@ -73,15 +73,13 @@ class D90AdisTS(SQLSubModel):
|
|||||||
major, minor, release = version.strip().split(".")
|
major, minor, release = version.strip().split(".")
|
||||||
created = False
|
created = False
|
||||||
|
|
||||||
if major == "0" and int(minor) <= 1:
|
if major == "0" and int(minor) < 2:
|
||||||
if int(release) < 6:
|
if cls.is_table_exists(execute, "d90_adists"):
|
||||||
|
cls._db_update_to_0_2_0(execute, data)
|
||||||
|
else:
|
||||||
cls._db_create(execute)
|
cls._db_create(execute)
|
||||||
created = True
|
created = True
|
||||||
|
|
||||||
if major == "0" and int(minor) < 2:
|
|
||||||
if not created:
|
|
||||||
cls._db_update_to_0_2_0(execute, data)
|
|
||||||
|
|
||||||
if not created:
|
if not created:
|
||||||
return cls._update_submodel(execute, version, data)
|
return cls._update_submodel(execute, version, data)
|
||||||
|
|
||||||
|
|||||||
@@ -70,16 +70,12 @@ class D90AdisTSSpec(SQLSubModel):
|
|||||||
@classmethod
|
@classmethod
|
||||||
def _db_update(cls, execute, version, data=None):
|
def _db_update(cls, execute, version, data=None):
|
||||||
major, minor, release = version.strip().split(".")
|
major, minor, release = version.strip().split(".")
|
||||||
created = False
|
|
||||||
|
|
||||||
if major == 0 and minor < 1:
|
|
||||||
if int(release) < 6:
|
|
||||||
cls._db_create(execute)
|
|
||||||
created = True
|
|
||||||
|
|
||||||
if major == "0" and int(minor) < 2:
|
if major == "0" and int(minor) < 2:
|
||||||
if not created:
|
if cls.is_table_exists(execute, "d90_spec"):
|
||||||
cls._db_update_to_0_2_0(execute, data)
|
cls._db_update_to_0_2_0(execute, data)
|
||||||
|
else:
|
||||||
|
cls._db_create(execute)
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
@@ -87,6 +83,7 @@ class D90AdisTSSpec(SQLSubModel):
|
|||||||
def _db_update_to_0_2_0(cls, execute, data):
|
def _db_update_to_0_2_0(cls, execute, data):
|
||||||
table = "d90_spec"
|
table = "d90_spec"
|
||||||
table_new = "d90_adists_spec"
|
table_new = "d90_adists_spec"
|
||||||
|
reachs = data['id2pid']['river_reach']
|
||||||
|
|
||||||
cls.update_db_add_pamhyr_id(execute, table, data)
|
cls.update_db_add_pamhyr_id(execute, table, data)
|
||||||
Scenario.update_db_add_scenario(execute, table)
|
Scenario.update_db_add_scenario(execute, table)
|
||||||
@@ -95,8 +92,10 @@ class D90AdisTSSpec(SQLSubModel):
|
|||||||
|
|
||||||
execute(
|
execute(
|
||||||
f"INSERT INTO {table_new}_tmp " +
|
f"INSERT INTO {table_new}_tmp " +
|
||||||
"(pamhyr_id, name, d90, enabled, scenario) " +
|
"(pamhyr_id, d90_default, name, reach, " +
|
||||||
"SELECT pamhyr_id, name, d90, enabled, scenario " +
|
" start_rk, end_rk, d90, enabled, scenario) " +
|
||||||
|
"SELECT pamhyr_id, d90_default, name, reach, " +
|
||||||
|
" start_rk, end_rk, d90, enabled, scenario " +
|
||||||
f"FROM {table}"
|
f"FROM {table}"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -104,12 +103,13 @@ class D90AdisTSSpec(SQLSubModel):
|
|||||||
execute(f"ALTER TABLE {table_new}_tmp RENAME TO {table_new}")
|
execute(f"ALTER TABLE {table_new}_tmp RENAME TO {table_new}")
|
||||||
|
|
||||||
cls._db_update_to_0_2_0_set_d90_pid(execute, data)
|
cls._db_update_to_0_2_0_set_d90_pid(execute, data)
|
||||||
|
cls._db_update_to_0_2_0_set_reach_pid(execute, table_new, reachs)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def _db_update_to_0_2_0_set_d90_pid(cls, execute, data):
|
def _db_update_to_0_2_0_set_d90_pid(cls, execute, data):
|
||||||
pid_d90 = data["id2pid"]["d90_adists"]
|
pid_d90 = data["id2pid"]["d90_adists"]
|
||||||
els = execute(
|
els = execute(
|
||||||
f"SELECT pamhyr_id, d90 FROM d90_adists_spec"
|
f"SELECT pamhyr_id, d90_default FROM d90_adists_spec"
|
||||||
)
|
)
|
||||||
|
|
||||||
for row in els:
|
for row in els:
|
||||||
@@ -122,7 +122,7 @@ class D90AdisTSSpec(SQLSubModel):
|
|||||||
|
|
||||||
execute(
|
execute(
|
||||||
f"UPDATE d90_adists_spec " +
|
f"UPDATE d90_adists_spec " +
|
||||||
f"SET d90 = {pid_d90[d90_id]} " +
|
f"SET d90_default = {pid_d90[d90_id]} " +
|
||||||
f"WHERE pamhyr_id = {pid}"
|
f"WHERE pamhyr_id = {pid}"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -167,7 +167,7 @@ class D90AdisTSSpec(SQLSubModel):
|
|||||||
owner_scenario=owner_scenario
|
owner_scenario=owner_scenario
|
||||||
)
|
)
|
||||||
if deleted:
|
if deleted:
|
||||||
new_spec.is_deleted()
|
new_spec.set_as_deleted()
|
||||||
|
|
||||||
new_spec.reach = reach
|
new_spec.reach = reach
|
||||||
new_spec.start_rk = start_rk
|
new_spec.start_rk = start_rk
|
||||||
@@ -196,12 +196,12 @@ class D90AdisTSSpec(SQLSubModel):
|
|||||||
"d90_default, name, reach, " +
|
"d90_default, name, reach, " +
|
||||||
"start_rk, end_rk, d90, enabled, scenario) " +
|
"start_rk, end_rk, d90, enabled, scenario) " +
|
||||||
"VALUES (" +
|
"VALUES (" +
|
||||||
f"{self.id}, {self._db_format(self.is_deleted())}" +
|
f"{self.id}, {self._db_format(self.is_deleted())}, " +
|
||||||
f"{d90_default}, " +
|
f"{d90_default}, " +
|
||||||
f"'{self._db_format(self._name_section)}', " +
|
f"'{self._db_format(self._name_section)}', " +
|
||||||
f"{self._reach}, " +
|
f"{self._reach}, " +
|
||||||
f"{self._start_rk}, {self._end_rk}, " +
|
f"{self._start_rk}, {self._end_rk}, " +
|
||||||
f"{self._d90}, {self._enabled}" +
|
f"{self._d90}, {self._enabled}, " +
|
||||||
f"{self._status.scenario_id}" +
|
f"{self._status.scenario_id}" +
|
||||||
")"
|
")"
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -79,15 +79,13 @@ class DIFAdisTS(SQLSubModel):
|
|||||||
major, minor, release = version.strip().split(".")
|
major, minor, release = version.strip().split(".")
|
||||||
created = False
|
created = False
|
||||||
|
|
||||||
if major == "0" and int(minor) <= 1:
|
if major == "0" and int(minor) < 2:
|
||||||
if int(release) < 6:
|
if cls.is_table_exists(execute, "dif_adists"):
|
||||||
|
cls._db_update_to_0_2_0(execute, data)
|
||||||
|
else:
|
||||||
cls._db_create(execute)
|
cls._db_create(execute)
|
||||||
created = True
|
created = True
|
||||||
|
|
||||||
if major == "0" and int(minor) < 2:
|
|
||||||
if not created:
|
|
||||||
cls._db_update_to_0_2_0(execute, data)
|
|
||||||
|
|
||||||
if not created:
|
if not created:
|
||||||
return cls._update_submodel(execute, version, data)
|
return cls._update_submodel(execute, version, data)
|
||||||
|
|
||||||
@@ -142,24 +140,24 @@ class DIFAdisTS(SQLSubModel):
|
|||||||
enabled = (next(it) == 1)
|
enabled = (next(it) == 1)
|
||||||
owner_scenario = next(it)
|
owner_scenario = next(it)
|
||||||
|
|
||||||
dif = cls(
|
DIF = cls(
|
||||||
id=pid,
|
id=pid,
|
||||||
name=name,
|
name=name,
|
||||||
status=status,
|
status=status,
|
||||||
owner_scenario=owner_scenario,
|
owner_scenario=owner_scenario,
|
||||||
)
|
)
|
||||||
|
|
||||||
dif.method = method
|
DIF.method = method
|
||||||
dif.dif = dif
|
DIF.dif = dif
|
||||||
dif.b = b
|
DIF.b = b
|
||||||
dif.c = c
|
DIF.c = c
|
||||||
dif.enabled = enabled
|
DIF.enabled = enabled
|
||||||
|
|
||||||
data['dif_default_id'] = pid
|
data['dif_default_id'] = pid
|
||||||
dif._data = DIFAdisTSSpec._db_load(execute, data)
|
DIF._data = DIFAdisTSSpec._db_load(execute, data)
|
||||||
|
|
||||||
loaded.add(pid)
|
loaded.add(pid)
|
||||||
new.append(dif)
|
new.append(DIF)
|
||||||
|
|
||||||
data["scenario"] = scenario.parent
|
data["scenario"] = scenario.parent
|
||||||
new += cls._db_load(execute, data)
|
new += cls._db_load(execute, data)
|
||||||
@@ -190,7 +188,7 @@ class DIFAdisTS(SQLSubModel):
|
|||||||
b = self.b
|
b = self.b
|
||||||
|
|
||||||
c = -1.
|
c = -1.
|
||||||
if self.dif is not None:
|
if self.c is not None:
|
||||||
c = self.c
|
c = self.c
|
||||||
|
|
||||||
sql = (
|
sql = (
|
||||||
@@ -201,7 +199,7 @@ class DIFAdisTS(SQLSubModel):
|
|||||||
"VALUES (" +
|
"VALUES (" +
|
||||||
f"{self.id}, '{self._db_format(self._name)}', " +
|
f"{self.id}, '{self._db_format(self._name)}', " +
|
||||||
f"'{self._db_format(self._method)}', " +
|
f"'{self._db_format(self._method)}', " +
|
||||||
f"{dif.id}, {b}, {c}, {self._enabled}, " +
|
f"{dif}, {b}, {c}, {self._enabled}, " +
|
||||||
f"{self._status.scenario_id}" +
|
f"{self._status.scenario_id}" +
|
||||||
")"
|
")"
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -33,16 +33,14 @@ class DIFAdisTSSpec(SQLSubModel):
|
|||||||
_id_cnt = 0
|
_id_cnt = 0
|
||||||
|
|
||||||
def __init__(self, id: int = -1, method: str = "",
|
def __init__(self, id: int = -1, method: str = "",
|
||||||
status=None):
|
status=None, owner_scenario=-1):
|
||||||
super(DIFAdisTSSpec, self).__init__()
|
super(DIFAdisTSSpec, self).__init__(
|
||||||
|
id=id, status=status,
|
||||||
|
owner_scenario=owner_scenario
|
||||||
|
)
|
||||||
|
|
||||||
self._status = status
|
self._status = status
|
||||||
|
|
||||||
if id == -1:
|
|
||||||
self.id = DIFAdisTSSpec._id_cnt
|
|
||||||
else:
|
|
||||||
self.id = id
|
|
||||||
|
|
||||||
self._method = method
|
self._method = method
|
||||||
self._reach = None
|
self._reach = None
|
||||||
self._start_rk = None
|
self._start_rk = None
|
||||||
@@ -52,8 +50,6 @@ class DIFAdisTSSpec(SQLSubModel):
|
|||||||
self._c = None
|
self._c = None
|
||||||
self._enabled = True
|
self._enabled = True
|
||||||
|
|
||||||
DIFAdisTSSpec._id_cnt = max(DIFAdisTSSpec._id_cnt + 1, self.id)
|
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def _db_create(cls, execute, ext=""):
|
def _db_create(cls, execute, ext=""):
|
||||||
execute(f"""
|
execute(f"""
|
||||||
@@ -81,16 +77,12 @@ class DIFAdisTSSpec(SQLSubModel):
|
|||||||
@classmethod
|
@classmethod
|
||||||
def _db_update(cls, execute, version, data=None):
|
def _db_update(cls, execute, version, data=None):
|
||||||
major, minor, release = version.strip().split(".")
|
major, minor, release = version.strip().split(".")
|
||||||
created = False
|
|
||||||
|
|
||||||
if major == 0 and minor < 1:
|
|
||||||
if int(release) < 6:
|
|
||||||
cls._db_create(execute)
|
|
||||||
created = True
|
|
||||||
|
|
||||||
if major == "0" and int(minor) < 2:
|
if major == "0" and int(minor) < 2:
|
||||||
if not created:
|
if cls.is_table_exists(execute, "dif_spec"):
|
||||||
cls._db_update_to_0_2_0(execute, data)
|
cls._db_update_to_0_2_0(execute, data)
|
||||||
|
else:
|
||||||
|
cls._db_create(execute)
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
@@ -195,7 +187,7 @@ class DIFAdisTSSpec(SQLSubModel):
|
|||||||
new_spec.c = c
|
new_spec.c = c
|
||||||
new_spec.enabled = enabled
|
new_spec.enabled = enabled
|
||||||
|
|
||||||
loaded.add(pid)
|
loaded.add(id)
|
||||||
new.append(new_spec)
|
new.append(new_spec)
|
||||||
|
|
||||||
data["scenario"] = scenario.parent
|
data["scenario"] = scenario.parent
|
||||||
@@ -210,22 +202,23 @@ class DIFAdisTSSpec(SQLSubModel):
|
|||||||
|
|
||||||
dif_default = data['dif_default_id']
|
dif_default = data['dif_default_id']
|
||||||
|
|
||||||
execute(
|
sql = (
|
||||||
"INSERT INTO " +
|
"INSERT INTO " +
|
||||||
"dif_adists_spec(pamhyr_id, deleted, " +
|
"dif_adists_spec(pamhyr_id, deleted, " +
|
||||||
"dif_default, method, reach, " +
|
"dif_default, method, reach, " +
|
||||||
"start_rk, end_rk, dif, b, c, enabled, scenario) " +
|
"start_rk, end_rk, dif, b, c, enabled, scenario) " +
|
||||||
"VALUES (" +
|
"VALUES (" +
|
||||||
f"{self.id}, {self._db_format(self.is_deleted())}" +
|
f"{self.id}, {self._db_format(self.is_deleted())}, " +
|
||||||
f"{dif_default}, " +
|
f"{dif_default}, " +
|
||||||
f"'{self._db_format(self._method)}', " +
|
f"'{self._db_format(self._method)}', " +
|
||||||
f"{self._reach}, " +
|
f"{self._reach}, " +
|
||||||
f"{self._start_rk}, {self._end_rk}, " +
|
f"{self._start_rk}, {self._end_rk}, " +
|
||||||
f"{self._dif}, {self._b}, {self._c}, " +
|
f"{self._dif}, {self._b}, {self._c}, " +
|
||||||
f"{self._enabled}" +
|
f"{self._enabled}, " +
|
||||||
f"{self._status.scenario_id}" +
|
f"{self._status.scenario_id}" +
|
||||||
")"
|
")"
|
||||||
)
|
)
|
||||||
|
execute(sql)
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
|||||||
@@ -228,9 +228,10 @@ class Friction(SQLSubModel):
|
|||||||
ssi = self._status.scenario_id
|
ssi = self._status.scenario_id
|
||||||
|
|
||||||
return (
|
return (
|
||||||
self._begin_strickler._owner_scenario == ssi
|
self._begin_strickler is not None
|
||||||
|
and (self._begin_strickler._owner_scenario == ssi
|
||||||
# or self._end_strickler._owner_scenario == ssi
|
# or self._end_strickler._owner_scenario == ssi
|
||||||
or super(Friction, self).must_be_saved()
|
or super(Friction, self).must_be_saved())
|
||||||
)
|
)
|
||||||
|
|
||||||
def _db_save(self, execute, data=None):
|
def _db_save(self, execute, data=None):
|
||||||
@@ -364,3 +365,9 @@ class Friction(SQLSubModel):
|
|||||||
# self.end_strickler.medium])
|
# self.end_strickler.medium])
|
||||||
|
|
||||||
return minor, medium
|
return minor, medium
|
||||||
|
|
||||||
|
def update_rks(self, begin_rk, end_rk):
|
||||||
|
self._begin_rk = begin_rk
|
||||||
|
self._end_rk = end_rk
|
||||||
|
|
||||||
|
self.modified()
|
||||||
@@ -0,0 +1,368 @@
|
|||||||
|
# GeoTIFF.py -- Pamhyr
|
||||||
|
# Copyright (C) 2024-2025 INRAE
|
||||||
|
#
|
||||||
|
# This program is free software: you can redistribute it and/or modify
|
||||||
|
# it under the terms of the GNU General Public License as published by
|
||||||
|
# the Free Software Foundation, either version 3 of the License, or
|
||||||
|
# (at your option) any later version.
|
||||||
|
#
|
||||||
|
# This program is distributed in the hope that it will be useful,
|
||||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
# GNU General Public License for more details.
|
||||||
|
#
|
||||||
|
# You should have received a copy of the GNU General Public License
|
||||||
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
import os
|
||||||
|
import struct
|
||||||
|
import logging
|
||||||
|
|
||||||
|
from functools import reduce
|
||||||
|
|
||||||
|
from tools import trace, timer
|
||||||
|
|
||||||
|
from Model.Tools.PamhyrDB import SQLSubModel
|
||||||
|
from Model.Except import NotImplementedMethodeError
|
||||||
|
from Model.Scenario import Scenario
|
||||||
|
|
||||||
|
try:
|
||||||
|
import rasterio
|
||||||
|
import rasterio.control
|
||||||
|
import rasterio.crs
|
||||||
|
import rasterio.sample
|
||||||
|
import rasterio.vrt
|
||||||
|
import rasterio._features
|
||||||
|
|
||||||
|
from rasterio.io import MemoryFile
|
||||||
|
_rasterio_loaded = True
|
||||||
|
except Exception as e:
|
||||||
|
print(f"Module 'rasterio' is not available: {e}")
|
||||||
|
_rasterio_loaded = False
|
||||||
|
|
||||||
|
logger = logging.getLogger()
|
||||||
|
|
||||||
|
|
||||||
|
class GeoTIFF(SQLSubModel):
|
||||||
|
_sub_classes = []
|
||||||
|
|
||||||
|
def __init__(self, id: int = -1, enabled=True,
|
||||||
|
name="", description="",
|
||||||
|
path="", coordinates=None,
|
||||||
|
status=None, owner_scenario=-1):
|
||||||
|
super(GeoTIFF, self).__init__(
|
||||||
|
id=id, status=status,
|
||||||
|
owner_scenario=owner_scenario
|
||||||
|
)
|
||||||
|
|
||||||
|
self._enabled = enabled
|
||||||
|
self._name = f"GeoTIFF #{self._pamhyr_id}" if name == "" else name
|
||||||
|
self._description = description
|
||||||
|
|
||||||
|
self._file_bytes = b''
|
||||||
|
if coordinates is None:
|
||||||
|
self._coordinates = {
|
||||||
|
"bottom": 0.0,
|
||||||
|
"top": 0.0,
|
||||||
|
"left": 0.0,
|
||||||
|
"right": 0.0,
|
||||||
|
}
|
||||||
|
else:
|
||||||
|
self._coordinates = coordinates
|
||||||
|
self._file_name = ""
|
||||||
|
|
||||||
|
if path != "":
|
||||||
|
self.read_file(path)
|
||||||
|
|
||||||
|
self._memfile = None
|
||||||
|
|
||||||
|
def __getitem__(self, key):
|
||||||
|
value = None
|
||||||
|
|
||||||
|
if key == "enabled":
|
||||||
|
value = self._enabled
|
||||||
|
elif key == "name":
|
||||||
|
value = self.name
|
||||||
|
elif key == "description":
|
||||||
|
value = self.description
|
||||||
|
elif key == "file_name":
|
||||||
|
value = self.file_name
|
||||||
|
elif key == "coordinates":
|
||||||
|
value = self.coordinates
|
||||||
|
elif key == "coordinates_bottom":
|
||||||
|
value = self.coordinates["bottom"]
|
||||||
|
elif key == "coordinates_top":
|
||||||
|
value = self.coordinates["top"]
|
||||||
|
elif key == "coordinates_left":
|
||||||
|
value = self.coordinates["left"]
|
||||||
|
elif key == "coordinates_right":
|
||||||
|
value = self.coordinates["right"]
|
||||||
|
elif key == "memfile":
|
||||||
|
value = self.memfile
|
||||||
|
|
||||||
|
return value
|
||||||
|
|
||||||
|
def __setitem__(self, key, value):
|
||||||
|
if key == "enabled":
|
||||||
|
self.enabled = value
|
||||||
|
elif key == "name":
|
||||||
|
self.name = value
|
||||||
|
elif key == "description":
|
||||||
|
self.description = value
|
||||||
|
elif key == "file_name":
|
||||||
|
if self._file_name != value:
|
||||||
|
self.read_file(value)
|
||||||
|
elif key == "coordinates":
|
||||||
|
self.coordinates = value
|
||||||
|
elif key == "coordinates_bottom" or key == "bottom":
|
||||||
|
self.coordinates["bottom"] = value
|
||||||
|
elif key == "coordinates_top" or key == "top":
|
||||||
|
self.coordinates["top"] = value
|
||||||
|
elif key == "coordinates_left" or key == "left":
|
||||||
|
self.coordinates["left"] = value
|
||||||
|
elif key == "coordinates_right" or key == "right":
|
||||||
|
self.coordinates["right"] = value
|
||||||
|
|
||||||
|
self.modified()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def enabled(self):
|
||||||
|
return self._enabled
|
||||||
|
|
||||||
|
@enabled.setter
|
||||||
|
def enabled(self, enabled):
|
||||||
|
self._enabled = enabled
|
||||||
|
self.modified()
|
||||||
|
|
||||||
|
def is_enabled(self):
|
||||||
|
return self._enabled
|
||||||
|
|
||||||
|
@property
|
||||||
|
def name(self):
|
||||||
|
return self._name
|
||||||
|
|
||||||
|
@name.setter
|
||||||
|
def name(self, name):
|
||||||
|
self._name = name
|
||||||
|
self.modified()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def description(self):
|
||||||
|
return self._description
|
||||||
|
|
||||||
|
@description.setter
|
||||||
|
def description(self, description):
|
||||||
|
self._description = description
|
||||||
|
self.modified()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def file_name(self):
|
||||||
|
return self._file_name
|
||||||
|
|
||||||
|
@file_name.setter
|
||||||
|
def file_name(self, file_name):
|
||||||
|
self._file_name = file_name
|
||||||
|
self.modified()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def coordinates(self):
|
||||||
|
return self._coordinates
|
||||||
|
|
||||||
|
@coordinates.setter
|
||||||
|
def coordinates(self, coordinates):
|
||||||
|
self._coordinates = coordinates
|
||||||
|
self.modified()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def coord_bottom(self):
|
||||||
|
if self._coordinates is None:
|
||||||
|
return 0.0
|
||||||
|
|
||||||
|
return self._coordinates["bottom"]
|
||||||
|
|
||||||
|
@property
|
||||||
|
def coord_top(self):
|
||||||
|
if self._coordinates is None:
|
||||||
|
return 0.0
|
||||||
|
|
||||||
|
return self._coordinates["top"]
|
||||||
|
|
||||||
|
@property
|
||||||
|
def coord_left(self):
|
||||||
|
if self._coordinates is None:
|
||||||
|
return 0.0
|
||||||
|
|
||||||
|
return self._coordinates["left"]
|
||||||
|
|
||||||
|
@property
|
||||||
|
def coord_right(self):
|
||||||
|
if self._coordinates is None:
|
||||||
|
return 0.0
|
||||||
|
|
||||||
|
return self._coordinates["right"]
|
||||||
|
|
||||||
|
@property
|
||||||
|
def memfile(self):
|
||||||
|
if not _rasterio_loaded:
|
||||||
|
return None
|
||||||
|
|
||||||
|
if self._file_bytes == b'':
|
||||||
|
return None
|
||||||
|
|
||||||
|
if self._memfile is None:
|
||||||
|
self._memfile = MemoryFile()
|
||||||
|
self._memfile.write(self._file_bytes)
|
||||||
|
|
||||||
|
return self._memfile
|
||||||
|
|
||||||
|
def read_file(self, path):
|
||||||
|
logger.debug(f"Read GeoTIFF file at : '{path}'")
|
||||||
|
|
||||||
|
self._file_name = path
|
||||||
|
self._file_bytes = b''
|
||||||
|
self._memfile = None
|
||||||
|
|
||||||
|
nbytes = 0
|
||||||
|
|
||||||
|
with open(path, "rb") as f:
|
||||||
|
while True:
|
||||||
|
data = f.read(4096)
|
||||||
|
if not data:
|
||||||
|
break
|
||||||
|
|
||||||
|
nbytes += len(data)
|
||||||
|
self._file_bytes += data
|
||||||
|
|
||||||
|
logger.debug(f"Read GeoTIFF: {nbytes} bytes readed")
|
||||||
|
|
||||||
|
def write_file(self, path):
|
||||||
|
with open(path, "w+b") as f:
|
||||||
|
f.write(self._file_bytes)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def _db_create(cls, execute, ext=""):
|
||||||
|
execute(f"""
|
||||||
|
CREATE TABLE geotiff{ext} (
|
||||||
|
{cls.create_db_add_pamhyr_id()},
|
||||||
|
enabled BOOLEAN NOT NULL,
|
||||||
|
deleted BOOLEAN NOT NULL DEFAULT FALSE,
|
||||||
|
name TEXT NOT NULL,
|
||||||
|
description TEXT NOT NULL,
|
||||||
|
file_name TEXT NOT NULL,
|
||||||
|
file_bytes BLOB NOT NULL,
|
||||||
|
coordinates_bottom REAL NOT NULL,
|
||||||
|
coordinates_top REAL NOT NULL,
|
||||||
|
coordinates_left REAL NOT NULL,
|
||||||
|
coordinates_right REAL NOT NULL,
|
||||||
|
{Scenario.create_db_add_scenario()},
|
||||||
|
{Scenario.create_db_add_scenario_fk()},
|
||||||
|
PRIMARY KEY(pamhyr_id, scenario)
|
||||||
|
)
|
||||||
|
""")
|
||||||
|
|
||||||
|
return cls._create_submodel(execute)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def _db_update(cls, execute, version, data=None):
|
||||||
|
major, minor, release = version.strip().split(".")
|
||||||
|
|
||||||
|
if major == "0" and int(minor) < 2:
|
||||||
|
cls._db_create(execute)
|
||||||
|
|
||||||
|
if major == "0" and int(minor) == 2:
|
||||||
|
if int(release) < 3:
|
||||||
|
cls._db_create(execute)
|
||||||
|
|
||||||
|
return True
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def _db_load(cls, execute, data=None):
|
||||||
|
new = []
|
||||||
|
scenario = data["scenario"]
|
||||||
|
loaded = data['loaded_pid']
|
||||||
|
|
||||||
|
if scenario is None:
|
||||||
|
return new
|
||||||
|
|
||||||
|
table = execute(
|
||||||
|
"SELECT pamhyr_id, enabled, deleted, " +
|
||||||
|
"name, description, file_name, file_bytes, " +
|
||||||
|
"coordinates_bottom, coordinates_top, " +
|
||||||
|
"coordinates_left, coordinates_right, " +
|
||||||
|
"scenario " +
|
||||||
|
"FROM geotiff " +
|
||||||
|
f"WHERE scenario = {scenario.id} " +
|
||||||
|
f"AND pamhyr_id NOT IN ({', '.join(map(str, loaded))})"
|
||||||
|
)
|
||||||
|
|
||||||
|
if table is not None:
|
||||||
|
for row in table:
|
||||||
|
it = iter(row)
|
||||||
|
|
||||||
|
id = next(it)
|
||||||
|
enabled = (next(it) == 1)
|
||||||
|
deleted = (next(it) == 1)
|
||||||
|
name = next(it)
|
||||||
|
description = next(it)
|
||||||
|
file_name = next(it)
|
||||||
|
file_bytes = next(it)
|
||||||
|
coordinates_bottom = next(it)
|
||||||
|
coordinates_top = next(it)
|
||||||
|
coordinates_left = next(it)
|
||||||
|
coordinates_right = next(it)
|
||||||
|
owner_scenario = next(it)
|
||||||
|
|
||||||
|
f = cls(
|
||||||
|
id=id, enabled=enabled, name=name,
|
||||||
|
description=description, coordinates={
|
||||||
|
"bottom": coordinates_bottom,
|
||||||
|
"top": coordinates_top,
|
||||||
|
"left": coordinates_left,
|
||||||
|
"right": coordinates_right,
|
||||||
|
},
|
||||||
|
status=data['status'],
|
||||||
|
owner_scenario=owner_scenario
|
||||||
|
)
|
||||||
|
if deleted:
|
||||||
|
f.set_as_deleted()
|
||||||
|
|
||||||
|
f._file_bytes = file_bytes
|
||||||
|
|
||||||
|
loaded.add(id)
|
||||||
|
new.append(f)
|
||||||
|
|
||||||
|
data["scenario"] = scenario.parent
|
||||||
|
new += cls._db_load(execute, data)
|
||||||
|
data["scenario"] = scenario
|
||||||
|
|
||||||
|
return new
|
||||||
|
|
||||||
|
def _db_save(self, execute, data=None):
|
||||||
|
if not self.must_be_saved():
|
||||||
|
return True
|
||||||
|
|
||||||
|
execute(
|
||||||
|
"INSERT INTO geotiff (" +
|
||||||
|
"pamhyr_id, enabled, deleted, " +
|
||||||
|
"name, description, file_name, file_bytes, " +
|
||||||
|
"coordinates_bottom, coordinates_top, " +
|
||||||
|
"coordinates_left, coordinates_right, " +
|
||||||
|
"scenario) " +
|
||||||
|
"VALUES (?,?,?,?,?,?,?,?,?,?,?,?)",
|
||||||
|
self._pamhyr_id,
|
||||||
|
self._enabled,
|
||||||
|
self.is_deleted(),
|
||||||
|
self.name,
|
||||||
|
self.description,
|
||||||
|
self.file_name,
|
||||||
|
self._file_bytes,
|
||||||
|
self.coordinates['bottom'],
|
||||||
|
self.coordinates['top'],
|
||||||
|
self.coordinates['left'],
|
||||||
|
self.coordinates['right'],
|
||||||
|
self._status.scenario_id,
|
||||||
|
)
|
||||||
|
|
||||||
|
return True
|
||||||
@@ -0,0 +1,59 @@
|
|||||||
|
# GeoTIFFList.py -- Pamhyr
|
||||||
|
# Copyright (C) 2024-2025 INRAE
|
||||||
|
#
|
||||||
|
# This program is free software: you can redistribute it and/or modify
|
||||||
|
# it under the terms of the GNU General Public License as published by
|
||||||
|
# the Free Software Foundation, either version 3 of the License, or
|
||||||
|
# (at your option) any later version.
|
||||||
|
#
|
||||||
|
# This program is distributed in the hope that it will be useful,
|
||||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
# GNU General Public License for more details.
|
||||||
|
#
|
||||||
|
# You should have received a copy of the GNU General Public License
|
||||||
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
from tools import trace, timer
|
||||||
|
|
||||||
|
from Model.Except import NotImplementedMethodeError
|
||||||
|
from Model.Tools.PamhyrListExt import PamhyrModelList
|
||||||
|
from Model.GeoTIFF.GeoTIFF import GeoTIFF
|
||||||
|
|
||||||
|
|
||||||
|
class GeoTIFFList(PamhyrModelList):
|
||||||
|
_sub_classes = [GeoTIFF]
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def _db_load(cls, execute, data=None):
|
||||||
|
new = cls(status=data["status"])
|
||||||
|
|
||||||
|
new._lst = GeoTIFF._db_load(execute, data)
|
||||||
|
|
||||||
|
return new
|
||||||
|
|
||||||
|
def _db_save(self, execute, data=None):
|
||||||
|
ok = True
|
||||||
|
|
||||||
|
# Delete previous data
|
||||||
|
execute(
|
||||||
|
"DELETE FROM geotiff " +
|
||||||
|
f"WHERE scenario = {self._status.scenario_id}"
|
||||||
|
)
|
||||||
|
|
||||||
|
for gt in self._lst:
|
||||||
|
ok &= gt._db_save(execute, data)
|
||||||
|
|
||||||
|
return ok
|
||||||
|
|
||||||
|
@property
|
||||||
|
def files(self):
|
||||||
|
return self.lst
|
||||||
|
|
||||||
|
def new(self, index):
|
||||||
|
n = GeoTIFF(status=self._status)
|
||||||
|
self.insert(index, n)
|
||||||
|
self._status.modified()
|
||||||
|
return n
|
||||||
@@ -66,6 +66,11 @@ class PointXYZ(Point):
|
|||||||
)
|
)
|
||||||
""")
|
""")
|
||||||
|
|
||||||
|
execute(
|
||||||
|
"CREATE INDEX idx_point " +
|
||||||
|
"ON geometry_pointXYZ(profile, scenario);"
|
||||||
|
)
|
||||||
|
|
||||||
return cls._create_submodel(execute)
|
return cls._create_submodel(execute)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
@@ -90,6 +95,13 @@ class PointXYZ(Point):
|
|||||||
"ADD COLUMN deleted BOOLEAN NOT NULL DEFAULT FALSE"
|
"ADD COLUMN deleted BOOLEAN NOT NULL DEFAULT FALSE"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
if major == "0" and minor == "2":
|
||||||
|
if int(release) < 5:
|
||||||
|
execute(
|
||||||
|
"CREATE INDEX idx_point " +
|
||||||
|
"ON geometry_pointXYZ(profile, scenario);"
|
||||||
|
)
|
||||||
|
|
||||||
return cls._update_submodel(execute, version, data)
|
return cls._update_submodel(execute, version, data)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
@@ -338,6 +350,13 @@ class PointXYZ(Point):
|
|||||||
self._z = float(value)
|
self._z = float(value)
|
||||||
self.modified()
|
self.modified()
|
||||||
|
|
||||||
|
def shift(self, x, y, z):
|
||||||
|
self.x += x
|
||||||
|
self.y += y
|
||||||
|
self.z += z
|
||||||
|
|
||||||
|
self.modified()
|
||||||
|
|
||||||
def get_coordinate(self):
|
def get_coordinate(self):
|
||||||
return (self._x, self._y, self._z)
|
return (self._x, self._y, self._z)
|
||||||
|
|
||||||
|
|||||||
@@ -76,11 +76,31 @@ class Profile(object):
|
|||||||
return self.points[index]
|
return self.points[index]
|
||||||
|
|
||||||
def named_point(self, name):
|
def named_point(self, name):
|
||||||
|
if name == "un":
|
||||||
|
return self.points[0]
|
||||||
|
elif name == "np":
|
||||||
|
return self.points[-1]
|
||||||
|
elif name == "talweg":
|
||||||
|
zm = self.z_min()
|
||||||
|
return next((p for p in self.points if p.z == zm), None)
|
||||||
|
else:
|
||||||
return next((p for p in self.points if p.name == name), None)
|
return next((p for p in self.points if p.name == name), None)
|
||||||
|
|
||||||
def named_point_index(self, name):
|
def named_point_index(self, name):
|
||||||
|
if name == "un":
|
||||||
|
return 0
|
||||||
|
elif name == "np":
|
||||||
|
return len(self.points) - 1
|
||||||
|
elif name == "talweg":
|
||||||
|
zm = self.z_min()
|
||||||
return next(
|
return next(
|
||||||
(p for p in enumerate(self.points) if p[1].name == name), None
|
(p for p in enumerate(self.points) if p[1].z == zm),
|
||||||
|
None
|
||||||
|
)[0]
|
||||||
|
else:
|
||||||
|
return next(
|
||||||
|
(p for p in enumerate(self.points) if p[1].name == name),
|
||||||
|
None
|
||||||
)[0]
|
)[0]
|
||||||
|
|
||||||
@property
|
@property
|
||||||
|
|||||||
@@ -24,7 +24,7 @@ from typing import List
|
|||||||
from functools import reduce
|
from functools import reduce
|
||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
|
|
||||||
from tools import timer, flatten
|
from tools import trace, timer, flatten
|
||||||
from shapely import geometry
|
from shapely import geometry
|
||||||
|
|
||||||
from Model.Tools.PamhyrDB import SQLSubModel
|
from Model.Tools.PamhyrDB import SQLSubModel
|
||||||
@@ -396,7 +396,6 @@ class ProfileXYZ(Profile, SQLSubModel):
|
|||||||
|
|
||||||
for point in self._points:
|
for point in self._points:
|
||||||
p = point.copy()
|
p = point.copy()
|
||||||
print(p)
|
|
||||||
new_p._points.append(p)
|
new_p._points.append(p)
|
||||||
|
|
||||||
new_p.modified()
|
new_p.modified()
|
||||||
@@ -1074,16 +1073,10 @@ class ProfileXYZ(Profile, SQLSubModel):
|
|||||||
|
|
||||||
def shift(self, x, y, z):
|
def shift(self, x, y, z):
|
||||||
for p in self._points:
|
for p in self._points:
|
||||||
p.x = p.x + x
|
p.shift(x, y, z)
|
||||||
p.y = p.y + y
|
|
||||||
p.z = p.z + z
|
|
||||||
|
|
||||||
self.modified()
|
self.modified()
|
||||||
|
|
||||||
def modified(self):
|
|
||||||
self.tab_up_to_date = False
|
|
||||||
self.station_up_to_date = False
|
|
||||||
|
|
||||||
def add_npoints(self, npoints):
|
def add_npoints(self, npoints):
|
||||||
# add npoints in a profile
|
# add npoints in a profile
|
||||||
for k in range(npoints):
|
for k in range(npoints):
|
||||||
@@ -1101,12 +1094,11 @@ class ProfileXYZ(Profile, SQLSubModel):
|
|||||||
self.point(i+1).z = 0.5 * self.point(i).z + 0.5 * self.point(i+2).z
|
self.point(i+1).z = 0.5 * self.point(i).z + 0.5 * self.point(i+2).z
|
||||||
|
|
||||||
def copy(self):
|
def copy(self):
|
||||||
p = ProfileXYZ(id=self.id,
|
p = ProfileXYZ(name=self.name,
|
||||||
name=self.name,
|
|
||||||
rk=self.rk,
|
rk=self.rk,
|
||||||
reach=self.reach,
|
reach=self.reach,
|
||||||
num=self.num,
|
|
||||||
status=self._status)
|
status=self._status)
|
||||||
|
|
||||||
for i, k in enumerate(self.points):
|
for i, k in enumerate(self.points):
|
||||||
p.insert_point(i, k.copy())
|
p.insert_point(i, k.copy())
|
||||||
|
|
||||||
|
|||||||
@@ -52,6 +52,11 @@ class Reach(SQLSubModel):
|
|||||||
self._guidelines_is_valid = False
|
self._guidelines_is_valid = False
|
||||||
self._guidelines = {}
|
self._guidelines = {}
|
||||||
|
|
||||||
|
super(Reach, self).__init__(
|
||||||
|
id=parent.pamhyr_id,
|
||||||
|
status=status,
|
||||||
|
)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def pamhyr_id(self):
|
def pamhyr_id(self):
|
||||||
return self._parent.pamhyr_id
|
return self._parent.pamhyr_id
|
||||||
@@ -60,6 +65,10 @@ class Reach(SQLSubModel):
|
|||||||
def _pamhyr_id(self):
|
def _pamhyr_id(self):
|
||||||
return self._parent.pamhyr_id
|
return self._parent.pamhyr_id
|
||||||
|
|
||||||
|
@_pamhyr_id.setter
|
||||||
|
def _pamhyr_id(self, id):
|
||||||
|
return
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def _db_create(cls, execute):
|
def _db_create(cls, execute):
|
||||||
return cls._create_submodel(execute)
|
return cls._create_submodel(execute)
|
||||||
@@ -103,6 +112,9 @@ class Reach(SQLSubModel):
|
|||||||
predicate=lambda obj, data: True,
|
predicate=lambda obj, data: True,
|
||||||
modifier=lambda obj, data: None,
|
modifier=lambda obj, data: None,
|
||||||
data={}):
|
data={}):
|
||||||
|
if predicate(self, data):
|
||||||
|
modifier(self, data)
|
||||||
|
|
||||||
for p in self._profiles:
|
for p in self._profiles:
|
||||||
p._data_traversal(predicate, modifier, data)
|
p._data_traversal(predicate, modifier, data)
|
||||||
|
|
||||||
@@ -195,13 +207,16 @@ class Reach(SQLSubModel):
|
|||||||
"""
|
"""
|
||||||
if profile in self._profiles:
|
if profile in self._profiles:
|
||||||
self.undelete([profile])
|
self.undelete([profile])
|
||||||
|
profile.num = index
|
||||||
else:
|
else:
|
||||||
self._profiles.insert(index, profile)
|
gi = self.get_global_profil_index(index)
|
||||||
|
profile.num = gi
|
||||||
|
self._profiles.insert(gi, profile)
|
||||||
|
|
||||||
self.modified()
|
self.modified()
|
||||||
|
|
||||||
def delete(self, indexes):
|
def delete(self, indexes):
|
||||||
"""Delete some elements in profile list
|
"""Set some elements as deleted in profile list
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
indexes: The list of index to delete
|
indexes: The list of index to delete
|
||||||
@@ -209,11 +224,12 @@ class Reach(SQLSubModel):
|
|||||||
Returns:
|
Returns:
|
||||||
Nothing.
|
Nothing.
|
||||||
"""
|
"""
|
||||||
|
gi = [self.get_global_profil_index(i) for i in indexes]
|
||||||
list(
|
list(
|
||||||
map(
|
map(
|
||||||
lambda p: p[1].set_as_deleted(),
|
lambda p: p[1].set_as_deleted(),
|
||||||
filter(
|
filter(
|
||||||
lambda e: e[0] in indexes,
|
lambda e: e[0] in gi,
|
||||||
enumerate(self._profiles)
|
enumerate(self._profiles)
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
@@ -229,11 +245,12 @@ class Reach(SQLSubModel):
|
|||||||
Returns:
|
Returns:
|
||||||
Nothing.
|
Nothing.
|
||||||
"""
|
"""
|
||||||
|
gi = [self.get_global_profil_index(i) for i in indexes]
|
||||||
self._profiles = list(
|
self._profiles = list(
|
||||||
map(
|
map(
|
||||||
lambda p: p[1],
|
lambda p: p[1],
|
||||||
filter(
|
filter(
|
||||||
lambda e: e[0] not in indexes,
|
lambda e: e[0] not in gi,
|
||||||
enumerate(self._profiles)
|
enumerate(self._profiles)
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
@@ -241,7 +258,7 @@ class Reach(SQLSubModel):
|
|||||||
self.modified()
|
self.modified()
|
||||||
|
|
||||||
def delete_profiles(self, profiles):
|
def delete_profiles(self, profiles):
|
||||||
"""Delete some elements in profile list
|
"""Set profiles list as deleted
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
profiles: The list of profile to delete
|
profiles: The list of profile to delete
|
||||||
@@ -929,3 +946,30 @@ class Reach(SQLSubModel):
|
|||||||
nr2._profiles = list(map(lambda p: p.cloned_for(nr2), nr2_profiles))
|
nr2._profiles = list(map(lambda p: p.cloned_for(nr2), nr2_profiles))
|
||||||
|
|
||||||
return nr1, nr2
|
return nr1, nr2
|
||||||
|
|
||||||
|
def get_local_profil_index(self, index):
|
||||||
|
local_index = -1
|
||||||
|
for i, p in enumerate(self._profiles):
|
||||||
|
if not p.is_deleted():
|
||||||
|
local_index += 1
|
||||||
|
if i == index:
|
||||||
|
break
|
||||||
|
|
||||||
|
return local_index
|
||||||
|
|
||||||
|
def get_global_profil_index(self, index):
|
||||||
|
local_index = -1
|
||||||
|
for i, p in enumerate(self._profiles):
|
||||||
|
if not p.is_deleted():
|
||||||
|
local_index += 1
|
||||||
|
if local_index == index:
|
||||||
|
global_index = i
|
||||||
|
break
|
||||||
|
|
||||||
|
return global_index
|
||||||
|
|
||||||
|
def update_rks(self, rks):
|
||||||
|
for rk, profile in zip(rks, self.profiles):
|
||||||
|
profile.rk = rk
|
||||||
|
|
||||||
|
self.modified()
|
||||||
@@ -183,7 +183,6 @@ class BasicHS(SQLSubModel):
|
|||||||
bhs._data = BHSValue._db_load(
|
bhs._data = BHSValue._db_load(
|
||||||
execute, data
|
execute, data
|
||||||
)
|
)
|
||||||
print(f"{bhs_pid} : {deleted}")
|
|
||||||
if deleted:
|
if deleted:
|
||||||
bhs.set_as_deleted()
|
bhs.set_as_deleted()
|
||||||
|
|
||||||
|
|||||||
@@ -274,6 +274,9 @@ class HydraulicStructure(SQLSubModel):
|
|||||||
return hs
|
return hs
|
||||||
|
|
||||||
def _db_save(self, execute, data=None):
|
def _db_save(self, execute, data=None):
|
||||||
|
if not self.must_be_saved():
|
||||||
|
return True
|
||||||
|
|
||||||
execute(
|
execute(
|
||||||
"DELETE FROM hydraulic_structures " +
|
"DELETE FROM hydraulic_structures " +
|
||||||
f"WHERE pamhyr_id = {self.pamhyr_id} " +
|
f"WHERE pamhyr_id = {self.pamhyr_id} " +
|
||||||
|
|||||||
@@ -225,6 +225,8 @@ class Data(SQLSubModel):
|
|||||||
return new
|
return new
|
||||||
|
|
||||||
def _db_save(self, execute, data=None):
|
def _db_save(self, execute, data=None):
|
||||||
|
if not self.must_be_saved():
|
||||||
|
return True
|
||||||
ind = data["ind"]
|
ind = data["ind"]
|
||||||
|
|
||||||
execute(
|
execute(
|
||||||
@@ -396,7 +398,7 @@ class InitialConditions(SQLSubModel):
|
|||||||
|
|
||||||
@reach.setter
|
@reach.setter
|
||||||
def reach(self, new):
|
def reach(self, new):
|
||||||
self._reach = reach
|
self._reach = new
|
||||||
self.modified()
|
self.modified()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
@@ -537,7 +539,10 @@ class InitialConditions(SQLSubModel):
|
|||||||
strickler = 25.0
|
strickler = 25.0
|
||||||
|
|
||||||
if not compute_discharge:
|
if not compute_discharge:
|
||||||
|
if profile.rk in data_discharge:
|
||||||
discharge = data_discharge[profile.rk]
|
discharge = data_discharge[profile.rk]
|
||||||
|
else:
|
||||||
|
discharge = 0.0
|
||||||
else:
|
else:
|
||||||
discharge = (
|
discharge = (
|
||||||
# ((width * 0.8)
|
# ((width * 0.8)
|
||||||
@@ -583,8 +588,6 @@ class InitialConditions(SQLSubModel):
|
|||||||
for data in self._data:
|
for data in self._data:
|
||||||
data_height[data["rk"].rk] = data["height"]
|
data_height[data["rk"].rk] = data["height"]
|
||||||
|
|
||||||
print(data_height)
|
|
||||||
|
|
||||||
incline = self._reach.reach.get_incline_median_mean()
|
incline = self._reach.reach.get_incline_median_mean()
|
||||||
logger.debug(f"incline = {incline}")
|
logger.debug(f"incline = {incline}")
|
||||||
|
|
||||||
@@ -605,7 +608,10 @@ class InitialConditions(SQLSubModel):
|
|||||||
strickler = 25.0
|
strickler = 25.0
|
||||||
|
|
||||||
if not compute_height:
|
if not compute_height:
|
||||||
|
if profile.rk in data_height:
|
||||||
height = data_height[profile.rk]
|
height = data_height[profile.rk]
|
||||||
|
else:
|
||||||
|
height = 0.0
|
||||||
else:
|
else:
|
||||||
if abs(incline) <= 0:
|
if abs(incline) <= 0:
|
||||||
height = 0.0
|
height = 0.0
|
||||||
@@ -656,7 +662,10 @@ class InitialConditions(SQLSubModel):
|
|||||||
for profile in profiles:
|
for profile in profiles:
|
||||||
|
|
||||||
if not compute_discharge:
|
if not compute_discharge:
|
||||||
|
if profile.rk in data_discharge:
|
||||||
d = data_discharge[profile.rk]
|
d = data_discharge[profile.rk]
|
||||||
|
else:
|
||||||
|
d = 0.0
|
||||||
else:
|
else:
|
||||||
d = discharge
|
d = discharge
|
||||||
elevation = interp(profile.rk,
|
elevation = interp(profile.rk,
|
||||||
|
|||||||
@@ -79,16 +79,12 @@ class InitialConditionsAdisTS(SQLSubModel):
|
|||||||
@classmethod
|
@classmethod
|
||||||
def _db_update(cls, execute, version, data=None):
|
def _db_update(cls, execute, version, data=None):
|
||||||
major, minor, release = version.strip().split(".")
|
major, minor, release = version.strip().split(".")
|
||||||
created = False
|
|
||||||
|
|
||||||
if major == "0" and int(minor) <= 1:
|
|
||||||
if int(release) < 6:
|
|
||||||
cls._db_create(execute)
|
|
||||||
created = True
|
|
||||||
|
|
||||||
if major == "0" and int(minor) < 2:
|
if major == "0" and int(minor) < 2:
|
||||||
if not created:
|
if cls.is_table_exists(execute, "initial_conditions_adists"):
|
||||||
cls._db_update_to_0_2_0(execute, data)
|
cls._db_update_to_0_2_0(execute, data)
|
||||||
|
else:
|
||||||
|
cls._db_create(execute)
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
@@ -267,7 +263,7 @@ class InitialConditionsAdisTS(SQLSubModel):
|
|||||||
@name.setter
|
@name.setter
|
||||||
def name(self, name):
|
def name(self, name):
|
||||||
self._name = name
|
self._name = name
|
||||||
self._status.modified()
|
self.modified()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def pollutant(self):
|
def pollutant(self):
|
||||||
@@ -276,7 +272,7 @@ class InitialConditionsAdisTS(SQLSubModel):
|
|||||||
@pollutant.setter
|
@pollutant.setter
|
||||||
def pollutant(self, pollutant):
|
def pollutant(self, pollutant):
|
||||||
self._pollutant = pollutant
|
self._pollutant = pollutant
|
||||||
self._status.modified()
|
self.modified()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def concentration(self):
|
def concentration(self):
|
||||||
@@ -285,7 +281,7 @@ class InitialConditionsAdisTS(SQLSubModel):
|
|||||||
@concentration.setter
|
@concentration.setter
|
||||||
def concentration(self, concentration):
|
def concentration(self, concentration):
|
||||||
self._concentration = concentration
|
self._concentration = concentration
|
||||||
self._status.modified()
|
self.modified()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def eg(self):
|
def eg(self):
|
||||||
@@ -294,7 +290,7 @@ class InitialConditionsAdisTS(SQLSubModel):
|
|||||||
@eg.setter
|
@eg.setter
|
||||||
def eg(self, eg):
|
def eg(self, eg):
|
||||||
self._eg = eg
|
self._eg = eg
|
||||||
self._status.modified()
|
self.modified()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def em(self):
|
def em(self):
|
||||||
@@ -303,7 +299,7 @@ class InitialConditionsAdisTS(SQLSubModel):
|
|||||||
@em.setter
|
@em.setter
|
||||||
def em(self, em):
|
def em(self, em):
|
||||||
self._em = em
|
self._em = em
|
||||||
self._status.modified()
|
self.modified()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def ed(self):
|
def ed(self):
|
||||||
@@ -312,7 +308,7 @@ class InitialConditionsAdisTS(SQLSubModel):
|
|||||||
@ed.setter
|
@ed.setter
|
||||||
def ed(self, ed):
|
def ed(self, ed):
|
||||||
self._ed = ed
|
self._ed = ed
|
||||||
self._status.modified()
|
self.modified()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def enabled(self):
|
def enabled(self):
|
||||||
@@ -321,28 +317,45 @@ class InitialConditionsAdisTS(SQLSubModel):
|
|||||||
@enabled.setter
|
@enabled.setter
|
||||||
def enabled(self, enabled):
|
def enabled(self, enabled):
|
||||||
self._enabled = enabled
|
self._enabled = enabled
|
||||||
self._status.modified()
|
self.modified()
|
||||||
|
|
||||||
def new(self, index):
|
def new(self, index):
|
||||||
n = ICAdisTSSpec(status=self._status)
|
n = ICAdisTSSpec(status=self._status)
|
||||||
self._data.insert(index, n)
|
self._data.insert(index, n)
|
||||||
self._status.modified()
|
self.modified()
|
||||||
return n
|
return n
|
||||||
|
|
||||||
def delete(self, data):
|
def delete(self, data):
|
||||||
self._data = list(
|
list(
|
||||||
filter(
|
map(
|
||||||
lambda x: x not in data,
|
lambda x: x.set_as_deleted(),
|
||||||
self._data
|
data
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
self._status.modified()
|
self.modified()
|
||||||
|
|
||||||
def delete_i(self, indexes):
|
def delete_i(self, indexes):
|
||||||
for ind in indexes:
|
list(
|
||||||
del self._data[ind]
|
map(
|
||||||
self._status.modified()
|
lambda e: e[1].set_as_deleted(),
|
||||||
|
filter(
|
||||||
|
lambda e: e[0] in indexes,
|
||||||
|
enumerate(self._data)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
self.modified()
|
||||||
|
|
||||||
def insert(self, index, data):
|
def insert(self, index, data):
|
||||||
|
if data in self._data:
|
||||||
|
self.undelete([data])
|
||||||
|
else:
|
||||||
self._data.insert(index, data)
|
self._data.insert(index, data)
|
||||||
self._status.modified()
|
|
||||||
|
self.modified()
|
||||||
|
|
||||||
|
def undelete(self, lst):
|
||||||
|
for x in lst:
|
||||||
|
x.set_as_not_deleted()
|
||||||
|
|
||||||
|
self.modified()
|
||||||
@@ -31,7 +31,7 @@ class ICAdisTSSpec(SQLSubModel):
|
|||||||
_sub_classes = []
|
_sub_classes = []
|
||||||
|
|
||||||
def __init__(self, id: int = -1, name: str = "",
|
def __init__(self, id: int = -1, name: str = "",
|
||||||
status=None):
|
status=None, owner_scenario=None):
|
||||||
super(ICAdisTSSpec, self).__init__()
|
super(ICAdisTSSpec, self).__init__()
|
||||||
|
|
||||||
self._status = status
|
self._status = status
|
||||||
@@ -77,16 +77,12 @@ class ICAdisTSSpec(SQLSubModel):
|
|||||||
@classmethod
|
@classmethod
|
||||||
def _db_update(cls, execute, version, data=None):
|
def _db_update(cls, execute, version, data=None):
|
||||||
major, minor, release = version.strip().split(".")
|
major, minor, release = version.strip().split(".")
|
||||||
created = False
|
|
||||||
|
|
||||||
if major == 0 and minor < 1:
|
|
||||||
if int(release) < 6:
|
|
||||||
cls._db_create(execute)
|
|
||||||
created = True
|
|
||||||
|
|
||||||
if major == "0" and int(minor) < 2:
|
if major == "0" and int(minor) < 2:
|
||||||
if not created:
|
if cls.is_table_exists(execute, "initial_conditions_adists_spec"):
|
||||||
cls._db_update_to_0_2_0(execute, data)
|
cls._db_update_to_0_2_0(execute, data)
|
||||||
|
else:
|
||||||
|
cls._db_create(execute)
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
@@ -194,7 +190,7 @@ class ICAdisTSSpec(SQLSubModel):
|
|||||||
new_spec.rate = rate
|
new_spec.rate = rate
|
||||||
new_spec.enabled = enabled
|
new_spec.enabled = enabled
|
||||||
|
|
||||||
loaded.add(pid)
|
# loaded.add(pid)
|
||||||
new.append(new_spec)
|
new.append(new_spec)
|
||||||
|
|
||||||
data["scenario"] = scenario.parent
|
data["scenario"] = scenario.parent
|
||||||
@@ -211,7 +207,7 @@ class ICAdisTSSpec(SQLSubModel):
|
|||||||
|
|
||||||
sql = (
|
sql = (
|
||||||
"INSERT INTO " +
|
"INSERT INTO " +
|
||||||
"initial_conditions_adists_spec(id, deleted, " +
|
"initial_conditions_adists_spec(pamhyr_id, deleted, " +
|
||||||
"ic_default, name, reach, " +
|
"ic_default, name, reach, " +
|
||||||
"start_rk, end_rk, concentration, eg, em, ed, rate, " +
|
"start_rk, end_rk, concentration, eg, em, ed, rate, " +
|
||||||
"enabled, scenario) " +
|
"enabled, scenario) " +
|
||||||
|
|||||||
@@ -147,9 +147,9 @@ class Data(SQLSubModel):
|
|||||||
it = iter(v)
|
it = iter(v)
|
||||||
|
|
||||||
pid = next(it)
|
pid = next(it)
|
||||||
delete = next(it)
|
deleted = next(it)
|
||||||
data0 = bc._types[0](next(it))
|
data0 = lc._types[0](next(it))
|
||||||
data1 = bc._types[1](next(it))
|
data1 = lc._types[1](next(it))
|
||||||
owner_scenario = next(it)
|
owner_scenario = next(it)
|
||||||
|
|
||||||
nd = cls(
|
nd = cls(
|
||||||
|
|||||||
@@ -69,14 +69,12 @@ class Data(SQLSubModel):
|
|||||||
major, minor, release = version.strip().split(".")
|
major, minor, release = version.strip().split(".")
|
||||||
created = False
|
created = False
|
||||||
|
|
||||||
if major == "0" and int(minor) <= 1:
|
|
||||||
if int(release) < 7:
|
|
||||||
cls._db_create(execute)
|
|
||||||
created = True
|
|
||||||
|
|
||||||
if major == "0" and int(minor) < 2:
|
if major == "0" and int(minor) < 2:
|
||||||
if not created:
|
if cls.is_table_exists(
|
||||||
|
execute, "lateral_contribution_data_adists"):
|
||||||
cls._db_update_to_0_2_0(execute, data)
|
cls._db_update_to_0_2_0(execute, data)
|
||||||
|
else:
|
||||||
|
cls._db_create(execute)
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
@@ -132,11 +130,13 @@ class Data(SQLSubModel):
|
|||||||
if scenario is None:
|
if scenario is None:
|
||||||
return new
|
return new
|
||||||
|
|
||||||
|
lca = data["lca"]
|
||||||
|
|
||||||
table = execute(
|
table = execute(
|
||||||
"SELECT pamhyr_id, deleted, data0, data1 " +
|
"SELECT pamhyr_id, deleted, data0, data1, scenario " +
|
||||||
"FROM lateral_contribution_data_adists " +
|
"FROM lateral_contribution_data_adists " +
|
||||||
f"WHERE scenario = {scenario.id} " +
|
f"WHERE scenario = {scenario.id} " +
|
||||||
f"AND pamhyr_id NOT IN ({', '.join(map(str, loaded))}) "
|
f"AND pamhyr_id NOT IN ({', '.join(map(str, loaded))}) " +
|
||||||
f"AND lca = '{lca.id}'"
|
f"AND lca = '{lca.id}'"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -150,16 +150,16 @@ class Data(SQLSubModel):
|
|||||||
data1 = next(it)
|
data1 = next(it)
|
||||||
owner_scenario = next(it)
|
owner_scenario = next(it)
|
||||||
|
|
||||||
data = cls(
|
lc = cls(
|
||||||
data0, data1,
|
data0, data1,
|
||||||
id=pid, status=status,
|
id=pid, status=status,
|
||||||
owner_scenario=owner_scenario
|
owner_scenario=owner_scenario
|
||||||
)
|
)
|
||||||
if deleted:
|
if deleted:
|
||||||
data.set_as_deleted()
|
lc.set_as_deleted()
|
||||||
|
|
||||||
loaded.add(pid)
|
loaded.add(pid)
|
||||||
new.append(data)
|
new.append(lc)
|
||||||
|
|
||||||
data["scenario"] = scenario.parent
|
data["scenario"] = scenario.parent
|
||||||
new += cls._db_load(execute, data)
|
new += cls._db_load(execute, data)
|
||||||
@@ -175,7 +175,7 @@ class Data(SQLSubModel):
|
|||||||
|
|
||||||
execute(
|
execute(
|
||||||
"INSERT INTO " +
|
"INSERT INTO " +
|
||||||
"lateral_contribution_adists(pamhyr_id, deleted," +
|
"lateral_contribution_data_adists(pamhyr_id, deleted," +
|
||||||
"data0, data1, lca, scenario) " +
|
"data0, data1, lca, scenario) " +
|
||||||
"VALUES (" +
|
"VALUES (" +
|
||||||
f"{self.id}, {self._db_format(self.is_deleted())}, " +
|
f"{self.id}, {self._db_format(self.is_deleted())}, " +
|
||||||
@@ -241,15 +241,13 @@ class LateralContributionAdisTS(SQLSubModel):
|
|||||||
major, minor, release = version.strip().split(".")
|
major, minor, release = version.strip().split(".")
|
||||||
created = False
|
created = False
|
||||||
|
|
||||||
if major == "0" and int(minor) <= 1:
|
if major == "0" and int(minor) < 2:
|
||||||
if int(release) < 7:
|
if cls.is_table_exists(execute, "lateral_contribution_adists"):
|
||||||
|
cls._db_update_to_0_2_0(execute, data)
|
||||||
|
else:
|
||||||
cls._db_create(execute)
|
cls._db_create(execute)
|
||||||
created = True
|
created = True
|
||||||
|
|
||||||
if major == "0" and int(minor) < 2:
|
|
||||||
if not created:
|
|
||||||
cls._db_update_to_0_2_0(execute, data)
|
|
||||||
|
|
||||||
if not created:
|
if not created:
|
||||||
return cls._update_submodel(execute, version, data)
|
return cls._update_submodel(execute, version, data)
|
||||||
|
|
||||||
@@ -372,16 +370,18 @@ class LateralContributionAdisTS(SQLSubModel):
|
|||||||
|
|
||||||
execute(
|
execute(
|
||||||
"INSERT INTO " +
|
"INSERT INTO " +
|
||||||
"lateral_contribution_adists(id, deleted," +
|
"lateral_contribution_adists(pamhyr_id, deleted, " +
|
||||||
"pollutant, reach, begin_rk, end_rk, scenario) " +
|
"pollutant, reach, begin_rk, end_rk, scenario) " +
|
||||||
"VALUES (" +
|
"VALUES (" +
|
||||||
f"{self.id}, {self._db_format(self.is_deleted())}, " +
|
f"{self.id}, {self._db_format(self.is_deleted())}, " +
|
||||||
f"{self._pollutant}, {self.reach}, " +
|
f"{self._pollutant}, {self.reach}, " +
|
||||||
f"{self._begin_rk}, {self._end_rk}" +
|
f"{self._begin_rk}, {self._end_rk}, " +
|
||||||
f"{self._status.scenario_id}" +
|
f"{self._status.scenario_id}" +
|
||||||
")"
|
")"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
data["lca"] = self
|
||||||
|
|
||||||
ind = 0
|
ind = 0
|
||||||
for d in self._data:
|
for d in self._data:
|
||||||
data["ind"] = ind
|
data["ind"] = ind
|
||||||
@@ -406,7 +406,7 @@ class LateralContributionAdisTS(SQLSubModel):
|
|||||||
return len(
|
return len(
|
||||||
list(
|
list(
|
||||||
filter(
|
filter(
|
||||||
lambda el: not el.is_deleted(),
|
lambda el: el is not None and not el.is_deleted(),
|
||||||
self._data
|
self._data
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
@@ -452,7 +452,12 @@ class LateralContributionAdisTS(SQLSubModel):
|
|||||||
|
|
||||||
@property
|
@property
|
||||||
def data(self):
|
def data(self):
|
||||||
return self._data.copy()
|
return list(
|
||||||
|
filter(
|
||||||
|
lambda el: el is not None and not el.is_deleted(),
|
||||||
|
self._data
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def begin_rk(self):
|
def begin_rk(self):
|
||||||
@@ -471,3 +476,57 @@ class LateralContributionAdisTS(SQLSubModel):
|
|||||||
def end_rk(self, end_rk):
|
def end_rk(self, end_rk):
|
||||||
self._end_rk = end_rk
|
self._end_rk = end_rk
|
||||||
self.modified()
|
self.modified()
|
||||||
|
|
||||||
|
@property
|
||||||
|
def _default_0(self):
|
||||||
|
return self._types[0](0)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def _default_1(self):
|
||||||
|
return self._types[1](0.0)
|
||||||
|
|
||||||
|
def add(self, index: int):
|
||||||
|
value = Data(self._default_0, self._default_1, status=self._status)
|
||||||
|
self._data.insert(index, value)
|
||||||
|
self.modified()
|
||||||
|
return value
|
||||||
|
|
||||||
|
def insert(self, index: int, data: Data):
|
||||||
|
self._data.insert(index, data)
|
||||||
|
self.modified()
|
||||||
|
|
||||||
|
def delete_i(self, indexes):
|
||||||
|
self._data = list(
|
||||||
|
map(
|
||||||
|
lambda e: e[1].set_as_deleted(),
|
||||||
|
filter(
|
||||||
|
lambda e: e[0] not in indexes,
|
||||||
|
enumerate(self.data)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
self.modified()
|
||||||
|
|
||||||
|
def index(self, bc):
|
||||||
|
self._data.index(bc)
|
||||||
|
|
||||||
|
def get_i(self, index: int):
|
||||||
|
return self._data[index]
|
||||||
|
|
||||||
|
def get_range(self, _range):
|
||||||
|
lst = []
|
||||||
|
for r in _range:
|
||||||
|
lst.append(r)
|
||||||
|
return lst
|
||||||
|
|
||||||
|
def _set_i_c_v(self, index: int, column: int, value):
|
||||||
|
v = self._data[index]
|
||||||
|
v[column] = self._types[column](value)
|
||||||
|
self._data[index] = v
|
||||||
|
self.modified()
|
||||||
|
|
||||||
|
def set_i_0(self, index: int, value):
|
||||||
|
self._set_i_c_v(index, 0, value)
|
||||||
|
|
||||||
|
def set_i_1(self, index: int, value):
|
||||||
|
self._set_i_c_v(index, 1, value)
|
||||||
@@ -95,16 +95,12 @@ class OutputRKAdists(SQLSubModel):
|
|||||||
@classmethod
|
@classmethod
|
||||||
def _db_update(cls, execute, version, data=None):
|
def _db_update(cls, execute, version, data=None):
|
||||||
major, minor, release = version.strip().split(".")
|
major, minor, release = version.strip().split(".")
|
||||||
created = False
|
|
||||||
|
|
||||||
if major == "0" and int(minor) <= 1:
|
|
||||||
if int(release) < 7:
|
|
||||||
cls._db_create(execute)
|
|
||||||
created = True
|
|
||||||
|
|
||||||
if major == "0" and int(minor) < 2:
|
if major == "0" and int(minor) < 2:
|
||||||
if not created:
|
if cls.is_table_exists(execute, "OutputRKAdists"):
|
||||||
cls._db_update_to_0_2_0(execute, data)
|
cls._db_update_to_0_2_0(execute, data)
|
||||||
|
else:
|
||||||
|
cls._db_create(execute)
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
@@ -193,7 +189,7 @@ class OutputRKAdists(SQLSubModel):
|
|||||||
"output_rk_adists(pamhyr_id, deleted, " +
|
"output_rk_adists(pamhyr_id, deleted, " +
|
||||||
"reach, rk, title, scenario) " +
|
"reach, rk, title, scenario) " +
|
||||||
"VALUES (" +
|
"VALUES (" +
|
||||||
f"{self.id}, {self._db_format(self.is_deleted())}" +
|
f"{self.id}, {self._db_format(self.is_deleted())}, " +
|
||||||
f"{self._reach}, {self._rk}, " +
|
f"{self._reach}, {self._rk}, " +
|
||||||
f"'{self._db_format(self._title)}', " +
|
f"'{self._db_format(self._title)}', " +
|
||||||
f"{self._status.scenario_id}" +
|
f"{self._status.scenario_id}" +
|
||||||
|
|||||||
@@ -28,6 +28,9 @@ from tools import (
|
|||||||
from Model.Tools.PamhyrDB import SQLSubModel
|
from Model.Tools.PamhyrDB import SQLSubModel
|
||||||
from Model.Except import NotImplementedMethodeError
|
from Model.Except import NotImplementedMethodeError
|
||||||
from Model.Scenario import Scenario
|
from Model.Scenario import Scenario
|
||||||
|
from Model.BoundaryConditionsAdisTS.BoundaryConditionAdisTS import (
|
||||||
|
BoundaryConditionAdisTS,
|
||||||
|
)
|
||||||
|
|
||||||
logger = logging.getLogger()
|
logger = logging.getLogger()
|
||||||
|
|
||||||
@@ -101,6 +104,12 @@ class PollutantCharacteristics(SQLSubModel):
|
|||||||
self._ac = value
|
self._ac = value
|
||||||
elif key == 8:
|
elif key == 8:
|
||||||
self._bc = value
|
self._bc = value
|
||||||
|
self.propagate_modified()
|
||||||
|
|
||||||
|
def propagate_modified(self):
|
||||||
|
self.modified()
|
||||||
|
if hasattr(self, "_pollutant") and self._pollutant:
|
||||||
|
self._pollutant.modified()
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def _db_create(cls, execute, ext=""):
|
def _db_create(cls, execute, ext=""):
|
||||||
@@ -129,16 +138,12 @@ class PollutantCharacteristics(SQLSubModel):
|
|||||||
@classmethod
|
@classmethod
|
||||||
def _db_update(cls, execute, version, data=None):
|
def _db_update(cls, execute, version, data=None):
|
||||||
major, minor, release = version.strip().split(".")
|
major, minor, release = version.strip().split(".")
|
||||||
created = False
|
|
||||||
|
|
||||||
if major == "0" and int(minor) <= 1:
|
|
||||||
if int(release) < 7:
|
|
||||||
cls._db_create(execute)
|
|
||||||
created = True
|
|
||||||
|
|
||||||
if major == "0" and int(minor) < 2:
|
if major == "0" and int(minor) < 2:
|
||||||
if not created:
|
if cls.is_table_exists(execute, "Pollutants_characteristics"):
|
||||||
cls._db_update_to_0_2_0(execute, data)
|
cls._db_update_to_0_2_0(execute, data)
|
||||||
|
else:
|
||||||
|
cls._db_create(execute)
|
||||||
|
|
||||||
return cls._update_submodel(execute, version, data)
|
return cls._update_submodel(execute, version, data)
|
||||||
|
|
||||||
@@ -253,6 +258,12 @@ class PollutantCharacteristics(SQLSubModel):
|
|||||||
if not self.must_be_saved():
|
if not self.must_be_saved():
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
execute(
|
||||||
|
"DELETE FROM pollutants_characteristics " +
|
||||||
|
f"WHERE pamhyr_id = {self.id} " +
|
||||||
|
f"AND scenario = {self._status.scenario_id}"
|
||||||
|
)
|
||||||
|
|
||||||
execute(
|
execute(
|
||||||
"INSERT INTO " +
|
"INSERT INTO " +
|
||||||
"pollutants_characteristics(pamhyr_id, deleted, " +
|
"pollutants_characteristics(pamhyr_id, deleted, " +
|
||||||
@@ -275,7 +286,10 @@ class PollutantCharacteristics(SQLSubModel):
|
|||||||
|
|
||||||
|
|
||||||
class Pollutants(SQLSubModel):
|
class Pollutants(SQLSubModel):
|
||||||
_sub_classes = [PollutantCharacteristics]
|
_sub_classes = [
|
||||||
|
PollutantCharacteristics,
|
||||||
|
# BoundaryConditionAdisTS
|
||||||
|
]
|
||||||
|
|
||||||
def __init__(self, id: int = -1, name: str = "",
|
def __init__(self, id: int = -1, name: str = "",
|
||||||
status=None, owner_scenario=-1):
|
status=None, owner_scenario=-1):
|
||||||
@@ -293,6 +307,11 @@ class Pollutants(SQLSubModel):
|
|||||||
self._enabled = True
|
self._enabled = True
|
||||||
|
|
||||||
self._data = []
|
self._data = []
|
||||||
|
self._boundary_conditions_adists = []
|
||||||
|
|
||||||
|
@property
|
||||||
|
def boundary_conditions_adists(self):
|
||||||
|
return self._boundary_conditions_adists
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def name(self):
|
def name(self):
|
||||||
@@ -301,18 +320,37 @@ class Pollutants(SQLSubModel):
|
|||||||
@name.setter
|
@name.setter
|
||||||
def name(self, name):
|
def name(self, name):
|
||||||
self._name = name
|
self._name = name
|
||||||
self._status.modified()
|
self.modified()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def data(self):
|
def data(self):
|
||||||
return self._data
|
return self._data
|
||||||
|
|
||||||
|
@data.setter
|
||||||
|
def data(self, data):
|
||||||
|
self._data = [
|
||||||
|
PollutantCharacteristics(
|
||||||
|
type=int(data[0]),
|
||||||
|
diametre=float(data[1]),
|
||||||
|
rho=float(data[2]),
|
||||||
|
porosity=float(data[3]),
|
||||||
|
cdc_riv=float(data[4]),
|
||||||
|
cdc_cas=float(data[5]),
|
||||||
|
apd=float(data[6]),
|
||||||
|
ac=float(data[7]),
|
||||||
|
bc=float(data[8]),
|
||||||
|
pollutant=self,
|
||||||
|
status=self._status
|
||||||
|
)
|
||||||
|
]
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def _db_create(cls, execute, ext=""):
|
def _db_create(cls, execute, ext=""):
|
||||||
execute(f"""
|
execute(f"""
|
||||||
CREATE TABLE pollutants{ext}(
|
CREATE TABLE pollutants{ext}(
|
||||||
{cls.create_db_add_pamhyr_id()},
|
{cls.create_db_add_pamhyr_id()},
|
||||||
deleted BOOLEAN NOT NULL DEFAULT FALSE,
|
deleted BOOLEAN NOT NULL DEFAULT FALSE,
|
||||||
|
enabled BOOLEAN NOT NULL DEFAULT TRUE,
|
||||||
name TEXT NOT NULL,
|
name TEXT NOT NULL,
|
||||||
{Scenario.create_db_add_scenario()},
|
{Scenario.create_db_add_scenario()},
|
||||||
{Scenario.create_db_add_scenario_fk()}
|
{Scenario.create_db_add_scenario_fk()}
|
||||||
@@ -329,14 +367,20 @@ class Pollutants(SQLSubModel):
|
|||||||
major, minor, release = version.strip().split(".")
|
major, minor, release = version.strip().split(".")
|
||||||
created = False
|
created = False
|
||||||
|
|
||||||
if major == "0" and int(minor) <= 1:
|
if major == "0" and int(minor) < 2:
|
||||||
if int(release) < 7:
|
if cls.is_table_exists(execute, "Pollutants"):
|
||||||
|
cls._db_update_to_0_2_0(execute, data)
|
||||||
|
else:
|
||||||
cls._db_create(execute)
|
cls._db_create(execute)
|
||||||
created = True
|
created = True
|
||||||
|
|
||||||
if major == "0" and int(minor) < 2:
|
if major == "0" and minor == "2":
|
||||||
if not created:
|
if int(release) < 6:
|
||||||
cls._db_update_to_0_2_0(execute, data)
|
execute(f"ALTER TABLE pollutants " +
|
||||||
|
f"ADD COLUMN enabled BOOLEAN NOT NULL DEFAULT TRUE")
|
||||||
|
|
||||||
|
# cls._db_update_to_0_2_5(execute, data)
|
||||||
|
# # created = True
|
||||||
|
|
||||||
if not created:
|
if not created:
|
||||||
return cls._update_submodel(execute, version, data)
|
return cls._update_submodel(execute, version, data)
|
||||||
@@ -375,7 +419,8 @@ class Pollutants(SQLSubModel):
|
|||||||
return new
|
return new
|
||||||
|
|
||||||
table = execute(
|
table = execute(
|
||||||
"SELECT pamhyr_id, deleted, name FROM pollutants " +
|
"SELECT pamhyr_id, deleted, enabled, name, scenario " +
|
||||||
|
"FROM pollutants " +
|
||||||
f"WHERE scenario = {scenario.id} " +
|
f"WHERE scenario = {scenario.id} " +
|
||||||
f"AND pamhyr_id NOT IN ({', '.join(map(str, loaded))})"
|
f"AND pamhyr_id NOT IN ({', '.join(map(str, loaded))})"
|
||||||
)
|
)
|
||||||
@@ -386,18 +431,30 @@ class Pollutants(SQLSubModel):
|
|||||||
|
|
||||||
pid = next(it)
|
pid = next(it)
|
||||||
deleted = (next(it) == 1)
|
deleted = (next(it) == 1)
|
||||||
|
enabled = (next(it) == 1)
|
||||||
name = next(it)
|
name = next(it)
|
||||||
|
owner_scenario = next(it)
|
||||||
|
|
||||||
new_pollutant = cls(
|
new_pollutant = cls(
|
||||||
id=pid, name=name,
|
id=pid, name=name,
|
||||||
status=status
|
status=status,
|
||||||
|
owner_scenario=owner_scenario
|
||||||
)
|
)
|
||||||
if deleted:
|
if deleted:
|
||||||
new_pollutant.set_as_deleted()
|
new_pollutant.set_as_deleted()
|
||||||
|
if not enabled:
|
||||||
|
new_pollutant.enabled = False
|
||||||
|
|
||||||
data["pollutant"] = new_pollutant
|
data["pollutant"] = new_pollutant
|
||||||
new_pollutant._data = PollutantCharacteristics._db_load(
|
new_pollutant._data = PollutantCharacteristics._db_load(
|
||||||
execute, data.copy()
|
execute, data=data
|
||||||
|
)
|
||||||
|
|
||||||
|
new_pollutant._boundary_conditions_adists = (
|
||||||
|
BoundaryConditionAdisTS._db_load(
|
||||||
|
execute,
|
||||||
|
data=data
|
||||||
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
loaded.add(pid)
|
loaded.add(pid)
|
||||||
@@ -421,16 +478,17 @@ class Pollutants(SQLSubModel):
|
|||||||
f"AND scenario = {self._status.scenario_id}"
|
f"AND scenario = {self._status.scenario_id}"
|
||||||
)
|
)
|
||||||
execute(
|
execute(
|
||||||
f"DELETE FROM pollutants_characteristics " +
|
"DELETE FROM pollutants_characteristics " +
|
||||||
f"WHERE pollutant = {self.id} " +
|
f"WHERE pollutant = {self.id} " +
|
||||||
f"AND scenario = {self._status.scenario_id}"
|
f"AND scenario = {self._status.scenario_id}"
|
||||||
)
|
)
|
||||||
|
|
||||||
execute(
|
execute(
|
||||||
"INSERT INTO " +
|
"INSERT INTO " +
|
||||||
"pollutants(pamhyr_id, deleted, name, scenario) " +
|
"pollutants(pamhyr_id, deleted, enabled, name, scenario) " +
|
||||||
"VALUES (" +
|
"VALUES (" +
|
||||||
f"{self.id}, {self._db_format(self.is_deleted())}, " +
|
f"{self.id}, {self._db_format(self.is_deleted())}, " +
|
||||||
|
f"{self._db_format(self._enabled)}, " +
|
||||||
f"'{self._db_format(self._name)}', " +
|
f"'{self._db_format(self._name)}', " +
|
||||||
f"{self._status.scenario_id}" +
|
f"{self._status.scenario_id}" +
|
||||||
")"
|
")"
|
||||||
@@ -439,6 +497,9 @@ class Pollutants(SQLSubModel):
|
|||||||
for d in self._data:
|
for d in self._data:
|
||||||
ok &= d._db_save(execute, data)
|
ok &= d._db_save(execute, data)
|
||||||
|
|
||||||
|
for bc in self._boundary_conditions_adists:
|
||||||
|
ok &= bc._db_save(execute, data)
|
||||||
|
|
||||||
return ok
|
return ok
|
||||||
|
|
||||||
def _data_traversal(self,
|
def _data_traversal(self,
|
||||||
|
|||||||
@@ -19,7 +19,7 @@
|
|||||||
from tools import trace, timer
|
from tools import trace, timer
|
||||||
|
|
||||||
from Model.Except import NotImplementedMethodeError
|
from Model.Except import NotImplementedMethodeError
|
||||||
from Model.Tools.PamhyrList import PamhyrModelList
|
from Model.Tools.PamhyrListExt import PamhyrModelList
|
||||||
from Model.Pollutants.Pollutants import Pollutants
|
from Model.Pollutants.Pollutants import Pollutants
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
@@ -263,8 +263,6 @@ class Reservoir(SQLSubModel):
|
|||||||
cls._db_update_to_0_2_0_set_node_pid(execute, table, nodes)
|
cls._db_update_to_0_2_0_set_node_pid(execute, table, nodes)
|
||||||
Scenario.update_db_add_scenario(execute, table)
|
Scenario.update_db_add_scenario(execute, table)
|
||||||
|
|
||||||
print(execute(f"SELECT * FROM {table}"))
|
|
||||||
|
|
||||||
cls._db_create(execute, ext="_tmp")
|
cls._db_create(execute, ext="_tmp")
|
||||||
execute(
|
execute(
|
||||||
f"INSERT INTO {table}_tmp " +
|
f"INSERT INTO {table}_tmp " +
|
||||||
@@ -310,11 +308,11 @@ class Reservoir(SQLSubModel):
|
|||||||
new_reservoir.set_as_deleted()
|
new_reservoir.set_as_deleted()
|
||||||
|
|
||||||
new_reservoir._node = None
|
new_reservoir._node = None
|
||||||
if node_id != -1:
|
if node_id != -1 and node_id is not None:
|
||||||
new_reservoir._node = next(
|
new_reservoir._node = next(
|
||||||
filter(
|
filter(
|
||||||
lambda n: n.id == node_id, data["nodes"]
|
lambda n: n.id == node_id, data["nodes"]
|
||||||
)
|
), None
|
||||||
)
|
)
|
||||||
|
|
||||||
data["reservoir"] = new_reservoir
|
data["reservoir"] = new_reservoir
|
||||||
|
|||||||
@@ -29,8 +29,134 @@ from Model.Results.River.River import River
|
|||||||
logger = logging.getLogger()
|
logger = logging.getLogger()
|
||||||
|
|
||||||
|
|
||||||
|
class AdditionalData(SQLSubModel):
|
||||||
|
_sub_classes = []
|
||||||
|
|
||||||
|
def __init__(self, id=-1, study=None, data=None):
|
||||||
|
super(AdditionalData, self).__init__(
|
||||||
|
id=id, status=study.status,
|
||||||
|
owner_scenario=study.status.scenario.id
|
||||||
|
)
|
||||||
|
|
||||||
|
self._study = study
|
||||||
|
self._data = data
|
||||||
|
|
||||||
|
@property
|
||||||
|
def data(self):
|
||||||
|
return self._data
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def _db_create(cls, execute, ext=""):
|
||||||
|
execute(f"""
|
||||||
|
CREATE TABLE results_add_data{ext} (
|
||||||
|
{cls.create_db_add_pamhyr_id()},
|
||||||
|
result INTEGER NOT NULL,
|
||||||
|
type_x TEXT NOT NULL,
|
||||||
|
type_y TEXT NOT NULL,
|
||||||
|
legend TEXT NOT NULL,
|
||||||
|
unit TEXT NOT NULL,
|
||||||
|
data_len INTEGER NOT NULL,
|
||||||
|
x BLOB NOT NULL,
|
||||||
|
y BLOB NOT NULL,
|
||||||
|
{Scenario.create_db_add_scenario()},
|
||||||
|
{Scenario.create_db_add_scenario_fk()},
|
||||||
|
FOREIGN KEY(result) REFERENCES results(pamhyr_id),
|
||||||
|
PRIMARY KEY(pamhyr_id, result, scenario)
|
||||||
|
)
|
||||||
|
""")
|
||||||
|
|
||||||
|
if ext != "":
|
||||||
|
return True
|
||||||
|
|
||||||
|
return cls._create_submodel(execute)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def _db_update(cls, execute, version, data=None):
|
||||||
|
major, minor, release = version.strip().split(".")
|
||||||
|
|
||||||
|
if major == "0" and int(minor) == 2 and int(release) <= 1:
|
||||||
|
cls._db_create(execute)
|
||||||
|
|
||||||
|
return cls._update_submodel(execute, version, data)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def _db_load(cls, execute, data=None):
|
||||||
|
new = []
|
||||||
|
|
||||||
|
study = data['study']
|
||||||
|
status = data['status']
|
||||||
|
scenario = data["scenario"]
|
||||||
|
|
||||||
|
table = execute(
|
||||||
|
"SELECT pamhyr_id, type_x, type_y, " +
|
||||||
|
"legend, unit, data_len, x, y, " +
|
||||||
|
"scenario " +
|
||||||
|
"FROM results_add_data " +
|
||||||
|
f"WHERE scenario = {scenario.id}"
|
||||||
|
)
|
||||||
|
|
||||||
|
if table is None:
|
||||||
|
return new
|
||||||
|
|
||||||
|
for v in table:
|
||||||
|
it = iter(v)
|
||||||
|
|
||||||
|
pid = next(it)
|
||||||
|
type_x = next(it)
|
||||||
|
type_y = next(it)
|
||||||
|
legend = next(it)
|
||||||
|
unit = next(it)
|
||||||
|
data_len = next(it)
|
||||||
|
bx = next(it)
|
||||||
|
by = next(it)
|
||||||
|
owner_scenario = next(it)
|
||||||
|
|
||||||
|
data_format = ">" + ''.join(itertools.repeat("d", data_len))
|
||||||
|
x = struct.unpack(data_format, bx)
|
||||||
|
y = struct.unpack(data_format, by)
|
||||||
|
|
||||||
|
data = {
|
||||||
|
'type_x': type_x,
|
||||||
|
'type_y': type_y,
|
||||||
|
'legend': legend,
|
||||||
|
'unit': unit,
|
||||||
|
'x': x, 'y': y
|
||||||
|
}
|
||||||
|
|
||||||
|
new_data = cls(study=study)
|
||||||
|
new_data._data = data
|
||||||
|
new.append(new_data)
|
||||||
|
|
||||||
|
return new
|
||||||
|
|
||||||
|
def _db_save(self, execute, data=None):
|
||||||
|
if self._status.scenario.id != self._owner_scenario:
|
||||||
|
return
|
||||||
|
|
||||||
|
pid = self._pamhyr_id
|
||||||
|
data_len = len(self._data["x"])
|
||||||
|
|
||||||
|
data_format = ">" + ''.join(itertools.repeat("d", data_len))
|
||||||
|
bx = struct.pack(data_format, *self._data["x"])
|
||||||
|
by = struct.pack(data_format, *self._data["y"])
|
||||||
|
|
||||||
|
execute(
|
||||||
|
"INSERT INTO " +
|
||||||
|
"results_add_data (pamhyr_id, result, " +
|
||||||
|
"type_x, type_y, " +
|
||||||
|
"legend, unit, data_len, x, y, " +
|
||||||
|
"scenario) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)",
|
||||||
|
self._pamhyr_id, data["result"],
|
||||||
|
self._data["type_x"], self._data["type_y"],
|
||||||
|
self._data["legend"], self._data["unit"],
|
||||||
|
data_len, bx, by, self._owner_scenario
|
||||||
|
)
|
||||||
|
|
||||||
|
return True
|
||||||
|
|
||||||
|
|
||||||
class Results(SQLSubModel):
|
class Results(SQLSubModel):
|
||||||
_sub_classes = [River]
|
_sub_classes = [River, AdditionalData]
|
||||||
|
|
||||||
def __init__(self, id=-1, study=None, solver=None,
|
def __init__(self, id=-1, study=None, solver=None,
|
||||||
repertory="", name="0"):
|
repertory="", name="0"):
|
||||||
@@ -50,6 +176,7 @@ class Results(SQLSubModel):
|
|||||||
# Keep results creation date
|
# Keep results creation date
|
||||||
"creation_date": datetime.now(),
|
"creation_date": datetime.now(),
|
||||||
"study_revision": study.status.version,
|
"study_revision": study.status.version,
|
||||||
|
"additional_data": [],
|
||||||
}
|
}
|
||||||
|
|
||||||
if solver is not None:
|
if solver is not None:
|
||||||
@@ -92,6 +219,11 @@ class Results(SQLSubModel):
|
|||||||
qlog=None,
|
qlog=None,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def bufferize(self, key):
|
||||||
|
if self.is_valid:
|
||||||
|
for reach in self._river._reachs:
|
||||||
|
reach.bufferize(self._meta_data["timestamps"], key)
|
||||||
|
|
||||||
def timestamps_to_struct(self):
|
def timestamps_to_struct(self):
|
||||||
ts = self._meta_data["timestamps"]
|
ts = self._meta_data["timestamps"]
|
||||||
sf = ">" + ''.join(itertools.repeat("d", len(ts)))
|
sf = ">" + ''.join(itertools.repeat("d", len(ts)))
|
||||||
@@ -184,6 +316,11 @@ class Results(SQLSubModel):
|
|||||||
data["timestamps"] = sorted(ts)
|
data["timestamps"] = sorted(ts)
|
||||||
new_results._river = River._db_load(execute, data)
|
new_results._river = River._db_load(execute, data)
|
||||||
|
|
||||||
|
new_results.set(
|
||||||
|
"additional_data",
|
||||||
|
AdditionalData._db_load(execute, data)
|
||||||
|
)
|
||||||
|
|
||||||
yield (solver_type, new_results)
|
yield (solver_type, new_results)
|
||||||
|
|
||||||
def _db_save_clear(self, execute, solver_type, data=None):
|
def _db_save_clear(self, execute, solver_type, data=None):
|
||||||
@@ -206,10 +343,21 @@ class Results(SQLSubModel):
|
|||||||
f"WHERE scenario = {self._owner_scenario} " +
|
f"WHERE scenario = {self._owner_scenario} " +
|
||||||
f"AND result = {pid}"
|
f"AND result = {pid}"
|
||||||
)
|
)
|
||||||
|
execute(
|
||||||
|
"DELETE FROM results_add_data " +
|
||||||
|
f"WHERE scenario = {self._owner_scenario} " +
|
||||||
|
f"AND result = {pid}"
|
||||||
|
)
|
||||||
|
|
||||||
|
def _is_valid(self):
|
||||||
|
return "timestamps" in self._meta_data
|
||||||
|
|
||||||
def _db_save(self, execute, data=None):
|
def _db_save(self, execute, data=None):
|
||||||
if self._status.scenario.id != self._owner_scenario:
|
if self._status.scenario.id != self._owner_scenario:
|
||||||
return
|
return True
|
||||||
|
|
||||||
|
if not self._is_valid():
|
||||||
|
return False
|
||||||
|
|
||||||
pid = self._pamhyr_id
|
pid = self._pamhyr_id
|
||||||
if self._solver is None:
|
if self._solver is None:
|
||||||
@@ -238,4 +386,7 @@ class Results(SQLSubModel):
|
|||||||
data["result"] = self._pamhyr_id
|
data["result"] = self._pamhyr_id
|
||||||
self._river._db_save(execute, data)
|
self._river._db_save(execute, data)
|
||||||
|
|
||||||
|
for add_data in self.get("additional_data"):
|
||||||
|
add_data._db_save(execute, data)
|
||||||
|
|
||||||
return True
|
return True
|
||||||
@@ -21,12 +21,13 @@ import os
|
|||||||
from copy import deepcopy
|
from copy import deepcopy
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
|
||||||
|
from Model.Tools.PamhyrDB import SQLSubModel
|
||||||
from Model.Results.River.RiverAdisTS import River
|
from Model.Results.River.RiverAdisTS import River
|
||||||
|
|
||||||
logger = logging.getLogger()
|
logger = logging.getLogger()
|
||||||
|
|
||||||
|
|
||||||
class Results(object):
|
class Results(SQLSubModel):
|
||||||
def __init__(self, study=None, solver=None,
|
def __init__(self, study=None, solver=None,
|
||||||
repertory="", name="0", type_pol=None):
|
repertory="", name="0", type_pol=None):
|
||||||
self._study = study
|
self._study = study
|
||||||
@@ -50,6 +51,9 @@ class Results(object):
|
|||||||
|
|
||||||
self._phys_var_list = ["C", "G", "M", "D", "L", "N", "R"]
|
self._phys_var_list = ["C", "G", "M", "D", "L", "N", "R"]
|
||||||
|
|
||||||
|
if solver is not None:
|
||||||
|
self.set("solver_type", solver._type)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def date(self):
|
def date(self):
|
||||||
date = self._meta_data["creation_date"]
|
date = self._meta_data["creation_date"]
|
||||||
@@ -87,3 +91,6 @@ class Results(object):
|
|||||||
self._repertory,
|
self._repertory,
|
||||||
qlog=None,
|
qlog=None,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def _db_save(self, execute, data=None):
|
||||||
|
return False
|
||||||
@@ -17,6 +17,7 @@
|
|||||||
import struct
|
import struct
|
||||||
import logging
|
import logging
|
||||||
import itertools
|
import itertools
|
||||||
|
import numpy as np
|
||||||
|
|
||||||
from tools import flatten
|
from tools import flatten
|
||||||
from functools import reduce
|
from functools import reduce
|
||||||
@@ -76,9 +77,12 @@ class Profile(SQLSubModel):
|
|||||||
return self._data[timestamp][key]
|
return self._data[timestamp][key]
|
||||||
return None
|
return None
|
||||||
|
|
||||||
def has_sediment(self):
|
def has_sediment_layers(self):
|
||||||
return any(map(lambda ts: "sl" in self._data[ts], self._data))
|
return any(map(lambda ts: "sl" in self._data[ts], self._data))
|
||||||
|
|
||||||
|
def has_bedload(self):
|
||||||
|
return any(map(lambda ts: "zfd" in self._data[ts], self._data))
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def _db_create(cls, execute, ext=""):
|
def _db_create(cls, execute, ext=""):
|
||||||
execute(f"""
|
execute(f"""
|
||||||
@@ -105,16 +109,6 @@ class Profile(SQLSubModel):
|
|||||||
@classmethod
|
@classmethod
|
||||||
def _db_update(cls, execute, version, data=None):
|
def _db_update(cls, execute, version, data=None):
|
||||||
major, minor, release = version.strip().split(".")
|
major, minor, release = version.strip().split(".")
|
||||||
create = False
|
|
||||||
|
|
||||||
if major == "0" and int(minor) < 2:
|
|
||||||
cls._db_create(execute)
|
|
||||||
create = True
|
|
||||||
|
|
||||||
if major == "0" and int(minor) == 2:
|
|
||||||
if int(release) < 1 and not create:
|
|
||||||
cls._db_create(execute)
|
|
||||||
create = True
|
|
||||||
|
|
||||||
return cls._update_submodel(execute, version, data)
|
return cls._update_submodel(execute, version, data)
|
||||||
|
|
||||||
@@ -285,6 +279,7 @@ class Reach(SQLSubModel):
|
|||||||
lambda p: p.name[0:8] != 'interpol', self._profiles
|
lambda p: p.name[0:8] != 'interpol', self._profiles
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
self._buffers = {}
|
||||||
|
|
||||||
def __len__(self):
|
def __len__(self):
|
||||||
return len(self._profiles)
|
return len(self._profiles)
|
||||||
@@ -305,6 +300,10 @@ class Reach(SQLSubModel):
|
|||||||
def profile_mask(self):
|
def profile_mask(self):
|
||||||
return self._profile_mask
|
return self._profile_mask
|
||||||
|
|
||||||
|
@property
|
||||||
|
def buffers(self):
|
||||||
|
return self._buffers
|
||||||
|
|
||||||
def profile(self, id):
|
def profile(self, id):
|
||||||
return self._profiles[id]
|
return self._profiles[id]
|
||||||
|
|
||||||
@@ -312,7 +311,16 @@ class Reach(SQLSubModel):
|
|||||||
self._profiles[profile_id].set(timestamp, key, data)
|
self._profiles[profile_id].set(timestamp, key, data)
|
||||||
|
|
||||||
def has_sediment(self):
|
def has_sediment(self):
|
||||||
return any(map(lambda profile: profile.has_sediment(), self._profiles))
|
return any(map(lambda profile: profile.has_sediment_layers(),
|
||||||
|
self._profiles))
|
||||||
|
|
||||||
|
def has_bedload(self):
|
||||||
|
return any(map(lambda profile: profile.has_bedload(), self._profiles))
|
||||||
|
|
||||||
|
def bufferize(self, timestamps, key):
|
||||||
|
self._buffers[key] = np.zeros((len(timestamps), len(self)))
|
||||||
|
for i, p in enumerate(self._profiles):
|
||||||
|
self._buffers[key][:, i] = p.get_key(key)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def _db_create(cls, execute, ext=""):
|
def _db_create(cls, execute, ext=""):
|
||||||
|
|||||||
+22
-3
@@ -59,6 +59,7 @@ from Model.LateralContributionsAdisTS.LateralContributionsAdisTSList \
|
|||||||
import LateralContributionsAdisTSList
|
import LateralContributionsAdisTSList
|
||||||
from Model.D90AdisTS.D90AdisTSList import D90AdisTSList
|
from Model.D90AdisTS.D90AdisTSList import D90AdisTSList
|
||||||
from Model.DIFAdisTS.DIFAdisTSList import DIFAdisTSList
|
from Model.DIFAdisTS.DIFAdisTSList import DIFAdisTSList
|
||||||
|
from Model.GeoTIFF.GeoTIFFList import GeoTIFFList
|
||||||
from Model.Results.Results import Results
|
from Model.Results.Results import Results
|
||||||
|
|
||||||
logger = logging.getLogger()
|
logger = logging.getLogger()
|
||||||
@@ -179,6 +180,9 @@ class RiverNode(Node):
|
|||||||
return nodes
|
return nodes
|
||||||
|
|
||||||
def _db_save(self, execute, data=None):
|
def _db_save(self, execute, data=None):
|
||||||
|
if not self.must_be_saved():
|
||||||
|
return True
|
||||||
|
|
||||||
execute(
|
execute(
|
||||||
"INSERT OR REPLACE INTO river_node(" +
|
"INSERT OR REPLACE INTO river_node(" +
|
||||||
"pamhyr_id, deleted, name, x, y, scenario" +
|
"pamhyr_id, deleted, name, x, y, scenario" +
|
||||||
@@ -360,7 +364,7 @@ class RiverReach(Edge):
|
|||||||
)
|
)
|
||||||
new.enable(enable=enabled)
|
new.enable(enable=enabled)
|
||||||
if deleted:
|
if deleted:
|
||||||
nd.set_as_deleted()
|
new.set_as_deleted()
|
||||||
|
|
||||||
data["reach"] = new
|
data["reach"] = new
|
||||||
new._reach = Reach._db_load(execute, data)
|
new._reach = Reach._db_load(execute, data)
|
||||||
@@ -376,6 +380,9 @@ class RiverReach(Edge):
|
|||||||
return reachs
|
return reachs
|
||||||
|
|
||||||
def _db_save(self, execute, data=None):
|
def _db_save(self, execute, data=None):
|
||||||
|
if not self.must_be_saved():
|
||||||
|
return True
|
||||||
|
|
||||||
execute(
|
execute(
|
||||||
"INSERT OR REPLACE INTO " +
|
"INSERT OR REPLACE INTO " +
|
||||||
"river_reach(" +
|
"river_reach(" +
|
||||||
@@ -385,7 +392,7 @@ class RiverReach(Edge):
|
|||||||
"VALUES (" +
|
"VALUES (" +
|
||||||
f"{self.pamhyr_id}, {self._db_format(self.is_deleted())}, " +
|
f"{self.pamhyr_id}, {self._db_format(self.is_deleted())}, " +
|
||||||
f"'{self._db_format(self._name)}', " +
|
f"'{self._db_format(self._name)}', " +
|
||||||
f"{self._db_format(self.is_enable())},"
|
f"{self._db_format(self.is_enable())}," +
|
||||||
f"{self.node1.pamhyr_id}, {self.node2.pamhyr_id}, " +
|
f"{self.node1.pamhyr_id}, {self.node2.pamhyr_id}, " +
|
||||||
f"{self._status.scenario_id}" +
|
f"{self._status.scenario_id}" +
|
||||||
")"
|
")"
|
||||||
@@ -468,6 +475,7 @@ class River(Graph):
|
|||||||
LateralContributionsAdisTSList,
|
LateralContributionsAdisTSList,
|
||||||
D90AdisTSList,
|
D90AdisTSList,
|
||||||
DIFAdisTSList,
|
DIFAdisTSList,
|
||||||
|
GeoTIFFList,
|
||||||
Results
|
Results
|
||||||
]
|
]
|
||||||
|
|
||||||
@@ -505,6 +513,8 @@ class River(Graph):
|
|||||||
self._D90AdisTS = D90AdisTSList(status=self._status)
|
self._D90AdisTS = D90AdisTSList(status=self._status)
|
||||||
self._DIFAdisTS = DIFAdisTSList(status=self._status)
|
self._DIFAdisTS = DIFAdisTSList(status=self._status)
|
||||||
|
|
||||||
|
self._geotiff = GeoTIFFList(status=self._status)
|
||||||
|
|
||||||
self._results = {}
|
self._results = {}
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
@@ -539,7 +549,7 @@ class River(Graph):
|
|||||||
new._nodes = RiverNode._db_load(
|
new._nodes = RiverNode._db_load(
|
||||||
execute, data
|
execute, data
|
||||||
)
|
)
|
||||||
data["nodes"] = new.nodes()
|
data["nodes"] = new._nodes
|
||||||
data['loaded_pid'] = set()
|
data['loaded_pid'] = set()
|
||||||
|
|
||||||
new._edges = RiverReach._db_load(
|
new._edges = RiverReach._db_load(
|
||||||
@@ -617,6 +627,8 @@ class River(Graph):
|
|||||||
|
|
||||||
new._DIFAdisTS = DIFAdisTSList._db_load(execute, data)
|
new._DIFAdisTS = DIFAdisTSList._db_load(execute, data)
|
||||||
|
|
||||||
|
new._geotiff = GeoTIFFList._db_load(execute, data)
|
||||||
|
|
||||||
return new
|
return new
|
||||||
|
|
||||||
def _db_load_results(self, execute, data=None):
|
def _db_load_results(self, execute, data=None):
|
||||||
@@ -650,6 +662,8 @@ class River(Graph):
|
|||||||
objs.append(self._D90AdisTS)
|
objs.append(self._D90AdisTS)
|
||||||
objs.append(self._DIFAdisTS)
|
objs.append(self._DIFAdisTS)
|
||||||
|
|
||||||
|
objs.append(self._geotiff)
|
||||||
|
|
||||||
for solv_type in self.results:
|
for solv_type in self.results:
|
||||||
objs.append(self.results[solv_type])
|
objs.append(self.results[solv_type])
|
||||||
|
|
||||||
@@ -726,6 +740,7 @@ class River(Graph):
|
|||||||
self._BoundaryConditionsAdisTS,
|
self._BoundaryConditionsAdisTS,
|
||||||
self._LateralContributionsAdisTS,
|
self._LateralContributionsAdisTS,
|
||||||
self._D90AdisTS, self._DIFAdisTS,
|
self._D90AdisTS, self._DIFAdisTS,
|
||||||
|
self._geotiff,
|
||||||
]
|
]
|
||||||
|
|
||||||
for solver in self._parameters:
|
for solver in self._parameters:
|
||||||
@@ -818,6 +833,10 @@ Last export at: @date."""
|
|||||||
def additional_files(self):
|
def additional_files(self):
|
||||||
return self._additional_files
|
return self._additional_files
|
||||||
|
|
||||||
|
@property
|
||||||
|
def geotiff(self):
|
||||||
|
return self._geotiff
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def rep_lines(self):
|
def rep_lines(self):
|
||||||
return self._rep_lines
|
return self._rep_lines
|
||||||
|
|||||||
+99
-25
@@ -17,21 +17,56 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
import os
|
import os
|
||||||
|
import logging
|
||||||
|
|
||||||
from tools import logger_exception
|
from tools import logger_exception, timer, flatten
|
||||||
|
|
||||||
from Model.Tools.PamhyrDB import SQLSubModel
|
from Model.Tools.PamhyrDB import SQLSubModel
|
||||||
|
|
||||||
|
logger = logging.getLogger()
|
||||||
|
|
||||||
|
|
||||||
class Scenario(SQLSubModel):
|
class Scenario(SQLSubModel):
|
||||||
_id_cnt = 0
|
_id_cnt = 0
|
||||||
_sub_classes = []
|
_sub_classes = []
|
||||||
|
|
||||||
|
tables_with_deleted_column = [
|
||||||
|
# Adists
|
||||||
|
"output_rk_adists",
|
||||||
|
"boundary_condition_adists", "boundary_condition_data_adists",
|
||||||
|
"lateral_contribution_adists", "lateral_contribution_data_adists",
|
||||||
|
"initial_conditions_adists", "initial_conditions_adists_spec",
|
||||||
|
"d90_adists", "d90_adists_spec",
|
||||||
|
"dif_adists_spec",
|
||||||
|
"pollutants", "pollutants_characteristics",
|
||||||
|
# Hydraulic
|
||||||
|
"additional_files",
|
||||||
|
"boundary_condition", "boundary_condition_data",
|
||||||
|
"lateral_contribution", "lateral_contribution_data",
|
||||||
|
"friction", "stricklers",
|
||||||
|
"hydraulic_structures",
|
||||||
|
"hydraulic_structures_basic",
|
||||||
|
"initial_conditions",
|
||||||
|
"sedimentary_layer", "sedimentary_layer_layer",
|
||||||
|
"reservoir", "reservoir_data",
|
||||||
|
"rep_lines",
|
||||||
|
"geometry_pointXYZ", "geometry_profileXYZ",
|
||||||
|
"river_reach", "river_node",
|
||||||
|
"geotiff", "reservoir", "reservoir_data",
|
||||||
|
]
|
||||||
|
|
||||||
|
related_tables = tables_with_deleted_column + [
|
||||||
|
"dif_adists",
|
||||||
|
"solver_parameter",
|
||||||
|
"hydraulic_structures_basic_value",
|
||||||
|
"results", "results_data", "results_add_data",
|
||||||
|
]
|
||||||
|
|
||||||
def __init__(self,
|
def __init__(self,
|
||||||
id: int = -1,
|
id: int = -1,
|
||||||
name: str = "",
|
name: str = "",
|
||||||
description: str = "",
|
description: str = "",
|
||||||
x: int = 0.0, y: int = 0.0,
|
x: int = 1000.0, y: int = 1000.0,
|
||||||
revision: int = 0,
|
revision: int = 0,
|
||||||
parent=None):
|
parent=None):
|
||||||
super(Scenario, self).__init__()
|
super(Scenario, self).__init__()
|
||||||
@@ -171,29 +206,7 @@ class Scenario(SQLSubModel):
|
|||||||
def drop_all(self, execute):
|
def drop_all(self, execute):
|
||||||
execute(f"DELETE FROM scenario WHERE id = {self.id}")
|
execute(f"DELETE FROM scenario WHERE id = {self.id}")
|
||||||
|
|
||||||
tables = [
|
tables = self.related_tables
|
||||||
# Adists
|
|
||||||
"output_rk_adists",
|
|
||||||
"boundary_condition_adists", "boundary_condition_data_adists",
|
|
||||||
"lateral_contribution_adists", "lateral_contribution_data_adists",
|
|
||||||
"initial_conditions_adists",
|
|
||||||
"d90_adists", "dif_adists",
|
|
||||||
"pollutants", "pollutants_characteristics",
|
|
||||||
# Hydraulic
|
|
||||||
"additional_files",
|
|
||||||
"boundary_condition", "boundary_condition_data",
|
|
||||||
"lateral_contribution", "lateral_contribution_data",
|
|
||||||
"friction", "stricklers",
|
|
||||||
"hydraulic_structures",
|
|
||||||
"hydraulic_structures_basic", "hydraulic_structures_basic_value",
|
|
||||||
"initial_conditions",
|
|
||||||
"sedimentary_layer", "sedimentary_layer_layer",
|
|
||||||
"reservoir", "reservoir_data",
|
|
||||||
"rep_lines",
|
|
||||||
"solver_parameter",
|
|
||||||
"geometry_pointXYZ", "geometry_profileXYZ",
|
|
||||||
"river_reach", "river_node",
|
|
||||||
]
|
|
||||||
|
|
||||||
for table in tables:
|
for table in tables:
|
||||||
execute(
|
execute(
|
||||||
@@ -203,6 +216,67 @@ class Scenario(SQLSubModel):
|
|||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
def get_parent_branch(self):
|
||||||
|
def aux(scenario, acc):
|
||||||
|
if scenario is None:
|
||||||
|
return acc
|
||||||
|
|
||||||
|
return aux(scenario.parent, [scenario.id] + acc)
|
||||||
|
|
||||||
|
return aux(self, [])
|
||||||
|
|
||||||
|
@timer
|
||||||
|
def clean_deleted_data(self, execute):
|
||||||
|
tables = self.tables_with_deleted_column
|
||||||
|
branch = self.get_parent_branch()
|
||||||
|
|
||||||
|
all_ids = []
|
||||||
|
|
||||||
|
for table in tables:
|
||||||
|
if self.parent is None:
|
||||||
|
# This scenario is the default scenario, so we can
|
||||||
|
# delete all data marked as deleted
|
||||||
|
execute(
|
||||||
|
f"DELETE FROM {table} " +
|
||||||
|
f"WHERE scenario = {self.id} " +
|
||||||
|
"AND deleted = TRUE"
|
||||||
|
)
|
||||||
|
continue
|
||||||
|
|
||||||
|
# Select pamhyr_id for each deleted data in this scenario
|
||||||
|
# who do not exists into parents scenarios
|
||||||
|
ids = execute(
|
||||||
|
f"""
|
||||||
|
SELECT pamhyr_id FROM {table}
|
||||||
|
WHERE deleted = TRUE
|
||||||
|
AND scenario = {self.id}
|
||||||
|
AND pamhyr_id NOT IN (
|
||||||
|
SELECT pamhyr_id FROM {table}
|
||||||
|
WHERE scenario IN
|
||||||
|
({', '.join(map(str, branch[:-1]))})
|
||||||
|
)
|
||||||
|
""", fetch_one=False
|
||||||
|
)
|
||||||
|
if ids is None or len(ids) == 0:
|
||||||
|
continue
|
||||||
|
|
||||||
|
ids = flatten(ids)
|
||||||
|
|
||||||
|
logger.debug(
|
||||||
|
f"({self.name}) Drop deleted data into '{table}' : {ids}"
|
||||||
|
)
|
||||||
|
|
||||||
|
execute(
|
||||||
|
f"DELETE FROM {table} " +
|
||||||
|
f"WHERE scenario = {self.id} " +
|
||||||
|
"AND pamhyr_id IN " +
|
||||||
|
f"({', '.join(map(str, ids))})"
|
||||||
|
)
|
||||||
|
|
||||||
|
all_ids += ids
|
||||||
|
|
||||||
|
return all_ids
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def id(self):
|
def id(self):
|
||||||
return self._id
|
return self._id
|
||||||
|
|||||||
@@ -16,6 +16,8 @@
|
|||||||
|
|
||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
from functools import reduce
|
||||||
|
|
||||||
from tools import logger_exception
|
from tools import logger_exception
|
||||||
|
|
||||||
from Model.Tools.PamhyrDict import PamhyrModelDict
|
from Model.Tools.PamhyrDict import PamhyrModelDict
|
||||||
@@ -78,3 +80,12 @@ class Scenarios(PamhyrModelDict):
|
|||||||
|
|
||||||
self._dict[key].set_as_deleted()
|
self._dict[key].set_as_deleted()
|
||||||
self._status.modified()
|
self._status.modified()
|
||||||
|
|
||||||
|
def is_leaf(self, scenario):
|
||||||
|
return not reduce(
|
||||||
|
lambda acc, s: (
|
||||||
|
acc or (not s.is_deleted() and s.parent is scenario)
|
||||||
|
),
|
||||||
|
self._dict.values(),
|
||||||
|
False
|
||||||
|
)
|
||||||
@@ -34,6 +34,7 @@ class StudyStatus(object):
|
|||||||
self._status = Status.EDIT
|
self._status = Status.EDIT
|
||||||
self._scenario = scenario
|
self._scenario = scenario
|
||||||
self._saved = True
|
self._saved = True
|
||||||
|
self._loading = False
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def scenario_id(self):
|
def scenario_id(self):
|
||||||
@@ -86,6 +87,15 @@ class StudyStatus(object):
|
|||||||
def is_saved(self):
|
def is_saved(self):
|
||||||
return self._saved
|
return self._saved
|
||||||
|
|
||||||
|
def is_loading(self):
|
||||||
|
return self._loading
|
||||||
|
|
||||||
|
def start_loading(self):
|
||||||
|
self._loading = True
|
||||||
|
|
||||||
|
def stop_loading(self):
|
||||||
|
self._loading = False
|
||||||
|
|
||||||
def save(self):
|
def save(self):
|
||||||
logger.debug("model status set as saved")
|
logger.debug("model status set as saved")
|
||||||
self._saved = True
|
self._saved = True
|
||||||
@@ -97,6 +107,9 @@ class StudyStatus(object):
|
|||||||
self._saved = True
|
self._saved = True
|
||||||
|
|
||||||
def modified(self):
|
def modified(self):
|
||||||
|
if self._loading:
|
||||||
|
return
|
||||||
|
|
||||||
if self._saved:
|
if self._saved:
|
||||||
self.version += 1
|
self.version += 1
|
||||||
|
|
||||||
|
|||||||
+75
-5
@@ -24,12 +24,21 @@ from datetime import datetime
|
|||||||
from tools import timer, timestamp
|
from tools import timer, timestamp
|
||||||
from functools import reduce
|
from functools import reduce
|
||||||
|
|
||||||
|
from Model.Tools.PamhyrListExt import PamhyrModelList
|
||||||
|
from Model.Tools.PamhyrDict import PamhyrModelDict
|
||||||
from Model.Tools.PamhyrDB import SQLModel
|
from Model.Tools.PamhyrDB import SQLModel
|
||||||
from Model.Scenarios import Scenarios
|
from Model.Scenarios import Scenarios
|
||||||
from Model.Scenario import Scenario
|
from Model.Scenario import Scenario
|
||||||
from Model.Status import StudyStatus
|
from Model.Status import StudyStatus
|
||||||
from Model.Except import NotImplementedMethodeError
|
from Model.Except import NotImplementedMethodeError
|
||||||
from Model.River import River
|
from Model.River import River
|
||||||
|
from Model.Geometry.Reach import Reach
|
||||||
|
from Model.HydraulicStructures.HydraulicStructures import (
|
||||||
|
HydraulicStructure
|
||||||
|
)
|
||||||
|
from Model.HydraulicStructures.Basic.HydraulicStructures import (
|
||||||
|
BasicHS
|
||||||
|
)
|
||||||
|
|
||||||
from Checker.Study import *
|
from Checker.Study import *
|
||||||
|
|
||||||
@@ -37,7 +46,7 @@ logger = logging.getLogger()
|
|||||||
|
|
||||||
|
|
||||||
class Study(SQLModel):
|
class Study(SQLModel):
|
||||||
_version = "0.2.1"
|
_version = "0.2.6"
|
||||||
|
|
||||||
_sub_classes = [
|
_sub_classes = [
|
||||||
Scenario,
|
Scenario,
|
||||||
@@ -449,8 +458,60 @@ class Study(SQLModel):
|
|||||||
[self.scenarios, self._river],
|
[self.scenarios, self._river],
|
||||||
data=progress
|
data=progress
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# Clear DB for current scenario
|
||||||
|
ids = self.status.scenario.clean_deleted_data(self.execute)
|
||||||
|
self.memory_clean(ids)
|
||||||
|
|
||||||
|
progress()
|
||||||
|
|
||||||
self.commit()
|
self.commit()
|
||||||
|
|
||||||
|
@timer
|
||||||
|
def memory_clean(self, ids):
|
||||||
|
if len(ids) == 0:
|
||||||
|
return
|
||||||
|
|
||||||
|
reach_class = Reach
|
||||||
|
hs_classes = [HydraulicStructure, BasicHS]
|
||||||
|
list_classes = set(PamhyrModelList.__subclasses__())
|
||||||
|
dict_classes = set(PamhyrModelDict.__subclasses__())
|
||||||
|
|
||||||
|
def modifier(obj, data):
|
||||||
|
t = type(obj)
|
||||||
|
|
||||||
|
if t is reach_class:
|
||||||
|
obj._profiles = list(
|
||||||
|
filter(
|
||||||
|
lambda el: el.id not in ids,
|
||||||
|
obj._profiles
|
||||||
|
)
|
||||||
|
)
|
||||||
|
elif t in hs_classes:
|
||||||
|
obj._data = list(
|
||||||
|
filter(
|
||||||
|
lambda el: el.id not in ids,
|
||||||
|
obj._data
|
||||||
|
)
|
||||||
|
)
|
||||||
|
elif t in dict_classes:
|
||||||
|
new = {}
|
||||||
|
for key in obj._dict:
|
||||||
|
if obj._dict[key].id not in ids:
|
||||||
|
new[key] = obj._dict[key]
|
||||||
|
obj._dict = new
|
||||||
|
elif t in list_classes:
|
||||||
|
obj._lst = list(
|
||||||
|
filter(
|
||||||
|
lambda el: el.id not in ids,
|
||||||
|
obj._lst
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
self.river._data_traversal(
|
||||||
|
lambda o, d: True, modifier, {}
|
||||||
|
)
|
||||||
|
|
||||||
def sql_save_request_count(self, *args, **kargs):
|
def sql_save_request_count(self, *args, **kargs):
|
||||||
return self._count()
|
return self._count()
|
||||||
|
|
||||||
@@ -459,7 +520,10 @@ class Study(SQLModel):
|
|||||||
[self.scenarios, self._river]
|
[self.scenarios, self._river]
|
||||||
)
|
)
|
||||||
logger.debug(cnt)
|
logger.debug(cnt)
|
||||||
return cnt + 7
|
# HACK: 8 is the number of SQL request do and dont counted in
|
||||||
|
# submodel save_count (let's count sql execute into the
|
||||||
|
# '_save' method)
|
||||||
|
return cnt + 8
|
||||||
|
|
||||||
def close(self):
|
def close(self):
|
||||||
"""Close db connection
|
"""Close db connection
|
||||||
@@ -485,7 +549,8 @@ class Study(SQLModel):
|
|||||||
self.status.scenario = scenario
|
self.status.scenario = scenario
|
||||||
|
|
||||||
if reduce(
|
if reduce(
|
||||||
lambda a, s: a or (s.parent is scenario),
|
lambda a, s: a or (s.parent is scenario
|
||||||
|
and not s.is_deleted()),
|
||||||
self.scenarios.lst,
|
self.scenarios.lst,
|
||||||
False
|
False
|
||||||
):
|
):
|
||||||
@@ -511,16 +576,21 @@ class Study(SQLModel):
|
|||||||
"scenario": scenario
|
"scenario": scenario
|
||||||
}
|
}
|
||||||
|
|
||||||
|
self.status.start_loading()
|
||||||
|
try:
|
||||||
# Load river data
|
# Load river data
|
||||||
river = River._db_load(
|
river = River._db_load(
|
||||||
sql_exec, data=data
|
sql_exec, data=data
|
||||||
)
|
)
|
||||||
data["study"] = self
|
data["study"] = self
|
||||||
river._db_load_results(sql_exec, data=data)
|
river._db_load_results(sql_exec, data=data)
|
||||||
|
finally:
|
||||||
self._river_scenario_cache[scenario] = river
|
self.status.stop_loading()
|
||||||
self.status.scenario = old_scenario
|
self.status.scenario = old_scenario
|
||||||
|
|
||||||
|
# FIXME: Disable scenario cache to save memory usage
|
||||||
|
# self._river_scenario_cache[scenario] = river
|
||||||
|
|
||||||
return river
|
return river
|
||||||
|
|
||||||
def copy_from_scenario(self, scenario):
|
def copy_from_scenario(self, scenario):
|
||||||
|
|||||||
@@ -26,6 +26,8 @@ from functools import reduce
|
|||||||
from SQL import SQL
|
from SQL import SQL
|
||||||
from Model.Except import NotImplementedMethodeError
|
from Model.Except import NotImplementedMethodeError
|
||||||
|
|
||||||
|
from tools import trace
|
||||||
|
|
||||||
from Model.Tools.PamhyrID import PamhyrID
|
from Model.Tools.PamhyrID import PamhyrID
|
||||||
|
|
||||||
logger = logging.getLogger()
|
logger = logging.getLogger()
|
||||||
@@ -48,11 +50,9 @@ class SQLModel(SQL):
|
|||||||
self._cur = self._db.cursor()
|
self._cur = self._db.cursor()
|
||||||
|
|
||||||
if is_new:
|
if is_new:
|
||||||
logger.info("Create database")
|
|
||||||
self._create() # Create db
|
self._create() # Create db
|
||||||
# self._save() # Save
|
# self._save() # Save
|
||||||
else:
|
else:
|
||||||
logger.info("Update database")
|
|
||||||
self._update() # Update db scheme if necessary
|
self._update() # Update db scheme if necessary
|
||||||
# self._load() # Load data
|
# self._load() # Load data
|
||||||
|
|
||||||
@@ -202,6 +202,8 @@ class SQLSubModel(PamhyrID):
|
|||||||
"""
|
"""
|
||||||
if self._status is None:
|
if self._status is None:
|
||||||
return
|
return
|
||||||
|
if self._status.is_loading():
|
||||||
|
return
|
||||||
|
|
||||||
self._owner_scenario = self._status.scenario_id
|
self._owner_scenario = self._status.scenario_id
|
||||||
self._status.modified()
|
self._status.modified()
|
||||||
@@ -218,6 +220,8 @@ class SQLSubModel(PamhyrID):
|
|||||||
"""
|
"""
|
||||||
if self._status is None:
|
if self._status is None:
|
||||||
return
|
return
|
||||||
|
if self._status.is_loading():
|
||||||
|
return
|
||||||
|
|
||||||
self._owner_scenario = self._status.scenario_id
|
self._owner_scenario = self._status.scenario_id
|
||||||
self._status.modified()
|
self._status.modified()
|
||||||
@@ -297,6 +301,15 @@ class SQLSubModel(PamhyrID):
|
|||||||
"""
|
"""
|
||||||
raise NotImplementedMethodeError(cls, cls._db_update)
|
raise NotImplementedMethodeError(cls, cls._db_update)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def is_table_exists(cls, execute, table):
|
||||||
|
return any(
|
||||||
|
execute(
|
||||||
|
"SELECT name FROM sqlite_master " +
|
||||||
|
f"WHERE type='table' AND name='{table}'"
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def _db_update_to_0_2_0_set_node_pid(cls, execute, table, nodes):
|
def _db_update_to_0_2_0_set_node_pid(cls, execute, table, nodes):
|
||||||
els = execute(
|
els = execute(
|
||||||
@@ -311,6 +324,14 @@ class SQLSubModel(PamhyrID):
|
|||||||
if node_id == -1:
|
if node_id == -1:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
|
if node_id not in nodes:
|
||||||
|
# ⚠️ cas important : probablement déjà migré
|
||||||
|
logger.warning(
|
||||||
|
f"node_id {node_id} not in nodes " +
|
||||||
|
"→ probably already migrated"
|
||||||
|
)
|
||||||
|
continue
|
||||||
|
|
||||||
execute(
|
execute(
|
||||||
f"UPDATE {table} " +
|
f"UPDATE {table} " +
|
||||||
f"SET node = {nodes[node_id]} " +
|
f"SET node = {nodes[node_id]} " +
|
||||||
|
|||||||
@@ -97,6 +97,9 @@ class PamhyrModelDict(SQLSubModel):
|
|||||||
Nothing
|
Nothing
|
||||||
|
|
||||||
"""
|
"""
|
||||||
|
if predicate(self, data):
|
||||||
|
modifier(self, data)
|
||||||
|
|
||||||
for key in self._dict:
|
for key in self._dict:
|
||||||
self._dict[key]\
|
self._dict[key]\
|
||||||
._data_traversal(predicate, modifier, data)
|
._data_traversal(predicate, modifier, data)
|
||||||
|
|||||||
@@ -100,6 +100,9 @@ class PamhyrModelList(SQLSubModel):
|
|||||||
Nothing
|
Nothing
|
||||||
|
|
||||||
"""
|
"""
|
||||||
|
if predicate(self, data):
|
||||||
|
modifier(self, data)
|
||||||
|
|
||||||
for el in self._lst:
|
for el in self._lst:
|
||||||
el._data_traversal(predicate, modifier, data)
|
el._data_traversal(predicate, modifier, data)
|
||||||
|
|
||||||
|
|||||||
@@ -107,6 +107,9 @@ class PamhyrModelList(SQLSubModel):
|
|||||||
Returns:
|
Returns:
|
||||||
Nothing
|
Nothing
|
||||||
"""
|
"""
|
||||||
|
if predicate(self, data):
|
||||||
|
modifier(self, data)
|
||||||
|
|
||||||
for el in self._lst:
|
for el in self._lst:
|
||||||
el._data_traversal(predicate, modifier, data)
|
el._data_traversal(predicate, modifier, data)
|
||||||
|
|
||||||
@@ -168,7 +171,7 @@ class PamhyrModelList(SQLSubModel):
|
|||||||
lambda x: x[1],
|
lambda x: x[1],
|
||||||
filter(
|
filter(
|
||||||
lambda x: x[0] in indexes,
|
lambda x: x[0] in indexes,
|
||||||
enumerate(self._lst)
|
enumerate(self.lst)
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -56,6 +56,7 @@ class Modules(IterableFlag):
|
|||||||
SEDIMENT_LAYER = auto()
|
SEDIMENT_LAYER = auto()
|
||||||
ADDITIONAL_FILES = auto()
|
ADDITIONAL_FILES = auto()
|
||||||
OUTPUT_RK = auto()
|
OUTPUT_RK = auto()
|
||||||
|
GEOTIFF = auto()
|
||||||
|
|
||||||
# Results
|
# Results
|
||||||
RESULTS = auto()
|
RESULTS = auto()
|
||||||
@@ -81,6 +82,7 @@ class Modules(IterableFlag):
|
|||||||
cls.RESULTS,
|
cls.RESULTS,
|
||||||
cls.WINDOW_LIST,
|
cls.WINDOW_LIST,
|
||||||
cls.OUTPUT_RK,
|
cls.OUTPUT_RK,
|
||||||
|
cls.GEOTIFF
|
||||||
]
|
]
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
@@ -99,6 +101,7 @@ class Modules(IterableFlag):
|
|||||||
| cls.HYDRAULIC_STRUCTURES
|
| cls.HYDRAULIC_STRUCTURES
|
||||||
| cls.RESERVOIR
|
| cls.RESERVOIR
|
||||||
| cls.SEDIMENT_LAYER
|
| cls.SEDIMENT_LAYER
|
||||||
|
| cls.GEOTIFF
|
||||||
)
|
)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
@@ -114,6 +117,7 @@ class Modules(IterableFlag):
|
|||||||
cls.HYDRAULIC_STRUCTURES,
|
cls.HYDRAULIC_STRUCTURES,
|
||||||
cls.RESERVOIR,
|
cls.RESERVOIR,
|
||||||
cls.SEDIMENT_LAYER,
|
cls.SEDIMENT_LAYER,
|
||||||
|
cls.GEOTIFF,
|
||||||
]
|
]
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
@@ -129,6 +133,7 @@ class Modules(IterableFlag):
|
|||||||
cls.HYDRAULIC_STRUCTURES: "Hydraulic structures",
|
cls.HYDRAULIC_STRUCTURES: "Hydraulic structures",
|
||||||
cls.RESERVOIR: "Reservoir",
|
cls.RESERVOIR: "Reservoir",
|
||||||
cls.SEDIMENT_LAYER: "Sediment layer",
|
cls.SEDIMENT_LAYER: "Sediment layer",
|
||||||
|
cls.GEOTIFF: "GeoTIFF",
|
||||||
}
|
}
|
||||||
|
|
||||||
def impact(self):
|
def impact(self):
|
||||||
@@ -168,4 +173,5 @@ _impact = {
|
|||||||
Modules.HYDRAULIC_STRUCTURES: [],
|
Modules.HYDRAULIC_STRUCTURES: [],
|
||||||
Modules.RESERVOIR: [],
|
Modules.RESERVOIR: [],
|
||||||
Modules.SEDIMENT_LAYER: [],
|
Modules.SEDIMENT_LAYER: [],
|
||||||
|
Modules.GEOTIFF: [],
|
||||||
}
|
}
|
||||||
+273
-57
@@ -96,18 +96,262 @@ class AdisTS(CommandLineSolver):
|
|||||||
|
|
||||||
return lst
|
return lst
|
||||||
|
|
||||||
|
##########
|
||||||
|
# Export #
|
||||||
|
##########
|
||||||
|
|
||||||
|
_alph = list(
|
||||||
|
map(
|
||||||
|
chr,
|
||||||
|
chain(
|
||||||
|
range(48, 58), # 0..9
|
||||||
|
range(65, 91), # A..Z
|
||||||
|
range(97, 123) # a..z
|
||||||
|
)
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
_l_alph = len(_alph)
|
||||||
|
|
||||||
|
_nodes_cnt = 0
|
||||||
|
_nodes_names = {}
|
||||||
|
_nodes_views = set()
|
||||||
|
|
||||||
|
def get_reach_name(self, reach):
|
||||||
|
return f"Reach_{reach.pamhyr_id:>3}".replace(" ", "0")
|
||||||
|
|
||||||
|
def get_node_name(self, node):
|
||||||
|
"""Generate a 3 char name for node
|
||||||
|
|
||||||
|
Args:
|
||||||
|
node: The node
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
A 3 char name string
|
||||||
|
"""
|
||||||
|
n = node.pamhyr_id
|
||||||
|
|
||||||
|
if n in self._nodes_names:
|
||||||
|
return self._nodes_names[n]
|
||||||
|
|
||||||
|
name = ""
|
||||||
|
|
||||||
|
checked_new = False
|
||||||
|
while not checked_new:
|
||||||
|
self._nodes_cnt += 1
|
||||||
|
nc = self._nodes_cnt
|
||||||
|
|
||||||
|
name = "".join(
|
||||||
|
map(
|
||||||
|
lambda i: self._alph[i % self._l_alph],
|
||||||
|
[
|
||||||
|
int(nc / (self._l_alph * self._l_alph)),
|
||||||
|
int(nc / self._l_alph),
|
||||||
|
nc
|
||||||
|
]
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
checked_new = name not in self._nodes_views
|
||||||
|
|
||||||
|
self._nodes_views.add(name)
|
||||||
|
self._nodes_names[n] = name
|
||||||
|
|
||||||
|
return name
|
||||||
|
|
||||||
def cmd_args(self, study):
|
def cmd_args(self, study):
|
||||||
lst = super(AdisTS, self).cmd_args(study)
|
lst = super(AdisTS, self).cmd_args(study)
|
||||||
return lst
|
return lst
|
||||||
|
|
||||||
def input_param(self):
|
def input_param(self):
|
||||||
name = self._study.name
|
name = self._study.name.replace(" ", "_")
|
||||||
return f"{name}.REP"
|
return f"{name}.REP"
|
||||||
|
|
||||||
def log_file(self):
|
def log_file(self):
|
||||||
name = self._study.name
|
name = self._study.name.replace(" ", "_")
|
||||||
return f"{name}.TRA"
|
return f"{name}.TRA"
|
||||||
|
|
||||||
|
def _export_ST(self, study, repertory, qlog, name="0"):
|
||||||
|
files = []
|
||||||
|
|
||||||
|
if qlog is not None:
|
||||||
|
qlog.put("Export ST file")
|
||||||
|
|
||||||
|
os.makedirs(os.path.join(repertory, "net"), exist_ok=True)
|
||||||
|
|
||||||
|
# Write header
|
||||||
|
edges = study.river.enable_edges()
|
||||||
|
for edge in edges:
|
||||||
|
name = f"Reach_{edge.id + 1:>3}".replace(" ", "0")
|
||||||
|
|
||||||
|
with adists_file_open(
|
||||||
|
os.path.join(repertory, "net", f"{name}.ST"),
|
||||||
|
"w+"
|
||||||
|
) as f:
|
||||||
|
files.append(str(os.path.join("net", f"{name}.ST")))
|
||||||
|
|
||||||
|
cnt_num = 1
|
||||||
|
for profile in edge.reach.profiles:
|
||||||
|
self._export_ST_profile_header(
|
||||||
|
f, files, profile, cnt_num
|
||||||
|
)
|
||||||
|
cnt_num += 1
|
||||||
|
|
||||||
|
# Points
|
||||||
|
for point in profile.points:
|
||||||
|
self._export_ST_point_line(
|
||||||
|
f, files, point
|
||||||
|
)
|
||||||
|
|
||||||
|
# Profile last line
|
||||||
|
f.write(f" 999.9990 999.9990 999.9990\n")
|
||||||
|
|
||||||
|
def _export_ST_profile_header(self, wfile, files,
|
||||||
|
profile, cnt):
|
||||||
|
num = f"{cnt:>6}"
|
||||||
|
c1 = f"{profile.code1:>6}"
|
||||||
|
c2 = f"{profile.code2:>6}"
|
||||||
|
t = f"{len(profile.points):>6}"
|
||||||
|
rk = f"{profile.rk:>12f}"[0:12]
|
||||||
|
pname = profile.name
|
||||||
|
if profile.name == "":
|
||||||
|
# Generate name from profile id prefixed with
|
||||||
|
# 'p' (and replace space char with '0' char)
|
||||||
|
pname = f"p{profile.id:>3}".replace(" ", "0")
|
||||||
|
name = f"{pname:<19}"
|
||||||
|
|
||||||
|
# Generate sediment additional data if available
|
||||||
|
sediment = ""
|
||||||
|
if profile.sl is not None:
|
||||||
|
if not any(filter(lambda f: ".GRA" in f, files)):
|
||||||
|
files.append(self._gra_file)
|
||||||
|
|
||||||
|
# Number of layers
|
||||||
|
nl = len(profile.sl)
|
||||||
|
sediment = f" {nl:>3}"
|
||||||
|
|
||||||
|
# Layers data
|
||||||
|
for layer in profile.sl.layers:
|
||||||
|
sediment += (
|
||||||
|
f" {layer.height:>10} {layer.d50:>10} " +
|
||||||
|
f"{layer.sigma:>10} " +
|
||||||
|
f"{layer.critical_constraint:>10}"
|
||||||
|
)
|
||||||
|
|
||||||
|
# Profile header line
|
||||||
|
wfile.write(f"{num}{c1}{c2}{t} {rk} {pname} {sediment}\n")
|
||||||
|
|
||||||
|
def _export_ST_point_line(self, wfile, files, point):
|
||||||
|
x = f"{point.x:<12.4f}"[0:12]
|
||||||
|
y = f"{point.y:<12.4f}"[0:12]
|
||||||
|
z = f"{point.z:<12.4f}"[0:12]
|
||||||
|
n = f"{point.name:<3}"
|
||||||
|
|
||||||
|
# Generate sediment additional data if available
|
||||||
|
sediment = ""
|
||||||
|
prev = point.z
|
||||||
|
if point.sl is not None:
|
||||||
|
# Number of layers
|
||||||
|
nl = len(point.sl)
|
||||||
|
sediment = f"{nl:>3}"
|
||||||
|
|
||||||
|
# Layers data
|
||||||
|
for layer in point.sl.layers:
|
||||||
|
prev = round(prev - layer.height, 5)
|
||||||
|
sediment += (
|
||||||
|
f" {prev:>10} {layer.d50:>10} " +
|
||||||
|
f"{layer.sigma:>10} " +
|
||||||
|
f"{layer.critical_constraint:>10}"
|
||||||
|
)
|
||||||
|
|
||||||
|
# Point line
|
||||||
|
wfile.write(f"{x} {y} {z} {n} {sediment}\n")
|
||||||
|
|
||||||
|
def _export_NET(self, study, repertory, qlog=None, name="0"):
|
||||||
|
|
||||||
|
if qlog is not None:
|
||||||
|
qlog.put("Export NET file")
|
||||||
|
|
||||||
|
with adists_file_open(
|
||||||
|
os.path.join(repertory, f"{name}.NET"), "w+") as f:
|
||||||
|
edges = study.river.enable_edges()
|
||||||
|
|
||||||
|
for e in edges:
|
||||||
|
name = f"Reach_{e.id + 1:>3}".replace(" ", "0")
|
||||||
|
id = name
|
||||||
|
|
||||||
|
n1 = f"{self.get_node_name(e.node1):3}".replace(" ", "x")
|
||||||
|
n2 = f"{self.get_node_name(e.node2):3}".replace(" ", "x")
|
||||||
|
file = os.path.join("net", name + ".ST")
|
||||||
|
|
||||||
|
f.write(f"{id} {n1} {n2} {file}\n")
|
||||||
|
|
||||||
|
def _export_fake_INI(self, study, repertory, qlog=None, name="0"):
|
||||||
|
if qlog is not None:
|
||||||
|
qlog.put("Export fake INI file")
|
||||||
|
|
||||||
|
with adists_file_open(
|
||||||
|
os.path.join(
|
||||||
|
repertory, "Mage_fin.ini"
|
||||||
|
), "w+"
|
||||||
|
) as f:
|
||||||
|
edges = study.river.enable_edges()
|
||||||
|
for i, edge in enumerate(edges):
|
||||||
|
lst = list(filter(
|
||||||
|
lambda f: f.is_full_defined(),
|
||||||
|
edge.frictions.frictions
|
||||||
|
))
|
||||||
|
rk_min = 9999999.9
|
||||||
|
rk_max = -9999999.9
|
||||||
|
coeff_min = -1.0
|
||||||
|
coeff_max = -1.0
|
||||||
|
for s in lst: # TODO optimise ?
|
||||||
|
if s.begin_rk > rk_max:
|
||||||
|
rk_max = s.begin_rk
|
||||||
|
coeff_max = s.begin_strickler
|
||||||
|
if s.begin_rk < rk_min:
|
||||||
|
rk_min = s.begin_rk
|
||||||
|
coeff_min = s.begin_strickler
|
||||||
|
if s.end_rk > rk_max:
|
||||||
|
rk_max = s.end_rk
|
||||||
|
coeff_max = s.end_strickler
|
||||||
|
if s.end_rk < rk_min:
|
||||||
|
rk_min = s.end_rk
|
||||||
|
coeff_min = s.end_strickler
|
||||||
|
|
||||||
|
print("min max", rk_min, rk_max)
|
||||||
|
|
||||||
|
def get_stricklers_from_rk(rk, lst):
|
||||||
|
print("rk", rk)
|
||||||
|
|
||||||
|
coeff = None
|
||||||
|
if rk > rk_max:
|
||||||
|
coeff = coeff_max
|
||||||
|
elif rk < rk_min:
|
||||||
|
coeff = coeff_min
|
||||||
|
else:
|
||||||
|
for s in lst:
|
||||||
|
if (rk >= s.begin_rk and rk <= s.end_rk or
|
||||||
|
rk <= s.begin_rk and rk >= s.end_rk):
|
||||||
|
coeff = s.begin_strickler # TODO: inerpolate
|
||||||
|
break
|
||||||
|
|
||||||
|
# TODO interpolation if rk is not in frictons
|
||||||
|
|
||||||
|
if coeff is None:
|
||||||
|
logger.error(
|
||||||
|
"Study frictions are not fully defined"
|
||||||
|
)
|
||||||
|
return None
|
||||||
|
|
||||||
|
return coeff.minor, coeff.medium
|
||||||
|
|
||||||
|
for j, profile in enumerate(edge.reach.profiles):
|
||||||
|
coef_min, coef_moy = get_stricklers_from_rk(profile.rk,
|
||||||
|
lst)
|
||||||
|
f.write(
|
||||||
|
f" {i+1:3}{j+1:4}{' '*116}{coef_min:9}{coef_moy:9}\n")
|
||||||
|
|
||||||
def _export_REP_additional_lines(self, study, rep_file):
|
def _export_REP_additional_lines(self, study, rep_file):
|
||||||
lines = filter(
|
lines = filter(
|
||||||
lambda line: line.is_enabled(),
|
lambda line: line.is_enabled(),
|
||||||
@@ -129,7 +373,7 @@ class AdisTS(CommandLineSolver):
|
|||||||
), "w+"
|
), "w+"
|
||||||
) as f:
|
) as f:
|
||||||
path = os.path.join("..", mage_rep, name)
|
path = os.path.join("..", mage_rep, name)
|
||||||
f.write(f"NET {path}.NET\n")
|
f.write(f"NET {name}.NET\n")
|
||||||
f.write(f"REP {path}.REP\n")
|
f.write(f"REP {path}.REP\n")
|
||||||
|
|
||||||
for file in files:
|
for file in files:
|
||||||
@@ -138,13 +382,25 @@ class AdisTS(CommandLineSolver):
|
|||||||
|
|
||||||
self._export_REP_additional_lines(study, f)
|
self._export_REP_additional_lines(study, f)
|
||||||
|
|
||||||
path_mage_net = os.path.join(os.path.abspath(
|
self._export_ST(study, repertory, qlog, name=name)
|
||||||
os.path.join(repertory, os.pardir)
|
self._export_NET(study, repertory, qlog, name=name)
|
||||||
), os.path.join(mage_rep, "net"))
|
|
||||||
path_adists_net = os.path.join(repertory, "net")
|
|
||||||
|
|
||||||
if os.path.exists(path_mage_net):
|
# fake mage_fin.ini:
|
||||||
shutil.copytree(path_mage_net, path_adists_net, dirs_exist_ok=True)
|
|
||||||
|
path_mage = os.path.join(os.path.abspath(
|
||||||
|
os.path.join(repertory, os.pardir)), mage_rep)
|
||||||
|
self._export_fake_INI(study, path_mage,
|
||||||
|
qlog, name=name)
|
||||||
|
|
||||||
|
# path_mage_net = os.path.join(os.path.abspath(
|
||||||
|
# os.path.join(repertory, os.pardir)
|
||||||
|
# ), os.path.join(mage_rep, "net"))
|
||||||
|
# path_adists_net = os.path.join(repertory, "net")
|
||||||
|
|
||||||
|
# if os.path.exists(path_mage_net):
|
||||||
|
# shutil.copytree(path_mage_net,
|
||||||
|
# path_adists_net,
|
||||||
|
# dirs_exist_ok=True)
|
||||||
|
|
||||||
@timer
|
@timer
|
||||||
def export(self, study, repertory, qlog=None):
|
def export(self, study, repertory, qlog=None):
|
||||||
@@ -197,48 +453,7 @@ class AdisTS(CommandLineSolver):
|
|||||||
_nodes_views = set()
|
_nodes_views = set()
|
||||||
|
|
||||||
def get_reach_name(self, reach):
|
def get_reach_name(self, reach):
|
||||||
return f"Reach_{reach.id:>3}".replace(" ", "0")
|
return f"Reach_{reach.pamhyr_id:>3}".replace(" ", "0")
|
||||||
|
|
||||||
def get_node_name(self, node):
|
|
||||||
"""Generate a 3 char name for node
|
|
||||||
|
|
||||||
Args:
|
|
||||||
node: The node
|
|
||||||
|
|
||||||
Returns:
|
|
||||||
A 3 char name string
|
|
||||||
"""
|
|
||||||
n = node.id
|
|
||||||
|
|
||||||
# print("node name id: ", n)
|
|
||||||
|
|
||||||
# if n in self._nodes_names:
|
|
||||||
# return self._nodes_names[n]
|
|
||||||
|
|
||||||
name = ""
|
|
||||||
|
|
||||||
checked_new = False
|
|
||||||
while not checked_new:
|
|
||||||
self._nodes_cnt += 1
|
|
||||||
nc = self._nodes_cnt
|
|
||||||
|
|
||||||
name = "".join(
|
|
||||||
map(
|
|
||||||
lambda i: self._alph[i % self._l_alph],
|
|
||||||
[
|
|
||||||
int(nc / (self._l_alph * self._l_alph)),
|
|
||||||
int(nc / self._l_alph),
|
|
||||||
nc
|
|
||||||
]
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
checked_new = name not in self._nodes_views
|
|
||||||
|
|
||||||
self._nodes_views.add(name)
|
|
||||||
self._nodes_names[n] = name
|
|
||||||
|
|
||||||
return name
|
|
||||||
|
|
||||||
#################################
|
#################################
|
||||||
# Adis-TS in weak coupling mode #
|
# Adis-TS in weak coupling mode #
|
||||||
@@ -338,9 +553,9 @@ class AdisTSwc(AdisTS):
|
|||||||
) as f:
|
) as f:
|
||||||
for LC in POL_LC:
|
for LC in POL_LC:
|
||||||
reach = next(filter(
|
reach = next(filter(
|
||||||
lambda edge: edge.id == LC.edge, study.river.edges()
|
lambda edge: edge.id == LC.edge, study.river.enable_edges()
|
||||||
)) # .name
|
)) # .name
|
||||||
reach_name = self.get_reach_name(self, reach)
|
reach_name = self.get_reach_name(reach)
|
||||||
f.write(f"${reach_name} {LC.begin_rk} {LC.end_rk}\n")
|
f.write(f"${reach_name} {LC.begin_rk} {LC.end_rk}\n")
|
||||||
f.write(f"*temps |débit massique (kg/s)\n")
|
f.write(f"*temps |débit massique (kg/s)\n")
|
||||||
f.write(f"*---------++++++++++\n")
|
f.write(f"*---------++++++++++\n")
|
||||||
@@ -365,8 +580,7 @@ class AdisTSwc(AdisTS):
|
|||||||
node = next(filter(
|
node = next(filter(
|
||||||
lambda x: x.id == BC.node, study.river._nodes
|
lambda x: x.id == BC.node, study.river._nodes
|
||||||
))
|
))
|
||||||
node_name = f"{node.id:3}".replace(" ", "x")
|
f.write(f"${self.get_node_name(node)}\n")
|
||||||
f.write(f"${node_name}\n")
|
|
||||||
|
|
||||||
if BC.type == "Concentration":
|
if BC.type == "Concentration":
|
||||||
f.write(f"*temps |concentration\n")
|
f.write(f"*temps |concentration\n")
|
||||||
@@ -731,6 +945,7 @@ class AdisTSwc(AdisTS):
|
|||||||
for file_bin in path_files:
|
for file_bin in path_files:
|
||||||
key_pol = os.path.basename(file_bin)[0:-4]
|
key_pol = os.path.basename(file_bin)[0:-4]
|
||||||
data_tmp[key_pol] = {}
|
data_tmp[key_pol] = {}
|
||||||
|
logger.info(f"read_bin: Start reading '{file_bin}' ...")
|
||||||
with open(file_bin, 'rb') as f:
|
with open(file_bin, 'rb') as f:
|
||||||
# header
|
# header
|
||||||
# first line
|
# first line
|
||||||
@@ -782,7 +997,9 @@ class AdisTSwc(AdisTS):
|
|||||||
# seventh line (useless)
|
# seventh line (useless)
|
||||||
for k in range(0, ismax, kbl):
|
for k in range(0, ismax, kbl):
|
||||||
data = np.fromfile(f, dtype=np.int32, count=1) # (start)
|
data = np.fromfile(f, dtype=np.int32, count=1) # (start)
|
||||||
zero = np.fromfile(f, dtype=np.int32, count=ismax)
|
zero = np.fromfile(
|
||||||
|
f, dtype=np.int32, count=min(k + kbl, ismax) - k
|
||||||
|
)
|
||||||
data = np.fromfile(f, dtype=np.int32, count=1) # (end)
|
data = np.fromfile(f, dtype=np.int32, count=1) # (end)
|
||||||
# end header
|
# end header
|
||||||
# data
|
# data
|
||||||
@@ -813,7 +1030,6 @@ class AdisTSwc(AdisTS):
|
|||||||
for r, reach in enumerate(reachs):
|
for r, reach in enumerate(reachs):
|
||||||
for i in range(is1[r]-1, is2[r]):
|
for i in range(is1[r]-1, is2[r]):
|
||||||
p_i = ip_to_ri(reach, i)
|
p_i = ip_to_ri(reach, i)
|
||||||
|
|
||||||
for t_data in timestamps_keys:
|
for t_data in timestamps_keys:
|
||||||
pol_view = []
|
pol_view = []
|
||||||
for pol in pollutants_keys:
|
for pol in pollutants_keys:
|
||||||
|
|||||||
+65
-24
@@ -19,6 +19,8 @@
|
|||||||
import os
|
import os
|
||||||
import logging
|
import logging
|
||||||
import numpy as np
|
import numpy as np
|
||||||
|
import time
|
||||||
|
from functools import reduce
|
||||||
|
|
||||||
from itertools import chain
|
from itertools import chain
|
||||||
from tools import timer, trace, logger_exception
|
from tools import timer, trace, logger_exception
|
||||||
@@ -325,7 +327,7 @@ class Mage(CommandLineSolver):
|
|||||||
v0 = d[0]
|
v0 = d[0]
|
||||||
v1 = d[1]
|
v1 = d[1]
|
||||||
|
|
||||||
if t in ["HYD", "QSO", "LIM"]:
|
if t in ["HYD", "LIM"]:
|
||||||
v0 /= 60 # Convert first column to minute
|
v0 /= 60 # Convert first column to minute
|
||||||
|
|
||||||
f.write(f"{v0:9} {v1:9} \n")
|
f.write(f"{v0:9} {v1:9} \n")
|
||||||
@@ -384,26 +386,26 @@ class Mage(CommandLineSolver):
|
|||||||
return files
|
return files
|
||||||
|
|
||||||
def _export_LC(self, study, lateral, f, qlog, name="0"):
|
def _export_LC(self, study, lateral, f, qlog, name="0"):
|
||||||
if lateral.edge is None:
|
if lateral.reach is None:
|
||||||
return
|
return
|
||||||
|
|
||||||
if lateral.is_deleted():
|
if lateral.is_deleted():
|
||||||
return
|
return
|
||||||
|
|
||||||
edges = study.river.enable_edges()
|
edges = study.river.enable_edges()
|
||||||
if lateral.edge not in edges:
|
if lateral.reach not in edges:
|
||||||
return
|
return
|
||||||
|
|
||||||
eid, _ = next(
|
eid, _ = next(
|
||||||
filter(
|
filter(
|
||||||
lambda e: e[1] == lateral.edge,
|
lambda e: e[1] == lateral.reach,
|
||||||
enumerate(edges)
|
enumerate(edges)
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
name = f"{eid+1:>3}"
|
name = f"{eid+1:>3}"
|
||||||
# name = f"Reach_{lateral.edge.id + 1:>3}".replace(" ", "0")
|
# name = f"Reach_{lateral.reach.id + 1:>3}".replace(" ", "0")
|
||||||
f.write(f"* {lateral.edge.name} ({name}) {lateral.lctype}\n")
|
f.write(f"* {lateral.reach.name} ({name}) {lateral.lctype}\n")
|
||||||
f.write(
|
f.write(
|
||||||
f"${name} " +
|
f"${name} " +
|
||||||
f"{lateral.begin_rk:>10.4f} {lateral.end_rk:>10.4f}\n"
|
f"{lateral.begin_rk:>10.4f} {lateral.end_rk:>10.4f}\n"
|
||||||
@@ -544,6 +546,15 @@ class Mage(CommandLineSolver):
|
|||||||
|
|
||||||
return files
|
return files
|
||||||
|
|
||||||
|
def skip_hs(self, hs):
|
||||||
|
return (
|
||||||
|
hs.input_reach is None
|
||||||
|
or not hs.input_reach.is_enable()
|
||||||
|
or not hs.enabled
|
||||||
|
or hs.input_section is None
|
||||||
|
or hs.is_deleted()
|
||||||
|
)
|
||||||
|
|
||||||
def _export_SIN(self, study, repertory, qlog, name="0"):
|
def _export_SIN(self, study, repertory, qlog, name="0"):
|
||||||
files = []
|
files = []
|
||||||
|
|
||||||
@@ -569,19 +580,7 @@ class Mage(CommandLineSolver):
|
|||||||
files.append(f"{name}.SIN")
|
files.append(f"{name}.SIN")
|
||||||
|
|
||||||
for hs in hydraulic_structures:
|
for hs in hydraulic_structures:
|
||||||
if hs.input_reach is None:
|
if self.skip_hs(hs):
|
||||||
continue
|
|
||||||
|
|
||||||
if not hs.input_reach.is_enable():
|
|
||||||
continue
|
|
||||||
|
|
||||||
if not hs.enabled:
|
|
||||||
continue
|
|
||||||
|
|
||||||
if hs.input_section is None:
|
|
||||||
continue
|
|
||||||
|
|
||||||
if hs.is_deleted():
|
|
||||||
continue
|
continue
|
||||||
|
|
||||||
f.write(
|
f.write(
|
||||||
@@ -988,7 +987,7 @@ class Mage8(Mage):
|
|||||||
|
|
||||||
# Data
|
# Data
|
||||||
for d in bound.data:
|
for d in bound.data:
|
||||||
f.write(f"{d[0]:10.3f}{d[1]:10.3f}\n")
|
f.write(f"{d[0]/60:10.3f}{d[1]:10.3f}\n")
|
||||||
|
|
||||||
return files
|
return files
|
||||||
|
|
||||||
@@ -1035,6 +1034,7 @@ class Mage8(Mage):
|
|||||||
@timer
|
@timer
|
||||||
def read_bin(self, study, fname, results, qlog=None, name="0"):
|
def read_bin(self, study, fname, results, qlog=None, name="0"):
|
||||||
logger.info(f"read_bin: Start reading '{fname}' ...")
|
logger.info(f"read_bin: Start reading '{fname}' ...")
|
||||||
|
start = time.time()
|
||||||
|
|
||||||
with mage_file_open(fname, "r") as f:
|
with mage_file_open(fname, "r") as f:
|
||||||
def newline(): return np.fromfile(f, dtype=np.int32, count=1)
|
def newline(): return np.fromfile(f, dtype=np.int32, count=1)
|
||||||
@@ -1117,7 +1117,8 @@ class Mage8(Mage):
|
|||||||
|
|
||||||
# Z and Y (4th line)
|
# Z and Y (4th line)
|
||||||
newline()
|
newline()
|
||||||
_ = read_float(3*nb_profile)
|
tmp = read_float(3*nb_profile)
|
||||||
|
zfd = tmp[::3]
|
||||||
endline()
|
endline()
|
||||||
|
|
||||||
# Data
|
# Data
|
||||||
@@ -1180,10 +1181,9 @@ class Mage8(Mage):
|
|||||||
logger.info(f"compute tab...")
|
logger.info(f"compute tab...")
|
||||||
for r in reachs:
|
for r in reachs:
|
||||||
for p in r.profiles:
|
for p in r.profiles:
|
||||||
if not p.geometry.tab_up_to_date:
|
|
||||||
p.geometry.compute_tabulation()
|
p.geometry.compute_tabulation()
|
||||||
|
|
||||||
logger.info(f"compute velocily...")
|
logger.info(f"compute velocity...")
|
||||||
|
|
||||||
for r in reachs:
|
for r in reachs:
|
||||||
for t in ts_list:
|
for t in ts_list:
|
||||||
@@ -1195,6 +1195,11 @@ class Mage8(Mage):
|
|||||||
r.set(i, t, "V", v)
|
r.set(i, t, "V", v)
|
||||||
logger.info(f"read_bin: ... end with {len(ts)} timestamp read")
|
logger.info(f"read_bin: ... end with {len(ts)} timestamp read")
|
||||||
|
|
||||||
|
results.bufferize("Z")
|
||||||
|
results.bufferize("Q")
|
||||||
|
results.bufferize("V")
|
||||||
|
logger.info(f"reading time: '{time.time() - start}'")
|
||||||
|
|
||||||
@timer
|
@timer
|
||||||
def read_gra(self, study, repertory, results, qlog=None, name="0"):
|
def read_gra(self, study, repertory, results, qlog=None, name="0"):
|
||||||
if not study.river.has_sediment():
|
if not study.river.has_sediment():
|
||||||
@@ -1202,6 +1207,7 @@ class Mage8(Mage):
|
|||||||
|
|
||||||
fname = os.path.join(repertory, f"{name}.GRA")
|
fname = os.path.join(repertory, f"{name}.GRA")
|
||||||
logger.info(f"read_gra: Start reading '{fname}' ...")
|
logger.info(f"read_gra: Start reading '{fname}' ...")
|
||||||
|
start = time.time()
|
||||||
|
|
||||||
with mage_file_open(fname, "r") as f:
|
with mage_file_open(fname, "r") as f:
|
||||||
def newline(): return np.fromfile(f, dtype=np.int32, count=1)
|
def newline(): return np.fromfile(f, dtype=np.int32, count=1)
|
||||||
@@ -1332,7 +1338,7 @@ class Mage8(Mage):
|
|||||||
|
|
||||||
i_pts = 0
|
i_pts = 0
|
||||||
i_data = 0
|
i_data = 0
|
||||||
# Loop on cross section
|
# Loop on cross sections
|
||||||
for i in range(n):
|
for i in range(n):
|
||||||
sec_sl = []
|
sec_sl = []
|
||||||
reach = ip_to_r(i)
|
reach = ip_to_r(i)
|
||||||
@@ -1361,8 +1367,43 @@ class Mage8(Mage):
|
|||||||
)
|
)
|
||||||
end = newline().size <= 0
|
end = newline().size <= 0
|
||||||
|
|
||||||
|
ts_list = sorted(ts)
|
||||||
|
|
||||||
|
logger.info(f"compute river bed elevation...")
|
||||||
|
|
||||||
|
for r in reachs:
|
||||||
|
z_min = r.geometry.get_z_min()
|
||||||
|
sls = list(map(
|
||||||
|
lambda p: p.get_ts_key(ts_list[0], "sl")[0],
|
||||||
|
r.profiles
|
||||||
|
))
|
||||||
|
z_br = list(map(
|
||||||
|
lambda z, sl: reduce(
|
||||||
|
lambda z, h: z - h[0],
|
||||||
|
sl, z
|
||||||
|
),
|
||||||
|
z_min, # Original geometry
|
||||||
|
sls # Original sediment layers
|
||||||
|
))
|
||||||
|
for t in ts_list:
|
||||||
|
sls = list(map(
|
||||||
|
lambda p: p.get_ts_key(t, "sl")[0],
|
||||||
|
r.profiles
|
||||||
|
))
|
||||||
|
zfd = list(map(
|
||||||
|
lambda z, sl: reduce(
|
||||||
|
lambda z, h: z + h[0],
|
||||||
|
sl, z
|
||||||
|
),
|
||||||
|
z_br, # bedrock
|
||||||
|
sls # Original sediment layers
|
||||||
|
))
|
||||||
|
for i, p in enumerate(r.profiles):
|
||||||
|
r.set(i, t, "zfd", zfd[i])
|
||||||
|
|
||||||
results.set("sediment_timestamps", ts)
|
results.set("sediment_timestamps", ts)
|
||||||
logger.info(f"read_gra: ... end with {len(ts)} timestamp read")
|
logger.info(f"read_gra: ... end with {len(ts)} timestamp read")
|
||||||
|
logger.debug(f"reading time: '{time.time() - start}'")
|
||||||
|
|
||||||
@timer
|
@timer
|
||||||
def results(self, study, repertory,
|
def results(self, study, repertory,
|
||||||
|
|||||||
+135
-13
@@ -58,12 +58,12 @@ class Rubar3(CommandLineSolver):
|
|||||||
("rubarbe_iovis", "n"),
|
("rubarbe_iovis", "n"),
|
||||||
("rubarbe_rep", "n"),
|
("rubarbe_rep", "n"),
|
||||||
("rubarbe_tinit", "000:00:00:00"),
|
("rubarbe_tinit", "000:00:00:00"),
|
||||||
("rubarbe_tmax", "999:99:99:00"),
|
("rubarbe_tmax", "000:01:00:00"), # 1 day
|
||||||
("rubarbe_tiopdt", "000:00:00:00"),
|
("rubarbe_tiopdt", "000:00:00:00"),
|
||||||
("rubarbe_dt", "5.0"),
|
("rubarbe_dt", "5.0"),
|
||||||
("rubarbe_ts", "999:99:99:00"),
|
("rubarbe_ts", "999:99:99:00"),
|
||||||
("rubarbe_dtsauv", "00:00:00:05"),
|
("rubarbe_dtsauv", "00:00:05:00"),
|
||||||
("rubarbe_psave", "00:00:00:05"),
|
("rubarbe_psave", "00:00:05:00"),
|
||||||
("rubarbe_fdeb1", "1"),
|
("rubarbe_fdeb1", "1"),
|
||||||
("rubarbe_fdeb2", "10"),
|
("rubarbe_fdeb2", "10"),
|
||||||
("rubarbe_fdeb3", "100"),
|
("rubarbe_fdeb3", "100"),
|
||||||
@@ -169,9 +169,8 @@ class Rubar3(CommandLineSolver):
|
|||||||
|
|
||||||
it = iter(params)
|
it = iter(params)
|
||||||
|
|
||||||
line = 0
|
param = next(it, None)
|
||||||
while line < 27:
|
while param is not None:
|
||||||
param = next(it)
|
|
||||||
name = param.name
|
name = param.name
|
||||||
value = param.value
|
value = param.value
|
||||||
|
|
||||||
@@ -204,7 +203,7 @@ class Rubar3(CommandLineSolver):
|
|||||||
# New line
|
# New line
|
||||||
f.write(f"\n")
|
f.write(f"\n")
|
||||||
|
|
||||||
line += 1
|
param = next(it, None)
|
||||||
|
|
||||||
def _export_ts(self, study, repertory, qlog, name="0"):
|
def _export_ts(self, study, repertory, qlog, name="0"):
|
||||||
if qlog is not None:
|
if qlog is not None:
|
||||||
@@ -427,7 +426,6 @@ class Rubar3(CommandLineSolver):
|
|||||||
coeff = coeff_min
|
coeff = coeff_min
|
||||||
else:
|
else:
|
||||||
for s in lst:
|
for s in lst:
|
||||||
print(s.begin_rk, s.end_rk)
|
|
||||||
if (rk >= s.begin_rk and rk <= s.end_rk or
|
if (rk >= s.begin_rk and rk <= s.end_rk or
|
||||||
rk <= s.begin_rk and rk >= s.end_rk):
|
rk <= s.begin_rk and rk >= s.end_rk):
|
||||||
coeff = s.begin_strickler # TODO: inerpolate
|
coeff = s.begin_strickler # TODO: inerpolate
|
||||||
@@ -474,7 +472,6 @@ class Rubar3(CommandLineSolver):
|
|||||||
last = profiles[-1]
|
last = profiles[-1]
|
||||||
|
|
||||||
if first not in data or last not in data:
|
if first not in data or last not in data:
|
||||||
print(data)
|
|
||||||
logger.error(
|
logger.error(
|
||||||
"Study initial condition is not fully defined"
|
"Study initial condition is not fully defined"
|
||||||
)
|
)
|
||||||
@@ -575,8 +572,12 @@ class Rubar3(CommandLineSolver):
|
|||||||
|
|
||||||
for bc in bcs:
|
for bc in bcs:
|
||||||
f.write(f"{len(bc)}\n")
|
f.write(f"{len(bc)}\n")
|
||||||
|
if bc.bctype == "ZD":
|
||||||
for d0, d1 in bc.data:
|
for d0, d1 in bc.data:
|
||||||
f.write(f"{d1} {d0}\n")
|
f.write(f"{d1} {d0}\n")
|
||||||
|
else:
|
||||||
|
for d0, d1 in bc.data:
|
||||||
|
f.write(f"{d0} {d1}\n")
|
||||||
|
|
||||||
def read_profil(self, study, fname, results, qlog=None, name="0"):
|
def read_profil(self, study, fname, results, qlog=None, name="0"):
|
||||||
logger.info(f"read: profil.{name}")
|
logger.info(f"read: profil.{name}")
|
||||||
@@ -587,10 +588,15 @@ class Rubar3(CommandLineSolver):
|
|||||||
# Add results reach to reach list
|
# Add results reach to reach list
|
||||||
res_reach = results.river.add(i)
|
res_reach = results.river.add(i)
|
||||||
reachs.append((res_reach, len(reach) + 1))
|
reachs.append((res_reach, len(reach) + 1))
|
||||||
|
results.set("nb_reach", f"{len(reachs)}")
|
||||||
|
results.set("nb_profile", f"{sum([r[1] - 1 for r in reachs])}")
|
||||||
|
|
||||||
def read_data_line(f):
|
def read_data_line(f):
|
||||||
line = f.readline().split()
|
line = f.readline().split()
|
||||||
|
if len(line) > 3:
|
||||||
return tuple(map(float, line))
|
return tuple(map(float, line))
|
||||||
|
else:
|
||||||
|
return (0.0, 0.0, 0.0, 0.0)
|
||||||
|
|
||||||
def set_and_compute_limites(reach, ind, z, q, s):
|
def set_and_compute_limites(reach, ind, z, q, s):
|
||||||
reach.set(ind, timestamp, "Z", z)
|
reach.set(ind, timestamp, "Z", z)
|
||||||
@@ -604,6 +610,28 @@ class Rubar3(CommandLineSolver):
|
|||||||
)
|
)
|
||||||
|
|
||||||
ts = set()
|
ts = set()
|
||||||
|
timestamp = next(filter(
|
||||||
|
lambda p: p.name == 'rubarbe_tinit',
|
||||||
|
study.river.get_params(self._type).parameters
|
||||||
|
)).value
|
||||||
|
if timestamp.count(':') == 3:
|
||||||
|
timestamp = old_pamhyr_date_to_timestamp(timestamp)
|
||||||
|
ts.add(timestamp)
|
||||||
|
|
||||||
|
# add initial condition
|
||||||
|
for r, edge in enumerate(study.river.enable_edges()):
|
||||||
|
reach = edge.reach
|
||||||
|
ics = study.river.initial_conditions.get(edge)
|
||||||
|
q = ics.get_discharge()
|
||||||
|
z = ics.get_elevation()
|
||||||
|
k = 0
|
||||||
|
for i, j in zip(z, q):
|
||||||
|
v = reach.profiles[k].speed(i, j)
|
||||||
|
set_and_compute_limites(reachs[r][0], k, i, j, v)
|
||||||
|
k += 1
|
||||||
|
|
||||||
|
# start read
|
||||||
|
|
||||||
end = False
|
end = False
|
||||||
while True:
|
while True:
|
||||||
line = f.readline()
|
line = f.readline()
|
||||||
@@ -628,9 +656,9 @@ class Rubar3(CommandLineSolver):
|
|||||||
h, s, q, z = read_data_line(f)
|
h, s, q, z = read_data_line(f)
|
||||||
set_and_compute_limites(
|
set_and_compute_limites(
|
||||||
reach, ind,
|
reach, ind,
|
||||||
(ph + pz + ph + h) / 2,
|
(pz + z + ph + h) / 2,
|
||||||
(pz + s) / 2,
|
(pq + q) / 2,
|
||||||
(pq + q) / 2
|
(ps + s) / 2
|
||||||
)
|
)
|
||||||
|
|
||||||
ph = h
|
ph = h
|
||||||
@@ -644,13 +672,107 @@ class Rubar3(CommandLineSolver):
|
|||||||
set_and_compute_limites(reach, ind, z+h, q, s)
|
set_and_compute_limites(reach, ind, z+h, q, s)
|
||||||
|
|
||||||
@timer
|
@timer
|
||||||
def results(self, study, repertory, qlog=None, name="0"):
|
def write_bin(self, study, fname, results, qlog=None, name="0"):
|
||||||
|
logger.info(f"write_bin: Start writing '{fname}' ...")
|
||||||
|
|
||||||
|
with open(fname, "wb") as f:
|
||||||
|
def newline(j): return np.asarray([j], dtype=np.int32).tofile(f)
|
||||||
|
def endline(j): return np.asarray([j], dtype=np.int32).tofile(f)
|
||||||
|
|
||||||
|
def write_int(i):
|
||||||
|
newline(len(i)*4)
|
||||||
|
np.array(i, dtype=np.int32).tofile(f)
|
||||||
|
endline(len(i)*4)
|
||||||
|
|
||||||
|
def write_float(i):
|
||||||
|
newline(len(i)*4)
|
||||||
|
np.array(i, dtype=np.float32).tofile(f)
|
||||||
|
endline(len(i)*4)
|
||||||
|
|
||||||
|
def write_float64(i):
|
||||||
|
newline(len(i)*8)
|
||||||
|
np.array(i, dtype=np.float64).tofile(f)
|
||||||
|
endline(len(i)*8)
|
||||||
|
|
||||||
|
def write_float64(i):
|
||||||
|
newline(len(i)*8)
|
||||||
|
np.array(i, dtype=np.float64).tofile(f)
|
||||||
|
endline(len(i)*8)
|
||||||
|
|
||||||
|
def write_data(npts, t, a, val):
|
||||||
|
newline(npts*4 + 13)
|
||||||
|
np.array(npts, dtype=np.int32).tofile(f)
|
||||||
|
np.array(t, dtype=np.float64).tofile(f)
|
||||||
|
np.array(bytearray(a.encode()), dtype=np.byte).tofile(f)
|
||||||
|
np.array(val, dtype=np.float32).tofile(f)
|
||||||
|
endline(npts*4 + 13)
|
||||||
|
|
||||||
|
ts_list = sorted(results.get("timestamps"))
|
||||||
|
profiles = []
|
||||||
|
for r in results.river.reachs:
|
||||||
|
profiles += r.profiles
|
||||||
|
|
||||||
|
# Meta data (1st line)
|
||||||
|
nb_reach = len(results.river)
|
||||||
|
nb_profile = len(profiles)
|
||||||
|
write_int([nb_reach, nb_profile, "82"])
|
||||||
|
|
||||||
|
# Reach information (2nd line)
|
||||||
|
is1 = []
|
||||||
|
is2 = []
|
||||||
|
ltmp = []
|
||||||
|
i = 1
|
||||||
|
for r in results.river.reachs:
|
||||||
|
j = i+len(r)-1
|
||||||
|
ltmp += [i, j]
|
||||||
|
is1.append(i)
|
||||||
|
is2.append(j)
|
||||||
|
i += i+len(r)
|
||||||
|
write_int(ltmp)
|
||||||
|
|
||||||
|
# X (3rd line)
|
||||||
|
rk = []
|
||||||
|
for r in results.river.reachs:
|
||||||
|
rk += r.geometry.get_rk()
|
||||||
|
write_float(rk)
|
||||||
|
|
||||||
|
# Z and Y (4th line)
|
||||||
|
# ltmp = []
|
||||||
|
# for r in results.river.reachs:
|
||||||
|
# for p in r.prifiles:
|
||||||
|
# ltpm.append(p.rk)
|
||||||
|
write_float(3*nb_profile*[0.0])
|
||||||
|
|
||||||
|
# Data
|
||||||
|
|
||||||
|
for timestamp in ts_list:
|
||||||
|
q = list(
|
||||||
|
map(
|
||||||
|
lambda p: p.get_ts_key(timestamp, "Q"),
|
||||||
|
profiles
|
||||||
|
)
|
||||||
|
)
|
||||||
|
write_data(nb_profile, timestamp, "Q", q)
|
||||||
|
z = list(
|
||||||
|
map(
|
||||||
|
lambda p: p.get_ts_key(timestamp, "Z"),
|
||||||
|
profiles
|
||||||
|
)
|
||||||
|
)
|
||||||
|
write_data(nb_profile, timestamp, "Z", z)
|
||||||
|
logger.info(f"write_bin: ... end with {len(ts_list)} timestamps")
|
||||||
|
|
||||||
|
@timer
|
||||||
|
def results(self, study, repertory, qlog=None, name=None):
|
||||||
results = Results(
|
results = Results(
|
||||||
study=study,
|
study=study,
|
||||||
solver=self,
|
solver=self,
|
||||||
repertory=repertory,
|
repertory=repertory,
|
||||||
name=name,
|
name=name,
|
||||||
)
|
)
|
||||||
|
if name is None:
|
||||||
|
name = study.name.replace(" ", "_")
|
||||||
|
|
||||||
results_file = f"profil.{name}"
|
results_file = f"profil.{name}"
|
||||||
|
|
||||||
fname = os.path.join(repertory, results_file)
|
fname = os.path.join(repertory, results_file)
|
||||||
|
|||||||
@@ -55,7 +55,7 @@ class AboutWindow(PamhyrDialog):
|
|||||||
|
|
||||||
label = self.get_label_text("label_version")
|
label = self.get_label_text("label_version")
|
||||||
label = label.replace("@version", version)
|
label = label.replace("@version", version)
|
||||||
label = label.replace("@codename", "(Adis-TS)")
|
label = label.replace("@codename", "")
|
||||||
self.set_label_text("label_version", label)
|
self.set_label_text("label_version", label)
|
||||||
|
|
||||||
# Authors
|
# Authors
|
||||||
|
|||||||
@@ -54,7 +54,7 @@ class Plot(PamhyrPlot):
|
|||||||
self._isometric_axis = False
|
self._isometric_axis = False
|
||||||
|
|
||||||
self._auto_relim_update = True
|
self._auto_relim_update = True
|
||||||
self._autoscale_update = True
|
self._autoscale_update = False
|
||||||
|
|
||||||
def custom_ticks(self):
|
def custom_ticks(self):
|
||||||
if self.data.header[0] != "time":
|
if self.data.header[0] != "time":
|
||||||
|
|||||||
@@ -58,7 +58,11 @@ logger = logging.getLogger()
|
|||||||
|
|
||||||
class TableModel(PamhyrTableModel):
|
class TableModel(PamhyrTableModel):
|
||||||
def get_true_data_row(self, row):
|
def get_true_data_row(self, row):
|
||||||
|
|
||||||
|
if len(self._data.data) > 0:
|
||||||
bc = self._data.data[row]
|
bc = self._data.data[row]
|
||||||
|
else:
|
||||||
|
return 0
|
||||||
|
|
||||||
return next(
|
return next(
|
||||||
map(
|
map(
|
||||||
@@ -103,6 +107,9 @@ class TableModel(PamhyrTableModel):
|
|||||||
if not index.isValid() or role != Qt.EditRole:
|
if not index.isValid() or role != Qt.EditRole:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
if self.is_same_data(index, value):
|
||||||
|
return False
|
||||||
|
|
||||||
row = index.row()
|
row = index.row()
|
||||||
column = index.column()
|
column = index.column()
|
||||||
|
|
||||||
|
|||||||
@@ -167,6 +167,9 @@ class TableModel(PamhyrTableModel):
|
|||||||
if not index.isValid() or role != Qt.EditRole:
|
if not index.isValid() or role != Qt.EditRole:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
if self.is_same_data(index, value):
|
||||||
|
return False
|
||||||
|
|
||||||
row = index.row()
|
row = index.row()
|
||||||
column = index.column()
|
column = index.column()
|
||||||
|
|
||||||
|
|||||||
@@ -55,7 +55,7 @@ class Plot(PamhyrPlot):
|
|||||||
self._isometric_axis = False
|
self._isometric_axis = False
|
||||||
|
|
||||||
self._auto_relim_update = True
|
self._auto_relim_update = True
|
||||||
self._autoscale_update = True
|
self._autoscale_update = False
|
||||||
|
|
||||||
def custom_ticks(self):
|
def custom_ticks(self):
|
||||||
if self.data.header[0] != "time":
|
if self.data.header[0] != "time":
|
||||||
|
|||||||
@@ -50,6 +50,22 @@ logger = logging.getLogger()
|
|||||||
|
|
||||||
|
|
||||||
class TableModel(PamhyrTableModel):
|
class TableModel(PamhyrTableModel):
|
||||||
|
def get_true_data_row(self, row):
|
||||||
|
if len(self._data.data) > 0:
|
||||||
|
bc = self._data.data[row]
|
||||||
|
else:
|
||||||
|
return 0
|
||||||
|
|
||||||
|
return next(
|
||||||
|
map(
|
||||||
|
lambda e: e[0],
|
||||||
|
filter(
|
||||||
|
lambda e: e[1] == bc,
|
||||||
|
enumerate(self._data._data)
|
||||||
|
)
|
||||||
|
), 0
|
||||||
|
)
|
||||||
|
|
||||||
def data(self, index, role):
|
def data(self, index, role):
|
||||||
if role == Qt.TextAlignmentRole:
|
if role == Qt.TextAlignmentRole:
|
||||||
return Qt.AlignHCenter | Qt.AlignVCenter
|
return Qt.AlignHCenter | Qt.AlignVCenter
|
||||||
@@ -63,6 +79,7 @@ class TableModel(PamhyrTableModel):
|
|||||||
value = QVariant()
|
value = QVariant()
|
||||||
|
|
||||||
if 0 <= column < 2:
|
if 0 <= column < 2:
|
||||||
|
row = self.get_true_data_row(row)
|
||||||
v = self._data._data[row][column]
|
v = self._data._data[row][column]
|
||||||
if self._data._types[column] == float:
|
if self._data._types[column] == float:
|
||||||
value = f"{v:.4f}"
|
value = f"{v:.4f}"
|
||||||
@@ -80,10 +97,15 @@ class TableModel(PamhyrTableModel):
|
|||||||
if not index.isValid() or role != Qt.EditRole:
|
if not index.isValid() or role != Qt.EditRole:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
if self.is_same_data(index, value):
|
||||||
|
return False
|
||||||
|
|
||||||
row = index.row()
|
row = index.row()
|
||||||
column = index.column()
|
column = index.column()
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
row = self.get_true_data_row(row)
|
||||||
|
|
||||||
self._undo.push(
|
self._undo.push(
|
||||||
SetDataCommand(
|
SetDataCommand(
|
||||||
self._data, row, column, value
|
self._data, row, column, value
|
||||||
@@ -110,14 +132,20 @@ class TableModel(PamhyrTableModel):
|
|||||||
|
|
||||||
def delete(self, rows, parent=QModelIndex()):
|
def delete(self, rows, parent=QModelIndex()):
|
||||||
self.beginRemoveRows(parent, rows[0], rows[-1])
|
self.beginRemoveRows(parent, rows[0], rows[-1])
|
||||||
|
_rows = list(
|
||||||
|
map(
|
||||||
|
lambda r: self.get_true_data_row(r),
|
||||||
|
rows
|
||||||
|
)
|
||||||
|
)
|
||||||
self._undo.push(
|
self._undo.push(
|
||||||
DelCommand(
|
DelCommand(
|
||||||
self._data, rows
|
self._data, _rows
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
self.endRemoveRows()
|
self.endRemoveRows()
|
||||||
|
self.layoutChanged.emit()
|
||||||
|
|
||||||
def sort(self, _reverse, parent=QModelIndex()):
|
def sort(self, _reverse, parent=QModelIndex()):
|
||||||
self.layoutAboutToBeChanged.emit()
|
self.layoutAboutToBeChanged.emit()
|
||||||
|
|||||||
@@ -61,9 +61,7 @@ class AddCommand(QUndoCommand):
|
|||||||
|
|
||||||
def redo(self):
|
def redo(self):
|
||||||
if self._new is None:
|
if self._new is None:
|
||||||
self._new = self._data.insert(self._index, (
|
self._new = self._data.add(self._index)
|
||||||
self._data._types[0](0), self._data._types[1](0.0)
|
|
||||||
))
|
|
||||||
else:
|
else:
|
||||||
self._data.insert(self._index, self._new)
|
self._data.insert(self._index, self._new)
|
||||||
|
|
||||||
@@ -82,10 +80,11 @@ class DelCommand(QUndoCommand):
|
|||||||
|
|
||||||
def undo(self):
|
def undo(self):
|
||||||
for row, el in self._bc:
|
for row, el in self._bc:
|
||||||
self._data.insert(row, el)
|
el.set_as_not_deleted()
|
||||||
|
|
||||||
def redo(self):
|
def redo(self):
|
||||||
self._data.delete_i(self._rows)
|
for row, el in self._bc:
|
||||||
|
el.set_as_deleted()
|
||||||
|
|
||||||
|
|
||||||
class PasteCommand(QUndoCommand):
|
class PasteCommand(QUndoCommand):
|
||||||
|
|||||||
@@ -63,6 +63,16 @@ class EditBoundaryConditionWindow(PamhyrWindow):
|
|||||||
|
|
||||||
name = trad[self._pamhyr_name]
|
name = trad[self._pamhyr_name]
|
||||||
|
|
||||||
|
if self._data is not None:
|
||||||
|
n = self._data.node
|
||||||
|
node_name = next(filter(
|
||||||
|
lambda x: x.id == n, study.river._nodes
|
||||||
|
)).name
|
||||||
|
name += (
|
||||||
|
f" - {study.name} " +
|
||||||
|
f"({node_name})"
|
||||||
|
)
|
||||||
|
|
||||||
super(EditBoundaryConditionWindow, self).__init__(
|
super(EditBoundaryConditionWindow, self).__init__(
|
||||||
title=name,
|
title=name,
|
||||||
study=study,
|
study=study,
|
||||||
@@ -71,16 +81,6 @@ class EditBoundaryConditionWindow(PamhyrWindow):
|
|||||||
parent=parent
|
parent=parent
|
||||||
)
|
)
|
||||||
|
|
||||||
if self._data is not None:
|
|
||||||
n = self._data.node
|
|
||||||
node_name = next(filter(
|
|
||||||
lambda x: x.id == n, self._study.river._nodes
|
|
||||||
)).name
|
|
||||||
name += (
|
|
||||||
f" - {study.name} " +
|
|
||||||
f"({node_name})"
|
|
||||||
)
|
|
||||||
|
|
||||||
self._hash_data.append(data)
|
self._hash_data.append(data)
|
||||||
|
|
||||||
self.setup_table()
|
self.setup_table()
|
||||||
@@ -89,6 +89,12 @@ class EditBoundaryConditionWindow(PamhyrWindow):
|
|||||||
|
|
||||||
def setup_table(self):
|
def setup_table(self):
|
||||||
table_headers = self._trad.get_dict("table_headers")
|
table_headers = self._trad.get_dict("table_headers")
|
||||||
|
|
||||||
|
if self._study.is_editable():
|
||||||
|
editable_headers = self._trad.get_dict("table_headers")
|
||||||
|
else:
|
||||||
|
editable_headers = []
|
||||||
|
|
||||||
if self._data.type == "Concentration":
|
if self._data.type == "Concentration":
|
||||||
self._data.header = ["time", "concentration"]
|
self._data.header = ["time", "concentration"]
|
||||||
else:
|
else:
|
||||||
@@ -108,7 +114,7 @@ class EditBoundaryConditionWindow(PamhyrWindow):
|
|||||||
self._table = TableModel(
|
self._table = TableModel(
|
||||||
table_view=table,
|
table_view=table,
|
||||||
table_headers=headers,
|
table_headers=headers,
|
||||||
editable_headers=self._data.header,
|
editable_headers=editable_headers,
|
||||||
delegates={
|
delegates={
|
||||||
# "time": self._delegate_time,
|
# "time": self._delegate_time,
|
||||||
},
|
},
|
||||||
@@ -141,6 +147,7 @@ class EditBoundaryConditionWindow(PamhyrWindow):
|
|||||||
self.plot.draw()
|
self.plot.draw()
|
||||||
|
|
||||||
def setup_connections(self):
|
def setup_connections(self):
|
||||||
|
if self._study.is_editable():
|
||||||
self.find(QAction, "action_add").triggered.connect(self.add)
|
self.find(QAction, "action_add").triggered.connect(self.add)
|
||||||
self.find(QAction, "action_del").triggered.connect(self.delete)
|
self.find(QAction, "action_del").triggered.connect(self.delete)
|
||||||
self.find(QAction, "action_sort").triggered.connect(self.sort)
|
self.find(QAction, "action_sort").triggered.connect(self.sort)
|
||||||
|
|||||||
@@ -43,7 +43,7 @@ from View.Tools.PamhyrTable import PamhyrTableModel
|
|||||||
|
|
||||||
from View.BoundaryConditionsAdisTS.UndoCommand import (
|
from View.BoundaryConditionsAdisTS.UndoCommand import (
|
||||||
SetNodeCommand, SetTypeCommand,
|
SetNodeCommand, SetTypeCommand,
|
||||||
AddCommand, DelCommand, SetPolCommand
|
AddCommand, DelCommand
|
||||||
)
|
)
|
||||||
from View.BoundaryCondition.translate import BC_types
|
from View.BoundaryCondition.translate import BC_types
|
||||||
|
|
||||||
@@ -112,18 +112,37 @@ class ComboBoxDelegate(QItemDelegate):
|
|||||||
|
|
||||||
|
|
||||||
class TableModel(PamhyrTableModel):
|
class TableModel(PamhyrTableModel):
|
||||||
def __init__(self, pollutant=None, bc_list=None, trad=None, **kwargs):
|
def __init__(self, bc_list=None, pollutant_bc_list=None,
|
||||||
|
trad=None, **kwargs):
|
||||||
self._trad = trad
|
self._trad = trad
|
||||||
self._bc_list = bc_list
|
self._bc_list = bc_list
|
||||||
self._pollutant = pollutant
|
self._pollutant = pollutant_bc_list.id
|
||||||
|
|
||||||
super(TableModel, self).__init__(trad=trad, **kwargs)
|
super(TableModel, self).__init__(trad=trad, **kwargs)
|
||||||
|
|
||||||
def rowCount(self, parent):
|
def _setup_lst(self):
|
||||||
return len(self._bc_list)
|
self._lst = list(
|
||||||
|
filter(
|
||||||
|
lambda bc: bc.pollutant == self._pollutant,
|
||||||
|
self._bc_list.lst
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
def get(self, row):
|
||||||
|
if row < 0 or row >= len(self._lst):
|
||||||
|
return None
|
||||||
|
return self._lst[row]
|
||||||
|
|
||||||
|
def _global_row(self, row):
|
||||||
|
bc = self.get(row)
|
||||||
|
if bc is None:
|
||||||
|
return None
|
||||||
|
return self._bc_list.index(bc)
|
||||||
|
|
||||||
|
def rowCount(self, parent=QModelIndex()):
|
||||||
|
return len(self._lst)
|
||||||
|
|
||||||
def data(self, index, role):
|
def data(self, index, role):
|
||||||
|
|
||||||
if role != Qt.ItemDataRole.DisplayRole:
|
if role != Qt.ItemDataRole.DisplayRole:
|
||||||
return QVariant()
|
return QVariant()
|
||||||
|
|
||||||
@@ -131,12 +150,12 @@ class TableModel(PamhyrTableModel):
|
|||||||
column = index.column()
|
column = index.column()
|
||||||
|
|
||||||
if self._headers[column] == "type":
|
if self._headers[column] == "type":
|
||||||
n = self._bc_list.get(row).type
|
n = self._lst[row].type
|
||||||
if n is None or n == "":
|
if n is None or n == "":
|
||||||
return self._trad["not_associated"]
|
return self._trad["not_associated"]
|
||||||
return n
|
return n
|
||||||
elif self._headers[column] == "node":
|
elif self._headers[column] == "node":
|
||||||
n = self._bc_list.get(row).node
|
n = self._lst[row].node
|
||||||
if n is None:
|
if n is None:
|
||||||
return self._trad["not_associated"]
|
return self._trad["not_associated"]
|
||||||
tmp = next(filter(lambda x: x.id == n, self._data._nodes), None)
|
tmp = next(filter(lambda x: x.id == n, self._data._nodes), None)
|
||||||
@@ -144,18 +163,6 @@ class TableModel(PamhyrTableModel):
|
|||||||
return tmp.name
|
return tmp.name
|
||||||
else:
|
else:
|
||||||
return self._trad["not_associated"]
|
return self._trad["not_associated"]
|
||||||
elif self._headers[column] == "pol":
|
|
||||||
n = self._bc_list.get(row).pollutant
|
|
||||||
if n is None or n == "not_associated" or n == "":
|
|
||||||
return self._trad["not_associated"]
|
|
||||||
tmp = next(filter(lambda x: x.id == n,
|
|
||||||
self._data._Pollutants.Pollutants_List
|
|
||||||
),
|
|
||||||
None)
|
|
||||||
if tmp is not None:
|
|
||||||
return tmp.name
|
|
||||||
else:
|
|
||||||
return self._trad["not_associated"]
|
|
||||||
|
|
||||||
return QVariant()
|
return QVariant()
|
||||||
|
|
||||||
@@ -163,38 +170,32 @@ class TableModel(PamhyrTableModel):
|
|||||||
if not index.isValid() or role != Qt.EditRole:
|
if not index.isValid() or role != Qt.EditRole:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
if self.is_same_data(index, value):
|
||||||
|
return False
|
||||||
|
|
||||||
row = index.row()
|
row = index.row()
|
||||||
column = index.column()
|
column = index.column()
|
||||||
|
|
||||||
try:
|
try:
|
||||||
if self._headers[column] == "type":
|
if self._headers[column] == "type":
|
||||||
|
global_row = self._global_row(row)
|
||||||
self._undo.push(
|
self._undo.push(
|
||||||
SetTypeCommand(
|
SetTypeCommand(
|
||||||
self._bc_list, row, value
|
self._bc_list, global_row, value
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
elif self._headers[column] == "node":
|
elif self._headers[column] == "node":
|
||||||
|
global_row = self._global_row(row)
|
||||||
|
node = next(
|
||||||
|
filter(lambda n: n.name == value, self._data.nodes()),
|
||||||
|
None
|
||||||
|
)
|
||||||
self._undo.push(
|
self._undo.push(
|
||||||
SetNodeCommand(
|
SetNodeCommand(
|
||||||
self._bc_list, row, self._data.node(value)
|
self._bc_list, global_row, node
|
||||||
)
|
|
||||||
)
|
|
||||||
elif self._headers[column] == "pol":
|
|
||||||
if value == self._trad["not_associated"]:
|
|
||||||
self._undo.push(
|
|
||||||
SetPolCommand(
|
|
||||||
self._bc_list, row, None
|
|
||||||
)
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
pol = next(filter(lambda x: x.name == value,
|
|
||||||
self._data._Pollutants.Pollutants_List)
|
|
||||||
)
|
|
||||||
self._undo.push(
|
|
||||||
SetPolCommand(
|
|
||||||
self._bc_list, row, pol.id
|
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.info(e)
|
logger.info(e)
|
||||||
logger.debug(traceback.format_exc())
|
logger.debug(traceback.format_exc())
|
||||||
@@ -203,33 +204,44 @@ class TableModel(PamhyrTableModel):
|
|||||||
return True
|
return True
|
||||||
|
|
||||||
def add(self, row, parent=QModelIndex()):
|
def add(self, row, parent=QModelIndex()):
|
||||||
self.beginInsertRows(parent, row, row - 1)
|
row = len(self._lst)
|
||||||
|
self.beginInsertRows(parent, row, row)
|
||||||
self._undo.push(
|
self._undo.push(
|
||||||
AddCommand(
|
AddCommand(
|
||||||
self._pollutant, self._bc_list, row
|
self._pollutant, self._bc_list, len(self._bc_list)
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
self._setup_lst()
|
||||||
self.endInsertRows()
|
self.endInsertRows()
|
||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
|
|
||||||
def delete(self, rows, parent=QModelIndex()):
|
def delete(self, rows, parent=QModelIndex()):
|
||||||
self.beginRemoveRows(parent, rows[0], rows[-1])
|
self.beginRemoveRows(parent, rows[0], rows[-1])
|
||||||
|
|
||||||
|
row_by_bc = {
|
||||||
|
id(bc): row for row, bc in enumerate(self._bc_list._lst)
|
||||||
|
}
|
||||||
|
global_rows = [
|
||||||
|
row_by_bc[id(self._lst[row])]
|
||||||
|
for row in rows
|
||||||
|
if 0 <= row < len(self._lst)
|
||||||
|
]
|
||||||
self._undo.push(
|
self._undo.push(
|
||||||
DelCommand(
|
DelCommand(
|
||||||
self._bc_list, rows
|
self._bc_list, global_rows
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
self._setup_lst()
|
||||||
self.endRemoveRows()
|
self.endRemoveRows()
|
||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
|
|
||||||
def undo(self):
|
def undo(self):
|
||||||
self._undo.undo()
|
self._undo.undo()
|
||||||
|
self._setup_lst()
|
||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
|
|
||||||
def redo(self):
|
def redo(self):
|
||||||
self._undo.redo()
|
self._undo.redo()
|
||||||
|
self._setup_lst()
|
||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
@@ -36,7 +36,7 @@ class SetNodeCommand(QUndoCommand):
|
|||||||
self._bcs = bcs
|
self._bcs = bcs
|
||||||
self._index = index
|
self._index = index
|
||||||
self._old = self._bcs.get(self._index).node
|
self._old = self._bcs.get(self._index).node
|
||||||
self._new = node.id
|
self._new = node.id if node is not None else None
|
||||||
|
|
||||||
def undo(self):
|
def undo(self):
|
||||||
self._bcs.get(self._index).node = self._old
|
self._bcs.get(self._index).node = self._old
|
||||||
@@ -62,22 +62,6 @@ class SetTypeCommand(QUndoCommand):
|
|||||||
self._bcs.get(self._index).type = self._new
|
self._bcs.get(self._index).type = self._new
|
||||||
|
|
||||||
|
|
||||||
class SetPolCommand(QUndoCommand):
|
|
||||||
def __init__(self, bcs, index, pollutant):
|
|
||||||
QUndoCommand.__init__(self)
|
|
||||||
|
|
||||||
self._bcs = bcs
|
|
||||||
self._index = index
|
|
||||||
self._old = self._bcs.get(self._index).pollutant
|
|
||||||
self._new = pollutant
|
|
||||||
|
|
||||||
def undo(self):
|
|
||||||
self._bcs.get(self._index).pollutant = self._old
|
|
||||||
|
|
||||||
def redo(self):
|
|
||||||
self._bcs.get(self._index).pollutant = self._new
|
|
||||||
|
|
||||||
|
|
||||||
class AddCommand(QUndoCommand):
|
class AddCommand(QUndoCommand):
|
||||||
def __init__(self, pollutant, bcs, index):
|
def __init__(self, pollutant, bcs, index):
|
||||||
QUndoCommand.__init__(self)
|
QUndoCommand.__init__(self)
|
||||||
@@ -106,7 +90,7 @@ class DelCommand(QUndoCommand):
|
|||||||
|
|
||||||
self._bc = []
|
self._bc = []
|
||||||
for row in rows:
|
for row in rows:
|
||||||
self._bc.append((row, self._bcs.get(row)))
|
self._bc.append((row, self._bcs._lst[row]))
|
||||||
self._bc.sort()
|
self._bc.sort()
|
||||||
|
|
||||||
def undo(self):
|
def undo(self):
|
||||||
|
|||||||
@@ -58,11 +58,22 @@ class BoundaryConditionAdisTSWindow(PamhyrWindow):
|
|||||||
_pamhyr_ui = "BoundaryConditionsAdisTS"
|
_pamhyr_ui = "BoundaryConditionsAdisTS"
|
||||||
_pamhyr_name = "Boundary conditions AdisTS"
|
_pamhyr_name = "Boundary conditions AdisTS"
|
||||||
|
|
||||||
def __init__(self, study=None, config=None, parent=None):
|
def __init__(self, data=None, pollutant_id=None, study=None,
|
||||||
|
config=None, parent=None):
|
||||||
|
self._data = data
|
||||||
|
self._pollutant_id = pollutant_id
|
||||||
trad = BCAdisTSTranslate()
|
trad = BCAdisTSTranslate()
|
||||||
|
|
||||||
|
self._pollutant_name = next(
|
||||||
|
(x.name for x in study._river._Pollutants.Pollutants_List
|
||||||
|
if x.id == self._pollutant_id),
|
||||||
|
None
|
||||||
|
)
|
||||||
|
|
||||||
name = (
|
name = (
|
||||||
trad[self._pamhyr_name] +
|
trad[self._pamhyr_name] +
|
||||||
" - " + study.name
|
" - " + study.name +
|
||||||
|
" - " + self._pollutant_name
|
||||||
)
|
)
|
||||||
|
|
||||||
super(BoundaryConditionAdisTSWindow, self).__init__(
|
super(BoundaryConditionAdisTSWindow, self).__init__(
|
||||||
@@ -73,15 +84,12 @@ class BoundaryConditionAdisTSWindow(PamhyrWindow):
|
|||||||
parent=parent
|
parent=parent
|
||||||
)
|
)
|
||||||
|
|
||||||
self._pollutants_lst = self._study._river._Pollutants
|
|
||||||
self._bcs = self._study.river.boundary_conditions_adists
|
self._bcs = self._study.river.boundary_conditions_adists
|
||||||
|
|
||||||
self.setup_graph()
|
self.setup_graph()
|
||||||
self.setup_table()
|
self.setup_table()
|
||||||
self.setup_connections()
|
self.setup_connections()
|
||||||
|
|
||||||
self.ui.setWindowTitle(self._title)
|
|
||||||
|
|
||||||
def setup_table(self):
|
def setup_table(self):
|
||||||
self._delegate_type = ComboBoxDelegate(
|
self._delegate_type = ComboBoxDelegate(
|
||||||
trad=self._trad,
|
trad=self._trad,
|
||||||
@@ -95,25 +103,24 @@ class BoundaryConditionAdisTSWindow(PamhyrWindow):
|
|||||||
mode="node",
|
mode="node",
|
||||||
parent=self
|
parent=self
|
||||||
)
|
)
|
||||||
self._delegate_pol = ComboBoxDelegate(
|
|
||||||
trad=self._trad,
|
if self._study.is_editable():
|
||||||
data=self._study.river,
|
editable_headers = self._trad.get_dict("table_headers")
|
||||||
mode="pol",
|
else:
|
||||||
parent=self
|
editable_headers = []
|
||||||
)
|
|
||||||
|
|
||||||
table = self.find(QTableView, f"tableView")
|
table = self.find(QTableView, f"tableView")
|
||||||
self._table = TableModel(
|
self._table = TableModel(
|
||||||
table_view=table,
|
table_view=table,
|
||||||
table_headers=self._trad.get_dict("table_headers"),
|
table_headers=self._trad.get_dict("table_headers"),
|
||||||
editable_headers=["type", "node", "pol"],
|
editable_headers=editable_headers,
|
||||||
delegates={
|
delegates={
|
||||||
"type": self._delegate_type,
|
"type": self._delegate_type,
|
||||||
"node": self._delegate_node,
|
"node": self._delegate_node,
|
||||||
"pol": self._delegate_pol,
|
|
||||||
},
|
},
|
||||||
trad=self._trad,
|
trad=self._trad,
|
||||||
bc_list=self._study.river.boundary_conditions_adists,
|
bc_list=self._study.river.boundary_conditions_adists,
|
||||||
|
pollutant_bc_list=self._data,
|
||||||
undo=self._undo_stack,
|
undo=self._undo_stack,
|
||||||
data=self._study.river
|
data=self._study.river
|
||||||
)
|
)
|
||||||
@@ -133,8 +140,10 @@ class BoundaryConditionAdisTSWindow(PamhyrWindow):
|
|||||||
self.graph_layout.addWidget(self.graph_widget)
|
self.graph_layout.addWidget(self.graph_widget)
|
||||||
|
|
||||||
def setup_connections(self):
|
def setup_connections(self):
|
||||||
|
if self._study.is_editable():
|
||||||
self.find(QAction, "action_add").triggered.connect(self.add)
|
self.find(QAction, "action_add").triggered.connect(self.add)
|
||||||
self.find(QAction, "action_del").triggered.connect(self.delete)
|
self.find(QAction, "action_del").triggered.connect(self.delete)
|
||||||
|
|
||||||
self.find(QAction, "action_edit").triggered.connect(self.edit)
|
self.find(QAction, "action_edit").triggered.connect(self.edit)
|
||||||
|
|
||||||
def index_selected_row(self):
|
def index_selected_row(self):
|
||||||
@@ -156,7 +165,7 @@ class BoundaryConditionAdisTSWindow(PamhyrWindow):
|
|||||||
)
|
)
|
||||||
|
|
||||||
def add(self):
|
def add(self):
|
||||||
self._table.add(len(self._bcs))
|
self._table.add(self._table.rowCount())
|
||||||
|
|
||||||
def delete(self):
|
def delete(self):
|
||||||
rows = self.index_selected_rows()
|
rows = self.index_selected_rows()
|
||||||
@@ -180,7 +189,7 @@ class BoundaryConditionAdisTSWindow(PamhyrWindow):
|
|||||||
def edit(self):
|
def edit(self):
|
||||||
rows = self.index_selected_rows()
|
rows = self.index_selected_rows()
|
||||||
for row in rows:
|
for row in rows:
|
||||||
data = self._bcs.get(row)
|
data = self._table.get(row)
|
||||||
|
|
||||||
if data.node is None:
|
if data.node is None:
|
||||||
continue
|
continue
|
||||||
|
|||||||
@@ -34,5 +34,4 @@ class BCAdisTSTranslate(MainTranslate):
|
|||||||
self._sub_dict["table_headers"] = {
|
self._sub_dict["table_headers"] = {
|
||||||
"type": self._dict["type"],
|
"type": self._dict["type"],
|
||||||
"node": _translate("BoundaryCondition", "Node"),
|
"node": _translate("BoundaryCondition", "Node"),
|
||||||
"pol": _translate("BoundaryCondition", "Pollutant")
|
|
||||||
}
|
}
|
||||||
@@ -46,25 +46,28 @@ _translate = QCoreApplication.translate
|
|||||||
|
|
||||||
|
|
||||||
class ComboBoxDelegate(QItemDelegate):
|
class ComboBoxDelegate(QItemDelegate):
|
||||||
def __init__(self, data=None, ic_spec_lst=None,
|
def __init__(self, data=None, d90_spec_lst=None,
|
||||||
trad=None, parent=None, mode="reaches"):
|
trad=None, parent=None, mode="reaches"):
|
||||||
super(ComboBoxDelegate, self).__init__(parent)
|
super(ComboBoxDelegate, self).__init__(parent)
|
||||||
|
|
||||||
self._data = data
|
self._data = data
|
||||||
self._mode = mode
|
self._mode = mode
|
||||||
self._trad = trad
|
self._trad = trad
|
||||||
self._ic_spec_lst = ic_spec_lst
|
self._d90_spec_lst = d90_spec_lst
|
||||||
|
|
||||||
def createEditor(self, parent, option, index):
|
def createEditor(self, parent, option, index):
|
||||||
self.editor = QComboBox(parent)
|
self.editor = QComboBox(parent)
|
||||||
|
|
||||||
val = []
|
val = []
|
||||||
if self._mode == "rk":
|
if self._mode == "rk":
|
||||||
reach_id = self._ic_spec_lst[index.row()].reach
|
reach_id = self._d90_spec_lst[index.row()].reach
|
||||||
|
|
||||||
reach = next(filter(
|
reach = next(
|
||||||
|
filter(
|
||||||
lambda edge: edge.id == reach_id, self._data.edges()
|
lambda edge: edge.id == reach_id, self._data.edges()
|
||||||
))
|
),
|
||||||
|
None
|
||||||
|
)
|
||||||
|
|
||||||
if reach_id is not None:
|
if reach_id is not None:
|
||||||
val = list(
|
val = list(
|
||||||
@@ -116,9 +119,16 @@ class D90TableModel(PamhyrTableModel):
|
|||||||
super(D90TableModel, self).__init__(data=data, **kwargs)
|
super(D90TableModel, self).__init__(data=data, **kwargs)
|
||||||
|
|
||||||
self._data = data
|
self._data = data
|
||||||
|
if self._undo is not None:
|
||||||
|
self._undo.indexChanged.connect(lambda _: self.update())
|
||||||
|
|
||||||
def _setup_lst(self):
|
def _setup_lst(self):
|
||||||
self._lst = self._data._data
|
self._lst = list(
|
||||||
|
filter(
|
||||||
|
lambda d90: d90._deleted is False,
|
||||||
|
self._data._data
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
def rowCount(self, parent):
|
def rowCount(self, parent):
|
||||||
return len(self._lst)
|
return len(self._lst)
|
||||||
@@ -164,6 +174,9 @@ class D90TableModel(PamhyrTableModel):
|
|||||||
if not index.isValid() or role != Qt.EditRole:
|
if not index.isValid() or role != Qt.EditRole:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
if self.is_same_data(index, value):
|
||||||
|
return False
|
||||||
|
|
||||||
row = index.row()
|
row = index.row()
|
||||||
column = index.column()
|
column = index.column()
|
||||||
|
|
||||||
@@ -197,25 +210,35 @@ class D90TableModel(PamhyrTableModel):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
self._setup_lst()
|
||||||
self.endInsertRows()
|
self.endInsertRows()
|
||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
|
|
||||||
def delete(self, rows, parent=QModelIndex()):
|
def delete(self, rows, parent=QModelIndex()):
|
||||||
self.beginRemoveRows(parent, rows[0], rows[-1])
|
self.beginRemoveRows(parent, rows[0], rows[-1])
|
||||||
|
|
||||||
|
data_rows = {
|
||||||
|
id(d90): i for i, d90 in enumerate(self._data._data)
|
||||||
|
}
|
||||||
self._undo.push(
|
self._undo.push(
|
||||||
DelCommand(
|
DelCommand(
|
||||||
self._data, self._lst, rows
|
self._data,
|
||||||
|
self._data._data,
|
||||||
|
[data_rows[id(self._lst[row])] for row in rows
|
||||||
|
if 0 <= row < len(self._lst)]
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
self._setup_lst()
|
||||||
self.endRemoveRows()
|
self.endRemoveRows()
|
||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
|
|
||||||
def undo(self):
|
def undo(self):
|
||||||
self._undo.undo()
|
self._undo.undo()
|
||||||
|
self._setup_lst()
|
||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
|
|
||||||
def redo(self):
|
def redo(self):
|
||||||
self._undo.redo()
|
self._undo.redo()
|
||||||
|
self._setup_lst()
|
||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
@@ -55,9 +55,9 @@ class D90TableDefaultModel(PamhyrTableModel):
|
|||||||
row = index.row()
|
row = index.row()
|
||||||
column = index.column()
|
column = index.column()
|
||||||
|
|
||||||
if self._headers[column] is "name":
|
if self._headers[column] == "name":
|
||||||
return self._data[row].name
|
return self._data[row].name
|
||||||
elif self._headers[column] is "d90":
|
elif self._headers[column] == "d90":
|
||||||
n = self._data[row].d90
|
n = self._data[row].d90
|
||||||
if n is None:
|
if n is None:
|
||||||
return self._trad['not_associated']
|
return self._trad['not_associated']
|
||||||
|
|||||||
@@ -112,11 +112,11 @@ class SetCommandSpec(QUndoCommand):
|
|||||||
|
|
||||||
|
|
||||||
class AddCommand(QUndoCommand):
|
class AddCommand(QUndoCommand):
|
||||||
def __init__(self, data, ics_spec, index):
|
def __init__(self, data, d90_spec, index):
|
||||||
QUndoCommand.__init__(self)
|
QUndoCommand.__init__(self)
|
||||||
|
|
||||||
self._data = data
|
self._data = data
|
||||||
self._ics_spec = ics_spec
|
self._d90_spec = d90_spec
|
||||||
self._index = index
|
self._index = index
|
||||||
self._new = None
|
self._new = None
|
||||||
|
|
||||||
@@ -131,21 +131,20 @@ class AddCommand(QUndoCommand):
|
|||||||
|
|
||||||
|
|
||||||
class DelCommand(QUndoCommand):
|
class DelCommand(QUndoCommand):
|
||||||
def __init__(self, data, ics_spec, rows):
|
def __init__(self, data, d90_spec, rows):
|
||||||
QUndoCommand.__init__(self)
|
QUndoCommand.__init__(self)
|
||||||
|
|
||||||
self._data = data
|
self._data = data
|
||||||
self._ics_spec = ics_spec
|
self._d90_spec = d90_spec
|
||||||
self._rows = rows
|
self._rows = rows
|
||||||
# self._data = data
|
|
||||||
|
|
||||||
self._ic = []
|
self._d90 = []
|
||||||
for row in rows:
|
for row in rows:
|
||||||
self._ic.append((row, self._ics_spec[row]))
|
self._d90.append((row, self._d90_spec[row]))
|
||||||
self._ic.sort()
|
self._d90.sort()
|
||||||
|
|
||||||
def undo(self):
|
def undo(self):
|
||||||
for row, el in self._ic:
|
for row, el in self._d90:
|
||||||
self._data.insert(row, el)
|
self._data.insert(row, el)
|
||||||
|
|
||||||
def redo(self):
|
def redo(self):
|
||||||
|
|||||||
@@ -43,7 +43,7 @@ from PyQt5.QtWidgets import (
|
|||||||
|
|
||||||
from Modules import Modules
|
from Modules import Modules
|
||||||
|
|
||||||
from View.InitialConditionsAdisTS.UndoCommand import (
|
from View.D90AdisTS.UndoCommand import (
|
||||||
SetCommand,
|
SetCommand,
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -93,18 +93,21 @@ class D90AdisTSWindow(PamhyrWindow):
|
|||||||
|
|
||||||
self.setup_table()
|
self.setup_table()
|
||||||
|
|
||||||
self.ui.setWindowTitle(self._title)
|
|
||||||
|
|
||||||
def setup_table(self):
|
def setup_table(self):
|
||||||
|
|
||||||
path_icons = os.path.join(self._get_ui_directory(), f"ressources")
|
path_icons = os.path.join(self._get_ui_directory(), f"ressources")
|
||||||
|
|
||||||
|
if self._study.is_editable():
|
||||||
|
editable_headers = self._trad.get_dict("table_headers")
|
||||||
|
else:
|
||||||
|
editable_headers = []
|
||||||
|
|
||||||
table_default = self.find(QTableView, f"tableView")
|
table_default = self.find(QTableView, f"tableView")
|
||||||
|
|
||||||
self._table = D90TableDefaultModel(
|
self._table = D90TableDefaultModel(
|
||||||
table_view=table_default,
|
table_view=table_default,
|
||||||
table_headers=self._trad.get_dict("table_headers"),
|
table_headers=self._trad.get_dict("table_headers"),
|
||||||
editable_headers=["name", "d90"],
|
editable_headers=editable_headers,
|
||||||
delegates={},
|
delegates={},
|
||||||
data=self._data,
|
data=self._data,
|
||||||
undo=self._undo_stack,
|
undo=self._undo_stack,
|
||||||
@@ -124,9 +127,14 @@ class D90AdisTSWindow(PamhyrWindow):
|
|||||||
|
|
||||||
action_add = QAction(self)
|
action_add = QAction(self)
|
||||||
action_add.setIcon(QIcon(os.path.join(path_icons, f"add.png")))
|
action_add.setIcon(QIcon(os.path.join(path_icons, f"add.png")))
|
||||||
|
|
||||||
|
if self._study.is_editable():
|
||||||
action_add.triggered.connect(self.add)
|
action_add.triggered.connect(self.add)
|
||||||
|
|
||||||
action_delete = QAction(self)
|
action_delete = QAction(self)
|
||||||
action_delete.setIcon(QIcon(os.path.join(path_icons, f"del.png")))
|
action_delete.setIcon(QIcon(os.path.join(path_icons, f"del.png")))
|
||||||
|
|
||||||
|
if self._study.is_editable():
|
||||||
action_delete.triggered.connect(self.delete)
|
action_delete.triggered.connect(self.delete)
|
||||||
|
|
||||||
toolBar.addAction(action_add)
|
toolBar.addAction(action_add)
|
||||||
@@ -138,22 +146,27 @@ class D90AdisTSWindow(PamhyrWindow):
|
|||||||
self._delegate_reach = ComboBoxDelegate(
|
self._delegate_reach = ComboBoxDelegate(
|
||||||
trad=self._trad,
|
trad=self._trad,
|
||||||
data=self._study.river,
|
data=self._study.river,
|
||||||
ic_spec_lst=self._data[0]._data,
|
d90_spec_lst=self._data[0]._data,
|
||||||
parent=self,
|
parent=self,
|
||||||
mode="reaches"
|
mode="reaches"
|
||||||
)
|
)
|
||||||
self._delegate_rk = ComboBoxDelegate(
|
self._delegate_rk = ComboBoxDelegate(
|
||||||
trad=self._trad,
|
trad=self._trad,
|
||||||
data=self._study.river,
|
data=self._study.river,
|
||||||
ic_spec_lst=self._data[0]._data,
|
d90_spec_lst=self._data[0]._data,
|
||||||
parent=self,
|
parent=self,
|
||||||
mode="rk"
|
mode="rk"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
if self._study.is_editable():
|
||||||
|
editable_headers_spec = self._trad.get_dict("table_headers_spec")
|
||||||
|
else:
|
||||||
|
editable_headers_spec = []
|
||||||
|
|
||||||
self._table_spec = D90TableModel(
|
self._table_spec = D90TableModel(
|
||||||
table_view=self.table_spec,
|
table_view=self.table_spec,
|
||||||
table_headers=self._trad.get_dict("table_headers_spec"),
|
table_headers=self._trad.get_dict("table_headers_spec"),
|
||||||
editable_headers=["name", "reach", "start_rk", "end_rk", "d90"],
|
editable_headers=editable_headers_spec,
|
||||||
delegates={
|
delegates={
|
||||||
"reach": self._delegate_reach,
|
"reach": self._delegate_reach,
|
||||||
"start_rk": self._delegate_rk,
|
"start_rk": self._delegate_rk,
|
||||||
@@ -227,15 +240,15 @@ class D90AdisTSWindow(PamhyrWindow):
|
|||||||
|
|
||||||
table = list(
|
table = list(
|
||||||
map(
|
map(
|
||||||
lambda eic: list(
|
lambda ed90: list(
|
||||||
map(
|
map(
|
||||||
lambda k: eic[1][k],
|
lambda k: ed90[1][k],
|
||||||
["rk", "discharge", "elevation"]
|
["rk", "discharge", "elevation"]
|
||||||
)
|
)
|
||||||
),
|
),
|
||||||
filter(
|
filter(
|
||||||
lambda eic: eic[0] in rows,
|
lambda ed90: ed90[0] in rows,
|
||||||
enumerate(self._ics.lst())
|
enumerate(self._d90.lst())
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
@@ -264,11 +277,27 @@ class D90AdisTSWindow(PamhyrWindow):
|
|||||||
self._update()
|
self._update()
|
||||||
|
|
||||||
def _undo(self):
|
def _undo(self):
|
||||||
|
undo_stack = self._undo_stack
|
||||||
|
if undo_stack is None or not undo_stack.canUndo():
|
||||||
|
return
|
||||||
|
|
||||||
|
if isinstance(undo_stack.command(undo_stack.index() - 1), SetCommand):
|
||||||
self._table.undo()
|
self._table.undo()
|
||||||
|
else:
|
||||||
|
self._table_spec.undo()
|
||||||
|
|
||||||
self._update()
|
self._update()
|
||||||
|
|
||||||
def _redo(self):
|
def _redo(self):
|
||||||
|
undo_stack = self._undo_stack
|
||||||
|
if undo_stack is None or not undo_stack.canRedo():
|
||||||
|
return
|
||||||
|
|
||||||
|
if isinstance(undo_stack.command(undo_stack.index()), SetCommand):
|
||||||
self._table.redo()
|
self._table.redo()
|
||||||
|
else:
|
||||||
|
self._table_spec.redo()
|
||||||
|
|
||||||
self._update()
|
self._update()
|
||||||
|
|
||||||
def add(self):
|
def add(self):
|
||||||
|
|||||||
@@ -46,25 +46,28 @@ _translate = QCoreApplication.translate
|
|||||||
|
|
||||||
|
|
||||||
class ComboBoxDelegate(QItemDelegate):
|
class ComboBoxDelegate(QItemDelegate):
|
||||||
def __init__(self, data=None, ic_spec_lst=None,
|
def __init__(self, data=None, dif_spec_lst=None,
|
||||||
trad=None, parent=None, mode="reaches"):
|
trad=None, parent=None, mode="reaches"):
|
||||||
super(ComboBoxDelegate, self).__init__(parent)
|
super(ComboBoxDelegate, self).__init__(parent)
|
||||||
|
|
||||||
self._data = data
|
self._data = data
|
||||||
self._mode = mode
|
self._mode = mode
|
||||||
self._trad = trad
|
self._trad = trad
|
||||||
self._ic_spec_lst = ic_spec_lst
|
self._dif_spec_lst = dif_spec_lst
|
||||||
|
|
||||||
def createEditor(self, parent, option, index):
|
def createEditor(self, parent, option, index):
|
||||||
self.editor = QComboBox(parent)
|
self.editor = QComboBox(parent)
|
||||||
|
|
||||||
val = []
|
val = []
|
||||||
if self._mode == "rk":
|
if self._mode == "rk":
|
||||||
reach_id = self._ic_spec_lst[index.row()].reach
|
reach_id = self._dif_spec_lst[index.row()].reach
|
||||||
|
|
||||||
reach = next(filter(
|
reach = next(
|
||||||
|
filter(
|
||||||
lambda edge: edge.id == reach_id, self._data.edges()
|
lambda edge: edge.id == reach_id, self._data.edges()
|
||||||
))
|
),
|
||||||
|
None
|
||||||
|
)
|
||||||
|
|
||||||
if reach_id is not None:
|
if reach_id is not None:
|
||||||
val = list(
|
val = list(
|
||||||
@@ -118,9 +121,16 @@ class DIFTableModel(PamhyrTableModel):
|
|||||||
super(DIFTableModel, self).__init__(data=data, **kwargs)
|
super(DIFTableModel, self).__init__(data=data, **kwargs)
|
||||||
|
|
||||||
self._data = data
|
self._data = data
|
||||||
|
if self._undo is not None:
|
||||||
|
self._undo.indexChanged.connect(lambda _: self.update())
|
||||||
|
|
||||||
def _setup_lst(self):
|
def _setup_lst(self):
|
||||||
self._lst = self._data._data
|
self._lst = list(
|
||||||
|
filter(
|
||||||
|
lambda dif: dif._deleted is False,
|
||||||
|
self._data._data
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
def rowCount(self, parent):
|
def rowCount(self, parent):
|
||||||
return len(self._lst)
|
return len(self._lst)
|
||||||
@@ -176,6 +186,9 @@ class DIFTableModel(PamhyrTableModel):
|
|||||||
if not index.isValid() or role != Qt.EditRole:
|
if not index.isValid() or role != Qt.EditRole:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
if self.is_same_data(index, value):
|
||||||
|
return False
|
||||||
|
|
||||||
row = index.row()
|
row = index.row()
|
||||||
column = index.column()
|
column = index.column()
|
||||||
|
|
||||||
@@ -183,7 +196,10 @@ class DIFTableModel(PamhyrTableModel):
|
|||||||
if self._headers[column] != "reach":
|
if self._headers[column] != "reach":
|
||||||
self._undo.push(
|
self._undo.push(
|
||||||
SetCommandSpec(
|
SetCommandSpec(
|
||||||
self._lst, row, self._headers[column], value
|
self._lst,
|
||||||
|
row,
|
||||||
|
self._headers[column],
|
||||||
|
value
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
elif self._headers[column] == "reach":
|
elif self._headers[column] == "reach":
|
||||||
@@ -211,25 +227,34 @@ class DIFTableModel(PamhyrTableModel):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
self._setup_lst()
|
||||||
self.endInsertRows()
|
self.endInsertRows()
|
||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
|
|
||||||
def delete(self, rows, parent=QModelIndex()):
|
def delete(self, rows, parent=QModelIndex()):
|
||||||
self.beginRemoveRows(parent, rows[0], rows[-1])
|
self.beginRemoveRows(parent, rows[0], rows[-1])
|
||||||
|
|
||||||
|
data_rows = {
|
||||||
|
id(dif): i for i, dif in enumerate(self._data._data)
|
||||||
|
}
|
||||||
self._undo.push(
|
self._undo.push(
|
||||||
DelCommand(
|
DelCommand(
|
||||||
self._data, self._lst, rows
|
self._data,
|
||||||
|
self._data._data,
|
||||||
|
[data_rows[id(self._lst[row])] for row in rows
|
||||||
|
if 0 <= row < len(self._lst)]
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
self._setup_lst()
|
||||||
self.endRemoveRows()
|
self.endRemoveRows()
|
||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
|
|
||||||
def undo(self):
|
def undo(self):
|
||||||
self._undo.undo()
|
self._undo.undo()
|
||||||
|
self._setup_lst()
|
||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
|
|
||||||
def redo(self):
|
def redo(self):
|
||||||
self._undo.redo()
|
self._undo.redo()
|
||||||
|
self._setup_lst()
|
||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
@@ -136,11 +136,11 @@ class SetCommandSpec(QUndoCommand):
|
|||||||
|
|
||||||
|
|
||||||
class AddCommand(QUndoCommand):
|
class AddCommand(QUndoCommand):
|
||||||
def __init__(self, data, ics_spec, index):
|
def __init__(self, data, dif_spec, index):
|
||||||
QUndoCommand.__init__(self)
|
QUndoCommand.__init__(self)
|
||||||
|
|
||||||
self._data = data
|
self._data = data
|
||||||
self._ics_spec = ics_spec
|
self._dif_spec = dif_spec
|
||||||
self._index = index
|
self._index = index
|
||||||
self._new = None
|
self._new = None
|
||||||
|
|
||||||
@@ -155,21 +155,20 @@ class AddCommand(QUndoCommand):
|
|||||||
|
|
||||||
|
|
||||||
class DelCommand(QUndoCommand):
|
class DelCommand(QUndoCommand):
|
||||||
def __init__(self, data, ics_spec, rows):
|
def __init__(self, data, dif_spec, rows):
|
||||||
QUndoCommand.__init__(self)
|
QUndoCommand.__init__(self)
|
||||||
|
|
||||||
self._data = data
|
self._data = data
|
||||||
self._ics_spec = ics_spec
|
self._dif_spec = dif_spec
|
||||||
self._rows = rows
|
self._rows = rows
|
||||||
# self._data = data
|
|
||||||
|
|
||||||
self._ic = []
|
self._dif = []
|
||||||
for row in rows:
|
for row in rows:
|
||||||
self._ic.append((row, self._ics_spec[row]))
|
self._dif.append((row, self._dif_spec[row]))
|
||||||
self._ic.sort()
|
self._dif.sort()
|
||||||
|
|
||||||
def undo(self):
|
def undo(self):
|
||||||
for row, el in self._ic:
|
for row, el in self._dif:
|
||||||
self._data.insert(row, el)
|
self._data.insert(row, el)
|
||||||
|
|
||||||
def redo(self):
|
def redo(self):
|
||||||
|
|||||||
@@ -51,6 +51,8 @@ from View.DIFAdisTS.Table import (
|
|||||||
DIFTableModel, ComboBoxDelegate,
|
DIFTableModel, ComboBoxDelegate,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
from View.DIFAdisTS.UndoCommand import SetCommand
|
||||||
|
|
||||||
from View.DIFAdisTS.translate import DIFAdisTSTranslate
|
from View.DIFAdisTS.translate import DIFAdisTSTranslate
|
||||||
|
|
||||||
from Solver.Mage import Mage8
|
from Solver.Mage import Mage8
|
||||||
@@ -89,18 +91,21 @@ class DIFAdisTSWindow(PamhyrWindow):
|
|||||||
|
|
||||||
self.setup_table()
|
self.setup_table()
|
||||||
|
|
||||||
self.ui.setWindowTitle(self._title)
|
|
||||||
|
|
||||||
def setup_table(self):
|
def setup_table(self):
|
||||||
|
|
||||||
path_icons = os.path.join(self._get_ui_directory(), f"ressources")
|
path_icons = os.path.join(self._get_ui_directory(), f"ressources")
|
||||||
|
|
||||||
table_default = self.find(QTableView, f"tableView")
|
table_default = self.find(QTableView, f"tableView")
|
||||||
|
|
||||||
|
if self._study.is_editable():
|
||||||
|
editable_headers = self._trad.get_dict("table_headers")
|
||||||
|
else:
|
||||||
|
editable_headers = []
|
||||||
|
|
||||||
self._delegate_method = ComboBoxDelegate(
|
self._delegate_method = ComboBoxDelegate(
|
||||||
trad=self._trad,
|
trad=self._trad,
|
||||||
data=self._study.river,
|
data=self._study.river,
|
||||||
ic_spec_lst=self._data[0]._data,
|
dif_spec_lst=self._data[0]._data,
|
||||||
parent=self,
|
parent=self,
|
||||||
mode="method"
|
mode="method"
|
||||||
)
|
)
|
||||||
@@ -108,7 +113,7 @@ class DIFAdisTSWindow(PamhyrWindow):
|
|||||||
self._table = DIFTableDefaultModel(
|
self._table = DIFTableDefaultModel(
|
||||||
table_view=table_default,
|
table_view=table_default,
|
||||||
table_headers=self._trad.get_dict("table_headers"),
|
table_headers=self._trad.get_dict("table_headers"),
|
||||||
editable_headers=["method", "dif", "b", "c"],
|
editable_headers=editable_headers,
|
||||||
delegates={
|
delegates={
|
||||||
"method": self._delegate_method
|
"method": self._delegate_method
|
||||||
},
|
},
|
||||||
@@ -130,9 +135,13 @@ class DIFAdisTSWindow(PamhyrWindow):
|
|||||||
|
|
||||||
action_add = QAction(self)
|
action_add = QAction(self)
|
||||||
action_add.setIcon(QIcon(os.path.join(path_icons, f"add.png")))
|
action_add.setIcon(QIcon(os.path.join(path_icons, f"add.png")))
|
||||||
|
|
||||||
|
if self._study.is_editable():
|
||||||
action_add.triggered.connect(self.add)
|
action_add.triggered.connect(self.add)
|
||||||
action_delete = QAction(self)
|
action_delete = QAction(self)
|
||||||
action_delete.setIcon(QIcon(os.path.join(path_icons, f"del.png")))
|
action_delete.setIcon(QIcon(os.path.join(path_icons, f"del.png")))
|
||||||
|
|
||||||
|
if self._study.is_editable():
|
||||||
action_delete.triggered.connect(self.delete)
|
action_delete.triggered.connect(self.delete)
|
||||||
|
|
||||||
toolBar.addAction(action_add)
|
toolBar.addAction(action_add)
|
||||||
@@ -144,23 +153,27 @@ class DIFAdisTSWindow(PamhyrWindow):
|
|||||||
self._delegate_reach = ComboBoxDelegate(
|
self._delegate_reach = ComboBoxDelegate(
|
||||||
trad=self._trad,
|
trad=self._trad,
|
||||||
data=self._study.river,
|
data=self._study.river,
|
||||||
ic_spec_lst=self._data[0]._data,
|
dif_spec_lst=self._data[0]._data,
|
||||||
parent=self,
|
parent=self,
|
||||||
mode="reaches"
|
mode="reaches"
|
||||||
)
|
)
|
||||||
self._delegate_rk = ComboBoxDelegate(
|
self._delegate_rk = ComboBoxDelegate(
|
||||||
trad=self._trad,
|
trad=self._trad,
|
||||||
data=self._study.river,
|
data=self._study.river,
|
||||||
ic_spec_lst=self._data[0]._data,
|
dif_spec_lst=self._data[0]._data,
|
||||||
parent=self,
|
parent=self,
|
||||||
mode="rk"
|
mode="rk"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
if self._study.is_editable():
|
||||||
|
editable_headers_spec = self._trad.get_dict("table_headers_spec")
|
||||||
|
else:
|
||||||
|
editable_headers_spec = []
|
||||||
|
|
||||||
self._table_spec = DIFTableModel(
|
self._table_spec = DIFTableModel(
|
||||||
table_view=self.table_spec,
|
table_view=self.table_spec,
|
||||||
table_headers=self._trad.get_dict("table_headers_spec"),
|
table_headers=self._trad.get_dict("table_headers_spec"),
|
||||||
editable_headers=["method", "reach", "start_rk",
|
editable_headers=editable_headers_spec,
|
||||||
"end_rk", "dif", "b", "c"],
|
|
||||||
delegates={
|
delegates={
|
||||||
"method": self._delegate_method,
|
"method": self._delegate_method,
|
||||||
"reach": self._delegate_reach,
|
"reach": self._delegate_reach,
|
||||||
@@ -272,11 +285,25 @@ class DIFAdisTSWindow(PamhyrWindow):
|
|||||||
self._update()
|
self._update()
|
||||||
|
|
||||||
def _undo(self):
|
def _undo(self):
|
||||||
|
undo_stack = self._undo_stack
|
||||||
|
if undo_stack is None or not undo_stack.canUndo():
|
||||||
|
return
|
||||||
|
|
||||||
|
if isinstance(undo_stack.command(undo_stack.index() - 1), SetCommand):
|
||||||
self._table.undo()
|
self._table.undo()
|
||||||
|
else:
|
||||||
|
self._table_spec.undo()
|
||||||
self._update()
|
self._update()
|
||||||
|
|
||||||
def _redo(self):
|
def _redo(self):
|
||||||
|
undo_stack = self._undo_stack
|
||||||
|
if undo_stack is None or not undo_stack.canRedo():
|
||||||
|
return
|
||||||
|
|
||||||
|
if isinstance(undo_stack.command(undo_stack.index()), SetCommand):
|
||||||
self._table.redo()
|
self._table.redo()
|
||||||
|
else:
|
||||||
|
self._table_spec.redo()
|
||||||
self._update()
|
self._update()
|
||||||
|
|
||||||
def add(self):
|
def add(self):
|
||||||
|
|||||||
@@ -17,6 +17,7 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
|
import traceback
|
||||||
|
|
||||||
from tools import trace, timer
|
from tools import trace, timer
|
||||||
|
|
||||||
@@ -92,7 +93,8 @@ class ReplWindow(PamhyrWindow):
|
|||||||
value = exec(rich_code)
|
value = exec(rich_code)
|
||||||
value = self.__debug_exec_result__
|
value = self.__debug_exec_result__
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
value = f"<font color=\"red\">" + str(e) + "</font>"
|
value = f"<font color=\"red\">" + str(e) + "</font>\n"
|
||||||
|
value += f"<font color=\"grey\">{traceback.format_exc()}</font>"
|
||||||
|
|
||||||
# Display code
|
# Display code
|
||||||
msg = f"<font color=\"grey\"> # " + code + " #</font>"
|
msg = f"<font color=\"grey\"> # " + code + " #</font>"
|
||||||
|
|||||||
@@ -43,7 +43,7 @@ class PlotRKZ(PlotRKC):
|
|||||||
self._isometric_axis = False
|
self._isometric_axis = False
|
||||||
|
|
||||||
self._auto_relim_update = True
|
self._auto_relim_update = True
|
||||||
self._autoscale_update = True
|
self._autoscale_update = False
|
||||||
self.parent = parent
|
self.parent = parent
|
||||||
|
|
||||||
def onpick(self, event):
|
def onpick(self, event):
|
||||||
|
|||||||
@@ -50,7 +50,7 @@ class PlotStricklers(PamhyrPlot):
|
|||||||
|
|
||||||
self._auto_relim = False
|
self._auto_relim = False
|
||||||
self._auto_relim_update = False
|
self._auto_relim_update = False
|
||||||
self._autoscale_update = True
|
self._autoscale_update = False
|
||||||
|
|
||||||
@timer
|
@timer
|
||||||
def draw(self, highlight=None):
|
def draw(self, highlight=None):
|
||||||
|
|||||||
@@ -134,6 +134,9 @@ class FrictionTableModel(PamhyrTableModel):
|
|||||||
if not index.isValid() or role != Qt.EditRole:
|
if not index.isValid() or role != Qt.EditRole:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
if self.is_same_data(index, value):
|
||||||
|
return False
|
||||||
|
|
||||||
row = index.row()
|
row = index.row()
|
||||||
column = index.column()
|
column = index.column()
|
||||||
|
|
||||||
@@ -274,15 +277,14 @@ class FrictionTableModel(PamhyrTableModel):
|
|||||||
with open(file_name, encoding="utf-8") as rug_file:
|
with open(file_name, encoding="utf-8") as rug_file:
|
||||||
for line in rug_file:
|
for line in rug_file:
|
||||||
if line.upper().startswith("K"):
|
if line.upper().startswith("K"):
|
||||||
line = line.split()
|
if int(line[1:4]) == reach_id:
|
||||||
if int(line[1]) == reach_id:
|
data.append(line[4:].split())
|
||||||
data.append(line[1:])
|
|
||||||
|
|
||||||
new_data = []
|
new_data = []
|
||||||
for d in data:
|
for d in data:
|
||||||
new = None
|
new = None
|
||||||
minor = float(d[3])
|
minor = float(d[2])
|
||||||
medium = float(d[4])
|
medium = float(d[3])
|
||||||
for s in self._study.river.stricklers.stricklers:
|
for s in self._study.river.stricklers.stricklers:
|
||||||
if s.minor == minor and s.medium == medium:
|
if s.minor == minor and s.medium == medium:
|
||||||
new = s
|
new = s
|
||||||
@@ -292,5 +294,5 @@ class FrictionTableModel(PamhyrTableModel):
|
|||||||
self._study.river.stricklers))
|
self._study.river.stricklers))
|
||||||
new.minor = minor
|
new.minor = minor
|
||||||
new.medium = medium
|
new.medium = medium
|
||||||
new_data.append([self._data, float(d[1]), float(d[2]), new, new])
|
new_data.append([self._data, float(d[0]), float(d[1]), new, new])
|
||||||
self.replace_data(new_data)
|
self.replace_data(new_data)
|
||||||
@@ -0,0 +1,296 @@
|
|||||||
|
# Window.py -- Pamhyr
|
||||||
|
# Copyright (C) 2024-2025 INRAE
|
||||||
|
#
|
||||||
|
# This program is free software: you can redistribute it and/or modify
|
||||||
|
# it under the terms of the GNU General Public License as published by
|
||||||
|
# the Free Software Foundation, either version 3 of the License, or
|
||||||
|
# (at your option) any later version.
|
||||||
|
#
|
||||||
|
# This program is distributed in the hope that it will be useful,
|
||||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
# GNU General Public License for more details.
|
||||||
|
#
|
||||||
|
# You should have received a copy of the GNU General Public License
|
||||||
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
import logging
|
||||||
|
|
||||||
|
from Modules import Modules
|
||||||
|
from View.Tools.PamhyrWindow import PamhyrWindow
|
||||||
|
|
||||||
|
from PyQt5.QtWidgets import (
|
||||||
|
QLabel, QPlainTextEdit, QPushButton, QCheckBox,
|
||||||
|
QFileDialog, QVBoxLayout, QDoubleSpinBox,
|
||||||
|
)
|
||||||
|
|
||||||
|
from PyQt5.QtCore import (
|
||||||
|
QSettings
|
||||||
|
)
|
||||||
|
|
||||||
|
from View.GeoTIFF.Translate import GeoTIFFTranslate
|
||||||
|
from View.GeoTIFF.UndoCommand import (
|
||||||
|
SetCommand
|
||||||
|
)
|
||||||
|
|
||||||
|
from View.Tools.Plot.PamhyrToolbar import PamhyrPlotToolbar
|
||||||
|
from View.Tools.Plot.PamhyrCanvas import MplCanvas
|
||||||
|
from View.PlotXY import PlotXY
|
||||||
|
|
||||||
|
try:
|
||||||
|
import rasterio
|
||||||
|
import rasterio.control
|
||||||
|
import rasterio.crs
|
||||||
|
import rasterio.sample
|
||||||
|
import rasterio.vrt
|
||||||
|
import rasterio._features
|
||||||
|
|
||||||
|
from rasterio.io import MemoryFile
|
||||||
|
_rasterio_loaded = True
|
||||||
|
except Exception as e:
|
||||||
|
print(f"Module 'rasterio' is not available: {e}")
|
||||||
|
_rasterio_loaded = False
|
||||||
|
|
||||||
|
logger = logging.getLogger()
|
||||||
|
|
||||||
|
|
||||||
|
class EditGeoTIFFWindow(PamhyrWindow):
|
||||||
|
_pamhyr_ui = "GeoTIFF"
|
||||||
|
_pamhyr_name = "Edit GeoTIFF"
|
||||||
|
|
||||||
|
def __init__(self, study=None, config=None, geotiff=None,
|
||||||
|
trad=None, undo=None, parent=None):
|
||||||
|
|
||||||
|
super(EditGeoTIFFWindow, self).__init__(
|
||||||
|
title=self._pamhyr_name,
|
||||||
|
study=study,
|
||||||
|
config=config,
|
||||||
|
trad=trad,
|
||||||
|
options=[],
|
||||||
|
parent=parent
|
||||||
|
)
|
||||||
|
|
||||||
|
self._geotiff = geotiff
|
||||||
|
self._file_name = geotiff.file_name
|
||||||
|
|
||||||
|
self._hash_data.append(self._geotiff)
|
||||||
|
|
||||||
|
self._undo = undo
|
||||||
|
|
||||||
|
self.setup_values()
|
||||||
|
self.setup_graph()
|
||||||
|
self.setup_connection()
|
||||||
|
|
||||||
|
def setup_graph(self):
|
||||||
|
self.canvas = MplCanvas(width=5, height=4, dpi=100)
|
||||||
|
self.canvas.setObjectName("canvas")
|
||||||
|
self.plot_layout = self.find(QVBoxLayout,
|
||||||
|
"verticalLayout_geotiff")
|
||||||
|
self._toolbar = PamhyrPlotToolbar(
|
||||||
|
self.canvas, self,
|
||||||
|
items=["home", "zoom", "save", "iso", "back/forward", "move"]
|
||||||
|
)
|
||||||
|
self.plot_layout.addWidget(self._toolbar)
|
||||||
|
self.plot_layout.addWidget(self.canvas)
|
||||||
|
|
||||||
|
self.plot = PlotXY(
|
||||||
|
canvas=self.canvas,
|
||||||
|
data=self._study.river.enable_edges(),
|
||||||
|
trad=self._trad,
|
||||||
|
toolbar=None,
|
||||||
|
parent=self
|
||||||
|
)
|
||||||
|
self.plot.update()
|
||||||
|
|
||||||
|
self._plot_img = None
|
||||||
|
|
||||||
|
memfile = self._geotiff.memfile
|
||||||
|
if memfile is not None:
|
||||||
|
self.draw_geotiff(memfile=memfile)
|
||||||
|
|
||||||
|
def setup_values(self):
|
||||||
|
self.set_check_box("checkBox", self._geotiff.enabled)
|
||||||
|
self.set_line_edit_text("lineEdit_name", self._geotiff.name)
|
||||||
|
self.set_line_edit_text("lineEdit_description",
|
||||||
|
self._geotiff.description)
|
||||||
|
|
||||||
|
bounds = list(self._geotiff.coordinates.values())
|
||||||
|
self._set_values_from_bounds(bounds)
|
||||||
|
self._set_default_values_from_bounds(bounds)
|
||||||
|
self._reset_spinboxes()
|
||||||
|
|
||||||
|
if self._study.is_read_only():
|
||||||
|
self.set_check_box_enable("checkBox", False)
|
||||||
|
self.set_line_edit_enable("lineEdit_name", False)
|
||||||
|
self.set_line_edit_enable("lineEdit_description", False)
|
||||||
|
|
||||||
|
for button in ["import", "bottom", "top", "left", "right"]:
|
||||||
|
self.find(QPushButton, f"pushButton_{button}")\
|
||||||
|
.setEnabled(False)
|
||||||
|
|
||||||
|
for spin in ["bottom", "top", "left", "right"]:
|
||||||
|
self.find(QDoubleSpinBox, f"doubleSpinBox_{spin}")\
|
||||||
|
.setEnabled(False)
|
||||||
|
|
||||||
|
def _set_values_from_bounds(self, bounds):
|
||||||
|
self._values = {
|
||||||
|
"bottom": bounds[0],
|
||||||
|
"top": bounds[1],
|
||||||
|
"left": bounds[2],
|
||||||
|
"right": bounds[3],
|
||||||
|
}
|
||||||
|
|
||||||
|
def _set_default_values_from_bounds(self, bounds):
|
||||||
|
self._values_default = {
|
||||||
|
"bottom": bounds[0],
|
||||||
|
"top": bounds[1],
|
||||||
|
"left": bounds[2],
|
||||||
|
"right": bounds[3],
|
||||||
|
}
|
||||||
|
|
||||||
|
def _reset_spinboxes(self):
|
||||||
|
for key in self._values:
|
||||||
|
self._reset_spinbox(key)
|
||||||
|
|
||||||
|
def _reset_spinbox(self, key):
|
||||||
|
self.set_double_spin_box(
|
||||||
|
f"doubleSpinBox_{key}", self._values_default[key]
|
||||||
|
)
|
||||||
|
|
||||||
|
def setup_connection(self):
|
||||||
|
self.find(QPushButton, "pushButton_cancel")\
|
||||||
|
.clicked.connect(self.close)
|
||||||
|
self.find(QPushButton, "pushButton_ok")\
|
||||||
|
.clicked.connect(self.accept)
|
||||||
|
self.find(QPushButton, "pushButton_import")\
|
||||||
|
.clicked.connect(self._import)
|
||||||
|
|
||||||
|
self.find(QPushButton, "pushButton_bottom")\
|
||||||
|
.clicked.connect(lambda: self._reset_spinbox("bottom"))
|
||||||
|
self.find(QPushButton, "pushButton_top")\
|
||||||
|
.clicked.connect(lambda: self._reset_spinbox("top"))
|
||||||
|
self.find(QPushButton, f"pushButton_left")\
|
||||||
|
.clicked.connect(lambda: self._reset_spinbox("left"))
|
||||||
|
self.find(QPushButton, f"pushButton_right")\
|
||||||
|
.clicked.connect(lambda: self._reset_spinbox("right"))
|
||||||
|
|
||||||
|
self.find(QDoubleSpinBox, f"doubleSpinBox_bottom")\
|
||||||
|
.valueChanged.connect(
|
||||||
|
lambda: self.update_values_from_spinbox("bottom")
|
||||||
|
)
|
||||||
|
self.find(QDoubleSpinBox, f"doubleSpinBox_top")\
|
||||||
|
.valueChanged.connect(
|
||||||
|
lambda: self.update_values_from_spinbox("top")
|
||||||
|
)
|
||||||
|
self.find(QDoubleSpinBox, f"doubleSpinBox_left")\
|
||||||
|
.valueChanged.connect(
|
||||||
|
lambda: self.update_values_from_spinbox("left")
|
||||||
|
)
|
||||||
|
self.find(QDoubleSpinBox, f"doubleSpinBox_right")\
|
||||||
|
.valueChanged.connect(
|
||||||
|
lambda: self.update_values_from_spinbox("right")
|
||||||
|
)
|
||||||
|
|
||||||
|
def update_values_from_spinbox(self, key):
|
||||||
|
self._values[key] = self.get_double_spin_box(f"doubleSpinBox_{key}")
|
||||||
|
|
||||||
|
left = self._values["left"]
|
||||||
|
right = self._values["right"]
|
||||||
|
bottom = self._values["bottom"]
|
||||||
|
top = self._values["top"]
|
||||||
|
|
||||||
|
self._plot_img.set_extent((left, right, bottom, top))
|
||||||
|
self.plot.idle()
|
||||||
|
|
||||||
|
def draw_geotiff(self, memfile=None):
|
||||||
|
if not _rasterio_loaded:
|
||||||
|
return
|
||||||
|
|
||||||
|
if memfile is None:
|
||||||
|
if self._file_name == "":
|
||||||
|
return
|
||||||
|
|
||||||
|
with rasterio.open(self._file_name) as data:
|
||||||
|
img = data.read()
|
||||||
|
b = data.bounds[:] # left, bottom, right, top
|
||||||
|
|
||||||
|
if b[2] > b[0] and b[1] < b[3]:
|
||||||
|
coord = [b[1], b[3], b[0], b[2]]
|
||||||
|
else:
|
||||||
|
xlim = self.canvas.axes.get_xlim()
|
||||||
|
ylim = self.canvas.axes.get_ylim()
|
||||||
|
coord = ylim + xlim
|
||||||
|
|
||||||
|
self._set_values_from_bounds(coord)
|
||||||
|
self._set_default_values_from_bounds(coord)
|
||||||
|
else:
|
||||||
|
with memfile.open() as gt:
|
||||||
|
img = gt.read()
|
||||||
|
|
||||||
|
if self._plot_img is not None:
|
||||||
|
self._plot_img.remove()
|
||||||
|
|
||||||
|
left = self._values["left"]
|
||||||
|
right = self._values["right"]
|
||||||
|
bottom = self._values["bottom"]
|
||||||
|
top = self._values["top"]
|
||||||
|
|
||||||
|
self._plot_img = self.canvas.axes.imshow(
|
||||||
|
img.transpose((1, 2, 0)),
|
||||||
|
extent=(left, right, bottom, top)
|
||||||
|
)
|
||||||
|
|
||||||
|
self.plot.idle()
|
||||||
|
self._reset_spinboxes()
|
||||||
|
|
||||||
|
def _import(self):
|
||||||
|
options = QFileDialog.Options()
|
||||||
|
settings = QSettings(QSettings.IniFormat,
|
||||||
|
QSettings.UserScope, 'MyOrg', )
|
||||||
|
options |= QFileDialog.DontUseNativeDialog
|
||||||
|
|
||||||
|
file_types = [
|
||||||
|
self._trad["file_geotiff"],
|
||||||
|
self._trad["file_all"],
|
||||||
|
]
|
||||||
|
|
||||||
|
filename, _ = QFileDialog.getOpenFileName(
|
||||||
|
self,
|
||||||
|
self._trad["open_file"],
|
||||||
|
"",
|
||||||
|
";; ".join(file_types),
|
||||||
|
options=options
|
||||||
|
)
|
||||||
|
|
||||||
|
if filename != "":
|
||||||
|
self._file_name = filename
|
||||||
|
self.draw_geotiff()
|
||||||
|
|
||||||
|
def accept(self):
|
||||||
|
if self._study.is_editable():
|
||||||
|
is_enabled = self.get_check_box("checkBox")
|
||||||
|
name = self.get_line_edit_text("lineEdit_name")
|
||||||
|
description = self.get_line_edit_text("lineEdit_description")
|
||||||
|
|
||||||
|
coord_bottom = self.get_double_spin_box("doubleSpinBox_bottom")
|
||||||
|
coord_top = self.get_double_spin_box("doubleSpinBox_top")
|
||||||
|
coord_left = self.get_double_spin_box("doubleSpinBox_left")
|
||||||
|
coord_right = self.get_double_spin_box("doubleSpinBox_right")
|
||||||
|
|
||||||
|
self._undo.push(
|
||||||
|
SetCommand(
|
||||||
|
self._geotiff, enabled=is_enabled,
|
||||||
|
name=name, description=description,
|
||||||
|
coordinates_bottom=coord_bottom,
|
||||||
|
coordinates_top=coord_top,
|
||||||
|
coordinates_left=coord_left,
|
||||||
|
coordinates_right=coord_right,
|
||||||
|
file_name=self._file_name,
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
self._propagate_update(key=Modules.GEOTIFF)
|
||||||
|
|
||||||
|
self.close()
|
||||||
@@ -0,0 +1,96 @@
|
|||||||
|
# List.py -- Pamhyr
|
||||||
|
# Copyright (C) 2024-2025 INRAE
|
||||||
|
#
|
||||||
|
# This program is free software: you can redistribute it and/or modify
|
||||||
|
# it under the terms of the GNU General Public License as published by
|
||||||
|
# the Free Software Foundation, either version 3 of the License, or
|
||||||
|
# (at your option) any later version.
|
||||||
|
#
|
||||||
|
# This program is distributed in the hope that it will be useful,
|
||||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
# GNU General Public License for more details.
|
||||||
|
#
|
||||||
|
# You should have received a copy of the GNU General Public License
|
||||||
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
import logging
|
||||||
|
|
||||||
|
from functools import reduce
|
||||||
|
from tools import trace, timer
|
||||||
|
|
||||||
|
from PyQt5.QtCore import (
|
||||||
|
Qt, QVariant,
|
||||||
|
)
|
||||||
|
|
||||||
|
from PyQt5.QtGui import (
|
||||||
|
QColor, QBrush,
|
||||||
|
)
|
||||||
|
|
||||||
|
from View.Tools.PamhyrList import PamhyrListModel
|
||||||
|
from View.GeoTIFF.UndoCommand import (
|
||||||
|
AddCommand, DelCommand
|
||||||
|
)
|
||||||
|
|
||||||
|
logger = logging.getLogger()
|
||||||
|
|
||||||
|
|
||||||
|
class ListModel(PamhyrListModel):
|
||||||
|
def get_true_data_row(self, row):
|
||||||
|
el = self._data.get(row)
|
||||||
|
|
||||||
|
return next(
|
||||||
|
map(
|
||||||
|
lambda e: e[0],
|
||||||
|
filter(
|
||||||
|
lambda e: e[1] == el,
|
||||||
|
enumerate(self._data._lst)
|
||||||
|
)
|
||||||
|
), 0
|
||||||
|
)
|
||||||
|
|
||||||
|
def data(self, index, role):
|
||||||
|
row = index.row()
|
||||||
|
column = index.column()
|
||||||
|
|
||||||
|
file = self._data.files[row]
|
||||||
|
|
||||||
|
if role == Qt.ForegroundRole:
|
||||||
|
color = Qt.gray
|
||||||
|
|
||||||
|
if file.is_enabled():
|
||||||
|
color = QColor("black")
|
||||||
|
else:
|
||||||
|
color = QColor("grey")
|
||||||
|
|
||||||
|
return QBrush(color)
|
||||||
|
|
||||||
|
if role == Qt.ItemDataRole.DisplayRole:
|
||||||
|
text = f"{file.name}: '{file.description}'"
|
||||||
|
|
||||||
|
if not file.is_enabled():
|
||||||
|
text += " (disabled)"
|
||||||
|
|
||||||
|
return text
|
||||||
|
|
||||||
|
return QVariant()
|
||||||
|
|
||||||
|
def add(self, row):
|
||||||
|
row = self.get_true_data_row(row)
|
||||||
|
|
||||||
|
self._undo.push(
|
||||||
|
AddCommand(
|
||||||
|
self._data, row
|
||||||
|
)
|
||||||
|
)
|
||||||
|
self.update()
|
||||||
|
|
||||||
|
def delete(self, row):
|
||||||
|
self._undo.push(
|
||||||
|
DelCommand(
|
||||||
|
self._data, self._data.files[row]
|
||||||
|
)
|
||||||
|
)
|
||||||
|
self.update()
|
||||||
@@ -0,0 +1,35 @@
|
|||||||
|
# Translate.py -- Pamhyr
|
||||||
|
# Copyright (C) 2024-2025 INRAE
|
||||||
|
#
|
||||||
|
# This program is free software: you can redistribute it and/or modify
|
||||||
|
# it under the terms of the GNU General Public License as published by
|
||||||
|
# the Free Software Foundation, either version 3 of the License, or
|
||||||
|
# (at your option) any later version.
|
||||||
|
#
|
||||||
|
# This program is distributed in the hope that it will be useful,
|
||||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
# GNU General Public License for more details.
|
||||||
|
#
|
||||||
|
# You should have received a copy of the GNU General Public License
|
||||||
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
from PyQt5.QtCore import QCoreApplication
|
||||||
|
from View.Translate import MainTranslate
|
||||||
|
|
||||||
|
_translate = QCoreApplication.translate
|
||||||
|
|
||||||
|
|
||||||
|
class GeoTIFFTranslate(MainTranslate):
|
||||||
|
def __init__(self):
|
||||||
|
super(GeoTIFFTranslate, self).__init__()
|
||||||
|
|
||||||
|
self._dict["GeoTIFF files"] = _translate(
|
||||||
|
"GeoTIFF", "GeoTIFF files"
|
||||||
|
)
|
||||||
|
|
||||||
|
self._dict["Edit additional file"] = _translate(
|
||||||
|
"GeoTIFF", "Edit GeoTIFF file"
|
||||||
|
)
|
||||||
@@ -0,0 +1,81 @@
|
|||||||
|
# UndoCommand.py -- Pamhyr
|
||||||
|
# Copyright (C) 2024-2025 INRAE
|
||||||
|
#
|
||||||
|
# This program is free software: you can redistribute it and/or modify
|
||||||
|
# it under the terms of the GNU General Public License as published by
|
||||||
|
# the Free Software Foundation, either version 3 of the License, or
|
||||||
|
# (at your option) any later version.
|
||||||
|
#
|
||||||
|
# This program is distributed in the hope that it will be useful,
|
||||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
# GNU General Public License for more details.
|
||||||
|
#
|
||||||
|
# You should have received a copy of the GNU General Public License
|
||||||
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
from tools import trace, timer
|
||||||
|
|
||||||
|
from PyQt5.QtWidgets import (
|
||||||
|
QMessageBox, QUndoCommand, QUndoStack,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
class SetCommand(QUndoCommand):
|
||||||
|
def __init__(self, geotiff, **kwargs):
|
||||||
|
QUndoCommand.__init__(self)
|
||||||
|
|
||||||
|
self._geotiff = geotiff
|
||||||
|
self._new = kwargs
|
||||||
|
self._old = None
|
||||||
|
|
||||||
|
def undo(self):
|
||||||
|
f = self._geotiff
|
||||||
|
|
||||||
|
for key in self._old:
|
||||||
|
f[key] = self._old[key]
|
||||||
|
|
||||||
|
def redo(self):
|
||||||
|
f = self._geotiff
|
||||||
|
|
||||||
|
if self._old is None:
|
||||||
|
self._old = {}
|
||||||
|
for key in self._new:
|
||||||
|
self._old[key] = f[key]
|
||||||
|
|
||||||
|
for key in self._new:
|
||||||
|
f[key] = self._new[key]
|
||||||
|
|
||||||
|
|
||||||
|
class AddCommand(QUndoCommand):
|
||||||
|
def __init__(self, files, row):
|
||||||
|
QUndoCommand.__init__(self)
|
||||||
|
|
||||||
|
self._files = files
|
||||||
|
self._row = row
|
||||||
|
self._new = None
|
||||||
|
|
||||||
|
def undo(self):
|
||||||
|
self._new.set_as_deleted()
|
||||||
|
|
||||||
|
def redo(self):
|
||||||
|
if self._new is None:
|
||||||
|
self._new = self._files.new(self._row)
|
||||||
|
else:
|
||||||
|
self._new.set_as_not_deleted()
|
||||||
|
|
||||||
|
|
||||||
|
class DelCommand(QUndoCommand):
|
||||||
|
def __init__(self, files, line):
|
||||||
|
QUndoCommand.__init__(self)
|
||||||
|
|
||||||
|
self._files = files
|
||||||
|
self._line = line
|
||||||
|
|
||||||
|
def undo(self):
|
||||||
|
self._line.set_as_not_deleted()
|
||||||
|
|
||||||
|
def redo(self):
|
||||||
|
self._line.set_as_deleted()
|
||||||
@@ -0,0 +1,223 @@
|
|||||||
|
# Window.py -- Pamhyr
|
||||||
|
# Copyright (C) 2024-2025 INRAE
|
||||||
|
#
|
||||||
|
# This program is free software: you can redistribute it and/or modify
|
||||||
|
# it under the terms of the GNU General Public License as published by
|
||||||
|
# the Free Software Foundation, either version 3 of the License, or
|
||||||
|
# (at your option) any later version.
|
||||||
|
#
|
||||||
|
# This program is distributed in the hope that it will be useful,
|
||||||
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||||
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||||
|
# GNU General Public License for more details.
|
||||||
|
#
|
||||||
|
# You should have received a copy of the GNU General Public License
|
||||||
|
# along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||||||
|
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
from tools import trace, timer, logger_exception
|
||||||
|
|
||||||
|
from PyQt5.QtWidgets import (
|
||||||
|
QAction, QListView, QVBoxLayout,
|
||||||
|
)
|
||||||
|
|
||||||
|
from matplotlib.patches import Rectangle
|
||||||
|
|
||||||
|
from Modules import Modules
|
||||||
|
|
||||||
|
from View.Tools.PamhyrWindow import PamhyrWindow
|
||||||
|
|
||||||
|
from View.GeoTIFF.List import ListModel
|
||||||
|
from View.GeoTIFF.Translate import GeoTIFFTranslate
|
||||||
|
from View.GeoTIFF.Edit.Window import EditGeoTIFFWindow
|
||||||
|
|
||||||
|
from View.Tools.Plot.PamhyrToolbar import PamhyrPlotToolbar
|
||||||
|
from View.Tools.Plot.PamhyrCanvas import MplCanvas
|
||||||
|
from View.PlotXY import PlotXY
|
||||||
|
|
||||||
|
try:
|
||||||
|
import rasterio
|
||||||
|
import rasterio.control
|
||||||
|
import rasterio.crs
|
||||||
|
import rasterio.sample
|
||||||
|
import rasterio.vrt
|
||||||
|
import rasterio._features
|
||||||
|
|
||||||
|
from rasterio.io import MemoryFile
|
||||||
|
_rasterio_loaded = True
|
||||||
|
except Exception as e:
|
||||||
|
print(f"Module 'rasterio' is not available: {e}")
|
||||||
|
_rasterio_loaded = False
|
||||||
|
|
||||||
|
|
||||||
|
class GeoTIFFListWindow(PamhyrWindow):
|
||||||
|
_pamhyr_ui = "GeoTIFFList"
|
||||||
|
_pamhyr_name = "GeoTIFF files"
|
||||||
|
|
||||||
|
def __init__(self, study=None, config=None,
|
||||||
|
parent=None):
|
||||||
|
trad = GeoTIFFTranslate()
|
||||||
|
name = trad[self._pamhyr_name] + " - " + study.name
|
||||||
|
|
||||||
|
super(GeoTIFFListWindow, self).__init__(
|
||||||
|
title=name,
|
||||||
|
study=study,
|
||||||
|
config=config,
|
||||||
|
trad=trad,
|
||||||
|
options=[],
|
||||||
|
parent=parent
|
||||||
|
)
|
||||||
|
|
||||||
|
self.setup_list()
|
||||||
|
self.setup_graph()
|
||||||
|
self.setup_connections()
|
||||||
|
|
||||||
|
def setup_list(self):
|
||||||
|
lst = self.find(QListView, "listView")
|
||||||
|
self._list = ListModel(
|
||||||
|
list_view=lst,
|
||||||
|
data=self._study.river.geotiff,
|
||||||
|
undo=self._undo_stack,
|
||||||
|
trad=self._trad,
|
||||||
|
)
|
||||||
|
|
||||||
|
def setup_graph(self):
|
||||||
|
self.canvas = MplCanvas(width=5, height=4, dpi=100)
|
||||||
|
self.canvas.setObjectName("canvas")
|
||||||
|
self._plot_layout = self.find(QVBoxLayout, "verticalLayout")
|
||||||
|
self._toolbar = PamhyrPlotToolbar(
|
||||||
|
self.canvas, self,
|
||||||
|
items=["home", "zoom", "save", "iso", "back/forward", "move"]
|
||||||
|
)
|
||||||
|
self._plot_layout.addWidget(self._toolbar)
|
||||||
|
self._plot_layout.addWidget(self.canvas)
|
||||||
|
|
||||||
|
self._plot = PlotXY(
|
||||||
|
canvas=self.canvas,
|
||||||
|
data=self._study.river.enable_edges(),
|
||||||
|
geotiff=self._study.river.geotiff,
|
||||||
|
trad=self._trad,
|
||||||
|
toolbar=None,
|
||||||
|
parent=self
|
||||||
|
)
|
||||||
|
self._plot_rect = []
|
||||||
|
|
||||||
|
self._plot.update()
|
||||||
|
|
||||||
|
def setup_connections(self):
|
||||||
|
if self._study.is_editable():
|
||||||
|
self.find(QAction, "action_add").triggered.connect(self.add)
|
||||||
|
self.find(QAction, "action_delete").triggered.connect(self.delete)
|
||||||
|
else:
|
||||||
|
self.find(QAction, "action_add").setEnabled(False)
|
||||||
|
self.find(QAction, "action_delete").setEnabled(False)
|
||||||
|
|
||||||
|
self.find(QAction, "action_edit").triggered.connect(self.edit)
|
||||||
|
|
||||||
|
self.find(QListView, "listView")\
|
||||||
|
.selectionModel()\
|
||||||
|
.selectionChanged\
|
||||||
|
.connect(self._update_rectangle)
|
||||||
|
|
||||||
|
def _propagated_update(self, key=Modules(0)):
|
||||||
|
if Modules.GEOMETRY not in key and Modules.GEOTIFF not in key:
|
||||||
|
return
|
||||||
|
|
||||||
|
self.update()
|
||||||
|
|
||||||
|
def update(self):
|
||||||
|
self._list.update()
|
||||||
|
self._plot.update()
|
||||||
|
|
||||||
|
self._update_rectangle()
|
||||||
|
|
||||||
|
def _update_rectangle(self):
|
||||||
|
for rect in self._plot_rect:
|
||||||
|
try:
|
||||||
|
rect.remove()
|
||||||
|
except Exception as e:
|
||||||
|
logger_exception(e)
|
||||||
|
|
||||||
|
self._plot_rect = []
|
||||||
|
|
||||||
|
rows = self.selected_rows()
|
||||||
|
if len(rows) <= 0:
|
||||||
|
return
|
||||||
|
|
||||||
|
for row in rows:
|
||||||
|
files = self._study.river.geotiff.files
|
||||||
|
if len(files) <= row:
|
||||||
|
continue
|
||||||
|
|
||||||
|
geotiff = files[row]
|
||||||
|
coord = geotiff.coordinates
|
||||||
|
|
||||||
|
xy = (coord["left"], coord["bottom"])
|
||||||
|
width = abs(coord["right"] - coord["left"])
|
||||||
|
height = abs(coord["top"] - coord["bottom"])
|
||||||
|
|
||||||
|
rect = Rectangle(
|
||||||
|
xy, width, height,
|
||||||
|
edgecolor='red', facecolor='none',
|
||||||
|
lw=2
|
||||||
|
)
|
||||||
|
|
||||||
|
self._plot_rect.append(rect)
|
||||||
|
|
||||||
|
self.canvas.axes.add_patch(
|
||||||
|
rect
|
||||||
|
)
|
||||||
|
|
||||||
|
self._plot.idle()
|
||||||
|
|
||||||
|
def selected_rows(self):
|
||||||
|
lst = self.find(QListView, f"listView")
|
||||||
|
return list(map(lambda i: i.row(), lst.selectedIndexes()))
|
||||||
|
|
||||||
|
def add(self):
|
||||||
|
rows = self.selected_rows()
|
||||||
|
if len(rows) > 0:
|
||||||
|
row = rows[0]
|
||||||
|
else:
|
||||||
|
row = 0
|
||||||
|
|
||||||
|
self._list.add(row)
|
||||||
|
|
||||||
|
def delete(self):
|
||||||
|
rows = self.selected_rows()
|
||||||
|
if len(rows) == 0:
|
||||||
|
return
|
||||||
|
|
||||||
|
self._list.delete(rows[0])
|
||||||
|
self.update()
|
||||||
|
|
||||||
|
def edit(self):
|
||||||
|
rows = self.selected_rows()
|
||||||
|
|
||||||
|
for row in rows:
|
||||||
|
geotiff = self._study.river.geotiff.files[row]
|
||||||
|
|
||||||
|
if self.sub_window_exists(
|
||||||
|
EditGeoTIFFWindow,
|
||||||
|
data=[self._study, self._config, geotiff]
|
||||||
|
):
|
||||||
|
continue
|
||||||
|
|
||||||
|
win = EditGeoTIFFWindow(
|
||||||
|
study=self._study,
|
||||||
|
config=self._config,
|
||||||
|
geotiff=geotiff,
|
||||||
|
trad=self._trad,
|
||||||
|
undo=self._undo_stack,
|
||||||
|
parent=self,
|
||||||
|
)
|
||||||
|
win.show()
|
||||||
|
|
||||||
|
def _undo(self):
|
||||||
|
self._undo_stack.undo()
|
||||||
|
self.update()
|
||||||
|
|
||||||
|
def _redo(self):
|
||||||
|
self._undo_stack.redo()
|
||||||
|
self.update()
|
||||||
@@ -38,7 +38,7 @@ class PlotAC(PamhyrPlot):
|
|||||||
self._isometric_axis = False
|
self._isometric_axis = False
|
||||||
|
|
||||||
self._auto_relim_update = True
|
self._auto_relim_update = True
|
||||||
self._autoscale_update = True
|
self._autoscale_update = False
|
||||||
|
|
||||||
self.label_x = self._trad["transverse_abscissa"]
|
self.label_x = self._trad["transverse_abscissa"]
|
||||||
self.label_y = self._trad["unit_elevation"]
|
self.label_y = self._trad["unit_elevation"]
|
||||||
@@ -143,7 +143,7 @@ class PlotAC(PamhyrPlot):
|
|||||||
]
|
]
|
||||||
else:
|
else:
|
||||||
color = self.color_incomplete_gl[
|
color = self.color_incomplete_gl[
|
||||||
lincomplete.index(txt)
|
lincomplete.index(txt) % len(self.color_incomplete_gl)
|
||||||
]
|
]
|
||||||
|
|
||||||
annotation = self.canvas.axes.annotate(
|
annotation = self.canvas.axes.annotate(
|
||||||
|
|||||||
@@ -114,6 +114,9 @@ class GeometryProfileTableModel(PamhyrTableModel):
|
|||||||
column = index.column()
|
column = index.column()
|
||||||
|
|
||||||
if role == Qt.EditRole:
|
if role == Qt.EditRole:
|
||||||
|
if self.is_same_data(index, value):
|
||||||
|
return False
|
||||||
|
|
||||||
try:
|
try:
|
||||||
if column == 0:
|
if column == 0:
|
||||||
self._undo.push(
|
self._undo.push(
|
||||||
|
|||||||
@@ -17,6 +17,10 @@
|
|||||||
# -*- coding: utf-8 -*-
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
from View.Tools.PamhyrWindow import PamhyrDialog
|
from View.Tools.PamhyrWindow import PamhyrDialog
|
||||||
|
from PyQt5.QtWidgets import QDoubleSpinBox
|
||||||
|
from View.Tools.FlexibleDoubleSpinBox import FlexibleDoubleSpinBox
|
||||||
|
from PyQt5.QtWidgets import QDoubleSpinBox
|
||||||
|
from View.Tools.FlexibleDoubleSpinBox import FlexibleDoubleSpinBox
|
||||||
|
|
||||||
|
|
||||||
class ShiftDialog(PamhyrDialog):
|
class ShiftDialog(PamhyrDialog):
|
||||||
@@ -31,8 +35,36 @@ class ShiftDialog(PamhyrDialog):
|
|||||||
parent=parent
|
parent=parent
|
||||||
)
|
)
|
||||||
|
|
||||||
|
self._replace_spinboxes()
|
||||||
self._init_default_values()
|
self._init_default_values()
|
||||||
|
|
||||||
|
def _replace_spinboxes(self):
|
||||||
|
for name in ["doubleSpinBox_X", "doubleSpinBox_Y", "doubleSpinBox_Z"]:
|
||||||
|
old = self.find(QDoubleSpinBox, name)
|
||||||
|
|
||||||
|
if old is None:
|
||||||
|
continue
|
||||||
|
|
||||||
|
new = FlexibleDoubleSpinBox(old.parent())
|
||||||
|
new.setObjectName(old.objectName())
|
||||||
|
|
||||||
|
# Copier propriétés utiles
|
||||||
|
new.setDecimals(old.decimals())
|
||||||
|
new.setMinimum(old.minimum())
|
||||||
|
new.setMaximum(old.maximum())
|
||||||
|
new.setSingleStep(old.singleStep())
|
||||||
|
new.setValue(old.value())
|
||||||
|
new.setGeometry(old.geometry())
|
||||||
|
|
||||||
|
# Remplacement dans layout si présent
|
||||||
|
layout = old.parent().layout()
|
||||||
|
if layout is not None:
|
||||||
|
layout.replaceWidget(old, new)
|
||||||
|
|
||||||
|
old.hide()
|
||||||
|
old.setParent(None)
|
||||||
|
old.deleteLater()
|
||||||
|
|
||||||
def _init_default_values(self):
|
def _init_default_values(self):
|
||||||
self._dx = 0.0
|
self._dx = 0.0
|
||||||
self._dy = 0.0
|
self._dy = 0.0
|
||||||
|
|||||||
@@ -95,6 +95,9 @@ class GeometryReachTableModel(PamhyrTableModel):
|
|||||||
column = index.column()
|
column = index.column()
|
||||||
|
|
||||||
if role == Qt.EditRole and index.column() != 2:
|
if role == Qt.EditRole and index.column() != 2:
|
||||||
|
if self.is_same_data(index, value):
|
||||||
|
return False
|
||||||
|
|
||||||
try:
|
try:
|
||||||
if index.column() == 0:
|
if index.column() == 0:
|
||||||
self._undo.push(
|
self._undo.push(
|
||||||
@@ -251,11 +254,22 @@ class GeometryReachTableModel(PamhyrTableModel):
|
|||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
|
|
||||||
def meshing(self, mesher, data, tableView):
|
def meshing(self, mesher, data, tableView):
|
||||||
|
new_profiles = mesher.meshing(
|
||||||
|
self._data,
|
||||||
|
**data
|
||||||
|
)
|
||||||
|
|
||||||
|
if new_profiles is None:
|
||||||
|
return
|
||||||
|
|
||||||
|
if len(new_profiles) == 0:
|
||||||
|
return
|
||||||
|
|
||||||
self.layoutAboutToBeChanged.emit()
|
self.layoutAboutToBeChanged.emit()
|
||||||
|
|
||||||
self._undo.push(
|
self._undo.push(
|
||||||
MeshingCommand(
|
MeshingCommand(
|
||||||
self._data, mesher, data, tableView
|
self._data, new_profiles, data, tableView
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -275,7 +289,6 @@ class GeometryReachTableModel(PamhyrTableModel):
|
|||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
|
|
||||||
def purge(self, np_purge):
|
def purge(self, np_purge):
|
||||||
|
|
||||||
self._undo.push(
|
self._undo.push(
|
||||||
PurgeCommand(
|
PurgeCommand(
|
||||||
self._data, np_purge
|
self._data, np_purge
|
||||||
@@ -284,7 +297,6 @@ class GeometryReachTableModel(PamhyrTableModel):
|
|||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
|
|
||||||
def shift(self, rows, dx, dy, dz):
|
def shift(self, rows, dx, dy, dz):
|
||||||
|
|
||||||
self._undo.push(
|
self._undo.push(
|
||||||
ShiftCommand(
|
ShiftCommand(
|
||||||
self._data, rows, dx, dy, dz
|
self._data, rows, dx, dy, dz
|
||||||
@@ -293,7 +305,6 @@ class GeometryReachTableModel(PamhyrTableModel):
|
|||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
|
|
||||||
def change_reach(self, new_reach, parent):
|
def change_reach(self, new_reach, parent):
|
||||||
|
|
||||||
self._undo.push(
|
self._undo.push(
|
||||||
ChangeReachCommand(
|
ChangeReachCommand(
|
||||||
new_reach,
|
new_reach,
|
||||||
|
|||||||
@@ -91,3 +91,21 @@ class GeometryTranslate(MainTranslate):
|
|||||||
self._dict["Shift"] = _translate(
|
self._dict["Shift"] = _translate(
|
||||||
"Geometry", "Shift"
|
"Geometry", "Shift"
|
||||||
)
|
)
|
||||||
|
self._dict["warning"] = _translate(
|
||||||
|
"Geometry", "Warning"
|
||||||
|
)
|
||||||
|
self._dict["format_not_exportable"] = _translate(
|
||||||
|
"Geometry", "The format of the file is not exportable."
|
||||||
|
)
|
||||||
|
self._dict["update_rk_error"] = _translate(
|
||||||
|
"Geometry",
|
||||||
|
"RK update can't be executed."
|
||||||
|
)
|
||||||
|
self._dict["update_rk_empty_profiles"] = _translate(
|
||||||
|
"Geometry",
|
||||||
|
"Some profiles don't contain any point."
|
||||||
|
)
|
||||||
|
self._dict["update_rk_empty_profiles_info"] = _translate(
|
||||||
|
"Geometry",
|
||||||
|
"Profiles concerned:"
|
||||||
|
)
|
||||||
@@ -247,14 +247,13 @@ class UpdateRKCommand(QUndoCommand):
|
|||||||
self._new_end_rk = None
|
self._new_end_rk = None
|
||||||
|
|
||||||
def undo(self):
|
def undo(self):
|
||||||
for rk, profile in zip(self._rks, self._reach.profiles):
|
|
||||||
profile.rk = rk
|
self._reach.update_rks(self._rks)
|
||||||
|
|
||||||
for begin_rk, end_rk, friction in zip(self._begin_rk,
|
for begin_rk, end_rk, friction in zip(self._begin_rk,
|
||||||
self._end_rk,
|
self._end_rk,
|
||||||
self._frictions):
|
self._frictions):
|
||||||
friction.begin_rk = begin_rk
|
friction.update_rks(begin_rk, end_rk)
|
||||||
friction.end_rk = end_rk
|
|
||||||
|
|
||||||
def redo(self):
|
def redo(self):
|
||||||
if self._new_rks is None:
|
if self._new_rks is None:
|
||||||
@@ -264,25 +263,23 @@ class UpdateRKCommand(QUndoCommand):
|
|||||||
**self._data
|
**self._data
|
||||||
)
|
)
|
||||||
|
|
||||||
for rk, profile in zip(self._new_rks, self._reach.profiles):
|
self._reach.update_rks(self._new_rks)
|
||||||
profile.rk = rk
|
|
||||||
for begin_rk, end_rk, friction in zip(self._new_begin_rk,
|
for begin_rk, end_rk, friction in zip(self._new_begin_rk,
|
||||||
self._new_end_rk,
|
self._new_end_rk,
|
||||||
self._frictions):
|
self._frictions):
|
||||||
friction.begin_rk = begin_rk
|
friction.update_rks(begin_rk, end_rk)
|
||||||
friction.end_rk = end_rk
|
|
||||||
|
|
||||||
|
|
||||||
class MeshingCommand(QUndoCommand):
|
class MeshingCommand(QUndoCommand):
|
||||||
def __init__(self, reach, mesher, data, tableView):
|
def __init__(self, reach, new_profiles, data, tableView):
|
||||||
QUndoCommand.__init__(self)
|
QUndoCommand.__init__(self)
|
||||||
|
|
||||||
self._reach = reach
|
self._reach = reach
|
||||||
self._data = data
|
self._data = data
|
||||||
self._limites = data["limites"]
|
self._limites = data["limites"]
|
||||||
self._mesher = mesher
|
|
||||||
|
|
||||||
self._new_profiles = None
|
self._new_profiles = new_profiles
|
||||||
self._new_profiles_indexes = None
|
self._new_profiles_indexes = None
|
||||||
self._tableView = tableView
|
self._tableView = tableView
|
||||||
self._indexes = tableView.selectionModel().selection()
|
self._indexes = tableView.selectionModel().selection()
|
||||||
@@ -308,12 +305,6 @@ class MeshingCommand(QUndoCommand):
|
|||||||
)
|
)
|
||||||
|
|
||||||
def redo(self):
|
def redo(self):
|
||||||
if self._new_profiles is None:
|
|
||||||
self._new_profiles = self._mesher.meshing(
|
|
||||||
self._reach,
|
|
||||||
**self._data
|
|
||||||
)
|
|
||||||
|
|
||||||
if self._new_profiles_indexes is None:
|
if self._new_profiles_indexes is None:
|
||||||
k = self._limites[0]
|
k = self._limites[0]
|
||||||
self._new_profiles_indexes = []
|
self._new_profiles_indexes = []
|
||||||
@@ -418,13 +409,13 @@ class ShiftCommand(QUndoCommand):
|
|||||||
def undo(self):
|
def undo(self):
|
||||||
for i in self._rows:
|
for i in self._rows:
|
||||||
profile = self._reach.profiles[i]
|
profile = self._reach.profiles[i]
|
||||||
self._reach.profiles[i].shift(
|
profile.shift(
|
||||||
-self._dx, -self._dy, -self._dz
|
-self._dx, -self._dy, -self._dz
|
||||||
)
|
)
|
||||||
|
|
||||||
def redo(self):
|
def redo(self):
|
||||||
for i in self._rows:
|
for i in self._rows:
|
||||||
profile = self._reach.profiles[i]
|
profile = self._reach.profiles[i]
|
||||||
self._reach.profiles[i].shift(
|
profile.shift(
|
||||||
self._dx, self._dy, self._dz
|
self._dx, self._dy, self._dz
|
||||||
)
|
)
|
||||||
@@ -130,8 +130,8 @@ class UpdateRKDialog(PamhyrDialog):
|
|||||||
return name
|
return name
|
||||||
|
|
||||||
def _init_default_values_guidelines(self):
|
def _init_default_values_guidelines(self):
|
||||||
bgl = ['un'] + self._gl + ['np']
|
bgl = ['un'] + self._gl + ['np', 'talweg']
|
||||||
egl = ['un'] + self._gl + ['np']
|
egl = ['un'] + self._gl + ['np', 'talweg']
|
||||||
|
|
||||||
self.combobox_add_items("comboBox_begin_gl", bgl)
|
self.combobox_add_items("comboBox_begin_gl", bgl)
|
||||||
self.combobox_add_items("comboBox_end_gl", egl)
|
self.combobox_add_items("comboBox_end_gl", egl)
|
||||||
|
|||||||
@@ -20,6 +20,7 @@ import os
|
|||||||
import sys
|
import sys
|
||||||
import time
|
import time
|
||||||
import pathlib
|
import pathlib
|
||||||
|
from pathlib import Path
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
from copy import deepcopy
|
from copy import deepcopy
|
||||||
@@ -301,7 +302,6 @@ class GeometryWindow(PamhyrWindow):
|
|||||||
self.tableView.model().blockSignals(False)
|
self.tableView.model().blockSignals(False)
|
||||||
|
|
||||||
def edit_meshing(self):
|
def edit_meshing(self):
|
||||||
|
|
||||||
rows = list(
|
rows = list(
|
||||||
set(
|
set(
|
||||||
(i.row() for i in self.tableView.selectedIndexes())
|
(i.row() for i in self.tableView.selectedIndexes())
|
||||||
@@ -325,6 +325,25 @@ class GeometryWindow(PamhyrWindow):
|
|||||||
logger_exception(e)
|
logger_exception(e)
|
||||||
return
|
return
|
||||||
|
|
||||||
|
ind = []
|
||||||
|
for i in range(self._reach.number_profiles):
|
||||||
|
if self._reach.profile(i).rk in selected_rk:
|
||||||
|
ind.append(i)
|
||||||
|
self.tableView.setFocus()
|
||||||
|
selection = self.tableView.selectionModel()
|
||||||
|
index = QItemSelection()
|
||||||
|
if len(ind) > 0:
|
||||||
|
for i in ind:
|
||||||
|
index.append(QItemSelectionRange(
|
||||||
|
self.tableView.model().index(i, 0))
|
||||||
|
)
|
||||||
|
selection.select(
|
||||||
|
index,
|
||||||
|
QItemSelectionModel.Rows |
|
||||||
|
QItemSelectionModel.ClearAndSelect |
|
||||||
|
QItemSelectionModel.Select
|
||||||
|
)
|
||||||
|
|
||||||
def _edit_meshing(self, data):
|
def _edit_meshing(self, data):
|
||||||
try:
|
try:
|
||||||
mesher = InternalMeshing()
|
mesher = InternalMeshing()
|
||||||
@@ -352,6 +371,29 @@ class GeometryWindow(PamhyrWindow):
|
|||||||
parent=self
|
parent=self
|
||||||
)
|
)
|
||||||
if dlg.exec():
|
if dlg.exec():
|
||||||
|
empty_profiles = [
|
||||||
|
profile for profile in self._reach.profiles
|
||||||
|
if profile.number_points == 0
|
||||||
|
]
|
||||||
|
if empty_profiles:
|
||||||
|
profile_names = ", ".join(
|
||||||
|
profile.name or str(profile.num)
|
||||||
|
for profile in empty_profiles
|
||||||
|
)
|
||||||
|
self.message_box(
|
||||||
|
(
|
||||||
|
self._trad["update_rk_error"] +
|
||||||
|
"\n" +
|
||||||
|
self._trad["update_rk_empty_profiles"]
|
||||||
|
),
|
||||||
|
(
|
||||||
|
self._trad["update_rk_empty_profiles_info"] +
|
||||||
|
f" {profile_names}"
|
||||||
|
),
|
||||||
|
window_title=self._trad["warning"]
|
||||||
|
)
|
||||||
|
return
|
||||||
|
|
||||||
data = {
|
data = {
|
||||||
"origin": dlg.origin,
|
"origin": dlg.origin,
|
||||||
"directrices": [dlg.begin_dir, dlg.end_dir],
|
"directrices": [dlg.begin_dir, dlg.end_dir],
|
||||||
@@ -370,9 +412,6 @@ class GeometryWindow(PamhyrWindow):
|
|||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger_exception(e)
|
logger_exception(e)
|
||||||
raise ExternFileMissingError(
|
raise ExternFileMissingError(
|
||||||
module="mage",
|
|
||||||
filename="MailleurTT",
|
|
||||||
path=MeshingWithMageMailleurTT._path(),
|
|
||||||
src_except=e
|
src_except=e
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -685,10 +724,24 @@ class GeometryWindow(PamhyrWindow):
|
|||||||
)
|
)
|
||||||
|
|
||||||
if filename != '' and filename is not None:
|
if filename != '' and filename is not None:
|
||||||
|
suffix = Path(filename).suffix
|
||||||
|
if suffix != "":
|
||||||
|
if suffix == ".st" or suffix == ".ST":
|
||||||
|
self._export_to_file_st(filename[:-3])
|
||||||
|
else:
|
||||||
|
# Warning popup when the format is not exportable
|
||||||
|
win = QtWidgets.QMessageBox()
|
||||||
|
win.setIcon(QtWidgets.QMessageBox.Warning)
|
||||||
|
win.setWindowTitle(self._trad["warning"])
|
||||||
|
win.setText(self._trad["format_not_exportable"])
|
||||||
|
win.exec()
|
||||||
|
|
||||||
|
pass
|
||||||
|
else:
|
||||||
self._export_to_file_st(filename)
|
self._export_to_file_st(filename)
|
||||||
|
|
||||||
def _export_to_file_st(self, filename):
|
def _export_to_file_st(self, filename):
|
||||||
with open(filename, "w+") as f:
|
with open(filename+".st", "w+") as f:
|
||||||
f.write("# Exported from Pamhyr2\n")
|
f.write("# Exported from Pamhyr2\n")
|
||||||
self._export_to_file_st_reach(f, self._reach)
|
self._export_to_file_st_reach(f, self._reach)
|
||||||
|
|
||||||
|
|||||||
@@ -138,6 +138,9 @@ class TableModel(PamhyrTableModel):
|
|||||||
if not index.isValid() or role != Qt.EditRole:
|
if not index.isValid() or role != Qt.EditRole:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
if self.is_same_data(index, value):
|
||||||
|
return False
|
||||||
|
|
||||||
row = index.row()
|
row = index.row()
|
||||||
column = index.column()
|
column = index.column()
|
||||||
|
|
||||||
@@ -288,6 +291,9 @@ class ParametersTableModel(PamhyrTableModel):
|
|||||||
if not index.isValid() or role != Qt.EditRole:
|
if not index.isValid() or role != Qt.EditRole:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
if self.is_same_data(index, value):
|
||||||
|
return False
|
||||||
|
|
||||||
if self._hs_index is None:
|
if self._hs_index is None:
|
||||||
return QVariant()
|
return QVariant()
|
||||||
|
|
||||||
|
|||||||
@@ -42,7 +42,7 @@ class PlotAC(PamhyrPlot):
|
|||||||
self._isometric_axis = False
|
self._isometric_axis = False
|
||||||
|
|
||||||
self._auto_relim_update = True
|
self._auto_relim_update = True
|
||||||
self._autoscale_update = True
|
self._autoscale_update = False
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def river(self):
|
def river(self):
|
||||||
|
|||||||
@@ -50,7 +50,7 @@ class PlotRKC(PamhyrPlot):
|
|||||||
self._isometric_axis = False
|
self._isometric_axis = False
|
||||||
|
|
||||||
self._auto_relim_update = True
|
self._auto_relim_update = True
|
||||||
self._autoscale_update = True
|
self._autoscale_update = False
|
||||||
self.parent = parent
|
self.parent = parent
|
||||||
self.anotate_lst = []
|
self.anotate_lst = []
|
||||||
|
|
||||||
@@ -136,13 +136,18 @@ class PlotRKC(PamhyrPlot):
|
|||||||
hs_color = []
|
hs_color = []
|
||||||
index = self.parent.tableView.selectedIndexes()
|
index = self.parent.tableView.selectedIndexes()
|
||||||
for i, hs in enumerate(lhs):
|
for i, hs in enumerate(lhs):
|
||||||
|
section = hs.input_section
|
||||||
|
if section is None:
|
||||||
|
continue
|
||||||
|
|
||||||
if i == index[0].row():
|
if i == index[0].row():
|
||||||
hs_color.append("blue")
|
color = "blue"
|
||||||
elif hs.enabled:
|
elif hs.enabled:
|
||||||
hs_color.append("red")
|
color = "red"
|
||||||
else:
|
else:
|
||||||
hs_color.append("darkgrey")
|
color = "darkgrey"
|
||||||
x = hs.input_section.rk
|
|
||||||
|
x = section.rk
|
||||||
if x is not None:
|
if x is not None:
|
||||||
a = self.canvas.axes.annotate(
|
a = self.canvas.axes.annotate(
|
||||||
" > " + hs.name,
|
" > " + hs.name,
|
||||||
@@ -151,12 +156,13 @@ class PlotRKC(PamhyrPlot):
|
|||||||
verticalalignment='top',
|
verticalalignment='top',
|
||||||
annotation_clip=True,
|
annotation_clip=True,
|
||||||
fontsize=9,
|
fontsize=9,
|
||||||
color=hs_color[-1],
|
color=color,
|
||||||
)
|
)
|
||||||
self.anotate_lst.append(a)
|
self.anotate_lst.append(a)
|
||||||
vx.append(x)
|
vx.append(x)
|
||||||
vymin.append(min(z_min))
|
vymin.append(min(z_min))
|
||||||
vymax.append(max(z_max))
|
vymax.append(max(z_max))
|
||||||
|
hs_color.append(color)
|
||||||
|
|
||||||
self.hs_vlines = self.canvas.axes.vlines(
|
self.hs_vlines = self.canvas.axes.vlines(
|
||||||
x=vx, ymin=vymin, ymax=vymax,
|
x=vx, ymin=vymin, ymax=vymax,
|
||||||
|
|||||||
@@ -169,6 +169,9 @@ class TableModel(PamhyrTableModel):
|
|||||||
if not index.isValid() or role != Qt.EditRole:
|
if not index.isValid() or role != Qt.EditRole:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
if self.is_same_data(index, value):
|
||||||
|
return False
|
||||||
|
|
||||||
row = index.row()
|
row = index.row()
|
||||||
column = index.column()
|
column = index.column()
|
||||||
na = self._trad['not_associated']
|
na = self._trad['not_associated']
|
||||||
|
|||||||
@@ -47,7 +47,7 @@ class PlotDRK(PamhyrPlot):
|
|||||||
self._isometric_axis = False
|
self._isometric_axis = False
|
||||||
|
|
||||||
self._auto_relim_update = True
|
self._auto_relim_update = True
|
||||||
self._autoscale_update = True
|
self._autoscale_update = False
|
||||||
|
|
||||||
@timer
|
@timer
|
||||||
def draw(self, highlight=None):
|
def draw(self, highlight=None):
|
||||||
|
|||||||
@@ -37,7 +37,7 @@ class PlotDischarge(PamhyrPlot):
|
|||||||
self._isometric_axis = False
|
self._isometric_axis = False
|
||||||
|
|
||||||
self._auto_relim_update = True
|
self._auto_relim_update = True
|
||||||
self._autoscale_update = True
|
self._autoscale_update = False
|
||||||
|
|
||||||
@timer
|
@timer
|
||||||
def draw(self):
|
def draw(self):
|
||||||
|
|||||||
@@ -153,6 +153,9 @@ class InitialConditionTableModel(PamhyrTableModel):
|
|||||||
if not index.isValid() or role != Qt.EditRole:
|
if not index.isValid() or role != Qt.EditRole:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
if self.is_same_data(index, value):
|
||||||
|
return False
|
||||||
|
|
||||||
row = index.row()
|
row = index.row()
|
||||||
column = index.column()
|
column = index.column()
|
||||||
|
|
||||||
@@ -173,7 +176,7 @@ class InitialConditionTableModel(PamhyrTableModel):
|
|||||||
def add(self, row, parent=QModelIndex()):
|
def add(self, row, parent=QModelIndex()):
|
||||||
self.beginInsertRows(parent, row, row - 1)
|
self.beginInsertRows(parent, row, row - 1)
|
||||||
|
|
||||||
row = self.get_true_data_row(row)
|
# row = self.get_true_data_row(row)
|
||||||
|
|
||||||
self._undo.push(
|
self._undo.push(
|
||||||
AddCommand(
|
AddCommand(
|
||||||
@@ -330,9 +333,27 @@ class InitialConditionTableModel(PamhyrTableModel):
|
|||||||
if not (line.startswith("#") or
|
if not (line.startswith("#") or
|
||||||
line.startswith("*") or
|
line.startswith("*") or
|
||||||
line.startswith("$")):
|
line.startswith("$")):
|
||||||
line = line.split()
|
line_split = line.split()
|
||||||
if int(line[0]) == reach_id:
|
formated = False
|
||||||
data.append([line[4], line[2], line[3]])
|
if len(line_split[0]) > 3:
|
||||||
|
formated = True
|
||||||
|
elif len(line_split[2]) > 10:
|
||||||
|
formated = True
|
||||||
|
elif len(line_split[3]) > 11:
|
||||||
|
formated = True
|
||||||
|
elif len(line_split[4]) > 9:
|
||||||
|
formated = True
|
||||||
|
|
||||||
|
if formated: # old PamHyr format
|
||||||
|
if int(line[1:4]) == reach_id:
|
||||||
|
data.append([line[31:40],
|
||||||
|
line[10:20],
|
||||||
|
line[20:31]])
|
||||||
|
else:
|
||||||
|
if int(line_split[0]) == reach_id:
|
||||||
|
data.append([line_split[4],
|
||||||
|
line_split[2],
|
||||||
|
line_split[3]])
|
||||||
|
|
||||||
self._undo.push(
|
self._undo.push(
|
||||||
ReplaceDataCommand(
|
ReplaceDataCommand(
|
||||||
|
|||||||
@@ -107,8 +107,6 @@ class InitialConditionsWindow(PamhyrWindow):
|
|||||||
self.setup_connections()
|
self.setup_connections()
|
||||||
self.setub_dialogs()
|
self.setub_dialogs()
|
||||||
|
|
||||||
self.ui.setWindowTitle(self._title)
|
|
||||||
|
|
||||||
def setup_table(self):
|
def setup_table(self):
|
||||||
table = self.find(QTableView, f"tableView")
|
table = self.find(QTableView, f"tableView")
|
||||||
self._delegate_rk = ComboBoxDelegate(
|
self._delegate_rk = ComboBoxDelegate(
|
||||||
@@ -224,6 +222,8 @@ class InitialConditionsWindow(PamhyrWindow):
|
|||||||
def _update_plot(self):
|
def _update_plot(self):
|
||||||
self.plot_1.draw()
|
self.plot_1.draw()
|
||||||
self.plot_2.draw()
|
self.plot_2.draw()
|
||||||
|
self.plot_1.idle()
|
||||||
|
self.plot_2.idle()
|
||||||
|
|
||||||
def _propagated_update(self, key=Modules(0)):
|
def _propagated_update(self, key=Modules(0)):
|
||||||
if Modules.GEOMETRY not in key:
|
if Modules.GEOMETRY not in key:
|
||||||
@@ -320,10 +320,12 @@ class InitialConditionsWindow(PamhyrWindow):
|
|||||||
def _import_from_results(self, results):
|
def _import_from_results(self, results):
|
||||||
logger.debug(f"import from results: {results}")
|
logger.debug(f"import from results: {results}")
|
||||||
self._table.import_from_results(results)
|
self._table.import_from_results(results)
|
||||||
|
self._update()
|
||||||
|
|
||||||
def _import_from_ini_file(self, file_name):
|
def _import_from_ini_file(self, file_name):
|
||||||
logger.debug(f"import from INI file: {file_name}")
|
logger.debug(f"import from INI file: {file_name}")
|
||||||
self._table.read_from_ini(file_name)
|
self._table.read_from_ini(file_name)
|
||||||
|
self._update()
|
||||||
|
|
||||||
def move_up(self):
|
def move_up(self):
|
||||||
row = self.index_selected_row()
|
row = self.index_selected_row()
|
||||||
|
|||||||
@@ -117,7 +117,12 @@ class InitialConditionTableModel(PamhyrTableModel):
|
|||||||
self._data = data
|
self._data = data
|
||||||
|
|
||||||
def _setup_lst(self):
|
def _setup_lst(self):
|
||||||
self._lst = self._data._data
|
self._lst = list(
|
||||||
|
filter(
|
||||||
|
lambda ica: ica._deleted is False,
|
||||||
|
self._data._data
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
def rowCount(self, parent):
|
def rowCount(self, parent):
|
||||||
return len(self._lst)
|
return len(self._lst)
|
||||||
@@ -183,6 +188,9 @@ class InitialConditionTableModel(PamhyrTableModel):
|
|||||||
if not index.isValid() or role != Qt.EditRole:
|
if not index.isValid() or role != Qt.EditRole:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
if self.is_same_data(index, value):
|
||||||
|
return False
|
||||||
|
|
||||||
row = index.row()
|
row = index.row()
|
||||||
column = index.column()
|
column = index.column()
|
||||||
|
|
||||||
@@ -216,25 +224,34 @@ class InitialConditionTableModel(PamhyrTableModel):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
self._setup_lst()
|
||||||
self.endInsertRows()
|
self.endInsertRows()
|
||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
|
|
||||||
def delete(self, rows, parent=QModelIndex()):
|
def delete(self, rows, parent=QModelIndex()):
|
||||||
self.beginRemoveRows(parent, rows[0], rows[-1])
|
self.beginRemoveRows(parent, rows[0], rows[-1])
|
||||||
|
|
||||||
|
data_rows = {
|
||||||
|
id(ica): i for i, ica in enumerate(self._data._data)
|
||||||
|
}
|
||||||
self._undo.push(
|
self._undo.push(
|
||||||
DelCommand(
|
DelCommand(
|
||||||
self._data, self._lst, rows
|
self._data,
|
||||||
|
self._data._data,
|
||||||
|
[data_rows[id(self._lst[row])] for row in rows]
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
self._setup_lst()
|
||||||
self.endRemoveRows()
|
self.endRemoveRows()
|
||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
|
|
||||||
def undo(self):
|
def undo(self):
|
||||||
self._undo.undo()
|
self._undo.undo()
|
||||||
|
self._setup_lst()
|
||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
|
|
||||||
def redo(self):
|
def redo(self):
|
||||||
self._undo.redo()
|
self._undo.redo()
|
||||||
|
self._setup_lst()
|
||||||
self.layoutChanged.emit()
|
self.layoutChanged.emit()
|
||||||
@@ -181,7 +181,6 @@ class DelCommand(QUndoCommand):
|
|||||||
self._data = data
|
self._data = data
|
||||||
self._ics_spec = ics_spec
|
self._ics_spec = ics_spec
|
||||||
self._rows = rows
|
self._rows = rows
|
||||||
# self._data = data
|
|
||||||
|
|
||||||
self._ic = []
|
self._ic = []
|
||||||
for row in rows:
|
for row in rows:
|
||||||
|
|||||||
@@ -68,15 +68,22 @@ class InitialConditionsAdisTSWindow(PamhyrWindow):
|
|||||||
_pamhyr_ui = "InitialConditionsAdisTS"
|
_pamhyr_ui = "InitialConditionsAdisTS"
|
||||||
_pamhyr_name = "Initial condition AdisTS"
|
_pamhyr_name = "Initial condition AdisTS"
|
||||||
|
|
||||||
def __init__(self, data=None, study=None, config=None, parent=None):
|
def __init__(self, data=None, pollutant_id=None, study=None,
|
||||||
|
config=None, parent=None):
|
||||||
self._data = []
|
self._data = []
|
||||||
self._data.append(data)
|
self._data.append(data)
|
||||||
|
self._pollutant_id = pollutant_id
|
||||||
trad = IcAdisTSTranslate()
|
trad = IcAdisTSTranslate()
|
||||||
|
|
||||||
|
self._pollutant_name = next(
|
||||||
|
(x.name for x in study._river._Pollutants.Pollutants_List
|
||||||
|
if x.id == self._pollutant_id),
|
||||||
|
None
|
||||||
|
)
|
||||||
name = (
|
name = (
|
||||||
trad[self._pamhyr_name] +
|
trad[self._pamhyr_name] +
|
||||||
" - " + study.name +
|
" - " + study.name +
|
||||||
" - " + self._data[0].name
|
" - " + self._pollutant_name
|
||||||
)
|
)
|
||||||
|
|
||||||
super(InitialConditionsAdisTSWindow, self).__init__(
|
super(InitialConditionsAdisTSWindow, self).__init__(
|
||||||
@@ -93,18 +100,21 @@ class InitialConditionsAdisTSWindow(PamhyrWindow):
|
|||||||
|
|
||||||
self.setup_table()
|
self.setup_table()
|
||||||
|
|
||||||
self.ui.setWindowTitle(self._title)
|
|
||||||
|
|
||||||
def setup_table(self):
|
def setup_table(self):
|
||||||
|
|
||||||
path_icons = os.path.join(self._get_ui_directory(), f"ressources")
|
path_icons = os.path.join(self._get_ui_directory(), f"ressources")
|
||||||
|
|
||||||
table_default = self.find(QTableView, f"tableView")
|
table_default = self.find(QTableView, f"tableView")
|
||||||
|
|
||||||
|
if self._study.is_editable():
|
||||||
|
editable_headers = self._trad.get_dict("table_headers")
|
||||||
|
else:
|
||||||
|
editable_headers = []
|
||||||
|
|
||||||
self._table = InitialConditionTableDefaultModel(
|
self._table = InitialConditionTableDefaultModel(
|
||||||
table_view=table_default,
|
table_view=table_default,
|
||||||
table_headers=self._trad.get_dict("table_headers"),
|
table_headers=self._trad.get_dict("table_headers"),
|
||||||
editable_headers=["name", "concentration", "eg", "em", "ed"],
|
editable_headers=editable_headers,
|
||||||
delegates={},
|
delegates={},
|
||||||
data=self._data,
|
data=self._data,
|
||||||
undo=self._undo_stack,
|
undo=self._undo_stack,
|
||||||
@@ -124,9 +134,13 @@ class InitialConditionsAdisTSWindow(PamhyrWindow):
|
|||||||
|
|
||||||
action_add = QAction(self)
|
action_add = QAction(self)
|
||||||
action_add.setIcon(QIcon(os.path.join(path_icons, f"add.png")))
|
action_add.setIcon(QIcon(os.path.join(path_icons, f"add.png")))
|
||||||
|
|
||||||
|
if self._study.is_editable():
|
||||||
action_add.triggered.connect(self.add)
|
action_add.triggered.connect(self.add)
|
||||||
action_delete = QAction(self)
|
action_delete = QAction(self)
|
||||||
action_delete.setIcon(QIcon(os.path.join(path_icons, f"del.png")))
|
action_delete.setIcon(QIcon(os.path.join(path_icons, f"del.png")))
|
||||||
|
|
||||||
|
if self._study.is_editable():
|
||||||
action_delete.triggered.connect(self.delete)
|
action_delete.triggered.connect(self.delete)
|
||||||
|
|
||||||
toolBar.addAction(action_add)
|
toolBar.addAction(action_add)
|
||||||
@@ -150,12 +164,15 @@ class InitialConditionsAdisTSWindow(PamhyrWindow):
|
|||||||
mode="rk"
|
mode="rk"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
if self._study.is_editable():
|
||||||
|
editable_headers_spec = self._trad.get_dict("table_headers_spec")
|
||||||
|
else:
|
||||||
|
editable_headers_spec = []
|
||||||
|
|
||||||
self._table_spec = InitialConditionTableModel(
|
self._table_spec = InitialConditionTableModel(
|
||||||
table_view=self.table_spec,
|
table_view=self.table_spec,
|
||||||
table_headers=self._trad.get_dict("table_headers_spec"),
|
table_headers=self._trad.get_dict("table_headers_spec"),
|
||||||
editable_headers=["name", "reach", "start_rk",
|
editable_headers=editable_headers_spec,
|
||||||
"end_rk", "concentration",
|
|
||||||
"eg", "em", "ed", "rate"],
|
|
||||||
delegates={
|
delegates={
|
||||||
"reach": self._delegate_reach,
|
"reach": self._delegate_reach,
|
||||||
"start_rk": self._delegate_rk,
|
"start_rk": self._delegate_rk,
|
||||||
@@ -266,11 +283,27 @@ class InitialConditionsAdisTSWindow(PamhyrWindow):
|
|||||||
self._update()
|
self._update()
|
||||||
|
|
||||||
def _undo(self):
|
def _undo(self):
|
||||||
|
undo_stack = self._undo_stack
|
||||||
|
if undo_stack is None or not undo_stack.canUndo():
|
||||||
|
return
|
||||||
|
|
||||||
|
if isinstance(undo_stack.command(undo_stack.index() - 1), SetCommand):
|
||||||
self._table.undo()
|
self._table.undo()
|
||||||
|
else:
|
||||||
|
self._table_spec.undo()
|
||||||
|
|
||||||
self._update()
|
self._update()
|
||||||
|
|
||||||
def _redo(self):
|
def _redo(self):
|
||||||
|
undo_stack = self._undo_stack
|
||||||
|
if undo_stack is None or not undo_stack.canRedo():
|
||||||
|
return
|
||||||
|
|
||||||
|
if isinstance(undo_stack.command(undo_stack.index()), SetCommand):
|
||||||
self._table.redo()
|
self._table.redo()
|
||||||
|
else:
|
||||||
|
self._table_spec.redo()
|
||||||
|
|
||||||
self._update()
|
self._update()
|
||||||
|
|
||||||
def add(self):
|
def add(self):
|
||||||
|
|||||||
Loaded 100 of 174 files, more files were not shown because too many files have changed in this diff.
Show more
Reference in new issue
Block a user