From 9646afbd8148f8a67fd553ab84e1fdb9b023b556 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jindra=20Pet=C5=99=C3=ADk?= <jindra.petrik@gmail.com>
Date: Tue, 7 Jul 2015 22:04:28 +0200
Subject: [PATCH] Issue #950 AddTrait setting modified fix

---
 src/com/jpexs/decompiler/flash/gui/abc/ABCPanel.java | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/com/jpexs/decompiler/flash/gui/abc/ABCPanel.java b/src/com/jpexs/decompiler/flash/gui/abc/ABCPanel.java
index 449065d50..d7d3dc6a9 100644
--- a/src/com/jpexs/decompiler/flash/gui/abc/ABCPanel.java
+++ b/src/com/jpexs/decompiler/flash/gui/abc/ABCPanel.java
@@ -61,6 +61,7 @@ import com.jpexs.decompiler.flash.gui.controls.JPersistentSplitPane;
 import com.jpexs.decompiler.flash.gui.editor.LinkHandler;
 import com.jpexs.decompiler.flash.gui.tagtree.TagTreeModel;
 import com.jpexs.decompiler.flash.tags.ABCContainerTag;
+import com.jpexs.decompiler.flash.tags.Tag;
 import com.jpexs.decompiler.flash.treeitems.TreeItem;
 import com.jpexs.decompiler.graph.CompilationException;
 import com.jpexs.helpers.CancellableWorker;
@@ -853,6 +854,11 @@ public class ABCPanel extends JPanel implements ItemListener, SearchListener<ABC
             } else {
                 traitId = abc.class_info.get(class_index).static_traits.traits.size() + abc.instance_info.get(class_index).instance_traits.addTrait(t);
             }
+            int scriptIndex = decompiledTextArea.getScriptLeaf().scriptIndex;
+            if (scriptIndex >= 0 && scriptIndex < abc.script_info.size()) {
+                abc.script_info.get(scriptIndex).setModified(true);
+            }
+            ((Tag) abc.parentTag).setModified(true);
             reload();
             decompiledTextArea.gotoTrait(traitId);
         }