Compare commits

...
225 Commits
Author SHA1 Message Date
Pierre-Antoine Rouby 28c23ee580 Translate: Add short RK name. 2024-07-03 15:25:45 +02:00
Pierre-Antoine Rouby 71c5aec876 doc: tools: Minor change. 2024-07-03 15:21:42 +02:00
Pierre-Antoine Rouby 3ce9e094e7 Translate: Rename KP to RK (River Kilometric). 2024-07-03 15:18:32 +02:00
Pierre-Antoine Rouby c4554d38c1 BC, LC: Edit: Disable auto sort. 2024-07-01 15:53:08 +02:00
Pierre-Antoine Rouby 4deed897ff BC, LC: Edit: Auto sort values. 2024-07-01 14:46:18 +02:00
Pierre-Antoine Rouby 460450f524 Pamhyr: Fix save as method and minor change. 2024-07-01 14:04:18 +02:00
Theophile Terraz 60336c25f1 pep8 2024-06-13 18:23:11 +02:00
Theophile Terraz 82093bcf95 remove slider in geometry 2024-06-13 18:21:09 +02:00
Theophile Terraz 7b8191e2f0 interactive selection in results 2024-06-13 18:20:46 +02:00
Theophile Terraz 88dd9cf15a remove slider in geometry window 2024-06-13 11:12:35 +02:00
Theophile Terraz 7d4091eb24 pep8 2024-06-12 11:40:52 +02:00
Theophile Terraz 59fc80a10b debug + interactive selection in lateral contrib 2024-06-11 16:19:52 +02:00
Theophile Terraz ea146cbe87 interactive profile selection in structures + debug 2024-06-11 14:17:38 +02:00
Theophile Terraz c6b28ba3a0 deal with conflicts between onpick and rect select in Profile plot 2024-06-11 11:55:52 +02:00
Theophile Terraz fe57890da4 work on ctrl+click 2024-06-11 10:02:43 +02:00
Theophile Terraz 8cd9ec939b mend 2024-06-10 15:59:32 +02:00
Theophile Terraz 872ff0ef62 PlotKPZ restructuration 2024-06-10 15:59:02 +02:00
Theophile Terraz 052faed48d PlotXY restructuration 2024-06-10 14:58:47 +02:00
Theophile Terraz 31b37a7792 add mouse selection in kpz geometry window 2024-06-07 16:23:09 +02:00
Theophile Terraz dfe12c8bcb work on multiple selection in profil plot 2024-06-07 14:18:56 +02:00
Theophile Terraz 89d5b2d576 work on interactive profil plot 2024-06-06 16:29:26 +02:00
Theophile Terraz 601ef73b5d implement rectangular selection in profile window 2024-06-06 14:30:15 +02:00
Theophile Terraz e4f8bc347d prepare rectangular selection 2024-06-06 14:29:31 +02:00
Pierre-Antoine Rouby 6c4b67d93a pamhyr: Fixing pep8 and use RD and RG by default at update PK (if exists). 2024-05-24 11:53:13 +02:00
Pierre-Antoine Rouby 2ee8420b94 Model: Add status versioning and update study version to 0.0.10. 2024-05-22 17:20:30 +02:00
Pierre-Antoine Rouby bbe127865b Merge branch 'master' of gitlab-ssh.irstea.fr:theophile.terraz/pamhyr 2024-05-22 17:08:25 +02:00
Pierre-Antoine Rouby 50686e1b65 Pamhyr: Disable native file dialog. 2024-05-22 15:56:45 +02:00
Theophile Terraz 7b3c3d891d Merge branch 'master' of gitlab-ssh.irstea.fr:theophile.terraz/pamhyr 2024-05-22 11:43:47 +02:00
Theophile Terraz 98feca7fca add orientation option for update KP 2024-05-22 11:36:12 +02:00
Pierre-Antoine Rouby e45eb3fd69 ci: build-linux: Fix 'pamhyr-x11' path. 2024-05-21 17:27:27 +02:00
Pierre-Antoine Rouby dd86b8b461 Package: Linux: Add script to force pamhyr running with x11. 2024-05-21 17:19:00 +02:00
Theophile Terraz 57d43cd5d3 Merge branch 'master' of gitlab-ssh.irstea.fr:theophile.terraz/pamhyr 2024-05-21 17:02:08 +02:00
Theophile Terraz 14b5b5e3df add purge dialog 2024-05-21 17:01:54 +02:00
Pierre-Antoine Rouby a7e43ca6a7 MainWindow: Info: Sync update with all modelling modules. 2024-05-21 11:08:59 +02:00
Pierre-Antoine Rouby 8ca21c413f IC: Fix display update and optimized discharge plot update. 2024-05-21 10:53:20 +02:00
Pierre-Antoine Rouby 61fd54c496 Geometry, Results: Fix HS display whene not associated. 2024-05-17 16:41:05 +02:00
Pierre-Antoine Rouby b017127b3a Windows: Add nsis icon. 2024-05-17 15:44:48 +02:00
Pierre-Antoine Rouby 52f2d15280 Package: Add icon for windows executable file. 2024-05-17 15:20:17 +02:00
Pierre-Antoine Rouby 5c86e46c72 Checker: Bypass study copy to improve performance. 2024-05-17 14:51:30 +02:00
Pierre-Antoine Rouby a5648851f2 Friction: Add strickler unit. 2024-05-17 14:31:34 +02:00
Pierre-Antoine Rouby cb41520a20 pamhyr: Fix Pep8. 2024-05-17 12:29:20 +02:00
Pierre-Antoine Rouby 537612e5e3 Geometry: Minor change. 2024-05-17 12:22:47 +02:00
Pierre-Antoine Rouby 4e0c671f68 Merge remote-tracking branch 'origin/v0.0.8' 2024-05-17 12:01:17 +02:00
Pierre-Antoine Rouby 507cfb9d53 Friction: Fix reach plot. 2024-05-17 11:55:29 +02:00
Pierre-Antoine Rouby 2013009e8a Mage: Update version. 2024-05-17 11:52:07 +02:00
Pierre-Antoine Rouby f1a1d08458 Pamhyr: Minor change. 2024-05-17 11:51:23 +02:00
Pierre-Antoine Rouby 7de6cc4468 Solver: RubarBE: Disable. 2024-05-17 11:50:07 +02:00
Pierre-Antoine Rouby 783d83e4cc Trad: Update french translate. 2024-05-17 11:48:37 +02:00
Theophile Terraz c366a1a4e7 increase precision of input sediment diameter 2024-05-16 18:06:53 +02:00
Theophile Terraz b784cab94b Merge branch 'v0.0.8' of gitlab-ssh.irstea.fr:theophile.terraz/pamhyr into v0.0.8 2024-05-16 13:01:14 +02:00
Theophile Terraz 78e361d486 debug issue #57 2024-05-16 13:01:03 +02:00
Pierre-Antoine Rouby 3b1f135fdb Geometry: Profile: Change print to logger. 2024-05-15 16:00:58 +02:00
Pierre-Antoine Rouby 2f93d1a83f LC: Use KP delegation. 2024-05-15 15:18:33 +02:00
Pierre-Antoine Rouby 653a26fee1 Geometry: Add hydro. struct. at KPZ plot. 2024-05-15 14:44:42 +02:00
Pierre-Antoine Rouby e33c9cb2f0 Results: Add hydro. struct. on kpc plot. 2024-05-15 14:37:31 +02:00
Pierre-Antoine Rouby 4be43528e3 Pamhyr: Fix Pep8. 2024-05-15 11:29:39 +02:00
Pierre-Antoine Rouby 9d343a42ce IC: Add import action (for BIN file). 2024-05-15 11:09:39 +02:00
Pierre-Antoine Rouby 459722610a IC: Implement copy/paste command. 2024-05-15 09:49:43 +02:00
Theophile Terraz f745dce04e debug 2024-05-14 15:14:29 +02:00
Pierre-Antoine Rouby c1f4aaf957 Merge branch 'arts' into v0.0.8 2024-05-14 14:41:14 +02:00
Pierre-Antoine Rouby f4bf8d3766 Results: Simplify and fix river bottom display. 2024-05-14 14:40:00 +02:00
Pierre-Antoine Rouby 6e94d7a707 Results: PlotH: Add current ts line. 2024-05-14 14:21:33 +02:00
Theophile Terraz 87e8e17e91 forgotten files 2024-05-14 14:13:46 +02:00
Theophile Terraz 9e79178d0f restructure meshing 2024-05-14 11:55:38 +02:00
Theophile Terraz 9d0168a9ee merge 2024-05-13 16:02:36 +02:00
Pierre-Antoine Rouby 9760d41561 Results: Add speed value into row data table. 2024-05-13 15:26:58 +02:00
Pierre-Antoine Rouby 0c039793df IC: Add speed column in table. 2024-05-13 15:24:25 +02:00
Pierre-Antoine Rouby 4cec45131e River: Add a default sediment layers. 2024-05-13 15:00:56 +02:00
Pierre-Antoine Rouby e6c08cb4e4 Geometry: Fix pep8. 2024-05-13 14:45:59 +02:00
Pierre-Antoine Rouby 2d17d8cfe4 Geometry: Add import exception handler. 2024-05-13 14:44:15 +02:00
Pierre-Antoine Rouby ef583fc9bb Geometry: Meshing: Prepare code for next 'mailleurTT' version. 2024-05-13 14:27:24 +02:00
Pierre-Antoine Rouby db87b1e214 Geometry: Add left and right line into XY plot. 2024-05-13 09:56:09 +02:00
Theophile Terraz 5790bdfe82 add purge in profile and geometry 2024-05-07 17:42:43 +02:00
Pierre-Antoine Rouby 731567369e Geometry: Remove old OnPickEvent file. 2024-05-07 17:02:42 +02:00
Pierre-Antoine Rouby 317cbf904d Geometry: Add water level at shift click, end of onpickevent refactoring. 2024-05-07 16:58:29 +02:00
Pierre-Antoine Rouby f4561494f6 Merge branch 'RubarBE' into v0.0.8. 2024-05-07 14:43:58 +02:00
Theophile Terraz 5164891928 add purge 2024-05-07 14:41:09 +02:00
Pierre-Antoine Rouby 5e79e29823 Geometry: Start OnPickEvent refactoring. 2024-05-07 14:12:53 +02:00
Pierre-Antoine Rouby cc3649fcb8 MeshingTools: Fix sync call. 2024-05-03 15:57:38 +02:00
Pierre-Antoine Rouby ab11d15729 Results: Plot: Fix display whene more than 10 sl. 2024-05-03 15:03:43 +02:00
Pierre-Antoine Rouby 0e75d67671 Geometry: Fix display with more than 10 guidelines. 2024-05-03 14:56:11 +02:00
Pierre-Antoine Rouby ba81f923d9 Results: CustomPlot: Use automatic x and y lim. 2024-05-03 14:40:53 +02:00
Theophile Terraz a0cdb182fd debug 2024-05-02 15:05:55 +02:00
Theophile Terraz ca7296ca8c translation 2024-04-30 16:31:53 +02:00
Theophile Terraz fb2d0d3e71 Merge branch 'master' of gitlab-ssh.irstea.fr:theophile.terraz/pamhyr 2024-04-30 14:24:15 +02:00
Theophile Terraz 2afd641f0f debug param init_internal 2024-04-30 14:22:42 +02:00
Theophile Terraz d871bbfc3b debug enable hydraulic structures 2024-04-30 14:18:12 +02:00
Pierre-Antoine Rouby 5e9ab81363 tools: Add relase mail generator. 2024-04-26 12:10:32 +02:00
Pierre-Antoine Rouby d0acface41 Geometry, doc: Fix pointXYZ foreign key. 2024-04-26 10:04:43 +02:00
Pierre-Antoine Rouby aea7cdb823 doc: dev: Add database scheme images. 2024-04-26 09:40:14 +02:00
Pierre-Antoine Rouby e066c6ab94 Pamhyr2: Fix pep8. 2024-04-24 16:38:25 +02:00
Pierre-Antoine Rouby 8e82772cfc Checker: Add modules impact and add IC and BC study checker. 2024-04-24 16:32:52 +02:00
Pierre-Antoine Rouby 441a064d84 Solver: CommandLine: Try fix Windows encoding crash. 2024-04-24 09:51:58 +02:00
Pierre-Antoine Rouby ee6b2a94eb Results: Add plot toolsbar iso view and back/forward option. 2024-04-24 09:51:28 +02:00
Pierre-Antoine Rouby 07761716e4 Solver: Mage: Check is node is enable before BC export. 2024-04-24 09:33:33 +02:00
Pierre-Antoine Rouby 6dc96b86b4 Merge branch 'shape' into v0.0.8 2024-04-23 14:35:01 +02:00
Pierre-Antoine Rouby 8c2874e04d Geometry: Compute KP at shape file import. 2024-04-23 14:32:21 +02:00
Pierre-Antoine Rouby c7001bf0fa Network: Fix pep8. 2024-04-23 14:10:55 +02:00
Pierre-Antoine Rouby f62b209ffb Merge branch 'multiple-edge-between-nodes' into v0.0.8 2024-04-23 14:07:54 +02:00
Pierre-Antoine Rouby 0fe71b2b92 Pamhyr2: Fix full requirements list. 2024-04-23 14:03:01 +02:00
Pierre-Antoine Rouby e528ba5278 Geometry: Add shapefile import file format. 2024-04-23 11:50:35 +02:00
Pierre-Antoine Rouby fd1b2b1041 doc: dev: Minor change. 2024-04-23 09:09:54 +02:00
Pierre-Antoine Rouby e5422c71f5 View: Network: Minor change. 2024-04-22 16:48:17 +02:00
Pierre-Antoine Rouby efd8960e1e Pamhyr2: Fix pep8. 2024-04-22 16:46:50 +02:00
Pierre-Antoine Rouby 8dcb0a298e Pamhyr2: View: Tools: Add comment. 2024-04-22 16:40:58 +02:00
Pierre-Antoine Rouby acbd3537c0 Pamhyr2: Minor change and method documentation. 2024-04-22 16:29:23 +02:00
Pierre-Antoine Rouby 145c301eed doc: dev: Update documentation. 2024-04-22 16:28:41 +02:00
Pierre-Antoine Rouby ec91a8b63f Solver: Fix SIN export and error handler at export. 2024-04-22 11:33:07 +02:00
Pierre-Antoine Rouby ffdeb28bfd Network: Fix potential crash with maptoscene. 2024-04-22 11:07:42 +02:00
Pierre-Antoine Rouby df1ac5586e IC: Fix incline computation in case of zero marging. 2024-04-19 16:31:12 +02:00
Pierre-Antoine Rouby ab1b7a2605 Save/Backup: Add mutual exclusion on save methode and block backup signal. 2024-04-19 10:39:25 +02:00
Pierre-Antoine Rouby 53456cf7fa IC: Fix generator spinbox max size. 2024-04-17 14:47:49 +02:00
Pierre-Antoine Rouby 1cbdda1ec5 Solver: Log: Fix new line spliting (depend of OS). 2024-04-17 14:46:17 +02:00
Pierre-Antoine Rouby ba2fa25636 Pamhyr2: Fix some fonts size and family. 2024-04-17 14:45:32 +02:00
Pierre-Antoine Rouby 5189cbaa6f Windows: Fix logging file encoding. 2024-04-17 14:27:01 +02:00
Pierre-Antoine Rouby f1ce2e4b10 Windows: Enable console but hide it. 2024-04-17 12:18:28 +02:00
Pierre-Antoine Rouby 50ae49c0ec Windows: Fix UTF-8 en/decode. 2024-04-17 12:17:25 +02:00
Pierre-Antoine Rouby 502c2240f7 Solver: Parameters: Fix Y/N french translate. 2024-04-17 09:42:17 +02:00
Pierre-Antoine Rouby b784f00b92 Solver: Mage: Fix pep8. 2024-04-16 15:39:51 +02:00
Pierre-Antoine Rouby 0c0d7f3481 Network: Fix edge delete whene there are multiple edge between two node. 2024-04-16 15:15:30 +02:00
Pierre-Antoine Rouby 457dac477a Solver: Mage: Minor change. 2024-04-16 14:01:59 +02:00
Pierre-Antoine Rouby dc114f759d Solver: Mage: Enable LC export. 2024-04-16 12:04:19 +02:00
Pierre-Antoine Rouby 672f81f216 MeshingTools: Fix mailleurTT path. 2024-04-16 10:47:07 +02:00
Pierre-Antoine Rouby 3516986ff2 Geometry: Meshing: Fix space step maximum value to 9999.99 (m). 2024-04-16 10:12:15 +02:00
Pierre-Antoine Rouby 7ecd98a5fb Solver: Mage: Add error file by default. 2024-04-16 08:32:19 +02:00
Pierre-Antoine Rouby 1fdeab9685 Geometry: Meshing: Minor change. 2024-04-15 15:42:37 +02:00
Pierre-Antoine Rouby ebd4e64cc3 Pamhyr2: Update Mage version to v8.3.5 2024-04-15 11:39:06 +02:00
Pierre-Antoine Rouby 1452f4129f Config: Fix configuration save whene click cancel. 2024-04-15 11:16:41 +02:00
Pierre-Antoine Rouby 04d43f3f28 Backup: Fix backup. 2024-04-15 11:16:20 +02:00
Pierre-Antoine Rouby b7ebf961ac Geometry: Fix geometry loading. 2024-04-15 10:29:56 +02:00
Pierre-Antoine Rouby 28c7a0177a HS: Basic: Fix plot AC init. 2024-04-15 09:35:10 +02:00
Pierre-Antoine Rouby 9c5ea0bdd9 Network: Reverse order for edges adding in scene to fix mouse selection. 2024-04-10 16:36:14 +02:00
Pierre-Antoine Rouby cd540b0146 Network: Display edge as an arc. 2024-04-10 16:24:12 +02:00
Pierre-Antoine Rouby 62860750f2 Config: Fix useless print. 2024-04-09 09:37:29 +02:00
Pierre-Antoine Rouby 4d8277c5c2 Pamhyr2: Fix pep8. 2024-04-08 16:25:25 +02:00
Pierre-Antoine Rouby b9fddcf78c LC: Add DMY datetime support. 2024-04-08 16:13:23 +02:00
Pierre-Antoine Rouby a834e1905d BC: Add DMY datetime support. 2024-04-08 16:10:39 +02:00
Pierre-Antoine Rouby de1f5c1944 RunSolver: Fix copy methode. 2024-04-08 15:37:00 +02:00
Pierre-Antoine Rouby 6d9b193d25 Solver: RubarBE: Add RubarBE at solvers and fix export. 2024-04-08 15:11:04 +02:00
Pierre-Antoine Rouby 982d346264 Solver: RubarBE: Add tps export. 2024-04-08 13:45:52 +02:00
Pierre-Antoine Rouby 4c927eb7c7 Model: Profile: Add speed computation methode. 2024-04-08 11:40:12 +02:00
Pierre-Antoine Rouby fead93f0b9 Solver: RubarBE: Add ts export. 2024-04-08 11:39:53 +02:00
Pierre-Antoine Rouby ca3ba3adb6 Solver: RubarBE: Add frictions export. 2024-04-05 16:45:13 +02:00
Pierre-Antoine Rouby 9f87a38107 Solver: RubarBE: Add MAIL file. 2024-04-05 16:03:55 +02:00
Pierre-Antoine Rouby 329572d191 Solver: RubarBE: Add rubarbe solver (WIP). 2024-04-05 15:28:05 +02:00
Pierre-Antoine Rouby ae21315593 Network: Fix crash when edge addition click on the void. 2024-04-05 11:15:09 +02:00
Pierre-Antoine Rouby b209547a5b Solver: Disable mage fake 7. 2024-04-03 17:48:09 +02:00
Pierre-Antoine Rouby 8a0f214144 Arts: Fix missing icon. 2024-04-03 16:12:29 +02:00
Pierre-Antoine Rouby 5ff3b9457a HydraulicStructures: Basic: Fix table setup. 2024-04-03 16:01:24 +02:00
Pierre-Antoine Rouby 8ec25cac46 Arts: Delete old and useless icons. 2024-04-03 15:59:58 +02:00
Pierre-Antoine Rouby 39180f8ea3 Arts: Add reload icon. 2024-04-03 15:53:42 +02:00
Pierre-Antoine Rouby 2203b2ca8b Arts: Add zoom icons. 2024-04-03 15:13:39 +02:00
Pierre-Antoine Rouby faab0d78cf tests_cases: Fix and update Saar study. 2024-04-03 14:49:19 +02:00
Pierre-Antoine Rouby c3a5ea1b91 Arts: Add meshing icon. 2024-04-03 11:56:47 +02:00
Pierre-Antoine Rouby 6349c4f752 Arts: Renew up/down/left/right and import/export. 2024-04-03 11:36:36 +02:00
Pierre-Antoine Rouby 8c5fa68f11 Arts: Add open, renew logo and minor change. 2024-04-03 10:42:51 +02:00
Pierre-Antoine Rouby a487cd8e64 AddFile, REPLine: Fix modified status. 2024-04-02 17:07:04 +02:00
Pierre-Antoine Rouby 92eac86c68 AddFiles, REPLine: Add undo commands. 2024-04-02 16:18:21 +02:00
Pierre-Antoine Rouby cd2e29a77d MainWindow: Set menu bar to not bold. 2024-04-02 09:28:11 +02:00
Pierre-Antoine Rouby ce090c6437 Plot: ToolBar: Minor change. 2024-04-02 09:26:11 +02:00
Pierre-Antoine Rouby f80be0f3da Arts: Update SolverLog window. 2024-03-29 17:41:11 +01:00
Pierre-Antoine Rouby fd9a338365 Arts: Minor change. 2024-03-29 17:37:54 +01:00
Pierre-Antoine Rouby 776164310b Arts: Add edit icon and some minor change. 2024-03-29 17:10:33 +01:00
Pierre-Antoine Rouby 52ab9f4ade Arts: MainWindow: Add menu icons. 2024-03-29 16:02:59 +01:00
Pierre-Antoine Rouby 88c75c0d35 Arts: Add save icon and some change. 2024-03-29 16:00:05 +01:00
Pierre-Antoine Rouby cd571b6ec4 Arts: Update geometry and minor change. 2024-03-29 15:05:18 +01:00
Pierre-Antoine Rouby 701f9ab69b arts: Renew friction, add run, sort, up and down icons. 2024-03-29 15:00:59 +01:00
Pierre-Antoine Rouby 6d47664767 arts: Add network, geometry, friction, etc. icons and add toolbar on mainwindow. 2024-03-29 12:02:12 +01:00
Pierre-Antoine Rouby 8cb8c50998 arts: Add new logo as ascii art. 2024-03-27 18:20:24 +01:00
Pierre-Antoine Rouby f14a149748 arts: Update Pamhyr2 logo. 2024-03-27 17:57:08 +01:00
Pierre-Antoine Rouby beb2c2fec0 arts: Renew add and del icons and add import and export logo. 2024-03-27 17:50:45 +01:00
Pierre-Antoine Rouby 48e010f195 Pamhyr2: Add new logo. 2024-03-27 16:14:52 +01:00
Pierre-Antoine Rouby e350b7d8f5 Solver: Mage8: Do not add additional files into REP file. 2024-03-27 10:50:23 +01:00
Pierre-Antoine Rouby e7e1582836 MainWindow: Checker: Minor change and fixes. 2024-03-27 09:57:53 +01:00
Pierre-Antoine Rouby 9d9a8a7ced MainWindow: Info: Simplify study section. 2024-03-27 09:40:14 +01:00
Pierre-Antoine Rouby aef81d088c Meshing: Add trivial MashingWithMAgeMailleurTT unit tests. 2024-03-27 09:36:34 +01:00
Pierre-Antoine Rouby 35b421b4df Pamhyr2: Fix Pep8. 2024-03-27 09:33:43 +01:00
Pierre-Antoine Rouby dbbae7a53f Translate: Rebuild Pamhyr2 french translate. 2024-03-27 09:27:37 +01:00
Pierre-Antoine Rouby b9ace9f58f MainWindow: Remove obsolete actions. 2024-03-26 17:54:32 +01:00
Pierre-Antoine Rouby 48aa5cb1aa Config: Update config with new mage_fake7 solver and new mage path. 2024-03-26 15:29:45 +01:00
Pierre-Antoine Rouby b517c38cb9 Solver: Mage: Add mage fake 7. 2024-03-26 15:18:05 +01:00
Pierre-Antoine Rouby d158c8ad7a Windows: Fix 'pwd' import. 2024-03-26 13:49:56 +01:00
Pierre-Antoine Rouby 8c9eeb6160 Windows: Fix NSI file. 2024-03-26 11:02:02 +01:00
Pierre-Antoine Rouby aedb8c46fc REPLines: Add REP lines into model, view and minor change. 2024-03-26 10:40:57 +01:00
Pierre-Antoine Rouby 1ff895368e pamhyr: Fix save after backup and minor change. 2024-03-26 08:36:45 +01:00
Pierre-Antoine Rouby 48e30125bb AdditionalFiles: Add model, view and solver export. 2024-03-25 17:21:31 +01:00
Pierre-Antoine Rouby db416d25de pamhyr: Fix Pep8. 2024-03-25 14:04:55 +01:00
Pierre-Antoine Rouby da18a8b712 Study: Add default description example. 2024-03-25 13:57:27 +01:00
Pierre-Antoine Rouby 48c98cddba Study: Switch rich text to markdown. 2024-03-22 17:07:13 +01:00
Pierre-Antoine Rouby 16a0e03d1d Study: Add riche text option for study description editing. 2024-03-22 16:12:57 +01:00
Pierre-Antoine Rouby e0f580daf6 windows: Fix widgets path. 2024-03-22 15:13:35 +01:00
Pierre-Antoine Rouby c1c16964eb windows: Disable console. 2024-03-22 15:03:47 +01:00
Pierre-Antoine Rouby d0134d6415 ci: Fix hash path. 2024-03-22 14:54:15 +01:00
Pierre-Antoine Rouby 9f4c759ea3 package: Fix exe and _internal position. 2024-03-22 14:51:10 +01:00
Pierre-Antoine Rouby 661e2071d1 Package: Rename mage to mage8 and reorganise package installation dir. 2024-03-22 14:43:51 +01:00
Pierre-Antoine Rouby 0f25334032 pamhyr: Close study DB if already init. 2024-03-22 14:03:46 +01:00
Pierre-Antoine Rouby 334ada3dcc pamhyr: Add missing dependencies. 2024-03-22 11:32:49 +01:00
Pierre-Antoine Rouby df0b5bb014 MainWindow: Minor change. 2024-03-22 11:24:58 +01:00
Pierre-Antoine Rouby 20686086d8 pamhyr: Add CONFIG modules. 2024-03-22 11:23:25 +01:00
Pierre-Antoine Rouby 1be8f6c434 MainWindow: Add status at save/backup. 2024-03-22 11:17:18 +01:00
Pierre-Antoine Rouby f8c16d8c7c pamhyr: Reopen backup if newer than last save in case of crash. 2024-03-22 10:55:48 +01:00
Pierre-Antoine Rouby 7f3cf46ee1 pamhyr: Add periodic backup. 2024-03-22 09:51:53 +01:00
Pierre-Antoine Rouby 632cd2cd4c Checker: Fix checker table update at study close. 2024-03-22 08:45:37 +01:00
Pierre-Antoine Rouby 0f5243f36a Geometry, checker: Minor change. 2024-03-22 08:29:04 +01:00
Pierre-Antoine Rouby 137368cc27 pamhyr: Fix pep8. 2024-03-20 17:08:11 +01:00
Pierre-Antoine Rouby e343d284d9 MainWindow: Checker: Add lists. 2024-03-20 17:03:37 +01:00
Pierre-Antoine Rouby a23692a29c MainWindow: Checker: Add worker. 2024-03-20 16:08:53 +01:00
Pierre-Antoine Rouby 30170df1b1 Mainwindow: Checker: Add table. 2024-03-20 12:24:13 +01:00
Pierre-Antoine Rouby 3be4e6c112 MainWindow: Fix missing file. 2024-03-20 09:25:59 +01:00
Pierre-Antoine Rouby e5650b5378 MainWindow: Checkers: Add checker tab template. 2024-03-19 17:21:27 +01:00
Pierre-Antoine Rouby 67f534a6f6 Checker: Add related module(s). 2024-03-19 12:00:44 +01:00
Pierre-Antoine Rouby 51031e7359 Pamhyr: Add pamhyr modules flags enum and checker status OR operator. 2024-03-19 09:18:28 +01:00
Pierre-Antoine Rouby e215e0f14a MainWindow: Info: Minor change. 2024-02-28 16:49:10 +01:00
Pierre-Antoine Rouby 1099d5f282 Pamhyr2: Fix Pep8. 2024-02-28 16:28:54 +01:00
Pierre-Antoine Rouby 330761a372 MainWindow: Info: Add some informations. 2024-02-28 16:27:09 +01:00
Pierre-Antoine Rouby 81f38b942c Reservoir: Edit: Fix undo/redo update. 2024-02-28 15:44:58 +01:00
Pierre-Antoine Rouby 07cefceed8 Network: Add reservoir shortcut at network edit. 2024-02-28 15:40:57 +01:00
Pierre-Antoine Rouby ed5e520b35 Network: Refacto graph context menu. 2024-02-28 10:49:38 +01:00
Pierre-Antoine Rouby 36b62260b1 Results: Fix results export. 2024-02-27 15:40:21 +01:00
Pierre-Antoine Rouby 3e085dd7c6 MainWindow: Add minimal info tab. 2024-02-27 15:40:08 +01:00
Pierre-Antoine Rouby a18fd29795 pamhyr: Fix Pep8. 2024-02-26 16:37:09 +01:00
Pierre-Antoine Rouby 76e4f4b80e Results: Add waiting window at results reading. 2024-02-26 16:33:45 +01:00
Pierre-Antoine Rouby 8466531916 MainWindow: Add BC short cut for sediment. 2024-02-26 14:21:49 +01:00
Pierre-Antoine Rouby a4c5d34720 Pamhyr: Centralize ND and NA translate. 2024-02-21 17:19:50 +01:00
Pierre-Antoine Rouby 2e158f4c1b Pamhyr: Factorize some translate. 2024-02-21 15:52:16 +01:00
354 changed files with 12116 additions and 6219 deletions

No files matched your search

