add norne example, needs opm-data

examples/norne.py

@@ -0,0 +1,45 @@
+import sys
+from os.path import isdir, join
+import sunbeam
+from datetime import datetime as dt
+
+def opmdatadir():
+    global OPMDATA_DIR
+    if isdir(OPMDATA_DIR):
+        return OPMDATA_DIR
+    if len(sys.argv) < 2:
+        return None
+    d = sys.argv[1]
+    if isdir(d) and isdir(join(d, 'norne')):
+        return d
+    return None
+
+def haveopmdata():
+    return opmdatadir() is not None
+
+def parse(fname):
+    s = dt.now()
+    es = sunbeam.parse(fname, ('PARSE_RANDOM_SLASH', sunbeam.action.ignore))
+    e = dt.now()
+    print('Parsing took %s sec' % (e - s).seconds)
+    return es
+
+def main():
+    es = parse(join(opmdatadir(), 'norne/NORNE_ATW2013.DATA'))
+    sc = es.schedule
+    wp = sc.wells[23] # producer
+    wi = sc.wells[20] # injector at ts 100
+    print('state:     %s' % es)
+    print('schedule:  %s' % sc)
+    print('at timestep 100 (%s)' % sc.timesteps[100])
+    print('prod well: %s' % wp)
+    print('inj  well: %s' % wi)
+    print('pos:       %s' % list(wp.pos()))
+
+if __name__ == '__main__':
+    global OPMDATA_DIR
+    OPMDATA_DIR = '../../opm-data'
+    if haveopmdata():
+        main()
+    else:
+        print('Need to have path "%s" or give opm-data as argument' % OPMDATA_DIR)