From 223da54ac571cb36f39e81aae660c684c9e887d2 Mon Sep 17 00:00:00 2001
From: Mathieu Bastian <mathieu.bastian@gmail.com>
Date: Tue, 9 Apr 2013 23:03:35 -0700
Subject: [PATCH] Minor changes on ImportAPI

---
 .../org/gephi/io/importer/api/ContainerUnloader.java   |  4 ++++
 .../org/gephi/io/importer/impl/ElementDraftImpl.java   |  7 +------
 .../gephi/io/importer/impl/ImportContainerImpl.java    | 10 ++++++++++
 3 files changed, 15 insertions(+), 6 deletions(-)

diff --git a/modules/ImportAPI/src/main/java/org/gephi/io/importer/api/ContainerUnloader.java b/modules/ImportAPI/src/main/java/org/gephi/io/importer/api/ContainerUnloader.java
index 90a6ef23c..62431f0f0 100644
--- a/modules/ImportAPI/src/main/java/org/gephi/io/importer/api/ContainerUnloader.java
+++ b/modules/ImportAPI/src/main/java/org/gephi/io/importer/api/ContainerUnloader.java
@@ -60,6 +60,10 @@ public interface ContainerUnloader {
 
     public Iterable<EdgeDraft> getEdges();
 
+    public boolean hasNodeColumn(String key);
+
+    public boolean hasEdgeColumn(String key);
+
     public Iterable<ColumnDraft> getNodeColumns();
 
     public Iterable<ColumnDraft> getEdgeColumns();
diff --git a/modules/ImportAPI/src/main/java/org/gephi/io/importer/impl/ElementDraftImpl.java b/modules/ImportAPI/src/main/java/org/gephi/io/importer/impl/ElementDraftImpl.java
index 48d3cc0d9..cb4594e00 100644
--- a/modules/ImportAPI/src/main/java/org/gephi/io/importer/impl/ElementDraftImpl.java
+++ b/modules/ImportAPI/src/main/java/org/gephi/io/importer/impl/ElementDraftImpl.java
@@ -100,11 +100,6 @@ public abstract class ElementDraftImpl implements ElementDraft {
         return labelColor;
     }
 
-    @Override
-    public void setValue(String key, Object value) {
-        throw new UnsupportedOperationException("Not supported yet.");
-    }
-
     @Override
     public void setLabel(String label) {
         this.label = label;
@@ -112,7 +107,7 @@ public abstract class ElementDraftImpl implements ElementDraft {
 
     @Override
     public void setColor(Color color) {
-        throw new UnsupportedOperationException("Not supported yet.");
+        this.color = color;
     }
 
     @Override
diff --git a/modules/ImportAPI/src/main/java/org/gephi/io/importer/impl/ImportContainerImpl.java b/modules/ImportAPI/src/main/java/org/gephi/io/importer/impl/ImportContainerImpl.java
index 9a96c17d1..82c15d20c 100644
--- a/modules/ImportAPI/src/main/java/org/gephi/io/importer/impl/ImportContainerImpl.java
+++ b/modules/ImportAPI/src/main/java/org/gephi/io/importer/impl/ImportContainerImpl.java
@@ -442,11 +442,21 @@ public class ImportContainerImpl implements Container, ContainerLoader, Containe
         return nodeColumns.get(key);
     }
 
+    @Override
+    public boolean hasNodeColumn(String key) {
+        return nodeColumns.containsKey(key);
+    }
+
     @Override
     public ColumnDraft getEdgeColumn(String key) {
         return edgeColumns.get(key);
     }
 
+    @Override
+    public boolean hasEdgeColumn(String key) {
+        return edgeColumns.containsKey(key);
+    }
+
     @Override
     public Iterable<ColumnDraft> getNodeColumns() {
         return nodeColumns.values();
-- 
GitLab