fargopy 0.2.0__py3-none-any.whl → 0.3.0__py3-none-any.whl

fargopy/sys.py

@@ -27,6 +27,7 @@ class Sys(object):
     QERROR = True
     STDERR = ''
     STDOUT = ''
+    OUT = ''
 
     @staticmethod
     def run(cmd,quiet=True):
@@ -35,8 +36,8 @@ class Sys(object):
         Parameters:
             cmd: string:
                 Command to run
-            verbose: boolean, default = True:
-                When True the output of the command is shown.
+            quiet: boolean, default = True:
+                When False the output of the command is shown.
             
         Output:
             error: integer:
@@ -82,6 +83,7 @@ class Sys(object):
             elif Sys.QERROR==0:
                 fargopy.Debug.trace(f"sysrun::Done. You're great. Check Sys.STDOUT for details.")
         
+        Sys.OUT = out
         return Sys.QERROR,out
 
     @staticmethod
@@ -99,23 +101,44 @@ class Sys(object):
     def simple(cmd):
         return os.system(cmd)
 
-    @staticmethod
-    def compile_fargo3d(self,clean=True):
-        if Conf._check_fargo(Conf.FARGO3D_FULLDIR):    
-            if clean:
-                Sys.sysrun(f'make -C {Conf.FARGO3D_FULLDIR} clean mrproper')
-            
-            error,out = Sys.sysrun(f'make -C {Conf.FARGO3D_FULLDIR} PARALLEL={self.parallel} GPU={self.gpu}',verbose=False)
-            if error:
-                if not Conf._check_fargo_binary(Conf.FARGO3D_FULLDIR,quiet=True):
-                    print("An error compiling the code arose. Check dependencies.")
-                    print(Sys.STDERR)
-                return False
-            
-            return True
-    
     @staticmethod
     def get_memory():
         svmem = psutil.virtual_memory()
         return svmem
-    
+
+    @staticmethod
+    def lock(pid):
+        """Create a lock file
+        """
+        if Sys.check_lock(verbose=False):
+            print(f"You cannot lock FARGO3D because it is already locked")
+            return
+        lock_file = open(fargopy.FP_FARGO3D_LOCKFILE,"w")
+        lock_file.write(pid)
+        lock_file.close()
+        
+    @staticmethod
+    def unlock():
+        """Unlock a fargopy instance
+        """
+        if Sys.check_lock(verbose=False):
+            print(f"Unlocking FARGO3D")
+            # Kill process 
+            error,output = Sys.run(f"kill -9 {pid}")
+            if error:
+                print(f"The locking process {pid} does not exist.")
+            else:
+                print(f"The locking process {pid} has been killed.")
+            # Remove locking file 
+            print(f"Removing locking file")
+            error,output = Sys.run(f"rm -rf {fargopy.FP_FARGO3D_LOCKFIL}")
+
+    @staticmethod
+    def check_lock(verbose=True):
+        """Check if fargopy is locked by a running process
+        """
+        if os.path.isfile(fargopy.FP_FARGO3D_LOCKFILE):
+            if verbose:
+                print(f"There is a lockfile in {fargopy.FP_FARGO3D_LOCKFILE}")
+            return True
+        return False

fargopy/util.py

@@ -8,6 +8,10 @@ import fargopy
 ###############################################################
 import matplotlib.pyplot as plt
 
+###############################################################
+# Constants
+###############################################################
+
 ###############################################################
 # Classes
 ###############################################################

fargopy/version.py

@@ -1 +1 @@
-version='0.2.0'
+version='0.3.0'

fargopy-0.3.0.dist-info/METADATA

