diff --git a/ide/static/js/content.js b/ide/static/js/content.js index 925bb9992..faab2efc1 100644 --- a/ide/static/js/content.js +++ b/ide/static/js/content.js @@ -41,11 +41,13 @@ class Content extends React.Component { rebuildNet: false, selectedPhase: 0, error: [], + multiple_layerId: [], load: false, modalIsOpen: false, totalParameters: 0, modelConfig: null, modelFramework: 'caffe' + }; this.addNewLayer = this.addNewLayer.bind(this); this.changeSelectedLayer = this.changeSelectedLayer.bind(this); @@ -89,6 +91,8 @@ class Content extends React.Component { // Might need to improve the logic of clickEvent this.clickEvent = false; this.handleClick = this.handleClick.bind(this); + this.store_list_layers = this.store_list_layers.bind(this); + this.delete_selected_layers = this.delete_selected_layers.bind(this); } openModal() { this.setState({modalIsOpen: true}); @@ -241,6 +245,27 @@ class Content extends React.Component { this.setState({ net, selectedLayer: null, nextLayerId: nextLayerId, totalParameters: totalParameters }); } + store_list_layers(layerId){ + var stored_list = this.state.multiple_layerId; + stored_list.push(layerId); + this.setState({multiple_layerId: stored_list}); + //console.log("Flag1"); + } + + delete_selected_layers(){ + const layers = this.state.multiple_layerId; + layers.forEach(layerId => { + this.deleteLayer(layerId); + }); + var multiple_layerId = []; + this.setState({multiple_layerId: multiple_layerId}); + + } + + + + + updateLayerShape(net, layerId) { const netData = JSON.parse(JSON.stringify(net)); Object.keys(netData[layerId].params).forEach(param => { @@ -1053,6 +1078,8 @@ class Content extends React.Component { copyTrain={this.copyTrain} trainOnly={this.trainOnly} updateLayerWithShape={this.modifyLayer} + store_list_layers={this.store_list_layers} + delete_selected_layers={this.delete_selected_layers} /> this.props.deleteLayer(this.props.selectedLayer)} > - DELETE LAYER + DELETE LAYER + + + + + ); @@ -175,7 +194,9 @@ SetParams.propTypes = { selectedPhase: React.PropTypes.number, copyTrain: React.PropTypes.func, changeSelectedLayer: React.PropTypes.func, - updateLayerWithShape: React.PropTypes.func + updateLayerWithShape: React.PropTypes.func, + store_list_layers: React.PropTypes.func, + delete_selected_layers: React.PropTypes.func }; export default SetParams;