examples/example5/Example3_generate_C++_code.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "61586d08-43b0-4d21-895a-c35dc2aa4526",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      " Computational graph of function No.1\n",
      "\n",
      " index | op  | parents\n",
      " ---------------------\n",
      "     1 | inp |         \n",
      "     2 | inp |         \n",
      "     3 | inp |         \n",
      "     4 | inp |         \n",
      "     5 | inp |         \n",
      "     6 |  p  |         \n",
      "     7 |  p  |         \n",
      "     8 |  p  |         \n",
      "     9 |  p  |         \n",
      "    10 |  p  |         \n",
      "    11 |  p  |         \n",
      "    12 |  p  |         \n",
      "    13 |  p  |         \n",
      "    14 | con |         \n",
      "    15 |  /  | 14   3  \n",
      "    16 | con |         \n",
      "    17 |  *  | 16   15 \n",
      "    18 | con |         \n",
      "    19 |  +  | 18   17 \n",
      "    20 | con |         \n",
      "    21 |  /  | 20   19 \n",
      "    22 | con |         \n",
      "    23 |  *  | 22   21 \n",
      "    24 | con |         \n",
      "    25 |  /  | 24   4  \n",
      "    26 |  *  | 12   25 \n",
      "    27 | con |         \n",
      "    28 |  *  | 4    27 \n",
      "    29 | con |         \n",
      "    30 |  +  | 29   26 \n",
      "    31 |  +  | 30   28 \n",
      "    32 | con |         \n",
      "    33 |  /  | 32   31 \n",
      "    34 | con |         \n",
      "    35 |  *  | 34   33 \n",
      "    36 |  +  | 5    4  \n",
      "    37 | con |         \n",
      "    38 |  -  | 36   37 \n",
      "    39 | con |         \n",
      "    40 |  *  | 11   39 \n",
      "    41 |  *  | 40   35 \n",
      "    42 |  *  | 41   2  \n",
      "    43 |  +  | 38   42 \n",
      "    44 |  +  | 5    4  \n",
      "    45 | con |         \n",
      "    46 |  -  | 44   45 \n",
      "    47 | con |         \n",
      "    48 |  *  | 47   43 \n",
      "    49 |  -  | 46   48 \n",
      "    50 | con |         \n",
      "    51 |  *  | 50   49 \n",
      "    52 | con |         \n",
      "    53 |  -  | 23   52 \n",
      "    54 |  *  | 53   1  \n",
      "    55 | out | 54      \n",
      "\n",
      "\n",
      " Computational graph of function No.2\n",
      "\n",
      " index | op  | parents\n",
      " ---------------------\n",
      "     1 | inp |         \n",
      "     2 | inp |         \n",
      "     3 | inp |         \n",
      "     4 | inp |         \n",
      "     5 | inp |         \n",
      "     6 |  p  |         \n",
      "     7 |  p  |         \n",
      "     8 |  p  |         \n",
      "     9 |  p  |         \n",
      "    10 |  p  |         \n",
      "    11 |  p  |         \n",
      "    12 |  p  |         \n",
      "    13 |  p  |         \n",
      "    14 | con |         \n",
      "    15 |  /  | 14   3  \n",
      "    16 | con |         \n",
      "    17 |  *  | 16   15 \n",
      "    18 | con |         \n",
      "    19 |  +  | 18   17 \n",
      "    20 | con |         \n",
      "    21 |  /  | 20   19 \n",
      "    22 | con |         \n",
      "    23 |  *  | 22   21 \n",
      "    24 | con |         \n",
      "    25 |  /  | 24   4  \n",
      "    26 |  *  | 12   25 \n",
      "    27 | con |         \n",
      "    28 |  *  | 4    27 \n",
      "    29 | con |         \n",
      "    30 |  +  | 29   26 \n",
      "    31 |  +  | 30   28 \n",
      "    32 | con |         \n",
      "    33 |  /  | 32   31 \n",
      "    34 | con |         \n",
      "    35 |  *  | 34   33 \n",
      "    36 |  +  | 5    4  \n",
      "    37 | con |         \n",
      "    38 |  -  | 36   37 \n",
      "    39 | con |         \n",
      "    40 |  *  | 11   39 \n",
      "    41 |  *  | 40   35 \n",
      "    42 |  *  | 41   2  \n",
      "    43 |  +  | 38   42 \n",
      "    44 |  +  | 5    4  \n",
      "    45 | con |         \n",
      "    46 |  -  | 44   45 \n",
      "    47 | con |         \n",
      "    48 |  *  | 47   43 \n",
      "    49 |  -  | 46   48 \n",
      "    50 | con |         \n",
      "    51 |  *  | 50   49 \n",
      "    52 | con |         \n",
      "    53 |  -  | 35   52 \n",
      "    54 |  *  | 53   2  \n",
      "    55 | out | 54      \n",
      "\n",
      "\n",
      " Computational graph of function No.3\n",
      "\n",
      " index | op  | parents\n",
      " ---------------------\n",
      "     1 | inp |         \n",
      "     2 | inp |         \n",
      "     3 | inp |         \n",
      "     4 | inp |         \n",
      "     5 | inp |         \n",
      "     6 |  p  |         \n",
      "     7 |  p  |         \n",
      "     8 |  p  |         \n",
      "     9 |  p  |         \n",
      "    10 |  p  |         \n",
      "    11 |  p  |         \n",
      "    12 |  p  |         \n",
      "    13 |  p  |         \n",
      "    14 | con |         \n",
      "    15 |  /  | 14   3  \n",
      "    16 | con |         \n",
      "    17 |  *  | 16   15 \n",
      "    18 | con |         \n",
      "    19 |  +  | 18   17 \n",
      "    20 | con |         \n",
      "    21 |  /  | 20   19 \n",
      "    22 | con |         \n",
      "    23 |  *  | 22   21 \n",
      "    24 | con |         \n",
      "    25 |  /  | 24   4  \n",
      "    26 |  *  | 12   25 \n",
      "    27 | con |         \n",
      "    28 |  *  | 4    27 \n",
      "    29 | con |         \n",
      "    30 |  +  | 29   26 \n",
      "    31 |  +  | 30   28 \n",
      "    32 | con |         \n",
      "    33 |  /  | 32   31 \n",
      "    34 | con |         \n",
      "    35 |  *  | 34   33 \n",
      "    36 |  +  | 5    4  \n",
      "    37 | con |         \n",
      "    38 |  -  | 36   37 \n",
      "    39 | con |         \n",
      "    40 |  *  | 11   39 \n",
      "    41 |  *  | 40   35 \n",
      "    42 |  *  | 41   2  \n",
      "    43 |  +  | 38   42 \n",
      "    44 |  +  | 5    4  \n",
      "    45 | con |         \n",
      "    46 |  -  | 44   45 \n",
      "    47 | con |         \n",
      "    48 |  *  | 47   43 \n",
      "    49 |  -  | 46   48 \n",
      "    50 | con |         \n",
      "    51 |  *  | 50   49 \n",
      "    52 | con |         \n",
      "    53 |  -  | 52   3  \n",
      "    54 | con |         \n",
      "    55 |  *  | 54   53 \n",
      "    56 |  *  | 6    23 \n",
      "    57 |  *  | 56   1  \n",
      "    58 |  -  | 55   57 \n",
      "    59 | out | 58      \n",
      "\n",
      "\n",
      " Computational graph of function No.4\n",
      "\n",
      " index | op  | parents\n",
      " ---------------------\n",
      "     1 | inp |         \n",
      "     2 | inp |         \n",
      "     3 | inp |         \n",
      "     4 | inp |         \n",
      "     5 | inp |         \n",
      "     6 |  p  |         \n",
      "     7 |  p  |         \n",
      "     8 |  p  |         \n",
      "     9 |  p  |         \n",
      "    10 |  p  |         \n",
      "    11 |  p  |         \n",
      "    12 |  p  |         \n",
      "    13 |  p  |         \n",
      "    14 | con |         \n",
      "    15 |  /  | 14   3  \n",
      "    16 | con |         \n",
      "    17 |  *  | 16   15 \n",
      "    18 | con |         \n",
      "    19 |  +  | 18   17 \n",
      "    20 | con |         \n",
      "    21 |  /  | 20   19 \n",
      "    22 | con |         \n",
      "    23 |  *  | 22   21 \n",
      "    24 | con |         \n",
      "    25 |  /  | 24   4  \n",
      "    26 |  *  | 12   25 \n",
      "    27 | con |         \n",
      "    28 |  *  | 4    27 \n",
      "    29 | con |         \n",
      "    30 |  +  | 29   26 \n",
      "    31 |  +  | 30   28 \n",
      "    32 | con |         \n",
      "    33 |  /  | 32   31 \n",
      "    34 | con |         \n",
      "    35 |  *  | 34   33 \n",
      "    36 |  +  | 5    4  \n",
      "    37 | con |         \n",
      "    38 |  -  | 36   37 \n",
      "    39 | con |         \n",
      "    40 |  *  | 11   39 \n",
      "    41 |  *  | 40   35 \n",
      "    42 |  *  | 41   2  \n",
      "    43 |  +  | 38   42 \n",
      "    44 |  +  | 5    4  \n",
      "    45 | con |         \n",
      "    46 |  -  | 44   45 \n",
      "    47 | con |         \n",
      "    48 |  *  | 47   43 \n",
      "    49 |  -  | 46   48 \n",
      "    50 | con |         \n",
      "    51 |  *  | 50   49 \n",
      "    52 | con |         \n",
      "    53 |  -  | 52   4  \n",
      "    54 | con |         \n",
      "    55 |  *  | 54   53 \n",
      "    56 |  *  | 7    23 \n",
      "    57 |  *  | 56   1  \n",
      "    58 |  +  | 55   57 \n",
      "    59 |  *  | 8    35 \n",
      "    60 |  *  | 59   2  \n",
      "    61 |  -  | 58   60 \n",
      "    62 | out | 61      \n",
      "\n",
      "\n",
      " Computational graph of function No.5\n",
      "\n",
      " index | op  | parents\n",
      " ---------------------\n",
      "     1 | inp |         \n",
      "     2 | inp |         \n",
      "     3 | inp |         \n",
      "     4 | inp |         \n",
      "     5 | inp |         \n",
      "     6 |  p  |         \n",
      "     7 |  p  |         \n",
      "     8 |  p  |         \n",
      "     9 |  p  |         \n",
      "    10 |  p  |         \n",
      "    11 |  p  |         \n",
      "    12 |  p  |         \n",
      "    13 |  p  |         \n",
      "    14 | con |         \n",
      "    15 |  /  | 14   3  \n",
      "    16 | con |         \n",
      "    17 |  *  | 16   15 \n",
      "    18 | con |         \n",
      "    19 |  +  | 18   17 \n",
      "    20 | con |         \n",
      "    21 |  /  | 20   19 \n",
      "    22 | con |         \n",
      "    23 |  *  | 22   21 \n",
      "    24 | con |         \n",
      "    25 |  /  | 24   4  \n",
      "    26 |  *  | 12   25 \n",
      "    27 | con |         \n",
      "    28 |  *  | 4    27 \n",
      "    29 | con |         \n",
      "    30 |  +  | 29   26 \n",
      "    31 |  +  | 30   28 \n",
      "    32 | con |         \n",
      "    33 |  /  | 32   31 \n",
      "    34 | con |         \n",
      "    35 |  *  | 34   33 \n",
      "    36 |  +  | 5    4  \n",
      "    37 | con |         \n",
      "    38 |  -  | 36   37 \n",
      "    39 | con |         \n",
      "    40 |  *  | 11   39 \n",
      "    41 |  *  | 40   35 \n",
      "    42 |  *  | 41   2  \n",
      "    43 |  +  | 38   42 \n",
      "    44 |  +  | 5    4  \n",
      "    45 | con |         \n",
      "    46 |  -  | 44   45 \n",
      "    47 | con |         \n",
      "    48 |  *  | 47   43 \n",
      "    49 |  -  | 46   48 \n",
      "    50 | con |         \n",
      "    51 |  *  | 50   49 \n",
      "    52 | con |         \n",
      "    53 |  *  | 52   5  \n",
      "    54 |  -  | 53   51 \n",
      "    55 |  *  | 9    23 \n",
      "    56 |  *  | 55   1  \n",
      "    57 |  *  | 10   35 \n",
      "    58 |  *  | 57   2  \n",
      "    59 |  +  | 54   56 \n",
      "    60 |  +  | 59   58 \n",
      "    61 | out | 60      \n",
      "\n",
      "\n",
      "\n",
      "\n",
      "C++ code for Adjoint mode computation of subgradients of McCormick relaxations is generated for Example 3 as reverseMC.cpp with its hearder file reverseMC.hpp\n"
     ]
    }
   ],
   "source": [
    "include(\"../src/ReverseADforVW.jl\")\n",
    "\n",
    "using .ReverseADforVW\n",
    "\n",
    "# construct AD tape for function\n",
    "nx = 5;\n",
    "np = 8;\n",
    "nf = 5;\n",
    "\n",
    "# function has to be constructed as f(x,p,i) instead of f(p,x,i) or f(i,x,p), which will lead to error\n",
    "function f(x,p,i)\n",
    "    \n",
    "    mu1 = 1.2*(1.0/(1.0+7.1*(1.0/x[3])));\n",
    "    mu2 = 0.74*(1.0/(1.0+p[7]*(1.0/x[4])+x[4]*(1.0/256.0)));\n",
    "    phi = x[5]+x[4]-34.0+p[6]*(1.0/19.8)*mu2*x[2];\n",
    "    q = 19.8*(x[5]+x[4]-50.0-0.5*phi);\n",
    "    \n",
    "    if i == 1\n",
    "        f = (mu1-0.2)*x[1];\n",
    "    elseif i == 2\n",
    "        f = (mu2-0.2)*x[2];\n",
    "    elseif i == 3\n",
    "        f = 0.4*(5.0-x[3])- p[1]*mu1*x[1];\n",
    "    elseif i == 4\n",
    "        f = 0.4*(80.0-x[4])+p[2]*mu1*x[1]-p[3]*mu2*x[2];\n",
    "    elseif i == 5\n",
    "        f = -0.4*x[5]-q+p[4]*mu1*x[1]+p[5]*mu2*x[2];\n",
    "    end\n",
    "end\n",
    "\n",
    "tape = record_tape(f,nx,np,1,nf) \n",
    "\n",
    "println(tape)\n",
    "println(\"\\nC++ code for Adjoint mode computation of subgradients of McCormick relaxations is generated for Example 3 as reverseMC.cpp with its hearder file reverseMC.hpp\")\n",
    "generate_revMC_c_code!(tape)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "4a5a1fcc-4e6c-4be5-a1a6-bf2605b5feeb",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Julia 1.8.5",
   "language": "julia",
   "name": "julia-1.8"
  },
  "language_info": {
   "file_extension": ".jl",
   "mimetype": "application/julia",
   "name": "julia",
   "version": "1.8.5"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}