| @@ -0,0 +1,447 @@ | |||
| #include <gtk/gtk.h> | |||
| #include <stdio.h> | |||
| #include <stdlib.h> | |||
| #include <string.h> | |||
| #include <ctype.h> | |||
| #include <list> | |||
| #include "callbacks.h" | |||
| #include "interface.h" | |||
| #include "../../svm.h" | |||
| using namespace std; | |||
| #define DEFAULT_PARAM "-t 2 -c 100" | |||
| #define XLEN 500 | |||
| #define YLEN 500 | |||
| GdkColor colors[] = | |||
| { | |||
| {0,0,0,0}, | |||
| {0,0,120<<8,120<<8}, | |||
| {0,120<<8,120<<8,0}, | |||
| {0,120<<8,0,120<<8}, | |||
| {0,0,200<<8,200<<8}, | |||
| {0,200<<8,200<<8,0}, | |||
| {0,200<<8,0,200<<8}, | |||
| }; | |||
| GdkGC *gc; | |||
| GdkPixmap *pixmap; | |||
| extern "C" GtkWidget *draw_main; | |||
| GtkWidget *draw_main; | |||
| extern "C" GtkWidget *entry_option; | |||
| GtkWidget *entry_option; | |||
| typedef struct { | |||
| double x, y; | |||
| signed char value; | |||
| } point; | |||
| list<point> point_list; | |||
| int current_value = 1; | |||
| extern "C" void svm_toy_initialize() | |||
| { | |||
| gboolean success[7]; | |||
| gdk_colormap_alloc_colors( | |||
| gdk_colormap_get_system(), | |||
| colors, | |||
| 7, | |||
| FALSE, | |||
| TRUE, | |||
| success); | |||
| gc = gdk_gc_new(draw_main->window); | |||
| pixmap = gdk_pixmap_new(draw_main->window,XLEN,YLEN,-1); | |||
| gdk_gc_set_foreground(gc,&colors[0]); | |||
| gdk_draw_rectangle(pixmap,gc,TRUE,0,0,XLEN,YLEN); | |||
| gtk_entry_set_text(GTK_ENTRY(entry_option),DEFAULT_PARAM); | |||
| } | |||
| void redraw_area(GtkWidget* widget, int x, int y, int w, int h) | |||
| { | |||
| gdk_draw_pixmap(widget->window, | |||
| gc, | |||
| pixmap, | |||
| x,y,x,y,w,h); | |||
| } | |||
| void draw_point(const point& p) | |||
| { | |||
| gdk_gc_set_foreground(gc,&colors[p.value+3]); | |||
| gdk_draw_rectangle(pixmap, gc, TRUE,int(p.x*XLEN),int(p.y*YLEN),4,4); | |||
| gdk_draw_rectangle(draw_main->window, gc, TRUE,int(p.x*XLEN),int(p.y*YLEN),4,4); | |||
| } | |||
| void draw_all_points() | |||
| { | |||
| for(list<point>::iterator p = point_list.begin(); p != point_list.end();p++) | |||
| draw_point(*p); | |||
| } | |||
| void clear_all() | |||
| { | |||
| point_list.clear(); | |||
| gdk_gc_set_foreground(gc,&colors[0]); | |||
| gdk_draw_rectangle(pixmap,gc,TRUE,0,0,XLEN,YLEN); | |||
| redraw_area(draw_main,0,0,XLEN,YLEN); | |||
| } | |||
| void | |||
| on_button_change_clicked (GtkButton *button, | |||
| gpointer user_data) | |||
| { | |||
| ++current_value; | |||
| if(current_value > 3) current_value = 1; | |||
| } | |||
| 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 = 100; | |||
| param.C = 1; | |||
| param.eps = 1e-3; | |||
| param.p = 0.1; | |||
| param.shrinking = 1; | |||
| param.probability = 0; | |||
| 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 = atoi(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 'b': | |||
| param.probability = 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.kernel_type == PRECOMPUTED) | |||
| { | |||
| } | |||
| else 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_free_and_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); | |||
| if (param.svm_type == ONE_CLASS && d<0) d=2; | |||
| 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_free_and_destroy_model(&model); | |||
| delete[] x_space; | |||
| delete[] prob.x; | |||
| delete[] prob.y; | |||
| } | |||
| free(param.weight_label); | |||
| free(param.weight); | |||
| draw_all_points(); | |||
| } | |||
| void | |||
| on_button_clear_clicked (GtkButton *button, | |||
| gpointer user_data) | |||
| { | |||
| clear_all(); | |||
| } | |||
| void | |||
| on_window1_destroy (GtkObject *object, | |||
| gpointer user_data) | |||
| { | |||
| gtk_exit(0); | |||
| } | |||
| gboolean | |||
| on_draw_main_button_press_event (GtkWidget *widget, | |||
| GdkEventButton *event, | |||
| gpointer user_data) | |||
| { | |||
| point p = {(double)event->x/XLEN, (double)event->y/YLEN, current_value}; | |||
| point_list.push_back(p); | |||
| draw_point(p); | |||
| return FALSE; | |||
| } | |||
| gboolean | |||
| on_draw_main_expose_event (GtkWidget *widget, | |||
| GdkEventExpose *event, | |||
| gpointer user_data) | |||
| { | |||
| redraw_area(widget, | |||
| event->area.x, event->area.y, | |||
| event->area.width, event->area.height); | |||
| return FALSE; | |||
| } | |||
| GtkWidget *fileselection; | |||
| static enum { SAVE, LOAD } fileselection_flag; | |||
| void show_fileselection() | |||
| { | |||
| fileselection = create_fileselection(); | |||
| gtk_signal_connect_object( | |||
| GTK_OBJECT(GTK_FILE_SELECTION(fileselection)->ok_button), | |||
| "clicked", GTK_SIGNAL_FUNC(gtk_widget_destroy), | |||
| (GtkObject *) fileselection); | |||
| gtk_signal_connect_object (GTK_OBJECT | |||
| (GTK_FILE_SELECTION(fileselection)->cancel_button), | |||
| "clicked", GTK_SIGNAL_FUNC(gtk_widget_destroy), | |||
| (GtkObject *) fileselection); | |||
| gtk_widget_show(fileselection); | |||
| } | |||
| void | |||
| on_button_save_clicked (GtkButton *button, | |||
| gpointer user_data) | |||
| { | |||
| fileselection_flag = SAVE; | |||
| show_fileselection(); | |||
| } | |||
| void | |||
| on_button_load_clicked (GtkButton *button, | |||
| gpointer user_data) | |||
| { | |||
| fileselection_flag = LOAD; | |||
| show_fileselection(); | |||
| } | |||
| 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"); | |||
| const char *p = gtk_entry_get_text(GTK_ENTRY(entry_option)); | |||
| const char* svm_type_str = strstr(p, "-s "); | |||
| int svm_type = C_SVC; | |||
| if(svm_type_str != NULL) | |||
| sscanf(svm_type_str, "-s %d", &svm_type); | |||
| if(fp) | |||
| { | |||
| if(svm_type == EPSILON_SVR || svm_type == NU_SVR) | |||
| { | |||
| for(list<point>::iterator p = point_list.begin(); p != point_list.end();p++) | |||
| fprintf(fp,"%f 1:%f\n", p->y, p->x); | |||
| } | |||
| else | |||
| { | |||
| 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) | |||
| { | |||
| point p = {x,y,v}; | |||
| point_list.push_back(p); | |||
| } | |||
| else if(sscanf(buf,"%lf%*d:%lf",&y,&x)==2) | |||
| { | |||
| point p = {x,y,current_value}; | |||
| point_list.push_back(p); | |||
| } | |||
| else | |||
| break; | |||
| } | |||
| fclose(fp); | |||
| draw_all_points(); | |||
| } | |||
| } | |||
| } | |||
| void | |||
| on_fileselection_destroy (GtkObject *object, | |||
| gpointer user_data) | |||
| { | |||
| } | |||
| void | |||
| on_filesel_cancel_clicked (GtkButton *button, | |||
| gpointer user_data) | |||
| { | |||
| } | |||