pro-well-plan / well_profile Goto Github PK

import wellpathpy
import welleng as we
import csv
from welleng.survey import Survey, SurveyHeader

# this is the ISCWSA dataset
filename = (
    "reference/standard-set-of-wellpaths"
    "-for-evaluating-clearance-scenarios-r4-17-may-2017.xlsx"
)

print("Loading data...")
try:
    data = we.io.import_iscwsa_collision_data(filename)
except:
    print(
        "Make sure you've updated filename to your local copy of ISCWSA's"
        " clearance scenarios"
    )

# Make a dictionary of surveys with welleng
print("Making surveys...")
surveys = {}
for well in data["wells"]:
    surveys[well] = {}
    sh = SurveyHeader(
        name=well,
        latitude=60.,
        b_total=50000.,
        dip=70.,
        declination=.0,
        convergence=0.,
        G=9.80665,
        azi_reference='grid'
    )
    if well == "Reference well":
        radius = 0.4572
    else:
        radius = 0.3048

    s = Survey(
        md=data["wells"][well]["MD"],
        inc=data["wells"][well]["IncDeg"],
        azi=data["wells"][well]["AziDeg"],
        deg=True,
        unit="meters"
    )
    surveys[well]['survey'] = s

    surveys[well]['errors'] = {
        'tvd': data["wells"][well]["TVD"] - s.tvd,
        'n': data["wells"][well]["N"] - s.n,
        'e': data["wells"][well]["E"] - s.e,
    }

# make a dictionary of wells with wellpathpy
surveys_wellpathpy = {}
for well in data['wells']:
    tvd, northing, easting, dls = wellpathpy.mincurve.minimum_curvature(
        md=data["wells"][well]["MD"],
        inc=data["wells"][well]["IncDeg"],
        azi=data["wells"][well]["AziDeg"],
    )

    surveys_wellpathpy[well] = {
        'tvd': tvd,
        'n': northing,
        'e': easting,
        'dls': dls,
        'errors_base': {
            'tvd': data["wells"][well]["TVD"] - tvd,
            'n': data["wells"][well]["N"] - northing,
            'e': data["wells"][well]["E"] - easting,
        },
        'errors_welleng': {
            'tvd': surveys[well]["survey"].tvd - tvd,
            'n': surveys[well]["survey"].n - northing,
            'e': surveys[well]["survey"].e - easting,
        }
    }

# generate a csv files to import into PWP web app
for well in data["wells"]:
    s_out = [['md', 'inc', 'azi']]
    s_out.extend([
        [md, inc, azi]
        for md, inc, azi in zip(
            data["wells"][well]["MD"],
            data["wells"][well]["IncDeg"],
            data["wells"][well]["AziDeg"],
        )
    ])

    with open(f'{well}.csv', 'w') as csv_file:
        writer = csv.writer(csv_file)
        for row in s_out:
            writer.writerow(row)

print("Done")

data = [md, inclination, azimuth, tvd]

if type(data[0]) is list:
    md, inc, azi, tvd = data

import numpy as np

if type(data[0]) is list or isinstance(data[0], np.ndarray):
    md, inc, azi, tvd = data