STEPWriter.h

Go to the documentation of this file.

#ifndef STEPWRITER_H
#define STEPWRITER_H
 
#include <TDocStd_Document.hxx>
#include <XCAFDoc_ShapeTool.hxx>
#include <XCAFDoc_ColorTool.hxx>
#include <Quantity_Color.hxx>
#include <TopoDS_Shape.hxx>
 
class STEPWriter {
public:
    STEPWriter();
    void addShape(TopoDS_Shape& shape, int colorIndex, int id, bool type);
    bool writeSTEP(std::string filename);
private:
    const std::vector<Quantity_Color> COLORS = {
        Quantity_Color(Quantity_NOC_RED),
        Quantity_Color(Quantity_NOC_GREEN),
        Quantity_Color(Quantity_NOC_BLUE),
        Quantity_Color(Quantity_NOC_YELLOW),
        Quantity_Color(Quantity_NOC_CYAN),
        Quantity_Color(Quantity_NOC_MAGENTA)
    };
 
    Handle(TDocStd_Document) doc;
    Handle(XCAFDoc_ShapeTool) shapeTool;
    Handle(XCAFDoc_ColorTool) colorTool;
};
 
#endif // STEPWRITER_H