coding.

diff --git a/jcfa/src/jp/igapyon/jcfa/JavaClassFileAnalyzer.java b/jcfa/src/jp/igapyon/jcfa/JavaClassFileAnalyzer.java
index c4d2949..25e9a2b 100644 (file)
--- a/jcfa/src/jp/igapyon/jcfa/JavaClassFileAnalyzer.java
+++ b/jcfa/src/jp/igapyon/jcfa/JavaClassFileAnalyzer.java
@@ -84,6 +84,7 @@ public class JavaClassFileAnalyzer {
 \r
        private void writeToFile(final JcfaUnit jcfaUnit) throws IOException {\r
                final StringBuffer result = new StringBuffer();\r
+\r
                for (JcfaClass jcfaClass : jcfaUnit.getClassList()) {\r
                        writeToBuffer(jcfaClass, result);\r
                }\r
@@ -97,6 +98,16 @@ public class JavaClassFileAnalyzer {
 \r
        private void writeToBuffer(final JcfaClass jcfaClass,\r
                        final StringBuffer result) {\r
+               if (jcfaClass.isMainClass()) {\r
+                       if (jcfaClass.getName().contains(".")) {\r
+                               result.append(" package "\r
+                                               + jcfaClass.getName().substring(0,\r
+                                                               jcfaClass.getName().lastIndexOf(".")) + ";");\r
+                       }\r
+               }\r
+\r
+               // TODO import\r
+\r
                result.append(jcfaClass.getAccess());\r
                result.append(" class " + jcfaClass.getLocalName());\r
                if (jcfaClass.getExtendsName() != null\r

diff --git a/jcfa/src/jp/igapyon/jcfa/vo/JcfaClass.java b/jcfa/src/jp/igapyon/jcfa/vo/JcfaClass.java
index e2f428d..7d9a7fd 100644 (file)
--- a/jcfa/src/jp/igapyon/jcfa/vo/JcfaClass.java
+++ b/jcfa/src/jp/igapyon/jcfa/vo/JcfaClass.java
@@ -10,6 +10,8 @@ public class JcfaClass extends JcfaNode {
 
        protected String extendsName;
 
+       protected boolean isMainClass = true;
+
        // TODO interfaces
 
        protected final List<JcfaField> fieldList = new ArrayList<JcfaField>();
@@ -40,6 +42,14 @@ public class JcfaClass extends JcfaNode {
                this.extendsName = extendsName;
        }
 
+       public boolean isMainClass() {
+               return isMainClass;
+       }
+
+       public void setMainClass(boolean isMainClass) {
+               this.isMainClass = isMainClass;
+       }
+
        public List<JcfaField> getFieldList() {
                return fieldList;
        }

diff --git a/jcfa/testJavaClass/output/test/TestJavaClass001.jcfa b/jcfa/testJavaClass/output/test/TestJavaClass001.jcfa
index a640bc3..d1d4c60 100644 (file)
--- a/jcfa/testJavaClass/output/test/TestJavaClass001.jcfa
+++ b/jcfa/testJavaClass/output/test/TestJavaClass001.jcfa
@@ -1,2 +1,4 @@
+package test;
+
 public class TestJavaClass001 {
 }
\ No newline at end of file

diff --git a/jcfa/testJavaClass/output/test/TestJavaClass002.jcfa b/jcfa/testJavaClass/output/test/TestJavaClass002.jcfa
index 27d5125..f9890a5 100644 (file)
--- a/jcfa/testJavaClass/output/test/TestJavaClass002.jcfa
+++ b/jcfa/testJavaClass/output/test/TestJavaClass002.jcfa
@@ -1,3 +1,5 @@
+package test;
+
 public class TestJavaClass002 {
        public static final String TEST_001;
        protected static String test002;