{
 "cells": [
  {
   "cell_type": "markdown",
   "id": "24164d4f",
   "metadata": {},
   "source": [
    "# GlobalMinimization\n",
    "Example based  in\n",
    "http://cran.r-project.org/web/packages/DEoptim/DEoptim.pdf\n",
    "Please install the R package DEoptim before run this example.\n",
    "\n",
    "\n",
    "\n",
    "\n",
    "**Author:** Omar Zapata  \n",
    "<i><small>This notebook tutorial was automatically generated with <a href= \"https://github.com/root-project/root/blob/master/documentation/doxygen/converttonotebook.py\">ROOTBOOK-izer</a> from the macro found in the ROOT repository  on Sunday, December 21, 2025 at 02:49 PM.</small></i>"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "a4eb9832",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "%%cpp -d\n",
    "#include<TRInterface.h>\n",
    "#include<TBenchmark.h>\n",
    "#include<cmath>\n",
    "#include<cstdlib>"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "e7d15536",
   "metadata": {},
   "source": [
    " In the next function the *double pointer should be changed by a TVectorD datatype,\n",
    "because the pointer has no meaning in R's enviroment.\n",
    "This is a generalization of the RosenBrock function, with the min xi=1 and i>0.\n",
    " "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "b977e427",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "%%cpp -d\n",
    "Double_t GenRosenBrock(const TVectorD xx )\n",
    "{\n",
    "   int length=xx.GetNoElements();\n",
    "\n",
    "   Double_t result=0;\n",
    "   for (int i=0;i<(length-1);i++) {\n",
    "      result+=pow(1-xx[i],2)+100*pow(xx[i+1]-pow(xx[i],2),2);\n",
    "   }\n",
    "   return result;\n",
    "}"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "de63b091",
   "metadata": {},
   "source": [
    " the min xi=0 i>0\n",
    " "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "c2c4b03e",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "%%cpp -d\n",
    "Double_t Rastrigin(const TVectorD xx)\n",
    "{\n",
    "   int length=xx.GetNoElements();\n",
    "   Double_t result=10*length;\n",
    "   for (int i=0;i<length;i++) {\n",
    "      result+=xx[i]*xx[i]-10*cos(6.2831853*xx[i]);\n",
    "   }\n",
    "   return result;\n",
    "}"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "85323bc2",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "TBenchmark bench;\n",
    "ROOT::R::TRInterface &r=ROOT::R::TRInterface::Instance();\n",
    "\n",
    "Bool_t installed=r.Eval(\"is.element('DEoptim', installed.packages()[,1])\");\n",
    "if (!installed) {\n",
    "   std::cout<<\"Package DEoptim no installed in R\"<<std::endl;\n",
    "   std::cout<<\"Run install.packages('DEoptim') in R's environment\"<<std::endl;\n",
    "   return;\n",
    "}"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "ddab2bbc",
   "metadata": {},
   "source": [
    "loading DEoptim"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "700a9af9",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "r<<\"suppressMessages(library(DEoptim, quietly = TRUE))\";"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "162472c9",
   "metadata": {},
   "source": [
    "passing RosenBrock function to R"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "e8d0f7b6",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "r[\"GenRosenBrock\"]<<ROOT::R::TRFunctionExport(GenRosenBrock);"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "a23d4bd2",
   "metadata": {},
   "source": [
    "maximun number of iterations"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "05610103",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "r[\"MaxIter\"]<<5000;"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "f2a50262",
   "metadata": {},
   "source": [
    "n = size of vector that is an argument for GenRosenBrock"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "62d35e97",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "r[\"n\"]<<3;"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "f82a4953",
   "metadata": {},
   "source": [
    "lower limits"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "e103fe05",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "r<<\"ll<-rep(-25, n)\";"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "824e64fa",
   "metadata": {},
   "source": [
    "upper limits"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "47f35dea",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "r<<\"ul<-rep(25, n)\";\n",
    "\n",
    "bench.Start(\"GlobalMinimizationRosenBrock\");"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "0a0c11df",
   "metadata": {},
   "source": [
    "calling minimization and timing it."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "66fb7cb0",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "r<<\"result1<-DEoptim(fn=GenRosenBrock,lower=ll,upper=ul,control=list(NP=10*n,itermax=MaxIter,trace=FALSE))\";\n",
    "std::cout<<\"-----------------------------------------\"<<std::endl;\n",
    "std::cout<<\"RosenBrock's minimum in: \"<<std::endl;\n",
    "r<<\"print(result1$optim$bestmem)\";\n",
    "std::cout<<\"Bechmark Times\"<<std::endl;"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "97c5d94b",
   "metadata": {},
   "source": [
    "printing times"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "8e377817",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "bench.Show(\"GlobalMinimizationRosenBrock\");"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "01c818c8",
   "metadata": {},
   "source": [
    "passing RosenBrock function to R"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "a0e9aaaa",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "r[\"Rastrigin\"]<<ROOT::R::TRFunctionExport(Rastrigin);"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "408b8305",
   "metadata": {},
   "source": [
    "maximun number of iterations"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "3f8c5fb2",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "r[\"MaxIter\"]<<2000;"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "b9a78edf",
   "metadata": {},
   "source": [
    "n = size of a vector which is an argument for Rastrigin"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "6a491504",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "r[\"n\"]<<3;"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "48caec09",
   "metadata": {},
   "source": [
    "lower limits"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "1b17f488",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "r<<\"ll<-rep(-5, n)\";"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "de126c3f",
   "metadata": {},
   "source": [
    "upper limits"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "a8759288",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "r<<\"ul<-rep(5, n)\";\n",
    "\n",
    "bench.Start(\"GlobalMinimizationRastrigin\");"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "ae2ca54c",
   "metadata": {},
   "source": [
    "calling minimization and timing it."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "9c1ef3cb",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "r<<\"result2<-DEoptim(fn=Rastrigin,lower=ll,upper=ul,control=list(NP=10*n,itermax=MaxIter,trace=FALSE))\";\n",
    "std::cout<<\"-----------------------------------------\"<<std::endl;\n",
    "std::cout<<\"Rastrigin's minimum in: \"<<std::endl;\n",
    "r<<\"print(result2$optim$bestmem)\";\n",
    "std::cout<<\"Bechmark Times\"<<std::endl;"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "6c720b33",
   "metadata": {},
   "source": [
    "printing times"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "84c9282b",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "bench.Show(\"GlobalMinimizationRastrigin\");"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "67f5b356",
   "metadata": {},
   "source": [
    "skip R plotting in batch mode"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "3bfd2c66",
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "if (!gROOT->IsBatch()) {\n",
    "   r<<\"dev.new(title='RosenBrock Convergence')\";\n",
    "   r<<\"plot(result1,type='o',pch='.')\";\n",
    "   r<<\"dev.off()\";\n",
    "   r<<\"dev.new(title='Rastrigin Convergence')\";\n",
    "   r<<\"plot(result2,type='o',pch='.')\";\n",
    "   r<<\"dev.off()\";\n",
    "}"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "ROOT C++",
   "language": "c++",
   "name": "root"
  },
  "language_info": {
   "codemirror_mode": "text/x-c++src",
   "file_extension": ".C",
   "mimetype": " text/x-c++src",
   "name": "c++"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}