KiCad PCB EDA Suite
lib_tree_model_adapter.h
Go to the documentation of this file.
1 /*
2  * This program source code file is part of KiCad, a free EDA CAD application.
3  *
4  * Copyright (C) 2017 Chris Pavlina <pavlina.chris@gmail.com>
5  * Copyright (C) 2014 Henner Zeller <h.zeller@acm.org>
6  * Copyright (C) 2014-2019 KiCad Developers, see AUTHORS.txt for contributors.
7  *
8  * This program is free software: you can redistribute it and/or modify it
9  * under the terms of the GNU General Public License as published by the
10  * Free Software Foundation, either version 3 of the License, or (at your
11  * option) any later version.
12  *
13  * This program is distributed in the hope that it will be useful, but
14  * WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16  * General Public License for more details.
17  *
18  * You should have received a copy of the GNU General Public License along
19  * with this program. If not, see <http://www.gnu.org/licenses/>.
20  */
21 
22 #ifndef LIB_TREE_MODEL_ADAPTER_H
23 #define LIB_TREE_MODEL_ADAPTER_H
24 
25 #include <lib_id.h>
26 #include <lib_tree_model.h>
27 #include <wx/hashmap.h>
28 #include <wx/dataview.h>
29 #include <wx/headerctrl.h>
30 #include <vector>
31 #include <functional>
32 
91 class TOOL_INTERACTIVE;
92 
93 class LIB_TREE_MODEL_ADAPTER: public wxDataViewModel
94 {
95 public:
96 
100  typedef wxObjectDataPtr<LIB_TREE_MODEL_ADAPTER> PTR;
101 
107 
112  {
115  };
116 
121  {
122  PART_COL = 0,
125  };
126 
131  void SaveColWidths();
132 
138  void SetFilter( CMP_FILTER_TYPE aFilter );
139 
143  CMP_FILTER_TYPE GetFilter() const { return m_filter; }
144 
151  void ShowUnits( bool aShow );
152 
160  void SetPreselectNode( LIB_ID const& aLibId, int aUnit );
161 
170  void DoAddLibrary( wxString const& aNodeName, wxString const& aDesc,
171  std::vector<LIB_TREE_ITEM*> const& aItemList, bool presorted );
172 
173 
178 
184  void UpdateSearchString( wxString const& aSearch );
185 
192  void AttachTo( wxDataViewCtrl* aDataViewCtrl );
193 
202  LIB_ID GetAliasFor( const wxDataViewItem& aSelection ) const;
203 
213  int GetUnitFor( const wxDataViewItem& aSelection ) const;
214 
223  LIB_TREE_NODE::TYPE GetTypeFor( const wxDataViewItem& aSelection ) const;
224 
225  virtual wxString GenerateInfo( LIB_ID const& aLibId, int aUnit ) { return wxEmptyString; };
226 
230  int GetItemCount() const;
231 
235  virtual int GetLibrariesCount() const
236  {
237  return m_tree.Children.size();
238  }
239 
246  wxDataViewItem FindItem( const LIB_ID& aLibId );
247 
253  unsigned int GetChildren( wxDataViewItem const& aItem,
254  wxDataViewItemArray& aChildren ) const override;
255 
256  // Freezing/Thawing. Used when updating the table model so that we don't try and fetch
257  // values during updating. Primarily a problem on OSX which doesn't pay attention to the
258  // wxDataViewCtrl's freeze count when updating the keyWindow.
259  void Freeze() { m_freeze++; }
260  void Thaw() { m_freeze--; }
261  bool IsFrozen() const { return m_freeze; }
262 
263  // Allows subclasses to nominate a context menu handler.
264  virtual TOOL_INTERACTIVE* GetContextMenuTool() { return nullptr; }
265 
266 protected:
267  static wxDataViewItem ToItem( LIB_TREE_NODE const* aNode );
268  static LIB_TREE_NODE const* ToNode( wxDataViewItem aItem );
269  static unsigned int IntoArray( LIB_TREE_NODE const& aNode, wxDataViewItemArray& aChildren );
270 
272 
274 
278  bool HasContainerColumns( wxDataViewItem const& aItem ) const override;
279 
283  bool IsContainer( wxDataViewItem const& aItem ) const override;
284 
290  wxDataViewItem GetParent( wxDataViewItem const& aItem ) const override;
291 
292  unsigned int GetColumnCount() const override { return NUM_COLS; }
293 
297  wxString GetColumnType( unsigned int aCol ) const override { return "string"; }
298 
306  void GetValue( wxVariant& aVariant,
307  wxDataViewItem const& aItem,
308  unsigned int aCol ) const override;
309 
314  bool SetValue( wxVariant const& aVariant,
315  wxDataViewItem const& aItem,
316  unsigned int aCol ) override { return false; }
317 
326  bool GetAttr( wxDataViewItem const& aItem,
327  unsigned int aCol,
328  wxDataViewItemAttr& aAttr ) const override;
329 
330 private:
335  int m_freeze;
336 
337  wxDataViewColumn* m_col_part;
338  wxDataViewColumn* m_col_desc;
339  wxDataViewCtrl* m_widget;
340 
342 
343  wxConfigBase* m_config;
344  wxString m_configPrefix;
345 
351  void FindAndExpand( LIB_TREE_NODE& aNode,
352  std::function<bool( LIB_TREE_NODE const* )> aFunc,
353  LIB_TREE_NODE** aHighScore );
354 
359 
364 
369 };
370 
371 #endif // LIB_TREE_MODEL_ADAPTER_H
372 
void DoAddLibrary(wxString const &aNodeName, wxString const &aDesc, std::vector< LIB_TREE_ITEM * > const &aItemList, bool presorted)
Add the given list of components by alias.
void AssignIntrinsicRanks()
Sort the tree and assign ranks after adding libraries.
bool IsContainer(wxDataViewItem const &aItem) const override
Check whether an item can have children.
CMP_FILTER_TYPE
This enum allows a selective filtering of components to list.
PTR_VECTOR Children
List of child nodes.
virtual int GetLibrariesCount() const
Return the number of libraries loaded in the tree.
int GetUnitFor(const wxDataViewItem &aSelection) const
Return the unit for the given item.
void SetPreselectNode(LIB_ID const &aLibId, int aUnit)
Set the component name to be selected if there are no search results.
void UpdateSearchString(wxString const &aSearch)
Set the search string provided by the user.
LIB_TREE_NODE * ShowResults()
Find and expand successful search results.
LIB_TREE_NODE::TYPE GetTypeFor(const wxDataViewItem &aSelection) const
Return node type for the given item.
LIB_TREE_NODE * ShowPreselect()
Find and expand preselected node.
unsigned int GetColumnCount() const override
wxString GetColumnType(unsigned int aCol) const override
Return the type of data stored in the column as indicated by wxVariant::GetType()
A logical library item identifier and consists of various portions much like a URI.
Definition: lib_id.h:51
virtual wxString GenerateInfo(LIB_ID const &aLibId, int aUnit)
static wxDataViewItem ToItem(LIB_TREE_NODE const *aNode)
Convert CMP_TREE_NODE -> wxDataViewItem.
static unsigned int IntoArray(LIB_TREE_NODE const &aNode, wxDataViewItemArray &aChildren)
Convert CMP_TREE_NODE's children to wxDataViewItemArray.
bool GetAttr(wxDataViewItem const &aItem, unsigned int aCol, wxDataViewItemAttr &aAttr) const override
Get any formatting for an item.
int GetItemCount() const
Return the number of components loaded in the tree.
bool HasContainerColumns(wxDataViewItem const &aItem) const override
Check whether a container has columns too.
void FindAndExpand(LIB_TREE_NODE &aNode, std::function< bool(LIB_TREE_NODE const *)> aFunc, LIB_TREE_NODE **aHighScore)
Find any results worth highlighting and expand them, according to given criteria (f(CMP_TREE_NODE con...
Node type: root.
void SetFilter(CMP_FILTER_TYPE aFilter)
Set the component filter type.
static LIB_TREE_NODE const * ToNode(wxDataViewItem aItem)
Convert wxDataViewItem -> CMP_TREE_NODE.
wxDataViewItem GetParent(wxDataViewItem const &aItem) const override
Get the parent of an item.
Model class in the component selector Model-View-Adapter (mediated MVC) architecture.
virtual TOOL_INTERACTIVE * GetContextMenuTool()
TREE_COLS
This enum defines the order of the columns in the tree view.
LIB_ID GetAliasFor(const wxDataViewItem &aSelection) const
Return the alias for the given item.
void AssignIntrinsicRanks(bool presorted=false)
Store intrinsic ranks on all children of this node.
void GetValue(wxVariant &aVariant, wxDataViewItem const &aItem, unsigned int aCol) const override
Get the value of an item.
bool SetValue(wxVariant const &aVariant, wxDataViewItem const &aItem, unsigned int aCol) override
Set the value of an item.
wxDataViewItem FindItem(const LIB_ID &aLibId)
Returns tree item corresponding to part.
wxObjectDataPtr< LIB_TREE_MODEL_ADAPTER > PTR
Reference-counting container for a pointer to CMP_TREE_MODEL_ADAPTER_BASE.
LIB_TREE_NODE * ShowSingleLibrary()
Find and expand a library if there is only one.
void SaveColWidths()
Save the column widths to the config file.
unsigned int GetChildren(wxDataViewItem const &aItem, wxDataViewItemArray &aChildren) const override
Populate a list of all the children of an item.
void AttachTo(wxDataViewCtrl *aDataViewCtrl)
Attach to a wxDataViewCtrl and initialize it.
CMP_FILTER_TYPE GetFilter() const
Return the active filter.
void ShowUnits(bool aShow)
Whether or not to show units.