fix terrain magic bug

Author: delebash

Content/Blueprints/ImportMap/Mapbox_BP.uasset

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6fc8a760f1e19225baceb1d614cc65ecf9219a50e85f2b532a585099f32293d8
-size 584909
+oid sha256:85908d2835e782b8cbbab5845d884457bee0ec5614e023e1987d17efaf3eb649
+size 730981

Resources/main.py

@@ -4,11 +4,11 @@
 alpha_brush_path = sys.argv[1]
 alpha_brush_name = sys.argv[2]
 heightmap_directory = sys.argv[3]
-alpha_textures_path = alpha_brush_path + "Textures/"
-
-# Duplicate stamp brush
-brush_template = "/Game/Brushes/PEAKS/Peak_10_brush"
-unreal.EditorAssetLibrary.duplicate_asset(brush_template, alpha_brush_path + alpha_brush_name)
+alpha_brush_template_path = sys.argv[4]
+alpha_textures = sys.argv[5]
+heightmap_property = sys.argv[6]
+stamp_tool = sys.argv[7]
+alpha_textures_path = alpha_brush_path + alpha_textures
 
 # Import known images as a list of Texture2D objects
 alpha_texture_png = heightmap_directory + "\\" + alpha_brush_name + ".png"
@@ -18,13 +18,17 @@
 data.set_editor_property('filenames', [alpha_texture_png])
 lst_texture2D = unreal.AssetToolsHelpers.get_asset_tools().import_assets_automated(data)
 
-# Load texture
-blueprint_generated = unreal.EditorAssetLibrary.load_blueprint_class(alpha_brush_path + alpha_brush_name)
-texture = unreal.EditorAssetLibrary.load_asset(alpha_textures_path + alpha_brush_name)
+if stamp_tool == 'Unreal Stamp Brush Plugin':
+    # Duplicate stamp brush
+    unreal.EditorAssetLibrary.duplicate_asset(alpha_brush_template_path, alpha_brush_path + alpha_brush_name)
+
+    # Load texture
+    blueprint_generated = unreal.EditorAssetLibrary.load_blueprint_class(alpha_brush_path + alpha_brush_name)
+    texture = unreal.EditorAssetLibrary.load_asset(alpha_textures_path + alpha_brush_name)
 
-# from that, get the class default object ( the actual template for the blueprint )
-blueprint_class_default = unreal.get_default_object(blueprint_generated)
+    # from that, get the class default object ( the actual template for the blueprint )
+    blueprint_class_default = unreal.get_default_object(blueprint_generated)
 
-#  set or get properties
-blueprint_class_default.set_editor_property("HeightMap", texture)
-unreal.EditorAssetLibrary.save_asset(alpha_brush_path + alpha_brush_name)
+    #  set or get properties
+    blueprint_class_default.set_editor_property(heightmap_property, texture)
+    unreal.EditorAssetLibrary.save_asset(alpha_brush_path + alpha_brush_name)