+55 -52
View File
@@ -1,5 +1,5 @@
# .gitlab-ci.yml -- Pamhyr gitlab-ci # .gitlab-ci.yml -- Pamhyr gitlab-ci
# Copyright (C) 2023 INRAE # Copyright (C) 2023-2024 INRAE
# #
# This program is free software: you can redistribute it and/or modify # 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 # it under the terms of the GNU General Public License as published by
@@ -29,9 +29,9 @@ stages:
############# #############
variables: variables:
MAGE_VERSION: "v8.3.4" MAGE_8_VERSION: "v8.3.6"
dl-mage-doc: dl-mage8-doc:
stage: downloads stage: downloads
tags: tags:
- linux - linux
@@ -40,46 +40,48 @@ dl-mage-doc:
script: script:
- mkdir -p doc - mkdir -p doc
- cd doc - cd doc
- curl -L -o mage8.pdf https://gitlab.irstea.fr/jean-baptiste.faure/mage/-/releases/$MAGE_VERSION/downloads/documents/Mage-8_Documentation.pdf - curl -L -o mage8.pdf https://gitlab.irstea.fr/jean-baptiste.faure/mage/-/releases/$MAGE_8_VERSION/downloads/documents/Mage-8_Documentation.pdf
artifacts: artifacts:
paths: paths:
- doc/mage8.pdf - doc/mage8.pdf
dl-mage-linux: dl-mage8-linux:
stage: downloads stage: downloads
tags: tags:
- linux - linux
rules: rules:
- if: $CI_COMMIT_BRANCH == 'ci-test' || $CI_COMMIT_BRANCH == 'master' || $CI_COMMIT_TAG - if: $CI_COMMIT_BRANCH == 'ci-test' || $CI_COMMIT_BRANCH == 'master' || $CI_COMMIT_TAG
script: script:
- curl -L -o mage.tgz https://gitlab.irstea.fr/jean-baptiste.faure/mage/-/releases/$MAGE_VERSION/downloads/packages/mage_linux.tgz - curl -L -o mage8.tgz https://gitlab.irstea.fr/jean-baptiste.faure/mage/-/releases/$MAGE_8_VERSION/downloads/packages/mage_linux.tgz
- mkdir -p mage-linux - mkdir -p mage8-linux
- cd mage-linux - cd mage8-linux
- tar xvf ../mage.tgz - tar xvf ../mage8.tgz
artifacts: artifacts:
paths: paths:
- mage-linux/mage - mage8-linux/mage
- mage-linux/mage_extraire - mage8-linux/mage_as7
- mage-linux/mailleurTT - mage8-linux/mage_extraire
- mage-linux/libbief.so - mage8-linux/mailleurTT
- mage8-linux/libbief.so
dl-mage-windows: dl-mage8-windows:
stage: downloads stage: downloads
tags: tags:
- linux - linux
rules: rules:
- if: $CI_COMMIT_BRANCH == 'ci-test' || $CI_COMMIT_BRANCH == 'master' || $CI_COMMIT_TAG - if: $CI_COMMIT_BRANCH == 'ci-test' || $CI_COMMIT_BRANCH == 'master' || $CI_COMMIT_TAG
script: script:
- curl -L -o mage.tgz https://gitlab.irstea.fr/jean-baptiste.faure/mage/-/releases/$MAGE_VERSION/downloads/packages/mage_windows.tgz - curl -L -o mage8.tgz https://gitlab.irstea.fr/jean-baptiste.faure/mage/-/releases/$MAGE_8_VERSION/downloads/packages/mage_windows.tgz
- mkdir -p mage-windows - mkdir -p mage8-windows
- cd mage-windows - cd mage8-windows
- tar xvf ../mage.tgz - tar xvf ../mage8.tgz
artifacts: artifacts:
paths: paths:
- mage-windows/mage.exe - mage8-windows/mage.exe
- mage-windows/mage_extraire.exe - mage8-windows/mage_as7.exe
- mage-windows/mailleurTT.exe - mage8-windows/mage_extraire.exe
- mage-windows/libbief.dll - mage8-windows/mailleurTT.exe
- mage8-windows/libbief.dll
############# #############
# CONFIGURE # # CONFIGURE #
@@ -222,9 +224,9 @@ build-linux:
rules: rules:
- if: $CI_COMMIT_BRANCH == 'ci-test' || $CI_COMMIT_BRANCH == 'master' || $CI_COMMIT_TAG - if: $CI_COMMIT_BRANCH == 'ci-test' || $CI_COMMIT_BRANCH == 'master' || $CI_COMMIT_TAG
needs: needs:
- job: dl-mage-linux - job: dl-mage8-linux
artifacts: true artifacts: true
- job: dl-mage-doc - job: dl-mage8-doc
artifacts: true artifacts: true
- job: set-version - job: set-version
artifacts: true artifacts: true
@@ -249,28 +251,32 @@ build-linux:
- pyinstaller -y --paths linux-venv/lib/python3.8/site-packages ../src/pamhyr.py - pyinstaller -y --paths linux-venv/lib/python3.8/site-packages ../src/pamhyr.py
# Create directory # Create directory
- mkdir -p pamhyr - mkdir -p pamhyr
- mkdir -p pamhyr/pamhyr - mkdir -p pamhyr/_internal
- mkdir -p pamhyr/pamhyr/_internal
# Copy Pamhyr info # Copy Pamhyr info
- cp -v ../VERSION pamhyr/pamhyr/_internal/ - cp -v ../VERSION pamhyr/_internal/
- cp -v ../AUTHORS pamhyr/pamhyr/_internal/ - cp -v ../AUTHORS pamhyr/_internal/
- cp -v ../LICENSE pamhyr/pamhyr/_internal/ - cp -v ../LICENSE pamhyr/_internal/
- cp -v ../src/motd.txt pamhyr/pamhyr/_internal/ - cp -v ../src/motd.txt pamhyr/_internal/
# Copy mage # Copy mage
- mkdir -p pamhyr/mage - mkdir -p pamhyr/mage8
- cp -v ../mage-linux/* pamhyr/mage/ - cp -v ../mage8-linux/* pamhyr/mage8/
# Copy Pamhyr # Copy Pamhyr
- cp -r dist/pamhyr pamhyr/ - cp -r dist/pamhyr/* pamhyr/
# Pamhyr script to force x11
- echo '#!/usr/bin/env bash' > pamhyr/pamhyr-x11
- echo "" >> pamhyr/pamhyr-x11
- echo "GDK_BACKEND=x11 ./pamhyr $@" >> pamhyr/pamhyr-x11
- chmod +x pamhyr/pamhyr-x11
# Copy Pamhyr resources # Copy Pamhyr resources
- mkdir -p pamhyr/pamhyr/_internal/View - mkdir -p pamhyr/_internal/View
- mkdir -p pamhyr/pamhyr/_internal/View/ui - mkdir -p pamhyr/_internal/View/ui
- mkdir -p pamhyr/pamhyr/_internal/View/ui/Widgets - mkdir -p pamhyr/_internal/View/ui/Widgets
- cp -r ../src/View/ui/*.ui pamhyr/pamhyr/_internal/View/ui/ - cp -r ../src/View/ui/*.ui pamhyr/_internal/View/ui/
- cp -r ../src/View/ui/ressources/ pamhyr/pamhyr/_internal/View/ui - cp -r ../src/View/ui/ressources/ pamhyr/_internal/View/ui
- cp -r ../src/View/ui/Widgets/*.ui pamhyr/pamhyr/_internal/View/ui/Widgets - cp -r ../src/View/ui/Widgets/*.ui pamhyr/_internal/View/ui/Widgets
# Copy lang # Copy lang
- mkdir -p pamhyr/pamhyr/_internal/lang - mkdir -p pamhyr/_internal/lang
- cp -r ../src/lang/*.qm pamhyr/pamhyr/_internal/lang - cp -r ../src/lang/*.qm pamhyr/_internal/lang
# Copy tests cases # Copy tests cases
- mkdir -p pamhyr/tests_cases/ - mkdir -p pamhyr/tests_cases/
- mkdir -p pamhyr/tests_cases/Saar - mkdir -p pamhyr/tests_cases/Saar
@@ -298,10 +304,6 @@ build-linux:
# Copy TP charriage # Copy TP charriage
- mkdir -p pamhyr/doc/TP_charriage/ - mkdir -p pamhyr/doc/TP_charriage/
- cp ../doc/users/TP_charriage/* pamhyr/doc/TP_charriage - cp ../doc/users/TP_charriage/* pamhyr/doc/TP_charriage
# Create running script
- echo "#!/bin/sh" > pamhyr/Pamhyr2
- echo "./pamhyr/pamhyr $@" >> pamhyr/Pamhyr2
- chmod u+x pamhyr/Pamhyr2
artifacts: artifacts:
paths: paths:
- linux/pamhyr - linux/pamhyr
@@ -313,9 +315,9 @@ build-windows:
rules: rules:
- if: $CI_COMMIT_BRANCH == 'ci-test' || $CI_COMMIT_BRANCH == 'master' || $CI_COMMIT_TAG - if: $CI_COMMIT_BRANCH == 'ci-test' || $CI_COMMIT_BRANCH == 'master' || $CI_COMMIT_TAG
needs: needs:
- job: dl-mage-windows - job: dl-mage8-windows
artifacts: true artifacts: true
- job: dl-mage-doc - job: dl-mage8-doc
artifacts: true artifacts: true
- job: set-version - job: set-version
artifacts: true artifacts: true
@@ -335,7 +337,8 @@ build-windows:
- python -m pip install -r ..\requirements.txt - python -m pip install -r ..\requirements.txt
- python -m pip install -U -r ..\requirements.txt - python -m pip install -U -r ..\requirements.txt
# Run Pyinstaller # Run Pyinstaller
- pyinstaller -y ..\src\pamhyr.py # - pyinstaller --noconsole -y ..\src\pamhyr.py
- pyinstaller -i ../src/View/ui/ressources/icon.ico --hide-console hide-early -y ..\src\pamhyr.py
# Create directory # Create directory
- mkdir pamhyr - mkdir pamhyr
- dir - dir
@@ -421,10 +424,10 @@ linux-build-hash:
- if: $CI_COMMIT_BRANCH == 'ci-test' || $CI_COMMIT_BRANCH == 'master' || $CI_COMMIT_TAG - if: $CI_COMMIT_BRANCH == 'ci-test' || $CI_COMMIT_BRANCH == 'master' || $CI_COMMIT_TAG
script: script:
- cd linux - cd linux
- sha256sum -b pamhyr/pamhyr/pamhyr > pamhyr/pamhyr/pamhyr.sha256 - sha256sum -b pamhyr/pamhyr > pamhyr/pamhyr.sha256
artifacts: artifacts:
paths: paths:
- linux/pamhyr/pamhyr/pamhyr.sha256 - linux/pamhyr/pamhyr.sha256
windows-build-hash: windows-build-hash:
stage: hash stage: hash
@@ -438,10 +441,10 @@ windows-build-hash:
- if: $CI_COMMIT_BRANCH == 'ci-test' || $CI_COMMIT_BRANCH == 'master' || $CI_COMMIT_TAG - if: $CI_COMMIT_BRANCH == 'ci-test' || $CI_COMMIT_BRANCH == 'master' || $CI_COMMIT_TAG
script: script:
- cd windows - cd windows
- sha256sum -b pamhyr/pamhyr/pamhyr.exe > pamhyr/pamhyr/pamhyr.exe.sha256 - sha256sum -b pamhyr/pamhyr.exe > pamhyr/pamhyr.exe.sha256
artifacts: artifacts:
paths: paths:
- windows/pamhyr/pamhyr/pamhyr.exe.sha256 - windows/pamhyr/pamhyr.exe.sha256
pkg-hash: pkg-hash:
stage: hash stage: hash
BIN
View File
Binary file not shown.
BIN
View File
Binary file not shown.
BIN
View File
Binary file not shown.
BIN
View File
Binary file not shown.
BIN
View File
Binary file not shown.
BIN
View File
Binary file not shown.
BIN
View File
Binary file not shown.
BIN
View File
Binary file not shown.
BIN
View File
Binary file not shown.
BIN
View File
Binary file not shown.
BIN
View File
Binary file not shown.
BIN
View File
Binary file not shown.
BIN
View File
Binary file not shown.
BIN
View File
Binary file not shown.
BIN
View File
Binary file not shown.
BIN
View File
Binary file not shown.
BIN
View File
Binary file not shown.
+83 -46
View File
@@ -1,5 +1,5 @@
# documentation.org -- Pamhyr developers documentation # documentation.org -- Pamhyr developers documentation
# Copyright (C) 2023 INRAE # Copyright (C) 2023-2024 INRAE
# #
# This program is free software: you can redistribute it and/or modify # 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 # it under the terms of the GNU General Public License as published by
@@ -369,6 +369,7 @@ of Bar (Listing [[sql-bar]] and [[sql-foo]]).
CREATE TABLE bar ( CREATE TABLE bar (
id INTEGER NOT NULL PRIMARY KEY, id INTEGER NOT NULL PRIMARY KEY,
x INTEGER NOT NULL, x INTEGER NOT NULL,
y INTEGER NOT NULL,
foo_id INTEGER NOT NULL, foo_id INTEGER NOT NULL,
FOREIGN KEY(foo_id) REFERENCES foo(id), FOREIGN KEY(foo_id) REFERENCES foo(id),
)""") )""")
@@ -455,6 +456,15 @@ of Bar (Listing [[sql-bar]] and [[sql-foo]]).
bar._sql_save(execute, data=data) bar._sql_save(execute, data=data)
#+end_src #+end_src
Let see the results database scheme for Pamhyr2 at version v0.0.7 in
Figure [[sql_schema]].
#+NAME: sql_schema
#+ATTR_LATEX: :width 16cm
#+CAPTION: SQLite database scheme at Pamhyr2 version v0.0.7 (generate with [[https://gitlab.com/Screwtapello/sqlite-schema-diagram]])
[[./images/schema_v0.0.7.png]]
[fn:sqlite] The SQLite web site: https://www.sqlite.org/index.html [fn:sqlite] The SQLite web site: https://www.sqlite.org/index.html
(last access 2023-09-20) (last access 2023-09-20)
@@ -538,44 +548,44 @@ generic value from optional parameters, for examples:
#+NAME: window #+NAME: window
#+CAPTION: Example of Pamhyr2 window #+CAPTION: Example of Pamhyr2 window
#+begin_src python :python python3 :results output :noweb yes #+begin_src python :python python3 :results output :noweb yes
from View.Tools.PamhyrWindow import PamhyrWindow from View.Tools.PamhyrWindow import PamhyrWindow
from View.My.Translate import MyTranslate from View.My.Translate import MyTranslate
from View.My.Table import MyTableModel from View.My.Table import MyTableModel
class MyWindow(PamhyrWindow): class MyWindow(PamhyrWindow):
_pamhyr_ui = "MyUI" _pamhyr_ui = "MyUI"
_pamhyr_name = "My window" _pamhyr_name = "My window"
def __init__(self, my_data=None, def __init__(self, study=None, config=None,
study=None, config=None, my_data=None,
parent=None): parent=None):
self._my_data = my_data self._my_data = my_data
super(MyWindow, self).__init__( super(MyWindow, self).__init__(
# Window title # Window title
title = self._pamhyr_name + " - " + study.name, title = self._pamhyr_name + " - " + study.name,
# Window standard data # Window standard data
study = study, config = config, study = study, config = config,
trad = MyTranslate(), trad = MyTranslate(),
parent = parent, parent = parent,
# Activate undo/redo and copy/paste shortcut # Activate undo/redo and copy/paste shortcut
options = ["undo", "copy"] options = ["undo", "copy"]
) )
# Add custom data to hash window computation # Add custom data to hash window computation
self._hash_data.append(self._my_data) self._hash_data.append(self._my_data)
# Setup custom window components # Setup custom window components
self.setup_table() self.setup_table()
self.setup_connections() self.setup_connections()
def setup_table(self): def setup_table(self):
# Init table(s)... # Init table(s)...
def setup_connections(self): def setup_connections(self):
# Init action connection(s)... # Init action connection(s)...
# ... # ...
#+end_src #+end_src
Typically we called method =setup_*=, the method to initialize some Typically we called method =setup_*=, the method to initialize some
@@ -636,7 +646,7 @@ class AddNodeCommand(QUndoCommand):
self._node = node self._node = node
def undo(self): def undo(self):
self._graph.remove_node(self._node.name) self._graph.remove_node(self._node)
def redo(self): def redo(self):
self._graph.insert_node(self._node) self._graph.insert_node(self._node)
@@ -661,11 +671,13 @@ creation, this stack is accessible at =self._undo_stack=.
*** Plot *** Plot
To define a new plot you can create a class who inherit to APlot. The To define a new plot you can create a class who inherit to
creator need threee argument: PamhyrPlot. The creator need at leaste five argument:
- A =canvas= of type =MplCanvas= - A =canvas= of type =MplCanvas=
- A (optional) =trad= of type =PamhyrTranslate=
- A =data= used in =draw= and =update= to create and update the plot - A =data= used in =draw= and =update= to create and update the plot
- A optional =toolbar= of type =PamhyrToolbar= - A optional =toolbar= of type =PamhyrToolbar=
- A =parent= window
This class must implement two method =draw= and =update=, the first This class must implement two method =draw= and =update=, the first
method to draw the plot from scratch, the second to update the plot if method to draw the plot from scratch, the second to update the plot if
data has changed. data has changed.
@@ -673,19 +685,36 @@ data has changed.
#+begin_src python :python python3 :results output :noweb yes #+begin_src python :python python3 :results output :noweb yes
from View.Tools.PamhyrPlot import PamhyrPlot from View.Tools.PamhyrPlot import PamhyrPlot
class MyPlot(APlot): class MyPlot(PamhyrPlot):
def __init__(self, canvas=None, data=None, toolbar=None): def __init__(self, canvas=None, trad=None, toolbar=None
data=None, parent=None):
super(MyPlot, self).__init__( super(MyPlot, self).__init__(
canvas=canvas, canvas=canvas,
trad=trad,
data=data, data=data,
toolbar=toolbar toolbar=toolbar,
parent=parent
) )
self.label_x = self._trad["x"]
self.label_y = self._trad["y"]
# Optional configuration
self._isometric_axis = False
self._auto_relim_update = True
self._autoscale_update = True
def draw(self): def draw(self):
# Draw function code... # Draw function code...
def update(self, ind = None): def update(self):
# Update function code... # Update function code...
def clear(self):
# Clear plot values...
# ...
#+end_src #+end_src
** Solver ** Solver
@@ -745,7 +774,7 @@ execute a code on distant computer, for example, over ssh.
style=dashed; style=dashed;
subgraph cluster021 { subgraph cluster021 {
label="Solver Classes"; label="Solver Classes";
classSolverM7[label="Mage7", fillcolor=6]; //classSolverM7[label="Mage7", fillcolor=6];
classSolverM8[label="Mage8", fillcolor=6]; classSolverM8[label="Mage8", fillcolor=6];
classSolverR[label="RubarBE", fillcolor=6]; classSolverR[label="RubarBE", fillcolor=6];
} }
@@ -828,7 +857,7 @@ solver and get results:
- (2.1) The solver read the input file(s) - (2.1) The solver read the input file(s)
- (2.2) The solver compute results and write it to solver output - (2.2) The solver compute results and write it to solver output
file(s) file(s)
- (3) Pamhyr2 create a Reuslts - (3) Pamhyr2 create a =Results= object
- (3.1) The Pamhyr2 solver class read solver output file(s) and - (3.1) The Pamhyr2 solver class read solver output file(s) and
complete Results with readed data complete Results with readed data
@@ -958,8 +987,16 @@ different methods:
** Unit tests ** Unit tests
The unit tests is actually not implemented in Pamhyr2, it is a *work A very small part of Pamhyr2 has unit test. This part is limited to the Model.
in progress*.
#+begin_src shell
python3 -m venv test
. test test/bin/activate
pip3 install -U -r ./full-requirements.txt
cd src/
python3 -Walways -m unittest discovert -v -t .
#+end_src
** The debug mode ** The debug mode
@@ -970,14 +1007,14 @@ line:
./Pamhyr2 debug ./Pamhyr2 debug
#+end_src #+end_src
This mode add some log and add two action in main window menu: This mode add some log and add two action in main window menu: "About
"About/Debug" open a window with Python Repl in current Python > Debug" open a window with Python Repl in current Python
environement, and "About/Debug SQLite" who open the application environement, and "About > Debug SQLite" who open the application
SQLiteBrowser (if installed) on current Study to explore the study SQLiteBrowser (if installed) on current Study to explore the study
data base file. data base file.
#+NAME: debug-repl #+NAME: debug-repl
#+ATTR_LATEX: :width 12cm #+ATTR_LATEX: :width 14cm
#+CAPTION: Pamhyr2 debug Python REPL #+CAPTION: Pamhyr2 debug Python REPL
[[./images/python-debug-repl.png]] [[./images/python-debug-repl.png]]
Binary file not shown.

Before

Width:  |  Height:  |  Size: 74 KiB

After

Width:  |  Height:  |  Size: 51 KiB

Binary file not shown.

After

Width:  |  Height:  |  Size: 224 KiB

+3 -2
View File
@@ -11,7 +11,8 @@
\usepackage[absolute]{textpos} \usepackage[absolute]{textpos}
\usepackage{titlesec} \usepackage{titlesec}
\usepackage{amssymb} \usepackage{amssymb}
\usepackage{tcolorbox} \usepackage[minted,xparse,breakable]{tcolorbox}
\usepackage{pmboxdraw}
\usepackage{enumitem} \usepackage{enumitem}
\setlist[enumerate]{itemsep=-0.5em} \setlist[enumerate]{itemsep=-0.5em}
@@ -140,7 +141,7 @@
\setminted[llvm]{breaklines,fontsize=\footnotesize,encoding=utf8} \setminted[llvm]{breaklines,fontsize=\footnotesize,encoding=utf8}
%% Add box arrond of minted %% Add box arrond of minted
\BeforeBeginEnvironment{minted}{\begin{tcolorbox}[boxsep=0pt, left=0.1cm, right=0.1cm, arc=0pt, boxrule=0.5pt, colback=white]}% \BeforeBeginEnvironment{minted}{\begin{tcolorbox}[breakable, boxsep=0pt, left=0.1cm, right=0.1cm, arc=0pt, boxrule=0.5pt, colback=white]}%
\AfterEndEnvironment{minted}{\end{tcolorbox}}% \AfterEndEnvironment{minted}{\end{tcolorbox}}%
\makeatletter \makeatletter
+2
View File
@@ -10,3 +10,5 @@ colorama>=0.4.3
pyinstaller>=5.11.0 pyinstaller>=5.11.0
shapely>=2.0.1 shapely>=2.0.1
lxml>=4.9.3 lxml>=4.9.3
platformdirs>=4.2.0
pyshp>=2.3.1
+3
View File
@@ -0,0 +1,3 @@
@echo off
start cmd /c pamhyr.exe debug
+3
View File
@@ -0,0 +1,3 @@
#! /bin/sh
./pamhyr debug
+19 -18
View File
@@ -16,34 +16,35 @@ rem along with this program. If not, see <https://www.gnu.org/licenses/>.
@ECHO ON @ECHO ON
mkdir pamhyr\pamhyr\_internal mkdir pamhyr\_internal
xcopy /s /e /y dist\pamhyr pamhyr\pamhyr xcopy /s /e /y dist\pamhyr pamhyr
rem Copy data rem Copy data
mkdir pamhyr\pamhyr\_internal\lang mkdir pamhyr\_internal\lang
mkdir pamhyr\pamhyr\_internal\View\ui\ressources mkdir pamhyr\_internal\View\ui\ressources
mkdir pamhyr\pamhyr\_internal\View\ui\Widgets mkdir pamhyr\_internal\View\ui\Widgets
rem UI rem UI
copy /y ..\src\View\ui\ressources\ pamhyr\pamhyr\_internal\View\ui\ressources copy /y ..\src\View\ui\ressources\ pamhyr\_internal\View\ui\ressources
copy /y ..\src\View\ui\Widgets\*.ui pamhyr\pamhyr\_internal\View\ui\Widgets copy /y ..\src\View\ui\Widgets\*.ui pamhyr\_internal\View\ui\Widgets
copy /y ..\src\View\ui\*.ui pamhyr\pamhyr\_internal\View\ui\ copy /y ..\src\View\ui\*.ui pamhyr\_internal\View\ui\
rem Lang rem Lang
copy /y ..\src\lang\*.qm pamhyr\pamhyr\_internal\lang\ copy /y ..\src\lang\*.qm pamhyr\_internal\lang\
rem Information rem Information
copy /y ..\VERSION pamhyr\pamhyr\_internal copy /y ..\VERSION pamhyr\_internal
copy /y ..\AUTHORS pamhyr\pamhyr\_internal copy /y ..\AUTHORS pamhyr\_internal
copy /y ..\LICENSE pamhyr\pamhyr\_internal copy /y ..\LICENSE pamhyr\_internal
copy /y ..\src\motd.txt pamhyr\pamhyr\_internal copy /y ..\src\motd.txt pamhyr\_internal
rem MAGE rem MAGE
mkdir pamhyr\mage mkdir pamhyr\mage8
copy /y ..\mage-windows\mage.exe pamhyr\mage\ copy /y ..\mage8-windows\mage.exe pamhyr\mage8\
copy /y ..\mage-windows\mage_extraire.exe pamhyr\mage\ copy /y ..\mage8-windows\mage_as7.exe pamhyr\mage8\
copy /y ..\mage-windows\mailleurTT.exe pamhyr\mage\ copy /y ..\mage8-windows\mage_extraire.exe pamhyr\mage8\
copy /y ..\mage-windows\libbief.dll pamhyr\mage\ copy /y ..\mage8-windows\mailleurTT.exe pamhyr\mage8\
copy /y ..\mage8-windows\libbief.dll pamhyr\mage8\
rem Copy tests_cases rem Copy tests_cases
mkdir pamhyr\tests_cases mkdir pamhyr\tests_cases
+4 -3
View File
@@ -3,9 +3,10 @@
!define LIC_NAME "LICENSE" !define LIC_NAME "LICENSE"
!define APP_NAME "PAMHYR" !define APP_NAME "PAMHYR"
Icon "pamhyr\_internal\View\ui\ressources\icon.ico"
Name "Pamhyr2" Name "Pamhyr2"
OutFile "pamhyr-windows.exe" OutFile "pamhyr-windows.exe"
LicenseData "pamhyr\pamhyr\_internal\LICENSE" LicenseData "pamhyr\_internal\LICENSE"
LicenseText "I Agree" LicenseText "I Agree"
RequestExecutionLevel admin RequestExecutionLevel admin
@@ -41,12 +42,12 @@ SectionEnd
Section "Start Menu Shortcuts" Section "Start Menu Shortcuts"
CreateDirectory "$SMPROGRAMS\Pamhyr2" CreateDirectory "$SMPROGRAMS\Pamhyr2"
CreateShortcut "$SMPROGRAMS\Pamhyr2\Uninstall.lnk" "$INSTDIR\uninstall.exe" CreateShortcut "$SMPROGRAMS\Pamhyr2\Uninstall.lnk" "$INSTDIR\uninstall.exe"
CreateShortcut "$SMPROGRAMS\Pamhyr2\Pamhyr2.lnk" "$INSTDIR\pamhyr\pamhyr.exe" CreateShortcut "$SMPROGRAMS\Pamhyr2\Pamhyr2.lnk" "$INSTDIR\pamhyr.exe"
SectionEnd SectionEnd
Section "Desktop Shortcuts" Section "Desktop Shortcuts"
SetShellVarContext all SetShellVarContext all
CreateShortcut "$desktop\Pamhyr2.lnk" "$instdir\pamhyr\pamhyr.exe" CreateShortcut "$desktop\Pamhyr2.lnk" "$instdir\pamhyr.exe"
SectionEnd SectionEnd
Section "Uninstall" Section "Uninstall"
+2
View File
@@ -10,3 +10,5 @@ colorama>=0.4.3
pyinstaller>=5.11.0 pyinstaller>=5.11.0
shapely>=2.0.1 shapely>=2.0.1
lxml>=4.9.3 lxml>=4.9.3
platformdirs>=4.2.0
pyshp>=2.3.1
+46 -1
View File
@@ -16,23 +16,53 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import logging
from enum import Enum from enum import Enum
from Modules import Modules
from Model.Except import NotImplementedMethodeError from Model.Except import NotImplementedMethodeError
logger = logging.getLogger()
class STATUS(Enum): class STATUS(Enum):
UNKNOWN = -1
OK = 0 OK = 0
WARNING = 1 WARNING = 1
ERROR = 2 ERROR = 2
UNKNOWN = 999
def __eq__(self, y):
return self.value == y.value
def __gt__(self, y):
return self.value > y.value
def __or__(self, y):
v = [self, y]
r = self.OK
if self.UNKNOWN in v:
r = self.UNKNOWN
if self.ERROR in v:
r = self.ERROR
if self.WARNING in v:
r = self.WARNING
logger.debug(f"CHECKER: STATUS: {self} | {y} = {r}")
return r
class AbstractModelChecker(object): class AbstractModelChecker(object):
def __init__(self): def __init__(self):
self._name = "" self._name = ""
self._description = "" self._description = ""
self._solver = "study"
self._modules = Modules.NONE
self.reset()
def reset(self):
self._status = STATUS.UNKNOWN self._status = STATUS.UNKNOWN
self._summary = "Unknown" self._summary = "Unknown"
@@ -62,6 +92,21 @@ class AbstractModelChecker(object):
def is_error(self): def is_error(self):
return self._status == STATUS.ERROR return self._status == STATUS.ERROR
# Basic check
def basic_check(self, study):
if study is None:
self._status = STATUS.ERROR
self._summary = "invalid_study"
return False
river = study.river
if river is None:
self._status = STATUS.ERROR
self._summary = "no_river_found"
return False
return True
# Abstract function # Abstract function
def _run(self, study): def _run(self, study):
+13 -6
View File
@@ -23,6 +23,8 @@ from tools import flatten, timer
from PyQt5.QtCore import QCoreApplication from PyQt5.QtCore import QCoreApplication
from Modules import Modules
from Checker.Checker import AbstractModelChecker, STATUS from Checker.Checker import AbstractModelChecker, STATUS
from Checker.Study import StudyGeometryChecker from Checker.Study import StudyGeometryChecker
@@ -30,7 +32,7 @@ _translate = QCoreApplication.translate
class MageNetworkGraphChecker(AbstractModelChecker): class MageNetworkGraphChecker(AbstractModelChecker):
def __init__(self, connectivity=True): def __init__(self, connectivity=True, version="mage8"):
super(MageNetworkGraphChecker, self).__init__() super(MageNetworkGraphChecker, self).__init__()
self._mode_conn = connectivity self._mode_conn = connectivity
@@ -44,6 +46,8 @@ class MageNetworkGraphChecker(AbstractModelChecker):
"Checker", f"Mage network graph {mode} checker") "Checker", f"Mage network graph {mode} checker")
self._description = _translate( self._description = _translate(
"Checker", "Check if the network graph is valid") "Checker", "Check if the network graph is valid")
self._solver = version
self._modules = Modules.NETWORK
@timer @timer
def _connectivity(self, summary, status, graph): def _connectivity(self, summary, status, graph):
@@ -199,15 +203,18 @@ class MageNetworkGraphChecker(AbstractModelChecker):
class MageGeometryGuideLineChecker(StudyGeometryChecker): class MageGeometryGuideLineChecker(StudyGeometryChecker):
def __init__(self): def __init__(self, version="mage8"):
super(MageGeometryGuideLineChecker, self).__init__() super(MageGeometryGuideLineChecker, self).__init__()
self._name = _translate("Checker", "Mage geometry guide line checker") self._name = _translate("Checker", "Mage geometry guideline checker")
self._description = _translate( self._description = _translate(
"Checker", "Checker",
"Check if exists geometry guide line is correctly defined \ "Check if exists geometry guidelines are correctly defined \
for each reaches of the study" for each reach"
) )
self._solver = version
self._modules = Modules.GEOMETRY
self._reachs = [] self._reachs = []
def run(self, study): def run(self, study):
@@ -234,7 +241,7 @@ for each reaches of the study"
for profile in profiles: for profile in profiles:
if not profile.has_standard_named_points(): if not profile.has_standard_named_points():
self._status = STATUS.WARNING self._status = STATUS.WARNING
self._summary = "no_standard_guileline_defined" self._summary = "no_standard_guideline_defined"
return False return False
for gl in gls[1:]: for gl in gls[1:]:
+133 -17
View File
@@ -17,33 +17,34 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import time import time
import logging
from functools import reduce
from PyQt5.QtCore import QCoreApplication from PyQt5.QtCore import QCoreApplication
from Modules import Modules
from Checker.Checker import AbstractModelChecker, STATUS from Checker.Checker import AbstractModelChecker, STATUS
_translate = QCoreApplication.translate _translate = QCoreApplication.translate
logger = logging.getLogger()
class StudyNetworkReachChecker(AbstractModelChecker): class StudyNetworkReachChecker(AbstractModelChecker):
def __init__(self): def __init__(self):
super(StudyNetworkReachChecker, self).__init__() super(StudyNetworkReachChecker, self).__init__()
self._name = _translate("Checker", "Study network reach checker") self._name = _translate("Checker", "Study reach network checker")
self._description = _translate( self._description = _translate(
"Checker", "Check if exists at least one reach for study") "Checker", "Check if exists at least one reach exists"
)
self._modules = Modules.NETWORK
def run(self, study): def run(self, study):
if study is None: if not self.basic_check(study):
self._status = STATUS.ERROR
self._summary = "invalid_study"
return False return False
river = study.river river = study.river
if river is None:
self._status = STATUS.ERROR
self._summary = "no_river_found"
return False
edges = list(filter(lambda e: e.is_enable(), river.edges())) edges = list(filter(lambda e: e.is_enable(), river.edges()))
if len(edges) == 0: if len(edges) == 0:
@@ -62,7 +63,10 @@ class StudyGeometryChecker(AbstractModelChecker):
self._name = _translate("Checker", "Study geometry checker") self._name = _translate("Checker", "Study geometry checker")
self._description = _translate( self._description = _translate(
"Checker", "Check if exists geometry for each reach of study") "Checker", "Check if the geometry of each reach exists"
)
self._modules = Modules.GEOMETRY
self._reachs = [] self._reachs = []
def run(self, study): def run(self, study):
@@ -71,16 +75,10 @@ class StudyGeometryChecker(AbstractModelChecker):
summary = "ok" summary = "ok"
status = STATUS.OK status = STATUS.OK
if study is None: if not self.basic_check(study):
self._status = STATUS.ERROR
self._summary = "invalid_study"
return False return False
river = study.river river = study.river
if river is None:
self._status = STATUS.ERROR
self._summary = "no_river_found"
return False
edges = river.enable_edges() edges = river.enable_edges()
if len(edges) == 0: if len(edges) == 0:
@@ -100,6 +98,124 @@ class StudyGeometryChecker(AbstractModelChecker):
return ok return ok
class StudyInitialConditionsChecker(AbstractModelChecker):
def __init__(self):
super(StudyInitialConditionsChecker, self).__init__()
self._name = _translate("Checker", "Study initial conditions checker")
self._description = _translate(
"Checker", "Check initial conditions for each node of study"
)
self._modules = Modules.INITIAL_CONDITION
def run(self, study):
ok = True
nerror = 0
self._summary = "ok"
self._status = STATUS.OK
if not self.basic_check(study):
return False
for reach in study.river.enable_edges():
ok &= self.check_reach(study, reach.reach)
return ok
def check_reach(self, study, reach):
ok = True
river = study.river
if reach not in river.initial_conditions:
self._summary = "missing_initial_condition_defined"
self._status = STATUS.WARNING
return ok
ic = river.initial_conditions[reach]
len_ic = len(ic)
len_reach = len(reach)
if len_ic < len_reach:
self._summary = "initial_condition_missing_profile"
self._status = STATUS.WARNING
if len_ic > len_reach:
self._summary = "more_initial_condition_than_profile"
self._status = STATUS.ERROR
ok = False
return ok
class StudyBoundaryConditionChecker(AbstractModelChecker):
def __init__(self):
super(StudyBoundaryConditionChecker, self).__init__()
self._name = _translate("Checker", "Study boundary conditions checker")
self._description = _translate(
"Checker", "Check boundary conditions for each node of study"
)
self._modules = Modules.BOUNDARY_CONDITION
def run(self, study):
ok = True
nerror = 0
self._summary = "ok"
self._status = STATUS.OK
if not self.basic_check(study):
return False
ok = self.check_liquid(study)
return ok
def check_liquid(self, study):
river = study.river
bcs = river.boundary_condition.get_tab('liquid')
if len(bcs) == 0:
self._status = STATUS.ERROR
self._summary = "no_boundary_condition_defined"
return False
upstream, downstream = reduce(
lambda acc, n: (
acc[0] + [n] if river.is_upstream_node(n) else acc[0],
acc[1] + [n] if river.is_downstream_node(n) else acc[1],
),
filter(
lambda n: river.is_enable_node(n),
river.nodes()
),
([], [])
)
bcs_nodes = set(map(lambda bc: bc.node, bcs))
upstream_ok = self.check_liquid_all_node_has_bc(
bcs_nodes, upstream
)
downstream_ok = self.check_liquid_all_node_has_bc(
bcs_nodes, downstream
)
ok = upstream_ok and downstream_ok
if not ok:
self._status = STATUS.ERROR
self._summary = "no_boundary_condition_at_boundary_node"
return ok
def check_liquid_all_node_has_bc(self, bcs_nodes, nodes):
return reduce(
lambda acc, n: (acc and (n in bcs_nodes)),
nodes,
True
)
class DummyOK(AbstractModelChecker): class DummyOK(AbstractModelChecker):
def __init__(self): def __init__(self):
super(DummyOK, self).__init__() super(DummyOK, self).__init__()
+122 -22
View File
@@ -30,7 +30,7 @@ from ctypes import (
) )
from PyQt5.QtCore import QProcess from PyQt5.QtCore import QProcess
from tools import logger_color_red, logger_color_reset from tools import logger_color_red, logger_color_reset, logger_exception
from Meshing.AMeshingTool import AMeshingTool from Meshing.AMeshingTool import AMeshingTool
logger = logging.getLogger() logger = logging.getLogger()
@@ -278,7 +278,7 @@ class MeshingWithMageMailleurTT(AMeshingTool):
return os.path.abspath( return os.path.abspath(
os.path.join( os.path.join(
os.path.dirname(__file__), os.path.dirname(__file__),
"..", "..", "..", "mage", f"mailleurTT{ext}" "..", "..", "mage8", f"mailleurTT{ext}"
) )
) )
@@ -293,10 +293,12 @@ class MeshingWithMageMailleurTT(AMeshingTool):
def meshing(self, reach, def meshing(self, reach,
step: float = 50, step: float = 50,
limites=[-1, -1], limites=[-1, -1],
origin=0,
directrices=['un', 'np'], directrices=['un', 'np'],
lplan: bool = False, lplan: bool = False,
lm: int = 3, lm: int = 3,
linear: bool = False): linear: bool = False,
origin_value=0.0):
if reach is None or len(reach.profiles) == 0: if reach is None or len(reach.profiles) == 0:
return reach return reach
@@ -308,37 +310,133 @@ class MeshingWithMageMailleurTT(AMeshingTool):
proc = QProcess() proc = QProcess()
proc.setWorkingDirectory(tmp) proc.setWorkingDirectory(tmp)
# Mage section index start at 1
origin += 1
limites[0] += 1
limites[1] += 1
lplan = 1 if lplan else 0 lplan = 1 if lplan else 0
linear = 1 if linear else 0 linear = 1 if linear else 0
logger.info( logger.info(
f"! mailleurTT {st_file} {m_file} " + f"! {self._exe_path()} " +
f"{st_file} {m_file} " +
f"mesh " +
f"{str(step)} " + f"{str(step)} " +
f"{limites[0]} {limites[1]} " + f"{limites[0]} {limites[1]} " +
f"{directrices[0]} {directrices[1]} " + f"{directrices[0]} {directrices[1]} " +
f"{lplan} {lm} {linear}" f"{lplan} {lm} {linear} " +
f"{origin} "
) )
proc.start( proc.start(
self._exe_path(), self._exe_path(),
list(map( list(
str, map(
[ str,
st_file, m_file, step, [
limites[0], limites[1], st_file, m_file,
directrices[0], directrices[1], "mesh", step,
lplan, lm, linear limites[0], limites[1],
] directrices[0], directrices[1],
)) lplan, lm, linear, origin, origin_value
]
)
)
) )
proc.waitForFinished() proc.waitForFinished()
errors = proc.readAllStandardError() if proc.exitCode() != 0:
if len(errors) != 0: logger.error(
f"{logger_color_red()}" +
f"MailleurTT error: {proc.exitCode()}" +
f"{logger_color_reset()}"
)
outputs = proc.readAllStandardOutput()
logger.debug(outputs)
errors = proc.readAllStandardError()
logger.error( logger.error(
f"{logger_color_red()}{errors}{logger_color_reset()}" f"{logger_color_red()}{errors}{logger_color_reset()}"
) )
else: return reach
self.import_m_file(reach, m_file)
self.import_m_file(reach, m_file)
return reach
def update_rk(self, reach,
step: float = 50,
limites=[-1, -1],
origin=0,
directrices=['un', 'np'],
lplan: bool = False,
lm: int = 3,
linear: bool = False,
origin_value=0.0,
orientation=0):
if reach is None or len(reach.profiles) == 0:
return reach
with tempfile.TemporaryDirectory() as tmp:
logger.debug(f"temp file: {tmp}")
st_file = self.export_reach_to_st(reach, tmp)
m_file = st_file.rsplit(".ST", 1)[0] + ".M"
os.sync()
proc = QProcess()
proc.setWorkingDirectory(tmp)
# Mage section indices start at 1
origin += 1
limites[0] += 1
limites[1] += 1
lplan = 1 if lplan else 0
linear = 1 if linear else 0
logger.info(
f"! {self._exe_path()} " +
f"{st_file} {m_file} " +
f"update_rk " +
f"{str(step)} " +
f"{limites[0]} {limites[1]} " +
f"{directrices[0]} {directrices[1]} " +
f"{orientation} {lm} {linear} " +
f"{origin} "
)
proc.start(
self._exe_path(),
list(
map(
str,
[
st_file, m_file,
"update_rk", step,
limites[0], limites[1],
directrices[0], directrices[1],
orientation, lm, linear, origin, origin_value
]
)
)
)
proc.waitForFinished()
if proc.exitCode() != 0:
logger.error(
f"{logger_color_red()}" +
f"MailleurTT error: {proc.exitCode()}" +
f"{logger_color_reset()}"
)
outputs = proc.readAllStandardOutput()
logger.debug(outputs)
errors = proc.readAllStandardError()
logger.error(
f"{logger_color_red()}{errors}{logger_color_reset()}"
)
return reach
self.import_m_file(reach, m_file)
return reach return reach
def export_reach_to_st(self, reach, tmp): def export_reach_to_st(self, reach, tmp):
@@ -350,7 +448,9 @@ class MeshingWithMageMailleurTT(AMeshingTool):
return tmp_st return tmp_st
def import_m_file(self, reach, m): def import_m_file(self, reach, m):
reach.purge() if os.path.exists(m):
reach.purge()
logger.debug(f"meshing: Import geometry from {m}") logger.debug(f"meshing: Import geometry from {m}")
reach.import_geometry(m) reach.import_geometry(m)
else:
logger.error(f"meshing: Meshed geometry file {m} do not exists")
+7 -1
View File
@@ -25,9 +25,10 @@ from Model.Saved import SavedStatus
from Model.Study import Study from Model.Study import Study
from Model.River import River from Model.River import River
from Meshing.Mage import MeshingWithMage from Meshing.Mage import MeshingWithMage, MeshingWithMageMailleurTT
mage_mesher = os.path.exists(MeshingWithMage._lib_path()) mage_mesher = os.path.exists(MeshingWithMage._lib_path())
mage_mesher_TT = os.path.exists(MeshingWithMageMailleurTT._exe_path())
class MeshingTestCase(unittest.TestCase): class MeshingTestCase(unittest.TestCase):
@@ -35,3 +36,8 @@ class MeshingTestCase(unittest.TestCase):
def test_create_meshingtool_mage(self): def test_create_meshingtool_mage(self):
mesher = MeshingWithMage() mesher = MeshingWithMage()
self.assertNotEqual(mesher, None) self.assertNotEqual(mesher, None)
@unittest.skipIf(not mage_mesher_TT, "mailleurTT is not available")
def test_create_meshingtool_mageTT(self):
mesher = MeshingWithMageMailleurTT()
self.assertNotEqual(mesher, None)
+178
View File
@@ -0,0 +1,178 @@
# AddFile.py -- Pamhyr
# Copyright (C) 2024 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 functools import reduce
from tools import trace, timer
from Model.Tools.PamhyrDB import SQLSubModel
from Model.Except import NotImplementedMethodeError
class AddFile(SQLSubModel):
_sub_classes = []
_id_cnt = 0
def __init__(self, id: int = -1, enabled=True,
name="", path="", text="",
status=None):
super(AddFile, self).__init__()
if id == -1:
self.id = AddFile._id_cnt
else:
self.id = id
self._status = status
self._enabled = enabled
self._name = f"File {self.id}" if name == "" else name
self._path = path
self._text = text
AddFile._id_cnt = max(id, AddFile._id_cnt+1)
def __getitem__(self, key):
value = None
if key == "enabled":
value = self._enabled
elif key == "name":
value = self._name
elif key == "path":
value = self._path
elif key == "text":
value = self._text
return value
def __setitem__(self, key, value):
if key == "enabled":
self._enabled = value
elif key == "name":
self._name = value
elif key == "path":
self._path = value
elif key == "text":
self._text = value
self._status.modified()
@property
def enabled(self):
return self._enabled
@enabled.setter
def enabled(self, enabled):
self._enabled = enabled
self._status.modified()
def is_enabled(self):
return self._enabled
@property
def name(self):
return self._name
@name.setter
def name(self, name):
self._name = name
self._status.modified()
@property
def path(self):
return self._path
@path.setter
def path(self, path):
self._path = path
self._status.modified()
@property
def text(self):
return self._text
@text.setter
def text(self, text):
self._text = text
self._status.modified()
@classmethod
def _db_create(cls, execute):
execute("""
CREATE TABLE additional_files(
id INTEGER NOT NULL PRIMARY KEY,
enabled BOOLEAN NOT NULL,
name TEXT NOT NULL,
path TEXT NOT NULL,
text TEXT NOT NULL
)
""")
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) < 8:
cls._db_create(execute)
return True
@classmethod
def _db_load(cls, execute, data=None):
new = []
table = execute(
"SELECT id, enabled, name, path, text " +
"FROM additional_files"
)
for row in table:
it = iter(row)
id = next(it)
enabled = (next(it) == 1)
name = next(it)
path = next(it)
text = next(it)
f = cls(
id=id, enabled=enabled, name=name, path=path, text=text,
status=data['status']
)
new.append(f)
return new
def _db_save(self, execute, data=None):
sql = (
"INSERT INTO " +
"additional_files(id, enabled, name, path, text) " +
"VALUES (" +
f"{self.id}, {self._enabled}, " +
f"'{self._db_format(self._name)}', " +
f"'{self._db_format(self._path)}', " +
f"'{self._db_format(self._text)}'" +
")"
)
execute(sql)
return True
+56
View File
@@ -0,0 +1,56 @@
# AddFileList.py -- Pamhyr
# Copyright (C) 2024 INRAE
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
# -*- coding: utf-8 -*-
from tools import trace, timer
from Model.Except import NotImplementedMethodeError
from Model.Tools.PamhyrList import PamhyrModelList
from Model.AdditionalFile.AddFile import AddFile
class AddFileList(PamhyrModelList):
_sub_classes = [AddFile]
@classmethod
def _db_load(cls, execute, data=None):
new = cls(status=data["status"])
new._lst = AddFile._db_load(execute, data)
return new
def _db_save(self, execute, data=None):
ok = True
# Delete previous data
execute("DELETE FROM additional_files")
for af in self._lst:
ok &= af._db_save(execute, data)
return ok
@property
def files(self):
return self.lst
def new(self, index):
n = AddFile(status=self._status)
self.insert(index, n)
self._status.modified()
return n
@@ -18,7 +18,12 @@
import logging import logging
from tools import trace, timer, old_pamhyr_date_to_timestamp from tools import (
trace, timer,
old_pamhyr_date_to_timestamp,
date_iso_to_timestamp,
date_dmy_to_timestamp,
)
from Model.Tools.PamhyrDB import SQLSubModel from Model.Tools.PamhyrDB import SQLSubModel
from Model.Except import NotImplementedMethodeError from Model.Except import NotImplementedMethodeError
@@ -189,6 +194,10 @@ class BoundaryCondition(SQLSubModel):
@classmethod @classmethod
def time_convert(cls, data): def time_convert(cls, data):
if type(data) is str: if type(data) is str:
if data.count("-") == 2:
return date_iso_to_timestamp(data)
if data.count("/") == 2:
return date_dmy_to_timestamp(data)
if data.count(":") == 3: if data.count(":") == 3:
return old_pamhyr_date_to_timestamp(data) return old_pamhyr_date_to_timestamp(data)
if data.count(":") == 2: if data.count(":") == 2:
@@ -306,6 +315,9 @@ class BoundaryCondition(SQLSubModel):
self._data.sort(reverse=_reverse, key=key) self._data.sort(reverse=_reverse, key=key)
self._status.modified() self._status.modified()
def index(self, bc):
self._data.index(bc)
def get_i(self, index): def get_i(self, index):
return self.data[index] return self.data[index]
+2 -1
View File
@@ -21,7 +21,7 @@ from PyQt5.QtCore import (
) )
from PyQt5.QtWidgets import ( from PyQt5.QtWidgets import (
QApplication, QMessageBox, QApplication, QMessageBox, QLabel,
) )
_translate = QCoreApplication.translate _translate = QCoreApplication.translate
@@ -34,6 +34,7 @@ _translate = QCoreApplication.translate
def exception_message_box(exception): def exception_message_box(exception):
msg = QMessageBox() msg = QMessageBox()
msg.setIcon(QMessageBox.Critical) msg.setIcon(QMessageBox.Critical)
msg.findChild(QLabel, "qt_msgbox_label").setFixedWidth(384)
msg.setText("Exception :") msg.setText("Exception :")
msg.setInformativeText(f"{exception}") msg.setInformativeText(f"{exception}")
+45 -31
View File
@@ -43,8 +43,8 @@ class Friction(SQLSubModel):
self._name = name self._name = name
self._edge = None self._edge = None
self._begin_kp = 0.0 self._begin_rk = 0.0
self._end_kp = 0.0 self._end_rk = 0.0
self._begin_strickler = None self._begin_strickler = None
self._end_strickler = None self._end_strickler = None
@@ -54,8 +54,8 @@ class Friction(SQLSubModel):
CREATE TABLE friction( CREATE TABLE friction(
id INTEGER NOT NULL PRIMARY KEY, id INTEGER NOT NULL PRIMARY KEY,
ind INTEGER NOT NULL, ind INTEGER NOT NULL,
begin_kp REAL NOT NULL, begin_rk REAL NOT NULL,
end_kp REAL NOT NULL, end_rk REAL NOT NULL,
reach INTEGER NOT NULL, reach INTEGER NOT NULL,
begin_strickler INTEGER NOT NULL, begin_strickler INTEGER NOT NULL,
end_strickler INTEGER NOT NULL, end_strickler INTEGER NOT NULL,
@@ -69,6 +69,12 @@ class Friction(SQLSubModel):
@classmethod @classmethod
def _db_update(cls, execute, version): def _db_update(cls, execute, version):
major, minor, release = version.strip().split(".")
if major == minor == "0":
if int(release) < 11:
execute("ALTER TABLE friction RENAME COLUMN begin_kp TO begin_rk")
execute("ALTER TABLE friction RENAME COLUMN end_kp TO end_rk")
return True return True
@classmethod @classmethod
@@ -80,7 +86,7 @@ class Friction(SQLSubModel):
stricklers = data["stricklers"].stricklers stricklers = data["stricklers"].stricklers
table = execute( table = execute(
"SELECT id, ind, begin_kp, end_kp, " + "SELECT id, ind, begin_rk, end_rk, " +
"begin_strickler, end_strickler " + "begin_strickler, end_strickler " +
f"FROM friction WHERE reach = {reach.id}" f"FROM friction WHERE reach = {reach.id}"
) )
@@ -102,8 +108,8 @@ class Friction(SQLSubModel):
# Create friction # Create friction
sec = cls(status=status) sec = cls(status=status)
sec.edge = reach sec.edge = reach
sec.begin_kp = row[2] sec.begin_rk = row[2]
sec.end_kp = row[3] sec.end_rk = row[3]
sec.begin_strickler = bs sec.begin_strickler = bs
sec.end_strickler = es sec.end_strickler = es
@@ -124,11 +130,11 @@ class Friction(SQLSubModel):
execute( execute(
"INSERT INTO " + "INSERT INTO " +
"friction(id, ind, begin_kp, end_kp, " + "friction(id, ind, begin_rk, end_rk, " +
"reach, begin_strickler, end_strickler) " + "reach, begin_strickler, end_strickler) " +
"VALUES (" + "VALUES (" +
f"{self.id}, {ind}, " + f"{self.id}, {ind}, " +
f"{self._begin_kp}, {self._end_kp}, " + f"{self._begin_rk}, {self._end_rk}, " +
f"{self._edge.id}, " + f"{self._edge.id}, " +
f"{b_s_id}, {e_s_id}" + f"{b_s_id}, {e_s_id}" +
")" ")"
@@ -157,10 +163,10 @@ class Friction(SQLSubModel):
def edge(self, edge): def edge(self, edge):
self._edge = edge self._edge = edge
if (edge is not None and if (edge is not None and
self._begin_kp == 0.0 and self._begin_rk == 0.0 and
self._end_kp == 0.0): self._end_rk == 0.0):
self._begin_kp = self._edge.reach.get_kp_min() self._begin_rk = self._edge.reach.get_rk_min()
self._end_kp = self._edge.reach.get_kp_max() self._end_rk = self._edge.reach.get_rk_max()
self._status.modified() self._status.modified()
def has_edge(self): def has_edge(self):
@@ -176,39 +182,47 @@ class Friction(SQLSubModel):
return self.has_edge() and self.has_coefficient() return self.has_edge() and self.has_coefficient()
@property @property
def begin_kp(self): def begin_rk(self):
return self._begin_kp return self._begin_rk
@begin_kp.setter @begin_rk.setter
def begin_kp(self, begin_kp): def begin_rk(self, begin_rk):
if self._edge is None: if self._edge is None:
self._begin_kp = begin_kp self._begin_rk = begin_rk
else: else:
_min = self._edge.reach.get_kp_min() _min = self._edge.reach.get_rk_min()
_max = self._edge.reach.get_kp_max() _max = self._edge.reach.get_rk_max()
if _min <= begin_kp <= _max: if _min <= begin_rk <= _max:
self._begin_kp = begin_kp self._begin_rk = begin_rk
self._status.modified() self._status.modified()
@property @property
def end_kp(self): def end_rk(self):
return self._end_kp return self._end_rk
@end_kp.setter @end_rk.setter
def end_kp(self, end_kp): def end_rk(self, end_rk):
if self._edge is None: if self._edge is None:
self._end_kp = end_kp self._end_rk = end_rk
else: else:
_min = self._edge.reach.get_kp_min() _min = self._edge.reach.get_rk_min()
_max = self._edge.reach.get_kp_max() _max = self._edge.reach.get_rk_max()
if _min <= end_kp <= _max: if _min <= end_rk <= _max:
self._end_kp = end_kp self._end_rk = end_rk
self._status.modified() self._status.modified()
def __contains__(self, rk):
return self.contains_rk(rk)
def contains_rk(self, rk):
return (
self._begin_rk <= rk <= self._end_rk
)
@property @property
def begin_strickler(self): def begin_strickler(self):
return self._begin_strickler return self._begin_strickler
+1 -1
View File
@@ -44,7 +44,7 @@ class FrictionList(PamhyrModelList):
logger.info(f"Update friction TABLE from {version}") logger.info(f"Update friction TABLE from {version}")
cls._db_update_0_0_1(execute, version) cls._db_update_0_0_1(execute, version)
return True return cls._update_submodel(execute, version)
@classmethod @classmethod
def _db_load(cls, execute, data=None): def _db_load(cls, execute, data=None):
+26 -2
View File
@@ -47,7 +47,7 @@ class PointXYZ(Point, SQLSubModel):
z INTEGER NOT NULL, z INTEGER NOT NULL,
profile INTEGER NOT NULL, profile INTEGER NOT NULL,
sl INTEGER, sl INTEGER,
FOREIGN KEY(profile) REFERENCES profileXYZ(id), FOREIGN KEY(profile) REFERENCES geometry_profileXYZ(id),
FOREIGN KEY(sl) REFERENCES sedimentary_layer(id) FOREIGN KEY(sl) REFERENCES sedimentary_layer(id)
) )
""") """)
@@ -192,6 +192,22 @@ class PointXYZ(Point, SQLSubModel):
def dist(self, p2): def dist(self, p2):
return PointXYZ.distance(self, p2) return PointXYZ.distance(self, p2)
def dist_2d(self, p2):
return PointXYZ.distance_2d(self, p2)
@staticmethod
def distance_2d(p1, p2):
"""Euclidean distance between p1 and p2.
Args:
p1: A XYZ Point
p2: A XYZ Point
Returns:
Euclidean 2D distance between the two points
"""
return dist((p1.x, p1.y), (p2.x, p2.y))
@staticmethod @staticmethod
def distance(p1, p2): def distance(p1, p2):
"""Euclidean distance between p1 and p2. """Euclidean distance between p1 and p2.
@@ -201,6 +217,14 @@ class PointXYZ(Point, SQLSubModel):
p2: A XYZ Point p2: A XYZ Point
Returns: Returns:
Euclidean distance between the two points Euclidean 3D distance between the two points
""" """
return dist((p1.x, p1.y, p1.z), (p2.x, p2.y, p2.z)) return dist((p1.x, p1.y, p1.z), (p2.x, p2.y, p2.z))
@staticmethod
def areatriangle3d(p1, p2, p3):
a = PointXYZ.distance(p1, p2)
b = PointXYZ.distance(p2, p3)
c = PointXYZ.distance(p3, p1)
s = (a + b + c) / 2
return (s*(s-a) * (s-b)*(s-c)) ** 0.5
+22 -7
View File
@@ -19,6 +19,7 @@
import logging import logging
from tools import timer from tools import timer
from shapely import geometry
from Model.Geometry.Point import Point from Model.Geometry.Point import Point
from Model.Except import NotImplementedMethodeError from Model.Except import NotImplementedMethodeError
@@ -30,7 +31,7 @@ class Profile(object):
_id_cnt = 0 _id_cnt = 0
def __init__(self, id: int = -1, num: int = 0, def __init__(self, id: int = -1, num: int = 0,
kp: float = 0.0, name: str = "", rk: float = 0.0, name: str = "",
code1: int = 0, code2: int = 0, code1: int = 0, code2: int = 0,
_type: str = "", reach=None, _type: str = "", reach=None,
status=None): status=None):
@@ -48,7 +49,7 @@ class Profile(object):
self._num = int(num) self._num = int(num)
self._code1 = int(code1) self._code1 = int(code1)
self._code2 = int(code2) self._code2 = int(code2)
self._kp = float(kp) self._rk = float(rk)
self._name = str(name) self._name = str(name)
self._reach = reach self._reach = reach
self._sl = None self._sl = None
@@ -135,16 +136,16 @@ class Profile(object):
return len(self.points) return len(self.points)
@property @property
def kp(self): def rk(self):
""" """
Returns: Returns:
Kilometer point. Kilometer point.
""" """
return self._kp return self._rk
@kp.setter @rk.setter
def kp(self, value: float): def rk(self, value: float):
self._kp = float(value) self._rk = float(value)
self._status.modified() self._status.modified()
@property @property
@@ -295,6 +296,11 @@ class Profile(object):
) )
self._status.modified() self._status.modified()
@timer
def reverse(self):
self._points.reverse()
self._status.modified()
# Sediment Layers # Sediment Layers
def get_sl(self): def get_sl(self):
@@ -341,3 +347,12 @@ class Profile(object):
# Abstract method for width approximation # Abstract method for width approximation
def get_water_limits(self, z): def get_water_limits(self, z):
raise NotImplementedMethodeError(self, self.get_water_limits) raise NotImplementedMethodeError(self, self.get_water_limits)
def wet_points(self, z):
raise NotImplementedMethodeError(self, self.wet_point)
def wet_perimeter(self, z):
raise NotImplementedMethodeError(self, self.wet_perimeter)
def wet_area(self, z):
raise NotImplementedMethodeError(self, self.wet_area)
+210 -107
View File
@@ -22,6 +22,7 @@ from typing import List
from functools import reduce from functools import reduce
from tools import timer from tools import timer
from shapely import geometry
from Model.Tools.PamhyrDB import SQLSubModel from Model.Tools.PamhyrDB import SQLSubModel
from Model.Except import ClipboardFormatError from Model.Except import ClipboardFormatError
@@ -40,7 +41,7 @@ class ProfileXYZ(Profile, SQLSubModel):
def __init__(self, def __init__(self,
id: int = -1, id: int = -1,
name: str = "", name: str = "",
kp: float = 0., rk: float = 0.,
reach=None, reach=None,
num=0, num=0,
nb_point: int = 0, nb_point: int = 0,
@@ -52,7 +53,7 @@ class ProfileXYZ(Profile, SQLSubModel):
num: The number of this profile num: The number of this profile
code1: The interpolation code 1 code1: The interpolation code 1
code2: The interpolation code 2 code2: The interpolation code 2
kp: Kilometer point rk: Kilometer point
name: The name of profile name: The name of profile
Returns: Returns:
@@ -62,7 +63,7 @@ class ProfileXYZ(Profile, SQLSubModel):
id=id, id=id,
num=num, num=num,
name=name, name=name,
kp=kp, rk=rk,
code1=code1, code2=code2, code1=code1, code2=code2,
_type="XYZ", _type="XYZ",
reach=reach, reach=reach,
@@ -77,7 +78,7 @@ class ProfileXYZ(Profile, SQLSubModel):
ind INTEGER NOT NULL, ind INTEGER NOT NULL,
name TEXT, name TEXT,
reach INTEGER NOT NULL, reach INTEGER NOT NULL,
kp REAL NOT NULL, rk REAL NOT NULL,
num INTEGER NOT NULL, num INTEGER NOT NULL,
code1 INTEGER NOT NULL, code1 INTEGER NOT NULL,
code2 INTEGER NOT NULL, code2 INTEGER NOT NULL,
@@ -93,7 +94,8 @@ class ProfileXYZ(Profile, SQLSubModel):
def _db_update(cls, execute, version): def _db_update(cls, execute, version):
major, minor, release = version.strip().split(".") major, minor, release = version.strip().split(".")
if major == minor == "0": if major == minor == "0":
if int(release) < 2: rl = int(release)
if rl < 2:
execute( execute(
""" """
ALTER TABLE geometry_profileXYZ ALTER TABLE geometry_profileXYZ
@@ -102,6 +104,14 @@ class ProfileXYZ(Profile, SQLSubModel):
""" """
) )
if rl < 11:
execute(
"""
ALTER TABLE geometry_profileXYZ
RENAME COLUMN kp TO rk
"""
)
return cls._update_submodel(execute, version) return cls._update_submodel(execute, version)
@classmethod @classmethod
@@ -111,7 +121,7 @@ class ProfileXYZ(Profile, SQLSubModel):
reach = data["reach"] reach = data["reach"]
table = execute( table = execute(
"SELECT id, ind, name, kp, num, code1, code2, sl " + "SELECT id, ind, name, rk, num, code1, code2, sl " +
"FROM geometry_profileXYZ " + "FROM geometry_profileXYZ " +
f"WHERE reach = {reach.id}" f"WHERE reach = {reach.id}"
) )
@@ -120,7 +130,7 @@ class ProfileXYZ(Profile, SQLSubModel):
id = row[0] id = row[0]
ind = row[1] ind = row[1]
name = row[2] name = row[2]
kp = row[3] rk = row[3]
num = row[5] num = row[5]
code1 = row[5] code1 = row[5]
code2 = row[6] code2 = row[6]
@@ -128,7 +138,7 @@ class ProfileXYZ(Profile, SQLSubModel):
new = cls( new = cls(
id=id, num=num, id=id, num=num,
name=name, kp=kp, name=name, rk=rk,
code1=code1, code2=code2, code1=code1, code2=code2,
reach=reach, reach=reach,
status=status status=status
@@ -158,10 +168,10 @@ class ProfileXYZ(Profile, SQLSubModel):
sql = ( sql = (
"INSERT OR REPLACE INTO " + "INSERT OR REPLACE INTO " +
"geometry_profileXYZ(id, ind, name, reach, " + "geometry_profileXYZ(id, ind, name, reach, " +
"kp, num, code1, code2, sl) " + "rk, num, code1, code2, sl) " +
"VALUES (" + "VALUES (" +
f"{self.id}, {ind}, '{self._db_format(self._name)}', " + f"{self.id}, {ind}, '{self._db_format(self._name)}', " +
f"{self.reach.id}, {self.kp}, {self.num}, " + f"{self.reach.id}, {self.rk}, {self.num}, " +
f"{self.code1}, {self.code1}, {sl}" + f"{self.code1}, {self.code1}, {sl}" +
")" ")"
) )
@@ -186,19 +196,19 @@ class ProfileXYZ(Profile, SQLSubModel):
try: try:
if len(header) == 0: if len(header) == 0:
name = data[0] name = data[0]
kp = data[1] rk = data[1]
reach = data[2] reach = data[2]
status = data[3] status = data[3]
profile = cls( profile = cls(
id=-1, id=-1,
name=name, name=name,
kp=kp, rk=rk,
reach=reach, reach=reach,
status=status status=status
) )
else: else:
valid_header = {'name', 'reach', 'kp', 'status'} valid_header = {'name', 'reach', 'rk', 'status'}
d = {} d = {}
for i, v in enumerate(data): for i, v in enumerate(data):
h = header[i].strip().lower().split(' ')[0] h = header[i].strip().lower().split(' ')[0]
@@ -322,7 +332,7 @@ class ProfileXYZ(Profile, SQLSubModel):
logger.debug(f"{e}") logger.debug(f"{e}")
raise IndexError( raise IndexError(
f"Invalid point name: {name} " + f"Invalid point name: {name} " +
f"for profile ({self.id}) kp = {self.kp}" f"for profile ({self.id}) rk = {self.rk}"
) )
def has_standard_named_points(self): def has_standard_named_points(self):
@@ -373,99 +383,13 @@ class ProfileXYZ(Profile, SQLSubModel):
""" """
return [x for x in lst if not np.isnan(x)] return [x for x in lst if not np.isnan(x)]
def _first_point_not_nan(self): def speed(self, q, z):
first_point = self.points[0] area = self.wet_area(z)
for point in self.points: if area == 0:
if not point.is_nan(): return 0
first_point = point
break
return first_point return q / area
def _last_point_not_nan(self):
last_point = self.points[-1]
for point in self.points[::-1]:
if not point.is_nan():
last_point = point
break
return last_point
@timer
def get_station(self) -> np.ndarray:
"""Projection of the points of the profile on a plane.
Args:
profile: The profile
Returns:
Projection of the points of the profile on a plane.
"""
if self.nb_points < 3:
return None
else:
first_named_point = None
index_first_named_point = None
last_named_point = None
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):
if point.point_is_named():
index_first_named_point = index
first_named_point = point
break
for point in reversed(self.points):
if point.point_is_named():
last_named_point = point
break
station = []
constant = 0.0
if (first_named_point is not None and
last_named_point is not None):
if (first_named_point != last_named_point and
first_named_point.x != last_named_point.x):
vector = Vector1d(first_named_point, last_named_point)
norm_dir_vec = vector.normalized_direction_vector()
else:
vector = Vector1d(first_point_not_nan, last_point_not_nan)
norm_dir_vec = vector.normalized_direction_vector()
for point in self.points:
xi = point.x - first_named_point.x
yi = point.y - first_named_point.y
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)
norm_dir_vec = vector.normalized_direction_vector()
for point in self.points:
xi = point.x - first_point_not_nan.x
yi = point.y - first_point_not_nan.y
station_i = (norm_dir_vec[0] * xi +
norm_dir_vec[1] * yi)
station.append(station_i)
z_min = self.z_min()
index_profile_z_min = list(
filter(
lambda z: z[1] == z_min,
enumerate(self.z())
)
)[0]
constant = station[index_profile_z_min[0]]
return list(map(lambda s: s - constant, station))
def width_approximation(self): def width_approximation(self):
if self.has_standard_named_points(): if self.has_standard_named_points():
@@ -477,6 +401,42 @@ class ProfileXYZ(Profile, SQLSubModel):
return abs(rg.dist(rd)) return abs(rg.dist(rd))
def wet_perimeter(self, z):
poly = self.wet_polygon(z)
if poly is None:
return 0
return poly.length
def wet_area(self, z):
poly = self.wet_polygon(z)
if poly is None:
return 0
return poly.area
def wet_polygon(self, z):
points = self.wet_points(z)
if len(points) < 3:
return None
z = map(lambda p: p.z, points)
station = self._get_station(points)
poly = geometry.Polygon(list(zip(station, z)))
return poly
def wet_points(self, z):
left, right = self.get_water_limits(z)
points = list(filter(lambda p: p.z < z, self._points))
points = [left] + points + [right]
points = sorted(points, key=lambda p: p.x)
return points
def get_water_limits(self, z): def get_water_limits(self, z):
""" """
Determine left and right limits of water elevation. Determine left and right limits of water elevation.
@@ -509,7 +469,7 @@ class ProfileXYZ(Profile, SQLSubModel):
[self.point(i_left).z, self.point(i_left - 1).z], [self.point(i_left).z, self.point(i_left - 1).z],
[self.point(i_left).y, self.point(i_left - 1).y] [self.point(i_left).y, self.point(i_left - 1).y]
) )
pt_left = PointXYZ(x, y, z) pt_left = PointXYZ(x, y, z, name="wl_left")
else: else:
pt_left = self.point(0) pt_left = self.point(0)
@@ -525,8 +485,151 @@ class ProfileXYZ(Profile, SQLSubModel):
[self.point(i_right).z, self.point(i_right + 1).z], [self.point(i_right).z, self.point(i_right + 1).z],
[self.point(i_right).y, self.point(i_right + 1).y] [self.point(i_right).y, self.point(i_right + 1).y]
) )
pt_right = PointXYZ(x, y, z) pt_right = PointXYZ(x, y, z, name="wl_right")
else: else:
pt_right = self.point(self.number_points - 1) pt_right = self.point(self.number_points - 1)
return pt_left, pt_right return pt_left, pt_right
def get_station(self):
"""Projection of the points of the profile on a plane.
Args:
self: The profile
Returns:
Projection of the points of the profile on a plane.
"""
if self.nb_points < 3:
return None
else:
return self._get_station(self.points)
@timer
def _get_station(self, points):
first_named_point = None
index_first_named_point = None
last_named_point = None
first_point_not_nan = self._first_point_not_nan(points)
last_point_not_nan = self._last_point_not_nan(points)
for index, point in enumerate(points):
if point.point_is_named():
index_first_named_point = index
first_named_point = point
break
for point in reversed(points):
if point.point_is_named():
last_named_point = point
break
station = []
constant = 0.0
if (first_named_point is not None
and last_named_point is not None):
if (first_named_point != last_named_point
and first_named_point.x != last_named_point.x):
vector = Vector1d(first_named_point, last_named_point)
norm_dir_vec = vector.normalized_direction_vector()
else:
vector = Vector1d(first_point_not_nan, last_point_not_nan)
norm_dir_vec = vector.normalized_direction_vector()
for point in points:
xi = point.x - first_named_point.x
yi = point.y - first_named_point.y
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)
norm_dir_vec = vector.normalized_direction_vector()
for point in points:
xi = point.x - first_point_not_nan.x
yi = point.y - first_point_not_nan.y
station_i = (norm_dir_vec[0] * xi +
norm_dir_vec[1] * yi)
station.append(station_i)
z_min = self.z_min()
index_profile_z_min = next(
filter(
lambda z: z[1] == z_min,
enumerate(self.z())
)
)
constant = station[index_profile_z_min[0]]
return list(map(lambda s: s - constant, station))
def _first_point_not_nan(self, points):
first_point = None
for point in points:
if not point.is_nan():
first_point = point
break
return first_point
def _last_point_not_nan(self, points):
last_point = None
for point in reversed(points):
if not point.is_nan():
last_point = point
break
return last_point
def purge(self, np_purge):
"""
Remove points to keep at most np_purge points.
"""
if (self.nb_points <= np_purge):
return
nb_named = 2 # we consider the first and last point as named
area = [0.0]
for i in range(1, self.nb_points-1):
if self.point(i).point_is_named():
area.append(9999999.999)
nb_named += 1
else:
area.append(
PointXYZ.areatriangle3d(
self.point(i-1),
self.point(i),
self.point(i+1))
)
area.append(0.0)
while self.nb_points > max(np_purge, nb_named):
to_rm = np.argmin(area[1:self.nb_points - 1]) + 1
self.delete_i([to_rm])
area.pop(to_rm)
for i in [to_rm-1, to_rm]:
if (i == 0):
continue
if (i == self.nb_points - 1):
continue
if self.point(i).point_is_named():
area[i] = 9999999.999
else:
area[i] = PointXYZ.areatriangle3d(
self.point(i-1),
self.point(i),
self.point(i+1)
)
+142 -34
View File
@@ -16,7 +16,9 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import os
import logging import logging
import shapefile
import numpy as np import numpy as np
from time import time from time import time
@@ -65,7 +67,7 @@ class Reach(SQLSubModel):
new._profiles = ProfileXYZ._db_load( new._profiles = ProfileXYZ._db_load(
execute, execute,
data=data data=data.copy()
) )
return new return new
@@ -111,7 +113,7 @@ class Reach(SQLSubModel):
def _get_profiles_list(self): def _get_profiles_list(self):
# Profiles list generator is type (int, Point) with the first # Profiles list generator is type (int, Point) with the first
# element the index of the Point in list # element the index of the Point in list
logger.info(f"Load profiles from reach {self.name}") logger.debug(f"Load profiles from reach {self.name}")
return list( return list(
map( map(
lambda p: p[1], lambda p: p[1],
@@ -139,10 +141,10 @@ class Reach(SQLSubModel):
def profiles(self, profiles): def profiles(self, profiles):
self._profiles = profiles self._profiles = profiles
def get_profiles_from_kp(self, kp): def get_profiles_from_rk(self, rk):
return list( return list(
filter( filter(
lambda p: p.kp == kp, self.profiles lambda p: p.rk == rk, self.profiles
) )
) )
@@ -333,18 +335,18 @@ class Reach(SQLSubModel):
) )
) )
def get_kp(self): def get_rk(self):
"""List of profiles kp """List of profiles rk
Returns: Returns:
List of profiles kp List of profiles rk
""" """
return [profile.kp for profile in self.profiles] return [profile.rk for profile in self.profiles]
def get_kp_complete_profiles(self): def get_rk_complete_profiles(self):
return list( return list(
map( map(
lambda profile: profile.kp, lambda profile: profile.rk,
filter( filter(
lambda profile: len(profile) > 0, lambda profile: len(profile) > 0,
self.profiles self.profiles
@@ -352,18 +354,42 @@ class Reach(SQLSubModel):
) )
) )
def get_kp_min(self): def get_rk_min(self):
if len(self.get_kp()) > 0: if len(self.get_rk()) > 0:
return min(self.get_kp()) return min(self.get_rk())
else: else:
return 0.0 return 0.0
def get_kp_max(self): def get_rk_max(self):
if len(self.get_kp()) > 0: if len(self.get_rk()) > 0:
return max(self.get_kp()) return max(self.get_rk())
else: else:
return 0.0 return 0.0
def inter_profiles_rk(self):
profiles = sorted(self.profiles, key=lambda p: p.rk)
first = profiles[0]
last = profiles[-1]
res_rk = [first.rk]
profiles = iter(profiles)
previous = next(profiles)
for profile in profiles:
prev = previous.rk
curr = profile.rk
diff = abs(previous.rk - profile.rk)
new = previous.rk + (diff / 2.0)
res_rk.append(new)
previous = profile
res_rk.append(last.rk)
return res_rk
# Sediment Layers # Sediment Layers
def get_sl(self): def get_sl(self):
@@ -510,7 +536,7 @@ class Reach(SQLSubModel):
def sort(self, is_reversed: bool = False): def sort(self, is_reversed: bool = False):
self.profiles = sorted( self.profiles = sorted(
self.profiles, self.profiles,
key=lambda profile: profile.kp, key=lambda profile: profile.rk,
reverse=is_reversed reverse=is_reversed
) )
self._status.modified() self._status.modified()
@@ -533,8 +559,66 @@ class Reach(SQLSubModel):
# Import/Export # Import/Export
@timer
def import_geometry(self, file_path_name: str): def import_geometry(self, file_path_name: str):
ext = file_path_name.split(".")
if len(ext) > 0:
ext = ext[-1].lower()
if ext in ['st', 'm']:
return self.import_geometry_st(file_path_name)
if ext in ['shp']:
return self.import_geometry_shp(file_path_name)
raise FileFormatError(
file_path_name,
f"Geometry file extention ('.{ext}') unkown"
)
@timer
def import_geometry_shp(self, file_path_name: str):
sf = shapefile.Reader(
os.path.abspath(file_path_name)
)
if sf.shapeType != shapefile.POLYLINEZ:
raise FileFormatError(
file_path_name,
"Shapefile is not a POLYLINEZ"
)
profiles = []
for i in range(len(sf)):
s_profile = sf.shape(i)
z = s_profile.z
ind = 0
points = []
for point in s_profile.points:
points.append([
point[0],
point[1],
z[ind]
])
ind += 1
prof = ProfileXYZ(
reach=self, status=self._status
)
prof.import_points(points)
profiles.append(prof)
self.profiles = profiles + self.profiles
self._update_profile_numbers()
self._recompute_rk()
return profiles
@timer
def import_geometry_st(self, file_path_name: str):
"""Import a geometry from file (.ST or .st) """Import a geometry from file (.ST or .st)
Args: Args:
@@ -549,7 +633,7 @@ class Reach(SQLSubModel):
try: try:
list_profile, list_header = self.read_file_st(str(file_path_name)) list_profile, list_header = self.read_file_st(str(file_path_name))
profile_header = ["num", "code1", profile_header = ["num", "code1",
"code2", "nb_point", "kp", "name"] "code2", "nb_point", "rk", "name"]
if list_profile and list_header: if list_profile and list_header:
for ind, profile in enumerate(list_profile): for ind, profile in enumerate(list_profile):
@@ -637,12 +721,12 @@ class Reach(SQLSubModel):
c1 = f"{profile.code1:>6}" c1 = f"{profile.code1:>6}"
c2 = f"{profile.code2:>6}" c2 = f"{profile.code2:>6}"
t = f"{len(profile.points):>6}" t = f"{len(profile.points):>6}"
kp = f"{profile.kp:>12f}"[0:12] rk = f"{profile.rk:>12f}"[0:12]
pname = profile.name pname = profile.name
if profile.name == "": if profile.name == "":
pname = f"p{profile.id:>3}".replace(" ", "0") pname = f"p{profile.id:>3}".replace(" ", "0")
st.write(f"{num}{c1}{c2}{t} {kp} {pname}\n") st.write(f"{num}{c1}{c2}{t} {rk} {pname}\n")
for point in profile.points: for point in profile.points:
x = f"{point.x:<12.4f}"[0:12] x = f"{point.x:<12.4f}"[0:12]
@@ -661,15 +745,15 @@ class Reach(SQLSubModel):
last = self.profile(len(self) - 1) last = self.profile(len(self) - 1)
z_first = first.z_min() z_first = first.z_min()
kp_first = first.kp rk_first = first.rk
z_last = last.z_min() z_last = last.z_min()
kp_last = last.kp rk_last = last.rk
return ( return (
(z_last - z_first) (z_last - z_first)
/ /
(kp_last - kp_first) (rk_last - rk_first)
) )
def get_incline_mean(self): def get_incline_mean(self):
@@ -680,15 +764,15 @@ class Reach(SQLSubModel):
for profile in profiles[1:]: for profile in profiles[1:]:
z_first = previous.z_min() z_first = previous.z_min()
kp_first = previous.kp rk_first = previous.rk
z_last = profile.z_min() z_last = profile.z_min()
kp_last = profile.kp rk_last = profile.rk
incline_acc += ( incline_acc += (
(z_last - z_first) (z_last - z_first)
/ /
(kp_last - kp_first) (rk_last - rk_first)
) )
previous = profile previous = profile
@@ -703,15 +787,15 @@ class Reach(SQLSubModel):
for profile in profiles[1:]: for profile in profiles[1:]:
z_first = previous.z_min() z_first = previous.z_min()
kp_first = previous.kp rk_first = previous.rk
z_last = profile.z_min() z_last = profile.z_min()
kp_last = profile.kp rk_last = profile.rk
incline_acc += [ incline_acc += [
(z_last - z_first) (z_last - z_first)
/ /
(kp_last - kp_first) (rk_last - rk_first)
] ]
previous = profile previous = profile
@@ -727,15 +811,15 @@ class Reach(SQLSubModel):
for profile in profiles[1:]: for profile in profiles[1:]:
z_first = previous.z_min() z_first = previous.z_min()
kp_first = previous.kp rk_first = previous.rk
z_last = profile.z_min() z_last = profile.z_min()
kp_last = profile.kp rk_last = profile.rk
incline_acc += [ incline_acc += [
(z_last - z_first) (z_last - z_first)
/ /
(kp_last - kp_first) (rk_last - rk_first)
] ]
previous = profile previous = profile
@@ -743,7 +827,10 @@ class Reach(SQLSubModel):
incline_acc.sort() incline_acc.sort()
marge = round(len(incline_acc) * 0.1) marge = round(len(incline_acc) * 0.1)
incline_set = incline_acc[marge:-marge] if marge > 0:
incline_set = incline_acc[marge:-marge]
else:
incline_set = incline_acc
logger.debug(f"+{incline_acc}") logger.debug(f"+{incline_acc}")
logger.debug(f"-{incline_set}") logger.debug(f"-{incline_set}")
@@ -759,3 +846,24 @@ class Reach(SQLSubModel):
except Exception as e: except Exception as e:
logger_exception(e) logger_exception(e)
return 0 return 0
def _recompute_rk(self, offset=0.0):
self._recompute_rk_no_gl(offset=offset)
def _recompute_rk_no_gl(self, offset=0.0):
profiles = iter(self.profiles)
previous = next(profiles)
previous.rk = offset
for profile in profiles:
prev_points = previous.points
curr_points = profile.points
dist = (
prev_points[0].dist_2d(curr_points[0]) +
prev_points[-1].dist_2d(curr_points[-1])
) / 2.0
profile.rk = previous.rk + dist
previous = profile
@@ -50,8 +50,8 @@ class HydraulicStructure(SQLSubModel):
self.id = id self.id = id
self._name = name self._name = name
self._input_kp = None self._input_rk = None
self._output_kp = None self._output_rk = None
self._input_reach = None self._input_reach = None
self._output_reach = None self._output_reach = None
self._enabled = True self._enabled = True
@@ -69,8 +69,8 @@ class HydraulicStructure(SQLSubModel):
id INTEGER NOT NULL PRIMARY KEY, id INTEGER NOT NULL PRIMARY KEY,
name TEXT NOT NULL, name TEXT NOT NULL,
enabled BOOLEAN NOT NULL, enabled BOOLEAN NOT NULL,
input_kp REAL NOT NULL, input_rk REAL NOT NULL,
output_kp REAL NOT NULL, output_rk REAL NOT NULL,
input_reach INTEGER, input_reach INTEGER,
output_reach INTEGER, output_reach INTEGER,
FOREIGN KEY(input_reach) REFERENCES river_reach(id), FOREIGN KEY(input_reach) REFERENCES river_reach(id),
@@ -84,9 +84,25 @@ class HydraulicStructure(SQLSubModel):
def _db_update(cls, execute, version): def _db_update(cls, execute, version):
major, minor, release = version.strip().split(".") major, minor, release = version.strip().split(".")
if major == minor == "0": if major == minor == "0":
if int(release) < 6: rl = int(release)
if rl < 6:
cls._db_create(execute) cls._db_create(execute)
if rl < 11:
execute(
"""
ALTER TABLE hydraulic_structures
RENAME COLUMN input_kp TO input_rk
"""
)
execute(
"""
ALTER TABLE hydraulic_structures
RENAME COLUMN output_kp TO output_rk
"""
)
return True return True
@classmethod @classmethod
@@ -95,7 +111,7 @@ class HydraulicStructure(SQLSubModel):
table = execute( table = execute(
"SELECT id, name, enabled, " + "SELECT id, name, enabled, " +
"input_kp, output_kp, " + "input_rk, output_rk, " +
"input_reach, output_reach " + "input_reach, output_reach " +
"FROM hydraulic_structures " "FROM hydraulic_structures "
) )
@@ -106,8 +122,8 @@ class HydraulicStructure(SQLSubModel):
hs_id = next(it) hs_id = next(it)
name = next(it) name = next(it)
enabled = (next(it) == 1) enabled = (next(it) == 1)
input_kp = next(it) input_rk = next(it)
output_kp = next(it) output_rk = next(it)
input_reach_id = next(it) input_reach_id = next(it)
output_reach_id = next(it) output_reach_id = next(it)
@@ -118,8 +134,8 @@ class HydraulicStructure(SQLSubModel):
) )
hs.enabled = enabled hs.enabled = enabled
hs.input_kp = input_kp if input_kp != -1 else None hs.input_rk = input_rk if input_rk != -1 else None
hs.output_kp = output_kp if output_kp != -1 else None hs.output_rk = output_rk if output_rk != -1 else None
hs.input_reach, hs.output_reach = reduce( hs.input_reach, hs.output_reach = reduce(
lambda acc, n: ( lambda acc, n: (
@@ -148,24 +164,24 @@ class HydraulicStructure(SQLSubModel):
if self._output_reach is not None: if self._output_reach is not None:
output_reach_id = self._output_reach.id output_reach_id = self._output_reach.id
input_kp = -1 input_rk = -1
if self.input_kp is not None: if self.input_rk is not None:
input_kp = self.input_kp input_rk = self.input_rk
output_kp = -1 output_rk = -1
if self.output_kp is not None: if self.output_rk is not None:
output_kp = self.output_kp output_rk = self.output_rk
sql = ( sql = (
"INSERT INTO " + "INSERT INTO " +
"hydraulic_structures(" + "hydraulic_structures(" +
" id, name, enabled, input_kp, output_kp, " + " id, name, enabled, input_rk, output_rk, " +
" input_reach, output_reach" + " input_reach, output_reach" +
") " + ") " +
"VALUES (" + "VALUES (" +
f"{self.id}, '{self._db_format(self._name)}', " + f"{self.id}, '{self._db_format(self._name)}', " +
f"{self._db_format(self.enabled)}, " + f"{self._db_format(self.enabled)}, " +
f"{input_kp}, {output_kp}, " + f"{input_rk}, {output_rk}, " +
f"{input_reach_id}, {output_reach_id}" + f"{input_reach_id}, {output_reach_id}" +
")" ")"
) )
@@ -198,24 +214,24 @@ class HydraulicStructure(SQLSubModel):
self._status.modified() self._status.modified()
@property @property
def input_kp(self): def input_rk(self):
return self._input_kp return self._input_rk
@input_kp.setter @input_rk.setter
def input_kp(self, input_kp): def input_rk(self, input_rk):
if input_kp is not None: if input_rk is not None:
input_kp = float(input_kp) input_rk = float(input_rk)
self._input_kp = input_kp self._input_rk = input_rk
self._status.modified() self._status.modified()
@property @property
def output_kp(self): def output_rk(self):
return self._output_kp return self._output_rk
@output_kp.setter @output_rk.setter
def output_kp(self, output_kp): def output_rk(self, output_rk):
self._output_kp = output_kp self._output_rk = output_rk
self._status.modified() self._status.modified()
@property @property
@@ -30,7 +30,7 @@ logger = logging.getLogger()
class Data(SQLSubModel): class Data(SQLSubModel):
def __init__(self, name: str = "", def __init__(self, name: str = "",
comment: str = "", reach=None, comment: str = "", reach=None,
kp: float = 0.0, discharge: float = 0.0, rk: float = 0.0, discharge: float = 0.0,
height: float = 0.0, height: float = 0.0,
status=None): status=None):
super(Data, self).__init__() super(Data, self).__init__()
@@ -42,14 +42,14 @@ class Data(SQLSubModel):
self._name = name self._name = name
self._comment = comment self._comment = comment
self._kp = kp self._rk = rk
self._discharge = discharge self._discharge = discharge
self._speed = 0.0 self._speed = 0.0
self._elevation = 0.0 self._elevation = 0.0
self._height = height self._height = height
if self._kp != 0.0: if self._rk != 0.0:
self._update_from_kp() self._update_from_rk()
if self._height != 0.0: if self._height != 0.0:
self._update_from_height() self._update_from_height()
if self._discharge != 0.0: if self._discharge != 0.0:
@@ -64,7 +64,7 @@ class Data(SQLSubModel):
name TEXT NOT NULL, name TEXT NOT NULL,
comment TEXT NOT NULL, comment TEXT NOT NULL,
reach INTEGER, reach INTEGER,
kp REAL NOT NULL, rk REAL NOT NULL,
discharge REAL NOT NULL, discharge REAL NOT NULL,
height REAL NOT NULL, height REAL NOT NULL,
FOREIGN KEY(reach) REFERENCES river_reach(id) FOREIGN KEY(reach) REFERENCES river_reach(id)
@@ -75,13 +75,20 @@ class Data(SQLSubModel):
@classmethod @classmethod
def _db_update(cls, execute, version): def _db_update(cls, execute, version):
major, minor, release = version.strip().split(".")
if major == minor == "0":
if int(release) < 11:
execute(
"ALTER TABLE initial_conditions RENAME COLUMN kp TO rk"
)
return cls._update_submodel(execute, version) return cls._update_submodel(execute, version)
@classmethod @classmethod
def _db_load(cls, execute, data=None): def _db_load(cls, execute, data=None):
id = data["reach"].id id = data["reach"].id
table = execute( table = execute(
"SELECT ind, name, comment, kp, discharge, height " + "SELECT ind, name, comment, rk, discharge, height " +
"FROM initial_conditions " + "FROM initial_conditions " +
f"WHERE reach = {id}" f"WHERE reach = {id}"
) )
@@ -95,7 +102,7 @@ class Data(SQLSubModel):
ind = row[0] ind = row[0]
name = row[1] name = row[1]
comment = row[2] comment = row[2]
kp = row[3] rk = row[3]
discharge = row[4] discharge = row[4]
height = row[5] height = row[5]
@@ -104,7 +111,7 @@ class Data(SQLSubModel):
status=data["status"], status=data["status"],
name=name, name=name,
comment=comment, comment=comment,
kp=kp, rk=rk,
discharge=discharge, discharge=discharge,
height=height, height=height,
) )
@@ -118,12 +125,12 @@ class Data(SQLSubModel):
execute( execute(
"INSERT INTO " + "INSERT INTO " +
"initial_conditions(ind, name, comment, kp, " + "initial_conditions(ind, name, comment, rk, " +
"discharge, height, reach) " + "discharge, height, reach) " +
"VALUES (" + "VALUES (" +
f"{ind}, '{self._db_format(self.name)}', " + f"{ind}, '{self._db_format(self.name)}', " +
f"'{self._db_format(self._comment)}', " + f"'{self._db_format(self._comment)}', " +
f"{self._kp}, {self._discharge}, {self._height}, " + f"{self._rk}, {self._discharge}, {self._height}, " +
f"{self._reach.id}" + f"{self._reach.id}" +
")" ")"
) )
@@ -134,7 +141,7 @@ class Data(SQLSubModel):
new = Data( new = Data(
name=self.name, name=self.name,
comment=self._comment, comment=self._comment,
kp=self._kp, rk=self._rk,
discharge=self._discharge, discharge=self._discharge,
height=self._height, height=self._height,
reach=self._reach, reach=self._reach,
@@ -154,8 +161,8 @@ class Data(SQLSubModel):
val = self._name val = self._name
elif key == "comment": elif key == "comment":
val = self._comment val = self._comment
elif key == "kp": elif key == "rk":
val = self._kp val = self._rk
elif key == "speed": elif key == "speed":
val = self._speed val = self._speed
elif key == "discharge": elif key == "discharge":
@@ -168,7 +175,7 @@ class Data(SQLSubModel):
return val return val
def _update_get_min(self): def _update_get_min(self):
profile = self._reach.reach.get_profiles_from_kp(self._kp) profile = self._reach.reach.get_profiles_from_rk(self._rk)
if len(profile) > 0: if len(profile) > 0:
min = profile[0].z_min() min = profile[0].z_min()
else: else:
@@ -176,7 +183,7 @@ class Data(SQLSubModel):
return min return min
def _update_from_kp(self): def _update_from_rk(self):
min = self._update_get_min() min = self._update_get_min()
self._elevation = min + self._height self._elevation = min + self._height
@@ -197,9 +204,9 @@ class Data(SQLSubModel):
self._name = str(value) self._name = str(value)
elif key == "comment": elif key == "comment":
self._comment = str(value) self._comment = str(value)
elif key == "kp": elif key == "rk":
self._kp = float(value) self._rk = float(value)
self._update_from_kp() self._update_from_rk()
elif key == "speed": elif key == "speed":
# Not supposed to be modified # Not supposed to be modified
self._speed = float(value) self._speed = float(value)
@@ -266,6 +273,9 @@ class InitialConditions(SQLSubModel):
def __len__(self): def __len__(self):
return len(self._data) return len(self._data)
def lst(self):
return self._data
@property @property
def reach(self): def reach(self):
return self._reach return self._reach
@@ -295,6 +305,15 @@ class InitialConditions(SQLSubModel):
self._data.insert(index, n) self._data.insert(index, n)
self._status.modified() self._status.modified()
def new_from_data(self, rk, discharge, elevation):
n = Data(reach=self._reach, status=self._status)
n['rk'] = rk
n['discharge'] = discharge
n['elevation'] = elevation
return n
def insert(self, index, data): def insert(self, index, data):
self._data.insert(index, data) self._data.insert(index, data)
self._status.modified() self._status.modified()
@@ -332,8 +351,8 @@ class InitialConditions(SQLSubModel):
) )
) )
def get_kp(self): def get_rk(self):
return self._data_get("kp") return self._data_get("rk")
def get_elevation(self): def get_elevation(self):
return self._data_get("elevation") return self._data_get("elevation")
@@ -341,10 +360,10 @@ class InitialConditions(SQLSubModel):
def get_discharge(self): def get_discharge(self):
return self._data_get("discharge") return self._data_get("discharge")
def _sort_by_z_and_kp(self, profiles): def _sort_by_z_and_rk(self, profiles):
profiles.sort( profiles.sort(
reverse=False, reverse=False,
key=lambda p: p.kp key=lambda p: p.rk
) )
first_z = profiles[0].z() first_z = profiles[0].z()
@@ -353,14 +372,14 @@ class InitialConditions(SQLSubModel):
if first_z > last_z: if first_z > last_z:
profiles.sort( profiles.sort(
reverse=True, reverse=True,
key=lambda p: p.kp key=lambda p: p.rk
) )
def generate_growing_constante_height(self, height: float): def generate_growing_constante_height(self, height: float):
self._data = [] self._data = []
profiles = self._reach.reach.profiles.copy() profiles = self._reach.reach.profiles.copy()
self._sort_by_z_and_kp(profiles) self._sort_by_z_and_rk(profiles)
incline = self._reach.reach.get_incline_median_mean() incline = self._reach.reach.get_incline_median_mean()
logger.debug(f"incline = {incline}") logger.debug(f"incline = {incline}")
@@ -381,13 +400,13 @@ class InitialConditions(SQLSubModel):
previous_elevation previous_elevation
) )
logger.debug(f"({profile.kp}):") logger.debug(f"({profile.rk}):")
logger.debug(f" width = {width}") logger.debug(f" width = {width}")
logger.debug(f" strickler = {strickler}") logger.debug(f" strickler = {strickler}")
logger.debug(f" discharge = {discharge}") logger.debug(f" discharge = {discharge}")
new = Data(reach=self._reach, status=self._status) new = Data(reach=self._reach, status=self._status)
new["kp"] = profile.kp new["rk"] = profile.rk
new["discharge"] = discharge new["discharge"] = discharge
new["elevation"] = elevation new["elevation"] = elevation
@@ -406,7 +425,7 @@ class InitialConditions(SQLSubModel):
def _generate_height_estimation_from_discharge(self, discharge: float): def _generate_height_estimation_from_discharge(self, discharge: float):
profiles = self._reach.reach.profiles.copy() profiles = self._reach.reach.profiles.copy()
self._sort_by_z_and_kp(profiles) self._sort_by_z_and_rk(profiles)
previous_elevation = -99999.99 previous_elevation = -99999.99
@@ -427,13 +446,13 @@ class InitialConditions(SQLSubModel):
previous_elevation previous_elevation
) )
logger.debug(f"({profile.kp}):") logger.debug(f"({profile.rk}):")
logger.debug(f" width = {width}") logger.debug(f" width = {width}")
logger.debug(f" strickler = {strickler}") logger.debug(f" strickler = {strickler}")
logger.debug(f" hieght = {height}") logger.debug(f" height = {height}")
new = Data(reach=self._reach, status=self._status) new = Data(reach=self._reach, status=self._status)
new["kp"] = profile.kp new["rk"] = profile.rk
new["discharge"] = discharge new["discharge"] = discharge
new["elevation"] = elevation new["elevation"] = elevation
@@ -444,10 +463,10 @@ class InitialConditions(SQLSubModel):
def _generate_resort_data(self, profiles): def _generate_resort_data(self, profiles):
is_reverse = False is_reverse = False
if profiles[0].kp > profiles[-1].kp: if profiles[0].rk > profiles[-1].rk:
is_reverse = True is_reverse = True
self._data.sort( self._data.sort(
reverse=not is_reverse, reverse=not is_reverse,
key=lambda d: d['kp'] key=lambda d: d['rk']
) )
@@ -18,7 +18,12 @@
import logging import logging
from tools import trace, timer, old_pamhyr_date_to_timestamp from tools import (
trace, timer,
old_pamhyr_date_to_timestamp,
date_iso_to_timestamp,
date_dmy_to_timestamp,
)
from Model.Tools.PamhyrDB import SQLSubModel from Model.Tools.PamhyrDB import SQLSubModel
from Model.Except import NotImplementedMethodeError from Model.Except import NotImplementedMethodeError
@@ -43,8 +48,8 @@ class LateralContribution(SQLSubModel):
self._name = name self._name = name
self._type = "" self._type = ""
self._edge = None self._edge = None
self._begin_kp = 0.0 self._begin_rk = 0.0
self._end_kp = 0.0 self._end_rk = 0.0
self._data = [] self._data = []
self._header = [] self._header = []
self._types = [float, float] self._types = [float, float]
@@ -61,8 +66,8 @@ class LateralContribution(SQLSubModel):
type TEXT NOT NULL, type TEXT NOT NULL,
tab TEXT NOT NULL, tab TEXT NOT NULL,
edge INTEGER, edge INTEGER,
begin_kp REAL NOT NULL, begin_rk REAL NOT NULL,
end_kp REAL NOT NULL, end_rk REAL NOT NULL,
FOREIGN KEY(edge) REFERENCES river_reach(id) FOREIGN KEY(edge) REFERENCES river_reach(id)
) )
""") """)
@@ -82,6 +87,22 @@ class LateralContribution(SQLSubModel):
@classmethod @classmethod
def _db_update(cls, execute, version): def _db_update(cls, execute, version):
major, minor, release = version.strip().split(".")
if major == minor == "0":
if int(release) < 11:
execute(
"""
ALTER TABLE lateral_contribution
RENAME COLUMN begin_kp TO begin_rk
"""
)
execute(
"""
ALTER TABLE lateral_contribution
RENAME COLUMN end_kp TO end_rk
"""
)
return True return True
@classmethod @classmethod
@@ -105,7 +126,7 @@ class LateralContribution(SQLSubModel):
tab = data["tab"] tab = data["tab"]
table = execute( table = execute(
"SELECT id, name, type, edge, begin_kp, end_kp " + "SELECT id, name, type, edge, begin_rk, end_rk " +
f"FROM lateral_contribution WHERE tab = '{tab}'" f"FROM lateral_contribution WHERE tab = '{tab}'"
) )
@@ -120,8 +141,8 @@ class LateralContribution(SQLSubModel):
lc.edge = None lc.edge = None
if row[3] != -1: if row[3] != -1:
lc.edge = next(filter(lambda e: e.id == row[3], data["edges"])) lc.edge = next(filter(lambda e: e.id == row[3], data["edges"]))
lc._begin_kp = row[4] lc._begin_rk = row[4]
lc._end_kp = row[5] lc._end_rk = row[5]
values = execute( values = execute(
"SELECT ind, data0, data1 FROM lateral_contribution_data " + "SELECT ind, data0, data1 FROM lateral_contribution_data " +
@@ -155,11 +176,11 @@ class LateralContribution(SQLSubModel):
sql = ( sql = (
"INSERT INTO " + "INSERT INTO " +
"lateral_contribution(id, name, type, tab, " + "lateral_contribution(id, name, type, tab, " +
"edge, begin_kp, end_kp) " + "edge, begin_rk, end_rk) " +
"VALUES (" + "VALUES (" +
f"{self.id}, '{self._db_format(self._name)}', " + f"{self.id}, '{self._db_format(self._name)}', " +
f"'{self._db_format(self._type)}', '{tab}', {edge}, " + f"'{self._db_format(self._type)}', '{tab}', {edge}, " +
f"{self._begin_kp}, {self._end_kp}" + f"{self._begin_rk}, {self._end_rk}" +
")" ")"
) )
execute(sql) execute(sql)
@@ -189,6 +210,10 @@ class LateralContribution(SQLSubModel):
@classmethod @classmethod
def time_convert(cls, data): def time_convert(cls, data):
if type(data) is str: if type(data) is str:
if data.count("-") == 2:
return date_iso_to_timestamp(data)
if data.count("/") == 2:
return date_dmy_to_timestamp(data)
if data.count(":") == 3: if data.count(":") == 3:
return old_pamhyr_date_to_timestamp(data) return old_pamhyr_date_to_timestamp(data)
if data.count(":") == 2: if data.count(":") == 2:
@@ -222,44 +247,44 @@ class LateralContribution(SQLSubModel):
def edge(self, edge): def edge(self, edge):
self._edge = edge self._edge = edge
if edge is not None: if edge is not None:
self._begin_kp = self._edge.reach.get_kp_min() self._begin_rk = self._edge.reach.get_rk_min()
self._end_kp = self._edge.reach.get_kp_max() self._end_rk = self._edge.reach.get_rk_max()
self._status.modified() self._status.modified()
def has_edge(self): def has_edge(self):
return self._edge is not None return self._edge is not None
@property @property
def begin_kp(self): def begin_rk(self):
return self._begin_kp return self._begin_rk
@begin_kp.setter @begin_rk.setter
def begin_kp(self, begin_kp): def begin_rk(self, begin_rk):
if self._edge is None: if self._edge is None:
self._begin_kp = begin_kp self._begin_rk = begin_rk
else: else:
_min = self._edge.reach.get_kp_min() _min = self._edge.reach.get_rk_min()
_max = self._edge.reach.get_kp_max() _max = self._edge.reach.get_rk_max()
if _min <= begin_kp <= _max: if _min <= begin_rk <= _max:
self._begin_kp = begin_kp self._begin_rk = begin_rk
self._status.modified() self._status.modified()
@property @property
def end_kp(self): def end_rk(self):
return self._end_kp return self._end_rk
@end_kp.setter @end_rk.setter
def end_kp(self, end_kp): def end_rk(self, end_rk):
if self._edge is None: if self._edge is None:
self._end_kp = end_kp self._end_rk = end_rk
else: else:
_min = self._edge.reach.get_kp_min() _min = self._edge.reach.get_rk_min()
_max = self._edge.reach.get_kp_max() _max = self._edge.reach.get_rk_max()
if _min <= end_kp <= _max: if _min <= end_rk <= _max:
self._end_kp = end_kp self._end_rk = end_rk
self._status.modified() self._status.modified()
@@ -368,8 +393,8 @@ class LateralContribution(SQLSubModel):
def convert(self, cls): 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.edge = self.edge
new.begin_kp = self.begin_kp new.begin_rk = self.begin_rk
new.end_kp = self.end_kp new.end_rk = self.end_rk
for i, _ in enumerate(self.data): for i, _ in enumerate(self.data):
new.add(i) new.add(i)
+27 -24
View File
@@ -67,6 +67,13 @@ class Graph(object):
def edges_counts(self): def edges_counts(self):
return len(self._edges) return len(self._edges)
def enable_nodes_counts(self):
return reduce(
lambda acc, n: acc + 1 if self.is_enable_node(n) else acc,
self._nodes,
0
)
def enable_edges_counts(self): def enable_edges_counts(self):
return reduce( return reduce(
lambda acc, e: acc + 1 if e.is_enable() else acc, lambda acc, e: acc + 1 if e.is_enable() else acc,
@@ -135,26 +142,21 @@ class Graph(object):
def insert_node(self, node): def insert_node(self, node):
return self._add_node(node) return self._add_node(node)
def remove_node(self, node_name: str): def remove_node(self, node):
self._nodes = list( self._nodes.remove(node)
filter( self._remove_associated_edge(node)
lambda n: n.name != node_name,
self._nodes
)
)
self._remove_associated_edge(node_name)
self._status.modified() self._status.modified()
def _remove_associated_edge(self, node_name: str): def _remove_associated_edge(self, node: str):
edges = list( edges = list(
filter( filter(
lambda e: (e.node1.name == node_name or lambda e: (e.node1 == node or
e.node2.name == node_name), e.node2 == node),
self._edges, self._edges,
) )
) )
for edge in edges: for edge in edges:
self.remove_edge(edge.name) self.remove_edge(edge)
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) node = self._create_node(x, y)
@@ -169,11 +171,11 @@ class Graph(object):
return edge return edge
def _add_edge(self, edge): def _add_edge(self, edge):
# This edge already exists ? # # This edge already exists ?
if any(filter(lambda e: (e.node1 == edge.node1 and # if any(filter(lambda e: (e.node1 == edge.node1 and
e.node2 == edge.node2), # e.node2 == edge.node2),
self._edges)): # self._edges)):
return None # return None
self._edges.append(edge) self._edges.append(edge)
@@ -190,13 +192,8 @@ class Graph(object):
def create_edge(self, n1: Node, n2: Node): def create_edge(self, n1: Node, n2: Node):
return self._create_edge(n1, n2) return self._create_edge(n1, n2)
def remove_edge(self, edge_name: str): def remove_edge(self, edge):
self._edges = list( self._edges.remove(edge)
filter(
lambda e: e.name != edge_name,
self._edges
)
)
self._status.modified() self._status.modified()
def is_upstream_node(self, node): def is_upstream_node(self, node):
@@ -213,6 +210,12 @@ class Graph(object):
True True
) )
def is_internal_node(self, node):
return not (
self.is_upstream_node(node) or
self.is_downstream_node(node)
)
def is_enable_node(self, node): def is_enable_node(self, node):
return reduce( return reduce(
lambda acc, e: ( lambda acc, e: (
+3 -3
View File
@@ -60,7 +60,7 @@ class GraphTestCase(unittest.TestCase):
e0 = g.add_edge(n0, n1) e0 = g.add_edge(n0, n1)
g.remove_edge(e0.name) g.remove_edge(e0)
self.assertEqual(g.nodes_counts(), 2) self.assertEqual(g.nodes_counts(), 2)
self.assertEqual(g.edges_counts(), 0) self.assertEqual(g.edges_counts(), 0)
@@ -73,7 +73,7 @@ class GraphTestCase(unittest.TestCase):
e0 = g.add_edge(n0, n1) e0 = g.add_edge(n0, n1)
g.remove_node(n0.name) g.remove_node(n0)
self.assertEqual(g.nodes_counts(), 1) self.assertEqual(g.nodes_counts(), 1)
self.assertEqual(g.edges_counts(), 0) self.assertEqual(g.edges_counts(), 0)
@@ -88,7 +88,7 @@ class GraphTestCase(unittest.TestCase):
e0 = g.add_edge(n0, n1) e0 = g.add_edge(n0, n1)
e1 = g.add_edge(n1, n2) e1 = g.add_edge(n1, n2)
g.remove_node(n1.name) g.remove_node(n1)
self.assertEqual(g.nodes_counts(), 2) self.assertEqual(g.nodes_counts(), 2)
self.assertEqual(g.edges_counts(), 0) self.assertEqual(g.edges_counts(), 0)
+180
View File
@@ -0,0 +1,180 @@
# REPLine.py -- Pamhyr
# Copyright (C) 2024 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 functools import reduce
from tools import trace, timer
from Model.Tools.PamhyrDB import SQLSubModel
from Model.Except import NotImplementedMethodeError
class REPLine(SQLSubModel):
_sub_classes = []
_id_cnt = 0
def __init__(self, id: int = -1, enabled=True,
name="", line="", solvers=set(),
status=None):
super(REPLine, self).__init__()
if id == -1:
self.id = REPLine._id_cnt
else:
self.id = id
self._status = status
self._enabled = enabled
self._name = f"Line{self.id}" if name == "" else name
self._line = line
self._solvers = solvers
REPLine._id_cnt = max(id, REPLine._id_cnt+1)
def __getitem__(self, key):
value = None
if key == "enabled":
value = self._enabled
elif key == "name":
value = self._name
elif key == "line":
value = self._line
elif key == "solvers":
value = self._solvers
return value
def __setitem__(self, key, value):
if key == "enabled":
self._enabled = value
elif key == "name":
self._name = value
elif key == "line":
self._line = value
elif key == "solvers":
self._solvers = value
self._status.modified()
@property
def enabled(self):
return self._enabled
@enabled.setter
def enabled(self, enabled):
self._enabled = enabled
self._status.modified()
def is_enabled(self):
return self._enabled
@property
def name(self):
return self._name
@name.setter
def name(self, name):
self._name = name
self._status.modified()
@property
def line(self):
return self._line
@line.setter
def line(self, line):
self._line = line
self._status.modified()
@property
def solvers(self):
return self._solvers
@solvers.setter
def solvers(self, solvers):
self._solvers = solvers
self._status.modified()
@classmethod
def _db_create(cls, execute):
execute("""
CREATE TABLE rep_lines(
id INTEGER NOT NULL PRIMARY KEY,
enabled BOOLEAN NOT NULL,
name TEXT NOT NULL,
line TEXT NOT NULL,
solvers TEXT NOT NULL
)
""")
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) < 9:
cls._db_create(execute)
return True
@classmethod
def _db_load(cls, execute, data=None):
new = []
table = execute(
"SELECT id, enabled, name, line, solvers " +
"FROM rep_lines"
)
for row in table:
it = iter(row)
id = next(it)
enabled = (next(it) == 1)
name = next(it)
line = next(it)
solvers = set(next(it).split(";;"))
f = cls(
id=id, enabled=enabled, name=name, line=line,
solvers=solvers, status=data['status']
)
new.append(f)
return new
def _db_save(self, execute, data=None):
solvers = ";;".join(self._solvers)
sql = (
"INSERT INTO " +
"rep_lines(id, enabled, name, line, solvers) " +
"VALUES (" +
f"{self.id}, {self._enabled}, " +
f"'{self._db_format(self._name)}', " +
f"'{self._db_format(self._line)}', " +
f"'{self._db_format(solvers)}'" +
")"
)
execute(sql)
return True
+56
View File
@@ -0,0 +1,56 @@
# REPLineList.py -- Pamhyr
# Copyright (C) 2024 INRAE
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
# -*- coding: utf-8 -*-
from tools import trace, timer
from Model.Except import NotImplementedMethodeError
from Model.Tools.PamhyrList import PamhyrModelList
from Model.REPLine.REPLine import REPLine
class REPLineList(PamhyrModelList):
_sub_classes = [REPLine]
@classmethod
def _db_load(cls, execute, data=None):
new = cls(status=data["status"])
new._lst = REPLine._db_load(execute, data)
return new
def _db_save(self, execute, data=None):
ok = True
# Delete previous data
execute("DELETE FROM rep_lines")
for af in self._lst:
ok &= af._db_save(execute, data)
return ok
@property
def lines(self):
return self.lst
def new(self, index):
n = REPLine(status=self._status)
self.insert(index, n)
self._status.modified()
return n
+10 -2
View File
@@ -35,8 +35,8 @@ class Profile(object):
return self._profile.name return self._profile.name
@property @property
def kp(self): def rk(self):
return self._profile.kp return self._profile.rk
@property @property
def geometry(self): def geometry(self):
@@ -126,3 +126,11 @@ class River(object):
self._reachs.append(new) self._reachs.append(new)
return new return new
def get_reach_by_geometry(self, geometry_reach):
return next(
filter(
lambda r: r.geometry is geometry_reach,
self._reachs
)
)
+63 -20
View File
@@ -40,6 +40,8 @@ from Model.Reservoir.ReservoirList import ReservoirList
from Model.HydraulicStructures.HydraulicStructuresList import ( from Model.HydraulicStructures.HydraulicStructuresList import (
HydraulicStructureList, HydraulicStructureList,
) )
from Model.AdditionalFile.AddFileList import AddFileList
from Model.REPLine.REPLineList import REPLineList
from Solver.Solvers import solver_type_list from Solver.Solvers import solver_type_list
@@ -224,6 +226,8 @@ class River(Graph, SQLSubModel):
SedimentLayerList, SedimentLayerList,
ReservoirList, ReservoirList,
HydraulicStructureList, HydraulicStructureList,
AddFileList,
REPLineList,
] ]
def __init__(self, status=None): def __init__(self, status=None):
@@ -245,6 +249,8 @@ class River(Graph, SQLSubModel):
self._hydraulic_structures = HydraulicStructureList( self._hydraulic_structures = HydraulicStructureList(
status=self._status status=self._status
) )
self._additional_files = AddFileList(status=self._status)
self._rep_lines = REPLineList(status=self._status)
@classmethod @classmethod
def _db_create(cls, execute): def _db_create(cls, execute):
@@ -263,66 +269,62 @@ class River(Graph, SQLSubModel):
# Stricklers (Stricklers is load in first because it's needed # Stricklers (Stricklers is load in first because it's needed
# for reachs) # for reachs)
new._stricklers = StricklersList._db_load( new._stricklers = StricklersList._db_load(
execute, execute, data
data
) )
data["stricklers"] = new._stricklers data["stricklers"] = new._stricklers
# Initial conditions # Initial conditions
new._sediment_layers = SedimentLayerList._db_load( new._sediment_layers = SedimentLayerList._db_load(
execute, execute, data
data
) )
data["sediment_layers_list"] = new._sediment_layers data["sediment_layers_list"] = new._sediment_layers
# Network # Network
new._nodes = RiverNode._db_load( new._nodes = RiverNode._db_load(
execute, execute, data
data
) )
data["nodes"] = new.nodes() data["nodes"] = new.nodes()
new._edges = RiverReach._db_load( new._edges = RiverReach._db_load(
execute, execute, data
data
) )
data["edges"] = new.edges() data["edges"] = new.edges()
# Boundary Condition # Boundary Condition
new._boundary_condition = BoundaryConditionList._db_load( new._boundary_condition = BoundaryConditionList._db_load(
execute, execute, data
data
) )
# Lateral Contribution # Lateral Contribution
new._lateral_contribution = LateralContributionList._db_load( new._lateral_contribution = LateralContributionList._db_load(
execute, execute, data
data
) )
# Initial conditions # Initial conditions
new._initial_conditions = InitialConditionsDict._db_load( new._initial_conditions = InitialConditionsDict._db_load(
execute, execute, data
data
) )
# Reservoir # Reservoir
new._reservoir = ReservoirList._db_load( new._reservoir = ReservoirList._db_load(
execute, execute, data
data
) )
# Hydraulic Structures # Hydraulic Structures
new._hydraulic_structures = HydraulicStructureList._db_load( new._hydraulic_structures = HydraulicStructureList._db_load(
execute, execute, data
data
) )
# Parameters # Parameters
new._parameters = SolverParametersList._db_load( new._parameters = SolverParametersList._db_load(
execute, execute, data
data
) )
# Additional Files
new._additional_files = AddFileList._db_load(
execute, data
)
new._rep_lines = REPLineList._db_load(execute, data)
return new return new
def _db_save(self, execute, data=None): def _db_save(self, execute, data=None):
@@ -336,6 +338,8 @@ class River(Graph, SQLSubModel):
objs.append(self._stricklers) objs.append(self._stricklers)
objs.append(self._reservoir) objs.append(self._reservoir)
objs.append(self._hydraulic_structures) objs.append(self._hydraulic_structures)
objs.append(self._additional_files)
objs.append(self._rep_lines)
for solver in self._parameters: for solver in self._parameters:
objs.append(self._parameters[solver]) objs.append(self._parameters[solver])
@@ -373,11 +377,42 @@ class River(Graph, SQLSubModel):
logger_exception(e) logger_exception(e)
def init_default(self): def init_default(self):
self.init_default_network()
self.init_default_sediment()
self.init_default_additional_files()
def init_default_network(self):
n1 = self.add_node(880.0, 950.0) n1 = self.add_node(880.0, 950.0)
n2 = self.add_node(1120.0, 1020.0) n2 = self.add_node(1120.0, 1020.0)
e = self.add_edge(n1, n2) e = self.add_edge(n1, n2)
def init_default_sediment(self):
sediment = self._sediment_layers
default = sediment.new(0)
default_0 = default.new(0)
default.name = "default"
default.comment = "Default sediment layers"
default_0.name = "L0"
default_0.height = 1.0
default_0.d50 = 0.002
default_0.sigma = 1.0
default_0.critical_constraint = 0.047
def init_default_additional_files(self):
add_file = self._additional_files.new(0)
add_file.name = "Pamhyr2 stamp file"
add_file.path = "Pamhyr2.txt"
add_file.text = """This repository has been generated by Pamhyr2 \
version "@version" !
All hand made file modification could be erased by the next solver
execution...
Last export at: @date."""
@property @property
def boundary_condition(self): def boundary_condition(self):
return self._boundary_condition return self._boundary_condition
@@ -419,6 +454,14 @@ class River(Graph, SQLSubModel):
def hydraulic_structures(self): def hydraulic_structures(self):
return self._hydraulic_structures return self._hydraulic_structures
@property
def additional_files(self):
return self._additional_files
@property
def rep_lines(self):
return self._rep_lines
@property @property
def parameters(self): def parameters(self):
return self._parameters return self._parameters
+17 -2
View File
@@ -22,10 +22,19 @@ logger = logging.getLogger()
class SavedStatus(object): class SavedStatus(object):
def __init__(self): def __init__(self, version=0):
super(SavedStatus, self).__init__() super(SavedStatus, self).__init__()
self._version = version
self._saved = True self._saved = True
@property
def version(self):
return self._version
@version.setter
def version(self, version):
self._version = version
def is_saved(self): def is_saved(self):
return self._saved return self._saved
@@ -34,5 +43,11 @@ class SavedStatus(object):
self._saved = True self._saved = True
def modified(self): def modified(self):
# logger.debug("model status set as modified") if self._saved:
self._version += 1
logger.debug(
"STATUS: Model status set as modified " +
f"at version {self._version}"
)
self._saved = False self._saved = False
+60 -13
View File
@@ -41,7 +41,7 @@ class Study(SQLModel):
def __init__(self, filename=None, init_new=True): def __init__(self, filename=None, init_new=True):
# Metadata # Metadata
self._version = "0.0.7" self._version = "0.0.11"
self.creation_date = datetime.now() self.creation_date = datetime.now()
self.last_modification_date = datetime.now() self.last_modification_date = datetime.now()
self.last_save_date = datetime.now() self.last_save_date = datetime.now()
@@ -71,6 +71,8 @@ class Study(SQLModel):
lst = [ lst = [
StudyNetworkReachChecker(), StudyNetworkReachChecker(),
StudyGeometryChecker(), StudyGeometryChecker(),
StudyInitialConditionsChecker(),
StudyBoundaryConditionChecker(),
# DummyOK(), # DummyOK(),
# DummyWARNING(), # DummyWARNING(),
# DummyERROR(), # DummyERROR(),
@@ -103,10 +105,23 @@ class Study(SQLModel):
except Exception as e: except Exception as e:
logger.error(e) logger.error(e)
is_new = False
fname = fname + "." + str(self._old_save_id) fname = fname + "." + str(self._old_save_id)
filename = os.path.join(fdir, "_PAMHYR_", "__old__", fname) if os.path.exists(self.filename) and ".backup" not in self.filename:
shutil.copy(self.filename, filename) filename = os.path.join(fdir, "_PAMHYR_", "__old__", fname)
self._old_save_id += 1 logger.debug(f"Backup previous version copy: {filename}")
shutil.copy(self.filename, filename)
self._old_save_id += 1
is_new = True
if ".backup" in self.filename:
is_new = True
if not os.path.exists(self.filename):
is_new = True
self._init_db_file(self.filename, is_new=is_new)
self.commit()
# Save # Save
self.last_save_date = datetime.now() self.last_save_date = datetime.now()
@@ -128,6 +143,16 @@ class Study(SQLModel):
@filename.setter @filename.setter
def filename(self, filename): def filename(self, filename):
if filename is None:
self._filename = None
self.status.modified()
return
if self._filename is not None and self._filename != "":
self._filename = str(filename)
self.status.modified()
return
self._filename = str(filename) self._filename = str(filename)
self._init_db_file(filename, is_new=True) self._init_db_file(filename, is_new=True)
self.status.modified() self.status.modified()
@@ -187,7 +212,12 @@ class Study(SQLModel):
def _create(self): def _create(self):
# Info (metadata) # Info (metadata)
self.execute( self.execute(
"CREATE TABLE info(key TEXT NOT NULL UNIQUE, value TEXT NOT NULL)") "INSERT INTO info VALUES ('study_release', '0')"
)
self.execute(
"CREATE TABLE info(key TEXT NOT NULL UNIQUE, value TEXT NOT NULL)"
)
self.execute( self.execute(
"INSERT INTO info VALUES ('version', " + "INSERT INTO info VALUES ('version', " +
f"'{self._db_format(self._version)}')", f"'{self._db_format(self._version)}')",
@@ -196,19 +226,20 @@ class Study(SQLModel):
self.execute("INSERT INTO info VALUES ('name', '')") self.execute("INSERT INTO info VALUES ('name', '')")
self.execute("INSERT INTO info VALUES ('description', '')") self.execute("INSERT INTO info VALUES ('description', '')")
self.execute( self.execute(
f"INSERT INTO info VALUES ('time_system', '{self._time_system}')") f"INSERT INTO info VALUES ('time_system', '{self._time_system}')"
)
self.execute( self.execute(
f"INSERT INTO info VALUES ('date', " + "INSERT INTO info VALUES ('date', " +
"'{self._date.timestamp()}')" f"'{self._date.timestamp()}')"
) )
self.execute( self.execute(
f"INSERT INTO info VALUES ('creation_date', " + "INSERT INTO info VALUES ('creation_date', " +
"'{self.creation_time.timestamp()}')" f"'{self.creation_date.timestamp()}')"
) )
self.execute( self.execute(
f"INSERT INTO info VALUES ('last_save_date', " + "INSERT INTO info VALUES ('last_save_date', " +
"'{self.last_save_time.timestamp()}')" f"'{self.last_save_date.timestamp()}')"
) )
self._create_submodel() self._create_submodel()
@@ -221,7 +252,13 @@ class Study(SQLModel):
if version[0] == self._version: if version[0] == self._version:
return True return True
logger.info("Update database") logger.debug("Update database")
major, minor, release = version[0].split('.')
if major == "0" and minor == "0" and int(release) < 10:
self.execute(
"INSERT INTO info VALUES ('study_release', '0')"
)
if self._update_submodel(version[0]): if self._update_submodel(version[0]):
self.execute( self.execute(
@@ -236,6 +273,11 @@ class Study(SQLModel):
def _load(cls, filename): def _load(cls, filename):
new = cls(init_new=False, filename=filename) new = cls(init_new=False, filename=filename)
version = new.execute(
"SELECT value FROM info WHERE key='study_release'"
)
new.status.version = int(version[0])
# TODO: Load metadata # TODO: Load metadata
new.name = new.execute("SELECT value FROM info WHERE key='name'")[0] new.name = new.execute("SELECT value FROM info WHERE key='name'")[0]
new.description = new.execute( new.description = new.execute(
@@ -270,6 +312,11 @@ class Study(SQLModel):
def _save(self, progress=None): def _save(self, progress=None):
progress = progress if progress is not None else lambda: None progress = progress if progress is not None else lambda: None
self.execute(
"INSERT INTO info VALUES ('study_release', " +
f"'{self.status.version}')"
)
self.execute( self.execute(
f"UPDATE info SET " + f"UPDATE info SET " +
f"value='{self._db_format(self.name)}' WHERE key='name'" f"value='{self._db_format(self.name)}' WHERE key='name'"
+4 -1
View File
@@ -35,8 +35,10 @@ class SQLModel(SQL):
_sub_classes = [] _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 self._db is not None:
self._db.close()
exists = Path(db).exists()
if exists and is_new: if exists and is_new:
os.remove(db) os.remove(db)
@@ -68,6 +70,7 @@ class SQLModel(SQL):
return True return True
def _create(self): def _create(self):
raise NotImplementedMethodeError(self, self._create) raise NotImplementedMethodeError(self, self._create)
def _update_submodel(self, version): def _update_submodel(self, version):
+1 -1
View File
@@ -56,7 +56,7 @@ class PamhyrModelDict(SQLSubModel):
def __len__(self): def __len__(self):
return len(self._dict) return len(self._dict)
def is_defined(self, key): def __contains__(self, key):
return key in self._dict return key in self._dict
def set(self, key, new): def set(self, key, new):
+79 -8
View File
@@ -62,16 +62,27 @@ class PamhyrModelList(SQLSubModel):
@property @property
def lst(self): def lst(self):
"""Return the PamhyrList as a Python list
Return the PamhyrList as a Python list. This list is the
current list used in the PamhyrList object. **If you need to
modify it, please make a copy.**
Returns: The Python list
"""
return self._lst return self._lst
def __len__(self): def __len__(self):
return len(self.lst) return len(self.lst)
def get(self, row): def index(self, el):
return self._lst[row] return self._lst.index(el)
def set(self, row, new): def get(self, index):
self._lst[row] = new return self._lst[index]
def set(self, index, new):
self._lst[index] = new
if self._status is not None: if self._status is not None:
self._status.modified() self._status.modified()
@@ -93,6 +104,14 @@ class PamhyrModelList(SQLSubModel):
self._status.modified() self._status.modified()
def delete(self, lst): def delete(self, lst):
"""Delete a list of elements
Args:
lst: The list of elements
Returns:
Nothing
"""
for el in lst: for el in lst:
self._lst.remove(el) self._lst.remove(el)
@@ -100,6 +119,14 @@ class PamhyrModelList(SQLSubModel):
self._status.modified() self._status.modified()
def delete_i(self, indexes): def delete_i(self, indexes):
"""Delete elements from list of indexes
Args:
indexes: The elements indexes
Returns:
Nothing
"""
lst = list( lst = list(
map( map(
lambda x: x[1], lambda x: x[1],
@@ -175,16 +202,32 @@ class PamhyrModelListWithTab(SQLSubModel):
################ ################
def len(self, lst): def len(self, lst):
"""Size of tab list
Args:
lst: The tab name
Returns:
The size of the tab list
"""
return len(self._tabs[lst]) return len(self._tabs[lst])
def get_tab(self, lst): def get_tab(self, lst):
"""Get tab list (copy) from name
Args:
lst: The tab name
Returns:
Nothing
"""
return self._tabs[lst].copy() return self._tabs[lst].copy()
def get(self, lst, row): def get(self, lst, index):
return self._tabs[lst][row] return self._tabs[lst][index]
def set(self, lst, row, new): def set(self, lst, index, new):
self._tabs[lst][row] = new self._tabs[lst][index] = new
self._status.modified() self._status.modified()
def new(self, lst, index): def new(self, lst, index):
@@ -200,15 +243,43 @@ class PamhyrModelListWithTab(SQLSubModel):
raise NotImplementedMethodeError(self, self.new) raise NotImplementedMethodeError(self, self.new)
def insert(self, lst, index, new): def insert(self, lst, index, new):
"""Insert element in tab
Args:
lst: The tab name
index: The index of new element
new: The new elements
Returns:
Nothing
"""
self._tabs[lst].insert(index, new) self._tabs[lst].insert(index, new)
self._status.modified() self._status.modified()
def delete(self, lst, els): def delete(self, lst, els):
"""Delete elements from specific tab
Args:
lst: The tab name
els: The elements list
Returns:
Nothing
"""
for el in els: for el in els:
self._tabs[lst].remove(el) self._tabs[lst].remove(el)
self._status.modified() self._status.modified()
def delete_i(self, lst, indexes): def delete_i(self, lst, indexes):
"""Delete elements from specific tab
Args:
lst: The tab name
indexes: The elements indexes
Returns:
Nothing
"""
els = list( els = list(
map( map(
lambda x: x[1], lambda x: x[1],
+164
View File
@@ -0,0 +1,164 @@
# Modules.py -- Pamhyr
# Copyright (C) 2024 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 enum import Flag, auto
logger = logging.getLogger()
class IterableFlag(Flag):
def __iter__(self):
all = filter(
lambda v: v in self,
type(self).values()
)
return all
@classmethod
def values(cls):
raise Exception("Not implemented yet")
class Modules(IterableFlag):
NONE = 0
# General
STUDY = auto()
CONFIG = auto()
# Modelling
NETWORK = auto()
GEOMETRY = auto()
BOUNDARY_CONDITION = auto()
LATERAL_CONTRIBUTION = auto()
FRICTION = auto()
INITIAL_CONDITION = auto()
HYDRAULIC_STRUCTURES = auto()
RESERVOIR = auto()
SEDIMENT_LAYER = auto()
ADDITIONAL_FILES = auto()
# Results
RESULTS = auto()
# Display
WINDOW_LIST = auto()
@classmethod
def values(cls):
return [
cls.STUDY, cls.CONFIG,
cls.NETWORK,
cls.GEOMETRY,
cls.BOUNDARY_CONDITION,
cls.LATERAL_CONTRIBUTION,
cls.FRICTION,
cls.INITIAL_CONDITION,
cls.HYDRAULIC_STRUCTURES,
cls.RESERVOIR,
cls.SEDIMENT_LAYER,
cls.ADDITIONAL_FILES,
cls.RESULTS,
cls.WINDOW_LIST,
]
@classmethod
def all(cls):
return ~cls.NONE
@classmethod
def modelling(cls):
return (
cls.NETWORK
| cls.GEOMETRY
| cls.BOUNDARY_CONDITION
| cls.LATERAL_CONTRIBUTION
| cls.FRICTION
| cls.INITIAL_CONDITION
| cls.HYDRAULIC_STRUCTURES
| cls.RESERVOIR
| cls.SEDIMENT_LAYER
)
@classmethod
def modelling_list(cls):
return [
cls.NETWORK,
cls.GEOMETRY,
cls.BOUNDARY_CONDITION,
cls.LATERAL_CONTRIBUTION,
cls.FRICTION,
cls.INITIAL_CONDITION,
cls.HYDRAULIC_STRUCTURES,
cls.RESERVOIR,
cls.SEDIMENT_LAYER,
]
@classmethod
def modelling_display_name(cls):
return {
cls.NETWORK: "Network",
cls.GEOMETRY: "Geometry",
cls.BOUNDARY_CONDITION: "Boundary condition",
cls.LATERAL_CONTRIBUTION: "Lateral contribution",
cls.FRICTION: "Friction",
cls.INITIAL_CONDITION: "Initial condition",
cls.HYDRAULIC_STRUCTURES: "Hydraulic structures",
cls.RESERVOIR: "Reservoir",
cls.SEDIMENT_LAYER: "Sediment layer",
}
def impact(self):
res = Modules(0)
for mod in self:
if mod in _impact:
for i in _impact[mod]:
res |= i
return res
def impact_set(self):
res = []
for mod in self:
if mod in _impact:
res += _impact[mod]
return set(res)
_impact = {
Modules.NETWORK: [
Modules.GEOMETRY, Modules.BOUNDARY_CONDITION,
Modules.LATERAL_CONTRIBUTION, Modules.FRICTION,
Modules.RESERVOIR, Modules.SEDIMENT_LAYER,
],
Modules.GEOMETRY: [
Modules.LATERAL_CONTRIBUTION, Modules.FRICTION,
Modules.INITIAL_CONDITION, Modules.SEDIMENT_LAYER,
],
Modules.BOUNDARY_CONDITION: [
Modules.SEDIMENT_LAYER
],
Modules.LATERAL_CONTRIBUTION: [],
Modules.FRICTION: [],
Modules.INITIAL_CONDITION: [],
Modules.HYDRAULIC_STRUCTURES: [],
Modules.RESERVOIR: [],
Modules.SEDIMENT_LAYER: [],
}
+59 -7
View File
@@ -19,7 +19,8 @@
import os import os
import logging import logging
from tools import timer, parse_command_line from datetime import datetime
from tools import timer, parse_command_line, get_version, logger_exception
try: try:
# Installation allow Unix-like signal # Installation allow Unix-like signal
@@ -124,6 +125,54 @@ class CommandLineSolver(AbstractSolver):
""" """
raise NotImplementedMethodeError(self, self.log_file) raise NotImplementedMethodeError(self, self.log_file)
def export_additional_files(self, study, repertory, qlog, name="0"):
files = []
if qlog is not None:
qlog.put("Export additional files")
add_files = study.river.additional_files.files
for add_file in add_files:
self.export_additional_file(
study, add_file, repertory, files
)
return files
def export_additional_file(self, study, add_file, repertory, files):
if add_file.path == "" or not add_file.is_enabled():
return files
path = os.path.join(repertory, add_file.path)
os.makedirs(
os.path.dirname(path),
exist_ok=True
)
with open(path, "w+") as f:
files.append(add_file.path)
txt = add_file.text
txt = txt.replace("@version", get_version())
txt = txt.replace("@date", datetime.now().isoformat(sep=" "))
f.write(txt)
return files
def export_study_description(self, study, repertory, qlog, name="0"):
files = []
path = os.path.join(
repertory, "pamhyr-study-description.txt"
)
with open(path, "w+", encoding='utf-8') as f:
txt = study.description\
.encode()\
.decode('utf-8', 'replace')
f.write(txt)
####### #######
# Run # # Run #
####### #######
@@ -132,7 +181,7 @@ class CommandLineSolver(AbstractSolver):
return os.path.abspath( return os.path.abspath(
os.path.join( os.path.join(
os.path.dirname(__file__), os.path.dirname(__file__),
"..", "..", ".." "..", ".."
) )
) )
@@ -224,12 +273,15 @@ class CommandLineSolver(AbstractSolver):
return True return True
def _data_ready(self): def _data_ready(self):
# Read process output and put lines in queue try:
s = self._process.readAll().data().decode() # Read process output and put lines in queue
s = self._process.readAll().data().decode('utf-8', "replace")
if self._output is not None: if self._output is not None:
for x in s.split('\n'): for x in s.split(os.linesep):
self._output.put(x) self._output.put(x)
except Exception as e:
logger_exception(e)
def _finished(self, study, exit_code, exit_status): def _finished(self, study, exit_code, exit_status):
if self._output is not None: if self._output is not None:
+151 -112
View File
@@ -20,7 +20,7 @@ import os
import logging import logging
import numpy as np import numpy as np
from tools import timer, trace from tools import timer, trace, logger_exception
from Solver.CommandLine import CommandLineSolver from Solver.CommandLine import CommandLineSolver
from Checker.Mage import ( from Checker.Mage import (
@@ -98,10 +98,9 @@ class Mage(CommandLineSolver):
@classmethod @classmethod
def checkers(cls): def checkers(cls):
lst = [ lst = [
MageNetworkGraphChecker(connectivity=True), MageNetworkGraphChecker(connectivity=True, version=cls._type),
MageNetworkGraphChecker(connectivity=False), MageNetworkGraphChecker(connectivity=False, version=cls._type),
MageGeometryGuideLineChecker(), MageGeometryGuideLineChecker(version=cls._type),
] ]
return lst return lst
@@ -129,7 +128,15 @@ class Mage(CommandLineSolver):
name = self._study.name name = self._study.name
return f"{name}.TRA" return f"{name}.TRA"
@timer def _export_REP_additional_lines(self, study, rep_file):
lines = filter(
lambda line: line.is_enabled(),
study.river.rep_lines.lines
)
for line in lines:
rep_file.write(line.line)
def _export_ST(self, study, repertory, qlog, name="0"): def _export_ST(self, study, repertory, qlog, name="0"):
files = [] files = []
@@ -139,14 +146,7 @@ class Mage(CommandLineSolver):
os.makedirs(os.path.join(repertory, "net"), exist_ok=True) os.makedirs(os.path.join(repertory, "net"), exist_ok=True)
# Write header # Write header
edges = study.river.edges() edges = study.river.enable_edges()
edges = list(
filter(
lambda e: e.is_enable(),
edges
)
)
for edge in edges: for edge in edges:
name = f"Reach_{edge.id + 1:>3}".replace(" ", "0") name = f"Reach_{edge.id + 1:>3}".replace(" ", "0")
@@ -180,7 +180,7 @@ class Mage(CommandLineSolver):
c1 = f"{profile.code1:>6}" c1 = f"{profile.code1:>6}"
c2 = f"{profile.code2:>6}" c2 = f"{profile.code2:>6}"
t = f"{len(profile.points):>6}" t = f"{len(profile.points):>6}"
kp = f"{profile.kp:>12f}"[0:12] rk = f"{profile.rk:>12f}"[0:12]
pname = profile.name pname = profile.name
if profile.name == "": if profile.name == "":
# Generate name from profile id prefixed with # Generate name from profile id prefixed with
@@ -207,7 +207,7 @@ class Mage(CommandLineSolver):
) )
# Profile header line # Profile header line
wfile.write(f"{num}{c1}{c2}{t} {kp} {pname} {sediment}\n") wfile.write(f"{num}{c1}{c2}{t} {rk} {pname} {sediment}\n")
def _export_ST_point_line(self, wfile, files, point): def _export_ST_point_line(self, wfile, files, point):
x = f"{point.x:<12.4f}"[0:12] x = f"{point.x:<12.4f}"[0:12]
@@ -235,7 +235,6 @@ class Mage(CommandLineSolver):
# Point line # Point line
wfile.write(f"{x} {y} {z} {n} {sediment}\n") wfile.write(f"{x} {y} {z} {n} {sediment}\n")
@timer
def _export_BC(self, t, bounds, repertory, qlog, name="0"): def _export_BC(self, t, bounds, repertory, qlog, name="0"):
files = [] files = []
@@ -269,7 +268,6 @@ class Mage(CommandLineSolver):
return files return files
@timer
def _export_bound_cond(self, study, repertory, qlog, name="0"): def _export_bound_cond(self, study, repertory, qlog, name="0"):
files = [] files = []
lst = study.river.boundary_condition lst = study.river.boundary_condition
@@ -284,6 +282,9 @@ class Mage(CommandLineSolver):
if bound.node is None: if bound.node is None:
continue continue
if not study.river.is_enable_node(bound.node):
continue
if bound.bctype == "ZD": if bound.bctype == "ZD":
AVA.append(bound) AVA.append(bound)
elif bound.bctype == "TD" or bound.bctype == "PC": elif bound.bctype == "TD" or bound.bctype == "PC":
@@ -300,42 +301,55 @@ class Mage(CommandLineSolver):
return files return files
# @timer def _export_lateral_contrib(self, study, repertory, qlog, name="0"):
# def _export_LC(self, lateral, repertory, qlog, name="0"): files = []
# files = [] lst = study.river.lateral_contribution
# if qlog is not None: with mage_file_open(
# qlog.put(f"Export LAT file") os.path.join(repertory, f"{name}.LAT"),
"w+"
) as f:
if qlog is not None:
qlog.put(f"Export LAT file")
files.append(f"{name}.LAT")
# with mage_file_open( for tab in ["liquid", "solid", "suspenssion"]:
# os.path.join(repertory, f"{name}.LAT"), for lateral in lst.get_tab(tab):
# "w+" self._export_LC(study, lateral, f, qlog)
# ) as f:
# files.append(f"{name}.LAT")
# name = f"{lateral.node.id:3}".replace(" ", "x") return files
# f.write(f"* {lateral.node.name} ({name}) {lateral.bctype}\n")
# f.write(f"${name}\n")
# header = lateral.header
# f.write(f"*{header[0]:>9}|{header[1]:>10}\n")
# for d in lateral.data: def _export_LC(self, study, lateral, f, qlog, name="0"):
# f.write(f"{d[0]:1{name}.3f}{d[1]:10.3f}\n") if lateral.edge is None:
return
# return files edges = study.river.enable_edges()
if lateral.edge not in edges:
return
# @timer eid, _ = next(
# def _export_lateral_contrib(self, study, repertory, qlog, name="0"): filter(
# files = [] lambda e: e[1] == lateral.edge,
# lst = study.river.lateral_contribution enumerate(edges)
)
)
# for tab in ["liquid", "solid", "suspenssion"]: name = f"{eid+1:>3}"
# for lateral in lst.get_tab(tab): # name = f"Reach_{lateral.edge.id + 1:>3}".replace(" ", "0")
# files = files + self._export_LC(lateral, repertory, qlog) f.write(f"* {lateral.edge.name} ({name}) {lateral.lctype}\n")
f.write(
f"${name} " +
f"{lateral.begin_rk:>10.4f} {lateral.end_rk:>10.4f}\n"
)
header = lateral.header
f.write(f"*{header[0]:>9}|{header[1]:>10}\n")
# return files for d in lateral.data:
if lateral.lctype in ["EV"]:
f.write(f"{d[0]:10.3f}{-d[1]:10.3f}\n")
else:
f.write(f"{d[0]:10.3f}{d[1]:10.3f}\n")
@timer
def _export_RUG(self, study, repertory, qlog, name="0"): def _export_RUG(self, study, repertory, qlog, name="0"):
files = [] files = []
@@ -346,13 +360,7 @@ class Mage(CommandLineSolver):
with mage_file_open(os.path.join(repertory, f"{name}.RUG"), "w+") as f: with mage_file_open(os.path.join(repertory, f"{name}.RUG"), "w+") as f:
files.append(f"{name}.RUG") files.append(f"{name}.RUG")
edges = study.river.edges() edges = study.river.enable_edges()
edges = list(
filter(
lambda e: e.is_enable(),
edges
)
)
id = 1 id = 1
for edge in edges: for edge in edges:
@@ -363,23 +371,22 @@ class Mage(CommandLineSolver):
continue continue
num = f"{id:>3}" num = f"{id:>3}"
bkp = f"{friction.begin_kp:>10.3f}" brk = f"{friction.begin_rk:>10.3f}"
ekp = f"{friction.end_kp:>10.3f}" erk = f"{friction.end_rk:>10.3f}"
# if friction.begin_kp != friction.end_kp: # if friction.begin_rk != friction.end_rk:
# print("TODO") # print("TODO")
strickler = friction.begin_strickler strickler = friction.begin_strickler
coef_1 = f"{strickler.minor:>10.3f}" coef_1 = f"{strickler.minor:>10.3f}"
coef_2 = f"{strickler.medium:>10.3f}" coef_2 = f"{strickler.medium:>10.3f}"
f.write(f"K{num} {bkp}{ekp}{coef_1}{coef_2}\n") f.write(f"K{num} {brk}{erk}{coef_1}{coef_2}\n")
id += 1 id += 1
return files return files
@timer
def _export_INI(self, study, repertory, qlog, name="0"): def _export_INI(self, study, repertory, qlog, name="0"):
files = [] files = []
@@ -394,7 +401,7 @@ class Mage(CommandLineSolver):
# TODO put real date... # TODO put real date...
f.write(f"$ date en minutes : 0.00\n") f.write(f"$ date en minutes : 0.00\n")
f.write(f"* IB IS discharge elevation kp\n") f.write(f"* IB IS discharge elevation rk\n")
id = 1 id = 1
for reach in reachs: for reach in reachs:
@@ -411,9 +418,9 @@ class Mage(CommandLineSolver):
IS = f"{id_sec}" IS = f"{id_sec}"
discharge = f"{d['discharge']:>10.5f}" discharge = f"{d['discharge']:>10.5f}"
elevation = f"{d['elevation']:>11.6f}" elevation = f"{d['elevation']:>11.6f}"
kp = f"{d['kp']:>9.2f}" rk = f"{d['rk']:>9.2f}"
f.write(f"{IR} {IS} {discharge} {elevation} {kp}\n") f.write(f"{IR} {IS} {discharge} {elevation} {rk}\n")
id_sec += 1 id_sec += 1
id += 1 id += 1
@@ -422,7 +429,6 @@ class Mage(CommandLineSolver):
files.append(f"{name}.INI") files.append(f"{name}.INI")
return files return files
@timer
def _export_CAS(self, study, repertory, qlog, name="0"): def _export_CAS(self, study, repertory, qlog, name="0"):
files = [] files = []
@@ -455,7 +461,6 @@ class Mage(CommandLineSolver):
return files return files
@timer
def _export_SIN(self, study, repertory, qlog, name="0"): def _export_SIN(self, study, repertory, qlog, name="0"):
files = [] files = []
@@ -487,35 +492,45 @@ class Mage(CommandLineSolver):
if not hs.input_reach.is_enable(): if not hs.input_reach.is_enable():
continue continue
if not hs.enabled:
continue
if hs.input_rk is None:
continue
f.write( f.write(
'* ouvrage au pk ' + '* ouvrage au pk ' +
f"{float(hs.input_kp):>12.1f}" + ' ' + f"{float(hs.input_rk):>12.1f}" + ' ' +
hs.name + '\n' hs.name + '\n'
) )
for bhs in hs.basic_structures: self._export_SIN_bhs(study, sin_dict, hs, f)
reach_id = study.river.get_edge_id(hs.input_reach) + 1
param_str = ' '.join(
[
f'{p:>10.3f}'
for p in self._export_SIN_parameters(bhs)
]
)
name = bhs.name
if name == "":
name = f"HS_{bhs.id:>3}".replace(" ", "0")
else:
name = name.replace(" ", "_")
f.write(
f"{sin_dict[bhs._type]} " +
f"{reach_id} {float(hs.input_kp):>12.3f} " +
f"{param_str} {name}\n"
)
return files return files
def _export_SIN_bhs(self, study, sin_dict, hs, f):
for bhs in hs.basic_structures:
if bhs.enabled:
reach_id = study.river.get_edge_id(hs.input_reach) + 1
param_str = ' '.join(
[
f'{p:>10.3f}'
for p in self._export_SIN_parameters(bhs)
]
)
name = bhs.name
if name == "":
name = f"HS_{bhs.id:>3}".replace(" ", "0")
else:
name = name.replace(" ", "_")
f.write(
f"{sin_dict[bhs._type]} " +
f"{reach_id} {float(hs.input_rk):>12.3f} " +
f"{param_str} {name}\n"
)
def _export_SIN_parameters(self, bhs): def _export_SIN_parameters(self, bhs):
res = [9999.999] * 5 res = [9999.999] * 5
@@ -574,7 +589,6 @@ class Mage(CommandLineSolver):
return res return res
@timer
def _export_VAR(self, study, repertory, qlog, name="0"): def _export_VAR(self, study, repertory, qlog, name="0"):
files = [] files = []
@@ -610,7 +624,6 @@ class Mage(CommandLineSolver):
return files return files
@timer
def _export_DEV(self, study, repertory, qlog, name="0"): def _export_DEV(self, study, repertory, qlog, name="0"):
files = [] files = []
@@ -633,7 +646,6 @@ class Mage(CommandLineSolver):
return files return files
@timer
def _export_REP(self, study, repertory, files, qlog, name="0"): def _export_REP(self, study, repertory, files, qlog, name="0"):
if qlog is not None: if qlog is not None:
qlog.put("Export REP file") qlog.put("Export REP file")
@@ -655,6 +667,7 @@ class Mage(CommandLineSolver):
f.write("* OUTPUT\n") f.write("* OUTPUT\n")
f.write(f"TRA {name}.TRA\n") f.write(f"TRA {name}.TRA\n")
f.write(f"BIN {name}.BIN\n") f.write(f"BIN {name}.BIN\n")
f.write(f"ERR {name}.ERR\n")
for file in files: for file in files:
EXT = file.split('.')[1] EXT = file.split('.')[1]
@@ -662,6 +675,8 @@ class Mage(CommandLineSolver):
if EXT in ["GRA"]: if EXT in ["GRA"]:
f.write(f"{EXT} {file}\n") f.write(f"{EXT} {file}\n")
self._export_REP_additional_lines(study, f)
@timer @timer
def export(self, study, repertory, qlog=None): def export(self, study, repertory, qlog=None):
self._study = study self._study = study
@@ -672,6 +687,8 @@ class Mage(CommandLineSolver):
self._bin_file = f"{name}.BIN" self._bin_file = f"{name}.BIN"
self._export_ST(study, repertory, qlog, name=name) self._export_ST(study, repertory, qlog, name=name)
self.export_additional_files(study, repertory, qlog, name=name)
self.export_study_description(study, repertory, qlog, name=name)
return True return True
@@ -763,7 +780,6 @@ class Mage8(Mage):
return lst return lst
@timer
def _export_PAR(self, study, repertory, qlog=None, name="0"): def _export_PAR(self, study, repertory, qlog=None, name="0"):
files = [] files = []
@@ -787,7 +803,7 @@ class Mage8(Mage):
value = "O" if value.lower() == "y" else "N" value = "O" if value.lower() == "y" else "N"
if name == "init_internal": if name == "init_internal":
value = ("p" if value.lower() in ["y", "yes", "true"] value = ("p" if value.lower() in ["y", "yes", "true", "o"]
else "") else "")
logger.debug( logger.debug(
@@ -798,7 +814,6 @@ class Mage8(Mage):
return files return files
@timer
def _export_NET(self, study, repertory, qlog=None, name="0"): def _export_NET(self, study, repertory, qlog=None, name="0"):
files = [] files = []
@@ -808,13 +823,7 @@ class Mage8(Mage):
with mage_file_open(os.path.join(repertory, f"{name}.NET"), "w+") as f: with mage_file_open(os.path.join(repertory, f"{name}.NET"), "w+") as f:
files.append(f"{name}.NET") files.append(f"{name}.NET")
edges = study.river.edges() edges = study.river.enable_edges()
edges = list(
filter(
lambda e: e.is_enable(),
edges
)
)
for e in edges: for e in edges:
name = f"Reach_{e.id + 1:>3}".replace(" ", "0") name = f"Reach_{e.id + 1:>3}".replace(" ", "0")
@@ -828,7 +837,6 @@ class Mage8(Mage):
return files return files
@timer
def _export_QSO(self, bounds, repertory, qlog, name="0"): def _export_QSO(self, bounds, repertory, qlog, name="0"):
files = [] files = []
@@ -864,6 +872,16 @@ class Mage8(Mage):
return files return files
def export_func_dict(self):
return [
self._export_ST, self._export_PAR,
self._export_NET, self._export_bound_cond,
self._export_RUG, self._export_INI,
self._export_SIN, self._export_VAR,
self._export_CAS, self._export_DEV,
self._export_lateral_contrib,
]
@timer @timer
def export(self, study, repertory, qlog=None, name="0"): def export(self, study, repertory, qlog=None, name="0"):
self._study = study self._study = study
@@ -876,20 +894,19 @@ class Mage8(Mage):
# Generate files # Generate files
files = [] files = []
files = self._export_ST(study, repertory, qlog, name=name) try:
files = files + self._export_PAR(study, repertory, qlog, name=name) for func in self.export_func_dict():
files = files + self._export_NET(study, repertory, qlog, name=name) files = files + func(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_SIN(study, repertory, qlog, name=name)
files = files + self._export_VAR(study, repertory, qlog, name=name)
files = files + self._export_CAS(study, repertory, qlog, name=name)
files = files + self._export_DEV(study, repertory, qlog, name=name)
self._export_REP(study, repertory, files, qlog, name=name)
return True self.export_additional_files(study, repertory, qlog, name=name)
self.export_study_description(study, repertory, qlog, name=name)
self._export_REP(study, repertory, files, qlog, name=name)
return True
except Exception as e:
logger.error(f"Failed to export study to {self._type}")
logger_exception(e)
return False
########### ###########
# RESULTS # # RESULTS #
@@ -1202,12 +1219,34 @@ class Mage8(Mage):
logger.info(f"read_gra: ... end with {len(ts)} timestamp read") logger.info(f"read_gra: ... end with {len(ts)} timestamp read")
@timer @timer
def results(self, study, repertory, qlog=None, name=None): def results(self, study, repertory,
qlog=None, name=None,
with_gra=True):
self._study = study self._study = study
if name is None: if name is None:
name = study.name.replace(" ", "_") name = study.name.replace(" ", "_")
results = super(Mage8, self).results(study, repertory, qlog, name=name) results = super(Mage8, self).results(study, repertory, qlog, name=name)
self.read_gra(study, repertory, results, qlog, name=name) if with_gra:
self.read_gra(study, repertory, results, qlog, name=name)
return results return results
class MageFake7(Mage8):
_type = "mage_fake7"
def __init__(self, name):
super(MageFake7, self).__init__(name)
self._type = "mage_fake7"
@timer
def results(self, study, repertory,
qlog=None, name=None,
with_gra=False):
results = super(MageFake7, self).results(
study, repertory,
qlog, name=name, with_gra=with_gra
)
return results
+441
View File
@@ -0,0 +1,441 @@
# RubarBE.py -- Pamhyr
# Copyright (C) 2024 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
import numpy as np
from tools import timer, trace, old_pamhyr_date_to_timestamp
from Solver.CommandLine import CommandLineSolver
from Model.Results.Results import Results
from Model.Results.River.River import River, Reach, Profile
logger = logging.getLogger()
class RubarBE(CommandLineSolver):
_type = "rubarbe"
def __init__(self, name):
super(RubarBE, self).__init__(name)
self._type = "rubarbe"
self._cmd_input = ""
self._cmd_solver = "@path @input -o @output"
self._cmd_output = ""
@classmethod
def default_parameters(cls):
lst = super(RubarBE, cls).default_parameters()
lst += [
("rubarbe_cfl", "0.50000E+00"),
("rubarbe_condam", "1"),
("rubarbe_condav", "3"),
("rubarbe_regime", "0"),
("rubarbe_iodev", "n"),
("rubarbe_iodebord", ""),
("rubarbe_iostockage", ""),
("rubarbe_iopdt", "y"),
("rubarbe_iovis", "n"),
("rubarbe_rep", "n"),
("rubarbe_tinit", "000:00:00:00"),
("rubarbe_tmax", "999:99:99:00"),
("rubarbe_tiopdt", "000:00:00:00"),
("rubarbe_dt", "3000.0"),
("rubarbe_ts", "999:99:99:00"),
("rubarbe_dtsauv", "999:99:99:00"),
("rubarbe_psave", "999:99:99:00"),
("rubarbe_fdeb1", "1"),
("rubarbe_fdeb2", "10"),
("rubarbe_fdeb3", "100"),
("rubarbe_tf_1", "y"),
("rubarbe_tf_2", "y"),
("rubarbe_tf_3", "y"),
("rubarbe_tf_4", "y"),
("rubarbe_tf_5", "y"),
("rubarbe_tf_6", "n"),
("rubarbe_trased", "y"),
("rubarbe_optfpc", "0"),
("rubarbe_ros", "2650.0"),
("rubarbe_dm", "0.1"),
("rubarbe_segma", "1.0"),
# Sediment parameters
("rubarbe_sediment_ros", "2650.0"),
("rubarbe_sediment_por", "0.4"),
("rubarbe_sediment_dcharg", "0.0"),
("rubarbe_sediment_halfa", "1.0"),
("rubarbe_sediment_mult_1", "1.0"),
("rubarbe_sediment_mult_2", ""),
("rubarbe_sediment_mult_3", ""),
("rubarbe_sediment_mult_4", ""),
("rubarbe_sediment_mult_5", ""),
("rubarbe_sediment_visc", "0.047"),
("rubarbe_sediment_opts", "6"),
("rubarbe_sediment_odchar", "0"),
("rubarbe_sediment_unisol", "1"),
("rubarbe_sediment_typdef", "3"),
("rubarbe_sediment_depot", "2"),
("rubarbe_sediment_choixc", "2"),
("rubarbe_sediment_option", "2"),
("rubarbe_sediment_capsol", "1"),
("rubarbe_sediment_bmiu", "0.85"),
("rubarbe_sediment_demix", "0"),
("rubarbe_sediment_defond", "1"),
("rubarbe_sediment_varcons", "1"),
("rubarbe_sediment_dchard", "0.0"),
("rubarbe_sediment_dchars", "0.0"),
]
return lst
@classmethod
def checkers(cls):
lst = [
]
return lst
##########
# Export #
##########
def cmd_args(self, study):
lst = super(RubarBE, self).cmd_args(study)
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"
def export(self, study, repertory, qlog=None):
self._study = study
name = study.name.replace(" ", "_")
self._export_donnee(study, repertory, qlog, name=name)
self._export_ts(study, repertory, qlog, name=name)
self._export_geomac_i(study, repertory, qlog, name=name)
self._export_mail(study, repertory, qlog, name=name)
self._export_tps(study, repertory, qlog, name=name)
self._export_stricklers(study, repertory, qlog, name=name)
def _export_donnee(self, study, repertory, qlog, name="0"):
if qlog is not None:
qlog.put("Export DONNEE file")
with open(
os.path.join(
repertory, f"donnee.{name}"
), "w+"
) as f:
params = filter(
lambda p: "rubarbe_sediment_" not in p.name,
study.river.get_params(self._type).parameters
)
it = iter(params)
line = 0
while line < 29:
param = next(it)
name = param.name
value = param.value
if value != "":
# Value format
if value.count(':') == 3:
value = old_pamhyr_date_to_timestamp(value)
value = f"{value:>12.5e}".upper()
if value.count('.') == 1:
value = f"{float(value):>12.5e}".upper()
if value == "y" or value == "n":
value = "O" if value == "y" else "N"
# Write value
f.write(f"{name:<50}{value}")
# Add values of 'rubarbe_iodebord' and
# 'rubarbe_iostockage'
if name == "rubarbe_iodev":
v2 = next(it).value
v3 = next(it).value
f.write(f"{v2}{v3}")
# New line
f.write(f"\n")
line += 1
def _export_ts(self, study, repertory, qlog, name="0"):
if qlog is not None:
qlog.put("Export TS file")
with open(
os.path.join(
repertory, f"ts.{name}"
), "w+"
) as f:
def float_format(string):
if "." in string:
return f"{float(string):>10.0f}"
return ""
params = filter(
lambda p: "rubarbe_sediment_" in p.name,
study.river.get_params(self.type).parameters
)
it = iter(params)
line = 0
while line < 20:
param = next(it)
name = param.name
value = param.value
if value != "":
# Value format
if value.count('.') == 1:
value = f"{float_format(value)}"
else:
value = f"{value:>10}"
# Write value
f.write(f"{name:<50}{value}")
# Add values of 'rubarbe_iodebord' and
# 'rubarbe_iostockage'
if name == "rubarbe_sediment_mult_1":
m2 = f"{float_format(next(it).value)}"
m3 = f"{float_format(next(it).value)}"
m4 = f"{float_format(next(it).value)}"
m5 = f"{float_format(next(it).value)}"
f.write(f"{m2}{m3}{m4}{m5}")
# New line
f.write(f"\n")
line += 1
def _export_geomac_i(self, study, repertory, qlog, name="0"):
if qlog is not None:
qlog.put("Export GEOMAC-i file")
with open(
os.path.join(
repertory, f"geomac-i.{name}"
), "w+"
) as f:
for edge in study.river.enable_edges():
reach = edge.reach
n_profiles = len(reach)
time = 0.0
f.write(f"{n_profiles:>5} {time:>11.3f}\n")
ind = 1
for profile in reach.profiles:
rk = profile.rk
n_points = len(profile)
f.write(f"{ind:>4} {rk:>11.3f} {n_points:>4}\n")
for point in profile.points:
label = point.name.lower()
if label != "":
if label[0] == "r":
label = label[1].upper()
else:
label = lable[0]
y = point.y
z = point.z
dcs = 0.001
scs = 1.0
tmcs = 0.0
f.write(
f"{label} {y:>11.5f}" +
f"{z:>13.5f}{dcs:>15.10f}" +
f"{scs:>15.10f}{tmcs:>15.5f}" +
"\n"
)
ind += 1
def _export_mail(self, study, repertory, qlog, name="0"):
if qlog is not None:
qlog.put("Export MAIL file")
with open(
os.path.join(
repertory, f"mail.{name}"
), "w+"
) as f:
for edge in study.river.enable_edges():
reach = edge.reach
lm = len(reach) + 1
f.write(f"{lm:>13}\n")
for mails in [reach.inter_profiles_rk(),
reach.get_rk()]:
ind = 0
for mail in mails:
f.write(f"{mail:15.3f}")
ind += 1
if ind % 3 == 0:
f.write("\n")
if ind % 3 != 0:
f.write("\n")
def _export_stricklers(self, study, repertory, qlog, name="0"):
self._export_frot(study, repertory, qlog, name=name, version="")
self._export_frot(study, repertory, qlog, name=name, version="2")
def _export_frot(self, study, repertory, qlog, name="0", version=""):
if qlog is not None:
qlog.put(f"Export FROT{version} file")
with open(
os.path.join(
repertory, f"frot{version}.{name}"
), "w+"
) as f:
for edge in study.river.enable_edges():
reach = edge.reach
lm = len(reach) + 1
f.write(f"{lm:>6}\n")
def get_stricklers_from_rk(rk):
return next(
map(
lambda s: (
s.begin_strickler.medium if version == "2"
else s.begin_strickler.minor
),
filter(
lambda f: rk in f,
edge.frictions.lst
)
)
)
ind = 1
for mail in edge.reach.inter_profiles_rk():
coef = get_stricklers_from_rk(mail)
f.write(f"{ind:>6} {coef:>12.5f}")
ind += 1
f.write("\n")
def _export_tps(self, study, repertory, qlog, name="0"):
if qlog is not None:
qlog.put("Export TPS file")
with open(
os.path.join(
repertory, f"tps.{name}"
), "w+"
) as f:
for edge in study.river.enable_edges():
reach = edge.reach
f.write(f"0.0\n")
ics = study.river.initial_conditions.get(edge)
data = self._export_tps_init_data(ics)
profiles = reach.profiles
first = profiles[0]
last = profiles[-1]
if first.rk not in data or last.rk not in data:
logger.error(
"Study initial condition is not fully defined"
)
return
f_h_s = self._export_tps_profile_height_speed(first, data)
l_h_s = self._export_tps_profile_height_speed(last, data)
# First mail
f.write(f"{1:>5} {f_h_s[0]} {f_h_s[1]}")
ind = 2
it = iter(profiles)
prev = next(it)
prev_h, prev_s = f_h_s
for profile in it:
if profile.rk not in data:
ind += 1
continue
cur_h, cur_s = self._export_tps_profile_height_speed(
profile, data
)
# Mean of height and speed
h = (prev_h + cur_h) / 2
s = (prev_s + cur_s) / 2
f.write(f"{ind:>5} {h} {s}\n")
prev_h, prev_s = cur_h, cur_s
ind += 1
# Last mail
f.write(f"{ind:>5} {f_h_s[0]} {f_h_s[1]}")
def _export_tps_init_data(self, ics):
data = {}
for d in ics.data:
data[d['rk']] = (
d['elevation'],
d['discharge'],
)
return data
def _export_tps_profile_height_speed(self, profile, data):
z = data[profile.rk][0]
q = data[profile.rk][1]
height = z - profile.z_min()
speed = profile.speed(q, z)
return height, speed
+8 -1
View File
@@ -19,7 +19,10 @@
from PyQt5.QtCore import QCoreApplication from PyQt5.QtCore import QCoreApplication
from Solver.GenericSolver import GenericSolver from Solver.GenericSolver import GenericSolver
from Solver.Mage import Mage7, Mage8 from Solver.Mage import (
Mage7, Mage8, MageFake7,
)
from Solver.RubarBE import RubarBE
_translate = QCoreApplication.translate _translate = QCoreApplication.translate
@@ -27,10 +30,14 @@ solver_long_name = {
# "generic": "Generic", # "generic": "Generic",
# "mage7": "Mage v7", # "mage7": "Mage v7",
"mage8": "Mage v8", "mage8": "Mage v8",
# "mage_fake7": "Mage fake v7",
# "rubarbe": "RubarBE",
} }
solver_type_list = { solver_type_list = {
# "generic": GenericSolver, # "generic": GenericSolver,
# "mage7": Mage7, # "mage7": Mage7,
"mage8": Mage8, "mage8": Mage8,
# "mage_fake7": MageFake7,
# "rubarbe": RubarBE,
} }
+1 -1
View File
@@ -67,5 +67,5 @@ class AboutWindow(PamhyrDialog):
label = f"\n - {author}" + label label = f"\n - {author}" + label
except StopIteration: except StopIteration:
label = _translate("About", "Contributors: ") + label label = _translate("About", "Contributors: ") + label
label = "Copyright © 2023-2024 INRAE\n" + label label = "Copyright © 2022-2024 INRAE\n" + label
self.set_label_text("label_copyright", label) self.set_label_text("label_copyright", label)
+90
View File
@@ -0,0 +1,90 @@
# Window.py -- Pamhyr
# Copyright (C) 2024 INRAE
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
# -*- coding: utf-8 -*-
import logging
from Modules import Modules
from View.Tools.PamhyrWindow import PamhyrWindow
from PyQt5.QtWidgets import (
QLabel, QPlainTextEdit, QPushButton,
QCheckBox,
)
from View.AdditionalFiles.Translate import AddFileTranslate
from View.AdditionalFiles.UndoCommand import (
SetCommand
)
logger = logging.getLogger()
class EditAddFileWindow(PamhyrWindow):
_pamhyr_ui = "AdditionalFile"
_pamhyr_name = "Edit additional file"
def __init__(self, study=None, config=None, add_file=None,
trad=None, undo=None, parent=None):
name = trad[self._pamhyr_name] + " - " + study.name
super(EditAddFileWindow, self).__init__(
title=name,
study=study,
config=config,
options=[],
parent=parent
)
self._add_file = add_file
self._hash_data.append(self._add_file)
self._undo = undo
self.setup_values()
self.setup_connection()
def setup_values(self):
self.set_check_box("checkBox", self._add_file.enabled)
self.set_line_edit_text("lineEdit_name", self._add_file.name)
self.set_line_edit_text("lineEdit_path", self._add_file.path)
self.set_plaintext_edit_text("plainTextEdit", self._add_file.text)
def setup_connection(self):
self.find(QPushButton, "pushButton_cancel")\
.clicked.connect(self.close)
self.find(QPushButton, "pushButton_ok")\
.clicked.connect(self.accept)
def accept(self):
is_enabled = self.get_check_box("checkBox")
name = self.get_line_edit_text("lineEdit_name")
path = self.get_line_edit_text("lineEdit_path")
text = self.get_plaintext_edit_text("plainTextEdit")
self._undo.push(
SetCommand(
self._add_file,
enabled=is_enabled,
name=name,
path=path,
text=text,
)
)
self._propagate_update(key=Modules.ADDITIONAL_FILES)
self.close()
+81
View File
@@ -0,0 +1,81 @@
# List.py -- Pamhyr
# Copyright (C) 2024 INRAE
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
# -*- coding: utf-8 -*-
import logging
from functools import reduce
from tools import trace, timer
from PyQt5.QtCore import (
Qt, QVariant,
)
from PyQt5.QtGui import (
QColor, QBrush,
)
from View.Tools.PamhyrList import PamhyrListModel
from View.AdditionalFiles.UndoCommand import (
AddCommand, DelCommand
)
logger = logging.getLogger()
class ListModel(PamhyrListModel):
def data(self, index, role):
row = index.row()
column = index.column()
file = self._data.files[row]
if role == Qt.ForegroundRole:
color = Qt.gray
if file.is_enabled():
color = QColor("black")
else:
color = QColor("grey")
return QBrush(color)
if role == Qt.ItemDataRole.DisplayRole:
text = f"{file.name}: '{file.path}'"
if not file.is_enabled():
text += " (disabled)"
return text
return QVariant()
def add(self, row):
self._undo.push(
AddCommand(
self._data, row
)
)
self.update()
def delete(self, row):
self._undo.push(
DelCommand(
self._data, row
)
)
self.update()
+35
View File
@@ -0,0 +1,35 @@
# Translate.py -- Pamhyr
# Copyright (C) 2024 INRAE
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
# -*- coding: utf-8 -*-
from PyQt5.QtCore import QCoreApplication
from View.Translate import MainTranslate
_translate = QCoreApplication.translate
class AddFileTranslate(MainTranslate):
def __init__(self):
super(AddFileTranslate, self).__init__()
self._dict["Additional files"] = _translate(
"AdditionalFiles", "Additional files"
)
self._dict["Edit additional file"] = _translate(
"AdditionalFiles", "Edit additional file"
)
+82
View File
@@ -0,0 +1,82 @@
# UndoCommand.py -- Pamhyr
# Copyright (C) 2024 INRAE
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
# -*- coding: utf-8 -*-
from tools import trace, timer
from PyQt5.QtWidgets import (
QMessageBox, QUndoCommand, QUndoStack,
)
class SetCommand(QUndoCommand):
def __init__(self, add_file, **kwargs):
QUndoCommand.__init__(self)
self._add_file = add_file
self._new = kwargs
self._old = None
def undo(self):
f = self._add_file
for key in self._old:
f[key] = self._old[key]
def redo(self):
f = self._add_file
if self._old is None:
self._old = {}
for key in self._new:
self._old[key] = f[key]
for key in self._new:
f[key] = self._new[key]
class AddCommand(QUndoCommand):
def __init__(self, files, row):
QUndoCommand.__init__(self)
self._files = files
self._row = row
self._new = None
def undo(self):
self._files.delete([self._new])
def redo(self):
if self._new is None:
self._new = self._files.new(self._row)
else:
self._files.insert(self._row, self._new)
class DelCommand(QUndoCommand):
def __init__(self, files, row):
QUndoCommand.__init__(self)
self._files = files
self._row = row
self._old = self._files.get(row)
def undo(self):
self._files.insert(self._row, self._old)
def redo(self):
self._files.delete_i([self._row])
+118
View File
@@ -0,0 +1,118 @@
# Window.py -- Pamhyr
# Copyright (C) 2024 INRAE
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
# -*- coding: utf-8 -*-
from tools import trace, timer
from PyQt5.QtWidgets import (
QAction, QListView,
)
from View.Tools.PamhyrWindow import PamhyrWindow
from View.AdditionalFiles.List import ListModel
from View.AdditionalFiles.Translate import AddFileTranslate
from View.AdditionalFiles.Edit.Window import EditAddFileWindow
class AddFileListWindow(PamhyrWindow):
_pamhyr_ui = "AdditionalFileList"
_pamhyr_name = "Additional files"
def __init__(self, study=None, config=None,
parent=None):
trad = AddFileTranslate()
name = trad[self._pamhyr_name] + " - " + study.name
super(AddFileListWindow, self).__init__(
title=name,
study=study,
config=config,
trad=trad,
options=[],
parent=parent
)
self.setup_list()
self.setup_connections()
def setup_list(self):
lst = self.find(QListView, f"listView")
self._list = ListModel(
list_view=lst,
data=self._study.river.additional_files,
undo=self._undo_stack,
trad=self._trad,
)
def setup_connections(self):
self.find(QAction, "action_add").triggered.connect(self.add)
self.find(QAction, "action_delete").triggered.connect(self.delete)
self.find(QAction, "action_edit").triggered.connect(self.edit)
def update(self):
self._list.update()
def selected_rows(self):
lst = self.find(QListView, f"listView")
return list(map(lambda i: i.row(), lst.selectedIndexes()))
def add(self):
rows = self.selected_rows()
if len(rows) > 0:
row = rows[0]
else:
row = 0
self._list.add(row)
def delete(self):
rows = self.selected_rows()
if len(rows) == 0:
return
self._list.delete(rows[0])
def edit(self):
rows = self.selected_rows()
for row in rows:
add_file = self._study.river.additional_files.files[row]
if self.sub_window_exists(
EditAddFileWindow,
data=[self._study, self._config, add_file]
):
continue
win = EditAddFileWindow(
study=self._study,
config=self._config,
add_file=add_file,
trad=self._trad,
undo=self._undo_stack,
parent=self,
)
win.show()
def _undo(self):
self._undo_stack.undo()
self.update()
def _redo(self):
self._undo_stack.redo()
self.update()
+14 -5
View File
@@ -99,7 +99,7 @@ class TableModel(PamhyrTableModel):
logger.info(e) logger.info(e)
logger.debug(traceback.format_exc()) logger.debug(traceback.format_exc())
self.dataChanged.emit(index, index) self.update()
return True return True
def add(self, row, parent=QModelIndex()): def add(self, row, parent=QModelIndex()):
@@ -112,7 +112,7 @@ class TableModel(PamhyrTableModel):
) )
self.endInsertRows() self.endInsertRows()
self.layoutChanged.emit() self.update()
def delete(self, rows, parent=QModelIndex()): def delete(self, rows, parent=QModelIndex()):
self.beginRemoveRows(parent, rows[0], rows[-1]) self.beginRemoveRows(parent, rows[0], rows[-1])
@@ -135,7 +135,7 @@ class TableModel(PamhyrTableModel):
) )
self.layoutAboutToBeChanged.emit() self.layoutAboutToBeChanged.emit()
self.layoutChanged.emit() self.update()
def move_up(self, row, parent=QModelIndex()): def move_up(self, row, parent=QModelIndex()):
if row <= 0: if row <= 0:
@@ -152,7 +152,7 @@ class TableModel(PamhyrTableModel):
) )
self.endMoveRows() self.endMoveRows()
self.layoutChanged.emit() self.update()
def move_down(self, index, parent=QModelIndex()): def move_down(self, index, parent=QModelIndex()):
if row > len(self._data): if row > len(self._data):
@@ -169,7 +169,7 @@ class TableModel(PamhyrTableModel):
) )
self.endMoveRows() self.endMoveRows()
self.layoutChanged.emit() self.update()
def paste(self, row, header, data): def paste(self, row, header, data):
if len(data) == 0: if len(data) == 0:
@@ -190,4 +190,13 @@ class TableModel(PamhyrTableModel):
) )
self.layoutAboutToBeChanged.emit() self.layoutAboutToBeChanged.emit()
self.update()
def auto_sort(self):
self.layoutAboutToBeChanged.emit()
self._data.sort(key=lambda x: x[0])
self.layoutAboutToBeChanged.emit()
def update(self):
# self.auto_sort()
self.layoutChanged.emit() self.layoutChanged.emit()
+2 -1
View File
@@ -108,7 +108,7 @@ class EditBoundaryConditionWindow(PamhyrWindow):
name = trad[self._pamhyr_name] name = trad[self._pamhyr_name]
if self._data is not None: if self._data is not None:
node_name = (self._data.node.name if self._data.node is not None node_name = (self._data.node.name if self._data.node is not None
else _translate("BoundaryCondition", "Not associate")) else trad['not_associated'])
name += ( name += (
f" - {study.name} " + f" - {study.name} " +
f" - {self._data.name} ({self._data.id}) " + f" - {self._data.name} ({self._data.id}) " +
@@ -195,6 +195,7 @@ class EditBoundaryConditionWindow(PamhyrWindow):
self.find(QAction, "action_sort").triggered.connect(self.sort) self.find(QAction, "action_sort").triggered.connect(self.sort)
self._table.dataChanged.connect(self.update) self._table.dataChanged.connect(self.update)
self._table.layoutChanged.connect(self.update)
if self._is_solid: if self._is_solid:
self._d50sigma.d50Changed.connect(self.d50_changed) self._d50sigma.d50Changed.connect(self.d50_changed)
+5 -5
View File
@@ -18,7 +18,7 @@
from PyQt5.QtCore import QCoreApplication from PyQt5.QtCore import QCoreApplication
from View.Tools.PamhyrTranslate import PamhyrTranslate from View.Translate import MainTranslate
from View.BoundaryCondition.translate import BCTranslate from View.BoundaryCondition.translate import BCTranslate
@@ -36,9 +36,9 @@ class BCETranslate(BCTranslate):
self._sub_dict["table_headers"] = { self._sub_dict["table_headers"] = {
"x": _translate("BoundaryCondition", "X"), "x": _translate("BoundaryCondition", "X"),
"y": _translate("BoundaryCondition", "Y"), "y": _translate("BoundaryCondition", "Y"),
"time": _translate("BoundaryCondition", "Time"), "time": self._dict["time"],
"date": _translate("BoundaryCondition", "Date"), "date": self._dict["date"],
"discharge": _translate("BoundaryCondition", "Discharge (m³/s)"), "discharge": self._dict["unit_discharge"],
"z": _translate("BoundaryCondition", "Z (m)"), "z": self._dict["unit_elevation"],
"solid": _translate("BoundaryCondition", "Solid (kg/s)"), "solid": _translate("BoundaryCondition", "Solid (kg/s)"),
} }
+2 -2
View File
@@ -85,7 +85,7 @@ class ComboBoxDelegate(QItemDelegate):
) )
else: else:
self.editor.addItems( self.editor.addItems(
[_translate("BoundaryCondition", "Not associate")] + [self._trad["not_associated"]] +
self._data.nodes_names() self._data.nodes_names()
) )
@@ -145,7 +145,7 @@ class TableModel(PamhyrTableModel):
elif self._headers[column] == "node": elif self._headers[column] == "node":
n = self._lst.get(self._tab, row).node n = self._lst.get(self._tab, row).node
if n is None: if n is None:
return _translate("BoundaryCondition", "Not associate") return self._trad["not_associated"]
return n.name return n.name
return QVariant() return QVariant()
+9 -1
View File
@@ -18,7 +18,7 @@
import logging import logging
from tools import trace, timer from tools import trace, timer, logger_exception
from View.Tools.PamhyrWindow import PamhyrWindow from View.Tools.PamhyrWindow import PamhyrWindow
@@ -37,6 +37,7 @@ from PyQt5.QtWidgets import (
QFileDialog, QTableView, QAbstractItemView, QFileDialog, QTableView, QAbstractItemView,
QUndoStack, QShortcut, QAction, QItemDelegate, QUndoStack, QShortcut, QAction, QItemDelegate,
QComboBox, QVBoxLayout, QHeaderView, QTabWidget, QComboBox, QVBoxLayout, QHeaderView, QTabWidget,
QWidget,
) )
from Model.BoundaryCondition.BoundaryConditionTypes import ( from Model.BoundaryCondition.BoundaryConditionTypes import (
@@ -170,6 +171,13 @@ class BoundaryConditionWindow(PamhyrWindow):
) )
) )
def set_active_tab(self, tab_id=0):
try:
tab_widget = self.find(QTabWidget, "tabWidget")
tab_widget.setCurrentIndex(tab_id)
except Exception as e:
logger_exception(e)
def add(self): def add(self):
tab = self.current_tab() tab = self.current_tab()
rows = self.index_selected_rows() rows = self.index_selected_rows()
+6 -6
View File
@@ -18,7 +18,7 @@
from PyQt5.QtCore import QCoreApplication from PyQt5.QtCore import QCoreApplication
from View.Tools.PamhyrTranslate import PamhyrTranslate from View.Translate import MainTranslate
from Model.BoundaryCondition.BoundaryConditionTypes import ( from Model.BoundaryCondition.BoundaryConditionTypes import (
NotDefined, PonctualContribution, NotDefined, PonctualContribution,
@@ -38,7 +38,7 @@ BC_types = {
} }
class BCTranslate(PamhyrTranslate): class BCTranslate(MainTranslate):
def __init__(self): def __init__(self):
super(BCTranslate, self).__init__() super(BCTranslate, self).__init__()
@@ -47,8 +47,8 @@ class BCTranslate(PamhyrTranslate):
) )
self._sub_dict["long_types"] = { self._sub_dict["long_types"] = {
"ND": _translate("BoundaryCondition", "Not defined"), "ND": self._dict["not_defined"],
"PC": _translate("BoundaryCondition", "Ponctual contribution"), "PC": _translate("BoundaryCondition", "Point sources"),
"TZ": _translate("BoundaryCondition", "Z(t)"), "TZ": _translate("BoundaryCondition", "Z(t)"),
"TD": _translate("BoundaryCondition", "Q(t)"), "TD": _translate("BoundaryCondition", "Q(t)"),
"ZD": _translate("BoundaryCondition", "Q(Z)"), "ZD": _translate("BoundaryCondition", "Q(Z)"),
@@ -56,7 +56,7 @@ class BCTranslate(PamhyrTranslate):
} }
self._sub_dict["table_headers"] = { self._sub_dict["table_headers"] = {
"name": _translate("BoundaryCondition", "Name"), "name": self._dict["name"],
"type": _translate("BoundaryCondition", "Type"), "type": self._dict["type"],
"node": _translate("BoundaryCondition", "Node") "node": _translate("BoundaryCondition", "Node")
} }
+67
View File
@@ -0,0 +1,67 @@
# List.py -- Pamhyr
# Copyright (C) 2024 INRAE
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
# -*- coding: utf-8 -*-
import logging
from functools import reduce
from tools import trace, timer
from PyQt5.QtCore import (
Qt, QVariant,
)
from PyQt5.QtGui import (
QColor, QBrush,
)
from Modules import Modules
from Checker.Checker import STATUS
from View.Tools.PamhyrList import PamhyrListModel
logger = logging.getLogger()
class TabListModel(PamhyrListModel):
def data(self, index, role):
row = index.row()
column = index.column()
checker = self._data[row]
if role == Qt.ForegroundRole:
color = Qt.gray
status = checker._status
if status is STATUS.OK:
color = Qt.green
elif status is STATUS.WARNING:
color = QColor("orange")
elif status is STATUS.ERROR:
color = Qt.red
return QBrush(color)
if role == Qt.ItemDataRole.DisplayRole:
msg = checker.name
if checker.is_error() or checker.is_warning():
msg += f": {checker.summary}"
return msg
return QVariant()
+96 -2
View File
@@ -16,6 +16,9 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import logging
from functools import reduce
from tools import trace, timer from tools import trace, timer
from PyQt5.QtCore import ( from PyQt5.QtCore import (
@@ -35,9 +38,11 @@ from PyQt5.QtWidgets import (
QComboBox, QComboBox,
) )
from Modules import Modules
from Checker.Checker import STATUS
from View.Tools.PamhyrTable import PamhyrTableModel from View.Tools.PamhyrTable import PamhyrTableModel
_translate = QCoreApplication.translate logger = logging.getLogger()
class TableModel(PamhyrTableModel): class TableModel(PamhyrTableModel):
@@ -52,7 +57,7 @@ class TableModel(PamhyrTableModel):
if self._data[row].is_ok(): if self._data[row].is_ok():
color = Qt.green color = Qt.green
elif self._data[row].is_warning(): elif self._data[row].is_warning():
color = Qt.yellow color = QColor("orange")
elif self._data[row].is_error(): elif self._data[row].is_error():
color = Qt.red color = Qt.red
@@ -68,3 +73,92 @@ class TableModel(PamhyrTableModel):
return self._data[row].summary return self._data[row].summary
return QVariant() return QVariant()
class TabTableModel(PamhyrTableModel):
def _setup_lst(self):
self._lst = self._opt_data
def data(self, index, role):
row = index.row()
column = index.column()
if role == Qt.ForegroundRole:
if self._headers[column] == "type":
return QVariant()
color = Qt.gray
status, _ = self.compute_status(row, column)
if status is STATUS.OK:
color = Qt.green
elif status is STATUS.WARNING:
color = QColor("orange")
elif status is STATUS.ERROR:
color = Qt.red
return QBrush(color)
if role == Qt.ItemDataRole.DisplayRole:
if self._headers[column] == "type":
return self.module_display_name(self._opt_data[row])
value = "UNKNOWN"
status, _ = self.compute_status(row, column)
if status is STATUS.OK:
value = "OK"
elif status is STATUS.WARNING:
value = "WARNING"
elif status is STATUS.ERROR:
value = "ERROR"
return value
return QVariant()
def module_display_name(self, module):
return Modules.modelling_display_name()[module]
def get_checkers(self, row, column):
module = self._opt_data[row]
solver = self._headers[column]
checkers = filter(
lambda checker: checker._modules is module,
filter(
lambda checker: checker._solver is solver,
self._data
)
)
return checkers
def get_checkers_from_indexes(self, indexes):
lst = []
for index in indexes:
row = index.row()
column = index.column()
lst += list(self.get_checkers(row, column))
return lst
def compute_status(self, row, column):
checkers = self.get_checkers(row, column)
checkers_status = list(
map(
lambda c: c._status,
checkers
)
)
status = STATUS.UNKNOWN
if len(checkers_status) > 0:
status = reduce(lambda acc, c: acc | c, checkers_status)
logger.debug(f"Checkers: {row}, {column}: {checkers_status} {status}")
return status, checkers_status
+3 -3
View File
@@ -18,12 +18,12 @@
from PyQt5.QtCore import QCoreApplication from PyQt5.QtCore import QCoreApplication
from View.Tools.PamhyrTranslate import PamhyrTranslate from View.Translate import MainTranslate
_translate = QCoreApplication.translate _translate = QCoreApplication.translate
class CheckListTranslate(PamhyrTranslate): class CheckListTranslate(MainTranslate):
def __init__(self): def __init__(self):
super(CheckListTranslate, self).__init__() super(CheckListTranslate, self).__init__()
@@ -32,6 +32,6 @@ class CheckListTranslate(PamhyrTranslate):
) )
self._sub_dict["table_headers"] = { self._sub_dict["table_headers"] = {
"name": _translate("CheckList", "Name"), "name": self._dict["name"],
"status": _translate("CheckList", "Status"), "status": _translate("CheckList", "Status"),
} }
+36 -1
View File
@@ -17,11 +17,14 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import time import time
import logging
from PyQt5.QtCore import ( from PyQt5.QtCore import (
Qt, QObject, pyqtSlot, pyqtSignal Qt, QObject, pyqtSlot, pyqtSignal
) )
logger = logging.getLogger()
class Worker(QObject): class Worker(QObject):
signalStatus = pyqtSignal(str) signalStatus = pyqtSignal(str)
@@ -43,6 +46,38 @@ class Worker(QObject):
self.signalStatus.emit(checker.name) self.signalStatus.emit(checker.name)
# Run checker # Run checker
checker._run(self._study) checker.run(self._study)
self.signalStatus.emit("progress") self.signalStatus.emit("progress")
class TabWorker(QObject):
signalStatus = pyqtSignal(str)
def __init__(self, study, checker_q, parent=None):
super(self.__class__, self).__init__(parent)
self._study = study
self._checker_q = checker_q
@property
def study(self):
return self._study
@study.setter
def study(self, study):
self._study = study
def process(self):
self._compute()
def _compute(self):
while True:
checker = self._checker_q.get()
if self._study is None:
self._checker_q.put(checker)
time.sleep(5)
else:
checker.run(self._study)
self.signalStatus.emit("progress")
+5 -5
View File
@@ -18,12 +18,12 @@
from PyQt5.QtCore import QCoreApplication from PyQt5.QtCore import QCoreApplication
from View.Tools.PamhyrTranslate import PamhyrTranslate from View.Translate import MainTranslate
_translate = QCoreApplication.translate _translate = QCoreApplication.translate
class ConfigureTranslate(PamhyrTranslate): class ConfigureTranslate(MainTranslate):
def __init__(self): def __init__(self):
super(ConfigureTranslate, self).__init__() super(ConfigureTranslate, self).__init__()
@@ -32,9 +32,9 @@ class ConfigureTranslate(PamhyrTranslate):
) )
self._sub_dict["table_headers"] = { self._sub_dict["table_headers"] = {
"name": _translate("Configure", "Name"), "name": self._dict["name"],
"type": _translate("Configure", "Type"), "type": self._dict["type"],
"description": _translate("Configure", "Description"), "description": self._dict["description"],
} }
# Solvers # Solvers
+18 -14
View File
@@ -45,6 +45,7 @@ from PyQt5.QtWidgets import (
QComboBox, QUndoStack, QShortcut, QHeaderView, QComboBox, QUndoStack, QShortcut, QHeaderView,
) )
from Modules import Modules
from Solver.Solvers import solver_long_name from Solver.Solvers import solver_long_name
logger = logging.getLogger() logger = logging.getLogger()
@@ -123,11 +124,11 @@ class ConfigureWindow(PamhyrDialog):
def setup_data(self): def setup_data(self):
# Backup # Backup
self.set_check_box("checkBox_backup", self._config.backup_enable) self.set_check_box("checkBox_backup", self._config.backup_enable)
self.set_line_edit_text("lineEdit_backup_path", # self.set_line_edit_text("lineEdit_backup_path",
self._config.backup_path) # self._config.backup_path)
self.set_time_edit("timeEdit_backup_frequence", self.set_time_edit("timeEdit_backup_frequence",
self._config.backup_frequence) self._config.backup_frequence)
self.set_spin_box("spinBox_backup_max", self._config.backup_max) # self.set_spin_box("spinBox_backup_max", self._config.backup_max)
# Editor # Editor
self.set_line_edit_text("lineEdit_editor_cmd", self.set_line_edit_text("lineEdit_editor_cmd",
@@ -155,12 +156,12 @@ class ConfigureWindow(PamhyrDialog):
"pushButton_stricklers_del": self.del_stricklers, "pushButton_stricklers_del": self.del_stricklers,
"pushButton_stricklers_sort": self.sort_stricklers, "pushButton_stricklers_sort": self.sort_stricklers,
# Others # Others
"pushButton_backup_path": lambda: self.file_dialog( # "pushButton_backup_path": lambda: self.file_dialog(
select_file=False, # select_file=False,
callback=lambda f: self.set_line_edit_text( # callback=lambda f: self.set_line_edit_text(
"lineEdit_backup_path", f[0] # "lineEdit_backup_path", f[0]
) # )
), # ),
} }
for button in buttons: for button in buttons:
@@ -172,11 +173,12 @@ class ConfigureWindow(PamhyrDialog):
# Backup # Backup
self._config.backup_enable = self.get_check_box("checkBox_backup") self._config.backup_enable = self.get_check_box("checkBox_backup")
self._config.backup_path = self.get_line_edit_text( # self._config.backup_path = self.get_line_edit_text(
"lineEdit_backup_path") # "lineEdit_backup_path")
self._config.backup_frequence = self.get_time_edit( self._config.backup_frequence = self.get_time_edit(
"timeEdit_backup_frequence") "timeEdit_backup_frequence"
self._config.backup_max = self.get_spin_box("spinBox_backup_max") )
# self._config.backup_max = self.get_spin_box("spinBox_backup_max")
# Stricklers # Stricklers
self._config.stricklers = deepcopy(self._stricklers) self._config.stricklers = deepcopy(self._stricklers)
@@ -188,6 +190,9 @@ class ConfigureWindow(PamhyrDialog):
self._config.lang = Config.languages( self._config.lang = Config.languages(
)[self.get_combobox_text("comboBox_language")] )[self.get_combobox_text("comboBox_language")]
self._propagate_update(key=Modules.CONFIG)
self._config.save()
self.end() self.end()
def reject(self): def reject(self):
@@ -195,7 +200,6 @@ class ConfigureWindow(PamhyrDialog):
self.end() self.end()
def end(self): def end(self):
self._config.save()
self.close() self.close()
# Debug # Debug
-20
View File
@@ -77,30 +77,10 @@ class ReplWindow(PamhyrWindow):
layout.insertWidget(0, self._editor) layout.insertWidget(0, self._editor)
def setup_connections(self): def setup_connections(self):
self._hup_sc = QShortcut(QKeySequence("ctrl+Up"), self)
self._hdown_sc = QShortcut(QKeySequence("ctrl+Down"), self)
self._hup_sc.activated.connect(self.history_up)
self._hdown_sc.activated.connect(self.history_down)
self.find(QPushButton, "pushButton").clicked.connect(self.eval_python) self.find(QPushButton, "pushButton").clicked.connect(self.eval_python)
def history_up(self):
if self._history_ind < len(self._history):
self._history_ind += 1
self._editor.setText(self._history[-self._history_ind])
def history_down(self):
if self._history_ind > 0:
self._history_ind -= 1
self._editor.setText(self._history[-self._history_ind])
else:
self._editor.setText("")
def eval_python(self): def eval_python(self):
code = self._editor.text() code = self._editor.text()
self._history.append(code)
self._editor.setText("")
self._history_ind = 0
# Code to rich_code # Code to rich_code
rich_code = code.strip().split("\n") rich_code = code.strip().split("\n")
+1 -1
View File
@@ -50,7 +50,7 @@ class DocWindow(PamhyrWindow):
return os.path.abspath( return os.path.abspath(
os.path.join( os.path.join(
os.path.dirname(__file__), os.path.dirname(__file__),
"..", "..", "..", "..", "doc", filename "..", "..", "..", "doc", filename
) )
) )
+50
View File
@@ -0,0 +1,50 @@
# PlotRKC.py -- Pamhyr
# Copyright (C) 2023-2024 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.HydraulicStructures.PlotRKC import PlotRKC
_translate = QCoreApplication.translate
class PlotRKZ(PlotRKC):
def __init__(self, canvas=None, trad=None, toolbar=None,
river=None, reach=None, profile=None,
parent=None):
super(PlotRKC, self).__init__(
canvas=canvas,
trad=trad,
data=river,
toolbar=toolbar,
parent=parent
)
self._current_reach = reach
self._current_profile = profile
self.label_x = self._trad["unit_rk"]
self.label_y = self._trad["unit_elevation"]
self._isometric_axis = False
self._auto_relim_update = True
self._autoscale_update = True
self.parent = parent
def onpick(self, event):
return
+16 -16
View File
@@ -41,10 +41,10 @@ class PlotStricklers(PamhyrPlot):
parent=parent parent=parent
) )
self.label_x = self._trad["kp"] self.label_x = self._trad["rk"]
self.label_y = self._trad["stricklers"] self.label_y = self._trad["strickler_plot"]
self.line_kp_elevation = [None, None] self.line_rk_elevation = [None, None]
self._isometric_axis = False self._isometric_axis = False
@@ -65,9 +65,9 @@ class PlotStricklers(PamhyrPlot):
self._init = True self._init = True
def draw_data(self): def draw_data(self):
kp = self.data.reach.get_kp() rk = self.data.reach.get_rk()
self.canvas.axes.set_xlim( self.canvas.axes.set_xlim(
left=min(kp), right=max(kp) left=min(rk), right=max(rk)
) )
frictions = self.data.frictions frictions = self.data.frictions
@@ -76,14 +76,14 @@ class PlotStricklers(PamhyrPlot):
self.draw_frictions(lst, self.color_plot) self.draw_frictions(lst, self.color_plot)
# HightLight # HightLight
kp_min, kp_max = (-1, -1) rk_min, rk_max = (-1, -1)
if self._highlight_data is not None: if self._highlight_data is not None:
kp_min, kp_max = self._highlight_data rk_min, rk_max = self._highlight_data
lst = list( lst = list(
filter( filter(
lambda s: (s.begin_kp == kp_min and lambda s: (s.begin_rk == rk_min and
s.end_kp == kp_max), s.end_rk == rk_max),
frictions.frictions frictions.frictions
) )
) )
@@ -96,7 +96,7 @@ class PlotStricklers(PamhyrPlot):
lambda f: f.is_full_defined(), lambda f: f.is_full_defined(),
frictions frictions
), ),
key=lambda s: s.begin_kp key=lambda s: s.begin_rk
) )
coef = flatten( coef = flatten(
@@ -106,9 +106,9 @@ class PlotStricklers(PamhyrPlot):
) )
) )
kp = flatten( rk = flatten(
map( map(
lambda s: [s.begin_kp, s.end_kp], lambda s: [s.begin_rk, s.end_rk],
lst lst
) )
) )
@@ -116,13 +116,13 @@ class PlotStricklers(PamhyrPlot):
coef_minor = list(map(lambda s: s.minor, coef)) coef_minor = list(map(lambda s: s.minor, coef))
coef_medium = list(map(lambda s: s.medium, coef)) coef_medium = list(map(lambda s: s.medium, coef))
self.line_kp_elevation[0] = self.canvas.axes.plot( self.line_rk_elevation[0] = self.canvas.axes.plot(
kp, coef_minor, rk, coef_minor,
color=color, lw=1. color=color, lw=1.
) )
self.line_kp_elevation[1] = self.canvas.axes.plot( self.line_rk_elevation[1] = self.canvas.axes.plot(
kp, coef_medium, rk, coef_medium,
color=color, lw=1. color=color, lw=1.
) )
+11 -10
View File
@@ -52,11 +52,12 @@ _translate = QCoreApplication.translate
class ComboBoxDelegate(QItemDelegate): class ComboBoxDelegate(QItemDelegate):
def __init__(self, data=None, study=None, def __init__(self, data=None, study=None,
mode="stricklers", parent=None): mode="stricklers", trad=None, parent=None):
super(ComboBoxDelegate, self).__init__(parent) super(ComboBoxDelegate, self).__init__(parent)
self._data = data self._data = data
self._study = study self._study = study
self._trad = trad
self._mode = mode self._mode = mode
def createEditor(self, parent, option, index): def createEditor(self, parent, option, index):
@@ -64,7 +65,7 @@ class ComboBoxDelegate(QItemDelegate):
if self._mode == "stricklers": if self._mode == "stricklers":
self.editor.addItems( self.editor.addItems(
[_translate("Frictions", "Not defined")] + [self._trad["not_defined"]] +
list( list(
map( map(
lambda s: str(s), lambda s: str(s),
@@ -112,19 +113,19 @@ class TableModel(PamhyrTableModel):
if self._headers[column] == "name": if self._headers[column] == "name":
return self._lst.get(row).name return self._lst.get(row).name
elif self._headers[column] == "begin_kp": elif self._headers[column] == "begin_rk":
return self._lst.get(row).begin_kp return self._lst.get(row).begin_rk
elif self._headers[column] == "end_kp": elif self._headers[column] == "end_rk":
return self._lst.get(row).end_kp return self._lst.get(row).end_rk
elif self._headers[column] == "begin_strickler": elif self._headers[column] == "begin_strickler":
value = self._lst.get(row).begin_strickler value = self._lst.get(row).begin_strickler
if value is None: if value is None:
return _translate("Frictions", "Not defined") return self._trad["not_defined"]
return str(value) return str(value)
elif self._headers[column] == "end_strickler": elif self._headers[column] == "end_strickler":
value = self._lst.get(row).end_strickler value = self._lst.get(row).end_strickler
if value is None: if value is None:
return _translate("Frictions", "Not defined") return self._trad["not_defined"]
return str(value) return str(value)
return QVariant() return QVariant()
@@ -143,13 +144,13 @@ class TableModel(PamhyrTableModel):
self._lst, row, value self._lst, row, value
) )
) )
elif self._headers[column] == "begin_kp": elif self._headers[column] == "begin_rk":
self._undo.push( self._undo.push(
SetBeginCommand( SetBeginCommand(
self._lst, row, value self._lst, row, value
) )
) )
elif self._headers[column] == "end_kp": elif self._headers[column] == "end_rk":
self._undo.push( self._undo.push(
SetEndCommand( SetEndCommand(
self._lst, row, value self._lst, row, value
+6 -6
View File
@@ -49,14 +49,14 @@ class SetBeginCommand(QUndoCommand):
self._frictions = frictions self._frictions = frictions
self._index = index self._index = index
self._old = self._frictions.get(self._index).begin_kp self._old = self._frictions.get(self._index).begin_rk
self._new = float(new_value) self._new = float(new_value)
def undo(self): def undo(self):
self._frictions.get(self._index).begin_kp = float(self._old) self._frictions.get(self._index).begin_rk = float(self._old)
def redo(self): def redo(self):
self._frictions.get(self._index).begin_kp = float(self._new) self._frictions.get(self._index).begin_rk = float(self._new)
class SetEndCommand(QUndoCommand): class SetEndCommand(QUndoCommand):
@@ -65,14 +65,14 @@ class SetEndCommand(QUndoCommand):
self._frictions = frictions self._frictions = frictions
self._index = index self._index = index
self._old = self._frictions.get(self._index).end_kp self._old = self._frictions.get(self._index).end_rk
self._new = float(new_value) self._new = float(new_value)
def undo(self): def undo(self):
self._frictions.get(self._index).end_kp = float(self._old) self._frictions.get(self._index).end_rk = float(self._old)
def redo(self): def redo(self):
self._frictions.get(self._index).end_kp = float(self._new) self._frictions.get(self._index).end_rk = float(self._new)
class SetBeginStricklerCommand(QUndoCommand): class SetBeginStricklerCommand(QUndoCommand):
+14 -8
View File
@@ -39,6 +39,8 @@ from PyQt5.QtWidgets import (
QComboBox, QVBoxLayout, QHeaderView, QTabWidget, QComboBox, QVBoxLayout, QHeaderView, QTabWidget,
) )
from Modules import Modules
from View.Frictions.UndoCommand import ( from View.Frictions.UndoCommand import (
PasteCommand, DuplicateCommand, PasteCommand, DuplicateCommand,
) )
@@ -48,7 +50,7 @@ from View.Frictions.Table import (
) )
from View.Tools.Plot.PamhyrCanvas import MplCanvas from View.Tools.Plot.PamhyrCanvas import MplCanvas
from View.Geometry.PlotKPZ import PlotKPZ from View.Frictions.PlotRKZ import PlotRKZ
from View.Frictions.PlotStricklers import PlotStricklers from View.Frictions.PlotStricklers import PlotStricklers
from View.Frictions.translate import FrictionsTranslate from View.Frictions.translate import FrictionsTranslate
@@ -99,6 +101,7 @@ class FrictionsWindow(PamhyrWindow):
data=self._reach, data=self._reach,
study=self._study, study=self._study,
mode="stricklers", mode="stricklers",
trad=self._trad,
parent=self parent=self
) )
@@ -107,7 +110,7 @@ class FrictionsWindow(PamhyrWindow):
table_view=table, table_view=table,
table_headers=self._trad.get_dict("table_headers"), table_headers=self._trad.get_dict("table_headers"),
editable_headers=[ editable_headers=[
"name", "begin_kp", "end_kp", "name", "begin_rk", "end_rk",
"begin_strickler", "end_strickler" "begin_strickler", "end_strickler"
], ],
delegates={ delegates={
@@ -115,6 +118,7 @@ class FrictionsWindow(PamhyrWindow):
"end_strickler": self._delegate_stricklers, "end_strickler": self._delegate_stricklers,
}, },
data=self._reach, data=self._reach,
trad=self._trad,
undo=self._undo_stack, undo=self._undo_stack,
opt_data=self._study opt_data=self._study
) )
@@ -129,11 +133,13 @@ class FrictionsWindow(PamhyrWindow):
self.plot_layout = self.find(QVBoxLayout, "verticalLayout") self.plot_layout = self.find(QVBoxLayout, "verticalLayout")
self.plot_layout.addWidget(self.canvas) self.plot_layout.addWidget(self.canvas)
self.plot = PlotKPZ( self.plot = PlotRKZ(
canvas=self.canvas, canvas=self.canvas,
data=self._reach.reach, reach=self._reach,
river=self._study.river,
trad=self._trad, trad=self._trad,
toolbar=None, toolbar=None,
parent=self
) )
self.plot.draw() self.plot.draw()
@@ -179,10 +185,10 @@ class FrictionsWindow(PamhyrWindow):
def update(self): def update(self):
self._set_current_reach() self._set_current_reach()
# self._propagate_update(key="friction") # self._propagate_update(key=Modules.FRICTION)
def _propagated_update(self, key=None): def _propagated_update(self, key=Modules(0)):
if key != "geometry": if Modules.GEOMETRY not in key:
return return
self.update_plot() self.update_plot()
@@ -198,7 +204,7 @@ class FrictionsWindow(PamhyrWindow):
data = self._reach data = self._reach
reach = self._reach.reach reach = self._reach.reach
sec = self._frictions.get(rows[0]) sec = self._frictions.get(rows[0])
highlight = (sec.begin_kp, sec.end_kp) highlight = (sec.begin_rk, sec.end_rk)
self.update_plot(highlight) self.update_plot(highlight)
+11 -10
View File
@@ -18,17 +18,18 @@
from PyQt5.QtCore import QCoreApplication from PyQt5.QtCore import QCoreApplication
from View.Tools.PamhyrTranslate import PamhyrTranslate from View.Translate import MainTranslate
_translate = QCoreApplication.translate _translate = QCoreApplication.translate
class FrictionsTranslate(PamhyrTranslate): class FrictionsTranslate(MainTranslate):
def __init__(self): def __init__(self):
super(FrictionsTranslate, self).__init__() super(FrictionsTranslate, self).__init__()
self._dict["kp"] = _translate( self._dict["rk"] = self._dict["unit_rk"]
"Frictions", "Kp (m)" self._dict["strickler_plot"] = _translate(
"Frictions", "Strickler ($m^{1/3}/s$)"
) )
self._dict["stricklers"] = _translate( self._dict["stricklers"] = _translate(
"Frictions", "Stricklers" "Frictions", "Stricklers"
@@ -39,10 +40,10 @@ class FrictionsTranslate(PamhyrTranslate):
) )
self._sub_dict["table_headers"] = { self._sub_dict["table_headers"] = {
# "name": _translate("Frictions", "Name"), # "name": self._dict["name"],
# "edge": _translate("Frictions", "Reach"), # "edge": self._dict["reach"],
"begin_kp": _translate("Frictions", "Begin kp (m)"), "begin_rk": _translate("Frictions", "Start (m)"),
"end_kp": _translate("Frictions", "End kp (m)"), "end_rk": _translate("Frictions", "End (m)"),
"begin_strickler": _translate("Frictions", "Begin strickler"), "begin_strickler": _translate("Frictions", "Start coefficient"),
"end_strickler": _translate("Frictions", "End strickler"), "end_strickler": _translate("Frictions", "End coefficient"),
} }
+13 -30
View File
@@ -58,6 +58,7 @@ class MeshingDialog(PamhyrDialog):
self._end_cs = -1 self._end_cs = -1
self._begin_dir = "un" self._begin_dir = "un"
self._end_dir = "np" self._end_dir = "np"
self._origin = self._reach.profile(0)
self._init_default_values_profiles() self._init_default_values_profiles()
self._init_default_values_guidelines() self._init_default_values_guidelines()
@@ -67,17 +68,6 @@ class MeshingDialog(PamhyrDialog):
self._space_step self._space_step
) )
lm_dict = self._trad.get_dict("lm_dict")
self.combobox_add_items(
"comboBox_lm", list(
map(
lambda x: lm_dict[x],
["1", "2", "3"]
)
)
)
self.set_combobox_text("comboBox_lm", self._lm)
if self._linear: if self._linear:
self.set_radio_button("radioButton_linear", True) self.set_radio_button("radioButton_linear", True)
else: else:
@@ -86,11 +76,11 @@ class MeshingDialog(PamhyrDialog):
def _init_default_values_profiles(self): def _init_default_values_profiles(self):
profiles = self.profiles profiles = self.profiles
self.combobox_add_items("comboBox_begin_kp", profiles) self.combobox_add_items("comboBox_begin_rk", profiles)
self.combobox_add_items("comboBox_end_kp", profiles) self.combobox_add_items("comboBox_end_rk", profiles)
self.set_combobox_text("comboBox_begin_kp", profiles[0]) self.set_combobox_text("comboBox_begin_rk", profiles[0])
self.set_combobox_text("comboBox_end_kp", profiles[-1]) self.set_combobox_text("comboBox_end_rk", profiles[-1])
@property @property
def profiles(self): def profiles(self):
@@ -108,9 +98,9 @@ class MeshingDialog(PamhyrDialog):
name = profile.name name = profile.name
if name == "": if name == "":
name = f"{profile.kp}" name = f"{profile.rk}"
else: else:
name += f" ({profile.kp})" name += f" ({profile.rk})"
return name return name
@@ -118,8 +108,8 @@ class MeshingDialog(PamhyrDialog):
gl, _ = self._reach.compute_guidelines() gl, _ = self._reach.compute_guidelines()
gl = list(gl) gl = list(gl)
bgl = ['un'] + gl bgl = ['un'] + gl + ['np']
egl = gl + ['np'] egl = ['un'] + gl + ['np']
self.combobox_add_items("comboBox_begin_gl", bgl) self.combobox_add_items("comboBox_begin_gl", bgl)
self.combobox_add_items("comboBox_end_gl", egl) self.combobox_add_items("comboBox_end_gl", egl)
@@ -135,21 +125,17 @@ class MeshingDialog(PamhyrDialog):
def lplan(self): def lplan(self):
return self._lplan return self._lplan
@property
def lm(self):
return int(self._lm)
@property @property
def linear(self): def linear(self):
return self._linear return self._linear
@property @property
def begin_cs(self): def begin_cs(self):
return self._begin_cs + 1 return self._begin_cs
@property @property
def end_cs(self): def end_cs(self):
return self._end_cs + 1 return self._end_cs
@property @property
def begin_dir(self): def begin_dir(self):
@@ -165,8 +151,8 @@ class MeshingDialog(PamhyrDialog):
) )
self._linear = self.get_radio_button("radioButton_linear") self._linear = self.get_radio_button("radioButton_linear")
p1 = self.get_combobox_text("comboBox_begin_kp") p1 = self.get_combobox_text("comboBox_begin_rk")
p2 = self.get_combobox_text("comboBox_end_kp") p2 = self.get_combobox_text("comboBox_end_rk")
self._begin_cs = self.profiles.index(p1) self._begin_cs = self.profiles.index(p1)
self._end_cs = self.profiles.index(p2) self._end_cs = self.profiles.index(p2)
@@ -174,9 +160,6 @@ class MeshingDialog(PamhyrDialog):
self._begin_dir = self.get_combobox_text("comboBox_begin_gl") self._begin_dir = self.get_combobox_text("comboBox_begin_gl")
self._end_dir = self.get_combobox_text("comboBox_end_gl") self._end_dir = self.get_combobox_text("comboBox_end_gl")
r_lm_dict = self._trad.get_dict("r_lm_dict")
self._lm = r_lm_dict[self.get_combobox_text("comboBox_lm")]
super().accept() super().accept()
def reject(self): def reject(self):
+23 -20
View File
@@ -21,12 +21,6 @@ import logging
from tools import timer from tools import timer
from View.Tools.PamhyrPlot import PamhyrPlot from View.Tools.PamhyrPlot import PamhyrPlot
from PyQt5.QtCore import (
QCoreApplication
)
_translate = QCoreApplication.translate
logger = logging.getLogger() logger = logging.getLogger()
@@ -46,20 +40,12 @@ class PlotAC(PamhyrPlot):
self._auto_relim_update = True self._auto_relim_update = True
self._autoscale_update = True self._autoscale_update = True
self.label_x = _translate( self.label_x = self._trad["transverse_abscissa"]
"Geometry", "Transverse abscissa (m)" self.label_y = self._trad["unit_height"]
)
self.label_y = _translate("Geometry", "Height (m)")
self.label_previous_plot_selected = _translate( self.label_previous_plot_selected = self._trad["prev_cs"]
"Geometry", "Previous cross-section" self.label_plot_selected = self._trad["cs"]
) self.label_next_plot_selected = self._trad["next_cs"]
self.label_plot_selected = _translate(
"Geometry", "Cross-section"
)
self.label_next_plot_selected = _translate(
"Geometry", "Next cross-section"
)
self.previous_plot_selected = None self.previous_plot_selected = None
self.plot_selected = None self.plot_selected = None
@@ -112,6 +98,7 @@ class PlotAC(PamhyrPlot):
label=self.label_next_plot_selected, label=self.label_next_plot_selected,
color=self.color_plot_next, lw=1.6, linestyle='--' color=self.color_plot_next, lw=1.6, linestyle='--'
) )
self.canvas.axes.set_visible(False)
def draw_gl(self): def draw_gl(self):
if self._current_data_update: if self._current_data_update:
@@ -120,12 +107,16 @@ class PlotAC(PamhyrPlot):
profile_id = 0 profile_id = 0
profile = self.data.profile(profile_id) profile = self.data.profile(profile_id)
if profile is None:
return
station = profile.get_station() station = profile.get_station()
elevation = profile.z() elevation = profile.z()
gl = profile.names() gl = profile.names()
self.annotation = [] self.annotation = []
self.complete_gl, self.incomplete_gl = self.data.compute_guidelines() self.complete_gl, self.incomplete_gl = self.data.compute_guidelines()
lcomplete = list(self.complete_gl) lcomplete = list(self.complete_gl)
lincomplete = list(self.incomplete_gl) lincomplete = list(self.incomplete_gl)
@@ -143,7 +134,7 @@ class PlotAC(PamhyrPlot):
if txt.strip() in self.complete_gl: if txt.strip() in self.complete_gl:
color = self.color_complete_gl[ color = self.color_complete_gl[
lcomplete.index(txt) lcomplete.index(txt) % len(self.color_complete_gl)
] ]
else: else:
color = self.color_incomplete_gl[ color = self.color_incomplete_gl[
@@ -174,6 +165,9 @@ class PlotAC(PamhyrPlot):
self.draw() self.draw()
return return
if self.data is None:
return
self.update_current() self.update_current()
self.update_gl() self.update_gl()
@@ -185,6 +179,10 @@ class PlotAC(PamhyrPlot):
previous_id = profile_id - 1 previous_id = profile_id - 1
next_id = profile_id + 1 next_id = profile_id + 1
self.previous_plot_selected.set_visible(False)
self.plot_selected.set_visible(False)
self.next_plot_selected.set_visible(False)
self.previous_plot_selected.set_data([], []) self.previous_plot_selected.set_data([], [])
self.plot_selected.set_data([], []) self.plot_selected.set_data([], [])
self.next_plot_selected.set_data([], []) self.next_plot_selected.set_data([], [])
@@ -194,18 +192,23 @@ class PlotAC(PamhyrPlot):
self.data.profile(previous_id).get_station(), self.data.profile(previous_id).get_station(),
self.data.profile(previous_id).z() self.data.profile(previous_id).z()
) )
self.previous_plot_selected.set_visible(True)
if 0 <= profile_id < self.data.number_profiles: if 0 <= profile_id < self.data.number_profiles:
self.plot_selected.set_data( self.plot_selected.set_data(
self.data.profile(profile_id).get_station(), self.data.profile(profile_id).get_station(),
self.data.profile(profile_id).z() self.data.profile(profile_id).z()
) )
self.plot_selected.set_visible(True)
if 0 <= next_id < self.data.number_profiles: if 0 <= next_id < self.data.number_profiles:
self.next_plot_selected.set_data( self.next_plot_selected.set_data(
self.data.profile(next_id).get_station(), self.data.profile(next_id).get_station(),
self.data.profile(next_id).z() self.data.profile(next_id).z()
) )
self.next_plot_selected.set_visible(True)
self.canvas.axes.set_visible(True)
def update_full(self): def update_full(self):
self.draw() self.draw()
-275
View File
@@ -1,275 +0,0 @@
# PlotKPC.py -- Pamhyr
# Copyright (C) 2023-2024 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 tools import timer
from View.Tools.PamhyrPlot import PamhyrPlot
from PyQt5.QtCore import (
QCoreApplication
)
_translate = QCoreApplication.translate
logger = logging.getLogger()
class PlotKPZ(PamhyrPlot):
def __init__(self, canvas=None, trad=None, data=None, toolbar=None,
parent=None):
super(PlotKPZ, self).__init__(
canvas=canvas,
trad=trad,
data=data,
toolbar=toolbar,
parent=parent
)
self._isometric_axis = False
self.line_kp_zgl = []
self.line_kp_zmin = None
self.line_kp_zmin_zmax = None
self.line_kp_zmin_zmax_highlight = None
self.label_x = _translate("MainWindow_reach", "Kp (m)")
self.label_y = _translate("MainWindow_reach", "Height (m)")
self.before_plot_selected = None
self.plot_selected = None
self.after_plot_selected = None
@timer
def draw(self):
self.init_axes()
if self.data is None:
return
if len(self.data.profiles) == 0:
return
profiles_defined = any(
filter(
lambda profile: len(profile) > 0,
self.data.profiles
)
)
if not profiles_defined:
self._init = False
return
self.draw_z_line()
self.draw_z_line_highlight()
self.draw_current()
self.draw_gl()
self.draw_bottom()
self.idle()
self._init = True
def draw_z_line(self):
kp = self.data.get_kp_complete_profiles()
z_min = self.data.get_z_min()
z_max = self.data.get_z_max()
self.line_kp_zmin_zmax = self.canvas.axes.vlines(
x=kp, ymin=z_min, ymax=z_max,
color=self.color_plot,
lw=1.
)
def draw_z_line_highlight(self):
if self._highlight_data is not None:
kp = self.data.get_kp_complete_profiles()
z_min = self.data.get_z_min()
z_max = self.data.get_z_max()
kp_min, kp_max = self._highlight_data
indexes = list(
map(
lambda x: x[0],
filter(
lambda x: kp_min <= x[1] <= kp_max,
enumerate(kp)
)
)
)
def indexes_filter(data): return list(
map(
lambda x: x[1],
filter(
lambda x: x[0] in indexes,
enumerate(data)
)
)
)
ikp = indexes_filter(kp)
imin = indexes_filter(z_min)
imax = indexes_filter(z_max)
self.line_kp_zmin_zmax_highlight = self.canvas.axes.vlines(
x=ikp,
ymin=imin, ymax=imax,
color=self.color_plot_highlight,
lw=1.
)
def draw_current(self):
kp = self.data.get_kp_complete_profiles()
z_min = self.data.get_z_min()
z_max = self.data.get_z_max()
self.plot_selected, = self.canvas.axes.plot(
(kp[0], kp[0]),
(z_min[0], z_max[0]),
color=self.color_plot_current, lw=1.5
)
self.plot_selected.set_visible(False)
self.before_plot_selected, = self.canvas.axes.plot(
(kp[0], kp[0]),
(z_min[0], z_max[0]),
color=self.color_plot_previous, lw=1.5, linestyle='--'
)
self.before_plot_selected.set_visible(False)
self.after_plot_selected, = self.canvas.axes.plot(
(kp[0], kp[0]),
(z_min[0], z_max[0]),
color=self.color_plot_next, lw=1.5, linestyle='--'
)
self.after_plot_selected.set_visible(False)
def draw_gl(self):
kp = self.data.get_kp_complete_profiles()
ind = 0
self.line_kp_zgl = []
for z in self.data.get_guidelines_z():
# Is incomplete guideline?
if len(z) != len(kp):
continue
self.line_kp_zgl.append(
self.canvas.axes.plot(
kp, z, lw=1., color=self.colors[ind]
)
)
ind += 1
def draw_bottom(self):
kp = self.data.get_kp_complete_profiles()
z_min = self.data.get_z_min()
self.line_kp_zmin, = self.canvas.axes.plot(
kp, z_min,
linestyle=":", lw=1.5,
color='lightgrey'
)
@timer
def update(self):
if not self._init:
self.draw()
return
self.update_gl()
self.update_current()
self.update_idle()
def update_current(self):
if self._current_data_update:
ind = self._current_data
before = ind - 1
after = ind + 1
self.before_plot_selected.set_visible(False)
self.plot_selected.set_visible(False)
self.after_plot_selected.set_visible(False)
if 0 <= before < self.data.number_profiles:
profile = self.data.profile(before)
if len(profile) > 0:
kp_i = profile.kp
z_min_i = profile.z_min()
z_max_i = profile.z_max()
self.before_plot_selected.set_data(
(kp_i, kp_i),
(z_min_i, z_max_i)
)
self.before_plot_selected.set_visible(True)
if 0 <= ind < self.data.number_profiles:
profile = self.data.profile(ind)
if len(profile) > 0:
kp_i = profile.kp
z_min_i = profile.z_min()
z_max_i = profile.z_max()
self.plot_selected.set_data(
(kp_i, kp_i),
(z_min_i, z_max_i)
)
self.plot_selected.set_visible(True)
if 0 <= after < self.data.number_profiles:
profile = self.data.profile(after)
if len(profile) > 0:
kp_i = profile.kp
z_min_i = profile.z_min()
z_max_i = profile.z_max()
self.after_plot_selected.set_data(
(kp_i, kp_i),
(z_min_i, z_max_i)
)
self.after_plot_selected.set_visible(True)
def update_gl(self):
if self._current_data_update:
return
kp = self.data.get_kp_complete_profiles()
z_min = self.data.get_z_min()
z_max = self.data.get_z_max()
self.line_kp_zmin.set_data(kp, z_min)
self.line_kp_zmin_zmax.remove()
self.line_kp_zmin_zmax = self.canvas.axes.vlines(
x=kp,
ymin=z_min, ymax=z_max,
color='r', lw=1.
)
z_complete = self.data.get_guidelines_z()
try:
for i in range(len(self.line_kp_zgl)):
self.line_kp_zgl[i][0].set_data(
kp, z_complete[i]
)
except Exception as e:
logger.warning(f"Failed to update graphic KPZ: {e}")
+372
View File
@@ -0,0 +1,372 @@
# PlotRKC.py -- Pamhyr
# Copyright (C) 2023-2024 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 tools import timer
from View.Tools.PamhyrPlot import PamhyrPlot
from PyQt5.QtWidgets import QApplication
from PyQt5.QtCore import (
Qt, QItemSelectionModel,
QItemSelection, QItemSelectionRange,
)
logger = logging.getLogger()
class PlotRKZ(PamhyrPlot):
def __init__(self, canvas=None, trad=None,
study=None, data=None, toolbar=None,
table=None, parent=None):
self._study = study
super(PlotRKZ, self).__init__(
canvas=canvas,
trad=trad,
data=data,
toolbar=toolbar,
table=table,
parent=parent
)
self._isometric_axis = False
self.line_rk_zgl = []
self.line_rk_zmin = None
self.line_rk_zmin_zmax = None
self.line_rk_zmin_zmax_highlight = None
self.label_x = self._trad["unit_rk"]
self.label_y = self._trad["unit_height"]
self.before_plot_selected = None
self.plot_selected = None
self.after_plot_selected = None
self.parent = parent
self._table = table
self._colors = []
def onpick(self, event):
if event.mouseevent.inaxes != self.canvas.axes:
return
if event.mouseevent.button.value != 1:
return
modifiers = QApplication.keyboardModifiers()
if modifiers not in [Qt.ControlModifier,
Qt.NoModifier,
Qt.ShiftModifier]:
return
ind, point = self._closest_rk(event)
if self.parent._table is None:
return
self.parent._table.blockSignals(True)
if modifiers == Qt.ControlModifier:
rows = list(
set(
(i.row() for i in self.parent.tableView.selectedIndexes())
)
)
if ind in rows:
rows.remove(ind)
else:
rows.append(ind)
self._select_in_table(rows)
elif modifiers == Qt.ShiftModifier:
rows = sorted(list(
set(
(i.row() for i in self.parent.tableView.selectedIndexes())
)
))
if len(rows) > 0:
i1 = min(rows[0], rows[-1], ind)
i2 = max(rows[0], rows[-1], ind)
else:
i1 = ind
i2 = ind
self._select_range_in_table(i1, i2)
else:
self._select_in_table([ind])
self.parent._table.blockSignals(False)
return
def _closest_rk(self, event):
s = event.artist.get_segments()
x = [i[0, 0] for i in s]
mx = event.mouseevent.xdata
points = enumerate(x)
def dist_mouse(point):
x = point[1]
d = abs(mx - x)
return d
closest = min(
points, key=dist_mouse
)
return closest
def _select_in_table(self, ind):
if self._table is None:
return
self._table.setFocus()
selection = self._table.selectionModel()
index = QItemSelection()
if len(ind) == 0:
return
for i in ind:
index.append(QItemSelectionRange(self._table.model().index(i, 0)))
selection.select(
index,
QItemSelectionModel.Rows |
QItemSelectionModel.ClearAndSelect |
QItemSelectionModel.Select
)
if len(ind) > 0:
self._table.scrollTo(self._table.model().index(ind[-1], 0))
def _select_range_in_table(self, ind1, ind2):
if self._table is not None:
self._table.setFocus()
selection = self._table.selectionModel()
index = QItemSelection(self._table.model().index(ind1, 0),
self._table.model().index(ind2, 0))
selection.select(
index,
QItemSelectionModel.Rows |
QItemSelectionModel.ClearAndSelect |
QItemSelectionModel.Select
)
self._table.scrollTo(self._table.model().index(ind2, 0))
@timer
def draw(self):
self.init_axes()
if self.data is None:
return
if len(self.data.profiles) == 0:
return
profiles_defined = any(
filter(
lambda profile: len(profile) > 0,
self.data.profiles
)
)
if not profiles_defined:
self._init = False
return
self.draw_z_line()
self.draw_lr()
self.draw_gl()
self.draw_bottom()
self.draw_profiles_hs(self._data)
self.idle()
self._init = True
def draw_z_line(self):
rk = self.data.get_rk_complete_profiles()
z_min = self.data.get_z_min()
z_max = self.data.get_z_max()
self._colors, self._style = self.color_hightlight()
self.line_rk_zmin_zmax = self.canvas.axes.vlines(
x=rk, ymin=z_min, ymax=z_max,
color=self._colors,
linestyle=self._style,
lw=1.,
picker=10,
)
def color_hightlight(self):
rows = sorted(list(
set(
(i.row() for i in self.parent.tableView.selectedIndexes())
)
))
colors = [self.color_plot for row in range(len(self._data))]
style = ["-" for row in range(len(self._data))]
if len(rows) > 0:
for row in rows:
colors[row] = self.color_plot_current
if rows[0] > 0:
colors[rows[0]-1] = self.color_plot_previous
style[rows[0]-1] = "--"
if rows[-1] < len(self._data)-1:
colors[rows[-1]+1] = self.color_plot_next
style[rows[-1]+1] = "--"
return colors, style
def draw_lr(self):
rk = self.data.get_rk_complete_profiles()
lz = []
rz = []
self.line_lr = []
for z in self.data.get_z():
lz.append(z[0])
rz.append(z[-1])
line = self.canvas.axes.plot(
rk, lz,
color=self.color_plot_river_bottom,
linestyle="dotted",
lw=1.,
)
self.line_lr.append(line)
line = self.canvas.axes.plot(
rk, rz,
color=self.color_plot_river_bottom,
linestyle="dotted",
lw=1.,
)
self.line_lr.append(line)
def draw_gl(self):
rk = self.data.get_rk_complete_profiles()
ind = 0
self.line_rk_zgl = []
for z in self.data.get_guidelines_z():
# Is incomplete guideline?
if len(z) != len(rk):
continue
self.line_rk_zgl.append(
self.canvas.axes.plot(
rk, z, lw=1.,
color=self.colors[ind % len(self.colors)],
linestyle=self.linestyle[ind // len(self.colors)]
)
)
ind += 1
def draw_bottom(self):
rk = self.data.get_rk_complete_profiles()
z_min = self.data.get_z_min()
self.line_rk_zmin, = self.canvas.axes.plot(
rk, z_min,
linestyle=":", lw=1.5,
color='lightgrey'
)
def draw_profiles_hs(self, reach):
lhs = filter(
lambda hs: hs._input_reach.reach is reach,
filter(
lambda hs: hs._input_reach is not None,
self._study.river.hydraulic_structures.lst
)
)
for hs in lhs:
x = hs.input_rk
if x is not None:
z_min = reach.get_z_min()
z_max = reach.get_z_max()
self.canvas.axes.plot(
[x, x],
[min(z_min), max(z_max)],
linestyle="--",
lw=1.,
color=self.color_plot_previous,
)
self.canvas.axes.annotate(
" > " + hs.name,
(x, max(z_max)),
horizontalalignment='left',
verticalalignment='top',
annotation_clip=True,
fontsize=9, color=self.color_plot_previous,
)
@timer
def update(self):
if not self._init:
self.draw()
return
self.update_lr()
self.update_gl()
self.update_current()
self.update_idle()
def update_current(self):
if self._current_data_update:
self._colors, self._style = self.color_hightlight()
self.line_rk_zmin_zmax.set_colors(self._colors)
self.line_rk_zmin_zmax.set_linestyle(self._style)
def update_gl(self):
if self._current_data_update:
return
rk = self.data.get_rk_complete_profiles()
z_min = self.data.get_z_min()
z_max = self.data.get_z_max()
self.line_rk_zmin.set_data(rk, z_min)
# TODO comprendre à quoi sert ce bout de code
# ========>
# self.line_rk_zmin_zmax.remove()
# self._colors, self._style = self.color_hightlight()
# self.line_rk_zmin_zmax = self.canvas.axes.vlines(
# x=rk,
# ymin=z_min,
# ymax=z_max,
# color=self._colors,
# linestyle = self._style,
# lw=1.
# )
# <========
z_complete = self.data.get_guidelines_z()
try:
for i in range(len(self.line_rk_zgl)):
self.line_rk_zgl[i][0].set_data(
rk, z_complete[i]
)
except Exception as e:
logger.warning(f"Failed to update graphic RKZ: {e}")
def update_lr(self):
for line in self.line_lr:
line[0].remove()
self.draw_lr()
+210 -91
View File
@@ -18,75 +18,233 @@
from tools import timer, trace from tools import timer, trace
from View.Tools.PamhyrPlot import PamhyrPlot from View.Tools.PamhyrPlot import PamhyrPlot
from matplotlib import collections
import numpy as np
from PyQt5.QtCore import ( from PyQt5.QtCore import (
QCoreApplication QCoreApplication, Qt, QItemSelectionModel,
QItemSelection, QItemSelectionRange,
) )
from PyQt5.QtWidgets import QApplication
_translate = QCoreApplication.translate _translate = QCoreApplication.translate
class PlotXY(PamhyrPlot): class PlotXY(PamhyrPlot):
def __init__(self, canvas=None, trad=None, data=None, toolbar=None, def __init__(self, canvas=None, trad=None, data=None, toolbar=None,
parent=None): table=None, parent=None):
super(PlotXY, self).__init__( super(PlotXY, self).__init__(
canvas=canvas, canvas=canvas,
trad=trad, trad=trad,
data=data, data=data,
toolbar=toolbar, toolbar=toolbar,
table=table,
parent=parent parent=parent
) )
self.line_xy = [] self.line_xy = []
self.line_gl = [] self.line_gl = []
self.label_x = _translate("Geometry", "X (m)") self.label_x = self._trad["x"]
self.label_y = _translate("Geometry", "Y (m)") self.label_y = self._trad["y"]
self.before_plot_selected = None self.before_plot_selected = None
self.plot_selected = None self.plot_selected = None
self.after_plot_selected = None self.after_plot_selected = None
self.parent = parent
self.line_xy_collection = None
self._table = table
self._colors = []
self._style = []
def onpick(self, event):
if event.mouseevent.inaxes != self.canvas.axes:
return
if event.mouseevent.button.value != 1:
return
modifiers = QApplication.keyboardModifiers()
if modifiers not in [Qt.ControlModifier,
Qt.NoModifier,
Qt.ShiftModifier]:
return
ind, point = self._closest_section(event)
if self._table is None:
return
self._table.blockSignals(True)
if modifiers == Qt.ControlModifier:
rows = list(
set(
(i.row() for i in self.parent.tableView.selectedIndexes())
)
)
if ind in rows:
rows.remove(ind)
self._select_in_table(rows)
else:
self._select_in_table(rows + [ind])
elif modifiers == Qt.ShiftModifier:
rows = list(
set(
(i.row() for i in self.parent.tableView.selectedIndexes())
)
)
if len(rows) > 0:
i1 = min(rows[0], rows[-1], ind)
i2 = max(rows[0], rows[-1], ind)
else:
i1 = ind
i2 = ind
self._select_range_in_table(i1, i2)
else:
self._select_in_table([ind])
self._table.blockSignals(False)
return
def _closest_section(self, event):
axes = self.canvas.axes
mx = event.mouseevent.xdata
my = event.mouseevent.ydata
bx, by = axes.get_xlim(), axes.get_ylim()
ratio = (bx[0] - bx[1]) / (by[0] - by[1])
segments = event.artist.get_segments()
ind = event.ind
points = []
for i in ind:
points = points + [[i, j] for j in segments[i]]
def dist_mouse(point):
x, y = point[1]
d2 = (((mx - x) / ratio) ** 2) + ((my - y) ** 2)
return d2
closest = min(
points, key=dist_mouse
)
return closest
def _select_in_table(self, ind):
if self._table is None:
return
self._table.setFocus()
selection = self._table.selectionModel()
index = QItemSelection()
if len(ind) == 0:
return
for i in ind:
index.append(QItemSelectionRange(self._table.model().index(i, 0)))
selection.select(
index,
QItemSelectionModel.Rows |
QItemSelectionModel.ClearAndSelect |
QItemSelectionModel.Select
)
if len(ind) > 0:
self._table.scrollTo(self._table.model().index(ind[-1], 0))
def _select_range_in_table(self, ind1, ind2):
if self._table is not None:
self._table.setFocus()
selection = self._table.selectionModel()
index = QItemSelection(self._table.model().index(ind1, 0),
self._table.model().index(ind2, 0))
selection.select(
index,
QItemSelectionModel.Rows |
QItemSelectionModel.ClearAndSelect |
QItemSelectionModel.Select
)
self._table.scrollTo(self._table.model().index(ind2, 0))
@timer @timer
def draw(self): def draw(self):
self.init_axes() self.init_axes()
if self.data is None: if self.data is None:
self.idle()
return return
if self.data.number_profiles == 0: if self.data.number_profiles == 0:
self._init = False self._init = False
self.idle()
return return
self.draw_xy() self.draw_xy()
self.draw_lr()
self.draw_gl() self.draw_gl()
self.draw_current()
self.idle() self.idle()
self._init = True self._init = True
def draw_xy(self): def draw_xy(self):
kp = self.data.get_kp_complete_profiles()
kp_min, kp_max = (-1, -1)
if self._highlight_data is not None:
kp_min, kp_max = self._highlight_data
def color_hightlight(kp):
if kp_min <= kp <= kp_max:
return self.color_plot_highlight
return self.color_plot
self.line_xy = [] self.line_xy = []
for x, y, kp in zip(self.data.get_x(), for xy in zip(self.data.get_x(), self.data.get_y()):
self.data.get_y(), self.line_xy.append(np.column_stack(xy))
kp):
line = self.canvas.axes.plot( self._colors, self._style = self.color_hightlight()
x, y, self.line_xy_collection = collections.LineCollection(
color=color_hightlight(kp), self.line_xy,
**self.plot_default_kargs colors=self._colors,
linestyle=self._style,
picker=10
)
self.canvas.axes.add_collection(self.line_xy_collection)
def color_hightlight(self):
rows = sorted(list(
set(
(i.row() for i in self.parent.tableView.selectedIndexes())
) )
self.line_xy.append(line) ))
colors = [self.color_plot for row in range(len(self._data))]
style = ["-" for row in range(len(self._data))]
if len(rows) > 0:
for row in rows:
colors[row] = self.color_plot_current
if rows[0] > 0:
colors[rows[0]-1] = self.color_plot_previous
style[rows[0]-1] = "--"
if rows[-1] < len(self._data)-1:
colors[rows[-1]+1] = self.color_plot_next
style[rows[-1]+1] = "--"
return colors, style
def draw_lr(self):
lx = []
ly = []
rx = []
ry = []
self.line_lr = []
for x, y in zip(self.data.get_x(),
self.data.get_y()):
lx.append(x[0])
ly.append(y[0])
rx.append(x[-1])
ry.append(y[-1])
line = self.canvas.axes.plot(
lx, ly,
color=self.color_plot_river_bottom,
linestyle="dotted",
lw=1.,
)
self.line_lr.append(line)
line = self.canvas.axes.plot(
rx, ry,
color=self.color_plot_river_bottom,
linestyle="dotted",
lw=1.,
)
self.line_lr.append(line)
def draw_gl(self): def draw_gl(self):
x_complete = self.data.get_guidelines_x() x_complete = self.data.get_guidelines_x()
@@ -96,38 +254,12 @@ class PlotXY(PamhyrPlot):
self.line_gl = [] self.line_gl = []
for x, y in zip(x_complete, y_complete): for x, y in zip(x_complete, y_complete):
line = self.canvas.axes.plot( line = self.canvas.axes.plot(
x, y, color=self.colors[ind] x, y, color=self.colors[ind % len(self.colors)],
linestyle=self.linestyle[ind // len(self.colors)]
) )
self.line_gl.append(line) self.line_gl.append(line)
ind += 1 ind += 1
def draw_current(self):
# Previous profile
self.before_plot_selected, = self.canvas.axes.plot(
self.data.profile(0).x(),
self.data.profile(0).y(),
color=self.color_plot_previous, linestyle="--",
**self.plot_default_kargs
)
self.before_plot_selected.set_visible(False)
# Current profile
self.plot_selected, = self.canvas.axes.plot(
self.data.profile(0).x(),
self.data.profile(0).y(),
color=self.color_plot_current, **self.plot_default_kargs
)
self.plot_selected.set_visible(False)
# Next profile
self.after_plot_selected, = self.canvas.axes.plot(
self.data.profile(0).x(),
self.data.profile(0).y(),
color=self.color_plot_next, linestyle='--',
**self.plot_default_kargs
)
self.after_plot_selected.set_visible(False)
@timer @timer
def update(self): def update(self):
if not self._init: if not self._init:
@@ -137,6 +269,7 @@ class PlotXY(PamhyrPlot):
if self.data is None: if self.data is None:
return return
self.update_lr()
self.update_gl() self.update_gl()
self.update_current() self.update_current()
@@ -147,21 +280,24 @@ class PlotXY(PamhyrPlot):
x_complete = list(self.data.get_guidelines_x()) x_complete = list(self.data.get_guidelines_x())
y_complete = list(self.data.get_guidelines_y()) y_complete = list(self.data.get_guidelines_y())
for i in range(self.data.number_profiles): # TODO comprendre à quoi sert ce bout de code
if i < len(self.line_xy): # ========>
self.line_xy[i][0].set_data( # for i in range(self.data.number_profiles):
self.data.profile(i).x(), # if i < len(self.line_xy):
self.data.profile(i).y() # self.line_xy[i][0].set_data(
) # self.data.profile(i).x(),
else: # self.data.profile(i).y()
self.line_xy.append( # )
self.canvas.axes.plot( # else:
self.data.profile(i).x(), # self.line_xy.append(
self.data.profile(i).y(), # self.canvas.axes.plot(
color='r', # self.data.profile(i).x(),
**self.plot_default_kargs # self.data.profile(i).y(),
) # color='r',
) # **self.plot_default_kargs
# )
# )
# <========
for i in range(len(x_complete)): for i in range(len(x_complete)):
if i < len(self.line_gl): if i < len(self.line_gl):
@@ -179,29 +315,12 @@ class PlotXY(PamhyrPlot):
def update_current(self): def update_current(self):
if self._current_data_update: if self._current_data_update:
ind = self._current_data self._colors, self._style = self.color_hightlight()
before = ind - 1 self.line_xy_collection.set_colors(self._colors)
after = ind + 1 self.line_xy_collection.set_linestyle(self._style)
self.before_plot_selected.set_visible(False) def update_lr(self):
self.plot_selected.set_visible(False) for line in self.line_lr:
self.after_plot_selected.set_visible(False) line[0].remove()
if 0 <= before < self.data.number_profiles: self.draw_lr()
self.before_plot_selected.set_data(
self.data.profile(before).x(),
self.data.profile(before).y()
)
self.before_plot_selected.set_visible(True)
if 0 <= ind < self.data.number_profiles:
self.plot_selected.set_data(self.data.profile(ind).x(),
self.data.profile(ind).y())
self.plot_selected.set_visible(True)
if 0 <= after < self.data.number_profiles:
self.after_plot_selected.set_data(
self.data.profile(after).x(),
self.data.profile(after).y()
)
self.after_plot_selected.set_visible(True)
+349 -47
View File
@@ -18,14 +18,19 @@
import logging import logging
from math import dist, sqrt
from tools import timer, trace from tools import timer, trace
from View.Tools.PamhyrPlot import PamhyrPlot from View.Tools.PamhyrPlot import PamhyrPlot
from View.Tools.Plot.OnPickEvent import OnpickEvent
from PyQt5.QtCore import ( from PyQt5.QtCore import (
QCoreApplication Qt, QCoreApplication, QItemSelectionModel,
QItemSelection, QItemSelectionRange,
) )
from PyQt5.QtWidgets import QApplication
from matplotlib.widgets import RectangleSelector
_translate = QCoreApplication.translate _translate = QCoreApplication.translate
logger = logging.getLogger() logger = logging.getLogger()
@@ -43,48 +48,289 @@ class Plot(PamhyrPlot):
) )
self._table = table self._table = table
self._parent = parent
self._z_note = None
self._z_line = None
self._z_fill_between = None
self.line_xy = [] self.line_xy = []
self.line_gl = [] self.line_gl = []
self.label_x = self._trad["unit_rk"]
self.label_y = self._trad["unit_height"]
self.before_plot_selected = None self.before_plot_selected = None
self.plot_selected = None self.plot_selected = None
self.after_plot_selected = None self.after_plot_selected = None
self._isometric_axis = False
self.hl_points = []
self.highlight = (
[], # Points list to highlight
None # Hydrolic values (z, wet_area,
# wet_preimeter, water_width)
)
self._onpickevent = None
self._rect_select = RectangleSelector(
ax=self.canvas.axes,
onselect=self.rect_select_callback,
useblit=True,
button=[1], # don't use middle nor right button
minspanx=2.0,
minspany=2.0,
spancoords='pixels',
interactive=False
)
def onrelease(self, event):
# we need to do that to prevent conflicst
# between onpick and rect_select_callback
modifiers = QApplication.keyboardModifiers()
points, hyd = self.highlight
if self._onpickevent is not None:
ind, point = self._closest_point(self._onpickevent)
if modifiers == Qt.ControlModifier:
rows = self._parent.index_selected_rows()
if ind in rows:
rows.remove(ind)
del (points[ind])
self.highlight = (points, hyd)
self._select_in_table(rows)
else:
self.highlight = (points+[point], hyd)
self._select_in_table(rows+[ind])
elif modifiers == Qt.ShiftModifier:
rows = self._parent.index_selected_rows()
if len(rows) > 0:
i1 = min(rows[0], rows[-1], ind)
i2 = max(rows[0], rows[-1], ind)
p = [
[self.data.points[i].x, self.data.points[i].y]
for i in range(i1, i2)
]
else:
i1 = ind
i2 = ind
p = [point]
self.highlight = (p, hyd)
self._select_range_in_table(i1, i2)
else:
self.highlight = ([point], hyd)
self._select_in_table([ind])
self._onpickevent = None
def onpick(self, event):
if event.mouseevent.inaxes != self.canvas.axes:
return
if event.mouseevent.button.value != 1:
return
modifiers = QApplication.keyboardModifiers()
if modifiers not in [Qt.ControlModifier,
Qt.NoModifier,
Qt.ShiftModifier]:
return
self._onpickevent = event
return
def onclick(self, event):
if event.inaxes != self.canvas.axes:
return
if event.button.value == 1:
return
points, _ = self.highlight
z = self._get_z_from_click(event)
if z < self.data.z_min() or event.button.value == 2:
self.highlight = (points, None)
self.update()
return
a, p, w = self._compute_hydraulics(z)
logger.debug(f"{z, a, p, w}")
self.highlight = (points, (z, a, p, w))
self.update()
return
def select_points_from_indices(self, indices):
data = self.data
_, hyd = self.highlight
points = list(
map(
lambda e: e[1],
filter(
lambda e: e[0] in indices,
enumerate(
zip(data.get_station(), data.z())
)
)
)
)
self.highlight = (points, hyd)
self.update()
def _select_in_table(self, ind):
if self._table is not None:
self._table.blockSignals(True)
self._table.setFocus()
selection = self._table.selectionModel()
index = QItemSelection()
if len(ind) > 0:
for i in ind:
index.append(QItemSelectionRange(
self._table.model().index(i, 0))
)
selection.select(
index,
QItemSelectionModel.Rows |
QItemSelectionModel.ClearAndSelect |
QItemSelectionModel.Select
)
if len(ind) > 0:
self._table.scrollTo(self._table.model().index(ind[-1], 0))
self._table.blockSignals(False)
def _select_range_in_table(self, ind1, ind2):
if self._table is not None:
self._table.blockSignals(True)
self._table.setFocus()
selection = self._table.selectionModel()
index = QItemSelection(self._table.model().index(ind1, 0),
self._table.model().index(ind2, 0))
selection.select(
index,
QItemSelectionModel.Rows |
QItemSelectionModel.ClearAndSelect |
QItemSelectionModel.Select
)
self._table.scrollTo(self._table.model().index(ind2, 0))
self._table.blockSignals(False)
def _closest_point(self, event):
points_ind = event.ind
axes = self.canvas.axes
bx, by = axes.get_xlim(), axes.get_ylim()
ratio = (bx[0] - bx[1]) / (by[0] - by[1])
x = event.artist.get_xdata()
y = event.artist.get_ydata()
# points = filter(
# lambda e: e[0] in points_ind,
# enumerate(zip(x, y))
# )
points = enumerate(zip(x, y))
mx = event.mouseevent.xdata
my = event.mouseevent.ydata
def dist_mouse(point):
x, y = point[1]
d2 = ((mx - x) / ratio) ** 2 + ((my - y) ** 2)
return d2
closest = min(
points, key=dist_mouse
)
return closest
def _get_z_from_click(self, event):
return event.ydata
def rect_select_callback(self, eclick, erelease):
points, hyd = self.highlight
x1, y1 = eclick.xdata, eclick.ydata
x2, y2 = erelease.xdata, erelease.ydata
if (max(abs(x1-x2), abs(y1-y2)) < 0.001):
return
modifiers = QApplication.keyboardModifiers()
x1, y1 = eclick.xdata, eclick.ydata
x2, y2 = erelease.xdata, erelease.ydata
inds, points2 = self._points_in_rectangle(x1, y1, x2, y2)
self._onclickevent = None
if modifiers == Qt.ControlModifier:
rows = self._parent.index_selected_rows()
if all(i in rows for i in inds):
for ind in sorted(inds, reverse=True):
rows.remove(ind)
del (points[ind])
self.highlight = (points, hyd)
self._select_in_table(rows)
else:
self.highlight = (points+points2, hyd)
self._select_in_table(rows+inds)
else:
self.highlight = (points2, hyd)
self._select_in_table(inds)
return
def _points_in_rectangle(self, x1, y1, x2, y2):
# TODO: use lambdas
listi = []
listp = []
station = self.data._get_station(self.data.points)
for i, p in enumerate(self.data.points):
if (min(x1, x2) < station[i] < max(x1, x2)
and min(y1, y2) < p.z < max(y1, y2)):
listi.append(i)
listp.append((station[i], p.z))
return listi, listp
def _compute_hydraulics(self, z):
profile = self.data
points = profile.wet_points(z)
station = profile._get_station(points)
width = abs(station[0] - station[-1])
poly = profile.wet_polygon(z)
area = poly.area
perimeter = poly.length
return area, perimeter, width
@timer @timer
def draw(self): def draw(self):
self.canvas.axes.cla() self.init_axes()
self.canvas.axes.grid(
color='grey', linestyle='--', linewidth=0.5
)
x = self.data.get_station() x = self.data.get_station()
y = self.data.z() y = self.data.z()
x_carto = self.data.x() x_carto = self.data.x()
y_carto = self.data.y() y_carto = self.data.y()
if (len(x_carto) < 3 or len(y_carto) < 3 or if (len(x_carto) < 3 or len(y_carto) < 3 or len(x) < 3):
len(x) < 3):
# Noting to do in this case # Noting to do in this case
return return
gl = map(lambda p: p.name, self.data.points)
self.profile_line2D, = self.canvas.axes.plot( self.profile_line2D, = self.canvas.axes.plot(
x, y, color='r', lw=1.5, x, y, color=self.color_plot,
markersize=7, marker='+', lw=1.5, markersize=7, marker='+',
picker=30 picker=10
)
self.canvas.axes.set_xlabel(
_translate("MainWindowProfile",
"Transverse abscissa (m)"),
color='black', fontsize=10
)
self.canvas.axes.set_ylabel(
_translate("MainWindowProfile", "Height (m)"),
color='black', fontsize=10
) )
self.draw_annotation(x, y)
self.draw_highligth()
self.idle()
def draw_annotation(self, x, y):
gl = map(lambda p: p.name, self.data.points)
# Add label on graph # Add label on graph
self.annotation = [] self.annotation = []
for i, name in enumerate(list(gl)): for i, name in enumerate(list(gl)):
@@ -120,32 +366,88 @@ class Plot(PamhyrPlot):
self.canvas.axes.set_facecolor('#F9F9F9') self.canvas.axes.set_facecolor('#F9F9F9')
self.canvas.figure.patch.set_facecolor('white') self.canvas.figure.patch.set_facecolor('white')
self.onpick_event = OnpickEvent( def draw_highligth(self):
self.canvas.axes, points, hyd = self.highlight
x, y, x_carto, y_carto, for p in self.hl_points:
self._table p[0].set_data([], [])
)
self.canvas.figure.canvas\
.mpl_connect(
'pick_event',
self.onpick_event.onpick
)
self.onclick_event = OnpickEvent( self.hl_points = []
self.canvas.axes,
x, y, x_carto, y_carto,
self._table
)
self.canvas.figure.canvas\
.mpl_connect(
'button_press_event',
self.onclick_event.onclick
)
self.canvas.figure.tight_layout() for x, y in points:
self.canvas.figure.canvas.draw_idle() self.hl_points.append(
self.canvas.axes.plot(
x, y,
color=self.color_plot_highlight,
lw=1.5, markersize=7, marker='+',
)
)
if hyd is not None:
self.draw_highligth_z_line(*hyd)
else:
if self._z_note is not None:
self._z_note.set_visible(False)
self._z_line[0].set_visible(False)
self._z_fill_between.set_visible(False)
def draw_highligth_z_line(self, z, a, p, w):
text = (
f"Z = {z:.3f} m, " +
f"{self._trad['width']} = {w:.3f} m,\n" +
f"{self._trad['area']} = {a:.3f} m², " +
f"{self._trad['perimeter']} = {p:.3f} m"
)
x = self.data.get_station()
xlim = (x[0], x[-1])
pos = (
xlim[0] + (abs(xlim[0] - xlim[1]) * 0.05),
z + 0.8
)
y = self.data.z()
if self._z_note is None:
self.draw_highligth_z_line_fill(x, y, z)
self._z_line = self.canvas.axes.plot(
xlim, [z, z],
color=self.color_plot_river_water
)
self._z_line[0].set_visible(True)
self._z_note = self.canvas.axes.annotate(
text, pos,
horizontalalignment='left',
verticalalignment='top',
annotation_clip=True,
color=self.color_plot_river_water,
fontsize=9,
fontweight='bold',
alpha=0.7
)
self._z_note.set_visible(True)
else:
self.draw_highligth_z_line_fill(x, y, z)
self._z_line[0].set_data(xlim, [z, z])
self._z_note.set_position(pos)
self._z_note.set_text(text)
self._z_line[0].set_visible(True)
self._z_note.set_visible(True)
def draw_highligth_z_line_fill(self, x, y, z):
if self._z_fill_between is not None:
self._z_fill_between.remove()
self._z_fill_between = self.canvas.axes.fill_between(
x, y, z,
where=y <= z,
facecolor=self.color_plot_river_water_zone,
interpolate=True, alpha=0.7
)
@timer @timer
def update(self, ind=None): def update(self):
logger.info("TODO: implemente update") self.draw_highligth()
self.draw()
self.update_idle()
+20 -20
View File
@@ -82,16 +82,6 @@ class GeometryProfileTableModel(PamhyrTableModel):
elif value == self._data.z_max(): elif value == self._data.z_max():
return QColor("blue") return QColor("blue")
if role == Qt.ToolTipRole:
if value == self._data.z_min():
return _translate("MainWindowProfile",
"La cote du fond",
"Z minimale")
elif value == self._data.z_max():
return _translate("MainWindowProfile",
"La cote maximale",
"Z maximale")
if index.column() == 3: if index.column() == 3:
value = self._data.point(index.row()).name value = self._data.point(index.row()).name
@@ -104,16 +94,6 @@ class GeometryProfileTableModel(PamhyrTableModel):
if role == Qt.ForegroundRole: if role == Qt.ForegroundRole:
return QColor("darkRed") return QColor("darkRed")
if role == Qt.ToolTipRole:
if value.strip().upper() == "RG":
return _translate(
"MainWindowProfile", "Rive gauche"
)
else:
return _translate(
"MainWindowProfile", "Rive droite"
)
return QVariant() return QVariant()
def setData(self, index, value, role=Qt.EditRole): def setData(self, index, value, role=Qt.EditRole):
@@ -250,6 +230,26 @@ class GeometryProfileTableModel(PamhyrTableModel):
self.endMoveRows() self.endMoveRows()
self.layoutChanged.emit() self.layoutChanged.emit()
def purge(self, np_purge):
self._undo.push(
PurgeCommand(
self._data, np_purge
)
)
self.layoutChanged.emit()
def reverse(self):
self._undo.push(
ReverseCommand(
self._data
)
)
self.layoutChanged.emit()
def paste(self, row, header, data): def paste(self, row, header, data):
if row > self._data.number_points: if row > self._data.number_points:
return return
+9 -5
View File
@@ -18,7 +18,7 @@
from PyQt5.QtCore import QCoreApplication from PyQt5.QtCore import QCoreApplication
from View.Tools.PamhyrTranslate import PamhyrTranslate from View.Translate import MainTranslate
from View.Geometry.Translate import GeometryTranslate from View.Geometry.Translate import GeometryTranslate
_translate = QCoreApplication.translate _translate = QCoreApplication.translate
@@ -32,10 +32,14 @@ class GeometryProfileTranslate(GeometryTranslate):
"Geometry", "Geometry cross-section" "Geometry", "Geometry cross-section"
) )
self._dict["width"] = _translate("Geometry", "Width")
self._dict["area"] = _translate("Geometry", "Area")
self._dict["perimeter"] = _translate("Geometry", "Perimeter")
self._sub_dict["table_headers"] = { self._sub_dict["table_headers"] = {
"x": _translate("Geometry", "X (m)"), "x": self._dict["x"],
"y": _translate("Geometry", "Y (m)"), "y": self._dict["y"],
"z": _translate("Geometry", "Z (m)"), "z": self._dict["z"],
"name": _translate("Geometry", "Name"), "name": self._dict["name"],
"abs": _translate("Geometry", "Traversal abs (m)"), "abs": _translate("Geometry", "Traversal abs (m)"),
} }
+28
View File
@@ -169,6 +169,34 @@ class MoveCommand(QUndoCommand):
self._profile.move_down_point(self._i) self._profile.move_down_point(self._i)
class ReverseCommand(QUndoCommand):
def __init__(self, profile):
QUndoCommand.__init__(self)
self._profile = profile
def undo(self):
self._profile.reverse()
def redo(self):
self._profile.reverse()
class PurgeCommand(QUndoCommand):
def __init__(self, profile, np_purge):
QUndoCommand.__init__(self)
self._profile = profile
self._old = self._profile.points.copy()
self._np_purge = np_purge
def undo(self):
self._profile._points = self._old.copy()
def redo(self):
self._profile.purge(self._np_purge)
class PasteCommand(QUndoCommand): class PasteCommand(QUndoCommand):
def __init__(self, profile, row, points): def __init__(self, profile, row, points):
QUndoCommand.__init__(self) QUndoCommand.__init__(self)
+65 -5
View File
@@ -20,8 +20,10 @@ import copy
import sys import sys
import csv import csv
from time import time from time import time
import logging
from tools import trace, timer from tools import trace, timer, logger_exception
from Modules import Modules
from PyQt5.QtGui import ( from PyQt5.QtGui import (
QKeySequence, QKeySequence,
@@ -45,9 +47,12 @@ from View.Tools.Plot.PamhyrCanvas import MplCanvas
from View.Geometry.Profile.Plot import Plot from View.Geometry.Profile.Plot import Plot
from View.Geometry.Profile.Table import GeometryProfileTableModel from View.Geometry.Profile.Table import GeometryProfileTableModel
from View.Geometry.Profile.Translate import GeometryProfileTranslate from View.Geometry.Profile.Translate import GeometryProfileTranslate
from View.Geometry.PurgeDialog import PurgeDialog
_translate = QCoreApplication.translate _translate = QCoreApplication.translate
logger = logging.getLogger()
class ProfileWindow(PamhyrWindow): class ProfileWindow(PamhyrWindow):
_pamhyr_ui = "GeometryCrossSection" _pamhyr_ui = "GeometryCrossSection"
@@ -59,7 +64,7 @@ class ProfileWindow(PamhyrWindow):
trad = GeometryProfileTranslate() trad = GeometryProfileTranslate()
name = ( name = (
trad[self._pamhyr_name] + trad[self._pamhyr_name] +
f" - {self._profile.name} {self._profile.kp}" f" - {self._profile.name} {self._profile.rk}"
) )
super(ProfileWindow, self).__init__( super(ProfileWindow, self).__init__(
@@ -108,8 +113,10 @@ class ProfileWindow(PamhyrWindow):
self._plot = Plot( self._plot = Plot(
canvas=self._canvas, canvas=self._canvas,
data=self._profile, data=self._profile,
trad=self._trad,
toolbar=self._toolbar, toolbar=self._toolbar,
table=self.find(QTableView, "tableView") table=self.find(QTableView, "tableView"),
parent=self
) )
self._plot.draw() self._plot.draw()
@@ -123,26 +130,49 @@ class ProfileWindow(PamhyrWindow):
"action_down": self.move_down, "action_down": self.move_down,
"action_add": self.add, "action_add": self.add,
"action_delete": self.delete, "action_delete": self.delete,
"action_purge": self.purge,
"action_reverse": self.reverse,
} }
for action in actions: for action in actions:
self.find(QAction, action)\ self.find(QAction, action)\
.triggered.connect(actions[action]) .triggered.connect(actions[action])
table = self.find(QTableView, f"tableView")
table.selectionModel()\
.selectionChanged\
.connect(self.update_points_selection)
self._tablemodel.dataChanged.connect(self.update) self._tablemodel.dataChanged.connect(self.update)
def update_points_selection(self):
rows = self.index_selected_rows()
self._plot.select_points_from_indices(rows)
def update(self): def update(self):
self.update_plot() self.update_plot()
self._propagate_update(key="geometry") self._propagate_update(key=Modules.GEOMETRY)
def _update(self, redraw=False, propagate=True):
if redraw:
self.update_plot()
if propagate:
self._propagate_update(key=Modules.GEOMETRY)
def update_plot(self): def update_plot(self):
self._tablemodel.blockSignals(True) self._tablemodel.blockSignals(True)
# TODO: Do not rebuild all graph
self._plot.update() self._plot.update()
self._tablemodel.blockSignals(False) self._tablemodel.blockSignals(False)
def _propagated_update(self, key=Modules(0)):
if Modules.GEOMETRY not in key:
return
self._tablemodel.layoutChanged.emit()
self._update(redraw=True, propagate=False)
def index_selected_row(self): def index_selected_row(self):
table = self.find(QTableView, "tableView") table = self.find(QTableView, "tableView")
rows = table.selectionModel()\ rows = table.selectionModel()\
@@ -152,6 +182,15 @@ class ProfileWindow(PamhyrWindow):
return rows[0].row() return rows[0].row()
def index_selected_rows(self):
table = self.find(QTableView, "tableView")
return list(
map(
lambda r: r.row(),
table.selectionModel().selectedRows()
)
)
def add(self): def add(self):
table = self.find(QTableView, "tableView") table = self.find(QTableView, "tableView")
if len(table.selectedIndexes()) == 0: if len(table.selectedIndexes()) == 0:
@@ -219,6 +258,27 @@ class ProfileWindow(PamhyrWindow):
self.update() self.update()
def purge(self):
self._tablemodel.purge()
self.update()
def purge(self):
try:
dlg = PurgeDialog(
trad=self._trad,
parent=self
)
if dlg.exec():
self._tablemodel.purge(dlg.np_purge)
self._plot.draw()
except Exception as e:
logger_exception(e)
return
def reverse(self):
self._tablemodel.reverse()
self.update()
def _copy(self): def _copy(self):
table = self.find(QTableView, "tableView") table = self.find(QTableView, "tableView")
rows = table.selectionModel().selectedRows() rows = table.selectionModel().selectedRows()
+60
View File
@@ -0,0 +1,60 @@
# PurgeDialog.py -- Pamhyr
# Copyright (C) 2023-2024 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 View.Tools.PamhyrWindow import PamhyrDialog
from PyQt5.QtGui import (
QKeySequence,
)
from PyQt5.QtCore import (
Qt, QVariant, QAbstractTableModel,
)
from PyQt5.QtWidgets import (
QUndoStack, QShortcut, QSpinBox,
)
class PurgeDialog(PamhyrDialog):
_pamhyr_ui = "PurgeOptions"
_pamhyr_name = "Purge"
def __init__(self, trad=None, parent=None):
super(PurgeDialog, self).__init__(
title=trad[self._pamhyr_name],
trad=trad,
options=[],
parent=parent
)
self._init_default_values()
def _init_default_values(self):
self._np_purge = 24
@property
def np_purge(self):
return self._np_purge
def accept(self):
self._np_purge = self.get_spin_box("spinBox_np_purge")
super().accept()
def reject(self):
self.close()
+26 -5
View File
@@ -55,8 +55,8 @@ class GeometryReachTableModel(PamhyrTableModel):
return self._data.profile(index.row()).name return self._data.profile(index.row()).name
if role == Qt.DisplayRole and index.column() == 1: if role == Qt.DisplayRole and index.column() == 1:
kp = self._data.profile(index.row()).kp rk = self._data.profile(index.row()).rk
return f"{kp:.4f}" return f"{rk:.4f}"
if role == Qt.DisplayRole and index.column() == 2: if role == Qt.DisplayRole and index.column() == 2:
return str(self._data.profile(index.row()).nb_points) return str(self._data.profile(index.row()).nb_points)
@@ -94,9 +94,9 @@ class GeometryReachTableModel(PamhyrTableModel):
if index.column() == 1: if index.column() == 1:
self._undo.push( self._undo.push(
SetKPCommand( SetRKCommand(
self._data, index.row(), self._data, index.row(),
self._data.profile(index.row()).kp, self._data.profile(index.row()).rk,
value value
) )
) )
@@ -240,9 +240,30 @@ class GeometryReachTableModel(PamhyrTableModel):
self._undo.push( self._undo.push(
MeshingCommand( MeshingCommand(
self._data, mesher, data self._data, mesher, data, "mesh"
) )
) )
self.layoutAboutToBeChanged.emit() self.layoutAboutToBeChanged.emit()
self.layoutChanged.emit() self.layoutChanged.emit()
def update_rk(self, mesher, data):
self.layoutAboutToBeChanged.emit()
self._undo.push(
MeshingCommand(
self._data, mesher, data, "update_rk"
)
)
self.layoutAboutToBeChanged.emit()
self.layoutChanged.emit()
def purge(self, np_purge):
self._undo.push(
PurgeCommand(
self._data, np_purge
)
)
self.layoutChanged.emit()
Loaded 100 of 354 files, more files were not shown because too many files have changed in this diff. Show more