Compare commits

...
108 Commits
Author SHA1 Message Date
Pierre-Antoine Rouby 4f91de4c97 Merge branch 'master' of gitlab-ssh.irstea.fr:theophile.terraz/pamhyr. 2023-11-29 14:49:02 +01:00
Pierre-Antoine Rouby 37b04961f1 IC: Add Manning/Strickler equation for IC approximation. 2023-11-29 14:47:35 +01:00
Theophile Terraz 0a63f16f62 Merge branch 'master' of gitlab-ssh.irstea.fr:theophile.terraz/pamhyr 2023-11-29 14:34:11 +01:00
Theophile Terraz 92eb729a39 add Reservoirs 2023-11-29 14:34:06 +01:00
Pierre-Antoine Rouby d891c9783d Model: Fix possible data removing durring save. 2023-11-27 16:58:46 +01:00
Pierre-Antoine Rouby 51858597cd Mage: ST export minor change. 2023-11-27 14:49:42 +01:00
Pierre-Antoine Rouby 8b3baf433f Solver run: Used merged channels to get stderr. 2023-11-27 14:31:04 +01:00
Pierre-Antoine Rouby 12e3d779c6 Mage: Fix BC export when node is not associate. 2023-11-27 11:40:59 +01:00
Pierre-Antoine Rouby c6274890a7 BC: Rename some BC types. 2023-11-27 11:29:39 +01:00
Theophile Terraz 907fc0c53a add step-by-step tutorial 2023-11-27 09:44:57 +01:00
Pierre-Antoine Rouby b4ee76416e BC, LC: Disable time delegate (fix #23). 2023-11-27 09:37:23 +01:00
Pierre-Antoine Rouby a88b42a8ef BC: Fix get_assoc_to_node in case of no results. 2023-11-27 09:21:23 +01:00
Pierre-Antoine Rouby 08f01aee69 Network: Fix #22. 2023-11-27 09:12:31 +01:00
Pierre-Antoine Rouby c69cbdb34e BC: Make node association unique (#25). 2023-11-27 08:55:38 +01:00
Pierre-Antoine Rouby 01fd7dbec8 Pamhyr2: Fix some type compare. 2023-11-22 15:47:57 +01:00
Pierre-Antoine Rouby 2af9b0018c SQL: Minor change. 2023-11-22 15:39:21 +01:00
Pierre-Antoine Rouby 7f8e8d5a9f SQL: Minor change and add exception handler. 2023-11-22 14:58:03 +01:00
Pierre-Antoine Rouby c9887a97e7 Friction: Disable name column because the value is not in database. 2023-11-22 10:20:50 +01:00
Pierre-Antoine Rouby 1a708921e3 BC: Fix #13 paste undo command. 2023-11-22 10:02:25 +01:00
Pierre-Antoine Rouby 9fe1b35a59 IC: Fix profile side effect on IC generator. 2023-11-21 16:06:37 +01:00
Pierre-Antoine Rouby edf744eee1 Geometry: Fix profiles list sorting. 2023-11-21 14:24:31 +01:00
Pierre-Antoine Rouby e400980103 Frictions: Fix plot drawing crash. 2023-11-21 11:07:36 +01:00
Pierre-Antoine Rouby b12e935a5d Geometry: Fix some translate. 2023-11-21 10:45:00 +01:00
Pierre-Antoine Rouby 1d4422e9c5 MainWindow: Disable window maximized. 2023-11-21 10:38:35 +01:00
Pierre-Antoine Rouby 1a10da9c89 MainWindow: Fix translate and re organize UI. 2023-11-21 10:36:13 +01:00
Pierre-Antoine Rouby 470eae8b92 pamhyr: Fix #20 study reload whene study is not saved on a file. 2023-11-21 09:26:57 +01:00
Pierre-Antoine Rouby 9b53b4302b BC,IC: Enable to get time int from float string (fix #21). 2023-11-21 09:11:35 +01:00
Pierre-Antoine Rouby 461202219d Window: Fix paste delimiter. 2023-11-20 17:29:37 +01:00
Pierre-Antoine Rouby ddfaa3a73f git: Add pip requirements. 2023-11-20 16:35:32 +01:00
Pierre-Antoine Rouby 3e531130fe Merge branch 'master' of gitlab-ssh.irstea.fr:theophile.terraz/pamhyr 2023-11-20 16:16:24 +01:00
Pierre-Antoine Rouby 595eb29c42 BC: Fix long type translate crash. 2023-11-20 16:15:43 +01:00
Pierre-Antoine Rouby ccbbc83502 IC: Fix sql_save as generator value type. 2023-11-20 16:10:10 +01:00
Pierre-Antoine Rouby fd0a8ddf51 BC: Fix paste command. 2023-11-20 15:55:57 +01:00
Pierre-Antoine Rouby ce5808e3ce Geometry: Fix delete profiles. 2023-11-20 15:38:52 +01:00
Pierre-Antoine Rouby 4dafa345b4 Geometry: Fix delete command crash. 2023-11-20 15:37:07 +01:00
Theophile Terraz 90776f0af9 Merge branch 'master' of gitlab-ssh.irstea.fr:theophile.terraz/pamhyr 2023-11-02 16:44:02 +01:00
Theophile Terraz 32777456b9 add scripts to use Mascaret 2023-11-02 16:43:57 +01:00
Pierre-Antoine Rouby dcaa7fafbd tools: Fix PEP8. 2023-10-30 09:30:16 +01:00
Pierre-Antoine Rouby 17eb3e8184 tools: parser: Minor change and tests fixes. 2023-10-27 17:21:09 +02:00
Pierre-Antoine Rouby 7bbca7cf54 unittests: Add parse tests for ' quote. 2023-10-27 15:10:16 +02:00
Pierre-Antoine Rouby 3c738ebcb9 Solver: Mage: Fix profile sediment export. 2023-10-27 14:32:51 +02:00
Pierre-Antoine Rouby 6522cea03f pamhyr: Set axis label color to 'black' instead 'green'. 2023-10-27 14:23:55 +02:00
Pierre-Antoine Rouby ea2dbfba3e pamhyr: Rename '_sql_*' function to '_db_*'. 2023-10-27 14:15:42 +02:00
Pierre-Antoine Rouby 8d92fbbdac pamhyr: Delete useless '__init__.py' file. 2023-10-27 14:13:01 +02:00
Pierre-Antoine Rouby 77f06aaa07 tools: Time the parser function. 2023-10-27 11:53:00 +02:00
Pierre-Antoine Rouby b79c5b30c0 tests: Add tests for flatten function. 2023-10-27 11:50:36 +02:00
Pierre-Antoine Rouby fdac834d48 Tests: Fix pep8. 2023-10-27 11:34:11 +02:00
Pierre-Antoine Rouby 4355be1a9c Solver: Mage: Split ST export function. 2023-10-27 11:31:20 +02:00
Pierre-Antoine Rouby 66f6d3505b Solver: CommandLine: Fix path replace. 2023-10-27 11:15:58 +02:00
Pierre-Antoine Rouby 65c261887c Tools: Command line parser: Add a test. 2023-10-27 11:14:10 +02:00
Pierre-Antoine Rouby 6c794d2eb4 Solver: Refactoring command line parser and add unittest. 2023-10-27 11:05:16 +02:00
Pierre-Antoine Rouby 51c912f3a8 Solver: Code refactoring. 2023-10-27 09:33:32 +02:00
Pierre-Antoine Rouby 76e90f417e Script: Fix pep8. 2023-10-24 15:26:55 +02:00
Pierre-Antoine Rouby e556bcff54 Script: Add script command list solver. 2023-10-24 15:13:35 +02:00
Pierre-Antoine Rouby 4833685e2e Script: Add run/export script. 2023-10-24 15:05:05 +02:00
Pierre-Antoine Rouby 2d7681086d ci: Add configure stage. 2023-10-24 11:11:34 +02:00
Pierre-Antoine Rouby f44ac69d68 ci: Change stages order (tests before build). 2023-10-24 10:58:22 +02:00
Pierre-Antoine Rouby 32684af292 tests: Minor change. 2023-10-24 10:56:23 +02:00
Pierre-Antoine Rouby 862ec88191 tests: Add test script and fix pep8 in unittest. 2023-10-24 10:48:54 +02:00
Pierre-Antoine Rouby d36a9cd340 tests: Fix test env path. 2023-10-24 10:43:33 +02:00
Pierre-Antoine Rouby afafbb0c5e tests: Add unittest for Model and minor change. 2023-10-24 10:38:34 +02:00
Pierre-Antoine Rouby e3f01b575a Solver: Mage: Fix minor in ST file export. 2023-10-23 10:31:59 +02:00
Pierre-Antoine Rouby d6924998d9 Results: CustomPlot: Fix pep8. 2023-10-20 17:25:36 +02:00
Pierre-Antoine Rouby 0ae5c02ea0 Friction: Fix open stricker edit. 2023-10-20 17:20:03 +02:00
Pierre-Antoine Rouby e395f9f575 BC, LC: Fix sub window open. 2023-10-20 17:16:37 +02:00
Pierre-Antoine Rouby f074e1fdf2 Geometry: Fix profile sub window open. 2023-10-20 17:01:28 +02:00
Pierre-Antoine Rouby 4d1eb0fecc View: Fix results window open. 2023-10-20 16:57:32 +02:00
Pierre-Antoine Rouby f5fdde9a5d Merge branch 'results-custom-plot'. 2023-10-20 16:51:26 +02:00
Pierre-Antoine Rouby 32aa5bece5 Results: CustomPlot: Add legend and change axes label fontsize for each plot. 2023-10-20 16:48:21 +02:00
Pierre-Antoine Rouby cdfb867037 Results: CustomPlot: Add custom time axes. 2023-10-20 16:27:07 +02:00
Pierre-Antoine Rouby 021971816a Results: CustomPlot: Implement plot with time on y axes. 2023-10-20 15:09:24 +02:00
Pierre-Antoine Rouby e4386e20a3 Network: Fix item selection. 2023-10-20 11:45:28 +02:00
Pierre-Antoine Rouby 8760cd364d Pamhyr: Implement window duplication deny mechanism with hash (#7). 2023-10-20 10:48:07 +02:00
Pierre-Antoine Rouby 4c40f675b5 IC: Fix #14. 2023-10-20 09:25:19 +02:00
Pierre-Antoine Rouby 325bb9ed02 Solver: Mage: Fix warning. 2023-10-18 16:43:01 +02:00
Pierre-Antoine Rouby 505d1aeb48 Solver: Add @output command line parameters. 2023-10-18 16:40:28 +02:00
Pierre-Antoine Rouby 1dd189cbd4 Results: Implement results reload (for same solver) #16. 2023-10-18 16:00:32 +02:00
Pierre-Antoine Rouby fa7490dee8 pep8: Minor fix. 2023-10-18 14:55:32 +02:00
Pierre-Antoine Rouby 7fda5b6b4a Mage: Fix #17 ST file export format. 2023-10-18 14:45:38 +02:00
Pierre-Antoine Rouby 3ff0883eca RunSolver: Fix #15 and minor change. 2023-10-18 14:24:14 +02:00
Pierre-Antoine Rouby 04c8f1ae5f Results: CustomPlot: Minor change to fix warnings. 2023-10-18 14:02:55 +02:00
Pierre-Antoine Rouby f73000056c Results: CustomPlot: Fix kp / elevation, water_elevation and discharge drawing. 2023-10-18 10:53:57 +02:00
Pierre-Antoine Rouby 4605950b94 Results: CustomPlot: Minor change. 2023-10-17 17:10:58 +02:00
Pierre-Antoine Rouby ae40097b51 Results: Fix pep8 format. 2023-10-17 11:27:00 +02:00
Pierre-Antoine Rouby daaa9b193e Resutls: CustomPlot: Prepare ploting mecanisme. 2023-10-17 11:25:30 +02:00
Pierre-Antoine Rouby 98ee3ee6ec CustomPlot: Add translation. 2023-10-16 17:25:08 +02:00
Pierre-Antoine Rouby b77aed2395 results: Add custom plot dialog scheme. 2023-10-16 15:52:29 +02:00
Pierre-Antoine Rouby 45afdc25f9 packages: Fix motd.txt missing and pip version in linux.sh. 2023-10-16 11:41:59 +02:00
Pierre-Antoine Rouby 2ae681cebc ci: Add a full requirements for doc build. 2023-10-11 16:52:09 +02:00
Pierre-Antoine Rouby cc465a050c ci: Allow doc build failed. 2023-10-11 16:50:14 +02:00
Pierre-Antoine Rouby 1fd840d33f ci: Update pip. 2023-10-11 16:47:52 +02:00
Pierre-Antoine Rouby cc237fffa8 pamhyr: Fix package requirements list. 2023-10-11 16:37:57 +02:00
Pierre-Antoine Rouby bb27d4ec20 ci: Fix env install package. 2023-10-11 16:26:43 +02:00
Pierre-Antoine Rouby 255f7302ea ci: Fix remove option -U from pip install 2023-10-11 16:24:45 +02:00
Pierre-Antoine Rouby 9e0b4fd243 ci: Fix missing package for pdoc, fix requirements and pep8. 2023-10-11 16:22:45 +02:00
Pierre-Antoine Rouby 2e926c9ece debug: Switch from QPlainTextEdit to PamhyrPythonEditor. 2023-10-11 16:17:21 +02:00
Pierre-Antoine Rouby bcc2d2f4d4 ci: Allow pep8 failed. 2023-10-11 15:16:04 +02:00
Pierre-Antoine Rouby e3555b2c51 pep8: Fix all pep8 format error. 2023-10-11 15:08:33 +02:00
Pierre-Antoine Rouby e99037223b pep8: Fix some View code format. 2023-10-11 11:56:48 +02:00
Pierre-Antoine Rouby 9f2b81d385 pep8: Fix code format for model. 2023-10-11 11:47:09 +02:00
Pierre-Antoine Rouby 2c592b25b9 pep8: Fix code format. 2023-10-11 11:23:46 +02:00
Pierre-Antoine Rouby bf05b5ddc1 ci: Add pep8 test job. 2023-10-11 11:12:32 +02:00
Pierre-Antoine Rouby fa2454d079 pep8: Some modifications to respect pep8 standard code. 2023-10-11 11:06:45 +02:00
Pierre-Antoine Rouby 8544dd1144 doc: Disactive documentation display in Pamhyr2 for lighter packages. 2023-10-11 10:40:58 +02:00
Pierre-Antoine Rouby 1f8b0bddb4 ui: Fix geometry cross section window lang. 2023-10-11 08:52:05 +02:00
Pierre-Antoine Rouby 5b2ee80329 geometry, results, configure: Some fixes. 2023-10-11 08:40:54 +02:00
Pierre-Antoine Rouby acfb876da0 doc: Change the Firefox button to an open button. 2023-10-10 15:28:59 +02:00
Pierre-Antoine Rouby ab9ebc6400 pamhyr: Add just-in-time database geometry loading. 2023-10-10 15:17:53 +02:00
218 changed files with 12613 additions and 2437 deletions

No files matched your search

+62 -36
View File
@@ -16,8 +16,9 @@
stages:
- downloads
- build
- configure
- test
- build
- package
- release
@@ -59,12 +60,12 @@ dl-mage-windows:
- mage-windows/mage_extraire.exe
- mage-windows/mailleurPF.exe
#########
# BUILD #
#########
#############
# CONFIGURE #
#############
set-version:
stage: build
stage: configure
tags:
- linux
script:
@@ -75,7 +76,7 @@ set-version:
- VERSION
build-lang:
stage: build
stage: configure
tags:
- linux
script:
@@ -85,6 +86,49 @@ build-lang:
paths:
- src/lang/*.qm
#########
# TESTS #
#########
unittest:
stage: test
tags:
- linux
needs:
- job: set-version
artifacts: true
script:
- python3 -m venv venv
- . venv/bin/activate
- pip3 install -U pip
- pip3 install -r ./full-requirements.txt
- pip3 install -U -r ./full-requirements.txt
- cd src
- python3 -m unittest discover -t .
test-pep8:
stage: test
tags:
- linux
needs:
- job: set-version
artifacts: true
- job: unittest
script:
# Setup virtual env
- python3 -m venv venv
- . venv/bin/activate
- pip3 install -U pip
- pip3 install -r ./requirements.txt
- pip3 install -U -r ./requirements.txt
- pip3 install pycodestyle
- pycodestyle ./src
allow_failure: true
#########
# BUILD #
#########
build-users-doc:
stage: build
tags:
@@ -100,6 +144,7 @@ build-users-doc:
- doc/users/documentation.pdf
- doc/users/documentation.html
- doc/users/images/
allow_failure: true
build-developers-doc:
stage: build
@@ -113,6 +158,12 @@ build-developers-doc:
- ./build.sh
# Create html documentation
- cd ../../src
- python3 -m venv doc-venv
- . doc-venv/bin/activate
- pip3 install -U pip
- pip3 install -r ../full-requirements.txt
- pip3 install -U -r ../full-requirements.txt
- pip3 install pdoc3
- pdoc --html .
# Create archive
- tar cf html.tar ./html
@@ -124,6 +175,7 @@ build-developers-doc:
- doc/dev/documentation.html
- doc/dev/html.tar
- doc/dev/images/
allow_failure: true
build-linux:
stage: build
@@ -148,6 +200,8 @@ build-linux:
# Setup virtual env
- python3 -m venv linux-venv
- . linux-venv/bin/activate
- pip3 install -U pip
- pip3 install -r ../requirements.txt
- pip3 install -U -r ../requirements.txt
# Run Pyinstaller
- pyinstaller -y --paths linux-venv/lib/python3.8/site-packages ../src/pamhyr.py
@@ -159,6 +213,7 @@ build-linux:
- cp -v ../VERSION pamhyr/pamhyr/_internal/
- cp -v ../AUTHORS pamhyr/pamhyr/_internal/
- cp -v ../LICENSE pamhyr/pamhyr/_internal/
- cp -v ../src/motd.txt pamhyr/pamhyr/_internal/
# Copy mage
- mkdir -p pamhyr/mage
- cp -v ../mage-linux/* pamhyr/mage/
@@ -218,6 +273,7 @@ build-windows:
- mkdir windows
- cd windows
# Setup env
- python -m pip install -r ..\requirements.txt
- python -m pip install -U -r ..\requirements.txt
# Run Pyinstaller
- pyinstaller -y ..\src\pamhyr.py
@@ -230,36 +286,6 @@ build-windows:
paths:
- windows/pamhyr
#########
# TESTS #
#########
# test-windows:
# stage: test
# tags:
# - wine
# needs:
# - job: set-version
# artifacts: true
# - job: build-windows
# artifacts: true
# script:
# - cd windows\pamhyr
# - pamhyr\pamhyr.exe hello
# test-linux:
# stage: test
# tags:
# - linux
# needs:
# - job: set-version
# artifacts: true
# - job: build-linux
# artifacts: true
# script:
# - cd linux/pamhyr
# - ./Pamhyr2 hello
############
# PACKAGES #
############
Binary file not shown.

After

Width:  |  Height:  |  Size: 23 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 150 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 22 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 8.4 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 15 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 14 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 5.3 KiB

File diff suppressed because it is too large. Load diff
+434
View File
@@ -0,0 +1,434 @@
1 0 0 8 31090.0000
31090.0000 0.0000 23.7300
31090.0000 2.5000 20.2800 rg
31090.0000 2.5000 19.2900
31090.0000 4.4000 19.2900
31090.0000 10.7000 19.5900
31090.0000 10.7000 20.5900 rd
31090.0000 14.5000 21.7100
31090.0000 17.6000 23.7300
999.9990 999.9990 999.9990
2 0 0 8 31165.0000
31165.0000 0.0000 22.8400
31165.0000 2.5000 20.1300 rg
31165.0000 2.5000 19.2300
31165.0000 4.4000 19.2300
31165.0000 10.7000 19.5300
31165.0000 10.7000 20.5300 rd
31165.0000 14.5000 21.7500
31165.0000 17.6000 23.5000
999.9990 999.9990 999.9990
3 0 0 10 31425.0000
31425.0000 -1.0000 22.8000
31425.0000 0.0000 21.9000
31425.0000 1.7000 20.3600 rg
31425.0000 2.5000 19.6300
31425.0000 2.5000 19.0300
31425.0000 4.4000 19.0300
31425.0000 10.7000 19.3300
31425.0000 10.7000 20.3300 rd
31425.0000 14.5000 21.9200
31425.0000 17.6000 23.4200
999.9990 999.9990 999.9990
4 0 0 10 31590.0000
31590.0000 -1.0000 22.7500
31590.0000 0.0000 22.0500
31590.0000 1.6700 20.2200 rg
31590.0000 2.5000 19.3100
31590.0000 2.5000 18.9000
31590.0000 4.4000 18.9000
31590.0000 10.7000 19.2000
31590.0000 10.7000 20.2000 rd
31590.0000 14.5000 22.0200
31590.0000 17.6000 23.2500
999.9990 999.9990 999.9990
5 0 0 10 31780.0000
31780.0000 -1.0000 22.7000
31780.0000 0.0000 22.0500
31780.0000 1.6000 20.0500 rg
31780.0000 2.5000 18.9400
31780.0000 2.5000 18.7500
31780.0000 4.4000 18.7500
31780.0000 10.7000 19.0500
31780.0000 10.7000 20.0500 rd
31780.0000 14.5000 22.1400
31780.0000 17.6000 23.0000
999.9990 999.9990 999.9990
6 0 0 9 31874.0000
31874.0000 0.0000 22.9900
31874.0000 0.0000 22.6900
31874.0000 2.5000 19.6900 rg
31874.0000 2.5000 18.6900
31874.0000 4.4000 18.6900
31874.0000 10.7000 18.9900
31874.0000 10.7000 19.9900 rd
31874.0000 14.5000 22.1900
31874.0000 17.6000 22.9900
999.9990 999.9990 999.9990
7 0 0 7 31875.0000 PontRD101m
31875.0000 0.5400 21.8900
31875.0000 2.5000 19.6900 rg
31875.0000 2.5000 18.6900
31875.0000 4.4000 18.6900
31875.0000 10.7000 18.9900
31875.0000 10.7000 19.9900 rd
31875.0000 13.8900 21.8900
999.9990 999.9990 999.9990
8 0 0 7 31885.0000 PontRD101v
31885.0000 0.5400 21.8900
31885.0000 2.5000 19.6900 rg
31885.0000 2.5000 18.6900
31885.0000 4.4000 18.6900
31885.0000 10.7000 18.9900
31885.0000 10.7000 19.9900 rd
31885.0000 13.8900 21.8900
999.9990 999.9990 999.9990
9 0 0 9 31886.0000
31886.0000 0.0000 22.9900
31886.0000 0.0000 22.6900
31886.0000 2.5000 19.6900 rg
31886.0000 2.5000 18.6900
31886.0000 4.4000 18.6900
31886.0000 10.7000 18.9900
31886.0000 10.7000 19.9900 rd
31886.0000 14.5000 22.1900
31886.0000 17.6000 22.9900
999.9990 999.9990 999.9990
10 0 0 13 32015.0000
32015.0000 0.0000 22.6400
32015.0000 0.2000 22.2200
32015.0000 1.3000 20.9900
32015.0000 2.0000 19.8800 rg
32015.0000 3.0000 18.6600
32015.0000 3.0000 18.5800
32015.0000 5.5000 18.5800
32015.0000 10.0000 18.9100
32015.0000 11.5000 19.2500
32015.0000 11.5000 20.0800 rd
32015.0000 15.1000 22.0600
32015.0000 17.6000 22.5100
32015.0000 18.0000 22.8100
999.9990 999.9990 999.9990
11 0 0 13 32205.0000
32205.0000 0.0000 22.3300
32205.0000 0.2000 22.1800
32205.0000 1.5000 21.0600
32205.0000 2.5000 19.7700 rg
32205.0000 3.6000 18.5100
32205.0000 3.6000 18.4500
32205.0000 6.9000 18.4500
32205.0000 8.8000 19.0500
32205.0000 12.5000 19.5700 rd
32205.0000 12.5000 20.1800
32205.0000 15.8000 21.9000
32205.0000 18.1000 22.4000
32205.0000 18.6000 22.5500
999.9990 999.9990 999.9990
12 0 0 13 32460.0000
32460.0000 0.0000 22.1500
32460.0000 0.3000 22.1300
32460.0000 1.8000 21.1600
32460.0000 3.0000 19.6400 rg
32460.0000 4.4000 18.3100
32460.0000 4.4000 18.2800
32460.0000 8.7000 18.2800
32460.0000 10.4000 19.2400 rd
32460.0000 13.7000 19.9900
32460.0000 13.7000 20.3300
32460.0000 16.8000 21.6800
32460.0000 18.9000 22.2500
32460.0000 19.3000 22.5500
999.9990 999.9990 999.9990
13 0 0 13 32655.0000
32655.0000 0.0000 22.4500
32655.0000 0.3000 22.0900
32655.0000 2.1000 21.2300
32655.0000 3.4000 19.5300 rg
32655.0000 5.0000 18.1600
32655.0000 5.0000 18.1400
32655.0000 10.1000 18.1400
32655.0000 11.6000 19.3900 rd
32655.0000 14.7000 20.3100
32655.0000 14.7000 20.4300
32655.0000 17.5000 21.5200
32655.0000 19.4000 22.1400
32655.0000 19.8000 22.4500
999.9990 999.9990 999.9990
14 0 0 13 32755.0000
32755.0000 0.0000 22.4500
32755.0000 0.3000 22.0900
32755.0000 2.1000 21.2300
32755.0000 3.4000 19.5300 rg
32755.0000 4.0000 18.0500
32755.0000 5.0000 18.0100
32755.0000 10.1000 18.0100
32755.0000 13.0000 19.3900 rd
32755.0000 14.7000 20.3100
32755.0000 14.7000 20.4300
32755.0000 17.5000 21.5200
32755.0000 19.4000 22.1400
32755.0000 19.8000 22.4500
999.9990 999.9990 999.9990
15 0 0 10 32770.0000 PontA2
32770.0000 0.0000 25.0000
32770.0000 1.9000 25.0000
32770.0000 1.9000 21.2700
32770.0000 3.3000 19.4700 rg
32770.0000 5.0000 18.0100
32770.0000 10.5000 18.0100
32770.0000 12.0000 19.4700 rd
32770.0000 19.4000 21.4700
32770.0000 19.4000 25.0000
32770.0000 20.3000 25.0000
999.9990 999.9990 999.9990
16 0 0 10 32800.0000 PontA2
32800.0000 0.0000 25.0000
32800.0000 1.9000 25.0000
32800.0000 1.9000 21.2700
32800.0000 3.3000 19.4700 rg
32800.0000 5.0000 18.0100
32800.0000 10.5000 18.0100
32800.0000 12.0000 19.4700 rd
32800.0000 19.4000 21.4700
32800.0000 19.4000 25.0000
32800.0000 20.3000 25.0000
999.9990 999.9990 999.9990
17 0 0 13 32820.0000
32820.0000 0.0000 22.4500
32820.0000 0.3000 22.0900
32820.0000 2.1000 21.2300
32820.0000 3.4000 19.5300 rg
32820.0000 4.0000 18.0500
32820.0000 5.0000 18.0100
32820.0000 10.1000 18.0100
32820.0000 13.0000 19.3900 rd
32820.0000 14.7000 20.3100
32820.0000 14.7000 20.4300
32820.0000 17.5000 21.5200
32820.0000 19.4000 22.1400
32820.0000 19.8000 22.4500
999.9990 999.9990 999.9990
18 0 0 11 32915.0000
32915.0000 0.0000 22.5500
32915.0000 0.4000 21.9500
32915.0000 1.7000 21.2200
32915.0000 2.4000 20.9700
32915.0000 3.8000 19.4100 rg
32915.0000 3.9000 19.3100
32915.0000 5.8000 18.0100
32915.0000 11.0000 18.0100
32915.0000 12.4000 19.3400 rd
32915.0000 19.4000 21.9900
32915.0000 19.7000 22.5500
999.9990 999.9990 999.9990
19 0 0 10 33140.0000
33140.0000 0.4000 21.7700
33140.0000 1.9000 20.7600
33140.0000 2.7000 20.5100
33140.0000 4.2000 19.1500 rg
33140.0000 4.3000 19.0600
33140.0000 6.4000 17.9200
33140.0000 11.3000 17.9200
33140.0000 12.6000 19.1400 rd
33140.0000 18.9000 21.8700
33140.0000 19.2000 22.1800
999.9990 999.9990 999.9990
20 0 0 10 33425.0000
33425.0000 0.0000 21.6800
33425.0000 0.5000 21.5300
33425.0000 2.2000 20.1600
33425.0000 3.0000 19.9400
33425.0000 4.8000 18.8200 rg
33425.0000 4.9000 18.7500
33425.0000 7.2000 17.8000
33425.0000 11.7000 17.8000
33425.0000 12.8000 18.8800 rd
33425.0000 18.2000 21.6800
999.9990 999.9990 999.9990
21 0 0 10 33685.0000
33685.0000 0.0000 21.5300
33685.0000 0.5000 21.3200
33685.0000 2.4000 19.6200
33685.0000 3.3000 19.4200
33685.0000 5.2000 18.5300 rg
33685.0000 5.4000 18.4700
33685.0000 7.9000 17.7000
33685.0000 12.0000 17.7000
33685.0000 12.9000 18.6500 rd
33685.0000 17.6000 21.5300
999.9990 999.9990 999.9990
22 0 0 9 33920.0000
33920.0000 0.6000 20.9800
33920.0000 2.6000 19.1300
33920.0000 3.6000 18.9400
33920.0000 5.7000 18.2600 rg
33920.0000 5.8000 18.2200
33920.0000 8.5000 17.6000
33920.0000 12.3000 17.6000
33920.0000 13.1000 18.4400 rd
33920.0000 17.0000 21.1300
999.9990 999.9990 999.9990
23 0 0 9 34095.0000
34095.0000 0.6000 20.9500
34095.0000 2.7000 18.7700
34095.0000 3.8000 18.5900
34095.0000 6.0000 18.0600 rg
34095.0000 6.1000 18.0300
34095.0000 9.0000 17.5300
34095.0000 12.5000 17.5300
34095.0000 13.2000 18.2800 rd
34095.0000 16.6000 20.9800
999.9990 999.9990 999.9990
24 0 0 10 34295.0000
34295.0000 0.0000 21.3600
34295.0000 0.6000 20.8300
34295.0000 2.4000 18.3500
34295.0000 3.5000 18.1900
34295.0000 5.8000 17.8300 rg
34295.0000 6.5000 17.3400
34295.0000 9.5000 17.3600
34295.0000 12.8000 17.4100
34295.0000 14.0000 18.1000 rd
34295.0000 16.1000 21.3600
999.9990 999.9990 999.9990
25 0 0 10 34334.0000 amontPontThivencel
34334.0000 0.0000 21.3600
34334.0000 0.6000 20.8300
34334.0000 2.4000 18.3500
34334.0000 3.5000 18.1900
34334.0000 5.8000 17.8300 rg
34334.0000 6.5000 17.3400
34334.0000 9.5000 17.3600
34334.0000 12.8000 17.4100
34334.0000 14.0000 18.1000 rd
34334.0000 16.1000 21.1300
999.9990 999.9990 999.9990
26 0 0 7 34335.0000 PontmThivencelle
34335.0000 2.9000 20.7400
34335.0000 2.9000 18.2900 rg
34335.0000 6.4000 17.7900
34335.0000 9.6000 17.2600
34335.0000 12.8000 17.2600
34335.0000 12.8000 18.3100 rd
34335.0000 12.8000 20.7400
999.9990 999.9990 999.9990
27 0 0 7 34343.0000 PontvThivencelle
34343.0000 2.9000 20.7400
34343.0000 2.9000 18.2900 rg
34343.0000 6.4000 17.7900
34343.0000 9.6000 17.2600
34343.0000 12.8000 17.2600
34343.0000 12.8000 18.2900 rd
34343.0000 12.8000 20.7400
999.9990 999.9990 999.9990
28 0 0 10 34345.0000 avalPontThivencell
34345.0000 0.0000 21.3600
34345.0000 0.6000 20.8300
34345.0000 2.4000 18.3500
34345.0000 3.5000 18.1900
34345.0000 5.8000 17.8300 rg
34345.0000 6.5000 17.3400
34345.0000 9.5000 17.2600
34345.0000 12.8000 17.4100
34345.0000 14.0000 18.1000 rd
34345.0000 16.1000 21.1300
999.9990 999.9990 999.9990
29 0 0 7 34405.0000
34405.0000 0.0000 21.1400
34405.0000 2.9000 17.8400 rg
34405.0000 6.4000 17.4400
34405.0000 9.6000 17.2000
34405.0000 12.8000 17.2200
34405.0000 13.9000 17.8400 rd
34405.0000 19.0000 21.0000
999.9990 999.9990 999.9990
30 0 0 7 34535.0000
34535.0000 0.0000 20.5900
34535.0000 2.9000 17.7400 rg
34535.0000 6.4000 17.2400
34535.0000 9.6000 17.0300
34535.0000 12.8000 17.0300
34535.0000 13.5000 17.7400 rd
34535.0000 15.8000 20.6000
999.9990 999.9990 999.9990
31 0 0 7 34705.0000
34705.0000 0.0000 20.5000
34705.0000 2.9000 17.6100 rg
34705.0000 6.4000 17.1100
34705.0000 9.6000 16.8500
34705.0000 12.8000 16.8500
34705.0000 14.0000 18.0000 rd
34705.0000 16.2000 20.5300
999.9990 999.9990 999.9990
32 0 0 7 34825.0000
34825.0000 0.0000 20.6200
34825.0000 2.9000 17.5200 rg
34825.0000 6.4000 17.0200
34825.0000 9.6000 16.7100
34825.0000 12.8000 16.7100
34825.0000 13.5000 17.5200 rd
34825.0000 16.2000 20.5000
999.9990 999.9990 999.9990
33 0 0 7 34995.0000
34995.0000 0.0000 20.3900
34995.0000 2.9000 17.3900 rg
34995.0000 6.4000 16.8900
34995.0000 9.6000 16.5300
34995.0000 12.8000 16.5300
34995.0000 13.6000 17.4300 rd
34995.0000 16.2000 20.3200
999.9990 999.9990 999.9990
34 0 0 7 35135.0000
35135.0000 0.0000 20.1600
35135.0000 2.9000 17.2900 rg
35135.0000 6.4000 16.7900
35135.0000 9.6000 16.4300
35135.0000 12.8000 16.4300
35135.0000 13.9000 17.3700 rd
35135.0000 16.2000 20.0600
999.9990 999.9990 999.9990
35 0 0 7 35360.0000
35360.0000 0.0000 19.8600
35360.0000 2.9000 17.1200 rg
35360.0000 6.4000 16.6200
35360.0000 9.6000 16.2600
35360.0000 12.8000 16.2600
35360.0000 13.7000 17.1200 rd
35360.0000 16.2000 19.4600
999.9990 999.9990 999.9990
36 0 0 7 35580.0000
35580.0000 0.0000 20.0000
35580.0000 2.9000 16.9500 rg
35580.0000 3.0000 16.0900
35580.0000 11.5000 16.0500
35580.0000 20.0000 16.0900
35580.0000 21.0000 17.0300 rd
35580.0000 23.0000 19.5000
999.9990 999.9990 999.9990
37 0 0 7 35920.0000
35920.0000 0.0000 20.0000
35920.0000 2.2000 16.0200 rg
35920.0000 3.0000 14.5500
35920.0000 11.5000 14.5000
35920.0000 20.0000 14.5500
35920.0000 20.9000 16.0200 rd
35920.0000 23.0000 19.5000
999.9990 999.9990 999.9990
38 0 0 7 36270.0000 S1
36270.0000 0.0000 20.0000
36270.0000 1.8000 17.0100 rg
36270.0000 3.0000 15.0800
36270.0000 11.5000 15.0500
36270.0000 20.0000 15.0800
36270.0000 21.3000 17.0100 rd
36270.0000 23.0000 19.5000
999.9990 999.9990 999.9990
39 0 0 7 36271.0000 S1bis
36271.0000 0.0000 20.0000
36271.0000 1.8000 17.0100 rg
36271.0000 3.0000 13.0800
36271.0000 11.5000 13.0500
36271.0000 20.0000 13.0800
36271.0000 21.3000 17.0100 rd
36271.0000 23.0000 19.5000
999.9990 999.9990 999.9990
+2
View File
@@ -0,0 +1,2 @@
0:00:00:00 15
0:00:01:00 15
+25
View File
@@ -0,0 +1,25 @@
0:00:00:00 5,53
0:11:37:00 5,72
0:16:17:00 5,82
0:22:57:00 5,92
1:00:00:00 6,6
1:00:57:00 8,04
1:02:29:00 14,3
1:04:38:00 24,4
1:08:48:00 40
1:11:27:00 46,3
1:13:47:00 48,8
1:16:07:00 47,9
1:17:17:00 44,8
1:20:23:00 27,8
1:21:57:00 24,3
2:01:15:00 19,2
2:05:15:00 15,8
2:11:25:00 11,5
2:11:57:00 11,1
2:13:17:00 10,5
2:19:07:00 9,84
2:23:59:00 8,43
3:04:51:00 7,02
3:09:43:00 5,61
3:14:35:00 5,53
+25
View File
@@ -0,0 +1,25 @@
**************** SINGULARITES **********************
*BieSec Nb Longueur Cote dev Cote charg Coef deb Impression
*===---===----------==========----------==========-----------------------------
*****************************************************
* Bief n° 1
*****************************************************
* Ouvrage au pk 36270.0 seuil_aval
*SeuilR type D
D 1 36270.00 18.250 15.500 0.000 0.400 99999.000 seuil palplanches
* Ouvrage au pk 31875.0 PontRS101
*OrificeR type O
O 1 31875.00 9.660 18.710 21.890 0.400 9999.000
*SeuilR type D
D 1 31875.00 14.000 22.890 0.000 0.400 99999.000
* Ouvrage au pk 34335.0 PontThivencelle
*SeuilR type D
D 1 34335.00 13.000 21.740 0.000 0.400 99999.000
*OrificeR type O
O 1 34335.00 9.070 17.280 20.740 0.400 9999.000
*****************************************************
* Bief n° 3
*****************************************************
* Ouvrage au pk 0.0 Orifice
*OrificeR type O
*O 3 0.000 6.000 19.000 19.001 0.400 20.000 Clapet_001
+1
View File
@@ -0,0 +1 @@
$Clapet_001 clapet
+2
View File
@@ -0,0 +1,2 @@
0:00:00:00 4
0:00:01:00 4
+24
View File
@@ -0,0 +1,24 @@
16,03 0
16,13 1,29
16,23 3,64
16,33 6,7
16,43 10,31
16,53 14,41
16,63 18,94
16,73 23,87
16,83 29,16
16,93 34,79
17,03 40,75
17,13 47,01
17,23 53,57
17,33 60,4
17,43 67,5
17,53 74,86
17,63 82,47
17,73 90,33
17,83 98,41
17,93 106,73
18,03 115,26
18,13 124,01
18,23 132,98
18,33 142,14
+9
View File
@@ -0,0 +1,9 @@
0:00:00:00 4
0:01:16:48 4
0:16:00:00 50
1:11:46:28 36,8
2:07:27:22 23,3
5:04:24:58 13,8
10:03:00:06 9,38
20:02:19:58 7,29
30:01:54:37 5,97
+10
View File
@@ -0,0 +1,10 @@
0:00:00:00 4
0:01:49:48 4
0:16:00:00 35
1:11:25:48 25
2:07:30:00 16,4
5:03:55:12 9,66
10:03:00:00 6,56
27:00:54:00 4,91
30:01:48:36 4
31:06:00:00 4
Binary file not shown.

After

Width:  |  Height:  |  Size: 16 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 69 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 57 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 48 KiB

BIN
View File
Binary file not shown.

After

Width:  |  Height:  |  Size: 7.7 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 71 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 18 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 23 KiB

+201
View File
@@ -0,0 +1,201 @@
%% LyX 2.0.2 created this file. For more info, see http://www.lyx.org/.
%% Do not edit unless you really know what you are doing.
\documentclass[12pt,french]{article}
\usepackage[T1]{fontenc}
\usepackage[latin9]{inputenc}
\usepackage{geometry}
\geometry{verbose,tmargin=2cm,bmargin=2cm,lmargin=2cm,rmargin=2cm,headheight=2cm,headsep=2cm,footskip=2cm}
\usepackage{textcomp}
\usepackage{graphicx}
\usepackage{hyperref}
\makeatletter
%%%%%%%%%%%%%%%%%%%%%%%%%%%%%% LyX specific LaTeX commands.
\newcommand{\noun}[1]{\textsc{#1}}
%% Because html converters don't know tabularnewline
\providecommand{\tabularnewline}{\\}
\makeatother
\usepackage{babel}
\addto\extrasfrench{%
\providecommand{\og}{\leavevmode\flqq~}%
\providecommand{\fg}{\ifdim\lastskip>\z@\unskip\fi~\frqq}%
}
\begin{document}
\includegraphics[width=5cm]{Logo-INRAE_Transparent.png}
~
~
~
\begin{center}
Tutorial for Pamhyr2
January 2024
\par\end{center}
~
~
\begin{center}
\textbf{\LARGE 1D Modeling of the Hogneau River (Nord, France)}
using Pamhyr2
\par\end{center}{\LARGE \par}
~
~
\begin{center}
{\large INRAE Lyon-Grenoble Auvergne-Rhône-Alpes}
\par\end{center}{\large \par}
\begin{center}
Riverly, river hydraulics
\par\end{center}
~
~
~
~
~
~
\begin{center}
\begin{tabular}{lll}
Autors : & Pierre-Antoine Rouby & pierre-antoine.rouby@inrae.fr\tabularnewline
& Théophile TERRAZ & theophile.terraz@inrae.fr\tabularnewline
\end{tabular}
\par\end{center}
~
\pagebreak{}
\begin{center}
\tableofcontents{}
\par\end{center}
~
\pagebreak{}
\section{Introduction}
TODO
\pagebreak{}
\section{Install Pamhyr2}
Pamhyr2 can be downloaded from \url{https://gitlab.irstea.fr/theophile.terraz/pamhyr}.
\begin{center}
\includegraphics[width=15cm]{dl.png}
\par\end{center}
Use the GNU Linux or the Windows download button depending on your system. On windows, launch the installer. On Linux, unpack the archive and launch Pamhyr2.
\section{Create your first study}
On the main windows, click on \texttt{[Files] => [New Study]} to create a new study.
Click on \texttt{[River Network] => [Edit River Network]} to create the reaches of your river. In this window, you must define an oriented graph that represents the reaches of your river network: the edges are the reaches and the nodes are either upstream boundary conditions, downstream boundary conditions or junctions.
Press the \includegraphics[width=0.5cm]{gtk_add.png} to enter the \textit{add} mode. Create two nodes by clicking in the grey zone of the window, and create a link by clicking again on each node. press \includegraphics[width=0.5cm]{gtk_add.png} again to exit the \textit{add} mode. You created your first reach, with an upstream node and a downstream node. In the lower part of the \textit{Edit River Network} window you can rename the nodes and the reaches. As the reach you created is automaticaly selected, all the next steps will apply to this reach. The window should look like that:
\begin{center}
\includegraphics[width=15cm]{network.png}
\par\end{center}
Close the \textit{Edit River Network} window.
\section{Edit the river geometry}
Click on \texttt{[Geometry] => [Edit Geometry]} to define the geometry of the selected bief. Click on the \texttt{[Import]} button and select the file \texttt{Data/Bief\_1.st}. You should see:
\begin{center}
\includegraphics[width=15cm]{Geo.png}
\par\end{center}
On the left panel is a list of all the cross sections with their name and longitudinal abscisa. In the top left plot you can see the top view of the river, on the top right panel the longitudinal cross-section of the river and in the bottom plot you can see the selected cross-section (blue) along with the next one (dashed purple) and previous one (dashed black). You can move in the section list using the slider at the very right of the window.
You can edit the selected cross section by clicking on the \includegraphics[width=0.5cm]{"../../../src/View/ui/ressources/edit.png"} icon.
select the cross section named \textit{PontRD101m} and open the edition window. You should see:
\begin{center}
\includegraphics[width=15cm]{editsect.png}
\par\end{center}
On the left panel is the list of all the points of the section, with their coordinates, their name and their transversal absisa. The Z coordinate of the highest point is written in blue and the lowest in red. Points can have a name. If a point with the same name exists in every section in a reach, it forms a longitudinal line. For example, here we have \textit{rg} and \textit{rd} which represent the left bank and the right bank of the main chanel.
On the plot is a projection of the cross section. You can use \texttt{[ctrl + click]} to select a point in the plot. For now we don't have to change anything. You can close the cross section edition window and the geometry edition window.
\section{Edit the boundary conditions}
From the main window, click on \texttt{[Hydraulics] => [Boundary conditions and punctual contributions]}.You should see:
\begin{center}
\includegraphics[width=15cm]{boundary.png}
\par\end{center}
Use the \textit{add} button on the top left of the window to add a liquid boundary condition.
On the new line, click to select the whole line, double click to select the cell.
Select the \textit{Type} cell to give a name to the boundary condition. here, we will define the flow discharge mesured during the february 2002 flood. You can name this boundary condition "flood2002".
Select the \textit{Type} cell and use the combo box to put a Q(t) law.
Select the \textit{Node} cell and atribute this condition to the upstream node.
Names of the nodes are recalled in network in the right panel.
Now select the whole line and click on the edit button \includegraphics[width=0.5cm]{"../../../src/View/ui/ressources/edit.png"}. You opened the \textit{Edit Boundary Conditions} window.
On a text editor, open the \texttt{Data/Fevrier\_2002.txt} file. Copy the content of the file (for example with \textit{ctrl+a ctrl+c}) and paste it in the left panel of the \textit{Edit Boundary Conditions} window with \textit{ctrl+v}. You can now see the flow discharge curve:
\begin{center}
\includegraphics[width=15cm]{fev2002.png}
\par\end{center}
Close this window. Go back on the \textit{Boundary Conditions} window. Add a new line, give it a name, give it the textit{Z(T)} type and associate it to the downstream node of the network. Open the \textit{Edit Boundary Conditions} window (\includegraphics[width=0.5cm]{"../../../src/View/ui/ressources/edit.png"}). Add two lines. In the first one, enter time: 0.00.00 and Z: 15.000.
On the second one, time: 1.00.00 and Z: 15.000. It creates a constant downstream water elevation. For the computaion, Mage will extrapolate continuously the water elevation, that's why we only need to define one hour.
You can close the the \textit{Edit Boundary Conditions} and the \textit{Boundary Conditions} window.
\section{Create initial conditions}
From the main window, click on \texttt{[Hydraulics] => [Initial conditions]}.
Click on \texttt{Generate minimal height} and enter a height of 2 meters in the pop-up window to generate an initial water height condition that verify the folowing conditions: a water height of at least 2 meters and a decreasing water surface elevation from upstream to downstream. You should see:
\begin{center}
\includegraphics[width=15cm]{ic.png}
\par\end{center}
Close the \textit{Initial conditions} window.
\section{Edit friction coefficients}
From the main window, click on \texttt{[Hydraulics] => [Edit friction]}.
You fist have to define sets of Strickler coefficients. Click on \includegraphics[width=0.5cm]{"../../../src/View/ui/ressources/edit.png"} to open the \textit{Strickler} window. Here you can create couples of Strickler coefficients, the first one for the minor bed, the second one for the medium bed. Click on \textit{add} four times to create four new couple. Give them the folowing values:
\begin{center}
\includegraphics[width=15cm]{K.png}
\par\end{center}
Close the \textit{Strickler} window. On the \textit{Edit friction} window, add four lines with the button \includegraphics[width=0.5cm]{gtk_add.png} to create fous friction zones. Each zone is defined by a \textit{begin} and \textit{end} PK and a \textit{begin} and \textit{end} Strickler couple. The strickler coefficient couples inside a zone are interpolated from the \textit{begin} and \textit{end} couples. In our case, we will use constant coefficients per zone. Set the zones as follow:
\begin{center}
\includegraphics[width=15cm]{frictions.png}
\par\end{center}
\pagebreak{}
\end{document}
+12
View File
@@ -0,0 +1,12 @@
PyQt5==5.15.4
PyQt5-Qt5==5.15.2
PyQt5-sip==12.12.2
PyQtWebEngine==5.15.6
QScintilla>=2.14.1
pyqtgraph>=0.12.1
matplotlib>=3.4.1
numpy>=1.24.2
colorama>=0.4.3
pyinstaller>=5.11.0
shapely>=2.0.1
lxml>=4.9.3
+2
View File
@@ -25,6 +25,7 @@ echo " *** SETUP ENV"
python3 -m venv linux-pamhyr-venv
. linux-pamhyr-venv/bin/activate
pip3 install -U pip
pip3 install -r ../requirements.txt
echo " *** PYINSTALL"
@@ -58,6 +59,7 @@ VERSION=$(cat ../VERSION)
echo "local-build-$VERSION" > dist/pamhyr/_internal/VERSION
cp ../AUTHORS dist/pamhyr/_internal/
cp ../LICENSE dist/pamhyr/_internal/
cp ../src/motd.txt dist/pamhyr/_internal/
mkdir -p dist/pamhyr/mage/
cp ../mage/mage dist/pamhyr/mage/
+1
View File
@@ -36,6 +36,7 @@ rem Information
copy /y ..\VERSION pamhyr\pamhyr\_internal
copy /y ..\AUTHORS pamhyr\pamhyr\_internal
copy /y ..\LICENSE pamhyr\pamhyr\_internal
copy /y ..\src\motd.txt pamhyr\pamhyr\_internal
rem MAGE
mkdir pamhyr\mage
+3 -1
View File
@@ -1,10 +1,12 @@
PyQt5==5.15.4
PyQt5-Qt5==5.15.2
PyQt5-sip==12.12.2
PyQtWebEngine==5.15.6
#PyQtWebEngine==5.15.6
QScintilla>=2.14.1
pyqtgraph>=0.12.1
matplotlib>=3.4.1
numpy>=1.24.2
colorama>=0.4.3
pyinstaller>=5.11.0
shapely>=2.0.1
lxml>=4.9.3
+8
View File
@@ -20,12 +20,14 @@ from enum import Enum
from Model.Except import NotImplementedMethodeError
class STATUS(Enum):
UNKNOWN = -1
OK = 0
WARNING = 1
ERROR = 2
class AbstractModelChecker(object):
def __init__(self):
self._name = ""
@@ -62,6 +64,12 @@ class AbstractModelChecker(object):
# Abstract function
def _run(self, study):
from Model.Study import Study
thread_study = Study.open(study._filename)
self.run(thread_study)
def run(self, study):
"""Run checker function
+6 -3
View File
@@ -27,8 +27,9 @@ from Checker.Checker import AbstractModelChecker, STATUS
_translate = QCoreApplication.translate
class MageNetworkGraphChecker(AbstractModelChecker):
def __init__(self, connectivity = True):
def __init__(self, connectivity=True):
super(MageNetworkGraphChecker, self).__init__()
self._mode_conn = connectivity
@@ -38,8 +39,10 @@ class MageNetworkGraphChecker(AbstractModelChecker):
else:
mode = "cycle"
self._name = _translate("Checker", f"Mage network graph {mode} checker")
self._description = _translate("Checker", "Check if the network graph is valid")
self._name = _translate(
"Checker", f"Mage network graph {mode} checker")
self._description = _translate(
"Checker", "Check if the network graph is valid")
@timer
def _connectivity(self, summary, status, graph):
+6 -2
View File
@@ -30,7 +30,8 @@ class StudyNetworkReachChecker(AbstractModelChecker):
super(StudyNetworkReachChecker, self).__init__()
self._name = _translate("Checker", "Study network reach checker")
self._description = _translate("Checker", "Check if exists at least one reach for study")
self._description = _translate(
"Checker", "Check if exists at least one reach for study")
def run(self, study):
if study is None:
@@ -54,12 +55,14 @@ class StudyNetworkReachChecker(AbstractModelChecker):
self._status = STATUS.OK
return True
class StudyGeometryChecker(AbstractModelChecker):
def __init__(self):
super(StudyGeometryChecker, self).__init__()
self._name = _translate("Checker", "Study geometry checker")
self._description = _translate("Checker", "Check if exists geometry for each reach of study")
self._description = _translate(
"Checker", "Check if exists geometry for each reach of study")
self._reachs = []
def run(self, study):
@@ -128,6 +131,7 @@ class DummyWARNING(AbstractModelChecker):
return True
class DummyERROR(AbstractModelChecker):
def __init__(self):
super(DummyERROR, self).__init__()
@@ -25,12 +25,13 @@ from Model.Except import NotImplementedMethodeError
logger = logging.getLogger()
class BoundaryCondition(SQLSubModel):
_sub_classes = []
_id_cnt = 0
def __init__(self, id:int = -1, name:str = "",
status = None):
def __init__(self, id: int = -1, name: str = "",
status=None):
super(BoundaryCondition, self).__init__()
self._status = status
@@ -50,7 +51,7 @@ class BoundaryCondition(SQLSubModel):
BoundaryCondition._id_cnt = max(BoundaryCondition._id_cnt + 1, self.id)
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
execute("""
CREATE TABLE boundary_condition(
id INTEGER NOT NULL PRIMARY KEY,
@@ -76,7 +77,7 @@ class BoundaryCondition(SQLSubModel):
return cls._create_submodel(execute)
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
return True
@classmethod
@@ -101,7 +102,7 @@ class BoundaryCondition(SQLSubModel):
return res
@classmethod
def _sql_load(cls, execute, data = None):
def _db_load(cls, execute, data=None):
new = []
tab = data["tab"]
@@ -115,9 +116,9 @@ class BoundaryCondition(SQLSubModel):
t = row[2]
ctor = cls._get_ctor_from_type(t)
bc = ctor(
id = row[0],
name = row[1],
status = data['status']
id=row[0],
name=row[1],
status=data['status']
)
bc.node = None
@@ -143,7 +144,7 @@ class BoundaryCondition(SQLSubModel):
return new
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
tab = data["tab"]
execute(f"DELETE FROM boundary_condition WHERE id = {self.id}")
@@ -155,22 +156,22 @@ class BoundaryCondition(SQLSubModel):
sql = (
"INSERT INTO " +
"boundary_condition(id, name, type, tab, node) "+
"boundary_condition(id, name, type, tab, node) " +
"VALUES (" +
f"{self.id}, '{self._sql_format(self._name)}', " +
f"'{self._sql_format(self._type)}', '{tab}', {node}" +
f"{self.id}, '{self._db_format(self._name)}', " +
f"'{self._db_format(self._type)}', '{tab}', {node}" +
")"
)
execute(sql)
ind = 0
for d in self._data:
data0 = self._sql_format(str(d[0]))
data1 = self._sql_format(str(d[1]))
data0 = self._db_format(str(d[0]))
data1 = self._db_format(str(d[1]))
sql = (
"INSERT INTO " +
"boundary_condition_data(ind, data0, data1, bc) "+
"boundary_condition_data(ind, data0, data1, bc) " +
f"VALUES ({ind}, '{data0}', {data1}, {self.id})"
)
execute(sql)
@@ -187,8 +188,13 @@ class BoundaryCondition(SQLSubModel):
@classmethod
def time_convert(cls, data):
if type(data) == str and data.count(":") == 3:
return old_pamhyr_date_to_timestamp(data)
if type(data) is str:
if data.count(":") == 3:
return old_pamhyr_date_to_timestamp(data)
if data.count(":") == 2:
return old_pamhyr_date_to_timestamp("00:" + data)
if data.count(".") == 1:
return round(float(data))
return int(data)
@@ -246,7 +252,7 @@ class BoundaryCondition(SQLSubModel):
new_1 = self._default_1
if len(header) != 0:
for i in [0,1]:
for i in [0, 1]:
for j in range(len(header)):
if self._header[i] == header[j]:
if i == 0:
@@ -259,13 +265,13 @@ class BoundaryCondition(SQLSubModel):
return (new_0, new_1)
def add(self, index:int):
def add(self, index: int):
value = (self._default_0, self._default_1)
self._data.insert(index, value)
self._status.modified()
return value
def insert(self, index:int, value):
def insert(self, index: int, value):
self._data.insert(index, value)
self._status.modified()
@@ -301,11 +307,10 @@ class BoundaryCondition(SQLSubModel):
return self.data[index]
def get_range(self, _range):
l = []
lst = []
for r in _range:
l.append(r)
return l
lst.append(r)
return lst
def _set_i_c_v(self, index, column, value):
v = list(self._data[index])
@@ -313,22 +318,22 @@ class BoundaryCondition(SQLSubModel):
self._data[index] = tuple(v)
self._status.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)
def set_i_1(self, index:int, value):
def set_i_1(self, index: int, value):
self._set_i_c_v(index, 1, value)
@timer
def convert(self, cls):
new = cls(name = self.name, status = self._status)
new = cls(name=self.name, status=self._status)
new.node = self.node
for i, _ in enumerate(self.data):
new.add(i)
for i in [0,1]:
for j in [0,1]:
for i in [0, 1]:
for j in [0, 1]:
if self._header[i] == new.header[j]:
for ind, v in self.data:
try:
@@ -30,6 +30,7 @@ from Model.BoundaryCondition.BoundaryConditionTypes import (
Solid,
)
class BoundaryConditionList(PamhyrModelListWithTab):
_tabs_list = ["liquid", "solid", "suspenssion"]
_sub_classes = [
@@ -37,21 +38,21 @@ class BoundaryConditionList(PamhyrModelListWithTab):
]
@classmethod
def _sql_load(cls, execute, data = None):
new = cls(status = data['status'])
def _db_load(cls, execute, data=None):
new = cls(status=data['status'])
if data is None:
data = {}
for tab in new._tabs:
data["tab"] = tab
new._tabs[tab] = BoundaryCondition._sql_load(
new._tabs[tab] = BoundaryCondition._db_load(
execute, data
)
return new
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
execute("DELETE FROM boundary_condition")
if data is None:
@@ -60,7 +61,7 @@ class BoundaryConditionList(PamhyrModelListWithTab):
for tab in self._tabs:
data["tab"] = tab
for bc in self._tabs[tab]:
bc._sql_save(execute, data = data)
bc._db_save(execute, data=data)
return True
@@ -73,18 +74,31 @@ class BoundaryConditionList(PamhyrModelListWithTab):
def __copy__(self):
new = BoundaryConditionList()
for l in self._tabs:
new.tabs[l] = self._tabs[l].copy()
for lst in self._tabs:
new.tabs[lst] = self._tabs[lst].copy()
return new
def __deepcopy__(self):
new = BoundaryConditionList()
for l in self._tabs:
new.tabs[l] = self._tabs[l].deepcopy()
for lst in self._tabs:
new.tabs[lst] = self._tabs[lst].deepcopy()
return new
def copy(self):
return copy(self)
def get_assoc_to_node(self, tab, node):
assoc = list(
filter(
lambda i: i.node is node,
self._tabs[tab]
)
)
if len(assoc) > 0:
return assoc[0]
return None
@@ -22,7 +22,7 @@ from Model.BoundaryCondition.BoundaryCondition import BoundaryCondition
class NotDefined(BoundaryCondition):
def __init__(self, id:int = -1, name:str = "", status = None):
def __init__(self, id: int = -1, name: str = "", status=None):
super(NotDefined, self).__init__(id=id, name=name, status=status)
self._type = "ND"
@@ -32,9 +32,11 @@ class NotDefined(BoundaryCondition):
def _default_0(self):
return 0.0
class PonctualContribution(BoundaryCondition):
def __init__(self, id:int = -1, name:str = "", status = None):
super(PonctualContribution, self).__init__(id=id, name=name, status=status)
def __init__(self, id: int = -1, name: str = "", status=None):
super(PonctualContribution, self).__init__(
id=id, name=name, status=status)
self._type = "PC"
self._header = ["time", "discharge"]
@@ -44,8 +46,9 @@ class PonctualContribution(BoundaryCondition):
def compatibility(cls):
return ["liquid"]
class TimeOverZ(BoundaryCondition):
def __init__(self, id:int = -1, name:str = "", status = None):
def __init__(self, id: int = -1, name: str = "", status=None):
super(TimeOverZ, self).__init__(id=id, name=name, status=status)
self._type = "TZ"
@@ -56,9 +59,11 @@ class TimeOverZ(BoundaryCondition):
def compatibility(cls):
return ["liquid"]
class TimeOverDischarge(BoundaryCondition):
def __init__(self, id:int = -1, name:str = "", status = None):
super(TimeOverDischarge, self).__init__(id=id, name=name, status=status)
def __init__(self, id: int = -1, name: str = "", status=None):
super(TimeOverDischarge, self).__init__(
id=id, name=name, status=status)
self._type = "TD"
self._header = ["time", "discharge"]
@@ -68,8 +73,9 @@ class TimeOverDischarge(BoundaryCondition):
def compatibility(cls):
return ["liquid"]
class ZOverDischarge(BoundaryCondition):
def __init__(self, id:int = -1, name:str = "", status = None):
def __init__(self, id: int = -1, name: str = "", status=None):
super(ZOverDischarge, self).__init__(id=id, name=name, status=status)
self._type = "ZD"
@@ -86,7 +92,7 @@ class ZOverDischarge(BoundaryCondition):
class Solid(BoundaryCondition):
def __init__(self, id:int = -1, name:str = "", status = None):
def __init__(self, id: int = -1, name: str = "", status=None):
super(Solid, self).__init__(id=id, name=name, status=status)
self.d50 = 0.002
+11 -5
View File
@@ -45,8 +45,9 @@ def exception_message_box(exception):
# Custom error #
################
class ExeceptionWithMessageBox(Exception):
def __init__(self, title = "Exeception"):
def __init__(self, title="Exeception"):
self.title = title
def header(self):
@@ -72,7 +73,7 @@ class ExeceptionWithMessageBox(Exception):
class NotImplementedMethodeError(ExeceptionWithMessageBox):
def __init__(self, obj, func):
super(NotImplementedMethodeError, self).__init__(
title = _translate("Exception", "Method not implemented")
title=_translate("Exception", "Method not implemented")
)
self.obj = obj
@@ -81,12 +82,16 @@ class NotImplementedMethodeError(ExeceptionWithMessageBox):
self.alert()
def __str__(self):
class_type = self.obj
if self.obj.__class__ != type:
class_type = self.obj.__class__
return (
_translate("Exception", "Method") +
f" '{self.func.__name__}' " +
_translate("Exception", "not implemented") +
_translate("Exception", "for class") +
f" '{self.obj.__class__ if self.obj.__class__ != type else self.obj}'"
f" '{class_type}'"
)
def header(self):
@@ -104,10 +109,11 @@ class NotImplementedMethodeError(ExeceptionWithMessageBox):
f" '{self.obj.__class__}'"
)
class FileFormatError(ExeceptionWithMessageBox):
def __init__(self, filename, reason):
super(FileFormatError, self).__init__(
title = _translate("Exception", "FileFormatError")
title=_translate("Exception", "FileFormatError")
)
self.reason = reason
@@ -137,7 +143,7 @@ class FileFormatError(ExeceptionWithMessageBox):
class ClipboardFormatError(ExeceptionWithMessageBox):
def __init__(self, mime=None, header=None, data=None):
super(ClipboardFormatError, self).__init__(
title = _translate("Exception", "Clipboard format error")
title=_translate("Exception", "Clipboard format error")
)
self._mime = mime
+20 -11
View File
@@ -20,8 +20,9 @@ from tools import trace, timer
from Model.Tools.PamhyrDB import SQLSubModel
class Friction(SQLSubModel):
def __init__(self, name:str = "", status = None):
def __init__(self, name: str = "", status=None):
super(Friction, self).__init__()
self._status = status
@@ -34,7 +35,7 @@ class Friction(SQLSubModel):
self._end_strickler = None
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
execute("""
CREATE TABLE friction(
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
@@ -53,11 +54,11 @@ class Friction(SQLSubModel):
return True
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
return True
@classmethod
def _sql_load(cls, execute, data = None):
def _db_load(cls, execute, data=None):
new = []
reach = data["parent"] # Reach object
status = data["status"]
@@ -78,22 +79,21 @@ class Friction(SQLSubModel):
es = next(filter(lambda s: s.id == row[4], stricklers))
# Create friction
sec = cls(status = status)
sec = cls(status=status)
sec.edge = reach
sec.begin_kp = row[1]
sec.end_kp = row[2]
sec.begin_strickler = bs
sec.end_strickler = es
new[ind] = sec
yield ind, sec
return new
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
ind = data["ind"]
execute(
"INSERT INTO " +
"friction(ind, begin_kp, end_kp, reach, begin_strickler, end_strickler) " +
"friction(ind, begin_kp, end_kp, " +
"reach, begin_strickler, end_strickler) " +
"VALUES (" +
f"{ind}, {self._begin_kp}, {self._end_kp}, " +
f"{self._edge.id}, " +
@@ -121,7 +121,7 @@ class Friction(SQLSubModel):
self._edge = edge
if (edge is not None and
self._begin_kp == 0.0 and
self._end_kp == 0.0):
self._end_kp == 0.0):
self._begin_kp = self._edge.reach.get_kp_min()
self._end_kp = self._edge.reach.get_kp_max()
self._status.modified()
@@ -129,6 +129,15 @@ class Friction(SQLSubModel):
def has_edge(self):
return self._edge is not None
def has_coefficient(self):
return (
self._begin_strickler is not None and
self._end_strickler is not None
)
def is_full_defined(self):
return self.has_edge() and self.has_coefficient()
@property
def begin_kp(self):
return self._begin_kp
+33 -10
View File
@@ -28,52 +28,75 @@ from Model.Friction.Friction import Friction
logger = logging.getLogger()
class FrictionList(PamhyrModelList):
_sub_classes = [
Friction
]
@classmethod
def _sql_update_0_0_1(cls, execute, version):
def _db_update_0_0_1(cls, execute, version):
execute("ALTER TABLE `section` RENAME TO `friction`")
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
if version == "0.0.0":
logger.info(f"Update friction TABLE from {version}")
cls._sql_update_0_0_1(execute, version)
cls._db_update_0_0_1(execute, version)
return True
@classmethod
def _sql_load(cls, execute, data = None):
new = cls(status = data['status'])
def _db_load(cls, execute, data=None):
new = cls(status=data['status'])
new._lst = Friction._sql_load(
new._lst = Friction._db_load(
execute, data
)
return new
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
frictions = self.lst
reach = data["reach"]
execute(f"DELETE FROM friction WHERE reach = {reach.id}")
ok = True
ind = 0
for friction in self._lst:
for friction in frictions:
data["ind"] = ind
ok &= friction._sql_save(execute, data = data)
ok &= friction._db_save(execute, data=data)
ind += 1
return ok
def _get_frictions_list(self):
# Frictions list generator is type (int, Point) with the first
# element the index of the Point in list
return list(
map(
lambda p: p[1],
sorted(
self._lst,
key=lambda p: p[0]
)
)
)
@property
def lst(self):
if not isinstance(self._lst, list):
self._lst = self._get_frictions_list()
return self._lst
@property
def frictions(self):
return self.lst
def new(self, index):
n = Friction(status = self._status)
n = Friction(status=self._status)
self._lst.insert(index, n)
self._status.modified()
return n
+2 -2
View File
@@ -18,8 +18,9 @@
from Model.Except import NotImplementedMethodeError
class Point(object):
def __init__(self, name:str = "", profile=None, status = None):
def __init__(self, name: str = "", profile=None, status=None):
super(Point, self).__init__()
self._status = status
@@ -28,7 +29,6 @@ class Point(object):
self._profile = profile
self._sl = None
@property
def sl(self):
if self._sl is None:
+4 -3
View File
@@ -20,10 +20,11 @@ from math import dist
from Model.Geometry.Point import Point
class PointAC(Point):
def __init__(self, a:float = 0.0, c:float = 0.0,
name: str = "", status = None):
super(PointXY, self).__init__(name = name, status = status)
def __init__(self, a: float = 0.0, c: float = 0.0,
name: str = "", status=None):
super(PointXY, self).__init__(name=name, status=status)
self._a = float(a)
self._c = float(c)
+17 -25
View File
@@ -22,19 +22,21 @@ import numpy as np
from Model.Tools.PamhyrDB import SQLSubModel
from Model.Geometry.Point import Point
class PointXYZ(Point, SQLSubModel):
_sub_classes = []
def __init__(self, x:float = 0.0, y:float = 0.0, z:float = 0.0,
name:str = "", profile = None, status = None):
super(PointXYZ, self).__init__(name=name, profile=profile, status=status)
def __init__(self, x: float = 0.0, y: float = 0.0, z: float = 0.0,
name: str = "", profile=None, status=None):
super(PointXYZ, self).__init__(
name=name, profile=profile, status=status)
self._x = float(x)
self._y = float(y)
self._z = float(z)
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
execute("""
CREATE TABLE geometry_pointXYZ(
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
@@ -53,7 +55,7 @@ class PointXYZ(Point, SQLSubModel):
return cls._create_submodel(execute)
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
cls._update_submodel(execute, version)
major, minor, release = version.strip().split(".")
@@ -69,9 +71,7 @@ class PointXYZ(Point, SQLSubModel):
return True
@classmethod
def _sql_load(cls, execute, data = None):
points = []
def _db_load(cls, execute, data=None):
status = data["status"]
profile = data["profile"]
@@ -81,10 +81,6 @@ class PointXYZ(Point, SQLSubModel):
f"WHERE profile = {profile.id}"
)
# Create points list
for _ in table:
points.append(None)
# Fill points list with new point
for row in table:
ind = row[0]
@@ -95,13 +91,13 @@ class PointXYZ(Point, SQLSubModel):
sl = row[5]
new = cls(
name = name,
x = x, y = y, z = z,
profile = profile,
status = status
name=name,
x=x, y=y, z=z,
profile=profile,
status=status
)
if sl == -1 or sl == None:
if sl == -1 or sl is None:
new._sl = None
else:
new._sl = next(
@@ -111,11 +107,9 @@ class PointXYZ(Point, SQLSubModel):
)
)
points[ind] = new
yield ind, new
return points
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
profile = data["profile"]
ind = data["ind"]
@@ -123,9 +117,9 @@ class PointXYZ(Point, SQLSubModel):
sql = (
"INSERT OR REPLACE INTO " +
"geometry_pointXYZ(ind, name, x, y, z, profile, sl) "+
"geometry_pointXYZ(ind, name, x, y, z, profile, sl) " +
"VALUES (" +
f"{ind}, '{self._sql_format(self._name)}', " +
f"{ind}, '{self._db_format(self._name)}', " +
f"{self.x}, {self.y}, {self.z}, " +
f"{profile.id}, {sl}" +
")"
@@ -134,7 +128,6 @@ class PointXYZ(Point, SQLSubModel):
return True
@classmethod
def from_data(cls, header, data):
point = None
@@ -157,7 +150,6 @@ class PointXYZ(Point, SQLSubModel):
return point
def __repr__(self):
return f"({self._x}, {self._y}, {self._z}, {self._name})"
+51 -30
View File
@@ -25,14 +25,15 @@ from Model.Except import NotImplementedMethodeError
logger = logging.getLogger()
class Profile(object):
_id_cnt = 0
def __init__(self, id:int = -1, num:int = 0,
kp:float = 0.0, name:str = "",
code1:int = 0, code2:int = 0,
_type:str = "", reach = None,
status = None):
def __init__(self, id: int = -1, num: int = 0,
kp: float = 0.0, name: str = "",
code1: int = 0, code2: int = 0,
_type: str = "", reach=None,
status=None):
super(Profile, self).__init__()
self._status = status
@@ -57,18 +58,34 @@ class Profile(object):
self._profile_type = _type
def __len__(self):
return len(self._points)
return len(self.points)
@property
def number_points(self):
return len(self._points)
return len(self.points)
def _get_points_list(self):
# Points list generator is type (int, Point) with the first
# element the index of the Point in list
return list(
map(
lambda p: p[1],
sorted(
self._points,
key=lambda p: p[0]
)
)
)
@property
def points(self):
return self._points.copy()
if not isinstance(self._points, list):
self._points = self._get_points_list()
return self._points
def point(self, index):
return self._points[index]
return self.points[index]
@property
def reach(self):
@@ -115,7 +132,7 @@ class Profile(object):
@property
def nb_points(self):
return len(self._points)
return len(self.points)
@property
def kp(self):
@@ -169,9 +186,9 @@ class Profile(object):
self._profile_type = value
self._status.modified()
def point(self, i:int):
if i < len(self._points):
return self._points[i]
def point(self, i: int):
if i < len(self.points):
return self.points[i]
return None
@@ -181,11 +198,10 @@ class Profile(object):
Returns:
The list of named point
"""
return [point for point in self._points
return [point for point in self.points
if point.point_is_named()]
def insert_point(self, index: int, point:Point):
def insert_point(self, index: int, point: Point):
"""Insert point at index.
Args:
@@ -195,10 +211,9 @@ class Profile(object):
Returns:
Nothing.
"""
self._points.insert(index, point)
self.points.insert(index, point)
self._status.modified()
def delete(self, indexes: int):
"""Delete points at index
@@ -218,7 +233,7 @@ class Profile(object):
)
)
self._points = list(
self.points = list(
filter(
lambda p: p not in points,
self.points
@@ -235,7 +250,7 @@ class Profile(object):
Returns:
Nothing.
"""
self._points = list(
self.points = list(
filter(
lambda p: p not in points,
self.points
@@ -246,10 +261,10 @@ class Profile(object):
# Move
def move_up_point(self, index: int):
if index < len(self._points):
if index < len(self.points):
next = index - 1
p = self._points
p = self.points
p[index], p[next] = p[next], p[index]
self._status.modified()
@@ -257,7 +272,7 @@ class Profile(object):
if index >= 0:
prev = index + 1
p = self._points
p = self.points
p[index], p[prev] = p[prev], p[index]
self._status.modified()
@@ -265,14 +280,14 @@ class Profile(object):
@timer
def sort(self, column, is_reversed: bool = False):
predicate = lambda p: p.x
def predicate(p): return p.x
if column == 'y':
predicate = lambda p: p.y
def predicate(p): return p.y
elif column == 'z':
predicate = lambda p: p.z
def predicate(p): return p.z
self._points = sorted(
self._points,
self.points = sorted(
self.points,
key=predicate,
reverse=is_reversed
)
@@ -280,10 +295,10 @@ class Profile(object):
@timer
def sort_with_indexes(self, indexes: list):
if len(self._points) != len(indexes):
if len(self.points) != len(indexes):
logger.critical("Indexes list do not correspond to point list")
self._points = list(
self.points = list(
map(
lambda x: x[1],
sorted(
@@ -330,3 +345,9 @@ class Profile(object):
# Abstract method, must be implemented for in non abstract class
def get_station(self):
raise NotImplementedMethodeError(self, self.get_station)
# Computation method
# Abstract method for width approximation
def width_approximation(self):
raise NotImplementedMethodeError(self, self.width_approximation)
+90 -55
View File
@@ -16,6 +16,7 @@
# -*- coding: utf-8 -*-
import logging
import numpy as np
from typing import List
@@ -27,6 +28,9 @@ from Model.Geometry.Profile import Profile
from Model.Geometry.PointXYZ import PointXYZ
from Model.Geometry.Vector_1d import Vector1d
logger = logging.getLogger()
class ProfileXYZ(Profile, SQLSubModel):
_sub_classes = [
PointXYZ,
@@ -36,11 +40,11 @@ class ProfileXYZ(Profile, SQLSubModel):
id: int = -1,
name: str = "",
kp: float = 0.,
reach = None,
num = 0,
reach=None,
num=0,
nb_point: int = 0,
code1: int = 0, code2: int = 0,
status = None):
status=None):
"""ProfileXYZ constructor
Args:
@@ -54,18 +58,18 @@ class ProfileXYZ(Profile, SQLSubModel):
Nothing.
"""
super(ProfileXYZ, self).__init__(
id = id,
num = num,
name = name,
kp = kp,
code1 = code1, code2 = code2,
_type = "XYZ",
reach = reach,
status = status,
id=id,
num=num,
name=name,
kp=kp,
code1=code1, code2=code2,
_type="XYZ",
reach=reach,
status=status,
)
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
execute("""
CREATE TABLE geometry_profileXYZ(
id INTEGER NOT NULL PRIMARY KEY,
@@ -85,7 +89,7 @@ class ProfileXYZ(Profile, SQLSubModel):
return cls._create_submodel(execute)
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
major, minor, release = version.strip().split(".")
if major == minor == "0":
if int(release) < 2:
@@ -100,7 +104,7 @@ class ProfileXYZ(Profile, SQLSubModel):
return cls._update_submodel(execute, version)
@classmethod
def _sql_load(cls, execute, data = None):
def _db_load(cls, execute, data=None):
profiles = []
status = data["status"]
reach = data["reach"]
@@ -108,7 +112,7 @@ class ProfileXYZ(Profile, SQLSubModel):
table = execute(
"SELECT id, ind, name, kp, num, code1, code2, sl " +
"FROM geometry_profileXYZ " +
f"WHERE reach = {reach}"
f"WHERE reach = {reach.id}"
)
for _ in table:
@@ -125,14 +129,14 @@ class ProfileXYZ(Profile, SQLSubModel):
sl = row[7]
new = cls(
id=id, num = num,
name = name, kp = kp,
code1 = code1, code2 = code2,
reach = data["parent"],
status = status
id=id, num=num,
name=name, kp=kp,
code1=code1, code2=code2,
reach=data["parent"],
status=status
)
if sl == -1 or sl == None:
if sl == -1 or sl is None:
new._sl = None
else:
new._sl = next(
@@ -143,13 +147,15 @@ class ProfileXYZ(Profile, SQLSubModel):
)
data["profile"] = new
new._points = PointXYZ._sql_load(execute, data)
new._points = PointXYZ._db_load(execute, data.copy())
profiles[ind] = new
yield ind, new
return profiles
# profiles[ind] = new
def _sql_save(self, execute, data = None):
# return profiles
def _db_save(self, execute, data=None):
ok = True
ind = data["ind"]
@@ -157,22 +163,25 @@ class ProfileXYZ(Profile, SQLSubModel):
sql = (
"INSERT OR REPLACE INTO " +
"geometry_profileXYZ(id, ind, name, reach, kp, num, code1, code2, sl) "+
"geometry_profileXYZ(id, ind, name, reach, " +
"kp, num, code1, code2, sl) " +
"VALUES (" +
f"{self.id}, {ind}, '{self._sql_format(self._name)}', " +
f"{self.id}, {ind}, '{self._db_format(self._name)}', " +
f"{self.reach.id}, {self.kp}, {self.num}, " +
f"{self.code1}, {self.code1}, {sl}" +
")"
)
execute(sql)
points = self.points
data["profile"] = self
execute(f"DELETE FROM geometry_pointXYZ WHERE profile = {self.id}")
ind = 0
for point in self._points:
for point in points:
data["ind"] = ind
ok &= point._sql_save(execute, data)
ok &= point._db_save(execute, data)
ind += 1
return ok
@@ -220,18 +229,17 @@ class ProfileXYZ(Profile, SQLSubModel):
return point
def x(self):
return [point.x for point in self._points]
return [point.x for point in self.points]
def y(self):
return [point.y for point in self._points]
return [point.y for point in self.points]
def z(self):
return [point.z for point in self._points]
return [point.z for point in self.points]
def names(self):
return [point.name for point in self._points]
return [point.name for point in self.points]
def x_max(self):
return max(self.filter_isnan(self.x()))
@@ -262,7 +270,7 @@ class ProfileXYZ(Profile, SQLSubModel):
"""
for point in list_points:
pt = PointXYZ(*point, profile=self, status=self._status)
self._points.append(pt)
self.points.append(pt)
self._status.modified()
def get_point_i(self, index: int) -> PointXYZ:
@@ -275,10 +283,31 @@ class ProfileXYZ(Profile, SQLSubModel):
The point.
"""
try:
return self._points[index]
return self.points[index]
except IndexError:
raise IndexError(f"Invalid point index: {index}")
def get_point_by_name(self, name: str) -> PointXYZ:
"""Get point by name.
Args:
name: Point name.
Returns:
The point.
"""
try:
n_name = name.lower().strip()
return next(
filter(
lambda p: p.name.lower().strip() == n_name,
self.points
)
)
except Exception as e:
logger.debug(f"{e}")
raise IndexError(f"Invalid point name: {name}")
def add(self):
"""Add a new PointXYZ to profile.
@@ -286,7 +315,7 @@ class ProfileXYZ(Profile, SQLSubModel):
Nothing.
"""
point_xyz = PointXYZ(0., 0., 0., profile=self, status=self._status)
self._points.append(point_xyz)
self.points.append(point_xyz)
self._status.modified()
def insert(self, index: int):
@@ -299,7 +328,7 @@ class ProfileXYZ(Profile, SQLSubModel):
The new point.
"""
point = PointXYZ(0., 0., 0., profile=self, status=self._status)
self._points.insert(index, point)
self.points.insert(index, point)
self._status.modified()
return point
@@ -315,9 +344,9 @@ class ProfileXYZ(Profile, SQLSubModel):
return [x for x in lst if not np.isnan(x)]
def _first_point_not_nan(self):
first_point = self._points[0]
first_point = self.points[0]
for point in self._points:
for point in self.points:
if not point.is_nan():
first_point = point
break
@@ -325,9 +354,9 @@ class ProfileXYZ(Profile, SQLSubModel):
return first_point
def _last_point_not_nan(self):
last_point = self._points[-1]
last_point = self.points[-1]
for point in self._points[::-1]:
for point in self.points[::-1]:
if not point.is_nan():
last_point = point
break
@@ -354,13 +383,13 @@ class ProfileXYZ(Profile, SQLSubModel):
first_point_not_nan = self._first_point_not_nan()
last_point_not_nan = self._last_point_not_nan()
for index, point in enumerate(self._points):
for index, point in enumerate(self.points):
if point.point_is_named():
index_first_named_point = index
first_named_point = point
break
for point in reversed(self._points):
for point in reversed(self.points):
if point.point_is_named():
last_named_point = point
break
@@ -369,32 +398,32 @@ class ProfileXYZ(Profile, SQLSubModel):
constant = 0.0
if (first_named_point is not None and
last_named_point is not None):
last_named_point is not None):
if (first_named_point != last_named_point and
first_named_point.x != last_named_point.x):
first_named_point.x != last_named_point.x):
vector = Vector1d(first_named_point, last_named_point)
normalized_direction_vec = vector.normalized_direction_vector()
norm_dir_vec = vector.normalized_direction_vector()
else:
vector = Vector1d(first_point_not_nan, last_point_not_nan)
normalized_direction_vec = vector.normalized_direction_vector()
norm_dir_vec = vector.normalized_direction_vector()
for point in self._points:
for point in self.points:
xi = point.x - first_named_point.x
yi = point.y - first_named_point.y
station_i = (normalized_direction_vec[0] * xi +
normalized_direction_vec[1] * yi)
station_i = (norm_dir_vec[0] * xi +
norm_dir_vec[1] * yi)
station.append(station_i)
constant = station[index_first_named_point]
elif first_named_point is None:
vector = Vector1d(first_point_not_nan, last_point_not_nan)
normalized_direction_vec = vector.normalized_direction_vector()
norm_dir_vec = vector.normalized_direction_vector()
for point in self._points:
for point in self.points:
xi = point.x - first_point_not_nan.x
yi = point.y - first_point_not_nan.y
station_i = (normalized_direction_vec[0] * xi +
normalized_direction_vec[1] * yi)
station_i = (norm_dir_vec[0] * xi +
norm_dir_vec[1] * yi)
station.append(station_i)
z_min = self.z_min()
@@ -407,3 +436,9 @@ class ProfileXYZ(Profile, SQLSubModel):
constant = station[index_profile_z_min[0]]
return list(map(lambda s: s - constant, station))
def width_approximation(self):
rg = self.get_point_by_name("rg")
rd = self.get_point_by_name("rd")
return abs(rg.dist(rd))
+164 -38
View File
@@ -36,6 +36,7 @@ from Model.Except import FileFormatError, exception_message_box
logger = logging.getLogger()
class Reach(SQLSubModel):
_sub_classes = [
ProfileXYZ,
@@ -51,36 +52,37 @@ class Reach(SQLSubModel):
self._guidelines = {}
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
return cls._create_submodel(execute)
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
return cls._update_submodel(execute, version)
@classmethod
def _sql_load(cls, execute, data = None):
new = cls(status = data["status"], parent = data["parent"])
def _db_load(cls, execute, data=None):
new = cls(status=data["status"], parent=data["parent"])
new._profiles = ProfileXYZ._sql_load(
new._profiles = ProfileXYZ._db_load(
execute,
data = data
data=data
)
return new
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
profiles = self.profiles
# Delete old data
execute(f"DELETE FROM geometry_profileXYZ WHERE reach = {self.id}")
# execute(f"DELETE FROM geometry_pointXYZ")
if data is None:
data = {}
ind = 0
for profile in self._profiles:
for profile in profiles:
data["ind"] = ind
profile._sql_save(execute, data)
profile._db_save(execute, data)
ind += 1
return True
@@ -94,29 +96,53 @@ class Reach(SQLSubModel):
Returns:
The profile at index i
"""
if i < len(self._profiles):
return self._profiles[i]
if i < len(self.profiles):
return self.profiles[i]
return None
@property
def name(self):
if self._parent == None:
if self._parent is None:
return ""
return self._parent.name
def _get_profiles_list(self):
# Profiles list generator is type (int, Point) with the first
# element the index of the Point in list
logger.info(f"Load profiles from reach {self.name}")
return list(
map(
lambda p: p[1],
sorted(
self._profiles,
key=lambda p: p[0]
)
)
)
def __len__(self):
if not isinstance(self._profiles, list):
self._profiles = self._get_profiles_list()
return len(self._profiles)
@property
def profiles(self):
return self._profiles.copy()
if not isinstance(self._profiles, list):
self._profiles = self._get_profiles_list()
return self._profiles
@profiles.setter
def profiles(self, profiles):
self._profiles = profiles
def get_profiles_from_kp(self, kp):
return list(
filter(
lambda p: p.kp == kp, self._profiles
lambda p: p.kp == kp, self.profiles
)
)
@@ -126,14 +152,14 @@ class Reach(SQLSubModel):
Returns:
Number of profiles
"""
return len(self._profiles)
return len(self.profiles)
def get_geometry(self) -> List[Profile]:
"""
Returns:
The profiles list.
"""
return self._profiles
return self.profiles
def _update_profile_numbers(self):
"""Update profiles index
@@ -155,7 +181,7 @@ class Reach(SQLSubModel):
"""
profile = ProfileXYZ(reach=self, status=self._status)
self._profiles.insert(index, profile)
self.profiles.insert(index, profile)
self._update_profile_numbers()
self._status.modified()
@@ -171,11 +197,10 @@ class Reach(SQLSubModel):
Returns:
Nothing.
"""
self._profiles.insert(index, profile)
self.profiles.insert(index, profile)
self._update_profile_numbers()
self._status.modified()
def delete(self, indexes):
"""Delete some elements in profile list
@@ -222,12 +247,11 @@ class Reach(SQLSubModel):
self._update_profile_numbers()
self._status.modified()
def move_up_profile(self, index: int):
if index < len(self.profiles):
next = index - 1
p = self._profiles
p = self.profiles
p[index], p[next] = p[next], p[index]
self._status.modified()
@@ -235,7 +259,7 @@ class Reach(SQLSubModel):
if index >= 0:
prev = index + 1
p = self._profiles
p = self.profiles
p[index], p[prev] = p[prev], p[index]
self._status.modified()
@@ -278,7 +302,6 @@ class Reach(SQLSubModel):
def get_kp_max(self):
return max(self.get_kp())
# Sediment Layers
def get_sl(self):
@@ -327,12 +350,12 @@ class Reach(SQLSubModel):
for guide in guide_set:
self._guidelines[guide] = flatten(
map(
lambda l: list(
lambda lst: list(
# Filter point with name (we assume we have
# only one point by profile)
filter(
lambda p: p.name == guide,
l
lst
)
),
named_points
@@ -347,7 +370,7 @@ class Reach(SQLSubModel):
Tuple of complete and incomplete guidelines name.
"""
# Get all point contains into a guideline
named_points = [profile.named_points() for profile in self._profiles]
named_points = [profile.named_points() for profile in self.profiles]
points_name = list(
map(
lambda lst: list(map(lambda p: p.name, lst)),
@@ -367,7 +390,9 @@ class Reach(SQLSubModel):
reduce(
lambda acc, h: acc + h,
map(
lambda l: list(set(l).symmetric_difference(guide_set)),
lambda lst: list(
set(lst).symmetric_difference(guide_set)
),
points_name
),
[]
@@ -417,13 +442,12 @@ class Reach(SQLSubModel):
def get_guidelines_z(self):
return self._map_guidelines_points(lambda p: p.z)
# Sort
@timer
def sort(self, is_reversed: bool = False):
self._profiles = sorted(
self._profiles,
self.profiles = sorted(
self.profiles,
key=lambda profile: profile.kp,
reverse=is_reversed
)
@@ -431,10 +455,10 @@ class Reach(SQLSubModel):
@timer
def sort_with_indexes(self, indexes: list):
if len(self._profiles) != len(indexes):
if len(self.profiles) != len(indexes):
logger.critical("Indexes list do not correspond to profile list")
self._profiles = list(
self.profiles = list(
map(
lambda x: x[1],
sorted(
@@ -452,7 +476,8 @@ class Reach(SQLSubModel):
"""Import a geometry from file (.ST or .st)
Args:
file_path_name: The absolute path of geometry file (.ST or .st) to import.
file_path_name: The absolute path of geometry file (.ST or .st)
to import.
Returns:
Nothing.
@@ -462,7 +487,8 @@ class Reach(SQLSubModel):
try:
list_profile, list_header = self.read_file_st(str(file_path_name))
profile_header = ["num", "code1", "code2", "nb_point", "kp", "name"]
profile_header = ["num", "code1",
"code2", "nb_point", "kp", "name"]
if list_profile and list_header:
for ind, profile in enumerate(list_profile):
@@ -474,7 +500,7 @@ class Reach(SQLSubModel):
**d, reach=self, status=self._status
)
prof.import_points(profile)
self._profiles.append(prof)
self.profiles.append(prof)
self._update_profile_numbers()
self._status.modified()
@@ -538,8 +564,8 @@ class Reach(SQLSubModel):
# TODO: Move this function to model reach
def export_reach(self, filename):
with open(f"{filename}", "w") as file_st:
for index in range(len(self._profiles)):
profile = self._profiles[index]
for index in range(len(self.profiles)):
profile = self.profiles[index]
for v in profile.header:
file_st.write(f"{v:>6}")
@@ -552,3 +578,103 @@ class Reach(SQLSubModel):
file_st.write(" 999.9990 999.9990 999.9990")
file_st.write("\n")
def get_incline(self):
first = self.profile(0)
last = self.profile(len(self) - 1)
z_first = first.z_min()
kp_first = first.kp
z_last = last.z_min()
kp_last = last.kp
return (
(z_last - z_first)
/
(kp_last - kp_first)
)
def get_incline_mean(self):
profiles = self.profiles
previous = profiles[0]
incline_acc = 0
for profile in profiles[1:]:
z_first = previous.z_min()
kp_first = previous.kp
z_last = profile.z_min()
kp_last = profile.kp
incline_acc += (
(z_last - z_first)
/
(kp_last - kp_first)
)
previous = profile
return (incline_acc / (len(profiles) - 1))
def get_incline_median(self):
profiles = self.profiles
previous = profiles[0]
incline_acc = []
for profile in profiles[1:]:
z_first = previous.z_min()
kp_first = previous.kp
z_last = profile.z_min()
kp_last = profile.kp
incline_acc += [
(z_last - z_first)
/
(kp_last - kp_first)
]
previous = profile
incline_acc.sort()
return incline_acc[round(len(profiles)/2)]
def get_incline_median_mean(self):
profiles = self.profiles
previous = profiles[0]
incline_acc = []
for profile in profiles[1:]:
z_first = previous.z_min()
kp_first = previous.kp
z_last = profile.z_min()
kp_last = profile.kp
incline_acc += [
(z_last - z_first)
/
(kp_last - kp_first)
]
previous = profile
incline_acc.sort()
marge = round(len(incline_acc) * 0.1)
incline_set = incline_acc[marge:-marge]
logger.debug(f"+{incline_acc}")
logger.debug(f"-{incline_set}")
return (
reduce(
lambda acc, x: acc + x,
incline_set,
0.0
) / (len(incline_set))
)
+1
View File
@@ -19,6 +19,7 @@
import numpy as np
from Model.Geometry.PointXYZ import PointXYZ
class Vector1d:
def __init__(self, a: PointXYZ, b: PointXYZ):
self.A = a
+119 -75
View File
@@ -16,18 +16,23 @@
# -*- coding: utf-8 -*-
import logging
from copy import copy, deepcopy
from tools import trace, timer
from functools import reduce
from Model.Tools.PamhyrDB import SQLSubModel
logger = logging.getLogger()
class Data(SQLSubModel):
def __init__(self, name:str = "",
comment:str = "", reach = None,
kp:float = 0.0, discharge:float = 0.0,
height:float = 0.0,
status = None):
def __init__(self, name: str = "",
comment: str = "", reach=None,
kp: float = 0.0, discharge: float = 0.0,
height: float = 0.0,
status=None):
super(Data, self).__init__()
self._status = status
@@ -51,7 +56,7 @@ class Data(SQLSubModel):
self._update_from_discharge()
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
execute("""
CREATE TABLE initial_conditions(
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
@@ -69,11 +74,11 @@ class Data(SQLSubModel):
return cls._create_submodel(execute)
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
return cls._update_submodel(execute, version)
@classmethod
def _sql_load(cls, execute, data = None):
def _db_load(cls, execute, data=None):
id = data["reach"].id
table = execute(
"SELECT ind, name, comment, kp, discharge, height " +
@@ -95,28 +100,29 @@ class Data(SQLSubModel):
height = row[5]
d = cls(
reach = data["reach"],
status = data["status"],
name = name,
comment = comment,
kp = kp,
discharge = discharge,
height = height,
reach=data["reach"],
status=data["status"],
name=name,
comment=comment,
kp=kp,
discharge=discharge,
height=height,
)
new[ind] = d
return new
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
ind = data["ind"]
execute(
"INSERT INTO " +
"initial_conditions(ind, name, comment, kp, discharge, height, reach) " +
"initial_conditions(ind, name, comment, kp, " +
"discharge, height, reach) " +
"VALUES (" +
f"{ind}, '{self._sql_format(self.name)}', " +
f"'{self._sql_format(self._comment)}', " +
f"{ind}, '{self._db_format(self.name)}', " +
f"'{self._db_format(self._comment)}', " +
f"{self._kp}, {self._discharge}, {self._height}, " +
f"{self._reach.id}" +
")"
@@ -124,16 +130,15 @@ class Data(SQLSubModel):
return True
def copy(self):
new = Data(
name = self.name,
comment = self._comment,
kp = self._kp,
discharge = self._discharge,
height = self._height,
reach = self._reach,
status = self._status,
name=self.name,
comment=self._comment,
kp=self._kp,
discharge=self._discharge,
height=self._height,
reach=self._reach,
status=self._status,
)
return new
@@ -173,7 +178,7 @@ class Data(SQLSubModel):
def _update_from_kp(self):
min = self._update_get_min()
self._elevation = min - self._height
self._elevation = min + self._height
def _update_from_elevation(self):
min = self._update_get_min()
@@ -216,7 +221,7 @@ class InitialConditions(SQLSubModel):
Data
]
def __init__(self, reach = None, status = None):
def __init__(self, reach=None, status=None):
super(InitialConditions, self).__init__()
self._status = status
@@ -225,37 +230,35 @@ class InitialConditions(SQLSubModel):
self._data = []
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
return cls._create_submodel(execute)
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
return cls._update_submodel(execute, version)
@classmethod
def _sql_load(cls, execute, data = None):
def _db_load(cls, execute, data=None):
new = cls(
reach = data["reach"],
status = data["status"]
reach=data["reach"],
status=data["status"]
)
new._data = Data._sql_load(
new._data = Data._db_load(
execute,
data = data
data=data
)
if new._data is None:
return None
if new._data is not None:
yield new
return new
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
ok = True
ind = 0
for d in self._data:
data["ind"] = ind
ok &= d._sql_save(execute, data)
ok &= d._db_save(execute, data)
ind += 1
return ok
@@ -288,7 +291,7 @@ class InitialConditions(SQLSubModel):
self._status.modified()
def new(self, index):
n = Data(reach = self._reach, status = self._status)
n = Data(reach=self._reach, status=self._status)
self._data.insert(index, n)
self._status.modified()
@@ -321,7 +324,6 @@ class InitialConditions(SQLSubModel):
self._data.sort(reverse=reverse, key=key)
self._status.modified()
def _data_get(self, key):
return list(
map(
@@ -341,8 +343,8 @@ class InitialConditions(SQLSubModel):
def _sort_by_z_and_kp(self, profiles):
profiles.sort(
reverse = False,
key = lambda p: p.kp
reverse=False,
key=lambda p: p.kp
)
first_z = profiles[0].z()
@@ -350,47 +352,89 @@ class InitialConditions(SQLSubModel):
if first_z > last_z:
profiles.sort(
reverse = True,
key = lambda p: p.kp
reverse=True,
key=lambda p: p.kp
)
def generate_growing_constante_height(self, height:float):
def generate_growing_constante_height(self, height: float):
self._data = []
profiles = self._reach.reach.profiles
profiles = self._reach.reach.profiles.copy()
self._sort_by_z_and_kp(profiles)
prev = None
for profile in profiles:
new = Data(reach = self._reach, status = self._status)
new["kp"] = profile.kp
incline = self._reach.reach.get_incline_median_mean()
logger.debug(f"incline = {incline}")
if prev is None:
new["elevation"] = profile.z_min() + height
else:
new["elevation"] = max(
profile.z_min() + height,
prev["elevation"]
)
previous_elevation = -99999.99
for profile in profiles:
width = profile.width_approximation()
strickler = 25
discharge = (
((width * 0.8)
* strickler
* (height ** (5/3))
* (abs(incline) ** (0.5)))
)
elevation= max(
profile.z_min() + height,
previous_elevation
)
logger.debug(f"({profile.kp}):")
logger.debug(f" width = {width}")
logger.debug(f" strickler = {strickler}")
logger.debug(f" discharge = {discharge}")
new = Data(reach=self._reach, status=self._status)
new["kp"] = profile.kp
new["discharge"] = discharge
new["elevation"] = elevation
self._data.append(new)
prev = new
previous_elevation = elevation
is_reverse = False
if profiles[0].kp > profiles[-1].kp:
is_reverse = True
self._data.sort(
reverse = not is_reverse,
key = lambda d: d['kp']
def generate_discharge(self, discharge: float):
self._data = []
self._generate_height_estimation_from_discharge(
discharge
)
def generate_discharge(self, discharge:float):
self._new = []
def _generate_height_estimation_from_discharge(self, discharge: float):
profiles = self._reach.reach.profiles.copy()
self._sort_by_z_and_kp(profiles)
for d in self._data:
n = d.copy()
n['discharge'] = discharge
self._new.append(n)
previous_elevation = -99999.99
self._data = self._new
incline = self._reach.reach.get_incline_median_mean()
logger.debug(f"incline = {incline}")
for profile in profiles:
width = profile.width_approximation()
strickler = 25
height = (
discharge
/
((width * 0.8) * strickler * (abs(incline) ** (0.5)))
) ** (0.6)
elevation= max(
profile.z_min() + height,
previous_elevation
)
logger.debug(f"({profile.kp}):")
logger.debug(f" width = {width}")
logger.debug(f" strickler = {strickler}")
logger.debug(f" hieght = {height}")
new = Data(reach=self._reach, status=self._status)
new["kp"] = profile.kp
new["discharge"] = discharge
new["elevation"] = elevation
previous_elevation = elevation
self._data.append(new)
@@ -16,27 +16,33 @@
# -*- coding: utf-8 -*-
import types
import logging
from copy import copy
from tools import trace, timer
from Model.Tools.PamhyrDict import PamhyrModelDict
from Model.InitialConditions.InitialConditions import InitialConditions
logger = logging.getLogger()
class InitialConditionsDict(PamhyrModelDict):
_sub_classes = [
InitialConditions,
]
@classmethod
def _sql_load(cls, execute, data = None):
new = cls(status = data["status"])
def _db_load(cls, execute, data=None):
new = cls(status=data["status"])
for reach in data["edges"]:
data["reach"] = reach
ic = InitialConditions._sql_load(
ic = InitialConditions._db_load(
execute,
data = data
data=data
)
if ic is not None:
@@ -44,20 +50,40 @@ class InitialConditionsDict(PamhyrModelDict):
return new
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
ok = True
if data is None:
data = {}
execute("DELETE FROM initial_conditions")
for reach in self._dict:
ics = self._dict
for reach in ics:
data["reach"] = reach
ok &= self._dict[reach]._sql_save(execute, data)
v = self._dict[reach]
if isinstance(v, types.GeneratorType):
self._dict[reach] = list(v)[0]
execute(
"DELETE FROM initial_conditions " +
f"WHERE reach = '{reach.id}'"
)
ok &= self._dict[reach]._db_save(execute, data)
return ok
def get(self, key):
if key in self._dict:
v = self._dict[key]
if isinstance(v, types.GeneratorType):
self._dict[key] = list(v)[0]
return self._dict[key]
new = self.new(key)
self.set(key, new)
return new
def new(self, reach):
new = InitialConditions(reach = reach, status = self._status)
new = InitialConditions(reach=reach, status=self._status)
self.set(reach, new)
return new
@@ -25,11 +25,12 @@ from Model.Except import NotImplementedMethodeError
logger = logging.getLogger()
class LateralContribution(SQLSubModel):
_sub_classes = []
_id_cnt = 0
def __init__(self, id:int = -1, name:str = "", status = None):
def __init__(self, id: int = -1, name: str = "", status=None):
super(LateralContribution, self).__init__()
self._status = status
@@ -39,7 +40,6 @@ class LateralContribution(SQLSubModel):
else:
self.id = id
self._name = name
self._type = ""
self._edge = None
@@ -49,10 +49,11 @@ class LateralContribution(SQLSubModel):
self._header = []
self._types = [float, float]
LateralContribution._id_cnt = max(LateralContribution._id_cnt + 1, self.id)
LateralContribution._id_cnt = max(
LateralContribution._id_cnt + 1, self.id)
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
execute("""
CREATE TABLE lateral_contribution(
id INTEGER NOT NULL PRIMARY KEY,
@@ -80,7 +81,7 @@ class LateralContribution(SQLSubModel):
return cls._create_submodel(execute)
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
return True
@classmethod
@@ -99,7 +100,7 @@ class LateralContribution(SQLSubModel):
return res
@classmethod
def _sql_load(cls, execute, data = None):
def _db_load(cls, execute, data=None):
new = []
tab = data["tab"]
@@ -112,9 +113,9 @@ class LateralContribution(SQLSubModel):
t = row[2]
ctor = cls._get_ctor_from_type(t)
lc = ctor(
id = row[0],
name = row[1],
status = data['status']
id=row[0],
name=row[1],
status=data['status']
)
lc.edge = None
if row[3] != -1:
@@ -141,7 +142,7 @@ class LateralContribution(SQLSubModel):
return new
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
tab = data["tab"]
execute(f"DELETE FROM lateral_contribution WHERE id = {self.id}")
@@ -153,10 +154,11 @@ class LateralContribution(SQLSubModel):
sql = (
"INSERT INTO " +
"lateral_contribution(id, name, type, tab, edge, begin_kp, end_kp) "+
"lateral_contribution(id, name, type, tab, " +
"edge, begin_kp, end_kp) " +
"VALUES (" +
f"{self.id}, '{self._sql_format(self._name)}', " +
f"'{self._sql_format(self._type)}', '{tab}', {edge}, " +
f"{self.id}, '{self._db_format(self._name)}', " +
f"'{self._db_format(self._type)}', '{tab}', {edge}, " +
f"{self._begin_kp}, {self._end_kp}" +
")"
)
@@ -164,12 +166,12 @@ class LateralContribution(SQLSubModel):
ind = 0
for d in self._data:
data0 = self._sql_format(str(d[0]))
data1 = self._sql_format(str(d[1]))
data0 = self._db_format(str(d[0]))
data1 = self._db_format(str(d[1]))
sql = (
"INSERT INTO " +
"lateral_contribution_data(ind, data0, data1, lc) "+
"lateral_contribution_data(ind, data0, data1, lc) " +
f"VALUES ({ind}, '{data0}', {data1}, {self.id})"
)
execute(sql)
@@ -186,8 +188,13 @@ class LateralContribution(SQLSubModel):
@classmethod
def time_convert(cls, data):
if type(data) == str and data.count(":") == 3:
return old_pamhyr_date_to_timestamp(data)
if type(data) is str:
if data.count(":") == 3:
return old_pamhyr_date_to_timestamp(data)
if data.count(":") == 2:
return old_pamhyr_date_to_timestamp("00:" + data)
if data.count(".") == 1:
return round(float(data))
return int(data)
@@ -282,7 +289,7 @@ class LateralContribution(SQLSubModel):
new_1 = self._default_1
if len(header) != 0:
for i in [0,1]:
for i in [0, 1]:
for j in range(len(header)):
if self._header[i] == header[j]:
if i == 0:
@@ -295,13 +302,13 @@ class LateralContribution(SQLSubModel):
return (new_0, new_1)
def add(self, index:int):
def add(self, index: int):
value = (self._default_0, self._default_1)
self._data.insert(index, value)
self._status.modified()
return value
def insert(self, index:int, value):
def insert(self, index: int, value):
self._data.insert(index, value)
self._status.modified()
@@ -337,11 +344,10 @@ class LateralContribution(SQLSubModel):
return self.data[index]
def get_range(self, _range):
l = []
lst = []
for r in _range:
l.append(r)
return l
lst.append(r)
return lst
def _set_i_c_v(self, index, column, value):
v = list(self._data[index])
@@ -349,15 +355,15 @@ class LateralContribution(SQLSubModel):
self._data[index] = tuple(v)
self._status.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)
def set_i_1(self, index:int, value):
def set_i_1(self, index: int, value):
self._set_i_c_v(index, 1, value)
@timer
def convert(self, cls):
new = cls(name = self.name, status = self._status)
new = cls(name=self.name, status=self._status)
new.edge = self.edge
new.begin_kp = self.begin_kp
new.end_kp = self.end_kp
@@ -365,8 +371,8 @@ class LateralContribution(SQLSubModel):
for i, _ in enumerate(self.data):
new.add(i)
for i in [0,1]:
for j in [0,1]:
for i in [0, 1]:
for j in [0, 1]:
if self._header[i] == new.header[j]:
for ind, v in self.data:
try:
@@ -27,6 +27,7 @@ from Model.LateralContribution.LateralContributionTypes import (
NotDefined, LateralContrib, Rain, Evaporation,
)
class LateralContributionList(PamhyrModelListWithTab):
_tabs_list = ["liquid", "solid", "suspenssion"]
_sub_classes = [
@@ -34,21 +35,21 @@ class LateralContributionList(PamhyrModelListWithTab):
]
@classmethod
def _sql_load(cls, execute, data = None):
new = cls(status = data['status'])
def _db_load(cls, execute, data=None):
new = cls(status=data['status'])
if data is None:
data = {}
for tab in new._tabs:
data["tab"] = tab
new._tabs[tab] = LateralContribution._sql_load(
new._tabs[tab] = LateralContribution._db_load(
execute, data
)
return new
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
execute("DELETE FROM lateral_contribution")
if data is None:
@@ -57,7 +58,7 @@ class LateralContributionList(PamhyrModelListWithTab):
for tab in self._tabs:
data["tab"] = tab
for lc in self._tabs[tab]:
lc._sql_save(execute, data = data)
lc._db_save(execute, data=data)
return True
@@ -70,16 +71,16 @@ class LateralContributionList(PamhyrModelListWithTab):
def __copy__(self):
new = LateralContributionList()
for l in self._tabs:
new.tabs[l] = self._tabs[l].copy()
for lst in self._tabs:
new.tabs[lst] = self._tabs[lst].copy()
return new
def __deepcopy__(self):
new = LateralContributionList()
for l in self._tabs:
new.tabs[l] = self._tabs[l].deepcopy()
for lst in self._tabs:
new.tabs[lst] = self._tabs[lst].deepcopy()
return new
@@ -22,7 +22,7 @@ from Model.LateralContribution.LateralContribution import LateralContribution
class NotDefined(LateralContribution):
def __init__(self, id:int = -1, name:str = "", status=None):
def __init__(self, id: int = -1, name: str = "", status=None):
super(NotDefined, self).__init__(id=id, name=name, status=status)
self._type = "ND"
@@ -32,8 +32,9 @@ class NotDefined(LateralContribution):
def _default_0(self):
return 0.0
class LateralContrib(LateralContribution):
def __init__(self, id:int = -1, name:str = "", status=None):
def __init__(self, id: int = -1, name: str = "", status=None):
super(LateralContrib, self).__init__(id=id, name=name, status=status)
self._type = "LC"
@@ -44,8 +45,9 @@ class LateralContrib(LateralContribution):
def compatibility(cls):
return ["liquid"]
class Rain(LateralContribution):
def __init__(self, id:int = -1, name:str = "", status=None):
def __init__(self, id: int = -1, name: str = "", status=None):
super(Rain, self).__init__(id=id, name=name, status=status)
self._type = "RA"
@@ -56,8 +58,9 @@ class Rain(LateralContribution):
def compatibility(cls):
return ["liquid"]
class Evaporation(LateralContribution):
def __init__(self, id:int = -1, name:str = "", status=None):
def __init__(self, id: int = -1, name: str = "", status=None):
super(Evaporation, self).__init__(id=id, name=name, status=status)
self._type = "EV"
+9 -5
View File
@@ -18,13 +18,14 @@
from Model.Network.Node import Node
class Edge(object):
_id_cnt = 0
def __init__(self, id:int, name:str,
node1:Node = None,
node2:Node = None,
status = None):
def __init__(self, id: int, name: str,
node1: Node = None,
node2: Node = None,
status=None):
super(Edge, self).__init__()
self._status = status
@@ -74,7 +75,10 @@ class Edge(object):
@property
def name(self):
return self._name if self._name != "" else f"{self.node1.name} -> {self.node2.name}"
name = self._name
if self._name == "":
name = f"{self.node1.name} -> {self.node2.name}"
return name
def is_enable(self):
return self._enable
+15 -15
View File
@@ -21,8 +21,9 @@ from functools import reduce
from Model.Network.Node import Node
from Model.Network.Edge import Edge
class Graph(object):
def __init__(self, status = None):
def __init__(self, status=None):
super(Graph, self).__init__()
self._status = status
@@ -83,7 +84,7 @@ class Graph(object):
False
)
def node(self, node_name:str):
def node(self, node_name: str):
node = list(
filter(
lambda n: n.name == node_name,
@@ -96,7 +97,7 @@ class Graph(object):
return node[0]
def edge(self, edge_name:str):
def edge(self, edge_name: str):
edge = list(
filter(
lambda e: e.name == edge_name,
@@ -109,12 +110,12 @@ class Graph(object):
return edge[0]
def _create_node(self, x:float, y:float):
def _create_node(self, x: float, y: float):
node = self._node_ctor(
-1,
"",
x = x, y = y,
status = self._status
x=x, y=y,
status=self._status
)
return node
@@ -123,14 +124,14 @@ class Graph(object):
self._status.modified()
return node
def add_node(self, x:float = 0.0, y:float = 0.0):
def add_node(self, x: float = 0.0, y: float = 0.0):
node = self._create_node(x, y)
return self._add_node(node)
def insert_node(self, node):
return self._add_node(node)
def remove_node(self, node_name:str):
def remove_node(self, node_name: str):
self._nodes = list(
filter(
lambda n: n.name != node_name,
@@ -139,19 +140,18 @@ class Graph(object):
)
self._status.modified()
def create_node(self, x:float = 0.0, y:float = 0.0):
def create_node(self, x: float = 0.0, y: float = 0.0):
node = self._create_node(x, y)
return node
def _create_edge(self, n1:Node, n2:Node):
def _create_edge(self, n1: Node, n2: Node):
edge = self._edge_ctor(
-1,
"", n1, n2,
status = self._status
status=self._status
)
return edge
def _add_edge(self, edge):
# This edge already exists ?
if any(filter(lambda e: (e.node1 == edge.node1 and
@@ -164,17 +164,17 @@ class Graph(object):
self._status.modified()
return edge
def add_edge(self, n1:Node, n2:Node):
def add_edge(self, n1: Node, n2: Node):
edge = self._create_edge(n1, n2)
return self._add_edge(edge)
def insert_edge(self, edge):
return self._add_edge(edge)
def create_edge(self, n1:Node, n2:Node):
def create_edge(self, n1: Node, n2: Node):
return self._create_edge(n1, n2)
def remove_edge(self, edge_name:str):
def remove_edge(self, edge_name: str):
self._edges = list(
filter(
lambda e: e.name != edge_name,
+4 -4
View File
@@ -18,12 +18,13 @@
from Model.Network.Point import Point
class Node(object):
_id_cnt = 0
def __init__(self, id:int, name:str,
x:float = 0.0, y:float = 0.0,
status = None):
def __init__(self, id: int, name: str,
x: float = 0.0, y: float = 0.0,
status=None):
super(Node, self).__init__()
self._status = status
@@ -73,7 +74,6 @@ class Node(object):
def y(self):
return self.pos.y
def setPos(self, x, y):
self.pos.x = x
self.pos.y = y
+1 -1
View File
@@ -17,7 +17,7 @@
# -*- coding: utf-8 -*-
class Point(object):
def __init__(self, x:float, y:float):
def __init__(self, x: float, y: float):
super(Point, self).__init__()
self.x = x
+268
View File
@@ -0,0 +1,268 @@
# Reservoir.py -- Pamhyr
# Copyright (C) 2023 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 Model.Tools.PamhyrDB import SQLSubModel
logger = logging.getLogger()
class Reservoir(SQLSubModel):
_sub_classes = []
_id_cnt = 0
def __init__(self, id: int = -1, name: str = "",
status=None):
super(Reservoir, self).__init__()
self._status = status
if id == -1:
self.id = Reservoir._id_cnt
else:
self.id = id
self._name = name
self._node = None
self._data = []
Reservoir._id_cnt = max(Reservoir._id_cnt + 1, self.id)
@classmethod
def _db_create(cls, execute):
execute("""
CREATE TABLE reservoir(
id INTEGER NOT NULL PRIMARY KEY,
name TEXT NOT NULL,
node INTEGER,
FOREIGN KEY(node) REFERENCES river_node(id)
)
""")
execute("""
CREATE TABLE reservoir_data(
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
ind INTEGER NOT NULL,
elevation REAL NOT NULL,
surface REAL NOT NULL,
reservoir INTEGER,
FOREIGN KEY(reservoir) REFERENCES reservoir(id)
)
""")
return cls._create_submodel(execute)
@classmethod
def _db_update(cls, execute, version):
major, minor, release = version.strip().split(".")
if major == minor == "0":
if int(release) < 5:
cls._db_create(execute)
return True
@classmethod
def _db_load(cls, execute, data=None):
new = []
table = execute(
"SELECT id, name, node " +
"FROM reservoir "
)
for row in table:
id = row[0]
name = row[1]
node_id = row[2]
new_reservoir = cls(id, name, status=data["status"])
new_reservoir._node = None
if node_id != -1:
new_reservoir._node = next(filter(lambda n: n.id == node_id, data["nodes"]))
new_data = []
table = execute(
"SELECT elevation, surface " +
"FROM reservoir_data " +
f"WHERE reservoir = {id} " +
"ORDER BY ind ASC"
)
for t in table:
new_data.append((t[0], t[1]))
new_reservoir._data = new_data
new.append(new_reservoir)
return new
def _db_save(self, execute, data=None):
execute(f"DELETE FROM reservoir WHERE id = {self.id}")
execute(f"DELETE FROM reservoir_data WHERE reservoir = {self.id}")
node_id = -1
if self._node is not None:
node_id = self._node.id
sql = (
"INSERT INTO " +
"reservoir(id, name, node) " +
"VALUES (" +
f"{self.id}, '{self._db_format(self._name)}', " +
f"{node_id}" +
")"
)
execute(sql)
ind = 0
for d in self._data:
sql = (
"INSERT INTO " +
"reservoir_data(ind, elevation, surface, reservoir) " +
f"VALUES ({ind}, '{d[0]}', {d[1]}, {self.id})"
)
execute(sql)
ind += 1
return True
def __len__(self):
return len(self._data)
@property
def name(self):
return self._name
@name.setter
def name(self, name):
self._name = name
self._status.modified()
@property
def node(self):
return self._node
@node.setter
def node(self, node):
self._node = node
self._status.modified()
def has_node(self):
return self._node is not None
@property
def data(self):
return self._data.copy()
@property
def _default_elevation(self):
return 0.0
@property
def _default_surface(self):
return 0.0
def is_define(self):
return len(self._data) != 0
def new_from_data(self, data):
try:
new_0 = float(data[0])
new_1 = float(data[1])
except Exception as e:
logger.error(e)
new_0 = None
new_1 = None
return (new_0, new_1)
def add(self, index: int):
value = (self._default_elevation, self._default_surface)
self._data.insert(index, value)
self._status.modified()
return value
def insert(self, index: int, value):
self._data.insert(index, value)
self._status.modified()
def delete_i(self, indexes):
self._data = list(
map(
lambda e: e[1],
filter(
lambda e: e[0] not in indexes,
enumerate(self.data)
)
)
)
self._status.modified()
def delete(self, els):
self._data = list(
filter(
lambda e: e not in els,
self.data
)
)
self._status.modified()
def sort(self, _reverse=False, key=None):
if key is None:
self._data.sort(reverse=_reverse)
else:
self._data.sort(reverse=_reverse, key=key)
self._status.modified()
def get_i(self, index):
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, column, value):
v = list(self._data[index])
v[column] = value
self._data[index] = tuple(v)
self._status.modified()
def set_i_elevation(self, index: int, value):
self._set_i_c_v(index, 0, value)
def set_i_surface(self, index: int, value):
self._set_i_c_v(index, 1, value)
def move_up(self, index):
if index < len(self):
next = index - 1
d = self._data
d[index], d[next] = d[next], d[index]
self._status.modified()
def move_down(self, index):
if index >= 0:
prev = index + 1
d = self._data
d[index], d[prev] = d[prev], d[index]
self._status.modified()
+87
View File
@@ -0,0 +1,87 @@
# ReservoirList.py -- Pamhyr
# Copyright (C) 2023 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 copy import copy
from tools import trace, timer
from Model.Tools.PamhyrList import PamhyrModelList
from Model.Reservoir.Reservoir import Reservoir
class ReservoirList(PamhyrModelList):
_sub_classes = [
Reservoir,
]
@classmethod
def _db_load(cls, execute, data=None):
new = cls(status=data['status'])
new._lst = Reservoir._db_load(
execute, data
)
return new
def _db_save(self, execute, data=None):
execute("DELETE FROM reservoir")
execute("DELETE FROM reservoir_data")
if data is None:
data = {}
for reservoir in self._lst:
reservoir._db_save(execute, data=data)
return True
def new(self, index):
r = Reservoir(status=self._status)
self._lst.insert(index, r)
self._status.modified()
return r
def __copy__(self):
new = ReservoirList()
new._lst = self._lst.copy()
return new
def __deepcopy__(self):
new = ReservoirList()
new._lst = self._lst.deepcopy()
return new
def copy(self):
return copy(self)
def get_assoc_to_node(self, node):
assoc = list(
filter(
lambda i: i.node is node,
self._lst
)
)
if len(assoc) > 0:
return assoc[0]
return None
+14 -1
View File
@@ -24,9 +24,15 @@ from Model.Results.River.River import River
logger = logging.getLogger()
class Results(object):
def __init__(self, study = None):
def __init__(self, study=None, solver=None,
repertory="", name="0"):
self._study = study
self._solver = solver
self._repertory = repertory
self._name = name
self._river = River(self._study)
self._meta_data = {
@@ -52,3 +58,10 @@ class Results(object):
def get(self, key):
return self._meta_data[key]
def reload(self):
return self._solver.results(
self._study,
self._repertory,
qlog=None,
)
+6 -3
View File
@@ -20,11 +20,12 @@ from datetime import datetime
logger = logging.getLogger()
class Profile(object):
def __init__(self, profile, study):
self._study = study
self._profile = profile # Source profile in the study
self._data = {} # Dict of dict {<ts>: {<key>: <value>, ...}, ...}
self._profile = profile # Source profile in the study
self._data = {} # Dict of dict {<ts>: {<key>: <value>, ...}, ...}
def __len__(self):
return len(self._data)
@@ -58,10 +59,11 @@ class Profile(object):
def get_ts_key(self, timestamp, key):
return self._data[timestamp][key]
class Reach(object):
def __init__(self, reach, study):
self._study = study
self._reach = reach # Source reach in the study
self._reach = reach # Source reach in the study
self._profiles = list(
map(
lambda p: Profile(p, self._study),
@@ -90,6 +92,7 @@ class Reach(object):
def set(self, profile_id, timestamp, key, data):
self._profiles[profile_id].set(timestamp, key, data)
class River(object):
def __init__(self, study):
self._study = study
+65 -47
View File
@@ -28,31 +28,35 @@ from Model.Geometry.Profile import Profile
from Model.Geometry.Reach import Reach
from Model.BoundaryCondition.BoundaryConditionList import BoundaryConditionList
from Model.LateralContribution.LateralContributionList import LateralContributionList
from Model.LateralContribution.LateralContributionList import (
LateralContributionList
)
from Model.InitialConditions.InitialConditionsDict import InitialConditionsDict
from Model.Stricklers.StricklersList import StricklersList
from Model.Friction.FrictionList import FrictionList
from Model.SolverParameters.SolverParametersList import SolverParametersList
from Model.SedimentLayer.SedimentLayerList import SedimentLayerList
from Model.Reservoir.ReservoirList import ReservoirList
from Solver.Solvers import solver_type_list
class RiverNode(Node, SQLSubModel):
_sub_classes = []
def __init__(self, id:str, name:str,
x:float, y:float,
status = None):
def __init__(self, id: str, name: str,
x: float, y: float,
status=None):
super(RiverNode, self).__init__(
id, name,
x, y,
status = status
status=status
)
self._locker = None
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
execute("""
CREATE TABLE river_node(
id INTEGER NOT NULL PRIMARY KEY,
@@ -66,11 +70,11 @@ class RiverNode(Node, SQLSubModel):
return True
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
return True
@classmethod
def _sql_load(cls, execute, data = None):
def _db_load(cls, execute, data=None):
nodes = []
table = execute("SELECT id, name, x, y FROM river_node")
@@ -82,11 +86,11 @@ class RiverNode(Node, SQLSubModel):
return nodes
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
sql = (
"INSERT OR REPLACE INTO river_node(id, name, x, y) VALUES (" +
f"{self.id}, '{self._sql_format(self.name)}', " +
f"{self.x}, {self.y}"+
f"{self.id}, '{self._db_format(self.name)}', " +
f"{self.x}, {self.y}" +
")"
)
execute(sql)
@@ -108,21 +112,21 @@ class RiverReach(Edge, SQLSubModel):
FrictionList,
]
def __init__(self, id:str, name:str,
node1:RiverNode = None,
node2:RiverNode = None,
status = None):
def __init__(self, id: str, name: str,
node1: RiverNode = None,
node2: RiverNode = None,
status=None):
super(RiverReach, self).__init__(
id, name,
node1, node2,
status = status
status=status
)
self._reach = Reach(status=self._status, parent=self)
self._frictions = FrictionList(status = self._status)
self._frictions = FrictionList(status=self._status)
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
execute("""
CREATE TABLE river_reach(
id INTEGER NOT NULL PRIMARY KEY,
@@ -139,17 +143,18 @@ class RiverReach(Edge, SQLSubModel):
return True
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
return cls._update_submodel(execute, version)
@classmethod
def _sql_load(cls, execute, data = None):
def _db_load(cls, execute, data=None):
reachs = []
if data is None:
data = {}
table = execute("SELECT id, name, enable, node1, node2 FROM river_reach")
table = execute(
"SELECT id, name, enable, node1, node2 FROM river_reach")
for row in table:
# Update id counter
cls._id_cnt = max(cls._id_cnt, row[0])
@@ -161,27 +166,27 @@ class RiverReach(Edge, SQLSubModel):
node1 = next(filter(lambda n: n.id == row[3], data["nodes"]))
node2 = next(filter(lambda n: n.id == row[4], data["nodes"]))
new = cls(id, name, node1, node2, status = data["status"])
new.enable(enable = enable)
new = cls(id, name, node1, node2, status=data["status"])
new.enable(enable=enable)
data["reach"] = id
data["parent"] = new
new._reach = Reach._sql_load(execute, data)
new._reach = Reach._db_load(execute, data)
new._frictions = FrictionList._sql_load(execute, data)
new._frictions = FrictionList._db_load(execute, data)
reachs.append(new)
return reachs
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
sql = (
"INSERT OR REPLACE INTO " +
"river_reach(id, name, enable, node1, node2) "+
"river_reach(id, name, enable, node1, node2) " +
"VALUES (" +
f"{self.id}, '{self._sql_format(self._name)}', " +
f"{self._sql_format(self.is_enable())},"
f"{self.node1.id}, {self.node2.id}"+
f"{self.id}, '{self._db_format(self._name)}', " +
f"{self._db_format(self.is_enable())},"
f"{self.node1.id}, {self.node2.id}" +
")"
)
execute(sql)
@@ -202,6 +207,7 @@ class RiverReach(Edge, SQLSubModel):
def frictions(self):
return self._frictions
class River(Graph, SQLSubModel):
_sub_classes = [
RiverNode,
@@ -212,6 +218,7 @@ class River(Graph, SQLSubModel):
StricklersList,
SolverParametersList,
SedimentLayerList,
ReservoirList,
]
def __init__(self, status=None):
@@ -223,87 +230,95 @@ class River(Graph, SQLSubModel):
self._current_reach = None
self._boundary_condition = BoundaryConditionList(status=self._status)
self._lateral_contribution = LateralContributionList(status=self._status)
self._lateral_contribution = LateralContributionList(
status=self._status)
self._initial_conditions = InitialConditionsDict(status=self._status)
self._stricklers = StricklersList(status=self._status)
self._parameters = {}
self._sediment_layers = SedimentLayerList(status=self._status)
self._reservoir = ReservoirList(status=self._status)
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
cls._create_submodel(execute)
return True
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
cls._update_submodel(execute, version)
return True
@classmethod
def _sql_load(cls, execute, data = None):
def _db_load(cls, execute, data=None):
new = cls(data["status"])
# Stricklers (Stricklers is load in first because it's needed
# for reachs)
new._stricklers = StricklersList._sql_load(
new._stricklers = StricklersList._db_load(
execute,
data
)
data["stricklers"] = new._stricklers
# Initial conditions
new._sediment_layers = SedimentLayerList._sql_load(
new._sediment_layers = SedimentLayerList._db_load(
execute,
data
)
data["sediment_layers_list"] = new._sediment_layers
# Network
new._nodes = RiverNode._sql_load(
new._nodes = RiverNode._db_load(
execute,
data
)
data["nodes"] = new.nodes()
new._edges = RiverReach._sql_load(
new._edges = RiverReach._db_load(
execute,
data
)
data["edges"] = new.edges()
# Boundary Condition
new._boundary_condition = BoundaryConditionList._sql_load(
new._boundary_condition = BoundaryConditionList._db_load(
execute,
data
)
# Lateral Contribution
new._lateral_contribution = LateralContributionList._sql_load(
new._lateral_contribution = LateralContributionList._db_load(
execute,
data
)
# Initial conditions
new._initial_conditions = InitialConditionsDict._sql_load(
new._initial_conditions = InitialConditionsDict._db_load(
execute,
data
)
# Reservoir
new._reservoir = ReservoirList._db_load(
execute,
data
)
# Parameters
new._parameters = SolverParametersList._sql_load(
new._parameters = SolverParametersList._db_load(
execute,
data
)
return new
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
objs = (self._nodes + self._edges)
objs.append(self._boundary_condition)
objs.append(self._initial_conditions)
objs.append(self._lateral_contribution)
objs.append(self._sediment_layers)
objs.append(self._stricklers)
objs.append(self._reservoir)
for solver in self._parameters:
objs.append(self._parameters[solver])
@@ -344,6 +359,9 @@ class River(Graph, SQLSubModel):
return ret[0]
@property
def reservoir(self):
return self._reservoir
@property
def parameters(self):
@@ -354,8 +372,8 @@ class River(Graph, SQLSubModel):
return self._parameters[solver]
new = SolverParametersList(
solver_type = solver_type_list[solver],
status = self._status
solver_type=solver_type_list[solver],
status=self._status
)
self._parameters[solver] = new
self._status.modified()
@@ -381,7 +399,7 @@ class River(Graph, SQLSubModel):
has = len(self._sediment_layers) != 0
has &= any(
filter(
lambda p: p.sl != None,
lambda p: p.sl is not None,
flatten(
map(lambda e: e.reach.profiles, self.edges())
)
+1
View File
@@ -20,6 +20,7 @@ import logging
logger = logging.getLogger()
class SavedStatus(object):
def __init__(self):
super(SavedStatus, self).__init__()
+46 -43
View File
@@ -5,16 +5,17 @@ from tools import trace, timer
from Model.Tools.PamhyrDB import SQLSubModel
from Model.Except import NotImplementedMethodeError
class Layer(SQLSubModel):
_sub_classes = []
_id_cnt = 0
def __init__(self,
id:int = -1, name:str = "",
type = "",
height = 0.0, d50 = 0.0, sigma = 0.0,
critical_constraint = 0.0,
sl = None, status = None):
id: int = -1, name: str = "",
type="",
height=0.0, d50=0.0, sigma=0.0,
critical_constraint=0.0,
sl=None, status=None):
super(Layer, self).__init__()
self._status = status
@@ -83,7 +84,7 @@ class Layer(SQLSubModel):
self._critical_constraint = float(critical_constraint)
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
execute("""
CREATE TABLE sedimentary_layer_layer(
id INTEGER NOT NULL PRIMARY KEY,
@@ -102,21 +103,22 @@ class Layer(SQLSubModel):
return cls._create_submodel(execute)
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
major, minor, release = version.strip().split(".")
if major == minor == "0":
if int(release) < 2:
cls._sql_create(execute)
cls._db_create(execute)
return True
@classmethod
def _sql_load(cls, execute, data = None):
def _db_load(cls, execute, data=None):
new = []
sl = data["sl"]
table = execute(
"SELECT id, ind, name, type, height, d50, sigma, critical_constraint " +
"SELECT id, ind, name, type, height, " +
"d50, sigma, critical_constraint " +
"FROM sedimentary_layer_layer " +
f"WHERE sl = {sl}"
)
@@ -128,27 +130,28 @@ class Layer(SQLSubModel):
ind = row[1]
layer = cls(
id = row[0], name = row[2],
type = row[3], height = row[4],
d50 = row[5], sigma = row[6],
critical_constraint = row[7],
sl = sl, status = data['status']
id=row[0], name=row[2],
type=row[3], height=row[4],
d50=row[5], sigma=row[6],
critical_constraint=row[7],
sl=sl, status=data['status']
)
new[ind] = layer
return new
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
ind = data["ind"]
sl = data["sl"]
sql = (
"INSERT INTO " +
"sedimentary_layer_layer(id, ind, name, type, height, d50, sigma, critical_constraint, sl) "+
"sedimentary_layer_layer(id, ind, name, type, height, " +
"d50, sigma, critical_constraint, sl) " +
"VALUES (" +
f"{self.id}, {ind}, '{self._sql_format(self._name)}', " +
f"'{self._sql_format(self._type)}', {self._height}, " +
f"{self.id}, {ind}, '{self._db_format(self._name)}', " +
f"'{self._db_format(self._type)}', {self._height}, " +
f"{self._d50}, {self._sigma}, {self._critical_constraint}, " +
f"{sl.id}" +
")"
@@ -162,9 +165,9 @@ class SedimentLayer(SQLSubModel):
_sub_classes = [Layer]
_id_cnt = 0
def __init__(self, id:int = -1,
name:str = "", comment:str = "",
status = None):
def __init__(self, id: int = -1,
name: str = "", comment: str = "",
status=None):
super(SedimentLayer, self).__init__()
self._status = status
@@ -196,7 +199,7 @@ class SedimentLayer(SQLSubModel):
def height(self):
return list(
map(lambda l: l.height, self._layers)
map(lambda layer: layer.height, self._layers)
)
@property
@@ -209,7 +212,7 @@ class SedimentLayer(SQLSubModel):
def names(self):
return list(
map(lambda l: l.name, self._layers)
map(lambda layer: layer.name, self._layers)
)
@property
@@ -221,7 +224,7 @@ class SedimentLayer(SQLSubModel):
self._comment = comment
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
execute("""
CREATE TABLE sedimentary_layer(
id INTEGER NOT NULL PRIMARY KEY,
@@ -233,16 +236,16 @@ class SedimentLayer(SQLSubModel):
return cls._create_submodel(execute)
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
major, minor, release = version.strip().split(".")
if major == minor == "0":
if int(release) < 2:
cls._sql_create(execute)
cls._db_create(execute)
return True
@classmethod
def _sql_load(cls, execute, data = None):
def _db_load(cls, execute, data=None):
new = []
table = execute(
@@ -252,36 +255,36 @@ class SedimentLayer(SQLSubModel):
for row in table:
sl = cls(
id = row[0],
name = row[1],
comment = row[2],
status = data['status']
id=row[0],
name=row[1],
comment=row[2],
status=data['status']
)
data["sl"] = sl.id
sl._layers = Layer._sql_load(execute, data)
sl._layers = Layer._db_load(execute, data)
new.append(sl)
return new
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
if data is None:
data = {}
sql = (
"INSERT INTO sedimentary_layer (id, name, comment) " +
f"VALUES ({self.id}, '{self._sql_format(self._name)}', " +
f"'{self._sql_format(self._comment)}')"
f"VALUES ({self.id}, '{self._db_format(self._name)}', " +
f"'{self._db_format(self._comment)}')"
)
execute(sql)
data["sl"] = self
ind = 0
for l in self._layers:
for layer in self._layers:
data["ind"] = ind
l._sql_save(execute, data)
layer._db_save(execute, data)
ind += 1
return True
@@ -298,7 +301,7 @@ class SedimentLayer(SQLSubModel):
self._status.modified()
def new(self, index):
n = Layer(sl=self, status = self._status)
n = Layer(sl=self, status=self._status)
self.insert(index, n)
self._status.modified()
return n
@@ -324,8 +327,8 @@ class SedimentLayer(SQLSubModel):
if index >= 0:
next = index - 1
l = self._layers
l[index], l[next] = l[next], l[index]
lst = self._layers
lst[index], lst[next] = lst[next], lst[index]
self._status.modified()
@@ -333,7 +336,7 @@ class SedimentLayer(SQLSubModel):
if index + 1 < len(self._layers):
prev = index + 1
l = self._layers
l[index], l[prev] = l[prev], l[index]
lst = self._layers
lst[index], lst[prev] = lst[prev], lst[index]
self._status.modified()
+7 -6
View File
@@ -6,18 +6,19 @@ from Model.Except import NotImplementedMethodeError
from Model.Tools.PamhyrList import PamhyrModelList
from Model.SedimentLayer.SedimentLayer import SedimentLayer
class SedimentLayerList(PamhyrModelList):
_sub_classes = [SedimentLayer]
@classmethod
def _sql_load(cls, execute, data = None):
new = cls(status = data["status"])
def _db_load(cls, execute, data=None):
new = cls(status=data["status"])
new._lst = SedimentLayer._sql_load(execute, data)
new._lst = SedimentLayer._db_load(execute, data)
return new
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
ok = True
# Delete previous data
@@ -25,7 +26,7 @@ class SedimentLayerList(PamhyrModelList):
execute("DELETE FROM sedimentary_layer_layer")
for sl in self._lst:
ok &= sl._sql_save(execute, data)
ok &= sl._db_save(execute, data)
return ok
@@ -34,7 +35,7 @@ class SedimentLayerList(PamhyrModelList):
return self.lst
def new(self, index):
n = SedimentLayer(status = self._status)
n = SedimentLayer(status=self._status)
self.insert(index, n)
self._status.modified()
return n
+1
View File
@@ -18,6 +18,7 @@
import pickle
class Serializable():
def __init__(self):
return
@@ -22,10 +22,11 @@ from Solver.Solvers import solver_type_list
from Model.Tools.PamhyrList import PamhyrModelList
class Parameter():
def __init__(self,
name = "", value = "",
status = None):
name="", value="",
status=None):
self._status = status
self._name = name
@@ -57,14 +58,15 @@ class Parameter():
@classmethod
def from_tuple(cls, data, status):
new = cls(status = status)
new = cls(status=status)
new["name"] = data[0]
new["value"] = data[1]
return new
class SolverParametersList(PamhyrModelList):
def __init__(self, solver_type = None, status = None):
def __init__(self, solver_type=None, status=None):
super(SolverParametersList, self).__init__()
self._status = status
@@ -76,9 +78,8 @@ class SolverParametersList(PamhyrModelList):
)
)
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
execute("""
CREATE TABLE solver_parameter(
id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT,
@@ -92,40 +93,47 @@ class SolverParametersList(PamhyrModelList):
return cls._create_submodel(execute)
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
major, minor, release = version.strip().split(".")
if major == minor == "0":
if int(release) < 3:
execute(f"UPDATE solver_parameter SET name='mage_implicitation' WHERE name='mage_implication'")
execute(f"UPDATE solver_parameter SET name='mage_iteration_type' WHERE name='mage_iter_type'")
execute(
"UPDATE solver_parameter SET name='mage_implicitation' " +
"WHERE name='mage_implication'"
)
execute(
"UPDATE solver_parameter SET name='mage_iteration_type' " +
"WHERE name='mage_iter_type'"
)
if int(release) < 4:
solvers = execute("SELECT DISTINCT solver FROM solver_parameter")
solvers = execute(
"SELECT DISTINCT solver FROM solver_parameter")
for solver in solvers:
if solver != "mage8":
continue
data = execute("SELECT ind, name, value " +
"FROM solver_parameter "+
f"WHERE solver = '{solver}' " +
"ORDER BY ind ACS")
"FROM solver_parameter " +
f"WHERE solver = '{solver}' " +
"ORDER BY ind ACS")
execute(
"DELETE FROM solver_parameter " +
f"WHERE solver = '{solver}'"
)
insert = lambda name, value, ind: execute(
"INSERT INTO " +
"solver_parameter(ind, name, value, solver) "+
"VALUES (" +
f"{ind}, " +
f"'{cls._sql_format(name)}', " +
f"'{cls._sql_format(value)}', " +
f"'{cls._sql_format(solver)}'" +
")"
def insert(name, value, ind): return execute(
"INSERT INTO " +
"solver_parameter(ind, name, value, solver) " +
"VALUES (" +
f"{ind}, " +
f"'{cls._db_format(name)}', " +
f"'{cls._db_format(value)}', " +
f"'{cls._db_format(solver)}'" +
")"
)
ind = 0
@@ -156,7 +164,7 @@ class SolverParametersList(PamhyrModelList):
return cls._update_submodel(execute, version)
@classmethod
def _sql_load(cls, execute, data = None):
def _db_load(cls, execute, data=None):
status = data["status"]
solvers = execute("SELECT DISTINCT solver FROM solver_parameter")
new = {}
@@ -167,7 +175,7 @@ class SolverParametersList(PamhyrModelList):
continue
st = solver_type_list[solver]
n = cls(solver_type = st, status = status)
n = cls(solver_type=st, status=status)
table = execute(
"SELECT ind, name, value " +
@@ -186,7 +194,7 @@ class SolverParametersList(PamhyrModelList):
return new
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
t = self._solver._type
execute(
"DELETE FROM solver_parameter " +
@@ -197,12 +205,12 @@ class SolverParametersList(PamhyrModelList):
for param in self._lst:
sql = (
"INSERT INTO " +
"solver_parameter(ind, name, value, solver) "+
"solver_parameter(ind, name, value, solver) " +
"VALUES (" +
f"{ind}, " +
f"'{self._sql_format(param.name)}', " +
f"'{self._sql_format(param.value)}', " +
f"'{self._sql_format(t)}'" +
f"'{self._db_format(param.name)}', " +
f"'{self._db_format(param.value)}', " +
f"'{self._db_format(t)}'" +
")"
)
execute(sql)
@@ -225,7 +233,7 @@ class SolverParametersList(PamhyrModelList):
self._lst
)
)["value"]
except:
except Exception:
return None
def set(self, index, new):
@@ -241,15 +249,15 @@ class SolverParametersList(PamhyrModelList):
self._lst.append(
Parameter(
name = key,
value = value,
status = self._status
name=key,
value=value,
status=self._status
)
)
self._status.modified()
def new(self, index):
n = Parameter(status = self._status)
n = Parameter(status=self._status)
self._lst.insert(index, n)
self._status.modified()
return n
+19 -19
View File
@@ -20,16 +20,17 @@ from tools import trace, timer
from Model.Tools.PamhyrDB import SQLSubModel
class Stricklers(SQLSubModel):
_id_cnt = 0
_sub_classes = []
def __init__(self, id:int = -1,
name:str = "",
comment:str = "",
minor:float = 35.0,
medium:float = 15.0,
status = None):
def __init__(self, id: int = -1,
name: str = "",
comment: str = "",
minor: float = 35.0,
medium: float = 15.0,
status=None):
super(Stricklers, self).__init__()
self._status = status
@@ -46,9 +47,8 @@ class Stricklers(SQLSubModel):
self._minor = minor
self._medium = medium
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
execute("""
CREATE TABLE stricklers(
id INTEGER NOT NULL PRIMARY KEY,
@@ -62,11 +62,11 @@ class Stricklers(SQLSubModel):
return cls._create_submodel(execute)
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
return cls._update_submodel(execute, version)
@classmethod
def _sql_load(cls, execute, data = None):
def _db_load(cls, execute, data=None):
stricklers = []
status = data["status"]
@@ -86,25 +86,25 @@ class Stricklers(SQLSubModel):
medium = row[4]
new = cls(
id = id,
name = name,
comment = comment,
minor = minor, medium = medium,
status = status
id=id,
name=name,
comment=comment,
minor=minor, medium=medium,
status=status
)
stricklers.append(new)
return stricklers
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
sql = (
"INSERT INTO " +
"stricklers(id, name, comment, minor, medium) "+
"stricklers(id, name, comment, minor, medium) " +
"VALUES (" +
f"{self.id}, " +
f"'{self._sql_format(self.name)}', " +
f"'{self._sql_format(self.comment)}', " +
f"'{self._db_format(self.name)}', " +
f"'{self._db_format(self.comment)}', " +
f"{float(self.minor)}, {float(self.medium)}" +
")"
)
+16 -13
View File
@@ -22,31 +22,32 @@ from Model.Saved import SavedStatus
from Model.Tools.PamhyrList import PamhyrModelList
from Model.Stricklers.Stricklers import Stricklers
class StricklersList(PamhyrModelList):
_sub_classes = [
Stricklers,
]
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
return cls._create_submodel(execute)
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
return cls._update_submodel(execute, version)
@classmethod
def _sql_load(cls, execute, data = None):
new = cls(status = data["status"])
def _db_load(cls, execute, data=None):
new = cls(status=data["status"])
new._lst = Stricklers._sql_load(
new._lst = Stricklers._db_load(
execute,
data = data
data=data
)
return new
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
execute("DELETE FROM stricklers")
objs = self._lst
@@ -57,19 +58,21 @@ class StricklersList(PamhyrModelList):
return self.lst
def new(self, index):
s = Stricklers(status = self._status)
s = Stricklers(status=self._status)
self.insert(index, s)
return s
@timer
def sort(self, reverse:bool = False, key=None):
f = lambda st: st.name
def sort(self, reverse: bool = False, key=None):
def f(st): return st.name
if key is not None:
f = key
self._lst = sorted(
self._lst,
key = f,
reverse = reverse,
key=f,
reverse=reverse,
)
self._status.modified()
if self._status is not None:
self._status.modified()
+73 -28
View File
@@ -33,20 +33,21 @@ from Checker.Study import *
logger = logging.getLogger()
class Study(SQLModel):
_sub_classes = [
River,
]
def __init__(self, filename = None, init_new = True):
def __init__(self, filename=None, init_new=True):
# Metadata
self._version = "0.0.4"
self._version = "0.0.5"
self.creation_date = datetime.now()
self.last_modification_date = datetime.now()
self.last_save_date = datetime.now()
self._filename = filename
super(Study, self).__init__(filename = filename)
super(Study, self).__init__(filename=filename)
self.status = SavedStatus()
@@ -59,9 +60,9 @@ class Study(SQLModel):
if init_new:
# Study data
self._river = River(status = self.status)
self._river = River(status=self.status)
else:
self._init_db_file(filename, is_new = False)
self._init_db_file(filename, is_new=False)
self._old_save_id = 0
@@ -128,7 +129,7 @@ class Study(SQLModel):
@filename.setter
def filename(self, filename):
self._filename = str(filename)
self._init_db_file(filename, is_new = True)
self._init_db_file(filename, is_new=True)
self.status.modified()
@property
@@ -139,7 +140,7 @@ class Study(SQLModel):
self._time_system = "time"
self.status.modified()
def use_date(self, date:datetime):
def use_date(self, date: datetime):
self._time_system = "date"
self._date = date
self.status.modified()
@@ -158,7 +159,7 @@ class Study(SQLModel):
# return cls()
@classmethod
def new(cls, name, description, date = None):
def new(cls, name, description, date=None):
me = cls()
me.name = name
me.description = description
@@ -176,16 +177,26 @@ class Study(SQLModel):
# SQL #
#######
def _db_insert_into_info(self, key, value, commit=False):
self.execute(
"INSERT INTO info VALUES " +
f"('{key}', '{self._db_format(value)}')",
commit=commit
)
def _create(self):
# Info (metadata)
self.execute("CREATE TABLE info(key TEXT NOT NULL UNIQUE, value TEXT NOT NULL)")
self.execute(
f"INSERT INTO info VALUES ('version', '{self._sql_format(self._version)}')",
commit = True
"CREATE TABLE info(key TEXT NOT NULL UNIQUE, value TEXT NOT NULL)")
self.execute(
"INSERT INTO info VALUES ('version', " +
f"'{self._db_format(self._version)}')",
commit=True
)
self.execute("INSERT INTO info VALUES ('name', '')")
self.execute("INSERT INTO info VALUES ('description', '')")
self.execute(f"INSERT INTO info VALUES ('time_system', '{self._time_system}')")
self.execute(
f"INSERT INTO info VALUES ('time_system', '{self._time_system}')")
self.execute(
f"INSERT INTO info VALUES ('date', " +
"'{self._date.timestamp()}')"
@@ -213,7 +224,8 @@ class Study(SQLModel):
logger.info("Update database")
if self._update_submodel(version[0]):
self.execute(f"UPDATE info SET value='{self._version}' WHERE key='version'")
self.execute(
f"UPDATE info SET value='{self._version}' WHERE key='version'")
return True
logger.info("TODO: update failed")
@@ -221,42 +233,75 @@ class Study(SQLModel):
@classmethod
def _load(cls, filename):
new = cls(init_new = False, filename = filename)
new = cls(init_new=False, filename=filename)
# TODO: Load metadata
new.name = new.execute("SELECT value FROM info WHERE key='name'")[0]
new.description = new.execute("SELECT value FROM info WHERE key='description'")[0]
new._time_system = new.execute("SELECT value FROM info WHERE key='time_system'")[0]
new.description = new.execute(
"SELECT value FROM info WHERE key='description'")[0]
new._time_system = new.execute(
"SELECT value FROM info WHERE key='time_system'")[0]
new._date = datetime.fromtimestamp(
float(new.execute("SELECT value FROM info WHERE key='date'")[0])
)
new.creation_date = datetime.fromtimestamp(
float(new.execute("SELECT value FROM info WHERE key='creation_date'")[0])
float(new.execute(
"SELECT value FROM info WHERE key='creation_date'")[0])
)
new.last_save_date = datetime.fromtimestamp(
float(new.execute("SELECT value FROM info WHERE key='last_save_date'")[0])
float(new.execute(
"SELECT value FROM info WHERE key='last_save_date'")[0])
)
# Load river data
new._river = River._sql_load(
new._river = River._db_load(
lambda sql: new.execute(
sql,
fetch_one = False,
commit = True
fetch_one=False,
commit=True
),
data = {"status": new.status}
data={"status": new.status}
)
return new
def _save(self):
self.execute(f"UPDATE info SET value='{self._sql_format(self.name)}' WHERE key='name'")
self.execute(f"UPDATE info SET value='{self._sql_format(self.description)}' WHERE key='description'")
self.execute(f"UPDATE info SET value='{self._time_system}' WHERE key='time_system'")
self.execute(f"UPDATE info SET value='{timestamp(self._date)}' WHERE key='date'")
self.execute(f"UPDATE info SET value='{timestamp(self.creation_date)}' WHERE key='creation_date'")
self.execute(f"UPDATE info SET value='{timestamp(self.last_save_date)}' WHERE key='last_save_date'")
self.execute(
f"UPDATE info SET " +
f"value='{self._db_format(self.name)}' WHERE key='name'"
)
self.execute(
f"UPDATE info SET " +
f"value='{self._db_format(self.description)}' " +
"WHERE key='description'"
)
self.execute(
f"UPDATE info SET " +
f"value='{self._time_system}' WHERE key='time_system'"
)
self.execute(
f"UPDATE info SET " +
f"value='{timestamp(self._date)}' WHERE key='date'"
)
self.execute(
f"UPDATE info SET " +
f"value='{timestamp(self.creation_date)}' " +
"WHERE key='creation_date'"
)
self.execute(
f"UPDATE info SET " +
f"value='{timestamp(self.last_save_date)}' " +
"WHERE key='last_save_date'"
)
self._save_submodel([self._river])
self.commit()
def close(self):
"""Close db connection
Returns:
Nothing.
"""
self._close()
+35 -32
View File
@@ -28,10 +28,12 @@ from Model.Except import NotImplementedMethodeError
logger = logging.getLogger()
# Top level model class
class SQLModel(SQL):
_sub_classes = []
def _init_db_file(self, db, is_new = True):
def _init_db_file(self, db, is_new=True):
exists = Path(db).exists()
if exists and is_new:
@@ -48,19 +50,18 @@ class SQLModel(SQL):
self._update() # Update db scheme if necessary
# self._load() # Load data
def __init__(self, filename = None):
def __init__(self, filename=None):
self._db = None
def _create_submodel(self):
fn = lambda sql: self.execute(
def fn(sql): return self.execute(
sql,
fetch_one = False,
commit = False
fetch_one=False,
commit=False
)
for cls in self._sub_classes:
requests = cls._sql_create(fn)
requests = cls._db_create(fn)
self.commit()
return True
@@ -69,15 +70,15 @@ class SQLModel(SQL):
raise NotImplementedMethodeError(self, self._create)
def _update_submodel(self, version):
fn = lambda sql: self.execute(
def fn(sql): return self.execute(
sql,
fetch_one = False,
commit = False
fetch_one=False,
commit=False
)
ok = True
for cls in self._sub_classes:
ok &= cls._sql_update(fn, version)
ok &= cls._db_update(fn, version)
self.commit()
return ok
@@ -85,16 +86,16 @@ class SQLModel(SQL):
def _update(self):
raise NotImplementedMethodeError(self, self._update)
def _save_submodel(self, objs, data = None):
fn = lambda sql: self.execute(
def _save_submodel(self, objs, data=None):
def fn(sql): return self.execute(
sql,
fetch_one = False,
commit = False
fetch_one=False,
commit=False
)
ok = True
for obj in objs:
ok &= obj._sql_save(fn)
ok &= obj._db_save(fn)
self.commit()
return ok
@@ -103,28 +104,30 @@ class SQLModel(SQL):
raise NotImplementedMethodeError(self, self._save)
@classmethod
def _load(cls, filename = None):
def _load(cls, filename=None):
raise NotImplementedMethodeError(cls, cls._load)
# Sub model class
class SQLSubModel(object):
_sub_classes = []
def _sql_format(self, value):
def _db_format(self, value):
# Replace ''' by '&#39;' to preserve SQL injection
if type(value) == str:
if type(value) is str:
value = value.replace("'", "&#39;")
elif type(value) == bool:
elif type(value) is bool:
value = 'TRUE' if value else 'FALSE'
return value
@classmethod
def _create_submodel(cls, execute):
for sc in cls._sub_classes:
sc._sql_create(execute)
sc._db_create(execute)
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
"""Create data base scheme
Args:
@@ -133,15 +136,15 @@ class SQLSubModel(object):
Returns:
Return true, otherelse false if an issue appear
"""
raise NotImplementedMethodeError(cls, cls._sql_create)
raise NotImplementedMethodeError(cls, cls._db_create)
@classmethod
def _update_submodel(cls, execute, version):
for sc in cls._sub_classes:
sc._sql_update(execute, version)
sc._db_update(execute, version)
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
"""Update data base scheme
Args:
@@ -151,10 +154,10 @@ class SQLSubModel(object):
Returns:
Return true, otherelse false if an issue appear
"""
raise NotImplementedMethodeError(cls, cls._sql_update)
raise NotImplementedMethodeError(cls, cls._db_update)
@classmethod
def _sql_load(cls, execute, data = None):
def _db_load(cls, execute, data=None):
"""Load instance of this class from SQL data base
Args:
@@ -164,13 +167,13 @@ class SQLSubModel(object):
Returns:
Return new instance of class
"""
raise NotImplementedMethodeError(cls, cls._sql_load)
raise NotImplementedMethodeError(cls, cls._db_load)
def _save_submodel(self, execute, objs, data = None):
def _save_submodel(self, execute, objs, data=None):
for o in objs:
o._sql_save(execute, data = data)
o._db_save(execute, data=data)
def _sql_save(self, execute, data = None):
def _db_save(self, execute, data=None):
"""Save class data to data base
Args:
@@ -181,4 +184,4 @@ class SQLSubModel(object):
Return true, otherelse false if an issue appear during
save
"""
raise NotImplementedMethodeError(self, self._sql_save)
raise NotImplementedMethodeError(self, self._db_save)
+14 -8
View File
@@ -16,6 +16,7 @@
# -*- coding: utf-8 -*-
import types
import logging
from copy import copy
@@ -30,7 +31,7 @@ logger = logging.getLogger()
class PamhyrModelDict(SQLSubModel):
_sub_classes = []
def __init__(self, status = None):
def __init__(self, status=None):
super(PamhyrModelDict, self).__init__()
self._status = status
@@ -38,19 +39,19 @@ class PamhyrModelDict(SQLSubModel):
self._dict = {}
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
return cls._create_submodel(execute)
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
return cls._update_submodel(execute, version)
@classmethod
def _sql_load(cls, execute, data = None):
raise NotImplementedMethodeError(cls, cls._sql_load)
def _db_load(cls, execute, data=None):
raise NotImplementedMethodeError(cls, cls._db_load)
def _sql_save(self, execute, data = None):
raise NotImplementedMethodeError(self, self._sql_save)
def _db_save(self, execute, data=None):
raise NotImplementedMethodeError(self, self._db_save)
def __len__(self):
return len(self._dict)
@@ -64,7 +65,12 @@ class PamhyrModelDict(SQLSubModel):
def get(self, key):
if key in self._dict:
return self._dict[key]
v = self._dict[key]
if type(v) == types.GeneratorType:
return list(v)
return v
new = self.new(key)
self.set(key, new)
+44 -32
View File
@@ -26,11 +26,12 @@ from Model.Except import NotImplementedMethodeError
logger = logging.getLogger()
class PamhyrModelList(SQLSubModel):
_sub_classes = [
]
def __init__(self, status = None):
def __init__(self, status=None):
super(PamhyrModelList, self).__init__()
self._status = status
@@ -41,37 +42,38 @@ class PamhyrModelList(SQLSubModel):
#######
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
return cls._create_submodel(execute)
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
return cls._update_submodel(execute, version)
@classmethod
def _sql_load(cls, execute, data = None):
raise NotImplementedMethodeError(cls, cls._sql_load)
def _db_load(cls, execute, data=None):
raise NotImplementedMethodeError(cls, cls._db_load)
def _sql_save(self, execute, data = None):
raise NotImplementedMethodeError(self, self._sql_save)
def _db_save(self, execute, data=None):
raise NotImplementedMethodeError(self, self._db_save)
################
# MODEL METHOD #
################
def __len__(self):
return len(self._lst)
@property
def lst(self):
return self._lst.copy()
return self._lst
def __len__(self):
return len(self.lst)
def get(self, row):
return self._lst[row]
def set(self, row, new):
self._lst[row] = new
self._status.modified()
if self._status is not None:
self._status.modified()
def new(self, index):
"""Create new elements and add it to list
@@ -86,12 +88,16 @@ class PamhyrModelList(SQLSubModel):
def insert(self, index, new):
self._lst.insert(index, new)
self._status.modified()
if self._status is not None:
self._status.modified()
def delete(self, lst):
for el in lst:
self._lst.remove(el)
self._status.modified()
if self._status is not None:
self._status.modified()
def delete_i(self, indexes):
lst = list(
@@ -107,23 +113,29 @@ class PamhyrModelList(SQLSubModel):
def sort(self, reverse=False, key=None):
self._lst.sort(reverse=reverse, key=key)
self._status.modified()
if self._status is not None:
self._status.modified()
def move_up(self, index):
if index < len(self._lst):
next = index - 1
l = self._lst
l[index], l[next] = l[next], l[index]
self._status.modified()
lst = self._lst
lst[index], lst[next] = lst[next], lst[index]
if self._status is not None:
self._status.modified()
def move_down(self, index):
if index >= 0:
prev = index + 1
l = self._lst
l[index], l[prev] = l[prev], l[index]
self._status.modified()
lst = self._lst
lst[index], lst[prev] = lst[prev], lst[index]
if self._status is not None:
self._status.modified()
class PamhyrModelListWithTab(SQLSubModel):
@@ -131,7 +143,7 @@ class PamhyrModelListWithTab(SQLSubModel):
_sub_classes = [
]
def __init__(self, status = None):
def __init__(self, status=None):
super(PamhyrModelListWithTab, self).__init__()
self._status = status
@@ -144,19 +156,19 @@ class PamhyrModelListWithTab(SQLSubModel):
#######
@classmethod
def _sql_create(cls, execute):
def _db_create(cls, execute):
return cls._create_submodel(execute)
@classmethod
def _sql_update(cls, execute, version):
def _db_update(cls, execute, version):
return cls._update_submodel(execute, version)
@classmethod
def _sql_load(cls, execute, data = None):
raise NotImplementedMethodeError(cls, cls._sql_load)
def _db_load(cls, execute, data=None):
raise NotImplementedMethodeError(cls, cls._db_load)
def _sql_save(self, execute, data = None):
raise NotImplementedMethodeError(self, self._sql_save)
def _db_save(self, execute, data=None):
raise NotImplementedMethodeError(self, self._db_save)
################
# MODEL METHOD #
@@ -216,14 +228,14 @@ class PamhyrModelListWithTab(SQLSubModel):
if index < len(self._tabs[lst]):
next = index - 1
l = self._tabs[lst]
l[index], l[next] = l[next], l[index]
lst = self._tabs[lst]
lst[index], lst[next] = lst[next], lst[index]
self._status.modified()
def move_down(self, lst, index):
if index >= 0:
prev = index + 1
l = self._tabs[lst]
l[index], l[prev] = l[prev], l[index]
lst = self._tabs[lst]
lst[index], lst[prev] = lst[prev], lst[index]
self._status.modified()
+17
View File
@@ -0,0 +1,17 @@
# __init__.py -- Pamhyr
# Copyright (C) 2023 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 -*-
+110
View File
@@ -0,0 +1,110 @@
# test_Model.py -- Pamhyr
# Copyright (C) 2023 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 unittest
import tempfile
from Model.Saved import SavedStatus
from Model.Study import Study
from Model.River import River
class StudyTestCase(unittest.TestCase):
def test_create_study(self):
study = Study.new("foo", "bar")
self.assertEqual(study.name, "foo")
self.assertEqual(study.description, "bar")
def test_open_study(self):
study = Study.open("../tests_cases/Enlargement/Enlargement.pamhyr")
self.assertNotEqual(study, None)
self.assertEqual(study.name, "Enlargement")
def test_save_open_study(self):
study = Study.new("foo", "bar")
dir = tempfile.mkdtemp()
f = os.path.join(dir, "foo.pamhyr")
# Save study
study.filename = f
study.save()
study.close()
# Reopen study
study = Study.open(f)
# Check
self.assertNotEqual(study, None)
self.assertEqual(study.name, "foo")
self.assertEqual(study.description, "bar")
def test_create_study_river(self):
study = Study.new("foo", "bar")
self.assertNotEqual(study.river, None)
class RiverTestCase(unittest.TestCase):
def test_create_river(self):
status = SavedStatus()
river = River(status=status)
self.assertNotEqual(river, None)
def test_create_river_nodes(self):
status = SavedStatus()
river = River(status=status)
self.assertNotEqual(river, None)
# Add nodes
n0 = river.add_node()
n1 = river.add_node(x=1.0, y=0.0)
n2 = river.add_node(x=0.0, y=1.0)
# Checks
self.assertEqual(river.nodes_counts(), 3)
nodes = river.nodes()
self.assertEqual(nodes[0], n0)
self.assertEqual(nodes[1], n1)
self.assertEqual(nodes[2], n2)
def test_create_river_edges(self):
status = SavedStatus()
river = River(status=status)
self.assertNotEqual(river, None)
# Add nodes
n0 = river.add_node()
n1 = river.add_node(x=1.0, y=0.0)
n2 = river.add_node(x=0.0, y=1.0)
self.assertEqual(river.nodes_counts(), 3)
# Add edges
e0 = river.add_edge(n0, n1)
e1 = river.add_edge(n1, n2)
# Checks
self.assertEqual(river.edges_counts(), 2)
edges = river.edges()
self.assertEqual(edges[0], e0)
self.assertEqual(edges[1], e1)
+1
View File
@@ -18,6 +18,7 @@
from Model.Except import NotImplementedMethodeError
class AScript(object):
name = ""
description = ""
+38
View File
@@ -0,0 +1,38 @@
# ListSolver.py -- Pamhyr
# Copyright (C) 2023 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 logging
from Scripts.AScript import AScript
logger = logging.getLogger()
class ScriptListSolver(AScript):
name = "ListSolver"
description = "List configured solver(s) for Pamhyr2"
def usage(self):
logger.info(f"Usage : {self._args[0]} {self._args[1]}")
def run(self):
for solver in self._conf.solvers:
print(f"{solver.name:<16} ({solver.type}): {solver.description}")
return 0
+2 -1
View File
@@ -29,6 +29,7 @@ from Model.Geometry.Reach import Reach
logger = logging.getLogger()
class Script3DST(AScript):
name = "3DST"
description = "Display a 3D plot of a river reach from ST file"
@@ -82,7 +83,7 @@ class Script3DST(AScript):
try:
status = SavedStatus()
my_reach = Reach(status = status)
my_reach = Reach(status=status)
my_reach.import_geometry(st_file)
my_reach.compute_guidelines()
+113
View File
@@ -0,0 +1,113 @@
# Run.py -- Pamhyr
# Copyright (C) 2023 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 logging
from queue import Queue
from PyQt5.QtCore import QProcess
from Scripts.AScript import AScript
from Model.Study import Study
logger = logging.getLogger()
class ScriptRun(AScript):
name = "Run"
description = "Run solver on Pamhyr2 a study"
def usage(self):
logger.info(
f"Usage : {self._args[0]} {self._args[1]} " +
"<SOLVER> <STUDY>"
)
def run(self):
if len(self._args) < 4:
return 1
command = self._args[1]
solver_name = self._args[2]
study_file = os.path.abspath(
self._args[3]
)
try:
solver = next(
filter(
lambda solver: solver.name == solver_name,
self._conf.solvers
)
)
except Exception as e:
logger.error(f"No solver found: {e}")
return 2
study = Study.open(study_file)
self._solver = solver
self._study = study
logger.info(
f"Run {solver.name} ({solver.type}) " +
"on study '{study.name}' ({study_file})"
)
# Workdir
workdir = os.path.join(
os.path.dirname(study.filename),
"_PAMHYR_",
study.name.replace(" ", "_"),
solver.name.replace(" ", "_")
)
os.makedirs(workdir, exist_ok=True)
logger.info(f"Set working dir to {workdir}")
# Preparate process
p = QProcess(None)
p.setWorkingDirectory(workdir)
self._q = Queue()
# Export and Run
logger.info(f"~Export~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~")
solver.export(study, workdir, qlog=self._q)
while self._q.qsize() != 0:
s = self._q.get()
logger.info(s)
if command == "run":
logger.info(f"~Run~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~")
solver.run(
study,
process=p,
output_queue=self._q
)
p.waitForFinished()
logger.info(f"~End~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~")
return 0
class ScriptExport(ScriptRun):
name = "Export"
description = "Export Pamhyr2 study for solver"
+341
View File
@@ -0,0 +1,341 @@
#!/bin/env python3
import sys
import importlib.util as imp
import os
import datetime as dt
import lxml.etree as ET
def mage_to_mascaret(filename):
# rep file
with open(filename, encoding = "ISO-8859-1", mode ="r") as rep:
lines = rep.readlines()
for line in lines:
l = line.split()
if (l[0] == 'PAR'):
parfile = l[1]
elif (l[0] == 'NET'):
netfile = l[1]
elif (l[0] == 'HYD'):
hydfile = l[1]
elif (l[0] == 'RUG'):
rugfile = l[1]
elif (l[0] == 'LIM'):
limfile = l[1]
elif (l[0] == 'INI'):
inifile = l[1]
# net file
with open(netfile, encoding = "ISO-8859-1", mode ="r") as net:
lines = net.readlines()
for line in lines:
if (line[0] != '*' and line[0] != '$'):
l = line.split()
biefname = l[0]
stfile = l[3]
break # on n'utilise que le premier bief pour le moment
# geometry file
print("geometrie fie : "+stfile)
wb.init_bief_from_geo_file(stfile, 0, 0)
wb.set_bief_name(biefname)
nb_sect = wb.get_nb_sections()
print("nb sections: "+str(nb_sect))
# hydrogramme file
with open(hydfile, encoding = "ISO-8859-1", mode ="r") as hyd:
HYD = []
lines = hyd.readlines()
for line in lines:
if (line[0] != '*' and line[0] != '$'):
HYD.append([x for x in line.split()])
# friction file
with open(rugfile, encoding = "ISO-8859-1", mode ="r") as lim:
RUG = []
lines = lim.readlines()
for line in lines:
if (line[0] != '*' and line[0] != '$'):
RUG.append([x for x in line.split()])
# tarage file
with open(limfile, encoding = "ISO-8859-1", mode ="r") as tar:
LIM = []
lines = tar.readlines()
for line in lines:
if (line[0] != '*' and line[0] != '$'):
LIM.append([x for x in line.split()])
# cond init file
with open(inifile, encoding = "ISO-8859-1", mode ="r") as ini:
INI = []
lines = ini.readlines()
for line in lines:
if (line[0] != '*' and line[0] != '$'):
INI.append([x for x in line.split()])
# PAR file
with open(parfile, encoding = "ISO-8859-1", mode ="r") as par:
PAR = {}
lines = par.readlines()
for line in lines:
l = line.split()
if l[0] == 'init_time':
t = l[1].split(":")
if len(t)>1:
PAR['init_time'] = int(t[0])*24*3600+int(t[1])*3600+int(t[2])*60+int(t[3])
else:
PAR['init_time'] = int(t[0])*60
elif l[0] == 'final_time':
t = l[1].split(":")
if len(t)>1:
PAR['final_time'] = int(t[0])*24*3600+int(t[1])*3600+int(t[2])*60+int(t[3])
else:
PAR['final_time'] = int(t[0])*60
elif l[0] == 'timestep':
PAR['timestep'] = float(l[1])
elif l[0] == 'timestep_bin':
PAR['timestep_bin'] = float(l[1])
elif l[0] == 'timestep_tra':
PAR['timestep_tra'] = float(l[1])
elif l[0] == 'min_height':
PAR['min_height'] = float(l[1])
#os.chdir(current_dir)
with open("hydrogramme.loi", mode ="w") as hyd:
hyd.write(" S\n")
for line in HYD:
hyd.write(" "+line[0]+" "+line[1]+"\n")
if float(HYD[-1][0]) < PAR['final_time']:
hyd.write(" "+str(PAR['final_time'])+" "+str(float(HYD[-1][1]))+"\n")
with open("limnigramme.loi", mode ="w") as lim:
lim.write(" S\n")
for line in LIM:
lim.write(" "+line[0]+" "+line[1]+"\n")
if float(LIM[-1][0]) < PAR['final_time']:
lim.write(" "+str(PAR['final_time'])+" "+str(float(LIM[-1][1]))+"\n")
with open("init.lig", mode ="w") as lig:
lig.write("RESULTATS CALCUL,DATE : "+dt.datetime.now().strftime("%d/%m/%y %H:%M"+"\n"))
lig.write("FICHIER RESULTAT MASCARET\n")
lig.write("-"*71+"\n")
lig.write(" IMAX = "+f"{len(INI):4d}"+" NBBIEF= 1\n")
lig.write(" I1,I2 = 1 "+str(len(INI))+"\n")
lig.write(" X\n")
for i in range(int(len(INI)/5)+1):
lig.write(" "+"".join([f"{float(INI[j][4]):12.2f}" for j in range(i*5, min(i*5+5, len(INI)))])+"\n")
lig.write(" Z\n")
for i in range(int(len(INI)/5)+1):
lig.write(" "+"".join([f"{float(INI[j][3]):12.3f}" for j in range(i*5, min(i*5+5, len(INI)))])+"\n")
lig.write(" Q\n")
for i in range(int(len(INI)/5)+1):
lig.write(" "+"".join([f"{float(INI[j][2]):12.3f}" for j in range(i*5, min(i*5+5, len(INI)))])+"\n")
lig.write(" FIN\n")
#os.remove("geometrie")
wb.output_bief_mascaret("geometrie")
pk = [wb.get_pk_section(i+1) for i in range(nb_sect)]
print(pk)
print(max(pk))
print(min(pk))
out_name = 'FichierCas.xcas'
root = ET.Element('fichierCas')
pc = ET.SubElement(root, 'parametresCas')
pg = ET.SubElement(pc, 'parametresGeneraux')
ET.SubElement(pg, 'versionCode').text='3'
ET.SubElement(pg, 'code').text = '3'
ET.SubElement(pg, 'fichMotsCles').text = out_name
ET.SubElement(pg, 'dictionaire').text='dico.txt'
ET.SubElement(pg, 'progPrincipal').text='princi.f'
ET.SubElement(pg, 'sauveModele').text='false'
ET.SubElement(pg, 'fichSauvModele').text='mascaret_exp.tmp'
ET.SubElement(pg, 'validationCode').text='false'
ET.SubElement(pg, 'typeValidation').text='1'
ET.SubElement(pg, 'presenceCasiers').text='false'
b = ET.SubElement(pg, 'bibliotheques')
ET.SubElement(b, 'bibliotheque').text='mascaretV5P1.a damoV3P0.a'
pmp = ET.SubElement(pc, 'parametresModelePhysique')
ET.SubElement(pmp, 'perteChargeConf').text='false'
ET.SubElement(pmp, 'compositionLits').text='1'
ET.SubElement(pmp, 'conservFrotVertical').text='false'
ET.SubElement(pmp, 'elevCoteArrivFront').text='0.05'
ET.SubElement(pmp, 'interpolLinStrickler').text='false'
d = ET.SubElement(pmp, 'debordement')
ET.SubElement(d, 'litMajeur').text='false'
ET.SubElement(d, 'zoneStock').text='false'
pn = ET.SubElement(pc, 'parametresNumeriques')
ET.SubElement(pn, 'calcOndeSubmersion').text='false'
ET.SubElement(pn, 'decentrement').text='false'
ET.SubElement(pn, 'froudeLimCondLim').text='1000.0'
ET.SubElement(pn, 'traitImplicitFrot').text='false'
ET.SubElement(pn, 'hauteurEauMini').text=str(PAR['min_height'])
ET.SubElement(pn, 'implicitNoyauTrans').text='true'
ET.SubElement(pn, 'optimisNoyauTrans').text='false'
ET.SubElement(pn, 'perteChargeAutoElargissement').text='false'
ET.SubElement(pn, 'termesNonHydrostatiques').text='false'
ET.SubElement(pn, 'apportDebit').text='0'
ET.SubElement(pn, 'attenuationConvection').text='false'
pt = ET.SubElement(pc, 'parametresTemporels')
ET.SubElement(pt, 'pasTemps').text=str(PAR['timestep'])
ET.SubElement(pt, 'tempsInit').text=str(PAR['init_time'])
ET.SubElement(pt, 'critereArret').text='1'
ET.SubElement(pt, 'nbPasTemps').text='2000'
ET.SubElement(pt, 'tempsMax').text=str(PAR['final_time'])
ET.SubElement(pt, 'pasTempsVar').text='true'
ET.SubElement(pt, 'nbCourant').text='0.8'
ET.SubElement(pt, 'coteMax').text='0.0'
ET.SubElement(pt, 'abscisseControle').text='0.0'
ET.SubElement(pt, 'biefControle').text='1'
pgr = ET.SubElement(pc, 'parametresGeometrieReseau')
g = ET.SubElement(pgr, 'geometrie')
ET.SubElement(g, 'fichier').text='geometrie'
ET.SubElement(g, 'format').text='2'
ET.SubElement(g, 'profilsAbscAbsolu').text='true'
lb = ET.SubElement(pgr, 'listeBranches')
ET.SubElement(lb, 'nb').text='1'
ET.SubElement(lb, 'numeros').text='1'
ET.SubElement(lb, 'abscDebut').text=str(min(pk))
ET.SubElement(lb, 'abscFin').text=str(max(pk))
ET.SubElement(lb, 'numExtremDebut').text='1'
ET.SubElement(lb, 'numExtremFin').text='2'
ln = ET.SubElement(pgr, 'listeNoeuds')
ET.SubElement(ln, 'nb').text='0'
ET.SubElement(ln, 'noeuds')
el = ET.SubElement(pgr, 'extrLibres')
ET.SubElement(el, 'nb').text='2'
ET.SubElement(el, 'num').text='1 2'
ET.SubElement(el, 'numExtrem').text='1 2'
noms = ET.SubElement(el, 'noms')
ET.SubElement(noms, 'string').text='limite1'
ET.SubElement(noms, 'string').text='limite2'
ET.SubElement(el, 'typeCond').text='1 2'
ET.SubElement(el, 'numLoi').text='1 2'
pc2 = ET.SubElement(pc, 'parametresConfluents')
ET.SubElement(pc2, 'nbConfluents').text='0'
ET.SubElement(pc2, 'confluents')
pcm = ET.SubElement(pc, 'parametresPlanimetrageMaillage')
ET.SubElement(pcm, 'methodeMaillage').text='1'
planim = ET.SubElement(pcm, 'planim')
ET.SubElement(planim, 'nbPas').text='101'
ET.SubElement(planim, 'nbZones').text='1'
ET.SubElement(planim, 'valeursPas').text='0.1'
ET.SubElement(planim, 'num1erProf').text='1'
ET.SubElement(planim, 'numDerProf').text=str(nb_sect)
maillage = ET.SubElement(pcm, 'maillage')
ET.SubElement(maillage, 'modeSaisie').text='2'
ET.SubElement(maillage, 'sauvMaillage').text='false'
maillageClavier = ET.SubElement(maillage, 'maillageClavier')
ET.SubElement(maillageClavier, 'nbSections').text='0'
ET.SubElement(maillageClavier, 'nbPlages').text='1'
ET.SubElement(maillageClavier, 'num1erProfPlage').text='1'
ET.SubElement(maillageClavier, 'numDerProfPlage').text=str(nb_sect)
ET.SubElement(maillageClavier, 'pasEspacePlage').text='10.0'
ET.SubElement(maillageClavier, 'nbZones').text='0'
ps = ET.SubElement(pc, 'parametresSingularite')
ET.SubElement(ps, 'nbSeuils').text='0'
pad = ET.SubElement(pc, 'parametresApportDeversoirs')
pc3 = ET.SubElement(pc, 'parametresCalage')
frt = ET.SubElement(pc3, 'frottement')
ET.SubElement(frt, 'loi').text='1'
ET.SubElement(frt, 'nbZone').text=str(len(RUG))
ET.SubElement(frt, 'numBranche').text=' '.join([i[1] for i in RUG])
ET.SubElement(frt, 'absDebZone').text=' '.join([i[2] for i in RUG])
ET.SubElement(frt, 'absFinZone').text=' '.join([i[3] for i in RUG])
ET.SubElement(frt, 'coefLitMin').text=' '.join([i[4] for i in RUG])
ET.SubElement(frt, 'coefLitMaj').text=' '.join([i[5] for i in RUG])
zst = ET.SubElement(pc3, 'zoneStockage')
ET.SubElement(zst, 'nbProfils').text='0'
ET.SubElement(zst, 'numProfil').text='-0'
ET.SubElement(zst, 'limGauchLitMaj').text='-0'
ET.SubElement(zst, 'limDroitLitMaj').text='-0'
plh = ET.SubElement(pc, 'parametresLoisHydrauliques')
ET.SubElement(plh, 'nb').text='2'
lois = ET.SubElement(plh, 'lois')
spl = ET.SubElement(lois, 'structureParametresLoi')
ET.SubElement(spl, 'nom').text='loi_1_hydrogramme'
ET.SubElement(spl, 'type').text='1'
donnees = ET.SubElement(spl, 'donnees')
ET.SubElement(donnees, 'modeEntree').text='1'
ET.SubElement(donnees, 'fichier').text='hydrogramme.loi'
ET.SubElement(donnees, 'uniteTps').text='-0'
ET.SubElement(donnees, 'nbPoints').text='-0'
ET.SubElement(donnees, 'nbDebitsDifferents').text='-0'
spl2 = ET.SubElement(lois, 'structureParametresLoi')
ET.SubElement(spl2, 'nom').text='loi_2_limnigramme'
ET.SubElement(spl2, 'type').text='2'
donnees = ET.SubElement(spl2, 'donnees')
ET.SubElement(donnees, 'modeEntree').text='1'
ET.SubElement(donnees, 'fichier').text='limnigramme.loi'
ET.SubElement(donnees, 'uniteTps').text='-0'
ET.SubElement(donnees, 'nbPoints').text='-0'
ET.SubElement(donnees, 'nbDebitsDifferents').text='-0'
pci = ET.SubElement(pc, 'parametresConditionsInitiales')
re = ET.SubElement(pci, 'repriseEtude')
ET.SubElement(re, 'repriseCalcul').text='false'
le = ET.SubElement(pci, 'ligneEau')
ET.SubElement(le, 'LigEauInit').text='true'
ET.SubElement(le, 'modeEntree').text='1'
ET.SubElement(le, 'fichLigEau').text='init.lig'
ET.SubElement(le, 'formatFichLig').text='2'
ET.SubElement(le, 'nbPts').text='-0'
pir = ET.SubElement(pc, 'parametresImpressionResultats')
ET.SubElement(pir, 'titreCalcul').text='Etude hydraulique1d'
impression = ET.SubElement(pir, 'impression')
ET.SubElement(impression, 'impressionGeometrie').text='false'
ET.SubElement(impression, 'impressionPlanimetrage').text='false'
ET.SubElement(impression, 'impressionReseau').text='false'
ET.SubElement(impression, 'impressionLoiHydraulique').text='false'
ET.SubElement(impression, 'impressionligneEauInitiale').text='false'
ET.SubElement(impression, 'impressionCalcul').text='false'
ps = ET.SubElement(pir, 'pasStockage')
ET.SubElement(ps, 'premPasTpsStock').text='1'
ET.SubElement(ps, 'pasStock').text=str(int(PAR['timestep_bin']/PAR['timestep']))
ET.SubElement(ps, 'pasImpression').text=str(int(PAR['timestep_tra']/PAR['timestep']))
results = ET.SubElement(pir, 'resultats')
ET.SubElement(results, 'fichResultat').text='mascaret_exp.opt'
ET.SubElement(results, 'postProcesseur').text='2'
listing = ET.SubElement(pir, 'listing')
ET.SubElement(listing, 'fichListing').text='mascaret_exp.lis'
reprise = ET.SubElement(pir, 'fichReprise')
ET.SubElement(reprise, 'fichRepriseEcr').text='mascaret_exp.rep'
rub = ET.SubElement(pir, 'rubens')
ET.SubElement(rub, 'ecartInterBranch').text='1.0'
stockage = ET.SubElement(pir, 'stockage')
ET.SubElement(stockage, 'option').text='1'
ET.SubElement(stockage, 'nbSite').text='0'
pvc = ET.SubElement(pc, 'parametresVariablesCalculees')
ET.SubElement(pvc, 'variablesCalculees').text="false false false false false false false false false false false false false false false"
pvs = ET.SubElement(pc, 'parametresVariablesStockees')
ET.SubElement(pvs, 'variablesStockees').text="true false false false false true true true false false false false false false false false false false false true false false true false false false false false false false false false false false false false false false false false false false"
tree = ET.ElementTree(root)
xmlstr = ET.tostring(root, encoding='unicode', pretty_print=True)
tree.write(out_name)
print("fin du programme")
parser = ET.XMLParser(remove_blank_text=True)
tree = ET.parse(out_name, parser)
tree.write(out_name, pretty_print=True)
if __name__ == "__main__":
current_dir = os.getcwd()
os.chdir('/home/theophile.terraz/Codes/mage/src')
spec = imp.spec_from_file_location("wb", "./wrapper_bief.py")
wb = imp.module_from_spec(spec)
spec.loader.exec_module(wb)
os.chdir(current_dir)
print( 'Number of arguments:', len(sys.argv), 'arguments.')
print( 'Argument List:', str(sys.argv))
mage_to_mascaret(sys.argv[1])
+99
View File
@@ -0,0 +1,99 @@
#!/bin/env python3
import sys
import importlib.util as imp
import os
import numpy as np
def lire_opt(filename, res):
with open(filename, encoding = "ISO-8859-1", mode ="r") as f:
variables = ['"t"','"Bief"','"Section"','"Pk"']
lines = f.readlines()
for start,line in enumerate(lines):
print("line n "+str(start)+" : ",line)
if line == "[variables]\n": continue
if line == "[resultats]\n": break
val = line.split(';')
variables.append(val[1])
if val[1] == '"ZREF"': izref = start+3
if val[1] == '"Q"': iq = start+3
if val[1] == '"Z"': iz = start+3
print ("variables : "+str(variables))
#resultats
res.mage_version = 83
# first pass to count sect and biefs
prev_timestep = float(lines[start+1].split(';')[0])
for line in lines[start+1:]:
ls = line.split(';')
t = float(ls[0])
if prev_timestep != t : break
ibief = int(ls[1].strip('"'))
isect = int(ls[2].strip('"'))
pk = float(ls[3])
if ibief > res.ibmax:
res.ibmax += 1
res.is1.append(isect)
res.is2.append(isect)
if isect > res.ismax:
res.ismax += 1
res.xgeo.append(pk)
res.ygeo.append(0.0)
res.zfd.append(float(ls[izref]))
res.ybas.append(0.0)
if isect > res.is2[ibief-1]:
res.is2[ibief-1] = isect
if isect < res.is1[ibief-1]:
res.is1[ibief-1] = isect
print(" nb sections : "+str(res.ismax))
# first pass to get the data
prev_bief = 0
prev_sect = 0
z = np.zeros(res.ismax, dtype=np.float32)
q = np.zeros(res.ismax, dtype=np.float32)
isect = 0
count = np.array([1],dtype=np.int32)
bZ = bytearray('Z'.encode())
bQ = bytearray('Q'.encode())
for line in lines[start+1:]:
ls = line.split(';')
t = float(ls[0])
ibief = int(ls[1].strip('"'))
isect = int(ls[2].strip('"'))
if prev_timestep != t :
res.values['Z'].append(prev_timestep, z)
res.values['Q'].append(prev_timestep, q)
count[0] = 4 + 8 + 1 + 8*len(z)
res.raw_data.append(count)
res.raw_data.append(np.array([len(z)],dtype=np.int32))
res.raw_data.append(np.array([prev_timestep],dtype=np.float64))
res.raw_data.append(np.array(bZ,dtype=np.byte))
res.raw_data.append(z)
res.raw_data.append(count)
res.raw_data.append(count)
res.raw_data.append(np.array([len(q)],dtype=np.int32))
res.raw_data.append(np.array([prev_timestep],dtype=np.float64))
res.raw_data.append(np.array(bQ,dtype=np.byte))
res.raw_data.append(q)
res.raw_data.append(count)
prev_timestep = t
z[isect-1] = float(ls[iz])
q[isect-1] = float(ls[iq])
if __name__ == "__main__":
current_dir = os.getcwd()
os.chdir('/home/theophile.terraz/Codes/mage/src')
spec = imp.spec_from_file_location("me", "./mage_extraire.py")
me = imp.module_from_spec(spec)
spec.loader.exec_module(me)
os.chdir(current_dir)
print( 'Number of arguments:', len(sys.argv), 'arguments.')
print( 'Argument List:', str(sys.argv))
res = me.data()
lire_opt(sys.argv[1], res)
res.write_bin_8(sys.argv[2][0:-4])
+13 -235
View File
@@ -25,7 +25,7 @@ try:
# Installation allow Unix-like signal
from signal import SIGTERM, SIGSTOP, SIGCONT
_signal = True
except:
except Exception:
_signal = False
from enum import Enum
@@ -37,12 +37,14 @@ from Model.Results.River.River import River, Reach, Profile
logger = logging.getLogger()
class STATUS(Enum):
NOT_LAUNCHED = -1
STOPED = 0
RUNNING = 1
PAUSED = 5
class AbstractSolver(object):
_type = ""
@@ -57,17 +59,6 @@ class AbstractSolver(object):
self._name = name
self._description = ""
self._path_input = ""
self._path_solver = ""
self._path_output = ""
self._cmd_input = ""
self._cmd_solver = ""
self._cmd_output = ""
self._process = None
self._output = None
# Last study running
self._study = None
@@ -90,7 +81,6 @@ class AbstractSolver(object):
("all_init_time", "000:00:00:00"),
("all_final_time", "999:99:00:00"),
("all_timestep", "300.0"),
("all_command_line_arguments", ""),
]
return lst
@@ -139,249 +129,37 @@ class AbstractSolver(object):
def description(self, description):
self._description = description
def set_input(self, path, cmd):
self._path_input = path
self._cmd_input = cmd
def set_solver(self, path, cmd):
self._path_solver = path
self._cmd_solver = cmd
def set_output(self, path, cmd):
self._path_output = path
self._cmd_output = cmd
##########
# Export #
##########
def export(self, study, repertory, qlog = None):
def export(self, study, repertory, qlog=None):
raise NotImplementedMethodeError(self, self.export)
def cmd_args(self, study):
"""Return solver command line arguments list
Returns:
Command line arguments list
"""
params = study.river.get_params(self.type)
args = params.get_by_key("all_command_line_arguments")
return args.split(" ")
def input_param(self):
"""Return input command line parameter(s)
Returns:
Returns input parameter(s) string
"""
raise NotImplementedMethodeError(self, self.input_param)
def log_file(self):
"""Return log file name
Returns:
Returns log file name as string
"""
raise NotImplementedMethodeError(self, self.log_file)
###########
# RESULTS #
###########
@timer
def results(self, study, repertory, qlog = None):
results = Results(study = study)
def results(self, study, repertory, qlog=None):
results = Results(study=study)
return results
#######
# Run #
#######
def _install_dir(self):
return os.path.abspath(
os.path.join(
os.path.dirname(__file__),
"..", ".."
)
)
def _format_command(self, study, cmd, path = ""):
"""Format command line
Args:
cmd: The command line
path: Optional path string (replace @path in cmd)
Returns:
The executable and list of arguments
"""
# HACK: Works in most case... Trust me i'm an engineer
cmd = cmd.replace("@install_dir", self._install_dir())
cmd = cmd.replace("@path", path.replace(" ", "\ "))
cmd = cmd.replace("@input", self.input_param())
cmd = cmd.replace("@dir", self._process.workingDirectory())
cmd = cmd.replace("@args", " ".join(self.cmd_args(study)))
logger.debug(f"! {cmd}")
if cmd[0] == "\"":
# Command line executable path is between " char
cmd = cmd.split("\"")
exe = cmd[1].replace("\ ", " ")
args = list(
filter(
lambda s: s != "",
"\"".join(cmd[2:]).split(" ")[1:]
)
)
else:
# We suppose the command line executable path as no space char
cmd = cmd.replace("\ ", "&_&").split(" ")
exe = cmd[0].replace("&_&", " ")
args = list(
filter(
lambda s: s != "",
map(lambda s: s.replace("&_&", "\ "), cmd[1:])
)
)
logger.info(f"! {exe} {args}")
return exe, args
def run_input_data_fomater(self, study):
if self._cmd_input == "":
self._run_next(study)
return True
cmd = self._cmd_input
exe, args = self._format_command(study, cmd, self._path_input)
if not os.path.exists(exe):
error = f"[ERROR] Path {exe} do not exists"
logger.info(error)
return error
self._process.start(
exe, args,
)
return True
def run_solver(self, study):
if self._cmd_solver == "":
self._run_next(study)
return True
cmd = self._cmd_solver
exe, args = self._format_command(study, cmd, self._path_solver)
if not os.path.exists(exe):
error = f"[ERROR] Path {exe} do not exists"
logger.info(error)
return error
self._process.start(
exe, args,
)
self._status = STATUS.RUNNING
return True
def run_output_data_fomater(self, study):
if self._cmd_output == "":
self._run_next(study)
return True
cmd = self._cmd_output
exe, args = self._format_command(study, cmd, self._path_output)
if not os.path.exists(exe):
error = f"[ERROR] Path {exe} do not exists"
logger.info(error)
return error
self._process.start(
exe, args,
)
return True
def _data_ready(self):
s = self._process.readAll().data().decode()
if self._output is not None:
for x in s.split('\n'):
self._output.put(x)
def _run_next(self, study):
self._step += 1
if self._step < len(self._runs):
res = self._runs[self._step](study)
if res is not True:
self._output.put(res)
else:
self._status = STATUS.STOPED
def _finished(self, study, exit_code, exit_status):
if self._output is not None:
self._output.put(exit_code)
self._run_next(study)
def run(self, study, process = None, output_queue = None):
if process is not None:
self._process = process
if output_queue is not None:
self._output = output_queue
self._process.readyRead.connect(self._data_ready)
self._process.finished.connect(lambda c, s: self._finished(study, c, s))
self._runs = [
self.run_input_data_fomater,
self.run_solver,
self.run_output_data_fomater,
]
self._step = 0
# Run first step
res = self._runs[0](study)
if res is not True:
self._output.put(res)
def run(self, study):
raise NotImplementedMethodeError(self, self.run)
def kill(self):
if self._process is None:
return True
raise NotImplementedMethodeError(self, self.kill)
self._process.kill()
self._status = STATUS.STOPED
return True
def start(self, study, process = None):
if _signal:
# Solver is PAUSED, so continue execution
if self._status == STATUS.PAUSED:
os.kill(self._process.pid(), SIGCONT)
self._status = STATUS.RUNNING
return True
self.run(study, process)
return True
def start(self, study, process=None):
raise NotImplementedMethodeError(self, self.start)
def pause(self):
if _signal:
if self._process is None:
return False
# Send SIGSTOP to PAUSED solver
os.kill(self._process.pid(), SIGSTOP)
self._status = STATUS.PAUSED
return True
return False
raise NotImplementedMethodeError(self, self.pause)
def stop(self):
if self._process is None:
return False
self._process.terminate()
self._status = STATUS.STOPED
return True
raise NotImplementedMethodeError(self, self.stop)
+309
View File
@@ -0,0 +1,309 @@
# CommandLine.py -- Pamhyr
# Copyright (C) 2023 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 logging
from tools import timer, parse_command_line
try:
# Installation allow Unix-like signal
from signal import SIGTERM, SIGSTOP, SIGCONT
_signal = True
except Exception:
_signal = False
from enum import Enum
from Model.Except import NotImplementedMethodeError
from Model.Results.Results import Results
from Model.Results.River.River import River, Reach, Profile
from Solver.ASolver import AbstractSolver, STATUS
logger = logging.getLogger()
class CommandLineSolver(AbstractSolver):
_type = ""
def __init__(self, name):
super(CommandLineSolver, self).__init__(name)
self._current_process = None
self._status = STATUS.NOT_LAUNCHED
self._path_input = ""
self._path_solver = ""
self._path_output = ""
self._cmd_input = ""
self._cmd_solver = ""
self._cmd_output = ""
self._process = None
self._output = None
# Last study running
self._study = None
@classmethod
def default_parameters(cls):
lst = super(CommandLineSolver, cls).default_parameters()
lst += [
("all_command_line_arguments", ""),
]
return lst
def set_input(self, path, cmd):
self._path_input = path
self._cmd_input = cmd
def set_solver(self, path, cmd):
self._path_solver = path
self._cmd_solver = cmd
def set_output(self, path, cmd):
self._path_output = path
self._cmd_output = cmd
##########
# Export #
##########
def cmd_args(self, study):
"""Return solver command line arguments list
Returns:
Command line arguments list
"""
params = study.river.get_params(self.type)
args = params.get_by_key("all_command_line_arguments")
return args.split(" ")
def input_param(self):
"""Return input command line parameter(s)
Returns:
Returns input parameter(s) string
"""
raise NotImplementedMethodeError(self, self.input_param)
def output_param(self):
"""Return output command line parameter(s)
Returns:
Returns output parameter(s) string
"""
raise NotImplementedMethodeError(self, self.output_param)
def log_file(self):
"""Return log file name
Returns:
Returns log file name as string
"""
raise NotImplementedMethodeError(self, self.log_file)
#######
# Run #
#######
def _install_dir(self):
return os.path.abspath(
os.path.join(
os.path.dirname(__file__),
"..", ".."
)
)
def _format_command(self, study, cmd, path=""):
"""Format command line
Args:
cmd: The command line
path: Optional path string (replace @path in cmd)
Returns:
The executable and list of arguments
"""
cmd = cmd.replace("@install_dir", self._install_dir())
cmd = cmd.replace("@path", "\"" + path + "\"")
cmd = cmd.replace("@input", self.input_param())
cmd = cmd.replace("@output", self.output_param())
cmd = cmd.replace("@dir", self._process.workingDirectory())
cmd = cmd.replace("@args", " ".join(self.cmd_args(study)))
logger.debug(f"! {cmd}")
words = parse_command_line(cmd)
exe = words[0]
args = words[1:]
logger.info(f"! {exe} {args}")
return exe, args
def run_input_data_fomater(self, study):
if self._cmd_input == "":
self._run_next(study)
return True
cmd = self._cmd_input
exe, args = self._format_command(study, cmd, self._path_input)
if not os.path.exists(exe):
error = f"[ERROR] Path {exe} do not exists"
logger.info(error)
return error
self._process.start(
exe, args,
)
return True
def run_solver(self, study):
if self._cmd_solver == "":
self._run_next(study)
return True
cmd = self._cmd_solver
exe, args = self._format_command(study, cmd, self._path_solver)
if not os.path.exists(exe):
error = f"[ERROR] Path {exe} do not exists"
logger.info(error)
return error
self._process.start(
exe, args,
)
self._process.waitForStarted()
self._status = STATUS.RUNNING
return True
def run_output_data_fomater(self, study):
if self._cmd_output == "":
self._run_next(study)
return True
cmd = self._cmd_output
exe, args = self._format_command(study, cmd, self._path_output)
if not os.path.exists(exe):
error = f"[ERROR] Path {exe} do not exists"
logger.info(error)
return error
self._process.start(
exe, args,
)
return True
def _data_ready(self):
# Read process output and put lines in queue
s = self._process.readAll().data().decode()
if self._output is not None:
for x in s.split('\n'):
self._output.put(x)
def _run_next(self, study):
self._step += 1
if self._step < len(self._runs):
res = self._runs[self._step](study)
if res is not True:
self._output.put(res)
else:
self._status = STATUS.STOPED
def _finished(self, study, exit_code, exit_status):
if self._output is not None:
self._output.put(exit_code)
self._run_next(study)
def run(self, study, process=None, output_queue=None):
self._study = study
# Replace old values if needed
if process is not None:
self._process = process
if output_queue is not None:
self._output = output_queue
# Connect / reconnect signal
self._process.readyRead.connect(self._data_ready)
self._process.finished.connect(
lambda c, s: self._finished(study, c, s))
# Prepare running step
self._runs = [
self.run_input_data_fomater,
self.run_solver,
self.run_output_data_fomater,
]
self._step = 0
# Run first step
res = self._runs[0](study)
if res is not True:
self._output.put(res)
def kill(self):
if self._process is None:
return True
self._process.kill()
self._status = STATUS.STOPED
return True
def start(self, study, process=None):
if _signal:
# Solver is PAUSED, so continue execution
if self._status == STATUS.PAUSED:
os.kill(self._process.pid(), SIGCONT)
self._status = STATUS.RUNNING
return True
self.run(study, process)
return True
def pause(self):
if _signal:
if self._process is None:
return False
# Send SIGSTOP to PAUSED solver
os.kill(self._process.pid(), SIGSTOP)
self._status = STATUS.PAUSED
return True
return False
def stop(self):
if self._process is None:
return False
self._process.terminate()
self._status = STATUS.STOPED
return True
+3 -4
View File
@@ -16,11 +16,10 @@
# -*- coding: utf-8 -*-
from Solver.ASolver import (
AbstractSolver, STATUS
)
from Solver.CommandLine import CommandLineSolver
class GenericSolver(AbstractSolver):
class GenericSolver(CommandLineSolver):
_type = "generic"
def __init__(self, name):
+186 -84
View File
@@ -22,7 +22,7 @@ import numpy as np
from tools import timer
from Solver.ASolver import AbstractSolver
from Solver.CommandLine import CommandLineSolver
from Checker.Mage import MageNetworkGraphChecker
from Model.Results.Results import Results
@@ -30,6 +30,7 @@ from Model.Results.River.River import River, Reach, Profile
logger = logging.getLogger()
def mage_file_open(filepath, mode):
f = open(filepath, mode)
@@ -39,7 +40,8 @@ def mage_file_open(filepath, mode):
return f
class Mage(AbstractSolver):
class Mage(CommandLineSolver):
_type = "mage"
def __init__(self, name):
@@ -51,7 +53,6 @@ class Mage(AbstractSolver):
self._cmd_solver = "@path @input -o @output"
self._cmd_output = ""
@classmethod
def default_parameters(cls):
lst = super(Mage, cls).default_parameters()
@@ -87,8 +88,8 @@ class Mage(AbstractSolver):
@classmethod
def checkers(cls):
lst = [
MageNetworkGraphChecker(connectivity = True),
MageNetworkGraphChecker(connectivity = False)
MageNetworkGraphChecker(connectivity=True),
MageNetworkGraphChecker(connectivity=False)
]
return lst
@@ -98,16 +99,20 @@ class Mage(AbstractSolver):
##########
def cmd_args(self, study):
l = super(Mage, self).cmd_args(study)
lst = super(Mage, self).cmd_args(study)
l.append("-r")
lst.append("-r")
return l
return lst
def input_param(self):
name = self._study.name
return f"{name}.REP"
def output_param(self):
name = self._study.name
return f"{name}.BIN"
def log_file(self):
name = self._study.name
return f"{name}.TRA"
@@ -120,7 +125,6 @@ class Mage(AbstractSolver):
qlog.put("Export ST file")
os.makedirs(os.path.join(repertory, "net"), exist_ok=True)
gra_file = f"{name}.GRA"
# Write header
edges = study.river.edges()
@@ -136,55 +140,91 @@ class Mage(AbstractSolver):
if edge._name == "":
name = f"Reach_{edge.id}"
with mage_file_open(os.path.join(repertory, "net", f"{name}.ST"), "w+") as f:
with mage_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:
num = f"{cnt_num:>6}"
c1 = f"{profile.code1:>6}"
c2 = f"{profile.code2:>6}"
t = f"{len(profile.points):>6}"
kp = f"{profile.kp:>13.4f}"
pname = profile.name
if profile.name == "":
pname = f"p{profile.id:>3}".replace(" ", "p")
name = f"{pname}"
sediment = ""
if profile.sl is not None:
if not any(filter(lambda f: ".GRA" in f, files)):
files.append(gra_file)
nl = len(profile.sl)
sediment = f" {nl:>3}"
for l in profile.sl.layers:
sediment += f" {l.height:>10} {l.d50:>10} {l.sigma:>10} {l.critical_constraint:>10}"
f.write(f"{num}{c1}{c2}{t}{kp} {name} {sediment}\n")
self._export_ST_profile_header(
f, files, profile, cnt_num
)
cnt_num += 1
# Points
for point in profile.points:
x = f"{point.x:>13.4f}"
y = f"{point.y:>13.4f}"
z = f"{point.z:>13.4f}"
n = f"{point.name:<3}"
sediment = ""
prev = point.z
if point.sl is not None:
nl = len(point.sl)
sediment = f"{nl:>3}"
for l in point.sl.layers:
prev = round(prev - l.height, 5)
sediment += f" {prev:>10} {l.d50:>10} {l.sigma:>10} {l.critical_constraint:>10}"
f.write(f"{x}{y}{z} {n} {sediment}\n")
self._export_ST_point_line(
f, files, point
)
# Profile last line
f.write(f" 999.9990 999.9990 999.9990\n")
return files
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}"
kp = f"{profile.kp:>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} {kp} {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")
@timer
def _export_BC(self, t, bounds, repertory, qlog, name="0"):
files = []
@@ -210,7 +250,7 @@ class Mage(AbstractSolver):
v1 = d[1]
if t in ["HYD", "QSO", "LIM"]:
v0 /= 60 # Convert first column to minute
v0 /= 60 # Convert first column to minute
f.write(f"{v0:10}{v1:10}\n")
@@ -228,6 +268,9 @@ class Mage(AbstractSolver):
for tab in ["liquid", "solid", "suspenssion"]:
for bound in lst.get_tab(tab):
if bound.node is None:
continue
if bound.bctype == "ZD":
AVA.append(bound)
elif bound.bctype == "TD" or bound.bctype == "PC":
@@ -251,7 +294,10 @@ class Mage(AbstractSolver):
# if qlog is not None:
# qlog.put(f"Export LAT file")
# with mage_file_open(os.path.join(repertory, f"{name}.LAT"), "w+") as f:
# with mage_file_open(
# os.path.join(repertory, f"{name}.LAT"),
# "w+"
# ) as f:
# files.append(f"{name}.LAT")
# name = f"{lateral.node.id:3}".replace(" ", "x")
@@ -366,6 +412,36 @@ class Mage(AbstractSolver):
files.append(f"{name}.INI")
return files
@timer
def _export_CAS(self, study, repertory, qlog, name="0"):
files = []
reservoirs = study.river.reservoir.lst
if len(reservoirs) == 0:
return files
if qlog is not None:
qlog.put("Export CAS file")
with mage_file_open(os.path.join(repertory, f"{name}.CAS"), "w+") as f:
files.append(f"{name}.CAS")
for reservoir in reservoirs:
reservoir.sort()
node = reservoir.node
name = f"{node.id:3}".replace(" ", "x")
f.write(f"* {node.name} ({name}) Reservoir\n")
f.write(f"${name}\n")
f.write(f"*{'Elev(m)':>9}|{'Area(ha)':>10}\n")
for d in reservoir.data:
v0 = d[0]
v1 = d[1]
f.write(f"{v0:>10.3f}{v1:>10.3f}\n")
return files
@timer
def _export_REP(self, study, repertory, files, qlog, name="0"):
if qlog is not None:
@@ -391,13 +467,16 @@ class Mage(AbstractSolver):
if EXT in ["GRA"]:
f.write(f"{EXT} {file}\n")
@timer
def export(self, study, repertory, qlog = None):
def export(self, study, repertory, qlog=None):
self._study = study
name = study.name.replace(" ", "_")
self._export_ST(study, repertory, qlog, name = name)
# Define GRA file name
self._gra_file = f"{name}.GRA"
self._bin_file = f"{name}.BIN"
self._export_ST(study, repertory, qlog, name=name)
return True
@@ -405,13 +484,17 @@ class Mage(AbstractSolver):
# RESULTS #
###########
def read_bin(self, study, repertory, results, qlog = None, name="0"):
def read_bin(self, study, repertory, results, qlog=None, name="0"):
return
@timer
def results(self, study, repertory, qlog = None, name="0"):
results = Results(study = study)
def results(self, study, repertory, qlog=None, name="0"):
results = Results(
study=study,
solver=self,
repertory=repertory,
name=name,
)
self.read_bin(study, repertory, results, qlog, name=name)
return results
@@ -420,6 +503,7 @@ class Mage(AbstractSolver):
# MAGE 7 #
##########
class Mage7(Mage):
_type = "mage7"
@@ -438,6 +522,7 @@ class Mage7(Mage):
# MAGE 8 #
##########
class Mage8(Mage):
_type = "mage8"
@@ -476,15 +561,15 @@ class Mage8(Mage):
##########
def cmd_args(self, study):
l = super(Mage8, self).cmd_args(study)
lst = super(Mage8, self).cmd_args(study)
if study.river.has_sediment():
l.append("-c=3")
lst.append("-c=3")
return l
return lst
@timer
def _export_PAR(self, study, repertory, qlog = None, name = "0"):
def _export_PAR(self, study, repertory, qlog=None, name="0"):
files = []
if qlog is not None:
@@ -503,7 +588,7 @@ class Mage8(Mage):
if name in ["command_line_arguments"]:
continue
if name is "mage_compute_reach_volume_balance":
if name == "mage_compute_reach_volume_balance":
value = "O" if value == "y" else "N"
f.write(f"{name} {value}\n")
@@ -511,7 +596,7 @@ class Mage8(Mage):
return files
@timer
def _export_NET(self, study, repertory, qlog = None, name="0"):
def _export_NET(self, study, repertory, qlog=None, name="0"):
files = []
if qlog is not None:
@@ -579,20 +664,26 @@ class Mage8(Mage):
return files
@timer
def export(self, study, repertory, qlog = None, name="0"):
def export(self, study, repertory, qlog=None, name="0"):
self._study = study
name = study.name.replace(" ", "_")
# Define GRA file name
self._gra_file = f"{name}.GRA"
self._bin_file = f"{name}.BIN"
# Generate files
files = []
files = self._export_ST(study, repertory, qlog, name=name)
files = files + self._export_PAR(study, repertory, qlog, name=name)
files = files + self._export_NET(study, repertory, qlog, name=name)
files = files + self._export_bound_cond(study, repertory, qlog, name=name)
files = files + \
self._export_bound_cond(study, repertory, qlog, name=name)
files = files + self._export_RUG(study, repertory, qlog, name=name)
files = files + self._export_INI(study, repertory, qlog, name=name)
files = files + self._export_CAS(study, repertory, qlog, name=name)
self._export_REP(study, repertory, files, qlog, name=name)
return True
@@ -602,17 +693,22 @@ class Mage8(Mage):
###########
@timer
def read_bin(self, study, repertory, results, qlog = None, name="0"):
def read_bin(self, study, repertory, results, qlog=None, name="0"):
fname = os.path.join(repertory, f"{name}.BIN")
logger.info(f"read_bin: Start reading '{fname}' ...")
with mage_file_open(fname, "r") as f:
newline = lambda: np.fromfile(f, dtype=np.int32, count=1)
endline = lambda: np.fromfile(f, dtype=np.int32, count=1)
def newline(): return np.fromfile(f, dtype=np.int32, count=1)
def endline(): return np.fromfile(f, dtype=np.int32, count=1)
read_int = lambda size: np.fromfile(f, dtype=np.int32, count=size)
read_float = lambda size: np.fromfile(f, dtype=np.float32, count=size)
read_float64 = lambda size: np.fromfile(f, dtype=np.float64, count=size)
def read_int(size): return np.fromfile(
f, dtype=np.int32, count=size)
def read_float(size): return np.fromfile(
f, dtype=np.float32, count=size)
def read_float64(size): return np.fromfile(
f, dtype=np.float64, count=size)
# Meta data (1st line)
newline()
@@ -686,7 +782,7 @@ class Mage8(Mage):
# Data
newline()
ip_to_r = lambda i: iprofiles[
def ip_to_r(i): return iprofiles[
next(
filter(
lambda k: k[0] <= i <= k[1],
@@ -694,14 +790,15 @@ class Mage8(Mage):
)
)
]
ip_to_ri = lambda r, i: i - reach_offset[r]
def ip_to_ri(r, i): return i - reach_offset[r]
ts = set()
end = False
while not end:
n = read_int(1)[0]
timestamp = read_float64(1)[0]
key = bytearray(np.fromfile(f, dtype=np.byte, count=1)).decode()
key = bytearray(np.fromfile(
f, dtype=np.byte, count=1)).decode()
data = read_float(n)
logger.debug(f"read_bin: timestamp = {timestamp} sec")
@@ -725,7 +822,7 @@ class Mage8(Mage):
logger.info(f"read_bin: ... end with {len(ts)} timestamp read")
@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():
return
@@ -733,12 +830,17 @@ class Mage8(Mage):
logger.info(f"read_gra: Start reading '{fname}' ...")
with mage_file_open(fname, "r") as f:
newline = lambda: np.fromfile(f, dtype=np.int32, count=1)
endline = lambda: np.fromfile(f, dtype=np.int32, count=1)
def newline(): return np.fromfile(f, dtype=np.int32, count=1)
def endline(): return np.fromfile(f, dtype=np.int32, count=1)
read_int = lambda size: np.fromfile(f, dtype=np.int32, count=size)
read_float = lambda size: np.fromfile(f, dtype=np.float32, count=size)
read_float64 = lambda size: np.fromfile(f, dtype=np.float64, count=size)
def read_int(size): return np.fromfile(
f, dtype=np.int32, count=size)
def read_float(size): return np.fromfile(
f, dtype=np.float32, count=size)
def read_float64(size): return np.fromfile(
f, dtype=np.float64, count=size)
# Meta data (1st line)
newline()
@@ -810,7 +912,7 @@ class Mage8(Mage):
endline()
# Data
ip_to_r = lambda i: iprofiles[
def ip_to_r(i): return iprofiles[
next(
filter(
lambda k: k[0] <= i <= k[1],
@@ -818,7 +920,7 @@ class Mage8(Mage):
)
)
]
ip_to_ri = lambda r, i: i - reach_offset[r]
def ip_to_ri(r, i): return i - reach_offset[r]
ts = set()
ind = 0
@@ -878,11 +980,11 @@ class Mage8(Mage):
logger.info(f"read_gra: ... end with {len(ts)} timestamp read")
@timer
def results(self, study, repertory, qlog = None):
def results(self, study, repertory, qlog=None):
self._study = study
name = study.name.replace(" ", "_")
results = super(Mage8, self).results(study, repertory, qlog, name = name)
self.read_gra(study, repertory, results, qlog, name = name)
results = super(Mage8, self).results(study, repertory, qlog, name=name)
self.read_gra(study, repertory, results, qlog, name=name)
return results
+7 -6
View File
@@ -26,6 +26,7 @@ from PyQt5.QtCore import QCoreApplication
_translate = QCoreApplication.translate
logger = logging.getLogger()
class AboutWindow(PamhyrDialog):
_pamhyr_ui = "about"
_pamhyr_name = "About"
@@ -38,13 +39,13 @@ class AboutWindow(PamhyrDialog):
)
)
def __init__(self, study = None, config = None, parent=None):
def __init__(self, study=None, config=None, parent=None):
super(AboutWindow, self).__init__(
title = _translate("About", "About"),
study = study,
config = config,
options = [],
parent = parent
title=_translate("About", "About"),
study=study,
config=config,
options=[],
parent=parent
)
# Version
+11 -11
View File
@@ -33,16 +33,17 @@ _translate = QCoreApplication.translate
logger = logging.getLogger()
class Plot(PamhyrPlot):
def __init__(self, mode = "time", data=None,
def __init__(self, mode="time", data=None,
trad=None, canvas=None, toolbar=None,
parent=None):
super(Plot, self).__init__(
canvas = canvas,
trad = trad,
data = data,
toolbar = toolbar,
parent = parent
canvas=canvas,
trad=trad,
data=data,
toolbar=toolbar,
parent=parent
)
self._table_headers = self._trad.get_dict("table_headers")
@@ -72,8 +73,8 @@ class Plot(PamhyrPlot):
map(
lambda v: str(
datetime.fromtimestamp(v[0]) - t0
).split(",")[0]\
.replace("days", _translate("BoundaryCondition", "days"))\
).split(",")[0]
.replace("days", _translate("BoundaryCondition", "days"))
.replace("day", _translate("BoundaryCondition", "day")),
fx
)
@@ -88,7 +89,6 @@ class Plot(PamhyrPlot):
self.canvas.axes.set_xticks(ticks=xx, labels=xt, rotation=45)
@timer
def draw(self):
self.canvas.axes.cla()
@@ -122,13 +122,13 @@ class Plot(PamhyrPlot):
self.canvas.axes.autoscale_view(True, True, True)
self.canvas.figure.tight_layout()
self.canvas.figure.canvas.draw_idle()
#self.toolbar.update()
# self.toolbar.update()
self._init = True
@timer
def update(self, ind=None):
if self._init == False:
if not self._init:
self.draw()
return
+1 -1
View File
@@ -44,13 +44,13 @@ from Model.BoundaryCondition.BoundaryConditionTypes import (
from View.BoundaryCondition.Edit.UndoCommand import (
SetDataCommand, AddCommand, DelCommand,
SortCommand, MoveCommand, PasteCommand,
DuplicateCommand,
)
_translate = QCoreApplication.translate
logger = logging.getLogger()
class TableModel(PamhyrTableModel):
def data(self, index, role):
if role == Qt.TextAlignmentRole:
+7 -18
View File
@@ -29,6 +29,7 @@ from Model.BoundaryCondition.BoundaryCondition import BoundaryCondition
logger = logging.getLogger()
class SetDataCommand(QUndoCommand):
def __init__(self, data, index, column, new_value):
QUndoCommand.__init__(self)
@@ -46,6 +47,7 @@ class SetDataCommand(QUndoCommand):
def redo(self):
self._data._set_i_c_v(self._index, self._column, self._new)
class SetMetaDataCommand(QUndoCommand):
def __init__(self, data, column, new_value):
QUndoCommand.__init__(self)
@@ -89,6 +91,7 @@ class AddCommand(QUndoCommand):
else:
self._data.insert(self._index, self._new)
class DelCommand(QUndoCommand):
def __init__(self, data, rows):
QUndoCommand.__init__(self)
@@ -108,6 +111,7 @@ class DelCommand(QUndoCommand):
def redo(self):
self._data.delete_i(self._rows)
class SortCommand(QUndoCommand):
def __init__(self, data, _reverse):
QUndoCommand.__init__(self)
@@ -170,25 +174,10 @@ class PasteCommand(QUndoCommand):
self._bcs.reverse()
def undo(self):
self._data.delete(self._bcs)
self._data.delete_i(
range(self._row, self._row + len(self._bcs))
)
def redo(self):
for bc in self._bcs:
self._data.insert(self._row, bc)
class DuplicateCommand(QUndoCommand):
def __init__(self, data, rows, bc):
QUndoCommand.__init__(self)
self._data = data
self._rows = rows
self._bc = deepcopy(bc)
self._bc.reverse()
def undo(self):
self._data.delete(self._bc)
def redo(self):
for bc in self._bcs:
self._data.insert(self._rows[0], bc)
+28 -23
View File
@@ -53,6 +53,7 @@ _translate = QCoreApplication.translate
logger = logging.getLogger()
class WD50Sigma(PamhyrWidget):
_pamhyr_ui = "d50sigma"
@@ -94,6 +95,7 @@ class WD50Sigma(PamhyrWidget):
def valueChangedSigma(self, value):
self.sigmaChanged.emit(value)
class EditBoundaryConditionWindow(PamhyrWindow):
_pamhyr_ui = "EditBoundaryConditions"
_pamhyr_name = "Edit Boundary Conditions"
@@ -115,13 +117,15 @@ class EditBoundaryConditionWindow(PamhyrWindow):
)
super(EditBoundaryConditionWindow, self).__init__(
title = name,
study = study,
config = config,
trad = trad,
parent = parent
title=name,
study=study,
config=config,
trad=trad,
parent=parent
)
self._hash_data.append(data)
self.setup_table()
self.setup_plot()
self.setup_data()
@@ -132,7 +136,7 @@ class EditBoundaryConditionWindow(PamhyrWindow):
if self._is_solid:
layout = self.find(QVBoxLayout, "verticalLayout_table")
self._d50sigma = WD50Sigma(parent = self)
self._d50sigma = WD50Sigma(parent=self)
layout.addWidget(self._d50sigma)
self._d50sigma.set_d50(self._data.d50)
@@ -145,22 +149,22 @@ class EditBoundaryConditionWindow(PamhyrWindow):
headers[h] = table_headers[h]
self._delegate_time = PamhyrExTimeDelegate(
data = self._data,
mode = self._study.time_system,
parent = self
data=self._data,
mode=self._study.time_system,
parent=self
)
table = self.find(QTableView, "tableView")
self._table = TableModel(
table_view = table,
table_headers = headers,
editable_headers = self._data.header,
delegates = {
"time": self._delegate_time,
table_view=table,
table_headers=headers,
editable_headers=self._data.header,
delegates={
#"time": self._delegate_time,
},
data = self._data,
undo = self._undo_stack,
opt_data = self._study.time_system
data=self._data,
undo=self._undo_stack,
opt_data=self._study.time_system
)
table.setModel(self._table)
@@ -178,11 +182,11 @@ class EditBoundaryConditionWindow(PamhyrWindow):
self.verticalLayout.addWidget(self.canvas)
self.plot = Plot(
canvas = self.canvas,
data = self._data,
mode = self._study.time_system,
trad = self._trad,
toolbar = self.toolbar,
canvas=self.canvas,
data=self._data,
mode=self._study.time_system,
trad=self._trad,
toolbar=self.toolbar,
)
self.plot.draw()
@@ -270,7 +274,6 @@ class EditBoundaryConditionWindow(PamhyrWindow):
self._table.move_down(row)
self.plot.update()
def _copy(self):
rows = self.index_selected_rows()
@@ -286,6 +289,8 @@ class EditBoundaryConditionWindow(PamhyrWindow):
def _paste(self):
header, data = self.parseClipboardTable()
logger.debug(f"paste: h:{header}, d:{data}")
if len(data) == 0:
return
@@ -24,6 +24,7 @@ from View.BoundaryCondition.translate import BCTranslate
_translate = QCoreApplication.translate
class BCETranslate(BCTranslate):
def __init__(self):
super(BCETranslate, self).__init__()
+17 -12
View File
@@ -44,7 +44,7 @@ from View.Tools.PamhyrTable import PamhyrTableModel
from View.BoundaryCondition.UndoCommand import (
SetNameCommand, SetNodeCommand, SetTypeCommand,
AddCommand, DelCommand, SortCommand,
MoveCommand, PasteCommand, DuplicateCommand,
MoveCommand, PasteCommand,
)
from View.BoundaryCondition.translate import BC_types
@@ -52,8 +52,10 @@ logger = logging.getLogger()
_translate = QCoreApplication.translate
class ComboBoxDelegate(QItemDelegate):
def __init__(self, data=None, mode="type", tab="", trad=None, parent=None):
def __init__(self, data=None, mode="type", tab="",
trad=None, parent=None):
super(ComboBoxDelegate, self).__init__(parent)
self._data = data
@@ -102,8 +104,9 @@ class ComboBoxDelegate(QItemDelegate):
def updateEditorGeometry(self, editor, option, index):
r = QRect(option.rect)
if self.editor.windowFlags() & Qt.Popup and editor.parent() is not None:
r.setTopLeft(self.editor.parent().mapToGlobal(r.topLeft()))
if self.editor.windowFlags() & Qt.Popup:
if editor.parent() is not None:
r.setTopLeft(self.editor.parent().mapToGlobal(r.topLeft()))
editor.setGeometry(r)
@pyqtSlot()
@@ -112,13 +115,14 @@ class ComboBoxDelegate(QItemDelegate):
class TableModel(PamhyrTableModel):
def __init__(self, **kwargs):
super(TableModel, self).__init__(**kwargs)
def __init__(self, trad = None, **kwargs):
self._trad = trad
self._long_types = {}
if self._trad is not None:
self._long_types = self._trad.get_dict("long_types")
super(TableModel, self).__init__(trad = trad, **kwargs)
def _setup_lst(self):
self._lst = self._data.boundary_condition
self._tab = self._opt_data
@@ -157,20 +161,21 @@ class TableModel(PamhyrTableModel):
if self._headers[column] == "name":
self._undo.push(
SetNameCommand(
self._lst, self._tab,row, value
self._lst, self._tab, row, value
)
)
elif self._headers[column] == "type":
key = next(k for k, v in self._long_types.items() if v == value)
key = next(k for k, v in self._long_types.items()
if v == value)
self._undo.push(
SetTypeCommand(
self._lst, self._tab,row, BC_types[key]
self._lst, self._tab, row, BC_types[key]
)
)
elif self._headers[column] == "node":
self._undo.push(
SetNodeCommand(
self._lst, self._tab,row, self._data.node(value)
self._lst, self._tab, row, self._data.node(value)
)
)
except Exception as e:
@@ -185,7 +190,7 @@ class TableModel(PamhyrTableModel):
self._undo.push(
AddCommand(
self._lst, self._tab,row
self._lst, self._tab, row
)
)
+17 -19
View File
@@ -26,6 +26,7 @@ from PyQt5.QtWidgets import (
from Model.BoundaryCondition.BoundaryCondition import BoundaryCondition
from Model.BoundaryCondition.BoundaryConditionList import BoundaryConditionList
class SetNameCommand(QUndoCommand):
def __init__(self, bcs, tab, index, new_value):
QUndoCommand.__init__(self)
@@ -42,6 +43,7 @@ class SetNameCommand(QUndoCommand):
def redo(self):
self._bcs.get(self._tab, self._index).name = self._new
class SetNodeCommand(QUndoCommand):
def __init__(self, bcs, tab, index, node):
QUndoCommand.__init__(self)
@@ -51,12 +53,20 @@ class SetNodeCommand(QUndoCommand):
self._index = index
self._old = self._bcs.get(self._tab, self._index).node
self._new = node
self._prev_assoc_to_node = self._bcs.get_assoc_to_node(tab, node)
def _previous_assoc_node(self, node):
if self._prev_assoc_to_node is not None:
self._prev_assoc_to_node.node = node
def undo(self):
self._bcs.get(self._tab, self._index).node = self._old
self._previous_assoc_node(self._new)
def redo(self):
self._bcs.get(self._tab, self._index).node = self._new
self._previous_assoc_node(None)
class SetTypeCommand(QUndoCommand):
def __init__(self, bcs, tab, index, _type):
@@ -76,6 +86,7 @@ class SetTypeCommand(QUndoCommand):
def redo(self):
self._bcs.set(self._tab, self._index, self._new)
class AddCommand(QUndoCommand):
def __init__(self, bcs, tab, index):
QUndoCommand.__init__(self)
@@ -94,6 +105,7 @@ class AddCommand(QUndoCommand):
else:
self._bcs.insert(self._tab, self._index, self._new)
class DelCommand(QUndoCommand):
def __init__(self, bcs, tab, rows):
QUndoCommand.__init__(self)
@@ -114,6 +126,7 @@ class DelCommand(QUndoCommand):
def redo(self):
self._bcs.delete_i(self._tab, self._rows)
class SortCommand(QUndoCommand):
def __init__(self, bcs, tab, _reverse):
QUndoCommand.__init__(self)
@@ -181,26 +194,11 @@ class PasteCommand(QUndoCommand):
self._bc.reverse()
def undo(self):
self._bcs.delete(self._tab, self._bc)
self._bcs.delete_i(
self._tab,
range(self._row, self._row + len(self._bc))
)
def redo(self):
for bc in self._bc:
self._bcs.insert(self._tab, self._row, bc)
class DuplicateCommand(QUndoCommand):
def __init__(self, bcs, tab, rows, bc):
QUndoCommand.__init__(self)
self._bcs = bcs
self._tab = tab
self._rows = rows
self._bc = deepcopy(bc)
self._bc.reverse()
def undo(self):
self._bcs.delete(self._tab, self._bc)
def redo(self):
for bc in self._bcs:
self._bcs.insert(self._tab, self._rows[0], bc)
+36 -35
View File
@@ -47,7 +47,7 @@ from Model.BoundaryCondition.BoundaryConditionTypes import (
from View.BoundaryCondition.UndoCommand import (
SetNameCommand, SetNodeCommand, SetTypeCommand,
AddCommand, DelCommand, SortCommand,
MoveCommand, PasteCommand, DuplicateCommand,
MoveCommand, PasteCommand,
)
from View.BoundaryCondition.Table import (
@@ -62,6 +62,7 @@ _translate = QCoreApplication.translate
logger = logging.getLogger()
class BoundaryConditionWindow(PamhyrWindow):
_pamhyr_ui = "BoundaryConditions"
_pamhyr_name = "Boundary conditions"
@@ -70,10 +71,10 @@ class BoundaryConditionWindow(PamhyrWindow):
name = self._pamhyr_name + " - " + study.name
super(BoundaryConditionWindow, self).__init__(
title = name,
study = study,
config = config,
trad = BCTranslate(),
title=name,
study=study,
config=config,
trad=BCTranslate(),
parent=parent
)
@@ -90,33 +91,33 @@ class BoundaryConditionWindow(PamhyrWindow):
for t in ["liquid", "solid", "suspenssion"]:
self._delegate_type = ComboBoxDelegate(
trad = self._trad,
data = self._study.river,
mode = "type",
tab = t,
trad=self._trad,
data=self._study.river,
mode="type",
tab=t,
parent=self
)
self._delegate_node = ComboBoxDelegate(
trad = self._trad,
data = self._study.river,
mode = "node",
tab = t,
trad=self._trad,
data=self._study.river,
mode="node",
tab=t,
parent=self
)
table = self.find(QTableView, f"tableView_{t}")
self._table[t] = TableModel(
table_view = table,
table_headers = self._trad.get_dict("table_headers"),
editable_headers = ["name", "type", "node"],
delegates = {
table_view=table,
table_headers=self._trad.get_dict("table_headers"),
editable_headers=["name", "type", "node"],
delegates={
"type": self._delegate_type,
"node": self._delegate_node,
},
trad = self._trad,
data = self._study.river,
undo = self._undo_stack,
opt_data = t,
trad=self._trad,
data=self._study.river,
undo=self._undo_stack,
opt_data=t,
)
table.setModel(self._table[t])
table.setSelectionBehavior(QAbstractItemView.SelectRows)
@@ -126,7 +127,7 @@ class BoundaryConditionWindow(PamhyrWindow):
def setup_graph(self):
self.graph_widget = GraphWidget(
self._study.river,
min_size=None, size=(200,200),
min_size=None, size=(200, 200),
only_display=True,
parent=self
)
@@ -213,17 +214,17 @@ class BoundaryConditionWindow(PamhyrWindow):
tab = self.current_tab()
rows = self.index_selected_rows()
for row in rows:
win = self.sub_win_filter_first(
"Edit boundary condition",
contain = [f"({self._bcs.get(tab, row).id})"]
)
data = self._bcs.get(tab, row)
if win is None:
win = EditBoundaryConditionWindow(
data=self._bcs.get(tab, row),
study=self._study,
parent=self
)
win.show()
else:
win.activateWindow()
if self.sub_window_exists(
EditBoundaryConditionWindow,
data=[self._study, None, data]
):
continue
win = EditBoundaryConditionWindow(
data=data,
study=self._study,
parent=self
)
win.show()
+4 -3
View File
@@ -37,6 +37,7 @@ BC_types = {
"SL": Solid,
}
class BCTranslate(PamhyrTranslate):
def __init__(self):
super(BCTranslate, self).__init__()
@@ -44,9 +45,9 @@ class BCTranslate(PamhyrTranslate):
self._sub_dict["long_types"] = {
"ND": _translate("BoundaryCondition", "Not defined"),
"PC": _translate("BoundaryCondition", "Ponctual contribution"),
"TZ": _translate("BoundaryCondition", "TimeZ"),
"TD": _translate("BoundaryCondition", "TimeDischarge"),
"ZD": _translate("BoundaryCondition", "ZDischarge"),
"TZ": _translate("BoundaryCondition", "Z(t)"),
"TD": _translate("BoundaryCondition", "Q(t)"),
"ZD": _translate("BoundaryCondition", "Q(Z)"),
"SL": _translate("BoundaryCondition", "Solid"),
}
+1
View File
@@ -22,6 +22,7 @@ from View.Tools.PamhyrTranslate import PamhyrTranslate
_translate = QCoreApplication.translate
class CheckListTranslate(PamhyrTranslate):
def __init__(self):
super(CheckListTranslate, self).__init__()
+20 -14
View File
@@ -44,13 +44,14 @@ from View.CheckList.Translate import CheckListTranslate
_translate = QCoreApplication.translate
class CheckListWindow(PamhyrWindow):
_pamhyr_ui = "CheckList"
_pamhyr_name = "Check list"
signalStatus = pyqtSignal(str)
def __init__(self, autorun:bool=True,
def __init__(self, autorun: bool = True,
study=None, config=None,
solver=None, parent=None):
self._autorun = autorun
@@ -59,14 +60,17 @@ class CheckListWindow(PamhyrWindow):
name = self._pamhyr_name + " - " + study.name
super(CheckListWindow, self).__init__(
title = name,
study = study,
config = config,
trad = CheckListTranslate(),
options = [],
parent = parent
title=name,
study=study,
config=config,
trad=CheckListTranslate(),
options=[],
parent=parent
)
# Add solver to hash computation data
self._hash_data.append(self._solver)
self._checker_list = (
self._study.checkers() +
self._solver.checkers()
@@ -81,9 +85,9 @@ class CheckListWindow(PamhyrWindow):
def setup_table(self):
table = self.find(QTableView, f"tableView")
self._table = TableModel(
table_view = table,
table_headers = self._trad.get_dict("table_headers"),
data = self._checker_list,
table_view=table,
table_headers=self._trad.get_dict("table_headers"),
data=self._checker_list,
)
def setup_progress_bar(self):
@@ -96,7 +100,8 @@ class CheckListWindow(PamhyrWindow):
def setup_connections(self):
self.find(QPushButton, "pushButton_ok").clicked.connect(self.accept)
self.find(QPushButton, "pushButton_retry").clicked.connect(self.retry)
self.find(QPushButton, "pushButton_cancel").clicked.connect(self.reject)
self.find(QPushButton, "pushButton_cancel")\
.clicked.connect(self.reject)
def setup_thread(self):
self._worker = Worker(self._study, self._checker_list)
@@ -120,7 +125,8 @@ class CheckListWindow(PamhyrWindow):
def _compute_status(self):
ok = len(list(filter(lambda c: c.is_ok(), self._checker_list)))
warning = len(list(filter(lambda c: c.is_warning(), self._checker_list)))
warning = len(
list(filter(lambda c: c.is_warning(), self._checker_list)))
error = len(list(filter(lambda c: c.is_error(), self._checker_list)))
return ok, warning, error
@@ -164,7 +170,7 @@ class CheckListWindow(PamhyrWindow):
self.update_statusbar()
def update(self, key:str):
def update(self, key: str):
if key == "start":
self.start_compute()
self.info_compute("Starting ...")
@@ -187,4 +193,4 @@ class CheckListWindow(PamhyrWindow):
def accept(self):
self._parent.solver_log(self._solver)
#self.end()
# self.end()
+2 -1
View File
@@ -22,6 +22,7 @@ from PyQt5.QtCore import (
Qt, QObject, pyqtSlot, pyqtSignal
)
class Worker(QObject):
signalStatus = pyqtSignal(str)
@@ -42,6 +43,6 @@ class Worker(QObject):
self.signalStatus.emit(checker.name)
# Run checker
checker.run(self._study)
checker._run(self._study)
self.signalStatus.emit("progress")
+17 -11
View File
@@ -24,6 +24,7 @@ from PyQt5.QtWidgets import (
QPushButton,
)
class ConfigureSolverWindow(PamhyrDialog):
_pamhyr_ui = "ConfigureAddSolverDialog"
_pamhyr_name = "Add/Edit Solver"
@@ -35,9 +36,9 @@ class ConfigureSolverWindow(PamhyrDialog):
name = "Add a new Solver"
super(ConfigureSolverWindow, self).__init__(
title = name,
config = config,
options = [],
title=name,
config=config,
options=[],
parent=parent
)
@@ -67,16 +68,19 @@ class ConfigureSolverWindow(PamhyrDialog):
# File button
buttons = {
"pushButton_input": (lambda: self.file_dialog(
select_file = True,
callback = lambda f: self.set_line_edit_text("lineEdit_input", f[0])
select_file=True,
callback=lambda f: self.set_line_edit_text(
"lineEdit_input", f[0])
)),
"pushButton_solver": (lambda: self.file_dialog(
select_file = True,
callback = lambda f: self.set_line_edit_text("lineEdit_solver", f[0])
select_file=True,
callback=lambda f: self.set_line_edit_text(
"lineEdit_solver", f[0])
)),
"pushButton_output": (lambda: self.file_dialog(
select_file = True,
callback = lambda f: self.set_line_edit_text("lineEdit_output", f[0])
select_file=True,
callback=lambda f: self.set_line_edit_text(
"lineEdit_output", f[0])
)),
}
@@ -92,9 +96,11 @@ class ConfigureSolverWindow(PamhyrDialog):
else:
# Build new solver from selected type
stype = self.get_combobox_text("comboBox_solver")
self.data = solver_type_list[stype](self.get_line_edit_text("lineEdit_name"))
self.data = solver_type_list[stype](
self.get_line_edit_text("lineEdit_name"))
self.data.description = self.get_line_edit_text("lineEdit_description")
self.data.description = self.get_line_edit_text(
"lineEdit_description")
self.data.set_input(
self.get_line_edit_text("lineEdit_input"),
self.get_line_edit_text("lineEdit_input_cmd")
+34
View File
@@ -0,0 +1,34 @@
# Translate.py -- Pamhyr
# Copyright (C) 2023 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.Tools.PamhyrTranslate import PamhyrTranslate
_translate = QCoreApplication.translate
class ConfigureTranslate(PamhyrTranslate):
def __init__(self):
super(ConfigureTranslate, self).__init__()
self._sub_dict["table_headers"] = {
"name": _translate("Configure", "Name"),
"type": _translate("Configure", "Type"),
"description": _translate("Configure", "Description"),
}
+57 -64
View File
@@ -22,11 +22,13 @@ from copy import deepcopy
from config import Config
from View.Tools.PamhyrWindow import PamhyrDialog
from View.Tools.PamhyrTable import PamhyrTableModel
from View.Stricklers.Table import TableModel
from View.Stricklers.translate import *
from View.Stricklers.translate import StricklersTranslate
from View.Stricklers.UndoCommand import *
from View.Configure.Translate import ConfigureTranslate
from View.Configure.Solver.Window import ConfigureSolverWindow
from PyQt5.QtGui import (
@@ -47,50 +49,29 @@ from Solver.Solvers import solver_long_name
logger = logging.getLogger()
class SolverTableModel(QAbstractTableModel):
def __init__(self, headers=[], rows=[]):
super(QAbstractTableModel, self).__init__()
self.rows = rows
self.headers = headers
def rowCount(self, parent):
return len(self.rows)
def columnCount(self, parent):
return len(self.headers)
class SolverTableModel(PamhyrTableModel):
def data(self, index, role):
if role != Qt.ItemDataRole.DisplayRole:
return QVariant()
ret = self.rows[index.row()][self.headers[index.column()]]
ret = self._data[index.row()][self._headers[index.column()]]
if self.headers[index.column()] == "type":
if self._headers[index.column()] == "type":
ret = solver_long_name[ret]
return ret
def headerData(self, section, orientation, role):
if (role == Qt.ItemDataRole.DisplayRole and
orientation == Qt.Orientation.Horizontal):
return self.headers[section].capitalize()
if (role == Qt.ItemDataRole.DisplayRole and
orientation == Qt.Orientation.Vertical):
return section + 1 # Start at 1
return QVariant()
def removeRow(self, index):
del self.rows[index.row()]
del self._data[index.row()]
self.layoutChanged.emit()
def add_solver(self, solver):
self.rows.append(solver)
self._data.append(solver)
self.layoutChanged.emit()
def change_solver(self, solver, index):
self.rows[index.row()] = solver
self._data[index.row()] = solver
self.layoutChanged.emit()
@@ -103,12 +84,15 @@ class ConfigureWindow(PamhyrDialog):
config = Config()
super(ConfigureWindow, self).__init__(
title = self._pamhyr_name,
config = config,
options = [],
title=self._pamhyr_name,
config=config,
trad=ConfigureTranslate(),
options=[],
parent=parent
)
self._trad_stricklers = StricklersTranslate()
self.setup_custom_sc()
self.setup_solver()
self.setup_stricklers()
@@ -118,43 +102,43 @@ class ConfigureWindow(PamhyrDialog):
def setup_solver(self):
table = self.find(QTableView, "tableView_solver")
self.solver_table_model = SolverTableModel(
headers = ["name", "type", "description"],
rows = self._config.solvers
table_view=table,
table_headers=self._trad.get_dict("table_headers"),
data=self._config.solvers
)
table.setModel(self.solver_table_model)
table.setSelectionBehavior(QAbstractItemView.SelectRows)
table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
table.setAlternatingRowColors(True)
table.resizeColumnsToContents()
def setup_stricklers(self):
table = self.find(QTableView, f"tableView_stricklers")
self._stricklers = deepcopy(self._config.stricklers)
self._stricklers_table = TableModel(
data = self._stricklers,
undo = self._undo_stack,
table_view=table,
table_headers=self._trad_stricklers.get_dict("table_headers"),
editable_headers=["name", "comment", "minor", "medium"],
data=self._stricklers,
undo=self._undo_stack,
)
table.setModel(self._stricklers_table)
table.setSelectionBehavior(QAbstractItemView.SelectRows)
table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
table.setAlternatingRowColors(True)
def setup_data(self):
# Meshing_Tool
self.set_line_edit_text("lineEdit_meshing_tool", self._config.meshing_tool)
self.set_line_edit_text("lineEdit_meshing_tool",
self._config.meshing_tool)
# Const
self.set_line_edit_text("lineEdit_segment", str(self._config.segment))
self.set_line_edit_text("lineEdit_max_listing", str(self._config.max_listing))
self.set_line_edit_text("lineEdit_max_listing",
str(self._config.max_listing))
# Backup
self.set_check_box("checkBox_backup", self._config.backup_enable)
self.set_line_edit_text("lineEdit_backup_path", self._config.backup_path)
self.set_time_edit("timeEdit_backup_frequence", self._config.backup_frequence)
self.set_line_edit_text("lineEdit_backup_path",
self._config.backup_path)
self.set_time_edit("timeEdit_backup_frequence",
self._config.backup_frequence)
self.set_spin_box("spinBox_backup_max", self._config.backup_max)
# Editor
self.set_line_edit_text("lineEdit_editor_cmd", str(self._config.editor))
self.set_line_edit_text("lineEdit_editor_cmd",
str(self._config.editor))
# Language
languages = Config.languages()
@@ -179,14 +163,14 @@ class ConfigureWindow(PamhyrDialog):
"pushButton_stricklers_sort": self.sort_stricklers,
# Others
"pushButton_backup_path": lambda: self.file_dialog(
select_file = False,
callback = lambda f: self.set_line_edit_text(
select_file=False,
callback=lambda f: self.set_line_edit_text(
"lineEdit_backup_path", f[0]
)
),
"pushButton_meshing_tool" : lambda: self.file_dialog(
select_file = True,
callback = lambda f: self.set_line_edit_text(
"pushButton_meshing_tool": lambda: self.file_dialog(
select_file=True,
callback=lambda f: self.set_line_edit_text(
"lineEdit_meshing_tool", f[0]
)
),
@@ -197,19 +181,23 @@ class ConfigureWindow(PamhyrDialog):
def accept(self):
# Solvers
self._config.solvers = self.solver_table_model.rows.copy()
self._config.solvers = self.solver_table_model._data.copy()
# Meshing_Tool
self._config.meshing_tool = self.get_line_edit_text("lineEdit_meshing_tool")
self._config.meshing_tool = self.get_line_edit_text(
"lineEdit_meshing_tool")
# Const
self._config.segment = self.get_line_edit_text("lineEdit_segment")
self._config.max_listing = self.get_line_edit_text("lineEdit_max_listing")
self._config.max_listing = self.get_line_edit_text(
"lineEdit_max_listing")
# Backup
self._config.backup_enable = self.get_check_box("checkBox_backup")
self._config.backup_path = self.get_line_edit_text("lineEdit_backup_path")
self._config.backup_frequence = self.get_time_edit("timeEdit_backup_frequence")
self._config.backup_path = self.get_line_edit_text(
"lineEdit_backup_path")
self._config.backup_frequence = self.get_time_edit(
"timeEdit_backup_frequence")
self._config.backup_max = self.get_spin_box("spinBox_backup_max")
# Stricklers
@@ -219,7 +207,8 @@ class ConfigureWindow(PamhyrDialog):
self._config.editor = self.get_line_edit_text("lineEdit_editor_cmd")
# Language
self._config.lang = Config.languages()[self.get_combobox_text("comboBox_language")]
self._config.lang = Config.languages(
)[self.get_combobox_text("comboBox_language")]
self.end()
@@ -232,6 +221,7 @@ class ConfigureWindow(PamhyrDialog):
self.close()
# Debug
def set_debug(self):
self._config.debug = not self._config.debug
logger.info(f"Debug mode set : {self._config.debug}")
@@ -240,15 +230,17 @@ class ConfigureWindow(PamhyrDialog):
# Solvers
def edit_solver(self):
indexes = self.find(QTableView, "tableView_solver").selectionModel().selectedRows()
indexes = self.find(
QTableView, "tableView_solver").selectionModel().selectedRows()
for index in indexes:
self.edit_solver = ConfigureSolverWindow(
data=self.solver_table_model.rows[index.row()],
data=self.solver_table_model._data[index.row()],
config=self._config,
parent=self
)
if self.edit_solver.exec_():
self.solver_table_model.change_solver(self.edit_solver.data, index)
self.solver_table_model.change_solver(
self.edit_solver.data, index)
def add_solver(self):
dialog_solver = ConfigureSolverWindow(parent=self)
@@ -256,7 +248,8 @@ class ConfigureWindow(PamhyrDialog):
self.solver_table_model.add_solver(dialog_solver.data)
def remove_solver(self):
indices = self.find(QTableView, "tableView_solver").selectionModel().selectedRows()
indices = self.find(
QTableView, "tableView_solver").selectionModel().selectedRows()
for index in sorted(indices):
self.solver_table_model.removeRow(index)
Loaded 100 of 218 files, more files were not shown because too many files have changed in this diff. Show more