7
mirror of https://gitlab.com/kicad/code/kicad.git synced 2024-11-22 17:55:01 +00:00
kicad/pcbnew/exporters/step/step_pcb_model.h
2024-11-04 16:15:02 +03:00

290 lines
12 KiB
C++

/*
* This program source code file is part of KiCad, a free EDA CAD application.
*
* Copyright (C) 2016 Cirilo Bernardo <cirilo.bernardo@gmail.com>
* Copyright (C) 2021-2024 KiCad Developers, see AUTHORS.txt for contributors.
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, you may find one here:
* http://www.gnu.org/licenses/old-licenses/gpl-2.0.html
* or you may search the http://www.gnu.org website for the version 2 license,
* or you may write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
*/
#ifndef OCE_VIS_OCE_UTILS_H
#define OCE_VIS_OCE_UTILS_H
#include <list>
#include <map>
#include <string>
#include <utility>
#include <vector>
#include <Standard_Handle.hxx>
#include <TDF_Label.hxx>
#include <TopoDS_Shape.hxx>
#include <math/vector2d.h>
#include <math/vector3.h>
#include <geometry/shape_poly_set.h>
#include <board_stackup_manager/board_stackup.h>
/**
* Default distance between points to treat them as separate ones (mm)
* 0.001 mm or less is a reasonable value. A too large value creates issues by
* merging points that should be different.
* Remember we are a 3D space, so a thin shape can be broken if 2 points
* are merged (in X, Y, Z coords) when they should not.
* round shapes converted to polygon can also be not good with a to large value
*/
static constexpr double OCC_MAX_DISTANCE_TO_MERGE_POINTS = 0.001;
// default PCB thickness in mm
static constexpr double BOARD_THICKNESS_DEFAULT_MM = 1.6;
// minimum PCB thickness in mm (10 microns assumes a very thin polyimide film)
// must be > OCC_MAX_DISTANCE_TO_MERGE_POINTS
static constexpr double BOARD_THICKNESS_MIN_MM = 0.01;
// default copper thickness in mm
// must be > OCC_MAX_DISTANCE_TO_MERGE_POINTS
static constexpr double COPPER_THICKNESS_DEFAULT_MM = 0.035;
// Max error to approximate an arc by segments (in mm)
static constexpr double ARC_TO_SEGMENT_MAX_ERROR_MM = 0.005;
class PAD;
class TDocStd_Document;
class XCAFApp_Application;
class XCAFDoc_ShapeTool;
typedef std::pair< std::string, TDF_Label > MODEL_DATUM;
typedef std::map< std::string, TDF_Label > MODEL_MAP;
extern void ReportMessage( const wxString& aMessage );
enum class OUTPUT_FORMAT
{
FMT_OUT_UNKNOWN = 0,
FMT_OUT_STEP,
FMT_OUT_IGES,
FMT_OUT_BREP,
FMT_OUT_XAO,
FMT_OUT_GLTF,
FMT_OUT_PLY,
FMT_OUT_STL
};
class STEP_PCB_MODEL
{
public:
STEP_PCB_MODEL( const wxString& aPcbName );
virtual ~STEP_PCB_MODEL();
// Update m_outFmt to aVariant, giving the output format variant
void SpecializeVariant( OUTPUT_FORMAT aVariant ) { m_outFmt = aVariant; }
// add a pad shape (must be in final position)
bool AddPadShape( const PAD* aPad, const VECTOR2D& aOrigin, bool aVia );
// add a pad hole or slot (must be in final position)
bool AddHole( const SHAPE_SEGMENT& aShape, int aPlatingThickness, PCB_LAYER_ID aLayerTop,
PCB_LAYER_ID aLayerBot, bool aVia, const VECTOR2D& aOrigin );
// add a plated hole shape (without the hole)
bool AddBarrel( const SHAPE_SEGMENT& aShape, PCB_LAYER_ID aLayerTop, PCB_LAYER_ID aLayerBot,
bool aVia, const VECTOR2D& aOrigin );
// add a set of polygons (must be in final position)
bool AddPolygonShapes( const SHAPE_POLY_SET* aPolyShapes, PCB_LAYER_ID aLayer,
const VECTOR2D& aOrigin );
// add a component at the given position and orientation
bool AddComponent( const std::string& aFileName, const std::string& aRefDes, bool aBottom,
VECTOR2D aPosition, double aRotation, VECTOR3D aOffset,
VECTOR3D aOrientation, VECTOR3D aScale, bool aSubstituteModels = true );
void SetCopperColor( double r, double g, double b );
void SetPadColor( double r, double g, double b );
void SetEnabledLayers( const LSET& aLayers );
void SetFuseShapes( bool aValue );
void SetSimplifyShapes( bool aValue );
void SetAddVisMaterials( bool aValue );
void SetStackup( const BOARD_STACKUP& aStackup );
void SetNetFilter( const wxString& aFilter );
// Set the max distance (in mm) to consider 2 points have the same coordinates
// and can be merged
void OCCSetMergeMaxDistance( double aDistance = OCC_MAX_DISTANCE_TO_MERGE_POINTS );
void SetMaxError( int aMaxError ) { m_maxError = aMaxError; }
// create the PCB model using the current outlines and drill holes
bool CreatePCB( SHAPE_POLY_SET& aOutline, VECTOR2D aOrigin, bool aPushBoardBody );
/**
* Convert a SHAPE_POLY_SET to TopoDS_Shape's (polygonal vertical prisms, or flat faces)
* @param aShapes is the TopoDS_Shape list to append to
* @param aPolySet is a polygon set
* @param aConvertToArcs set to approximate with arcs
* @param aThickness is the height of the created prism, or 0.0: flat face pointing up, -0.0: down.
* @param aOrigin is the origin of the coordinates
* @return true if success
*/
bool MakeShapes( std::vector<TopoDS_Shape>& aShapes, const SHAPE_POLY_SET& aPolySet,
bool aConvertToArcs, double aThickness, double aZposition,
const VECTOR2D& aOrigin );
/**
* Make a segment shape based on start and end point. If they're too close, make a cylinder.
* It is a specialized version of MakeShape()
* @param aShape is the TopoDS_Shape to initialize (must be empty)
* @param aStartPoint is the start point of the segment
* @param aEndPoint is the end point of the segment
* @param aWidth is the width of the segment
* @param aThickness is the height of the created segment, or 0.0: flat face pointing up, -0.0: down.
* @param aOrigin is the origin of the coordinates
* @return true if success
*/
bool MakeShapeAsThickSegment( TopoDS_Shape& aShape,
VECTOR2D aStartPoint, VECTOR2D aEndPoint,
double aWidth, double aThickness, double aZposition,
const VECTOR2D& aOrigin );
#ifdef SUPPORTS_IGES
// write the assembly model in IGES format
bool WriteIGES( const wxString& aFileName );
#endif
// write the assembly model in STEP format
bool WriteSTEP( const wxString& aFileName, bool aOptimize );
// write the assembly in BREP format
bool WriteBREP( const wxString& aFileName );
// write the assembly in XAO format with pad faces as groups
bool WriteXAO( const wxString& aFileName );
/**
* Write the assembly in binary GLTF Format
*
* We only support binary GLTF because GLTF is weird
* Officially, binary GLTF is actually json+binary in one file
* If we elected non-binary output with opecascade, it will generate
* that one file as two separate files, one containing json that references the binary
* Which is actually more annoying to deal with (to do the temp file rename, since we dont
* control the binary name) and silly when you can just have the one file.
*
* @param aFileName Output file path
*
* @return true if the write succeeded without error
*/
bool WriteGLTF( const wxString& aFileName );
// write the assembly in PLY format (mesh)
bool WritePLY( const wxString& aFileName );
// write the assembly in STL format (mesh)
bool WriteSTL( const wxString& aFileName );
private:
/**
* @return true if the board(s) outline is valid. False otherwise
*/
bool isBoardOutlineValid();
void getLayerZPlacement( const PCB_LAYER_ID aLayer, double& aZPos, double& aThickness );
void getCopperLayerZPlacement( const PCB_LAYER_ID aLayer, double& aZPos, double& aThickness );
void getBoardBodyZPlacement( double& aZPos, double& aThickness );
/**
* Load a 3D model data.
*
* @param aFileNameUTF8 is the filename encoded UTF8 (different formats allowed)
* but for WRML files a model data can be loaded instead of the vrml data,
* not suitable in a step file.
* @param aScale is the X,Y,Z scaling factors.
* @param aLabel is the TDF_Label to store the data.
* @param aSubstituteModels = true to allows data substitution, false to disallow.
* @param aErrorMessage (can be nullptr) is an error message to be displayed on error.
* @return true if successfully loaded, false on error.
*/
bool getModelLabel( const std::string& aFileNameUTF8, VECTOR3D aScale, TDF_Label& aLabel,
bool aSubstituteModels, wxString* aErrorMessage = nullptr );
bool getModelLocation( bool aBottom, VECTOR2D aPosition, double aRotation, VECTOR3D aOffset,
VECTOR3D aOrientation, TopLoc_Location& aLocation );
bool readIGES( Handle( TDocStd_Document ) & aDoc, const char* aFname );
bool readSTEP( Handle( TDocStd_Document ) & aDoc, const char* aFname );
bool readVRML( Handle( TDocStd_Document ) & aDoc, const char* aFname );
bool performMeshing( Handle( XCAFDoc_ShapeTool ) & aShapeTool );
TDF_Label transferModel( Handle( TDocStd_Document )& source, Handle( TDocStd_Document ) & dest,
VECTOR3D aScale );
Handle( XCAFApp_Application ) m_app;
Handle( TDocStd_Document ) m_doc;
Handle( XCAFDoc_ShapeTool ) m_assy;
TDF_Label m_assy_label;
bool m_hasPCB; // set true if CreatePCB() has been invoked
bool m_simplifyShapes; // convert parts of outlines to arcs where possible
bool m_fuseShapes; // fuse geometry together
std::vector<TDF_Label> m_pcb_labels; // labels for the PCB model (one by main outline)
MODEL_MAP m_models; // map of file names to model labels
int m_components; // number of successfully loaded components;
double m_precision; // model (length unit) numeric precision
double m_angleprec; // angle numeric precision
double m_copperColor[3]; // copper, RGB values
double m_padColor[3]; // pads, RGB values
BOARD_STACKUP m_stackup; // board stackup
LSET m_enabledLayers; // a set of layers enabled for export
wxString m_netFilter; // remove nets not matching this wildcard
double m_minx; // leftmost curve point
double m_mergeOCCMaxDist; // minimum distance (mm) below which two
// points are considered coincident by OCC
// Holes in copper and main outlines
std::vector<TopoDS_Shape> m_copperCutouts;
std::vector<TopoDS_Shape> m_boardCutouts;
// Main outlines (more than one board)
std::vector<TopoDS_Shape> m_board_outlines;
std::vector<TopoDS_Shape> m_board_copper;
std::vector<TopoDS_Shape> m_board_copper_pads;
std::vector<TopoDS_Shape> m_board_copper_vias;
std::vector<TopoDS_Shape> m_board_copper_fused;
std::vector<TopoDS_Shape> m_board_silkscreen;
std::vector<TopoDS_Shape> m_board_soldermask;
// Data for pads
std::map<wxString, std::pair<gp_Pnt, TopoDS_Shape>> m_pad_points;
/// Name of the PCB, which will most likely be the file name of the path.
wxString m_pcbName;
int m_maxError;
/// The current output format for created file
OUTPUT_FORMAT m_outFmt;
};
#endif // OCE_VIS_OCE_UTILS_H