Yes, I want to add my OuputSHPFile (contour feature class - param3) to my currend MXD, the script is described below.
So instead of expressing "OutputSHPFile", I should write "str = inputWorkspace + "\\" + OutputSHPFile", or something like that ? Because I don't want to use the path of the class, to allow my script to be used by other users.
Thank you very much.
import arcpy
import os
class Toolbox(object):
def __init__(self):
"""Define the toolbox (the name of the toolbox is the name of the
.pyt file)."""
self.label = "Clip and contour Analysis"
self.alias = ""
# List of tool classes associated with this toolbox
self.tools = [Tool]
class Tool(object):
def __init__(self):
"""Define the tool (tool name is the name of the class)."""
self.label = "clip and contour"
self.description = ""
self.canRunInBackground = False
def getParameterInfo(self):
"""Define parameter definitions"""
param0 = arcpy.Parameter(
displayName="Select shapefile (output extent):",
name="in_SourceSHP",
datatype="GPFeatureLayer",
parameterType="Required",
direction="Input")
param1 = arcpy.Parameter(
displayName="Raster",
name="in_TargetRaster",
datatype="GPRasterLayer",
parameterType="Required",
direction="Input")
param2 = arcpy.Parameter(
displayName="Raster clip",
name="out_raster",
datatype="DERasterDataset",
parameterType="Required",
direction="Output")
param3 = arcpy.Parameter(
displayName="Contour",
name="out_features",
datatype="GPFeatureLayer",
parameterType="Required",
direction="Output")
param4 = arcpy.Parameter(
displayName="Interval contour:",
name="in_Interval",
datatype="GPDouble",
parameterType="Required",
direction="Input")
param4.value = 0.5
params = [param0, param1, param2, param3, param4]
return params
def execute(self, parameters, messages):
"""The source code of the tool."""
mxd =arcpy.mapping.MapDocument("CURRENT")
Interval = float(parameters[4].valueAsText)
OutputSHPFile = parameters[3].valueAsText
RasterFile = parameters[1].valueAsText
OutRasterFile = parameters[2].valueAsText
ShapeFileName = parameters[0].valueAsText
# Process : Clip
arcpy.Clip_management(RasterFile, "#", OutRasterFile, ShapeFileName, "0", "ClippingGeometry")
# Process: Contour
arcpy.Contour_3d(OutRasterFile, OutputSHPFile, Interval, "0", "1")
# Add shp to mxd
mxd =arcpy.mapping.MapDocument("CURRENT")
df = arcpy.mapping.ListDataFrames(mxd)[0]
addLayer = arcpy.mapping.Layer(OutputSHPFile)
arcpy.mapping.AddLayer(df, addLayer, "TOP")
arcpy.RefreshActiveView()
arcpy.RefreshTOC()