@@ -0,0 +1,243 @@
+Metadata-Version: 2.1
+Name: fargopy
+Version: 0.3.0
+Summary: FARGO3D Wrapping
+Home-page: https://pypi.org/project/fargopy
+Author: Jorge Zuluaga, Matias Montesinos
+Author-email: jorge.zuluaga@gmail.com
+License: MIT
+Keywords: astronomy MHD CFD
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: scipy
+Requires-Dist: matplotlib
+Requires-Dist: tqdm
+Requires-Dist: numpy
+Requires-Dist: ipython
+Requires-Dist: celluloid
+Requires-Dist: psutil
+Requires-Dist: gdown
+
+# FARGOpy
+## Wrapping FRAGO3D
+
+<!-- This are visual tags that you may add to your package at the beginning with useful information on your package --> 
+[![version](https://img.shields.io/pypi/v/fargopy?color=blue)](https://pypi.org/project/fargopy/)
+[![downloads](https://img.shields.io/pypi/dw/fargopy)](https://pypi.org/project/fargopy/)
+
+`FARGOpy` is a python wrapping for [`FARGO3D`](https://fargo3d.bitbucket.io/intro.html)., the well-knwon hydrodynamics and magnetohydrodynamics parallel code. This wrapping is intended to ensue the interaction with FARGO3D especially for those starting using the code, for instance for teaching and training purposes, but also provide functionalities for most advanced users in tasks related to the postprocessing of output files and plotting.
+
+This is an example of what can be done using `FARGOpy`:
+
+<p align="center"><img src="https://github.com/seap-udea/fargopy/blob/main/gallery/fargo-animation.gif?raw=true" alt="Animation""/></p>
+
+## Install `FARGOpy` 
+
+To install the package run:
+
+```bash
+$ pip3 install fargopy
+```
+
+Since `FARGOpy` is a python wrap for `FARGO3D` the ideal environment to work with the package is `IPython` or other similar environments such as `jupyter-lab`, `Google Colab`, etc. 
+
+If you are working on a Linux server it is better to run the package for the first time using the `IPython` initialization command:
+
+```bash
+$ ifargopy
+```
+
+The first time you run this command, it will create the configuration directory `$HOME/.fargopy`. This directory contains a set of basic configuration variables `$HOME/.fargopy/fargopyrc`, that can be customized, and the `IPython` initialization script `$HOME/.fargopy/ifargopy.py`.
+
+If you are working in `Jupyter` or in `Google Colab`, the configuration directory and its content will be crated the first time you import the package:
+
+
+```python
+import fargopy as fp
+%load_ext autoreload
+%autoreload 2
+```
+
+    Running FARGOpy version 0.2.1
+
+
+## Download and install FARGO3D
+
+`FARGOpy` can be used either to *control* `FARGO3D`, ie. download, compile and run it, and/or read and manipulate the output of simulations.  In either case it would be convenient to obtain a copy of the package and all its prerequisites. For a detailed guide please see the [FARGO documentation](https://fargo3d.bitbucket.io/index.html) or the [project repo at bitbucket](https://bitbucket.org/fargo3d/public/src/ae0fcdc67bb7c83aed85fc9a4d4a2d5061324597/?at=release%2Fpublic). 
+
+> **NOTE**: It is important to understand that `FARGO3D` works especially well on Linux plaforms (including `MacOS`). The same condition applies for `FARGOpy`. Because of that, most internal as well as public features of the packages are designed to work in a `Linux` environment. For working in another operating systems, for instance for teaching or training purposes, please consider to use virtual machines.
+
+`FARGOpy` provides a simple way to get the latest version of `FARGO3D`. In the terminal run:
+
+```shell
+$ ifargopy download
+```
+
+A copy of `FARGO3D` will be download in the HOME directory `~/fargo3d/`.
+
+If you are working in a `Jupyter` environment you may run:
+
+
+```python
+fp.initialize('download')
+```
+
+    Downloading FARGOpy...
+    	FARGO3D directory already present in '/home/jzuluaga/fargo3d/'
+    Header file for FARGO3D is in the fargo directory /home/jzuluaga/fargo3d/
+
+
+> **NOTE**: As you may see the default directory where FARGO3D is download is `~/` and the name of the directory is `fargo3d`. You may change this by setting the configuration variables `FP_FARGO3D_BASEDIR` or `FP_FARGO3D_PACKDIR` brefore downloading the package.
+
+## Quickstart
+
+Here we will illustrate the minimal commands you may run to test the package. A more detailed set of examples can be found in [this file](TUTORIAL.md). 
+
+For this example we will assume that you already have a set of FARGO3D simulation results. You may download a precomputed set of results prepared by the developers of `FARGOpy` using the command: 
+
+
+```python
+fp.Simulation.download_precomputed(setup='fargo')
+```
+
+    Precomputed output directory '/tmp/fargo' already exist
+
+
+
+
+
+    '/tmp/fargo'
+
+
+
+Create a simulation object:
+
+
+```python
+sim = fp.Simulation()
+```
+
+    Your simulation is now connected with '/home/jzuluaga/fargo3d/'
+
+
+Set the output directory (in this case, the directory where the precomputed simulation has been stored):
+
+
+```python
+sim.set_output_dir('/tmp/fargo')
+```
+
+    Now you are connected with output directory '/tmp/fargo'
+
+
+Load the properties of the simulation:
+
+
+```python
+sim.load_properties()
+```
+
+    Loading variables
+    84 variables loaded
+    Simulation in 2 dimensions
+    Loading domain in cylindrical coordinates:
+    	Variable phi: 384 [[0, -3.1334114227210694], [-1, 3.1334114227210694]]
+    	Variable r: 128 [[0, 0.408203125], [-1, 2.491796875]]
+    	Variable z: 1 [[0, 0.0], [-1, 0.0]]
+    Number of snapshots in output directory: 51
+    Configuration variables and domains load into the object. See e.g. <sim>.vars
+
+
+Load gas density from a given snapshot:
+
+
+```python
+gasdens = sim.load_field('gasdens',snapshot=20)
+```
+
+Create a `meshslice` of the field:
+
+
+```python
+gasdens_r, mesh = gasdens.meshslice(slice='z=0,phi=0')
+```
+
+And plot the slice:
+
+
+```python
+import matplotlib.pyplot as plt
+plt.ioff() # Drop this out of this tutorial
+fig,ax = plt.subplots()
+ax.semilogy(mesh.r,gasdens_r)
+ax.set_xlabel(r"$r$ [cu]")
+ax.set_ylabel(r"$\rho$ [cu]")
+fp.Util.fargopy_mark(ax)
+fig.savefig('gallery/example-dens_r.png') # Drop this out of this tutorial
+```
+
+<p align="center"><img src="https://github.com/seap-udea/fargopy/blob/main/gallery/example-dens_r.png?raw=true" alt="Animation""/></p>
+
+You may also create a 2-dimensional slice:
+
+
+```python
+gasdens_plane, mesh = gasdens.meshslice(slice='z=0')
+```
+
+And plot it:
+
+
+```python
+plt.ioff() # Drop this out of this tutorial
+fig,axs = plt.subplots(1,2,figsize=(12,6))
+
+ax = axs[0]
+ax.pcolormesh(mesh.phi,mesh.r,gasdens_plane,cmap='prism')
+ax.set_xlabel('$\phi$ [rad]')
+ax.set_ylabel('$r$ [UL]')
+fp.Util.fargopy_mark(ax)
+
+ax = axs[1]
+ax.pcolormesh(mesh.x,mesh.y,gasdens_plane,cmap='prism')
+ax.set_xlabel('$x$ [UL]')
+ax.set_ylabel('$y$ [UL]')
+fp.Util.fargopy_mark(ax)
+ax.axis('equal')
+fig.savefig('gallery/example-dens_disk.png') # Drop this out of this tutorial
+```
+
+<p align="center"><img src="https://github.com/seap-udea/fargopy/blob/main/gallery/example-dens_disk.png?raw=true" alt="Animation""/></p>
+
+## What's new
+
+
+Version 0.3.*:
+
+- Refactoring of initializing routines.
+- Improvements in documentation of basic classes in `__init__.py`.
+
+Version 0.2.*:
+
+- First real applications tested with FARGOpy.
+- All basic routines for reading output created.
+- Major refactoring. 
+
+Version 0.1.*:
+
+- Package is now provided with a script 'ifargopy' to run 'ipython' with fargopy initialized.
+- A new 'progress' mode has been added to status method.
+- All the dynamics of loading/compiling/running/stoppìng/resuming FARGO3D has been developed.
+
+Version 0.0.*:
+
+- First classes created.
+- The project is started!
+
+------------
+
+This package has been designed and written mostly by Jorge I. Zuluaga with advising and contributions by Matías Montesinos (C) 2023
+

fargopy-0.3.0.dist-info/RECORD

@@ -0,0 +1,13 @@
+fargopy/__init__.py,sha256=q_tpw5_Mi4zmowrMUxQFrhHlTPVCUsmU3RX88-pRBE8,9326
+fargopy/fields.py,sha256=-f0Q-3J1dq2oZYpmiaAuT-B48qECCLhvmjiIUSjyXhg,8700
+fargopy/simulation.py,sha256=NPZes8MZ8xtbmfZqrhuGAGhYvEs2vUdJSpRFNB1hfjo,32901
+fargopy/sys.py,sha256=-RnUZJ5GnYHwIlI7UtjPlnyS88ptnrSBa9FvnF_2gDM,4311
+fargopy/util.py,sha256=y39aTcvK1lM3vbuK11lE2bfj-eCcOzc0lKMvyBRW2-M,1584
+fargopy/version.py,sha256=14vICHbg-AX_96xiJkVH7VdMy15awdqU6glK1dWQttc,16
+fargopy-0.3.0.data/scripts/ifargopy,sha256=t5E07jKzOLtNnlNyElQZbzjCW6wnR0bc8uCaZpw9yS8,382
+fargopy-0.3.0.dist-info/LICENSE,sha256=aIckKnNVrkXQMqEKlzGn_REfTwc82TF35BHMCRwPXCI,1097
+fargopy-0.3.0.dist-info/METADATA,sha256=nUDcNFKH-P0X7UM_cn0JuEaPIpFANqb1wPp_ahdkfeg,7823
+fargopy-0.3.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+fargopy-0.3.0.dist-info/entry_points.txt,sha256=555NPKYbLCN0fgJbdW4b4azZ5sqjhqVqTUxujBBYEJY,49
+fargopy-0.3.0.dist-info/top_level.txt,sha256=_r66v1_-9T7dB5sQa12AdxaAdsv9-OTwtR-vft4OPBU,8
+fargopy-0.3.0.dist-info/RECORD,,

{fargopy-0.2.0.dist-info → fargopy-0.3.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.41.3)
+Generator: bdist_wheel (0.42.0)
 Root-Is-Purelib: true
 Tag: py3-none-any