Logo ROOT  
Reference Guide
pyroot004_NumbaDeclare.py File Reference

Namespaces

namespace  pyroot004_NumbaDeclare
 

Detailed Description

View in nbviewer Open in SWAN
This tutorial illustrates how PyROOT supports declaring C++ callables from Python callables making them, for example, usable with RDataFrame.

The feature uses the numba Python package for just-in-time compilation of the Python callable and supports fundamental types and ROOT::RVec thereof.

import ROOT
# To mark a Python callable to be used from C++, you have to use the decorator
# provided by PyROOT passing the C++ types of the input arguments and the return
# value.
@ROOT.Numba.Declare(['float', 'int'], 'float')
def pypow(x, y):
return x**y
# The Python callable is now available from C++ in the Numba namespace.
# For example, we can use it from the interpreter.
ROOT.gInterpreter.ProcessLine('cout << "2^3 = " << Numba::pypow(2, 3) << endl;')
# Or we can use the callable as well within a RDataFrame workflow.
data = ROOT.RDataFrame(4).Define('x', '(float)rdfentry_')\
.Define('x_pow3', 'Numba::pypow(x, 3)')\
.AsNumpy()
print('pypow({}, 3) = {}'.format(data['x'], data['x_pow3']))
# ROOT uses the numba Python package to create C++ functions from python ones.
# We support as input and return types of the callable fundamental types and
# ROOT::RVec thereof. See the following callable computing the power of the
# elements in an array.
@ROOT.Numba.Declare(['RVecF', 'int'], 'RVecF')
def pypowarray(x, y):
return x**y
ROOT.gInterpreter.ProcessLine('''
ROOT::RVecF x = {0, 1, 2, 3};
cout << "pypowarray(" << x << ", 3) = " << Numba::pypowarray(x, 3) << endl;
''')
# and now with RDataFrame
s = ROOT.RDataFrame(1).Define('x', 'ROOT::RVecF{1,2,3}')\
.Define('x2', 'Numba::pypowarray(x, 2)')\
.Sum('x2') # 1 + 4 + 9 == 14
print('sum(pypowarray({ 1, 2, 3 }, 2)) = ', s.GetValue())
Option_t Option_t TPoint TPoint const char GetTextMagnitude GetFillStyle GetLineColor GetLineWidth GetMarkerStyle GetTextAlign GetTextColor GetTextSize void char Point_t Rectangle_t WindowAttributes_t Float_t Float_t Float_t Int_t Int_t UInt_t UInt_t Rectangle_t Int_t Int_t Window_t TString Int_t GCValues_t GetPrimarySelectionOwner GetDisplay GetScreen GetColormap GetNativeEvent const char const char dpyName wid window const char font_name cursor keysym reg const char only_if_exist regb h Point_t winding char text const char depth char const char Int_t count const char ColorStruct_t color const char Pixmap_t Pixmap_t PictureAttributes_t attr const char char ret_data h unsigned char height h Atom_t Int_t ULong_t ULong_t unsigned char prop_list Atom_t Atom_t Atom_t Time_t format
ROOT's RDataFrame offers a modern, high-level interface for analysis of data stored in TTree ,...
Definition: RDataFrame.hxx:40
T Sum(const RVec< T > &v, const T zero=T(0))
Sum elements of an RVec.
Definition: RVec.hxx:1900
2^3 = 8
pypowarray({ 0, 1, 2, 3 }, 3) = { 0, 1, 8, 27 }
pypow([0. 1. 2. 3.], 3) = [ 0. 1. 8. 27.]
sum(pypowarray({ 1, 2, 3 }, 2)) = 14.0
Date
March 2020
Author
Stefan Wunsch

Definition in file pyroot004_NumbaDeclare.py.