|
@@ -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) |
|
|
|
|
|
{ |
|
|
|
|
|
} |