Added another enumConstantRef overload (#68)

phax · Apr 15, 2019 · d5620e0 · d5620e0
1 parent 50c78ed
commit d5620e0
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/src/main/java/com/helger/jcodemodel/AbstractJClass.java b/src/main/java/com/helger/jcodemodel/AbstractJClass.java
@@ -550,8 +550,9 @@ public final JLambdaMethodRef methodRef (@Nonnull final String sMethod)
   }
 
   /**
-   * @param Enumeration
-   *        constant name. May neither be <code>null</code> nor empty.
+   * @param sName
+   *        Enumeration constant name. May neither be <code>null</code> nor
+   *        empty.
    * @return <code>class.name</code>
    * @since 3.2.4
    */
@@ -561,6 +562,18 @@ public final IJExpression enumConstantRef (@Nonnull final String sName)
     return JExpr.enumConstantRef (this, sName);
   }
 
+  /**
+   * @param aEnumEntry
+   *        Enumeration item. May not be <code>null</code>.
+   * @return <code>class.name</code>
+   * @since 3.2.4
+   */
+  @Nonnull
+  public final IJExpression enumConstantRef (@Nonnull final Enum <?> aEnumEntry)
+  {
+    return enumConstantRef (aEnumEntry.name ());
+  }
+
   public void generate (@Nonnull final IJFormatter f)
   {
     f.type (this);