← Back to branch summary

~mortenoh/+junk/dhis2-detailed-import-export

« back to all changes in this revision

Viewing changes to dhis-2/dhis-web/dhis-web-commons/src/main/java/org/hisp/dhis/ouwt/action/ExpandSubtreeAction.java

  • Committer: larshelge at gmail
  • Date: 2009-03-03 16:46:36 UTC
  • Revision ID: larshelge@gmail.com-20090303164636-2sjlrquo7ib1gf7r
Initial check-in

Show diffs side-by-side

added added

removed removed

Lines of Context:
dhis-2/dhis-web/dhis-web-commons/src/main/java/org/hisp/dhis/ouwt/action/ExpandSubtreeAction.java
 
1
package org.hisp.dhis.ouwt.action;
 
2
 
 
3
/*
 
4
 * Copyright (c) 2004-2007, University of Oslo
 
5
 * All rights reserved.
 
6
 *
 
7
 * Redistribution and use in source and binary forms, with or without
 
8
 * modification, are permitted provided that the following conditions are met:
 
9
 * * Redistributions of source code must retain the above copyright notice, this
 
10
 *   list of conditions and the following disclaimer.
 
11
 * * Redistributions in binary form must reproduce the above copyright notice,
 
12
 *   this list of conditions and the following disclaimer in the documentation
 
13
 *   and/or other materials provided with the distribution.
 
14
 * * Neither the name of the HISP project nor the names of its contributors may
 
15
 *   be used to endorse or promote products derived from this software without
 
16
 *   specific prior written permission.
 
17
 *
 
18
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
 
19
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
 
20
 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 
21
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
 
22
 * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
 
23
 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
 
24
 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
 
25
 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 
26
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 
27
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
28
 */
 
29
 
 
30
import java.util.ArrayList;
 
31
import java.util.Collections;
 
32
import java.util.HashMap;
 
33
import java.util.List;
 
34
import java.util.Map;
 
35
 
 
36
import org.apache.commons.logging.Log;
 
37
import org.apache.commons.logging.LogFactory;
 
38
import org.hisp.dhis.organisationunit.OrganisationUnit;
 
39
import org.hisp.dhis.organisationunit.OrganisationUnitService;
 
40
import org.hisp.dhis.organisationunit.comparator.OrganisationUnitNameComparator;
 
41
import org.hisp.dhis.ouwt.manager.TreeStateManager;
 
42
 
 
43
import com.opensymphony.xwork.Action;
 
44
 
 
45
/**
 
46
 * @author Torgeir Lorange Ostby
 
47
 * @version $Id: ExpandSubtreeAction.java 6251 2008-11-10 14:37:05Z larshelg $
 
48
 */
 
49
public class ExpandSubtreeAction
 
50
    implements Action
 
51
{
 
52
    private static final Log LOG = LogFactory.getLog( ExpandSubtreeAction.class );
 
53
 
 
54
    // -------------------------------------------------------------------------
 
55
    // Dependencies
 
56
    // -------------------------------------------------------------------------
 
57
 
 
58
    private OrganisationUnitService organisationUnitService;
 
59
 
 
60
    public void setOrganisationUnitService( OrganisationUnitService organisationUnitService )
 
61
    {
 
62
        this.organisationUnitService = organisationUnitService;
 
63
    }
 
64
 
 
65
    private TreeStateManager treeStateManager;
 
66
 
 
67
    public void setTreeStateManager( TreeStateManager treeStateManager )
 
68
    {
 
69
        this.treeStateManager = treeStateManager;
 
70
    }
 
71
 
 
72
    // -------------------------------------------------------------------------
 
73
    // Input
 
74
    // -------------------------------------------------------------------------
 
75
 
 
76
    private int parentId;
 
77
 
 
78
    public void setParentId( int organisationUnitId )
 
79
    {
 
80
        this.parentId = organisationUnitId;
 
81
    }
 
82
 
 
83
    // -------------------------------------------------------------------------
 
84
    // Output
 
85
    // -------------------------------------------------------------------------
 
86
 
 
87
    private List<OrganisationUnit> parents = new ArrayList<OrganisationUnit>();
 
88
 
 
89
    public List<OrganisationUnit> getParents()
 
90
    {
 
91
        return parents;
 
92
    }
 
93
 
 
94
    private Map<OrganisationUnit, List<OrganisationUnit>> childrenMap = new HashMap<OrganisationUnit, List<OrganisationUnit>>();
 
95
 
 
96
    public Map<OrganisationUnit, List<OrganisationUnit>> getChildrenMap()
 
97
    {
 
98
        return childrenMap;
 
99
    }
 
100
 
 
101
    // -------------------------------------------------------------------------
 
102
    // Action implementation
 
103
    // -------------------------------------------------------------------------
 
104
 
 
105
    public String execute()
 
106
        throws Exception
 
107
    {
 
108
        try
 
109
        {
 
110
            OrganisationUnit parent = organisationUnitService.getOrganisationUnit( parentId );
 
111
 
 
112
            if ( parent == null )
 
113
            {
 
114
                throw new RuntimeException( "OrganisationUnit with id " + parentId + " doesn't exist" );
 
115
            }
 
116
 
 
117
            treeStateManager.setSubtreeExpanded( parent );
 
118
 
 
119
            addParentWithChildren( parent );
 
120
        }
 
121
        catch ( Exception e )
 
122
        {
 
123
            LOG.error( e.getMessage(), e );
 
124
 
 
125
            throw e;
 
126
        }
 
127
 
 
128
        return SUCCESS;
 
129
    }
 
130
 
 
131
    private void addParentWithChildren( OrganisationUnit parent )
 
132
        throws Exception
 
133
    {
 
134
        List<OrganisationUnit> children = getChildren( parent );
 
135
 
 
136
        parents.add( parent );
 
137
 
 
138
        childrenMap.put( parent, children );
 
139
 
 
140
        for ( OrganisationUnit child : children )
 
141
        {
 
142
            boolean hasChildren = child.getChildren().size() > 0; // Dirty loading
 
143
 
 
144
            LOG.debug( "OrganisationUnit " + child.getId() + " has children = " + hasChildren );
 
145
 
 
146
            if ( treeStateManager.isSubtreeExpanded( child ) )
 
147
            {
 
148
                addParentWithChildren( child );
 
149
            }
 
150
        }
 
151
    }
 
152
 
 
153
    private final List<OrganisationUnit> getChildren( OrganisationUnit parent )
 
154
    {
 
155
        List<OrganisationUnit> children = new ArrayList<OrganisationUnit>( parent.getChildren() );
 
156
 
 
157
        Collections.sort( children, new OrganisationUnitNameComparator() );
 
158
 
 
159
        return children;
 
160
    }
 
161
}