f069ba924675a339fbe503e7c51eebc3449fbde3
[chesshog.git] / chesshog-format / src / main / java / org / hedgecode / chess / img / piece / PieceSetLoader.java
1 /*
2  * Copyright (c) 2018-2019. Developed by Hedgecode.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *   http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package org.hedgecode.chess.img.piece;
18
19 import java.io.File;
20 import java.io.IOException;
21 import java.util.HashMap;
22 import java.util.Map;
23
24 import javax.imageio.ImageIO;
25
26 import org.hedgecode.chess.img.AbstractImageLoader;
27 import org.hedgecode.chess.img.ImageConstants;
28 import org.hedgecode.chess.img.ImageException;
29 import org.hedgecode.chess.img.ImageFilter;
30
31 /**
32  *
33  *
34  * @author Dmitry Samoshin aka gotty
35  */
36 public class PieceSetLoader extends AbstractImageLoader {
37
38     private static final File PIECES_PATH = new File(IMAGES_PATH, ImageConstants.RESOURCE_PIECES_DIR);
39
40     private static final String[] PIECES_NAMES = PieceSet.getAllPieceNames();
41
42     private Map<String, PieceSet> pieceSetMap = new HashMap<>();
43
44     @Override
45     public PieceSet load(String name) throws ImageException {
46         PieceSet pieceSet = null;
47         if (Type.STATELESS.equals(loadType()) || !pieceSetMap.containsKey(name)) {
48             File piecePath = new File(PIECES_PATH, name);
49             if (piecePath.exists() && piecePath.isDirectory()) {
50                 pieceSet = loadPieces(piecePath);
51             }
52             if (Type.STATEFUL.equals(loadType()) && pieceSet != null) {
53                 pieceSetMap.put(name, pieceSet);
54             }
55         } else {
56             pieceSet = pieceSetMap.get(name);
57         }
58         return pieceSet;
59     }
60
61     @Override
62     public void unload(String name) {
63         if (Type.STATEFUL.equals(loadType()) && pieceSetMap.containsKey(name)) {
64             pieceSetMap.remove(name);
65         }
66     }
67
68     @Override
69     protected void clear() {
70         pieceSetMap.clear();
71     }
72
73     private PieceSet loadPieces(File piecePath) throws ImageException {
74         PieceSet pieceSet = PieceSet.create();
75         File[] images = piecePath.listFiles(new ImageFilter(PIECES_NAMES));
76         if (images != null) {
77             for (File image : images) {
78                 if (image.isFile()) {
79                     String filename = image.getName();
80                     try {
81                         pieceSet.add(
82                                 filename.substring(0, filename.lastIndexOf('.')),
83                                 ImageIO.read(image)
84                         );
85                     } catch (IOException e) {
86                         throw new ImageException(e.getMessage()); // todo: locale
87                     }
88                 }
89             }
90         }
91         return pieceSet.isFull() ? pieceSet : null;
92     }
93
94 }