Add more custom commands and properties

Author: ThexXTURBOXx

src/main/java/com/github/sh0nk/matplotlib4j/Plot.java

@@ -15,6 +15,8 @@ static Plot create(PythonConfig pythonConfig) {
         return new PlotImpl(pythonConfig, false);
     }
 
+    CustomBuilder cmd(String key);
+
     GridBuilder grid();
 
     LegendBuilder legend();

src/main/java/com/github/sh0nk/matplotlib4j/PlotImpl.java

@@ -26,6 +26,11 @@ public class PlotImpl implements Plot {
         this(PythonConfig.systemDefaultPythonConfig(), dryRun);
     }
 
+    @Override
+    public CustomBuilder cmd(String key) {
+        return new CustomBuilderImpl(key);
+    }
+
     @Override
     public LegendBuilder legend() {
         LegendBuilder builder = new LegendBuilderImpl();

src/main/java/com/github/sh0nk/matplotlib4j/builder/AxLineBuilder.java

@@ -1,13 +1,13 @@
 package com.github.sh0nk.matplotlib4j.builder;
 
+import com.github.sh0nk.matplotlib4j.kwargs.KwArgsBuilder;
 import com.github.sh0nk.matplotlib4j.kwargs.Line2DBuilder;
-import java.util.List;
 
 
 /**
  * matplotlib.pyplot.plot(*args, **kwargs)
  */
-public interface AxLineBuilder extends Builder, Line2DBuilder<AxLineBuilder> {
+public interface AxLineBuilder extends Builder, Line2DBuilder<AxLineBuilder>, KwArgsBuilder<AxLineBuilder> {
 
     AxLineBuilder at(Number value);
 

src/main/java/com/github/sh0nk/matplotlib4j/builder/AxLineBuilderImpl.java

@@ -66,6 +66,31 @@ public AxLineBuilder color(String arg) {
         return line2DBuilder.color(arg);
     }
 
+    @Override
+    public AxLineBuilder addToKwargs(String k, String v) {
+        return innerBuilder.addToKwargs(k, v);
+    }
+
+    @Override
+    public AxLineBuilder addToKwargsWithoutQuoting(String k, String v) {
+        return innerBuilder.addToKwargsWithoutQuoting(k, v);
+    }
+
+    @Override
+    public AxLineBuilder addToKwargs(String k, Number n) {
+        return innerBuilder.addToKwargs(k, n);
+    }
+
+    @Override
+    public AxLineBuilder addToKwargs(String k, List<? extends Number> v) {
+        return innerBuilder.addToKwargs(k, v);
+    }
+
+    @Override
+    public AxLineBuilder addToKwargs(String k, boolean v) {
+        return innerBuilder.addToKwargs(k, v);
+    }
+
     @Override
     public String build() {
         return innerBuilder.build();

src/main/java/com/github/sh0nk/matplotlib4j/builder/BarBuilder.java

@@ -1,13 +1,14 @@
 package com.github.sh0nk.matplotlib4j.builder;
 
+import com.github.sh0nk.matplotlib4j.kwargs.KwArgsBuilder;
 import com.github.sh0nk.matplotlib4j.kwargs.Line2DBuilder;
 import java.util.List;
 
 
 /**
  * matplotlib.pyplot.bar(*args, **kwargs)
  */
-public interface BarBuilder extends Builder, Line2DBuilder<BarBuilder> {
+public interface BarBuilder extends Builder, Line2DBuilder<BarBuilder>, KwArgsBuilder<BarBuilder> {
 
     BarBuilder add(List<? extends Number> nums);
 

src/main/java/com/github/sh0nk/matplotlib4j/builder/BarBuilderImpl.java

@@ -49,6 +49,31 @@ public BarBuilder color(String arg) {
         return line2DBuilder.color(arg);
     }
 
+    @Override
+    public BarBuilder addToKwargs(String k, String v) {
+        return innerBuilder.addToKwargs(k, v);
+    }
+
+    @Override
+    public BarBuilder addToKwargsWithoutQuoting(String k, String v) {
+        return innerBuilder.addToKwargsWithoutQuoting(k, v);
+    }
+
+    @Override
+    public BarBuilder addToKwargs(String k, Number n) {
+        return innerBuilder.addToKwargs(k, n);
+    }
+
+    @Override
+    public BarBuilder addToKwargs(String k, List<? extends Number> v) {
+        return innerBuilder.addToKwargs(k, v);
+    }
+
+    @Override
+    public BarBuilder addToKwargs(String k, boolean v) {
+        return innerBuilder.addToKwargs(k, v);
+    }
+
     @Override
     public String build() {
         return innerBuilder.build();

src/main/java/com/github/sh0nk/matplotlib4j/builder/Builder.java

@@ -3,5 +3,13 @@
 public interface Builder {
     String build();
 
+    default boolean returns() {
+        return true;
+    }
+
+    default String getMethodPrefix() {
+        return "plt.";
+    }
+
     String getMethodName();
 }

src/main/java/com/github/sh0nk/matplotlib4j/builder/CLabelBuilder.java

@@ -1,6 +1,8 @@
 package com.github.sh0nk.matplotlib4j.builder;
 
-public interface CLabelBuilder extends Builder {
+import com.github.sh0nk.matplotlib4j.kwargs.KwArgsBuilder;
+
+public interface CLabelBuilder extends Builder, KwArgsBuilder<CLabelBuilder> {
 
     CLabelBuilder fontsize(String arg);
 

src/main/java/com/github/sh0nk/matplotlib4j/builder/CLabelBuilderImpl.java

@@ -1,5 +1,7 @@
 package com.github.sh0nk.matplotlib4j.builder;
 
+import java.util.List;
+
 public class CLabelBuilderImpl implements CLabelBuilder {
 
     private final CompositeBuilder<CLabelBuilder> innerBuilder = new CompositeBuilder<>(this);
@@ -48,6 +50,31 @@ public CLabelBuilder useClabeltext(boolean arg) {
         return innerBuilder.addToKwargs("use_clabeltext", arg);
     }
 
+    @Override
+    public CLabelBuilder addToKwargs(String k, String v) {
+        return innerBuilder.addToKwargs(k, v);
+    }
+
+    @Override
+    public CLabelBuilder addToKwargsWithoutQuoting(String k, String v) {
+        return innerBuilder.addToKwargsWithoutQuoting(k, v);
+    }
+
+    @Override
+    public CLabelBuilder addToKwargs(String k, Number n) {
+        return innerBuilder.addToKwargs(k, n);
+    }
+
+    @Override
+    public CLabelBuilder addToKwargs(String k, List<? extends Number> v) {
+        return innerBuilder.addToKwargs(k, v);
+    }
+
+    @Override
+    public CLabelBuilder addToKwargs(String k, boolean v) {
+        return innerBuilder.addToKwargs(k, v);
+    }
+
     @Override
     public String build() {
         return innerBuilder.build();

src/main/java/com/github/sh0nk/matplotlib4j/builder/CompositeBuilder.java

@@ -1,6 +1,7 @@
 package com.github.sh0nk.matplotlib4j.builder;
 
 import com.github.sh0nk.matplotlib4j.TypeConversion;
+import com.github.sh0nk.matplotlib4j.kwargs.KwArgsBuilder;
 import com.google.common.base.Joiner;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -14,7 +15,7 @@
  *
  * @param <T> Owner builder class
  */
-public class CompositeBuilder<T extends Builder> implements Builder {
+public class CompositeBuilder<T extends Builder> implements Builder, KwArgsBuilder<T> {
 
     private final static Logger LOGGER = LoggerFactory.getLogger(CompositeBuilder.class);
 
@@ -107,9 +108,10 @@ public String build() {
         }
 
         // retName
-        sb.append(retName).append(" = ");
+        if (ownerBuilder.returns())
+            sb.append(retName).append(" = ");
 
-        sb.append("plt.");
+        sb.append(ownerBuilder.getMethodPrefix());
         sb.append(ownerBuilder.getMethodName());
         sb.append("(");