| @@ -0,0 +1,437 @@ | |||||
| #include <QtGui> | |||||
| #include <stdio.h> | |||||
| #include <stdlib.h> | |||||
| #include <string.h> | |||||
| #include <ctype.h> | |||||
| #include <list> | |||||
| #include "../../svm.h" | |||||
| using namespace std; | |||||
| #define DEFAULT_PARAM "-t 2 -c 100" | |||||
| #define XLEN 500 | |||||
| #define YLEN 500 | |||||
| QRgb colors[] = | |||||
| { | |||||
| qRgb(0,0,0), | |||||
| qRgb(0,120,120), | |||||
| qRgb(120,120,0), | |||||
| qRgb(120,0,120), | |||||
| qRgb(0,200,200), | |||||
| qRgb(200,200,0), | |||||
| qRgb(200,0,200) | |||||
| }; | |||||
| class SvmToyWindow : public QWidget | |||||
| { | |||||
| Q_OBJECT | |||||
| public: | |||||
| SvmToyWindow(); | |||||
| ~SvmToyWindow(); | |||||
| protected: | |||||
| virtual void mousePressEvent( QMouseEvent* ); | |||||
| virtual void paintEvent( QPaintEvent* ); | |||||
| private: | |||||
| QPixmap buffer; | |||||
| QPixmap icon1; | |||||
| QPixmap icon2; | |||||
| QPixmap icon3; | |||||
| QPushButton button_change_icon; | |||||
| QPushButton button_run; | |||||
| QPushButton button_clear; | |||||
| QPushButton button_save; | |||||
| QPushButton button_load; | |||||
| QLineEdit input_line; | |||||
| QPainter buffer_painter; | |||||
| struct point { | |||||
| double x, y; | |||||
| signed char value; | |||||
| }; | |||||
| list<point> point_list; | |||||
| int current_value; | |||||
| const QPixmap& choose_icon(int v) | |||||
| { | |||||
| if(v==1) return icon1; | |||||
| else if(v==2) return icon2; | |||||
| else return icon3; | |||||
| } | |||||
| void clear_all() | |||||
| { | |||||
| point_list.clear(); | |||||
| buffer.fill(Qt::black); | |||||
| repaint(); | |||||
| } | |||||
| void draw_point(const point& p) | |||||
| { | |||||
| const QPixmap& icon = choose_icon(p.value); | |||||
| buffer_painter.drawPixmap((int)(p.x*XLEN),(int)(p.y*YLEN),icon); | |||||
| repaint(); | |||||
| } | |||||
| void draw_all_points() | |||||
| { | |||||
| for(list<point>::iterator p = point_list.begin(); p != point_list.end();p++) | |||||
| draw_point(*p); | |||||
| } | |||||
| private slots: | |||||
| void button_change_icon_clicked() | |||||
| { | |||||
| ++current_value; | |||||
| if(current_value > 3) current_value = 1; | |||||
| button_change_icon.setIcon(choose_icon(current_value)); | |||||
| } | |||||
| void button_run_clicked() | |||||
| { | |||||
| // 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 = input_line.text().toAscii().constData(); | |||||
| 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)); | |||||
| } | |||||
| buffer_painter.setPen(colors[0]); | |||||
| buffer_painter.drawLine(0,0,0,YLEN-1); | |||||
| int p = (int)(param.p * YLEN); | |||||
| for(i = 1; i < XLEN; i++) | |||||
| { | |||||
| buffer_painter.setPen(colors[0]); | |||||
| buffer_painter.drawLine(i,0,i,YLEN-1); | |||||
| buffer_painter.setPen(colors[5]); | |||||
| buffer_painter.drawLine(i-1,j[i-1],i,j[i]); | |||||
| if(param.svm_type == EPSILON_SVR) | |||||
| { | |||||
| buffer_painter.setPen(colors[2]); | |||||
| buffer_painter.drawLine(i-1,j[i-1]+p,i,j[i]+p); | |||||
| buffer_painter.setPen(colors[2]); | |||||
| buffer_painter.drawLine(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; | |||||
| buffer_painter.setPen(colors[(int)d]); | |||||
| buffer_painter.drawPoint(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 button_clear_clicked() | |||||
| { | |||||
| clear_all(); | |||||
| } | |||||
| void button_save_clicked() | |||||
| { | |||||
| QString filename = QFileDialog::getSaveFileName(); | |||||
| if(!filename.isNull()) | |||||
| { | |||||
| FILE *fp = fopen(filename.toAscii().constData(),"w"); | |||||
| const char *p = input_line.text().toAscii().constData(); | |||||
| 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); | |||||
| } | |||||
| } | |||||
| } | |||||
| void button_load_clicked() | |||||
| { | |||||
| QString filename = QFileDialog::getOpenFileName(); | |||||
| if(!filename.isNull()) | |||||
| { | |||||
| FILE *fp = fopen(filename.toAscii().constData(),"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(); | |||||
| } | |||||
| } | |||||
| } | |||||
| }; | |||||
| #include "svm-toy.moc" | |||||
| SvmToyWindow::SvmToyWindow() | |||||
| :button_change_icon(this) | |||||
| ,button_run("Run",this) | |||||
| ,button_clear("Clear",this) | |||||
| ,button_save("Save",this) | |||||
| ,button_load("Load",this) | |||||
| ,input_line(this) | |||||
| ,current_value(1) | |||||
| { | |||||
| buffer = QPixmap(XLEN,YLEN); | |||||
| buffer.fill(Qt::black); | |||||
| buffer_painter.begin(&buffer); | |||||
| QObject::connect(&button_change_icon, SIGNAL(clicked()), this, | |||||
| SLOT(button_change_icon_clicked())); | |||||
| QObject::connect(&button_run, SIGNAL(clicked()), this, | |||||
| SLOT(button_run_clicked())); | |||||
| QObject::connect(&button_clear, SIGNAL(clicked()), this, | |||||
| SLOT(button_clear_clicked())); | |||||
| QObject::connect(&button_save, SIGNAL(clicked()), this, | |||||
| SLOT(button_save_clicked())); | |||||
| QObject::connect(&button_load, SIGNAL(clicked()), this, | |||||
| SLOT(button_load_clicked())); | |||||
| QObject::connect(&input_line, SIGNAL(returnPressed()), this, | |||||
| SLOT(button_run_clicked())); | |||||
| // don't blank the window before repainting | |||||
| setAttribute(Qt::WA_NoBackground); | |||||
| icon1 = QPixmap(4,4); | |||||
| icon2 = QPixmap(4,4); | |||||
| icon3 = QPixmap(4,4); | |||||
| QPainter painter; | |||||
| painter.begin(&icon1); | |||||
| painter.fillRect(0,0,4,4,QBrush(colors[4])); | |||||
| painter.end(); | |||||
| painter.begin(&icon2); | |||||
| painter.fillRect(0,0,4,4,QBrush(colors[5])); | |||||
| painter.end(); | |||||
| painter.begin(&icon3); | |||||
| painter.fillRect(0,0,4,4,QBrush(colors[6])); | |||||
| painter.end(); | |||||
| button_change_icon.setGeometry( 0, YLEN, 50, 25 ); | |||||
| button_run.setGeometry( 50, YLEN, 50, 25 ); | |||||
| button_clear.setGeometry( 100, YLEN, 50, 25 ); | |||||
| button_save.setGeometry( 150, YLEN, 50, 25); | |||||
| button_load.setGeometry( 200, YLEN, 50, 25); | |||||
| input_line.setGeometry( 250, YLEN, 250, 25); | |||||
| input_line.setText(DEFAULT_PARAM); | |||||
| button_change_icon.setIcon(icon1); | |||||
| } | |||||
| SvmToyWindow::~SvmToyWindow() | |||||
| { | |||||
| buffer_painter.end(); | |||||
| } | |||||
| void SvmToyWindow::mousePressEvent( QMouseEvent* event ) | |||||
| { | |||||
| point p = {(double)event->x()/XLEN, (double)event->y()/YLEN, current_value}; | |||||
| point_list.push_back(p); | |||||
| draw_point(p); | |||||
| } | |||||
| void SvmToyWindow::paintEvent( QPaintEvent* ) | |||||
| { | |||||
| // copy the image from the buffer pixmap to the window | |||||
| QPainter p(this); | |||||
| p.drawPixmap(0, 0, buffer); | |||||
| } | |||||
| int main( int argc, char* argv[] ) | |||||
| { | |||||
| QApplication myapp( argc, argv ); | |||||
| SvmToyWindow* mywidget = new SvmToyWindow(); | |||||
| mywidget->setGeometry( 100, 100, XLEN, YLEN+25 ); | |||||
| mywidget->show(); | |||||
| return myapp.exec(); | |||||
| } | |||||