geoloop 0.0.1__py3-none-any.whl → 1.0.0b1__py3-none-any.whl

geoloop/bin/__init__.py

@@ -0,0 +1,3 @@
+"""
+This package contains the main scripts to run for using Geoloop.
+"""

geoloop/cli/batch.py

@@ -0,0 +1,106 @@
+import json
+from pathlib import Path
+
+import typer
+
+from geoloop.bin.Runbatch import run_batch_from_json
+
+batch_app = typer.Typer(help="Batch processing commands for geoloop.")
+
+
+@batch_app.command("run")
+def batch_run(batchfile: Path = typer.Argument(..., help="Path to batch JSON file.")):
+    """Run a batch JSON file."""
+    run_batch_from_json(str(batchfile))
+
+
+@batch_app.command("create")
+def batch_create(
+    output: Path = typer.Argument(..., help="Where to save the new batch JSON file."),
+    stochastic_run: list[str] = typer.Option([], help="Runmain config files."),
+    single_run: list[str] = typer.Option([], help="SingleRunSim config files."),
+    process_lithology: list[str] = typer.Option([], help="Lithology config files."),
+    calculate_loadprofile: list[str] = typer.Option(
+        [], help="Loadprofile config files."
+    ),
+    calculate_flowdata: list[str] = typer.Option([], help="FlowData config files."),
+    plot: list[str] = typer.Option([], help="Plotmain config files."),
+):
+    """
+    Create a batch JSON non-interactively.
+
+    Example:
+        geoloop batch create batch.json --stochastic-run a.json --plot p.json
+    """
+
+    commands = []
+
+    for file in stochastic_run:
+        commands.append({"command": "Runmain", "args": file})
+
+    for file in single_run:
+        commands.append({"command": "SingleRunSim", "args": file})
+
+    for file in process_lithology:
+        commands.append({"command": "Lithology", "args": file})
+
+    for file in calculate_loadprofile:
+        commands.append({"command": "Loadprofile", "args": file})
+
+    for file in calculate_flowdata:
+        commands.append({"command": "FlowData", "args": file})
+
+    for file in plot:
+        commands.append({"command": "Plotmain", "args": file})
+
+    output.parent.mkdir(parents=True, exist_ok=True)
+    output.write_text(json.dumps({"commands": commands}, indent=2))
+
+    typer.secho(f"Batch file created: {output}", fg=typer.colors.GREEN)
+
+
+@batch_app.command("wizard")
+def batch_wizard():
+    """
+    Interactive wizard to build a batch JSON file.
+    """
+
+    typer.echo("Welcome to the Geoloop Batch Wizard!")
+    typer.echo("Let's build your batch sequence.\n")
+
+    commands = []
+    while True:
+        typer.echo("\nSelect command to add:")
+        choice = typer.prompt(
+            "Options: stochastic-run, single-run, process-lithology, calculate-loadprofile, calculate-flowdata, plot, done",
+            type=str,
+        )
+
+        if choice.lower() == "done":
+            break
+
+        valid = {
+            "stochastic-run": "Runmain",
+            "single-run": "SingleRunSim",
+            "process-lithology": "Lithology",
+            "calculate-loadprofile": "Loadprofile",
+            "calculate-flowdata": "FlowData",
+            "plot": "Plotmain",
+        }
+
+        if choice.lower() not in valid:
+            typer.secho("Invalid command type!", fg=typer.colors.RED)
+            continue
+
+        filepath = typer.prompt("Enter path to the config JSON file:")
+
+        commands.append({"command": valid[choice.lower()], "args": filepath})
+
+        typer.secho("Command added!\n", fg=typer.colors.GREEN)
+
+    output_file = typer.prompt("Where should the batch JSON be saved?")
+    output = Path(output_file)
+
+    output.write_text(json.dumps({"commands": commands}, indent=2))
+
+    typer.secho(f"\nBatch file saved: {output}", fg=typer.colors.GREEN)

geoloop/cli/main.py

@@ -0,0 +1,105 @@
+import typer
+
+from geoloop.bin.Flowdatamain import main_flow_data
+from geoloop.bin.Lithologymain import main_lithology
+from geoloop.bin.Loadprofilemain import main_load_profile
+from geoloop.bin.Plotmain import main_plotmain
+from geoloop.bin.Runbatch import run_batch_from_json
+from geoloop.bin.Runmain import main_runmain
+from geoloop.bin.SingleRunSim import main_single_run_sim
+from geoloop.cli.batch import batch_app
+
+# Set up the main app
+simulation = typer.Typer(help="Geoloop simulation CLI.")
+simulation.add_typer(batch_app, name="batch")
+
+
+@simulation.command(name="batch-run")
+def batch_run(
+    config_file_path: str = typer.Argument(
+        ...,
+        help="Path to the Json file with a sequence of the command scripts and configuration arguments",
+    ),
+) -> None:
+    """
+    Run a batch of scripts, in sequence, for BHE simulation.
+    """
+    run_batch_from_json(config_file_path)
+
+
+@simulation.command(name="single-run")
+def single_run(
+    config_file_path: str = typer.Argument(
+        ..., help="Path to the Json file with the main model configuration parameters"
+    ),
+) -> None:
+    """
+    Run main script for a single BHE simulation.
+    """
+    main_single_run_sim(config_file_path)
+
+
+@simulation.command(name="stochastic-run")
+def stochastic_run(
+    config_file_path: str = typer.Argument(
+        ..., help="Path to the Json file with the main model configuration parameters"
+    ),
+) -> None:
+    """
+    Run main script for a stochastic BHE simulation.
+    """
+    main_runmain(config_file_path)
+
+
+@simulation.command(name="process-lithology")
+def process_lithology(
+    config_file_path: str = typer.Argument(
+        ...,
+        help="Path to the Json file with the subsurface model configuration parameters",
+    ),
+) -> None:
+    """
+    Run main script for calculating and plotting subsurface thermal properties.
+    """
+    main_lithology(config_file_path)
+
+
+@simulation.command(name="calculate-loadprofile")
+def calculate_loadprofile(
+    config_file_path: str = typer.Argument(
+        ...,
+        help="Path to the Json file with the subsurface model configuration parameters",
+    ),
+) -> None:
+    """
+    Run main script for calculating a time-profile of heat load.
+    """
+    main_load_profile(config_file_path)
+
+
+@simulation.command(name="calculate-flowdata")
+def calculate_flowdata(
+    config_file_path: str = typer.Argument(
+        ..., help="Path to the Json file with the flow data configuration parameters"
+    ),
+) -> None:
+    """
+    Run main script for calculating a time-profile of flow rate.
+    """
+    main_flow_data(config_file_path)
+
+
+@simulation.command(name="plot")
+def plot_results(
+    config_file_path: str = typer.Argument(
+        ..., help="Path to the Json file with the plotting configuration parameters"
+    ),
+) -> None:
+    """
+    Run main script for plotting the simulation results.
+    """
+    main_plotmain(config_file_path)
+
+
+if __name__ == "__main__":
+    simulation()