package water.util;

// Tight/tiny StringBuilder wrapper.
// Short short names on purpose; so they don't obscure the printing.
// Can't believe this wasn't done long long ago.
public class SB {
  public final StringBuilder _sb;
  private int _indent = 0;
  public SB(        ) { _sb = new StringBuilder( ); }
  public SB(String s) { _sb = new StringBuilder(s); }
  public SB p( String s ) { _sb.append(s); return this; }
  public SB p( float  s ) { if(  Float.isNaN(s) ) _sb.append( "Float.NaN"); else _sb.append(s); return this; }
  public SB p( double s ) { if( Double.isNaN(s) ) _sb.append("Double.NaN"); else _sb.append(s); return this; }
  public SB p( char   s ) { _sb.append(s); return this; }
  public SB p( int    s ) { _sb.append(s); return this; }
  // Not spelled "p" on purpose: too easy to accidentally say "p(1.0)" and
  // suddenly call the the autoboxed version.
  public SB pobj( Object s ) { _sb.append(s.toString()); return this; }
  public SB i( int d ) { for( int i=0; i<d+_indent; i++ ) p("  "); return this; }
  public SB i( ) { return i(0); }
  // Java specific append of float
  public SB pj( float  s ) { _sb.append(s).append('f'); return this; }
  // Increase indentation
  public SB ii( int i) { _indent += i; return this; }
  // Decrease indentation
  public SB di( int i) { _indent -= i; return this; }
  public SB nl( ) { return p('\n'); }
  // Convert a String[] into a valid Java String initializer
  public SB toJavaStringInit( String[] ss ) {
    p('{');
    for( int i=0; i<ss.length-1; i++ )  p('"').p(ss[i]).p("\",");
    if( ss.length > 0 ) p('"').p(ss[ss.length-1]).p('"');
    return p('}');
  }
  public SB toJSArray(float[] nums) {
    p('[');
    for (int i=0; i<nums.length; i++) {
      if (i>0) p(',');
      p(nums[i]);
    }
    return p(']');
  }
  public SB toJSArray(String[] ss) {
    p('[');
    for (int i=0; i<ss.length; i++) {
      if (i>0) p(',');
      p('"').p(ss[i]).p('"');
    }
    return p(']');
  }

  // Mostly a fail, since we should just dump into the same SB.
  public SB p( SB sb ) { _sb.append(sb._sb); return this;  }
  @Override public String toString() { return _sb.toString(); }
}

@@ -1120,24 +1120,211 @@
 float  s ) %7B
+%0A    if (Float.isInfinite(s))%0A      _sb.append(%22Float.%22).append(s%3E0? %22POSITIVE_INFINITY%22 : %22NEGATIVE_INFINITY%22);%0A    else if (Float.isNaN(s))%0A      _sb.append(%22Float.NaN%22);%0A    else%0A     
  _sb.append(
@@ -1338,16 +1338,20 @@
 nd('f');
+%0A   
  return 
@@ -1347,32 +1347,34 @@
     return this;
+%0A 
  %7D%0A  // Increase

Fix of PUB-117 - generated model representing Java code now correctly serializes +/- Infinities and NaN as Float literals.