1package eu.siacs.conversations.services;
2
3import android.graphics.Bitmap;
4import android.graphics.Canvas;
5import android.graphics.Paint;
6import android.graphics.Rect;
7import android.graphics.Typeface;
8import android.net.Uri;
9
10import java.util.ArrayList;
11import java.util.List;
12import java.util.Locale;
13
14import eu.siacs.conversations.entities.Account;
15import eu.siacs.conversations.entities.Bookmark;
16import eu.siacs.conversations.entities.Contact;
17import eu.siacs.conversations.entities.Conversation;
18import eu.siacs.conversations.entities.ListItem;
19import eu.siacs.conversations.entities.MucOptions;
20
21public class AvatarService {
22
23 private static final int FG_COLOR = 0xFFFAFAFA;
24 private static final int TRANSPARENT = 0x00000000;
25
26 private static final String PREFIX_CONTACT = "contact";
27 private static final String PREFIX_CONVERSATION = "conversation";
28 private static final String PREFIX_ACCOUNT = "account";
29 private static final String PREFIX_GENERIC = "generic";
30
31 final private ArrayList<Integer> sizes = new ArrayList<>();
32
33 protected XmppConnectionService mXmppConnectionService = null;
34
35 public AvatarService(XmppConnectionService service) {
36 this.mXmppConnectionService = service;
37 }
38
39 public Bitmap get(final Contact contact, final int size) {
40 final String KEY = key(contact, size);
41 Bitmap avatar = this.mXmppConnectionService.getBitmapCache().get(KEY);
42 if (avatar != null) {
43 return avatar;
44 }
45 if (contact.getProfilePhoto() != null) {
46 avatar = mXmppConnectionService.getFileBackend().cropCenterSquare(Uri.parse(contact.getProfilePhoto()), size);
47 }
48 if (avatar == null && contact.getAvatar() != null) {
49 avatar = mXmppConnectionService.getFileBackend().getAvatar(contact.getAvatar(), size);
50 }
51 if (avatar == null) {
52 avatar = get(contact.getDisplayName(), size);
53 }
54 this.mXmppConnectionService.getBitmapCache().put(KEY, avatar);
55 return avatar;
56 }
57
58 public void clear(Contact contact) {
59 for (Integer size : sizes) {
60 this.mXmppConnectionService.getBitmapCache().remove(
61 key(contact, size));
62 }
63 }
64
65 private String key(Contact contact, int size) {
66 synchronized (this.sizes) {
67 if (!this.sizes.contains(size)) {
68 this.sizes.add(size);
69 }
70 }
71 return PREFIX_CONTACT + "_" + contact.getAccount().getJid() + "_"
72 + contact.getJid() + "_" + String.valueOf(size);
73 }
74
75 public Bitmap get(ListItem item, int size) {
76 if (item instanceof Contact) {
77 return get((Contact) item, size);
78 } else if (item instanceof Bookmark) {
79 Bookmark bookmark = (Bookmark) item;
80 if (bookmark.getConversation() != null) {
81 return get(bookmark.getConversation(), size);
82 } else {
83 return get(bookmark.getDisplayName(), size);
84 }
85 } else {
86 return get(item.getDisplayName(), size);
87 }
88 }
89
90 public Bitmap get(Conversation conversation, int size) {
91 if (conversation.getMode() == Conversation.MODE_SINGLE) {
92 return get(conversation.getContact(), size);
93 } else {
94 return get(conversation.getMucOptions(), size);
95 }
96 }
97
98 public void clear(Conversation conversation) {
99 if (conversation.getMode() == Conversation.MODE_SINGLE) {
100 clear(conversation.getContact());
101 } else {
102 clear(conversation.getMucOptions());
103 }
104 }
105
106 public Bitmap get(MucOptions mucOptions, int size) {
107 final String KEY = key(mucOptions, size);
108 Bitmap bitmap = this.mXmppConnectionService.getBitmapCache().get(KEY);
109 if (bitmap != null) {
110 return bitmap;
111 }
112 List<MucOptions.User> users = mucOptions.getUsers();
113 int count = users.size();
114 bitmap = Bitmap.createBitmap(size, size, Bitmap.Config.ARGB_8888);
115 Canvas canvas = new Canvas(bitmap);
116 bitmap.eraseColor(TRANSPARENT);
117
118 if (count == 0) {
119 String name = mucOptions.getConversation().getName();
120 String letter = name.substring(0, 1);
121 int color = this.getColorForName(name);
122 drawTile(canvas, letter, color, 0, 0, size, size);
123 } else if (count == 1) {
124 drawTile(canvas, users.get(0), 0, 0, size, size);
125 } else if (count == 2) {
126 drawTile(canvas, users.get(0), 0, 0, size / 2 - 1, size);
127 drawTile(canvas, users.get(1), size / 2 + 1, 0, size, size);
128 } else if (count == 3) {
129 drawTile(canvas, users.get(0), 0, 0, size / 2 - 1, size);
130 drawTile(canvas, users.get(1), size / 2 + 1, 0, size, size / 2 - 1);
131 drawTile(canvas, users.get(2), size / 2 + 1, size / 2 + 1, size,
132 size);
133 } else if (count == 4) {
134 drawTile(canvas, users.get(0), 0, 0, size / 2 - 1, size / 2 - 1);
135 drawTile(canvas, users.get(1), 0, size / 2 + 1, size / 2 - 1, size);
136 drawTile(canvas, users.get(2), size / 2 + 1, 0, size, size / 2 - 1);
137 drawTile(canvas, users.get(3), size / 2 + 1, size / 2 + 1, size,
138 size);
139 } else {
140 drawTile(canvas, users.get(0), 0, 0, size / 2 - 1, size / 2 - 1);
141 drawTile(canvas, users.get(1), 0, size / 2 + 1, size / 2 - 1, size);
142 drawTile(canvas, users.get(2), size / 2 + 1, 0, size, size / 2 - 1);
143 drawTile(canvas, "\u2026", 0xFF202020, size / 2 + 1, size / 2 + 1,
144 size, size);
145 }
146 this.mXmppConnectionService.getBitmapCache().put(KEY, bitmap);
147 return bitmap;
148 }
149
150 public void clear(MucOptions options) {
151 for (Integer size : sizes) {
152 this.mXmppConnectionService.getBitmapCache().remove(
153 key(options, size));
154 }
155 }
156
157 private String key(MucOptions options, int size) {
158 synchronized (this.sizes) {
159 if (!this.sizes.contains(size)) {
160 this.sizes.add(size);
161 }
162 }
163 return PREFIX_CONVERSATION + "_" + options.getConversation().getUuid()
164 + "_" + String.valueOf(size);
165 }
166
167 public Bitmap get(Account account, int size) {
168 final String KEY = key(account, size);
169 Bitmap avatar = mXmppConnectionService.getBitmapCache().get(KEY);
170 if (avatar != null) {
171 return avatar;
172 }
173 avatar = mXmppConnectionService.getFileBackend().getAvatar(
174 account.getAvatar(), size);
175 if (avatar == null) {
176 avatar = get(account.getJid().toString(), size);
177 }
178 mXmppConnectionService.getBitmapCache().put(KEY, avatar);
179 return avatar;
180 }
181
182 public void clear(Account account) {
183 for (Integer size : sizes) {
184 this.mXmppConnectionService.getBitmapCache().remove(
185 key(account, size));
186 }
187 }
188
189 private String key(Account account, int size) {
190 synchronized (this.sizes) {
191 if (!this.sizes.contains(size)) {
192 this.sizes.add(size);
193 }
194 }
195 return PREFIX_ACCOUNT + "_" + account.getUuid() + "_"
196 + String.valueOf(size);
197 }
198
199 public Bitmap get(final String name, final int size) {
200 final String KEY = key(name, size);
201 Bitmap bitmap = mXmppConnectionService.getBitmapCache().get(KEY);
202 if (bitmap != null) {
203 return bitmap;
204 }
205 bitmap = Bitmap.createBitmap(size, size, Bitmap.Config.ARGB_8888);
206 Canvas canvas = new Canvas(bitmap);
207 String letter = name.substring(0, 1);
208 int color = this.getColorForName(name);
209 drawTile(canvas, letter, color, 0, 0, size, size);
210 mXmppConnectionService.getBitmapCache().put(KEY, bitmap);
211 return bitmap;
212 }
213
214 private String key(String name, int size) {
215 synchronized (this.sizes) {
216 if (!this.sizes.contains(size)) {
217 this.sizes.add(size);
218 }
219 }
220 return PREFIX_GENERIC + "_" + name + "_" + String.valueOf(size);
221 }
222
223 private void drawTile(Canvas canvas, String letter, int tileColor,
224 int left, int top, int right, int bottom) {
225 letter = letter.toUpperCase(Locale.getDefault());
226 Paint tilePaint = new Paint(), textPaint = new Paint();
227 tilePaint.setColor(tileColor);
228 textPaint.setFlags(Paint.ANTI_ALIAS_FLAG);
229 textPaint.setColor(FG_COLOR);
230 textPaint.setTypeface(Typeface.create("sans-serif-light",
231 Typeface.NORMAL));
232 textPaint.setTextSize((float) ((right - left) * 0.8));
233 Rect rect = new Rect();
234
235 canvas.drawRect(new Rect(left, top, right, bottom), tilePaint);
236 textPaint.getTextBounds(letter, 0, 1, rect);
237 float width = textPaint.measureText(letter);
238 canvas.drawText(letter, (right + left) / 2 - width / 2, (top + bottom)
239 / 2 + rect.height() / 2, textPaint);
240 }
241
242 private void drawTile(Canvas canvas, MucOptions.User user, int left,
243 int top, int right, int bottom) {
244 Contact contact = user.getContact();
245 if (contact != null) {
246 Uri uri = null;
247 if (contact.getProfilePhoto() != null) {
248 uri = Uri.parse(contact.getProfilePhoto());
249 } else if (contact.getAvatar() != null) {
250 uri = mXmppConnectionService.getFileBackend().getAvatarUri(
251 contact.getAvatar());
252 }
253 if (uri != null) {
254 Bitmap bitmap = mXmppConnectionService.getFileBackend()
255 .cropCenter(uri, bottom - top, right - left);
256 if (bitmap != null) {
257 drawTile(canvas, bitmap, left, top, right, bottom);
258 } else {
259 String letter = user.getName().substring(0, 1);
260 int color = this.getColorForName(user.getName());
261 drawTile(canvas, letter, color, left, top, right, bottom);
262 }
263 } else {
264 String letter = user.getName().substring(0, 1);
265 int color = this.getColorForName(user.getName());
266 drawTile(canvas, letter, color, left, top, right, bottom);
267 }
268 } else {
269 String letter = user.getName().substring(0, 1);
270 int color = this.getColorForName(user.getName());
271 drawTile(canvas, letter, color, left, top, right, bottom);
272 }
273 }
274
275 private void drawTile(Canvas canvas, Bitmap bm, int dstleft, int dsttop,
276 int dstright, int dstbottom) {
277 Rect dst = new Rect(dstleft, dsttop, dstright, dstbottom);
278 canvas.drawBitmap(bm, null, dst, null);
279 }
280
281 private int getColorForName(String name) {
282 int holoColors[] = {0xFFe91e63, 0xFF9c27b0, 0xFF673ab7, 0xFF3f51b5,
283 0xFF5677fc, 0xFF03a9f4, 0xFF00bcd4, 0xFF009688, 0xFFff5722,
284 0xFF795548, 0xFF607d8b};
285 return holoColors[(int) ((name.hashCode() & 0xffffffffl) % holoColors.length)];
286 }
287
288}