static void draw(void)
{
/* We don't push the matrix stack, we just keep adding rotations. */
rotate(yrot*0.01, xrot*0.01, zrot*0.01);
glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
glColorMask(GL_TRUE, GL_TRUE, GL_TRUE, GL_TRUE);
draw_all_surfaces();
draw_all_curves();
draw_all_points();
if (draw_axes)
draw_gl_axes_old(10, axis_length,
axis_thickness*0.01, axis_thickness*0.04, 0.1);
glFlush();
glutSwapBuffers();
glutPostRedisplay();
mouse_movement=0;
}
void
on_filesel_ok_clicked (GtkButton *button,
gpointer user_data)
{
gtk_widget_hide(fileselection);
const char *filename = gtk_file_selection_get_filename(GTK_FILE_SELECTION(fileselection));
if(fileselection_flag == SAVE)
{
FILE *fp = fopen(filename,"w");
if(fp)
{
for(list<point>::iterator p = point_list.begin(); p != point_list.end();p++)
fprintf(fp,"%d 1:%f 2:%f\n", p->value, p->x, p->y);
fclose(fp);
}
}
else if(fileselection_flag == LOAD)
{
FILE *fp = fopen(filename,"r");
if(fp)
{
clear_all();
char buf[4096];
while(fgets(buf,sizeof(buf),fp))
{
int v;
double x,y;
if(sscanf(buf,"%d%*d:%lf%*d:%lf",&v,&x,&y)!=3)
break;
point p = {x,y,v};
point_list.push_back(p);
}
fclose(fp);
draw_all_points();
}
}
}
void
on_button_run_clicked (GtkButton *button,
gpointer user_data)
{
// guard
if(point_list.empty()) return;
svm_parameter param;
int i,j;
// default values
param.svm_type = C_SVC;
param.kernel_type = RBF;
param.degree = 3;
param.gamma = 0;
param.coef0 = 0;
param.nu = 0.5;
param.cache_size = 40;
param.C = 1;
param.eps = 1e-3;
param.p = 0.1;
param.shrinking = 1;
param.nr_weight = 0;
param.weight_label = NULL;
param.weight = NULL;
// parse options
const char *p = gtk_entry_get_text(GTK_ENTRY(entry_option));
while (1) {
while (*p && *p != '-')
p++;
if (*p == '\0')
break;
p++;
switch (*p++) {
case 's':
param.svm_type = atoi(p);
break;
case 't':
param.kernel_type = atoi(p);
break;
case 'd':
param.degree = atof(p);
break;
case 'g':
param.gamma = atof(p);
break;
case 'r':
param.coef0 = atof(p);
break;
case 'n':
param.nu = atof(p);
break;
case 'm':
param.cache_size = atof(p);
break;
case 'c':
param.C = atof(p);
break;
case 'e':
param.eps = atof(p);
break;
case 'p':
param.p = atof(p);
break;
case 'h':
param.shrinking = atoi(p);
break;
case 'w':
++param.nr_weight;
param.weight_label = (int *)realloc(param.weight_label,sizeof(int)*param.nr_weight);
param.weight = (double *)realloc(param.weight,sizeof(double)*param.nr_weight);
param.weight_label[param.nr_weight-1] = atoi(p);
while(*p && !isspace(*p)) ++p;
param.weight[param.nr_weight-1] = atof(p);
break;
}
}
// build problem
svm_problem prob;
prob.l = point_list.size();
prob.y = new double[prob.l];
if(param.svm_type == EPSILON_SVR ||
param.svm_type == NU_SVR)
{
if(param.gamma == 0) param.gamma = 1;
svm_node *x_space = new svm_node[2 * prob.l];
prob.x = new svm_node *[prob.l];
i = 0;
for (list <point>::iterator q = point_list.begin(); q != point_list.end(); q++, i++)
{
x_space[2 * i].index = 1;
x_space[2 * i].value = q->x;
x_space[2 * i + 1].index = -1;
prob.x[i] = &x_space[2 * i];
prob.y[i] = q->y;
}
// build model & classify
svm_model *model = svm_train(&prob, ¶m);
svm_node x[2];
x[0].index = 1;
x[1].index = -1;
int *j = new int[XLEN];
for (i = 0; i < XLEN; i++)
{
x[0].value = (double) i / XLEN;
j[i] = (int)(YLEN*svm_predict(model, x));
}
gdk_gc_set_foreground(gc,&colors[0]);
gdk_draw_line(pixmap,gc,0,0,0,YLEN-1);
gdk_draw_line(draw_main->window,gc,0,0,0,YLEN-1);
int p = (int)(param.p * YLEN);
for(i = 1; i < XLEN; i++)
{
gdk_gc_set_foreground(gc,&colors[0]);
gdk_draw_line(pixmap,gc,i,0,i,YLEN-1);
gdk_draw_line(draw_main->window,gc,i,0,i,YLEN-1);
gdk_gc_set_foreground(gc,&colors[5]);
gdk_draw_line(pixmap,gc,i-1,j[i-1],i,j[i]);
gdk_draw_line(draw_main->window,gc,i-1,j[i-1],i,j[i]);
if(param.svm_type == EPSILON_SVR)
{
gdk_gc_set_foreground(gc,&colors[2]);
gdk_draw_line(pixmap,gc,i-1,j[i-1]+p,i,j[i]+p);
gdk_draw_line(draw_main->window,gc,i-1,j[i-1]+p,i,j[i]+p);
gdk_gc_set_foreground(gc,&colors[2]);
gdk_draw_line(pixmap,gc,i-1,j[i-1]-p,i,j[i]-p);
gdk_draw_line(draw_main->window,gc,i-1,j[i-1]-p,i,j[i]-p);
}
}
svm_destroy_model(model);
delete[] j;
delete[] x_space;
delete[] prob.x;
delete[] prob.y;
}
else
{
if(param.gamma == 0) param.gamma = 0.5;
svm_node *x_space = new svm_node[3 * prob.l];
prob.x = new svm_node *[prob.l];
i = 0;
for (list <point>::iterator q = point_list.begin(); q != point_list.end(); q++, i++)
{
x_space[3 * i].index = 1;
x_space[3 * i].value = q->x;
x_space[3 * i + 1].index = 2;
x_space[3 * i + 1].value = q->y;
x_space[3 * i + 2].index = -1;
prob.x[i] = &x_space[3 * i];
prob.y[i] = q->value;
}
// build model & classify
svm_model *model = svm_train(&prob, ¶m);
svm_node x[3];
x[0].index = 1;
x[1].index = 2;
x[2].index = -1;
for (i = 0; i < XLEN; i++)
for (j = 0; j < YLEN; j++) {
x[0].value = (double) i / XLEN;
x[1].value = (double) j / YLEN;
double d = svm_predict(model, x);
gdk_gc_set_foreground(gc,&colors[(int)d]);
gdk_draw_point(pixmap,gc,i,j);
gdk_draw_point(draw_main->window,gc,i,j);
}
svm_destroy_model(model);
delete[] x_space;
delete[] prob.x;
delete[] prob.y;
}
free(param.weight_label);
free(param.weight);
draw_all_points();
}
