363 lines
12 KiB
C++
363 lines
12 KiB
C++
/*
|
|
* This program source code file is part of KiCad, a free EDA CAD application.
|
|
*
|
|
* Copyright (C) 2017 Oliver Walters
|
|
* Copyright (C) 2017 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 _EESCHEMA_BOM_TABLE_MODEL_H_
|
|
#define _EESCHEMA_BOM_TABLE_MODEL_H_
|
|
|
|
#include <wx/dataview.h>
|
|
#include <wx/regex.h>
|
|
|
|
#include <vector>
|
|
#include <map>
|
|
|
|
#include <sch_component.h>
|
|
#include <sch_reference_list.h>
|
|
#include <netlist_object.h>
|
|
#include <template_fieldnames.h>
|
|
|
|
#include "bom_table_column.h"
|
|
|
|
// Forward-declare classes
|
|
class BOM_TABLE_ROW; // Base-class for table row data model
|
|
class BOM_TABLE_GROUP; // Class for displaying a group of components
|
|
class BOM_TABLE_COMPONENT; // Class for displaying a single component
|
|
|
|
// Map column IDs to field values (for quick lookup)
|
|
typedef std::map<unsigned int, wxString> FIELD_VALUE_MAP;
|
|
|
|
|
|
/**
|
|
* The BOM_FIELD_VALUES class provides quick lookup of component values
|
|
* (based on Field ID)
|
|
* This is done for the following reasons:
|
|
* - Increase lookup speed for table values
|
|
* - Allow field values to be reverted to original values
|
|
* - Allow duplicate components to reference the same data
|
|
*/
|
|
class BOM_FIELD_VALUES
|
|
{
|
|
public:
|
|
BOM_FIELD_VALUES( const wxString& aRefDes, FIELD_VALUE_MAP* aTemplate );
|
|
|
|
/**
|
|
* Return the current value for the provided field ID
|
|
* @return true if the field exists
|
|
* @param aFieldId = the field index
|
|
* @param aValue = a string to return the field value
|
|
*/
|
|
bool GetFieldValue( unsigned int aFieldId, wxString& aValue ) const;
|
|
|
|
/**
|
|
* Return the backup value for the provided field ID
|
|
* @return true if the field exists
|
|
* @param aFieldId = the field index
|
|
* @param aValue = a string to return the field backup value
|
|
*/
|
|
bool GetBackupValue( unsigned int aFieldId, wxString& aValue ) const;
|
|
|
|
/**
|
|
* Return the template value for a provided field ID (if it exists)
|
|
* @return true if the field exists
|
|
* @param aFieldId = the field index
|
|
* @param aValue = a string to return the field template value
|
|
*/
|
|
bool GetTemplateValue( unsigned int aFieldId, wxString& aValue ) const;
|
|
|
|
/**
|
|
* Set the value for the provided field ID
|
|
* Field value is set under any of the following conditions:
|
|
* - param aOverwrite is true
|
|
* - There is no current value
|
|
* - The current value is empty
|
|
*/
|
|
void SetFieldValue( unsigned int aFieldId, const wxString& aValue, bool aOverwrite = false );
|
|
|
|
wxString GetReference() const { return m_refDes; }
|
|
|
|
bool HasValueChanged( unsigned int aFieldId ) const;
|
|
|
|
void RevertChanges( unsigned int aFieldId );
|
|
|
|
void SetBackupPoint();
|
|
|
|
protected:
|
|
//! The RefDes to which these values correspond
|
|
wxString m_refDes;
|
|
|
|
//! Current values for each column
|
|
FIELD_VALUE_MAP m_currentValues;
|
|
|
|
//! Backup values for each column
|
|
FIELD_VALUE_MAP m_backupValues;
|
|
|
|
//! Template values for each column
|
|
FIELD_VALUE_MAP* m_templateValues;
|
|
};
|
|
|
|
/**
|
|
* Virtual base class determining how a row is displayed
|
|
* There are three types of rows:
|
|
* GROUP - Displays a group of (one or more) components
|
|
* COMPONENT - Displays a single component
|
|
* UNIT - Child of COMPONENT for multi-unit components
|
|
*/
|
|
class BOM_TABLE_ROW
|
|
{
|
|
public:
|
|
BOM_TABLE_ROW();
|
|
virtual ~BOM_TABLE_ROW() {}
|
|
|
|
/// Set display properties for a cell
|
|
virtual bool GetAttr( unsigned int aFieldId, wxDataViewItemAttr& aAttr ) const;
|
|
|
|
/// Get the row value associated with provided field ID
|
|
virtual wxString GetFieldValue( unsigned int aFieldId ) const = 0;
|
|
|
|
/// Set the field value associated with the provided field ID
|
|
virtual bool SetFieldValue( unsigned int aFieldId, const wxString aValue, bool aOverwrite = false ) = 0;
|
|
|
|
/// Return parent item
|
|
virtual BOM_TABLE_ROW* GetParent() const { return nullptr; }
|
|
|
|
/// Test if row has any child rows
|
|
virtual bool HasChildren() const { return false; }
|
|
|
|
/// Return any child rows
|
|
virtual unsigned int GetChildren( wxDataViewItemArray& aChildren ) const { return 0; }
|
|
|
|
/// Determine if a value has changed
|
|
virtual bool HasValueChanged( BOM_COLUMN* aField ) const { return false; }
|
|
|
|
/// Determine if any values have changed
|
|
bool HasChanged() const;
|
|
|
|
protected:
|
|
|
|
/// Pointer to list of columns
|
|
BOM_COLUMN_LIST* m_columnList;
|
|
};
|
|
|
|
/**
|
|
* BOM_TABLE_GROUP class displays a group of similar components
|
|
* If the group contains more than one component,
|
|
* they are each displayed as child items of the group
|
|
*/
|
|
class BOM_TABLE_GROUP : public BOM_TABLE_ROW
|
|
{
|
|
public:
|
|
/// List of components stored in this group
|
|
std::vector<BOM_TABLE_COMPONENT*> Components;
|
|
|
|
BOM_TABLE_GROUP( BOM_COLUMN_LIST* aColumnList );
|
|
virtual ~BOM_TABLE_GROUP() {}
|
|
|
|
/// Set display properties for a group row
|
|
virtual bool GetAttr( unsigned int aFieldId, wxDataViewItemAttr& aAttr ) const override;
|
|
|
|
/// Get group row value
|
|
virtual wxString GetFieldValue( unsigned int aFieldId ) const override;
|
|
|
|
/// Set group row value
|
|
virtual bool SetFieldValue( unsigned int aFieldId, const wxString aValue, bool aOverwrite = false ) override;
|
|
|
|
/// Attempt to add a new component to the group
|
|
bool AddComponent( BOM_TABLE_COMPONENT* aComponent );
|
|
|
|
/// Test if this group should display children
|
|
virtual bool HasChildren() const override { return Components.size() > 1; }
|
|
|
|
/// Return a list of children items of this group
|
|
virtual unsigned int GetChildren( wxDataViewItemArray& aChildren ) const override;
|
|
|
|
/// Test if any children have changed
|
|
virtual bool HasValueChanged( BOM_COLUMN* aField ) const override;
|
|
|
|
/// Return the number of child items in this group
|
|
unsigned int GroupSize( void ) const { return Components.size(); }
|
|
|
|
/// Return a sorted, concatenated list of references
|
|
wxArrayString GetReferences( bool aSort = true ) const;
|
|
|
|
/// Function for sorting two reference strings
|
|
static int SortReferences( const wxString& aFirst, const wxString& aSecond );
|
|
|
|
/// Function for sorting two value strings
|
|
static int SortValues( const wxString& aFirst, const wxString& aSecond );
|
|
|
|
protected:
|
|
/// Test if a particular field matches against another component
|
|
bool TestField( BOM_COLUMN* aField, BOM_TABLE_COMPONENT* aComponent ) const;
|
|
};
|
|
|
|
class BOM_TABLE_COMPONENT : public BOM_TABLE_ROW
|
|
{
|
|
public:
|
|
// List of units associated with this component
|
|
std::vector<SCH_REFERENCE> Units;
|
|
|
|
BOM_TABLE_COMPONENT( BOM_TABLE_GROUP* aParent, BOM_COLUMN_LIST* aColumnList, BOM_FIELD_VALUES* aValues );
|
|
|
|
bool AddUnit( const SCH_REFERENCE& aUnit );
|
|
|
|
virtual wxString GetFieldValue( unsigned int aFieldId ) const override;
|
|
|
|
virtual bool SetFieldValue( unsigned int aFieldId, const wxString aValue, bool aOverwrite = false ) override;
|
|
|
|
virtual bool HasValueChanged( BOM_COLUMN* aField ) const override;
|
|
|
|
/// Return the reference of the first unit (all units must be the same
|
|
wxString GetReference() const;
|
|
|
|
/**
|
|
* @return the prefix of a component e.g. "R23" returns "R"
|
|
*/
|
|
wxString GetPrefix() const;
|
|
|
|
void ApplyFieldChanges();
|
|
|
|
void RevertFieldChanges();
|
|
|
|
void SetParent( BOM_TABLE_GROUP* aParent ) { m_parent = aParent; }
|
|
|
|
virtual BOM_TABLE_ROW* GetParent() const override { return m_parent; }
|
|
|
|
protected:
|
|
BOM_TABLE_GROUP* m_parent;
|
|
|
|
BOM_FIELD_VALUES* m_fieldValues;
|
|
};
|
|
|
|
/**
|
|
* BOM_TABLE_MODEL class
|
|
*
|
|
* Contains complete BOM information:
|
|
* a) List of columns (fields) to display
|
|
* b) List of groups of consolidated components
|
|
*/
|
|
class BOM_TABLE_MODEL : public wxDataViewModel
|
|
{
|
|
protected:
|
|
BOM_TABLE_MODEL();
|
|
|
|
// Vector of unique component rows
|
|
std::vector<std::unique_ptr<BOM_TABLE_COMPONENT>> m_components;
|
|
|
|
// Vector of field values mapped to field IDs
|
|
std::vector<std::unique_ptr<BOM_FIELD_VALUES>> m_fieldValues;
|
|
|
|
// Template field values
|
|
FIELD_VALUE_MAP m_fieldTemplates;
|
|
|
|
// BOM Preferences
|
|
//! Group components based on values
|
|
bool m_groupColumns = true;
|
|
//! Filter components
|
|
bool m_filterColumns = false;
|
|
//! Allow blank fields to be merged with otherwise matching groups
|
|
bool m_mergeBlankFields = false;
|
|
|
|
wxDataViewCtrl* m_widget;
|
|
|
|
//! ID of column to sort by
|
|
unsigned int m_sortingColumn;
|
|
bool m_sortingOrder;
|
|
|
|
void AddDefaultColumns();
|
|
void ClearColumns();
|
|
|
|
virtual bool HasContainerColumns( const wxDataViewItem& aItem ) const override { return true; }
|
|
|
|
virtual bool IsContainer( const wxDataViewItem& aItem ) const override;
|
|
|
|
virtual wxDataViewItem GetParent( const wxDataViewItem& aItem ) const override;
|
|
|
|
virtual unsigned int GetChildren( const wxDataViewItem& aItem, wxDataViewItemArray& aChildren ) const override;
|
|
|
|
virtual unsigned int GetColumnCount() const override { return ColumnList.ColumnCount( false ); }
|
|
|
|
virtual wxString GetColumnType( unsigned int aFieldId ) const override { return wxString( "string" ); }
|
|
|
|
virtual void GetValue( wxVariant& aVariant, const wxDataViewItem& aItem, unsigned int aFieldId ) const override;
|
|
|
|
virtual bool SetValue( const wxVariant& aVariant, const wxDataViewItem& item, unsigned int aFieldId ) override;
|
|
|
|
public:
|
|
|
|
virtual ~BOM_TABLE_MODEL();
|
|
|
|
BOM_COLUMN_LIST ColumnList;
|
|
|
|
/// List of component groups
|
|
std::vector<std::unique_ptr<BOM_TABLE_GROUP>> Groups;
|
|
|
|
typedef wxObjectDataPtr<BOM_TABLE_MODEL> MODEL_PTR;
|
|
|
|
static MODEL_PTR Create();
|
|
|
|
void AttachTo( wxDataViewCtrl* aView );
|
|
|
|
wxDataViewColumn* AddColumn( BOM_COLUMN* aColumn, int aPosition = -1 );
|
|
bool RemoveColumn( BOM_COLUMN* aColumn );
|
|
|
|
// wxDataViewModel functions
|
|
virtual bool GetAttr( const wxDataViewItem& aItem, unsigned int aFieldId, wxDataViewItemAttr& aAttr ) const override;
|
|
virtual bool HasDefaultCompare() const override { return false; }
|
|
virtual int Compare( const wxDataViewItem& aItem1,
|
|
const wxDataViewItem& aItem2,
|
|
unsigned int aColumnId,
|
|
bool aAscending ) const override;
|
|
|
|
void ReloadTable();
|
|
|
|
unsigned int ColumnCount() const { return ColumnList.ColumnCount(); }
|
|
unsigned int GroupCount() const { return (unsigned int) Groups.size(); }
|
|
unsigned int ComponentCount() const;
|
|
|
|
void SetColumnGrouping( const bool aGroup = true ) { m_groupColumns = aGroup; }
|
|
bool GetColumnGrouping() const { return m_groupColumns; }
|
|
|
|
void SetColumnFiltering( const bool aFilter = true ) { m_filterColumns = aFilter; }
|
|
bool GetColumnFiltering() const { return m_filterColumns; }
|
|
|
|
void SetBlankMerging( const bool aMerge = true ) { m_mergeBlankFields = aMerge; }
|
|
bool GetBlankMerging() const { return m_mergeBlankFields; }
|
|
|
|
wxArrayString GetRowData( unsigned int aRow, std::vector<BOM_COLUMN*> aColumns ) const;
|
|
|
|
void SetComponents( SCH_REFERENCE_LIST aRefs, const TEMPLATE_FIELDNAMES& aTemplateFields );
|
|
void AddComponentFields( SCH_COMPONENT* aCmp );
|
|
|
|
void RevertFieldChanges();
|
|
void ApplyFieldChanges();
|
|
|
|
bool HaveFieldsChanged() const;
|
|
|
|
void SetBackupPoint();
|
|
|
|
std::vector<SCH_REFERENCE> GetChangedComponents();
|
|
unsigned int CountChangedComponents();
|
|
};
|
|
|
|
#endif // _EESCHEMA_BOM_TABLE_MODEL_H_
|