a72a63232a
Every basic data type is stored in Base module, color is standing out as one that does not. Moving it to Base opens possibilities to integrate it better with the rest of FreeCAD.
118 lines
4.0 KiB
C++
118 lines
4.0 KiB
C++
/****************************************************************************
|
|
* Copyright (c) 2018 Zheng, Lei (realthunder) <realthunder.dev@gmail.com>*
|
|
* *
|
|
* This file is part of the FreeCAD CAx development system. *
|
|
* *
|
|
* This library is free software; you can redistribute it and/or *
|
|
* modify it under the terms of the GNU Library General Public *
|
|
* License as published by the Free Software Foundation; either *
|
|
* version 2 of the License, or (at your option) any later version. *
|
|
* *
|
|
* This library 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 Library General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU Library General Public *
|
|
* License along with this library; see the file COPYING.LIB. If not, *
|
|
* write to the Free Software Foundation, Inc., 59 Temple Place, *
|
|
* Suite 330, Boston, MA 02111-1307, USA *
|
|
* *
|
|
****************************************************************************/
|
|
|
|
#ifndef IMPORT_EXPORTOCAF2_H
|
|
#define IMPORT_EXPORTOCAF2_H
|
|
|
|
#include <functional>
|
|
#include <map>
|
|
#include <set>
|
|
#include <string>
|
|
#include <unordered_map>
|
|
#include <vector>
|
|
|
|
#include <TDocStd_Document.hxx>
|
|
|
|
#include <Mod/Import/ImportGlobal.h>
|
|
#include "Tools.h"
|
|
|
|
namespace App
|
|
{
|
|
class DocumentObject;
|
|
}
|
|
|
|
namespace Part
|
|
{
|
|
class TopoShape;
|
|
}
|
|
|
|
namespace Import
|
|
{
|
|
|
|
struct ImportExport ExportOCAFOptions
|
|
{
|
|
ExportOCAFOptions();
|
|
Base::Color defaultColor;
|
|
bool exportHidden = true;
|
|
bool keepPlacement = false;
|
|
};
|
|
|
|
class ImportExport ExportOCAF2
|
|
{
|
|
public:
|
|
using GetShapeColorsFunc =
|
|
std::function<std::map<std::string, Base::Color>(App::DocumentObject*, const char*)>;
|
|
explicit ExportOCAF2(Handle(TDocStd_Document) hDoc,
|
|
GetShapeColorsFunc func = GetShapeColorsFunc());
|
|
|
|
static ExportOCAFOptions customExportOptions();
|
|
void setExportOptions(ExportOCAFOptions opts)
|
|
{
|
|
options = opts;
|
|
}
|
|
void setExportHiddenObject(bool enable)
|
|
{
|
|
options.exportHidden = enable;
|
|
}
|
|
void setKeepPlacement(bool enable)
|
|
{
|
|
options.keepPlacement = enable;
|
|
}
|
|
void exportObjects(std::vector<App::DocumentObject*>& objs, const char* name = nullptr);
|
|
bool canFallback(std::vector<App::DocumentObject*> objs);
|
|
|
|
private:
|
|
TDF_Label exportObject(App::DocumentObject* obj,
|
|
const char* sub,
|
|
TDF_Label parent,
|
|
const char* name = nullptr);
|
|
void setupObject(TDF_Label label,
|
|
App::DocumentObject* obj,
|
|
const Part::TopoShape& shape,
|
|
const std::string& prefix,
|
|
const char* name = nullptr,
|
|
bool force = false);
|
|
void setName(TDF_Label label, App::DocumentObject* obj, const char* name = nullptr);
|
|
TDF_Label findComponent(const char* subname, TDF_Label label, TDF_LabelSequence& labels);
|
|
|
|
private:
|
|
Handle(TDocStd_Document) pDoc;
|
|
Handle(XCAFDoc_ShapeTool) aShapeTool;
|
|
Handle(XCAFDoc_ColorTool) aColorTool;
|
|
|
|
std::unordered_map<App::DocumentObject*, TDF_Label> myObjects;
|
|
|
|
std::unordered_map<TDF_Label, std::vector<std::string>, LabelHasher> myNames;
|
|
|
|
std::set<std::pair<App::DocumentObject*, std::string>> mySetups;
|
|
|
|
std::vector<App::DocumentObject*> groupLinks;
|
|
|
|
GetShapeColorsFunc getShapeColors;
|
|
|
|
ExportOCAFOptions options;
|
|
};
|
|
|
|
} // namespace Import
|
|
|
|
#endif // IMPORT_EXPORTOCAF2_H
|