osqp_demo.c 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758
  1. #include "osqp.h"
  2. int main(int argc, char **argv) {
  3. // Load problem data
  4. c_float P_x[3] = { 4.0, 1.0, 2.0, };
  5. c_int P_nnz = 3;
  6. c_int P_i[3] = { 0, 0, 1, };
  7. c_int P_p[3] = { 0, 1, 3, };
  8. c_float q[2] = { 1.0, 1.0, };
  9. c_float A_x[4] = { 1.0, 1.0, 1.0, 1.0, };
  10. c_int A_nnz = 4;
  11. c_int A_i[4] = { 0, 1, 0, 2, };
  12. c_int A_p[3] = { 0, 2, 4, };
  13. c_float l[3] = { 1.0, 0.0, 0.0, };
  14. c_float u[3] = { 1.0, 0.7, 0.7, };
  15. c_int n = 2;
  16. c_int m = 3;
  17. // Exitflag
  18. c_int exitflag = 0;
  19. // Workspace structures
  20. OSQPWorkspace *work;
  21. OSQPSettings *settings = (OSQPSettings *)c_malloc(sizeof(OSQPSettings));
  22. OSQPData *data = (OSQPData *)c_malloc(sizeof(OSQPData));
  23. // Populate data
  24. if (data) {
  25. data->n = n;
  26. data->m = m;
  27. data->P = csc_matrix(data->n, data->n, P_nnz, P_x, P_i, P_p);
  28. data->q = q;
  29. data->A = csc_matrix(data->m, data->n, A_nnz, A_x, A_i, A_p);
  30. data->l = l;
  31. data->u = u;
  32. }
  33. // Define solver settings as default
  34. if (settings) osqp_set_default_settings(settings);
  35. // Setup workspace
  36. exitflag = osqp_setup(&work, data, settings);
  37. // Solve Problem
  38. osqp_solve(work);
  39. // Clean workspace
  40. osqp_cleanup(work);
  41. if (data) {
  42. if (data->A) c_free(data->A);
  43. if (data->P) c_free(data->P);
  44. c_free(data);
  45. }
  46. if (settings) c_free(settings);
  47. return exitflag;
  48. }