3333import  static  uk .co .real_logic .sbe .generation .java .JavaUtil .formatPropertyName ;
3434import  static  uk .co .real_logic .sbe .generation .java .JavaUtil .javaTypeName ;
3535
36- public  class  JavaPojoGenerator  implements  CodeGenerator 
36+ public  class  JavaMockPojoGenerator  implements  CodeGenerator 
3737{
3838 private  static  final  String  BASE_INDENT  = "" ;
3939 private  static  final  String  INDENT  = " " ;
40-  public  static  final  String  MOCK  = "Mock" ;
40+  private  static  final  String  MOCK  = "Mock" ;
4141
4242 private  final  Ir  ir ;
4343 private  final  OutputManager  outputManager ;
4444
45-  public  JavaPojoGenerator (final  Ir  ir , final  OutputManager  outputManager )
45+  public  JavaMockPojoGenerator (final  Ir  ir , final  OutputManager  outputManager )
4646 throws  IOException 
4747 {
4848 Verify .notNull (ir , "ir" );
@@ -109,20 +109,20 @@ private CharSequence generateChoices(final String bitSetClassName, final List<To
109109 {
110110 final  String  choiceName  = token .name ();
111111 sb .append (String .format (
112-    "\n "  +
113-    " private boolean %s;\n "  +
114-    " public boolean %s()\n "  +
115-    " {\n "  +
116-    " return %s;\n "  +
117-    " }\n \n "  +
118-    " public %s %s(final boolean value)\n "  +
119-    " {\n "  +
120-    " %s = value;\n "  +
121-    " return this;\n "  +
122-    " }\n " ,
123-    choiceName , choiceName , choiceName ,
124-    bitSetClassName ,
125-    choiceName , choiceName 
112+  "\n "  +
113+  " private boolean %s;\n "  +
114+  " public boolean %s()\n "  +
115+  " {\n "  +
116+  " return %s;\n "  +
117+  " }\n \n "  +
118+  " public %s %s(final boolean value)\n "  +
119+  " {\n "  +
120+  " %s = value;\n "  +
121+  " return this;\n "  +
122+  " }\n " ,
123+  choiceName , choiceName , choiceName ,
124+  bitSetClassName ,
125+  choiceName , choiceName 
126126 ));
127127 }
128128 }
@@ -197,11 +197,7 @@ private int collectRootFields(final List<Token> tokens, int index, final List<To
197197 }
198198
199199 private  int  generateGroups (
200-  final  StringBuilder  sb ,
201-  final  String  parentMessageClassName ,
202-  final  List <Token > tokens ,
203-  int  index ,
204-  final  String  indent )
200+  final  StringBuilder  sb , final  String  parentMessageClassName , final  List <Token > tokens , int  index , final  String  indent )
205201 {
206202 for  (int  size  = tokens .size (); index  < size ; index ++)
207203 {
@@ -211,7 +207,7 @@ private int generateGroups(
211207 final  String  groupName  = groupToken .name ();
212208 final  String  groupClassName  = formatClassName (groupName );
213209 sb .append (generateSingleValueProperty (parentMessageClassName ,
214-    JavaUtil .toLowerFirstChar (groupName ), groupClassName , indent ));
210+  JavaUtil .toLowerFirstChar (groupName ), groupClassName , indent ));
215211
216212 generateGroupClassHeader (sb , groupName , indent  + INDENT );
217213
@@ -231,10 +227,7 @@ private int generateGroups(
231227 return  index ;
232228 }
233229
234-  private  void  generateGroupClassHeader (
235-  final  StringBuilder  sb ,
236-  final  String  groupName ,
237-  final  String  indent )
230+  private  void  generateGroupClassHeader (final  StringBuilder  sb , final  String  groupName , final  String  indent )
238231 {
239232 sb .append (String .format (
240233 "\n "  +
@@ -369,69 +362,71 @@ else if (arrayLength > 1)
369362 }
370363
371364 private  CharSequence  generateArrayProperty (
372-  final  Encoding  encoding ,
373-  final  String  containingClassName , final  String  propertyName ,
374-  final  String  javaTypeName , final  String  indent )
365+  final  Encoding  encoding ,
366+  final  String  containingClassName ,
367+  final  String  propertyName ,
368+  final  String  javaTypeName ,
369+  final  String  indent )
375370 {
376371 final  StringBuilder  sb  = new  StringBuilder ();
377372
378373 sb .append (String .format (
379-    "\n "  + indent  + " private %s[] %s;\n " ,
380-    javaTypeName ,
381-    propertyName 
374+  "\n "  + indent  + " private %s[] %s;\n " ,
375+  javaTypeName ,
376+  propertyName 
382377 ));
383378
384379 sb .append (String .format (
385-    indent  + " public %1$s %2$s(final int index)\n "  +
386-    indent  + " {\n "  +
387-    indent  + " return %2$s[index];\n "  +
388-    indent  + " }\n \n " ,
389-    javaTypeName ,
390-    propertyName 
380+  indent  + " public %1$s %2$s(final int index)\n "  +
381+  indent  + " {\n "  +
382+  indent  + " return %2$s[index];\n "  +
383+  indent  + " }\n \n " ,
384+  javaTypeName ,
385+  propertyName 
391386 ));
392387
393388 sb .append (String .format (
394-    indent  + " public void %1$s(final int index, final %2$s value)\n "  +
395-    indent  + " {\n "  +
396-    indent  + " %1$s[index] = value;\n "  +
397-    indent  + " }\n " ,
398-    propertyName ,
399-    javaTypeName 
389+  indent  + " public void %1$s(final int index, final %2$s value)\n "  +
390+  indent  + " {\n "  +
391+  indent  + " %1$s[index] = value;\n "  +
392+  indent  + " }\n " ,
393+  propertyName ,
394+  javaTypeName 
400395 ));
401396
402397 if  (encoding .primitiveType () == PrimitiveType .CHAR )
403398 {
404399 sb .append (String .format (
405-    "\n "  +
406-    indent  + " public int get%1$s(final byte[] dst, final int dstOffset)\n "  +
407-    indent  + " {\n "  +
408-    indent  + " System.arraycopy(%2$s, 0, dst, dstOffset, %2$sLength());\n "  +
409-    indent  + " return %2$sLength();\n "  +
410-    indent  + " }\n \n " ,
411-    JavaUtil .toUpperFirstChar (propertyName ),
412-    propertyName 
400+  "\n "  +
401+  indent  + " public int get%1$s(final byte[] dst, final int dstOffset)\n "  +
402+  indent  + " {\n "  +
403+  indent  + " System.arraycopy(%2$s, 0, dst, dstOffset, %2$sLength());\n "  +
404+  indent  + " return %2$sLength();\n "  +
405+  indent  + " }\n \n " ,
406+  JavaUtil .toUpperFirstChar (propertyName ),
407+  propertyName 
413408 ));
414409
415410 sb .append (String .format (
416-    indent  + " public %1$s put%2$s(final byte[] src, final int srcOffset)\n "  +
417-    indent  + " {\n "  +
418-    indent  + " System.arraycopy(src, srcOffset, %3$s, 0, src.length - srcOffset);\n "  +
419-    indent  + " return this;\n "  +
420-    indent  + " }\n " ,
421-    containingClassName ,
422-    JavaUtil .toUpperFirstChar (propertyName ),
423-    propertyName 
411+  indent  + " public %1$s put%2$s(final byte[] src, final int srcOffset)\n "  +
412+  indent  + " {\n "  +
413+  indent  + " System.arraycopy(src, srcOffset, %3$s, 0, src.length - srcOffset);\n "  +
414+  indent  + " return this;\n "  +
415+  indent  + " }\n " ,
416+  containingClassName ,
417+  JavaUtil .toUpperFirstChar (propertyName ),
418+  propertyName 
424419 ));
425420
426421 sb .append (String .format (
427-    indent  + " public %1$s put%2$s(final byte[] src)\n "  +
428-    indent  + " {\n "  +
429-    indent  + " %3$s = Arrays.copyOf(src, %3$sLength());\n "  +
430-    indent  + " return this;\n "  +
431-    indent  + " }\n " ,
432-    containingClassName ,
433-    JavaUtil .toUpperFirstChar (propertyName ),
434-    propertyName 
422+  indent  + " public %1$s put%2$s(final byte[] src)\n "  +
423+  indent  + " {\n "  +
424+  indent  + " %3$s = Arrays.copyOf(src, %3$sLength());\n "  +
425+  indent  + " return this;\n "  +
426+  indent  + " }\n " ,
427+  containingClassName ,
428+  JavaUtil .toUpperFirstChar (propertyName ),
429+  propertyName 
435430 ));
436431 }
437432
@@ -444,9 +439,9 @@ private CharSequence generateSingleValueProperty(
444439 final  StringBuilder  sb  = new  StringBuilder ();
445440
446441 sb .append (String .format (
447-    "\n "  + indent  + " private %s %s;\n " ,
448-    javaTypeName ,
449-    propertyName 
442+  "\n "  + indent  + " private %s %s;\n " ,
443+  javaTypeName ,
444+  propertyName 
450445 ));
451446
452447 sb .append (String .format (
@@ -506,10 +501,10 @@ private CharSequence generateFields(final String containingClassName, final List
506501 }
507502
508503 private  CharSequence  generateEnumProperty (
509-    final  String  containingClassName , final  String  propertyName , final  Token  token , final  String  indent )
504+  final  String  containingClassName , final  String  propertyName , final  Token  token , final  String  indent )
510505 {
511- 
512506 final  String  enumName  = token .name ();
507+ 
513508 return  generateSingleValueProperty (containingClassName , propertyName , enumName , indent );
514509 }
515510}
0 commit comments