From b3193d8f5cb4e4202e2537e7d94dfe8371e84ec2 Mon Sep 17 00:00:00 2001
From: jonnymaserati <jonnycorcutt@gmail.com>
Date: Sun, 7 Jul 2024 15:42:39 +0200
Subject: [PATCH] Update quick start example to reflect updated  module.

---
 README.md                  | 25 +++++++++++++++----------
 examples/simple_example.py | 14 ++++++--------
 welleng/visual.py          |  2 +-
 3 files changed, 22 insertions(+), 19 deletions(-)

diff --git a/README.md b/README.md
index 8841f56..3f17e61 100644
--- a/README.md
+++ b/README.md
@@ -197,6 +197,7 @@ Here's an example using `welleng` to construct a couple of simple well trajector
 import welleng as we
 from tabulate import tabulate
 
+
 # construct simple well paths
 print("Constructing wells...")
 connector_reference = we.survey.from_connections(
@@ -207,7 +208,7 @@ connector_reference = we.survey.from_connections(
         pos2=[-100., 0., 2000.],
         inc2=90,
         azi2=60,
-        ),
+    ),
     step=50
 )
 
@@ -266,16 +267,20 @@ c = we.clearance.Clearance(
 )
 
 print("Calculating ISCWSA clearance...")
-clearance_ISCWSA = we.clearance.ISCWSA(c)
+clearance_ISCWSA = we.clearance.IscwsaClearance(
+    survey_reference, survey_offset
+)
 
 print("Calculating mesh clearance...")
-clearance_mesh = we.clearance.MeshClearance(c, sigma=2.445)
+clearance_mesh = we.clearance.MeshClearance(
+    survey_reference, survey_offset, sigma=2.445
+)
 
 # tabulate the Separation Factor results and print them
 results = [
     [md, sf0, sf1]
     for md, sf0, sf1
-    in zip(c.reference.md, clearance_ISCWSA.SF, clearance_mesh.SF)
+    in zip(c.reference.md, clearance_ISCWSA.sf, clearance_mesh.sf)
 ]
 
 print("RESULTS\n-------")
@@ -285,12 +290,12 @@ print(tabulate(results, headers=['md', 'SF_ISCWSA', 'SF_MESH']))
 lines = we.visual.get_lines(clearance_mesh)
 
 # plot the result
-we.visual.plot(
-    [mesh_reference.mesh, mesh_offset.mesh],  # list of meshes
-    names=['reference', 'offset'],  # list of names
-    colors=['red', 'blue'],  # list of colors
-    lines=lines
-)
+plot = we.visual.Plotter()
+plot.add(mesh_reference, c='red')
+plot.add(mesh_offset, c='blue')
+plot.add(lines)
+plot.show()
+plot.close()
 
 print("Done!")
 
diff --git a/examples/simple_example.py b/examples/simple_example.py
index fed9997..258b579 100644
--- a/examples/simple_example.py
+++ b/examples/simple_example.py
@@ -13,9 +13,7 @@
 '''
 import welleng as we
 from tabulate import tabulate
-# import os
 
-# os.environ['DISPLAY'] = ':1'
 
 # construct simple well paths
 print("Constructing wells...")
@@ -109,11 +107,11 @@
 lines = we.visual.get_lines(clearance_mesh)
 
 # plot the result
-we.visual.plot(
-    [mesh_reference.mesh, mesh_offset.mesh],  # list of meshes
-    names=['reference', 'offset'],  # list of names
-    colors=['red', 'blue'],  # list of colors
-    lines=lines
-)
+plot = we.visual.Plotter()
+plot.add(mesh_reference, c='red')
+plot.add(mesh_offset, c='blue')
+plot.add(lines)
+plot.show()
+plot.close()
 
 print("Done!")
diff --git a/welleng/visual.py b/welleng/visual.py
index f052b42..f81055a 100644
--- a/welleng/visual.py
+++ b/welleng/visual.py
@@ -353,7 +353,7 @@ def get_lines(clearance):
     c = clearance.sf
     start_points, end_points = clearance.get_lines()
     lines = Lines(start_points, end_points).cmap('hot_r', c, on='cells')
-    lines.addScalarBar(title='SF')
+    lines.add_scalarbar(title='SF')
 
     return lines