104 lines
3.3 KiB
C++
104 lines
3.3 KiB
C++
// SPDX-License-Identifier: LGPL-2.1-or-later
|
|
/****************************************************************************
|
|
* *
|
|
* Copyright (c) 2023 Ondsel <development@ondsel.com> *
|
|
* *
|
|
* This file is part of FreeCAD. *
|
|
* *
|
|
* FreeCAD is free software: you can redistribute it and/or modify it *
|
|
* under the terms of the GNU Lesser General Public License as *
|
|
* published by the Free Software Foundation, either version 2.1 of the *
|
|
* License, or (at your option) any later version. *
|
|
* *
|
|
* FreeCAD 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 *
|
|
* Lesser General Public License for more details. *
|
|
* *
|
|
* You should have received a copy of the GNU Lesser General Public *
|
|
* License along with FreeCAD. If not, see *
|
|
* <https://www.gnu.org/licenses/>. *
|
|
* *
|
|
***************************************************************************/
|
|
|
|
|
|
#ifndef ASSEMBLY_BomObject_H
|
|
#define ASSEMBLY_BomObject_H
|
|
|
|
#include <App/PropertyFile.h>
|
|
|
|
#include <Mod/Assembly/AssemblyGlobal.h>
|
|
|
|
#include <Mod/Spreadsheet/App/Sheet.h>
|
|
#include <App/PropertyLinks.h>
|
|
|
|
namespace App
|
|
{
|
|
class DocumentObject;
|
|
}
|
|
|
|
namespace Assembly
|
|
{
|
|
|
|
class AssemblyObject;
|
|
|
|
class BomDataElement
|
|
{
|
|
public:
|
|
BomDataElement(std::string objName, std::string columnName, std::string value)
|
|
: objName(objName)
|
|
, columnName(columnName)
|
|
, value(value)
|
|
{}
|
|
~BomDataElement()
|
|
{}
|
|
|
|
std::string objName;
|
|
std::string columnName;
|
|
std::string value;
|
|
};
|
|
|
|
class AssemblyExport BomObject: public Spreadsheet::Sheet
|
|
{
|
|
PROPERTY_HEADER_WITH_OVERRIDE(Assembly::BomObject);
|
|
|
|
public:
|
|
BomObject();
|
|
~BomObject() override;
|
|
|
|
PyObject* getPyObject() override;
|
|
|
|
const char* getViewProviderName() const override
|
|
{
|
|
return "AssemblyGui::ViewProviderBom";
|
|
}
|
|
|
|
App::DocumentObjectExecReturn* execute() override;
|
|
|
|
void generateBOM();
|
|
void addObjectToBom(App::DocumentObject* obj, size_t row, std::string index);
|
|
void
|
|
addObjectChildrenToBom(std::vector<App::DocumentObject*> objs, size_t& row, std::string index);
|
|
void saveCustomColumnData();
|
|
|
|
AssemblyObject* getAssembly();
|
|
|
|
bool hasQuantityColumn();
|
|
int getColumnIndex(std::string name);
|
|
std::string getText(size_t row, size_t col);
|
|
|
|
App::PropertyStringList columnsNames;
|
|
App::PropertyBool detailSubAssemblies;
|
|
App::PropertyBool detailParts;
|
|
App::PropertyBool onlyParts;
|
|
|
|
std::vector<BomDataElement> dataElements;
|
|
std::vector<App::DocumentObject*> obj_list;
|
|
};
|
|
|
|
|
|
} // namespace Assembly
|
|
|
|
|
|
#endif // ASSEMBLY_BomObject_H
|