Merge branch 'terencehill/vehicle_hud_fix' into 'master'
[xonotic/xonotic-data.pk3dir.git] / qcsrc / menu / xonotic / screenshotlist.qc
1 #ifndef SCREENSHOTLIST_H
2 #define SCREENSHOTLIST_H
3 #include "listbox.qc"
4 CLASS(XonoticScreenshotList, XonoticListBox)
5         METHOD(XonoticScreenshotList, configureXonoticScreenshotList, void(entity))
6         ATTRIB(XonoticScreenshotList, rowsPerItem, float, 1)
7         METHOD(XonoticScreenshotList, resizeNotify, void(entity, vector, vector, vector, vector))
8         METHOD(XonoticScreenshotList, setSelected, void(entity, float))
9         METHOD(XonoticScreenshotList, draw, void(entity))
10         METHOD(XonoticScreenshotList, drawListBoxItem, void(entity, int, vector, bool, bool))
11         METHOD(XonoticScreenshotList, getScreenshots, void(entity))
12         METHOD(XonoticScreenshotList, previewScreenshot, void(entity))
13         METHOD(XonoticScreenshotList, startScreenshot, void(entity))
14         METHOD(XonoticScreenshotList, screenshotName, string(entity, float))
15         METHOD(XonoticScreenshotList, doubleClickListBoxItem, void(entity, float, vector))
16         METHOD(XonoticScreenshotList, keyDown, float(entity, float, float, float))
17         METHOD(XonoticScreenshotList, destroy, void(entity))
18         METHOD(XonoticScreenshotList, showNotify, void(entity))
19         ATTRIB(XonoticScreenshotList, listScreenshot, float, -1)
20         ATTRIB(XonoticScreenshotList, realFontSize, vector, '0 0 0')
21         ATTRIB(XonoticScreenshotList, columnNameOrigin, float, 0)
22         ATTRIB(XonoticScreenshotList, columnNameSize, float, 0)
23         ATTRIB(XonoticScreenshotList, realUpperMargin, float, 0)
24         ATTRIB(XonoticScreenshotList, origin, vector, '0 0 0')
25         ATTRIB(XonoticScreenshotList, itemAbsSize, vector, '0 0 0')
26         ATTRIB(XonoticScreenshotList, filterString, string, string_null)
27         ATTRIB(XonoticScreenshotList, filterBox, entity, NULL)
28         ATTRIB(XonoticScreenshotList, filterTime, float, 0)
29
30         ATTRIB(XonoticScreenshotList, newScreenshotTime, float, 0)
31         ATTRIB(XonoticScreenshotList, newSlideShowScreenshotTime, float, 0)
32         ATTRIB(XonoticScreenshotList, prevSelectedItem, float, 0)
33
34         ATTRIB(XonoticScreenshotList, screenshotBrowserDialog, entity, NULL)
35         ATTRIB(XonoticScreenshotList, screenshotPreview, entity, NULL)
36         ATTRIB(XonoticScreenshotList, screenshotViewerDialog, entity, NULL)
37         METHOD(XonoticScreenshotList, goScreenshot, void(entity, float))
38         METHOD(XonoticScreenshotList, startSlideShow, void(entity))
39         METHOD(XonoticScreenshotList, stopSlideShow, void(entity))
40 ENDCLASS(XonoticScreenshotList)
41
42 entity makeXonoticScreenshotList();
43 void StartScreenshot_Click(entity btn, entity me);
44 void ScreenshotList_Refresh_Click(entity btn, entity me);
45 void ScreenshotList_Filter_Would_Change(entity box, entity me);
46 void ScreenshotList_Filter_Change(entity box, entity me);
47 #endif
48
49 #ifdef IMPLEMENTATION
50
51 entity makeXonoticScreenshotList()
52 {
53         entity me;
54         me = NEW(XonoticScreenshotList);
55         me.configureXonoticScreenshotList(me);
56         return me;
57 }
58
59 void XonoticScreenshotList_configureXonoticScreenshotList(entity me)
60 {
61         me.configureXonoticListBox(me);
62         me.getScreenshots(me);
63 }
64
65 string XonoticScreenshotList_screenshotName(entity me, float i)
66 {
67         string s;
68         s = bufstr_get(me.listScreenshot, i);
69
70         if(substring(s, 0, 1) == "/")
71                 s = substring(s, 1, strlen(s) - 1);  // remove the first forward slash
72
73         return s;
74 }
75
76 // if subdir is true look in subdirectories too (1 level)
77 void getScreenshots_for_ext(entity me, string ext, float subdir)
78 {
79         string s;
80         if (subdir)
81                 s="screenshots/*/";
82         else
83                 s="screenshots/";
84         if(me.filterString)
85                 s=strcat(s, me.filterString, ext);
86         else
87                 s=strcat(s, "*", ext);
88
89         float list, i, n;
90         list = search_begin(s, false, true);
91         if(list >= 0)
92         {
93                 n = search_getsize(list);
94                 for(i = 0; i < n; ++i)
95                 {
96                         s = search_getfilename(list, i); // get initial full file name
97                         s = substring(s, 12, (strlen(s) - 12 - 4)); // remove "screenshots/" prefix and ".<ext>" suffix
98                         s = strdecolorize(s); // remove any pre-existing colors
99                         if(subdir)
100                         {
101                                 s = strreplace("/", "^7/", s); // clear colors at the forward slash
102                                 s = strcat("/", rgb_to_hexcolor(SKINCOLOR_SCREENSHOTLIST_SUBDIR), s); // add a forward slash for sorting, then color
103                                 bufstr_add(me.listScreenshot, s, true);
104                         }
105                         else { bufstr_add(me.listScreenshot, s, true); }
106                 }
107                 search_end(list);
108         }
109
110         if (subdir)
111                 getScreenshots_for_ext(me, ext, false);
112 }
113
114 void XonoticScreenshotList_getScreenshots(entity me)
115 {
116         if (me.listScreenshot >= 0)
117                 buf_del(me.listScreenshot);
118         me.listScreenshot = buf_create();
119         if (me.listScreenshot < 0)
120         {
121                 me.nItems = 0;
122                 return;
123         }
124         getScreenshots_for_ext(me, ".jpg", true);
125         getScreenshots_for_ext(me, ".tga", true);
126         getScreenshots_for_ext(me, ".png", true);
127         me.nItems = buf_getsize(me.listScreenshot);
128         if(me.nItems > 0)
129                 buf_sort(me.listScreenshot, 128, false);
130 }
131
132 void XonoticScreenshotList_destroy(entity me)
133 {
134         if(me.nItems > 0)
135                 buf_del(me.listScreenshot);
136 }
137
138 void XonoticScreenshotList_resizeNotify(entity me, vector relOrigin, vector relSize, vector absOrigin, vector absSize)
139 {
140         me.itemAbsSize = '0 0 0';
141         SUPER(XonoticScreenshotList).resizeNotify(me, relOrigin, relSize, absOrigin, absSize);
142
143         me.realFontSize_y = me.fontSize / (me.itemAbsSize_y = (absSize.y * me.itemHeight));
144         me.realFontSize_x = me.fontSize / (me.itemAbsSize_x = (absSize.x * (1 - me.controlWidth)));
145         me.realUpperMargin = 0.5 * (1 - me.realFontSize.y);
146
147         me.columnNameOrigin = me.realFontSize.x;
148         me.columnNameSize = 1 - 2 * me.realFontSize.x;
149 }
150
151 void XonoticScreenshotList_setSelected(entity me, float i)
152 {
153         if (me.newSlideShowScreenshotTime)
154                 me.startSlideShow(me);
155         me.prevSelectedItem = me.selectedItem;
156         SUPER(XonoticScreenshotList).setSelected(me, i);
157         if (me.pressed && me.selectedItem != me.prevSelectedItem)
158         {
159                 // while dragging the scrollbar (or an item)
160                 // for a smooth mouse movement do not load immediately the new selected images
161                 me.newScreenshotTime = time + 0.22; // dragging an item we need a delay > 0.2 (from listbox: me.dragScrollTimer = time + 0.2;)
162         }
163         else if (time > me.newScreenshotTime)
164         {
165                 me.newScreenshotTime = 0;
166                 me.previewScreenshot(me); // load the preview on selection change
167         }
168 }
169
170 void XonoticScreenshotList_drawListBoxItem(entity me, int i, vector absSize, bool isSelected, bool isFocused)
171 {
172         string s;
173         if(isSelected)
174                 draw_Fill('0 0 0', '1 1 0', SKINCOLOR_LISTBOX_SELECTED, SKINALPHA_LISTBOX_SELECTED);
175         else if(isFocused)
176         {
177                 me.focusedItemAlpha = getFadedAlpha(me.focusedItemAlpha, SKINALPHA_LISTBOX_FOCUSED, SKINFADEALPHA_LISTBOX_FOCUSED);
178                 draw_Fill('0 0 0', '1 1 0', SKINCOLOR_LISTBOX_FOCUSED, me.focusedItemAlpha);
179         }
180
181         s = me.screenshotName(me,i);
182         s = draw_TextShortenToWidth(s, me.columnNameSize, 0, me.realFontSize);
183         draw_Text(me.realUpperMargin * eY + (me.columnNameOrigin + 0.00 * (me.columnNameSize - draw_TextWidth(s, 0, me.realFontSize))) * eX, s, me.realFontSize, '1 1 1', SKINALPHA_TEXT, 1);
184 }
185
186 void XonoticScreenshotList_showNotify(entity me)
187 {
188         me.getScreenshots(me);
189         me.previewScreenshot(me);
190 }
191
192 void ScreenshotList_Refresh_Click(entity btn, entity me)
193 {
194         me.getScreenshots(me);
195         me.setSelected(me, 0); //always select the first element after a list update
196 }
197
198 void ScreenshotList_Filter_Change(entity box, entity me)
199 {
200         if(me.filterString)
201                 strunzone(me.filterString);
202
203         if(box.text != "")
204         {
205                 if (strstrofs(box.text, "*", 0) >= 0 || strstrofs(box.text, "?", 0) >= 0)
206                         me.filterString = strzone(box.text);
207                 else
208                         me.filterString = strzone(strcat("*", box.text, "*"));
209         }
210         else
211                 me.filterString = string_null;
212
213         ScreenshotList_Refresh_Click(world, me);
214 }
215
216 void ScreenshotList_Filter_Would_Change(entity box, entity me)
217 {
218         me.filterBox = box;
219         me.filterTime = time + 0.5;
220 }
221
222 void XonoticScreenshotList_draw(entity me)
223 {
224         if (me.filterTime && time > me.filterTime)
225         {
226                 ScreenshotList_Filter_Change(me.filterBox, me);
227                 me.filterTime = 0;
228         }
229         if (me.newScreenshotTime && time > me.newScreenshotTime)
230         {
231                 me.previewScreenshot(me);
232                 me.newScreenshotTime = 0;
233         }
234         else if (me.newSlideShowScreenshotTime && time > me.newSlideShowScreenshotTime)
235         {
236                 if (me.selectedItem == me.nItems - 1) //last screenshot?
237                 {
238                         // restart from the first screenshot
239                         me.setSelected(me, 0);
240                         me.goScreenshot(me, +0);
241                 }
242                 else
243                         me.goScreenshot(me, +1);
244         }
245         SUPER(XonoticScreenshotList).draw(me);
246 }
247
248 void XonoticScreenshotList_startSlideShow(entity me)
249 {
250         me.newSlideShowScreenshotTime = time + 3;
251 }
252
253 void XonoticScreenshotList_stopSlideShow(entity me)
254 {
255         me.newSlideShowScreenshotTime = 0;
256 }
257
258 void XonoticScreenshotList_goScreenshot(entity me, float d)
259 {
260         if(!me.screenshotViewerDialog)
261                 return;
262         me.setSelected(me, me.selectedItem + d);
263         me.screenshotViewerDialog.loadScreenshot(me.screenshotViewerDialog, strcat("/screenshots/", strdecolorize(me.screenshotName(me,me.selectedItem))));
264 }
265
266 void XonoticScreenshotList_startScreenshot(entity me)
267 {
268         me.screenshotViewerDialog.loadScreenshot(me.screenshotViewerDialog, strcat("/screenshots/", strdecolorize(me.screenshotName(me,me.selectedItem))));
269         // pop up screenshot
270         DialogOpenButton_Click_withCoords(NULL, me.screenshotViewerDialog, me.origin + eX * (me.columnNameOrigin * me.size.x) + eY * ((me.itemHeight * me.selectedItem - me.scrollPos) * me.size.y), eY * me.itemAbsSize.y + eX * (me.itemAbsSize.x * me.columnNameSize));
271 }
272
273 void XonoticScreenshotList_previewScreenshot(entity me)
274 {
275         if(!me.screenshotBrowserDialog)
276                 return;
277         if (me.nItems <= 0)
278                 me.screenshotBrowserDialog.loadPreviewScreenshot(me.screenshotBrowserDialog, "");
279         else
280                 me.screenshotBrowserDialog.loadPreviewScreenshot(me.screenshotBrowserDialog, strcat("/screenshots/", strdecolorize(me.screenshotName(me,me.selectedItem))));
281 }
282
283 void StartScreenshot_Click(entity btn, entity me)
284 {
285         me.startScreenshot(me);
286 }
287
288 void XonoticScreenshotList_doubleClickListBoxItem(entity me, float i, vector where)
289 {
290         me.startScreenshot(me);
291 }
292
293 float XonoticScreenshotList_keyDown(entity me, float scan, float ascii, float shift)
294 {
295         if(scan == K_ENTER || scan == K_KP_ENTER || scan == K_MOUSE2 || scan == K_SPACE) {
296                 me.startScreenshot(me);
297                 return 1;
298         }
299         if(scan == K_MWHEELUP || scan == K_MWHEELDOWN)
300                 me.newScreenshotTime = time + 0.2;
301         return SUPER(XonoticScreenshotList).keyDown(me, scan, ascii, shift);
302 }
303 #endif