Create a gist now

Instantly share code, notes, and snippets.

Run code cells in an iPython notebook until a breakpoint is encountered Updated version with div.breakpoint element to display breakpoint indicator.
//----------------------------------------------------------------------------
// Copyright (C) 2013 The IPython Development Team
//
// Distributed under the terms of the BSD License. The full license is in
// the file COPYING, distributed as part of this software.
//----------------------------------------------------------------------------
//============================================================================
// Breakpoint extension - execute notebook cells until breakpoint
// If a breakpoint is set at the currently selected cell, run cell anyway,
// allowing to step through the notebook
//============================================================================
/**
* Toggle breakpoint marker on/off
*
* @param {Object} current codecell
* @param {Boolean} turn breakpoint on/off
*/
function makeMarker(cell,val) {
var input = cell.element.find('div.breakpoint');
cell.metadata.run_control.breakpoint = val ;
if (val == true) {
input.html('');
console.log('true');
} else {
input.html('');
console.log('false');
}
}
/**
* Clear all breakpoints in notebook
*
*/
var clear_breakpoints = function() {
var cells = IPython.notebook.get_cells();
for(var i in cells){
var cell = cells[i];
if ((cell instanceof IPython.CodeCell)) {
if (cell.metadata.run_control == undefined){cell.metadata.run_control = {}} {
makeMarker(cell,false);
}
}
}
};
/**
* Run code cells from current cell to next breakpoint
*
*/
var run_breakpoint = function () {
var start = IPython.notebook.get_selected_index();
var end = IPython.notebook.ncells()
for (var i=start; i<end; i++) {
IPython.notebook.select(i);
var cell = IPython.notebook.get_selected_cell();
if ((cell instanceof IPython.CodeCell)) {
if (cell.metadata.run_control != undefined) {
if (cell.metadata.run_control.breakpoint == true && i > start) {
break;
}
}
IPython.notebook.execute_selected_cell({add_new:false});
}
}
};
/**
* Loop through all code cells to find which input prompt was clicked
* and turn breakpoint on/off
*
* @param {Object} div.input_prompt object where the mouse was clicked at
*/
function clickPrompt(prompt_clicked) {
var cells = IPython.notebook.get_cells();
for(var i in cells){
var cell = cells[i];
if ((cell instanceof IPython.CodeCell)) {
var prompt = cell.element.find('div.input_prompt');
if (prompt_clicked.is(prompt)) {
if (cell.metadata.run_control == undefined){cell.metadata.run_control = {}}
makeMarker(cell,!cell.metadata.run_control.breakpoint);
}
}
}
};
/**
* Add run control buttons to toolbar and mark cells with breakpoint enabled,
* and add callback function to set/remove breakpoints
*
*/
var init_breakpoint = function(){
IPython.toolbar.add_buttons_group([
{
id : 'run_c',
label : 'Run Cell',
icon : 'ui-icon-seek-next',
callback : function () {
IPython.notebook.execute_selected_cell();
}
},
{
id : 'run_cb',
label : 'Run Cells Below',
icon : 'ui-icon-seek-end',
callback : function () {
IPython.notebook.execute_cells_below();
}
},
{
id : 'run_a',
label : 'Run All',
icon : 'ui-icon-play',
callback : function () {
IPython.notebook.execute_all_cells();
}
},
{
id : 'interrupt_b',
label : 'Interrupt',
icon : 'ui-icon-stop',
callback : function () {
IPython.notebook.kernel.interrupt();
}
},
{
id : 'run_until_break',
label : 'Run Until Breakpoint',
icon : 'ui-icon-bullet',
callback : run_breakpoint
},
{
id : 'clear_all_breakpoints',
label : 'Clear all Breakpoints',
icon : 'ui-icon-radio-off',
callback : clear_breakpoints
}
]);
var cells = IPython.notebook.get_cells();
for(var i in cells){
var cell = cells[i];
if ((cell instanceof IPython.CodeCell)) {
var prompt = cell.element.find('div.input_prompt');
prompt.click( function() {clickPrompt($(this)); } )
prompt.append('<div class="breakpoint" style="color: rgb(136, 34, 34);"></div');
if (cell.metadata.run_control != undefined) {
makeMarker(cell,cell.metadata.run_control.breakpoint);
}
}
}
};
$([IPython.events]).on('notebook_loaded.Notebook',init_breakpoint);
console.log("Breakpoint extension loaded correctly")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment