Modify traits management.

- Compute all traits to generate properties (attributes & accessors).
- Only get traits from the class when writing the imports.
This is to avoid redundancy in classes definition but still get all correct properties.

Author: ClotildeToullec

lib/src/main/java/ch/akuhn/fame/codegen/CodeGeneration.java

@@ -340,7 +340,7 @@ private void acceptClass(MetaDescription metaDescription) throws IOException {
         }
         // Properties from my traits that are not in my own properties
         Set<PropertyDescription> propertyDescriptionSet = new HashSet<>();
-        metaDescription.getTraits().stream()
+        metaDescription.computeAllTraits().stream()
                 .map(c -> c.getProperties().stream()
                         .filter(traitProperty -> propertyDescriptions.stream().noneMatch(myProperty -> myProperty.getName().equals(traitProperty.getName())))
                         .collect(Collectors.toList()))
@@ -365,7 +365,7 @@ private void acceptTrait(FM3Trait m) throws IOException {
         code.setTraits(m.getTraits().stream().map(FM3Type::getName).collect(Collectors.toList()));
         code.addImport(FameDescription.class);
         code.addImport(FamePackage.class);
-        for (FM3Trait t : m.computeAllTraits()) {
+        for (FM3Trait t : m.getTraits()) {
             code.addImport(packageName(t.getPackage()), t.getName());
         }
 

lib/src/main/java/ch/akuhn/fame/fm3/FM3Type.java

@@ -156,7 +156,7 @@ public void setTraits(Collection<FM3Trait> traits) {
         }
     }
 
-    @FameProperty(name="traits", opposite = "owner")
+    @FameProperty(name="traits", opposite = "users")
     public Collection<FM3Trait> getTraits() {
         return traits.values();
     }
@@ -174,11 +174,16 @@ public boolean isRoot() {
     }
 
     public Set<FM3Trait> computeAllTraits() {
-        Set<FM3Trait> traits = new HashSet<>();
-        traits.addAll(getTraits());
-        for (FM3Trait t : traits) {
-            traits.addAll(t.computeAllTraits());
+        Set<FM3Trait> computedTraits = new HashSet<>();
+        computedTraits.addAll(getTraits());
+        Set<FM3Trait> transitiveTraits = new HashSet<>();
+        
+        for (FM3Trait t : computedTraits) {
+            transitiveTraits.addAll(t.computeAllTraits());
         }
-        return traits;
+
+        computedTraits.addAll(transitiveTraits);
+        return computedTraits;
     }
+    
 }