gimu 0.5.0__tar.gz → 0.6.0__tar.gz

{gimu-0.5.0 → gimu-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gimu
-Version: 0.5.0
+Version: 0.6.0
 Summary: A toolkit and python library for modelling at Geothermal Institute, University of Auckland.
 Project-URL: Documentation, https://github.com/cyeh015/gimu/blob/main/README.md
 Project-URL: Issues, https://github.com/cyeh015/gimu/issues

gimu-0.6.0/src/gimu/gmf/boundary_conditions.py

@@ -0,0 +1,64 @@
+""" handle GMF's boundary condition
+
+
+
+"""
+
+def cell2block(geo, cell):
+    """ returns mulgrid block name (str) of a cell id (int) """
+    return geo.block_name_list[geo.num_atmosphere_blocks:][cell]
+
+def update_waiwera_model_heat_bc(geo, wjson, bcjson):
+    """ bcjson is loaded from the GMF modifier _boundary_conditions.json
+    """
+    # heat component
+    if wjson['eos']['name'] == 'wae':
+        heat_components = [3, 'energy']
+    else:
+        raise NotImplementedError(f"(gimu)    EOS {wjson['eos']['name']} not implemented")
+
+
+    if 'Heat' not in bcjson:
+        return
+    heat_bc = bcjson['Heat']
+    default_flux = float(heat_bc['Default']['Heat_Flux'])
+    count = 0
+    for src in wjson['source']:
+        if 'component' in src:
+            if src['component'] in heat_components:
+                block_name = cell2block(geo, src['cell'])
+                column_name = geo.column_name(block_name)
+                area = geo.column[column_name].area
+                flux = default_flux
+                for zone_name, zone in heat_bc.get('Zones', {}).items():
+                    if column_name in zone['Columns']:
+                        flux = float(zone['Heat_Flux'])
+                        break
+                src['rate'] = area * flux
+                count += 1
+                # print(f"Updated {src['name']} heatflow")
+    print(f"(gimu)    Updated heatflow in {count} sources")
+
+
+def update_aut2_model_heat_bc(geo, dat, bcjson):
+    if 'Heat' not in bcjson:
+        return
+    heat_bc = bcjson['Heat']
+    default_flux = float(heat_bc['Default']['Heat_Flux'])
+    count = 0
+    for gen in dat.generatorlist:
+        if gen.type == 'HEAT':
+            block_name = gen.block
+            column_name = geo.column_name(block_name)
+            area = geo.column[column_name].area
+            flux = default_flux
+            for zone_name, zone in heat_bc.get('Zones', {}).items():
+                if column_name in zone['Columns']:
+                    flux = float(zone['Heat_Flux'])
+                    break
+            gen.gx = area * flux
+            count += 1
+            # print(f"Updated {gen.name} heatflow")
+    print(f"(gimu)    Updated heatflow in {count} generators")
+
+

{gimu-0.5.0 → gimu-0.6.0}/src/gimu/gmf/modifier.py

@@ -99,14 +99,14 @@ def compute_rocktypes(rocktype_json, warn=True):
                     rt['conductivity'] = [rocktype_json[base_name]['conductivity'][0] * c_modifier]
                 else:
                     if 'conductivity' in rt:
-                        print(f"Rocktype {rt_name} remove conductivity because no conductivity in base rocktype {base_name}")
+                        print(f"(gimu)    Rocktype {rt_name} remove conductivity because no conductivity in base rocktype {base_name}")
                         del rt['conductivity']
             else:
                 msg = f"Base rocktype {base_name} not found for rocktype {rt_name}"
                 if warn:
                     raise Exception(msg)
                 else:
-                    print("Warning: " + msg)
+                    print("(gimu)    Warning: " + msg)
                     rt['error'] = msg
 
 
@@ -116,10 +116,10 @@ def update_waiwera_model_rocktypes(wjson, rtjson, skip_missing=True):
     for rt in wjson['rock']['types']:
         if rt['name'] not in rtjson:
             if skip_missing:
-                print(f"Rocktype {rt['name']} not found in _rocktypes.json... skipped")
+                print(f"(gimu)    Rocktype {rt['name']} not found in _rocktypes.json... skipped")
                 continue
             else:
-                raise Exception(f"Rocktype {rt['name']} not found in _rocktypes.json")
+                raise Exception(f"(gimu)    Rocktype {rt['name']} not found in _rocktypes.json")
         modifier = rtjson[rt['name']]
         rt['permeability'] = modifier['permeability']
         rt['porosity'] = modifier['porosity'][0]
@@ -133,10 +133,10 @@ def update_aut2_model_rocktypes(dat, rtjson, skip_missing=True):
     for rt in dat.grid.rocktypelist:
         if rt.name not in rtjson:
             if skip_missing:
-                print(f"Rocktype {rt.name} not found in _rocktypes.json... skipped")
+                print(f"(gimu)    Rocktype {rt.name} not found in _rocktypes.json... skipped")
                 continue
             else:
-                raise Exception(f"Rocktype {rt.name} not found in _rocktypes.json")
+                raise Exception(f"(gimu)    Rocktype {rt.name} not found in _rocktypes.json")
         modifier = rtjson[rt.name]
         rt.permeability = modifier['permeability']
         rt.porosity = modifier['porosity'][0]