BirthmarkSpi -> BirthmarkService

[stigmata/stigmata.git] / src / main / java / jp / sourceforge / stigmata / birthmarks / kgram / KGramBasedBirthmarkExtractor.java

diff --git a/src/main/java/jp/sourceforge/stigmata/birthmarks/kgram/KGramBasedBirthmarkExtractor.java b/src/main/java/jp/sourceforge/stigmata/birthmarks/kgram/KGramBasedBirthmarkExtractor.java
index 24d5d68..9c8e527 100644 (file)
--- a/src/main/java/jp/sourceforge/stigmata/birthmarks/kgram/KGramBasedBirthmarkExtractor.java
+++ b/src/main/java/jp/sourceforge/stigmata/birthmarks/kgram/KGramBasedBirthmarkExtractor.java
@@ -2,10 +2,11 @@ package jp.sourceforge.stigmata.birthmarks.kgram;
 
 import jp.sourceforge.stigmata.Birthmark;
 import jp.sourceforge.stigmata.BirthmarkContext;
+import jp.sourceforge.stigmata.BirthmarkElement;
 import jp.sourceforge.stigmata.ExtractionUnit;
 import jp.sourceforge.stigmata.birthmarks.ASMBirthmarkExtractor;
 import jp.sourceforge.stigmata.birthmarks.BirthmarkExtractVisitor;
-import jp.sourceforge.stigmata.spi.BirthmarkSpi;
+import jp.sourceforge.stigmata.spi.BirthmarkService;
 
 import org.objectweb.asm.ClassWriter;
 
@@ -15,7 +16,7 @@ import org.objectweb.asm.ClassWriter;
 public class KGramBasedBirthmarkExtractor extends ASMBirthmarkExtractor{
     private int kvalue = 4;
 
-    public KGramBasedBirthmarkExtractor(BirthmarkSpi spi){
+    public KGramBasedBirthmarkExtractor(BirthmarkService spi){
         super(spi);
     }
 
@@ -44,4 +45,19 @@ public class KGramBasedBirthmarkExtractor extends ASMBirthmarkExtractor{
             ExtractionUnit.CLASS, ExtractionUnit.PACKAGE, ExtractionUnit.ARCHIVE, 
         };
     }
+
+
+    @Override
+    public BirthmarkElement buildElement(String value) {
+        value = value.trim();
+        if(value.startsWith("{") && value.endsWith("}")){
+            String[] param = value.substring(1, value.length() - 1).split(", *");
+            KGram<Integer> kgram = new KGram<Integer>(param.length);
+            for(int i = 0; i < param.length; i++){
+                kgram.set(i, new Integer(param[i].trim()));
+            }
+            return new KGramBasedBirthmarkElement<Integer>(kgram);
+        }
+        return null;
+    }
 }