Write more labels on score axis of eval graph
[xboard.git] / evalgraph.c
1 /*
2  * evalgraph.c - Evaluation graph back-end part
3  *
4  * Author: Alessandro Scotti (Dec 2005)
5  *
6  * Copyright 2005 Alessandro Scotti
7  *
8  * ------------------------------------------------------------------------
9  *
10  * GNU XBoard is free software: you can redistribute it and/or modify
11  * it under the terms of the GNU General Public License as published by
12  * the Free Software Foundation, either version 3 of the License, or (at
13  * your option) any later version.
14  *
15  * GNU XBoard is distributed in the hope that it will be useful, but
16  * WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18  * General Public License for more details.
19  *
20  * You should have received a copy of the GNU General Public License
21  * along with this program. If not, see http://www.gnu.org/licenses/.  *
22  *
23  *------------------------------------------------------------------------
24  ** See the file ChangeLog for a revision history.  */
25
26 // code refactored by HGM to obtain front-end / back-end separation
27
28 #include "config.h"
29
30 #include <stdio.h>
31
32 #if STDC_HEADERS
33 # include <stdlib.h>
34 # include <string.h>
35 #else /* not STDC_HEADERS */
36 # if HAVE_STRING_H
37 #  include <string.h>
38 # else /* not HAVE_STRING_H */
39 #  include <strings.h>
40 # endif /* not HAVE_STRING_H */
41 #endif /* not STDC_HEADERS */
42
43 #include "common.h"
44 #include "frontend.h"
45 #include "backend.h"
46 #include "evalgraph.h"
47
48 /* Module globals */
49 ChessProgramStats_Move * currPvInfo;
50 int currFirst = 0;
51 int currLast = 0;
52 int currCurrent = -1;
53 int range = 1;
54
55 int nWidthPB = 0;
56 int nHeightPB = 0;
57
58 int MarginX = 18;
59 int MarginW = 4;
60 int MarginH = 4;
61
62 // back-end
63 static void DrawLine( int x1, int y1, int x2, int y2, int penType )
64 {
65     DrawSegment( x1, y1, NULL, NULL, PEN_NONE );
66     DrawSegment( x2, y2, NULL, NULL, penType );
67 }
68
69 // back-end
70 static void DrawLineEx( int x1, int y1, int x2, int y2, int penType )
71 {
72     int savX, savY;
73     DrawSegment( x1, y1, &savX, &savY, PEN_NONE );
74     DrawSegment( x2, y2, NULL, NULL, penType );
75     DrawSegment( savX, savY, NULL, NULL, PEN_NONE );
76 }
77
78 // back-end
79 static int GetPvScore( int index )
80 {
81     int score = currPvInfo[ index ].score;
82
83     if( index & 1 ) score = -score; /* Flip score for black */
84
85     return score;
86 }
87
88 char* MakeEvalTitle( char * title)
89 {
90     int score = currPvInfo[ currCurrent ].score;
91     int depth = currPvInfo[ currCurrent ].depth;
92     static char buf[MSG_SIZ];
93
94     if( depth <=0 ) return title;
95     if( currCurrent & 1 ) score = -score; /* Flip score for black */
96     snprintf(buf, MSG_SIZ, "%s {%s%.2f/%-2d %d}", title, score>0 ? "+" : " ", score/100., depth, (currPvInfo[currCurrent].time+50)/100);
97
98     return buf;
99 }
100
101 // back-end
102 /*
103     For a centipawn value, this function returns the height of the corresponding
104     histogram, centered on the reference axis.
105
106     Note: height can be negative!
107 */
108 static int GetValueY( int value )
109 {
110     if( value < -range*700 ) value = -range*700;
111     if( value > +range*700 ) value = +range*700;
112     if(value > 100*range)  value += appData.zoom * 100 - 100*range; else
113     if(value < -100*range) value -= appData.zoom * 100 - 100*range; else
114         value *= appData.zoom;
115     return (nHeightPB / 2) - (int)(value * (nHeightPB - 2*MarginH) / ((1200. + 200.*appData.zoom)*range));
116 }
117
118 // the brush selection is made part of the DrawLine, by passing a style argument
119 // the wrapper for doing the text output makes this back-end
120 static void DrawAxisSegmentHoriz( int value, Boolean drawValue )
121 {
122     int y = GetValueY( range*value*100 );
123
124     if( drawValue ) {
125         char buf[MSG_SIZ], *b = buf;
126
127         if( value > 0 ) *b++ = '+';
128         sprintf(b, "%d", range*value);
129
130         DrawEvalText(buf, strlen(buf), y);
131     }
132     // [HGM] counts on DrawEvalText to have select transparent background for dotted line!
133     DrawLine( MarginX, y, MarginX + MarginW, y, PEN_BLACK ); // Y-axis tick marks
134     DrawLine( MarginX + MarginW, y, nWidthPB - MarginW, y, PEN_DOTTED ); // hor grid
135 }
136
137 // The DrawLines again must select their own brush.
138 // the initial brush selection is useless? BkMode needed for dotted line and text
139 static void DrawAxis()
140 {
141     int cy = nHeightPB / 2, space = nHeightPB/(6 + appData.zoom);
142     
143     DrawAxisSegmentHoriz( +5, TRUE );
144     DrawAxisSegmentHoriz( +3, space >= 20 );
145     DrawAxisSegmentHoriz( +1, space >= 20 && space*appData.zoom >= 40 );
146     DrawAxisSegmentHoriz(  0, TRUE );
147     DrawAxisSegmentHoriz( -1, space >= 20 && space*appData.zoom >= 40 );
148     DrawAxisSegmentHoriz( -3, space >= 20 );
149     DrawAxisSegmentHoriz( -5, TRUE );
150
151     DrawLine( MarginX + MarginW, cy, nWidthPB - MarginW, cy, PEN_BLACK ); // x-axis
152     DrawLine( MarginX + MarginW, MarginH, MarginX + MarginW, nHeightPB - MarginH, PEN_BLACK ); // y-axis
153 }
154
155 // back-end
156 static void DrawHistogram( int x, int y, int width, int value, int side )
157 {
158     int left, top, right, bottom;
159
160     if( value > -appData.evalThreshold*range && value < +appData.evalThreshold*range ) return;
161
162     left = x;
163     right = left + width + 1;
164
165     if( value > 0 ) {
166         top = GetValueY( value );
167         bottom = y+1;
168     }
169     else {
170         top = y;
171         bottom = GetValueY( value ) + 1;
172     }
173
174
175     if( width == MIN_HIST_WIDTH ) {
176         right--;
177         DrawRectangle( left, top, right, bottom, side, FILLED );
178     }
179     else {
180         DrawRectangle( left, top, right, bottom, side, OPEN );
181     }
182 }
183
184 // back-end
185 static void DrawSeparator( int index, int x )
186 {
187     if( index > 0 ) {
188         if( index == currCurrent ) {
189             DrawLineEx( x, MarginH, x, nHeightPB - MarginH, PEN_BLUEDOTTED );
190         }
191         else if( (index % 20) == 0 ) {
192             DrawLineEx( x, MarginH, x, nHeightPB - MarginH, PEN_DOTTED );
193         }
194     }
195 }
196
197 // made back-end by replacing MoveToEx and LineTo by DrawSegment
198 /* Actually draw histogram as a diagram, cause there's too much data */
199 static void DrawHistogramAsDiagram( int cy, int paint_width, int hist_count )
200 {
201     double step;
202     int i;
203
204     /* Rescale the graph every few moves (as opposed to every move) */
205     hist_count -= hist_count % 8;
206     hist_count += 8;
207     hist_count /= 2;
208
209     step = (double) paint_width / (hist_count + 1);
210
211     for( i=0; i<2; i++ ) {
212         int index = currFirst;
213         int side = (currCurrent + i + 1) & 1; /* Draw current side last */
214         double x = MarginX + MarginW;
215
216         if( (index & 1) != side ) {
217             x += step / 2;
218             index++;
219         }
220
221         DrawSegment( (int) x, cy, NULL, NULL, PEN_NONE );
222
223         index += 2;
224
225         while( index < currLast ) {
226             x += step;
227
228             DrawSeparator( index, (int) x );
229
230             /* Extend line up to current point */
231             if( currPvInfo[index].depth > 0 ) {
232                 DrawSegment((int) x, GetValueY( GetPvScore(index) ), NULL, NULL, PEN_BOLD + side );
233             }
234
235             index += 2;
236         }
237     }
238 }
239
240 // back-end, delete pen selection
241 static void DrawHistogramFull( int cy, int hist_width, int hist_count )
242 {
243     int i;
244
245 //    SelectObject( hdcPB, GetStockObject(BLACK_PEN) );
246
247     for( i=0; i<hist_count; i++ ) {
248         int index = currFirst + i;
249         int x = MarginX + MarginW + index * hist_width;
250
251         /* Draw a separator every 10 moves */
252         DrawSeparator( index, x );
253
254         /* Draw histogram */
255         if( currPvInfo[i].depth > 0 ) {
256             DrawHistogram( x, cy, hist_width, GetPvScore(index), index & 1 );
257         }
258     }
259 }
260
261 typedef struct {
262     int cy;
263     int hist_width;
264     int hist_count;
265     int paint_width;
266 } VisualizationData;
267
268 // back-end
269 static Boolean InitVisualization( VisualizationData * vd )
270 {
271     Boolean result = FALSE;
272
273     vd->cy = nHeightPB / 2;
274     vd->hist_width = MIN_HIST_WIDTH;
275     vd->hist_count = currLast - currFirst;
276     vd->paint_width = nWidthPB - MarginX - 2*MarginW;
277
278     if( vd->hist_count > 0 ) {
279         result = TRUE;
280
281         /* Compute width */
282         vd->hist_width = vd->paint_width / vd->hist_count;
283
284         if( vd->hist_width > MAX_HIST_WIDTH ) vd->hist_width = MAX_HIST_WIDTH;
285
286         vd->hist_width -= vd->hist_width % 2;
287     }
288
289     return result;
290 }
291
292 // back-end
293 static void DrawHistograms()
294 {
295     VisualizationData vd;
296
297     if( InitVisualization( &vd ) ) {
298         if( vd.hist_width < MIN_HIST_WIDTH ) {
299             DrawHistogramAsDiagram( vd.cy, vd.paint_width, vd.hist_count );
300         }
301         else {
302             DrawHistogramFull( vd.cy, vd.hist_width, vd.hist_count );
303         }
304     }
305 }
306
307 // back-end
308 int GetMoveIndexFromPoint( int x, int y )
309 {
310     int result = -1;
311     int start_x = MarginX + MarginW;
312     VisualizationData vd;
313
314     if( x >= start_x && InitVisualization( &vd ) ) {
315         /* Almost an hack here... we duplicate some of the paint logic */
316         if( vd.hist_width < MIN_HIST_WIDTH ) {
317             double step;
318
319             vd.hist_count -= vd.hist_count % 8;
320             vd.hist_count += 8;
321             vd.hist_count /= 2;
322
323             step = (double) vd.paint_width / (vd.hist_count + 1);
324             step /= 2;
325
326             result = (int) (0.5 + (double) (x - start_x) / step);
327         }
328         else {
329             result = (x - start_x) / vd.hist_width;
330         }
331     }
332
333     if( result >= currLast ) {
334         result = -1;
335     }
336
337     return result;
338 }
339
340 // init and display part split of so they can be moved to front end
341 void PaintEvalGraph( void )
342 {
343     VariantClass v = gameInfo.variant;
344     range = (gameInfo.holdingsWidth && v != VariantSuper && v != VariantGreat && v != VariantSChess) ? 2 : 1; // [HGM] double range in drop games
345     /* Draw */
346     DrawRectangle(0, 0, nWidthPB, nHeightPB, 2, FILLED);
347     DrawAxis();
348     DrawHistograms();
349 }
350