statistics/src/main/java/cz/xelfi/quoridor/statistics/OpeningTreeNode.java
author Jaroslav Tulach <jaroslav.tulach@apidesign.org>
Tue, 14 Sep 2010 09:46:43 +0200
changeset 266 15fcdfc4cd4a
parent 264 d60370059c3c
permissions -rw-r--r--
Using maven-license-plugin and updating all missing headers
jaroslav@266
     1
/**
jaroslav@264
     2
 * Quoridor server and related libraries
jaroslav@264
     3
 * Copyright (C) 2009-2010 Martin Rexa
jaroslav@178
     4
 *
jaroslav@264
     5
 * This program is free software: you can redistribute it and/or modify
jaroslav@264
     6
 * it under the terms of the GNU General Public License as published by
jaroslav@264
     7
 * the Free Software Foundation, either version 3 of the License.
jaroslav@178
     8
 *
jaroslav@264
     9
 * This program is distributed in the hope that it will be useful,
jaroslav@264
    10
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
jaroslav@264
    11
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
jaroslav@264
    12
 * GNU General Public License for more details.
jaroslav@178
    13
 *
jaroslav@264
    14
 * You should have received a copy of the GNU General Public License
jaroslav@264
    15
 * along with this program. Look for COPYING file in the top folder.
jaroslav@264
    16
 * If not, see http://www.gnu.org/licenses/.
jaroslav@178
    17
 */
jaroslav@266
    18
jaroslav@178
    19
package cz.xelfi.quoridor.statistics;
jaroslav@178
    20
jaroslav@178
    21
import cz.xelfi.quoridor.webidor.Game;
jaroslav@178
    22
import cz.xelfi.quoridor.webidor.GameId;
jaroslav@178
    23
import cz.xelfi.quoridor.webidor.GameStatus;
jaroslav@178
    24
import cz.xelfi.quoridor.webidor.User;
jaroslav@178
    25
import cz.xelfi.quoridor.Move;
jaroslav@178
    26
import java.util.HashMap;
jaroslav@178
    27
import java.util.Map;
jaroslav@178
    28
import java.util.List;
jaroslav@178
    29
import java.util.ArrayList;
jaroslav@178
    30
import javax.xml.bind.annotation.XmlAttribute;
jaroslav@178
    31
jaroslav@178
    32
/**
jaroslav@178
    33
 *
jaroslav@178
    34
 * @author Martin Rexa
jaroslav@178
    35
 */
jaroslav@178
    36
public class OpeningTreeNode {
jaroslav@178
    37
    @XmlAttribute
jaroslav@178
    38
    String hashCode;
jaroslav@178
    39
    List<GameId> gameIds;
jaroslav@178
    40
    List<GameId> finishedGames;
jaroslav@178
    41
    Map<Move, OpeningTreeNode> children;
jaroslav@178
    42
jaroslav@178
    43
    public OpeningTreeNode(){
jaroslav@178
    44
    }
jaroslav@178
    45
jaroslav@178
    46
    public OpeningTreeNode(String hashCode){
jaroslav@178
    47
        this.hashCode = hashCode;
jaroslav@178
    48
        gameIds = new ArrayList<GameId>();
jaroslav@178
    49
        finishedGames = new ArrayList<GameId>();
jaroslav@178
    50
        children = new HashMap<Move,OpeningTreeNode>();
jaroslav@178
    51
    }
jaroslav@178
    52
jaroslav@178
    53
    public OpeningTreeNode addGame(Game game){
jaroslav@178
    54
        gameIds.add(game.getId());
jaroslav@178
    55
        return this;
jaroslav@178
    56
    }
jaroslav@178
    57
jaroslav@178
    58
    public OpeningTreeNode addFinishedGame(Game game){
jaroslav@178
    59
        finishedGames.add(game.getId());
jaroslav@178
    60
        return this;
jaroslav@178
    61
    }
jaroslav@178
    62
jaroslav@178
    63
    public OpeningTreeNode addGameId(GameId gameId){
jaroslav@178
    64
        gameIds.add(gameId);
jaroslav@178
    65
        return this;
jaroslav@178
    66
    }
jaroslav@178
    67
jaroslav@178
    68
    public OpeningTreeNode addChild(Move move, OpeningTreeNode child){
jaroslav@178
    69
        children.put(move, child);
jaroslav@178
    70
        return this;
jaroslav@178
    71
    }
jaroslav@178
    72
jaroslav@178
    73
    public OpeningNodeView getView(User user){
jaroslav@178
    74
        return getView(user.getId());
jaroslav@178
    75
    }
jaroslav@178
    76
jaroslav@178
    77
    public OpeningNodeView getView(String userId){
jaroslav@178
    78
        OpeningNodeView view = new OpeningNodeView(hashCode);
jaroslav@178
    79
        for(Map.Entry<Move, OpeningTreeNode> e: children.entrySet()){
jaroslav@178
    80
            for(GameId gId: e.getValue().gameIds){
jaroslav@178
    81
                if(User.canSee(gId, userId)){
jaroslav@178
    82
                    view.processGame(e.getKey(), e.getValue().hashCode, gId);
jaroslav@178
    83
                }
jaroslav@178
    84
            }
jaroslav@178
    85
        }
jaroslav@178
    86
        for(GameId gId: finishedGames){
martin@191
    87
            if(User.canSee(gId, userId)){
martin@191
    88
                view.addFinishedGame(gId);
martin@191
    89
            }
jaroslav@178
    90
        }
jaroslav@178
    91
        return view;
jaroslav@178
    92
    }
jaroslav@178
    93
jaroslav@178
    94
    public List<GameId> filterGames(String userId, GameStatus status){
jaroslav@178
    95
        List<GameId> result = new ArrayList<GameId>();
jaroslav@178
    96
        for(GameId gId: gameIds){
jaroslav@178
    97
            if(User.canSee(gId, userId))
jaroslav@178
    98
                if(gId.getStatus().equals(status))
jaroslav@178
    99
                    result.add(gId);
jaroslav@178
   100
        }
jaroslav@178
   101
        return result;
jaroslav@178
   102
    }
jaroslav@178
   103
jaroslav@178
   104
    public List<GameId> getGameIds(){
jaroslav@178
   105
        return gameIds;
jaroslav@178
   106
    }
jaroslav@178
   107
jaroslav@178
   108
//    @XmlAttribute
jaroslav@178
   109
//    @XmlElement
jaroslav@178
   110
    public String getHashCode(){
jaroslav@178
   111
        return this.hashCode;
jaroslav@178
   112
    }
jaroslav@178
   113
jaroslav@178
   114
jaroslav@178
   115
    @Override
jaroslav@178
   116
    public boolean equals(Object obj) {
jaroslav@178
   117
        if (obj == null) {
jaroslav@178
   118
            return false;
jaroslav@178
   119
        }
jaroslav@178
   120
        if (getClass() != obj.getClass()) {
jaroslav@178
   121
            return false;
jaroslav@178
   122
        }
jaroslav@178
   123
        OpeningTreeNode other = (OpeningTreeNode)obj;
jaroslav@178
   124
        return hashCode.equals(other.hashCode);
jaroslav@178
   125
    }
jaroslav@178
   126
jaroslav@178
   127
    @Override
jaroslav@178
   128
    public int hashCode() {
jaroslav@178
   129
        return hashCode.hashCode();
jaroslav@178
   130
    }
jaroslav@178
   131
jaroslav@178
   132
}