diff --git a/demos/python_scripts_examples/gen_gerber_and_drill_files_board.py b/demos/python_scripts_examples/gen_gerber_and_drill_files_board.py index 8a9b2c4a04..12e00ff777 100644 --- a/demos/python_scripts_examples/gen_gerber_and_drill_files_board.py +++ b/demos/python_scripts_examples/gen_gerber_and_drill_files_board.py @@ -72,7 +72,9 @@ plot_plan = [ for layer_info in plot_plan: pctl.SetLayer(layer_info[1]) pctl.OpenPlotfile(layer_info[0], PLOT_FORMAT_GERBER, layer_info[2]) - pctl.PlotLayer() + print 'plot %s' % pctl.GetPlotFileName() + if pctl.PlotLayer() == False: + print "plot error" #generate internal copper layers, if any lyrcnt = board.GetCopperLayerCount(); @@ -81,7 +83,9 @@ for innerlyr in range ( 1, lyrcnt-1 ): pctl.SetLayer(innerlyr) lyrname = 'inner%s' % innerlyr pctl.OpenPlotfile(lyrname, PLOT_FORMAT_GERBER, "inner") - pctl.PlotLayer() + print 'plot %s' % pctl.GetPlotFileName() + if pctl.PlotLayer() == False: + print "plot error" # At the end you have to close the last plot, otherwise you don't know when @@ -96,8 +100,7 @@ drlwriter.SetMapFileFormat( PLOT_FORMAT_PDF ) mirror = False minimalHeader = False offset = wxPoint(0,0) -mergeNPTH = False -drlwriter.SetOptions( mirror, minimalHeader, offset, mergeNPTH ) +drlwriter.SetOptions( mirror, minimalHeader, offset ) metricFmt = True drlwriter.SetFormat( metricFmt ) @@ -107,5 +110,5 @@ genMap = True drlwriter.CreateDrillandMapFilesSet( plotDir, genDrl, genMap ); # One can create a text file to report drill statistics -rptfn = plotDir + '/drill_report.txt' +rptfn = pctl.GetPlotDirName() + 'drill_report.txt' drlwriter.GenDrillReportFile( rptfn ); diff --git a/pcbnew/pcbplot.cpp b/pcbnew/pcbplot.cpp index 77b5d61df2..6d1b1c3137 100644 --- a/pcbnew/pcbplot.cpp +++ b/pcbnew/pcbplot.cpp @@ -371,7 +371,9 @@ bool PLOT_CONTROLLER::OpenPlotfile( const wxString &aSuffix, if( EnsureFileDirectoryExists( &outputDir, boardFilename ) ) { - wxFileName fn( boardFilename ); + // outputDir contains now the full path of plot files + m_plotFile = boardFilename; + m_plotFile.SetPath( outputDir.GetPath() ); wxString fileExt = GetDefaultPlotExtension( aFormat ); // Gerber format can use specific file ext, depending on layers @@ -380,10 +382,11 @@ bool PLOT_CONTROLLER::OpenPlotfile( const wxString &aSuffix, GetPlotOptions().GetUseGerberProtelExtensions() ) fileExt = GetGerberProtelExtension( GetLayer() ); - BuildPlotFileName( &fn, outputDirName, aSuffix, fileExt ); + // Build plot filenames from the board name and layer names: + BuildPlotFileName( &m_plotFile, outputDir.GetPath(), aSuffix, fileExt ); m_plotter = StartPlotBoard( m_board, &GetPlotOptions(), ToLAYER_ID( GetLayer() ), - fn.GetFullPath(), aSheetDesc ); + m_plotFile.GetFullPath(), aSheetDesc ); } return( m_plotter != NULL ); diff --git a/pcbnew/plotcontroller.h b/pcbnew/plotcontroller.h index 480644649e..013afed20c 100644 --- a/pcbnew/plotcontroller.h +++ b/pcbnew/plotcontroller.h @@ -40,6 +40,7 @@ class BOARD; /** * Batch plotter state object. Keeps the plot options and handles multiple * plot requests + * Especially useful in Python scripts */ class PLOT_CONTROLLER { @@ -83,7 +84,27 @@ public: */ bool PlotLayer(); - void SetColorMode( bool aColorMode ); + /** + * @return the current plot full filename, set by OpenPlotfile + */ + const wxString GetPlotFileName() { return m_plotFile.GetFullPath(); } + + /** + * @return the current plot full filename, set by OpenPlotfile + */ + const wxString GetPlotDirName() { return m_plotFile.GetPathWithSep(); } + + /** + * Plotters can plot in Black and White mode or Color mode + * SetColorMode activate/de-actiavte the Color mode. + * @param aColorMode = true to activate the plot color mode + */ + void SetColorMode( bool ); + + /** + * @return true if the current plot color mode is Color, + * false if the current plot color mode is Black and White + */ bool GetColorMode(); private: @@ -99,6 +120,9 @@ private: /// The board we're plotting BOARD* m_board; + + /// The current plot filename, set by OpenPlotfile + wxFileName m_plotFile; }; #endif