~dhis2-devs-core/dhis2/dhis2-ccei-2.18

package org.hisp.dhis.dataentry.action;

 

import java.util.ArrayList;

import java.util.Calendar;

import java.util.Collection;

import java.util.Collections;

import java.util.List;

 

import org.hisp.dhis.dataset.DataSet;

import org.hisp.dhis.dataset.DataSetService;

import org.hisp.dhis.i18n.I18nFormat;

import org.hisp.dhis.period.CalendarPeriodType;

import org.hisp.dhis.period.Period;

import org.hisp.dhis.period.PeriodType;

import org.hisp.dhis.system.filter.PastAndCurrentPeriodFilter;

import org.hisp.dhis.system.util.FilterUtils;

 

import com.opensymphony.xwork2.Action;

 

/**

 * @author Mithilesh Kumar Thakur

 */

 

public class LoadPeriodsAction implements Action

{

    // -------------------------------------------------------------------------

    // Dependencies

    // -------------------------------------------------------------------------

 

    private DataSetService dataSetService;

    

    public void setDataSetService( DataSetService dataSetService )

    {

        this.dataSetService = dataSetService;

    }

    

    private I18nFormat format;

 

    public void setFormat( I18nFormat format )

    {

        this.format = format;

    }

 

    // -------------------------------------------------------------------------

    // Input

    // -------------------------------------------------------------------------

    

    private int dataSetId;

    

    public void setDataSetId( int dataSetId )

    {

        this.dataSetId = dataSetId;

    }

    

    // -------------------------------------------------------------------------

    // Output

    // -------------------------------------------------------------------------

 

 

    private List<Period> periods = new ArrayList<Period>();

 

    public Collection<Period> getPeriods()

    {

        return periods;

    }

 

    // -------------------------------------------------------------------------

    // Action implementation

    // -------------------------------------------------------------------------

 

    public String execute() throws Exception

    { 

        

        DataSet dataset = dataSetService.getDataSet( dataSetId );

        

        String periodType = dataset.getPeriodType().getName();

        

        CalendarPeriodType _periodType = (CalendarPeriodType) CalendarPeriodType.getPeriodTypeByName( periodType );

        

        Calendar cal = PeriodType.createCalendarInstance();

        

        periods = _periodType.generatePeriods( cal.getTime() );

        

        //periods = new ArrayList<Period>( periodService.getPeriodsByPeriodType( periodType ) );

        

        FilterUtils.filter( periods, new PastAndCurrentPeriodFilter() );

 

        Collections.reverse( periods );

        //Collections.sort( periods );

        for ( Period period : periods )

        {

            //System.out.println("ISO Date : " + period.getIsoDate() );

            

            period.setName( format.formatPeriod( period ) );

        }

        

        return SUCCESS;

    }

}