=== modified file 'dhis-2/dhis-web/dhis-web-api/src/main/java/org/hisp/dhis/api/utils/FormUtils.java' --- dhis-2/dhis-web/dhis-web-api/src/main/java/org/hisp/dhis/api/utils/FormUtils.java 2013-08-20 13:23:29 +0000 +++ dhis-2/dhis-web/dhis-web-api/src/main/java/org/hisp/dhis/api/utils/FormUtils.java 2013-08-21 07:28:04 +0000 @@ -39,6 +39,8 @@ import org.hisp.dhis.datavalue.DataValue; import org.hisp.dhis.program.Program; import org.hisp.dhis.program.ProgramStage; +import org.hisp.dhis.program.ProgramStageDataElement; +import org.hisp.dhis.program.ProgramStageSection; import org.springframework.util.Assert; import org.springframework.util.StringUtils; @@ -126,9 +128,50 @@ form.getOptions().put( "captureCoordinates", programStage.getCaptureCoordinates() ); + if ( programStage.getProgramStageSections().size() > 0 ) + { + for ( ProgramStageSection section : programStage.getProgramStageSections() ) + { + List fields = inputsFromProgramStageDataElements( section.getProgramStageDataElements() ); + + if ( !fields.isEmpty() ) + { + Group s = new Group(); + s.setLabel( section.getDisplayName() ); + s.setFields( fields ); + form.getGroups().add( s ); + } + } + } + else + { + List fields = inputsFromProgramStageDataElements( + new ArrayList( programStage.getProgramStageDataElements() ) ); + + if ( !fields.isEmpty() ) + { + Group s = new Group(); + s.setLabel( "default" ); + s.setFields( fields ); + form.getGroups().add( s ); + } + } + return form; } + private static List inputsFromProgramStageDataElements( List programStageDataElements ) + { + List dataElements = new ArrayList(); + + for ( ProgramStageDataElement programStageDataElement : programStageDataElements ) + { + dataElements.add( programStageDataElement.getDataElement() ); + } + + return inputsFromDataElements( dataElements, new ArrayList() ); + } + private static List inputsFromDataElements( List dataElements ) { return inputsFromDataElements( dataElements, new ArrayList() );