/***************/ /* treeprj.cpp */ /***************/ #ifdef KICAD_PYTHON #include <pyhandler.h> #endif #include "fctsys.h" #include "gr_basic.h" #include "common.h" #include "kicad.h" #include "protos.h" #include "wx/image.h" #include "wx/imaglist.h" #include "wx/treectrl.h" #include "wx/regex.h" #include "wx/dir.h" #include "bitmaps.h" #include "bitmaps/icon_gerbview_small.xpm" #include "bitmaps/icon_cvpcb_small.xpm" #include "bitmaps/unknown.xpm" //#include "bitmaps/new_gerb.xpm" //#include "bitmaps/new_pcb.xpm" //#include "bitmaps/new_sch.xpm" //#include "bitmaps/new_cvpcb.xpm" #include "id.h" /********************************************/ /* Methodes pour l'arbre gestion du projet */ /********************************************/ IMPLEMENT_ABSTRACT_CLASS(WinEDA_TreePrj, wxTreeCtrl) WinEDA_TreePrj::WinEDA_TreePrj(WinEDA_PrjFrame * parent) : wxTreeCtrl(parent, ID_PROJECT_TREE, wxDefaultPosition, wxDefaultSize, wxTR_HAS_BUTTONS | wxTR_EDIT_LABELS, wxDefaultValidator, wxT("EDATreeCtrl")) { m_Parent = parent; // Make an image list containing small icons m_ImageList = new wxImageList(16, 16, TRUE, TREE_MAX); m_ImageList->Add(wxBitmap(kicad_icon_small_xpm)); // TREE_PROJECT m_ImageList->Add(wxBitmap(eeschema_xpm)); // TREE_SCHEMA m_ImageList->Add(wxBitmap(pcbnew_xpm)); // TREE_PCB m_ImageList->Add(wxBitmap(icon_python_small_xpm)); // TREE_PY m_ImageList->Add(wxBitmap(icon_gerbview_small_xpm)); // TREE_GERBER m_ImageList->Add(wxBitmap(datasheet_xpm)); // TREE_PDF m_ImageList->Add(wxBitmap(icon_txt_xpm)); // TREE_TXT m_ImageList->Add(wxBitmap(icon_cvpcb_small_xpm)); // TREE_NET m_ImageList->Add(wxBitmap(unknown_xpm)); // TREE_UNKNOWN m_ImageList->Add(wxBitmap(directory_xpm)); // TREE_DIRECTORY SetImageList(m_ImageList); } /***************************************************************************************/ int WinEDA_TreePrj::OnCompareItems(const wxTreeItemId& item1, const wxTreeItemId& item2) /***************************************************************************************/ /* sort function for tree items. items are sorted : directory names first by alphabetic order root file names after file names last by alphabetic order */ { TreePrjItemData* myitem1 = (TreePrjItemData*)GetItemData(item1); TreePrjItemData* myitem2 = (TreePrjItemData*)GetItemData(item2); if ( (myitem1->m_Type == TREE_DIRECTORY) && ( myitem2->m_Type != TREE_DIRECTORY ) ) return -1; if ( (myitem2->m_Type == TREE_DIRECTORY) && ( myitem1->m_Type != TREE_DIRECTORY ) ) return 1; if ( myitem1->m_IsRootFile && ! myitem2->m_IsRootFile ) return -1; if ( myitem2->m_IsRootFile && ! myitem1->m_IsRootFile ) return 1; return myitem1->m_FileName.CmpNoCase(myitem2->m_FileName) ; } /****************************************************************************************************/ TreePrjItemData::TreePrjItemData(enum TreeFileType type, const wxString & data, wxTreeCtrl * parent) : wxTreeItemData() /****************************************************************************************************/ { m_Type = type; m_Parent = parent; m_FileName = data; m_IsRootFile = false; } #ifdef KICAD_PYTHON using namespace boost::python; /**************************************/ object TreePrjItemData::GetIdPy() const /**************************************/ // Convert the data to an id { wxTreeItemId * id = new wxTreeItemId(); *id = GetId(); return object( handle<>( borrowed( wxPyConstructObject( id, wxT("wxTreeItemId"), true ) ) ) ); } #endif /*******************************************/ void TreePrjItemData::SetState( int state ) /*******************************************/ // Set the state used in the icon list { wxImageList* imglist = m_Parent->GetImageList(); if ( !imglist || state < 0 || state >= imglist->GetImageCount() / ( TREE_MAX - 2 ) ) return; m_State = state; int imgid = m_Type - 1 + state * ( TREE_MAX - 1 ); m_Parent->SetItemImage( GetId(), imgid ); m_Parent->SetItemImage( GetId(), imgid, wxTreeItemIcon_Selected ); } /*******************************************/ wxString TreePrjItemData::GetDir() const /*******************************************/ /* Get the directory containing the file */ { if (TREE_DIRECTORY == m_Type ) return m_FileName; wxFileName filename = wxFileName( m_FileName ); filename.MakeRelativeTo( wxGetCwd() ); wxArrayString dirs = filename.GetDirs(); wxString dir; for ( unsigned int i = 0; i < dirs.Count(); i ++ ) { dir += dirs[i] + filename.GetPathSeparator(); } return dir; } /****************************************************************/ void TreePrjItemData::OnRename( wxTreeEvent & event, bool check ) /****************************************************************/ /* Called upon tree item rename */ { if ( !Rename( event.GetLabel(), check ) ) event.Veto(); } /****************************************************/ void TreePrjItemData::Move( TreePrjItemData * dest ) /****************************************************/ // Move the object to dest { const wxString sep = wxFileName().GetPathSeparator(); if (!dest) return; if ( m_Parent != dest->m_Parent ) return;// Can not cross move! if ( dest == this ) return; // Can not move to ourself... wxTreeItemId parent = m_Parent->GetItemParent( GetId() ); if ( dest == dynamic_cast<TreePrjItemData*>( m_Parent->GetItemData( parent ) ) ) return; // same parent ? // We need to create a new item from us, and move // data to there ... // First move file on the disk wxFileName fname( m_FileName ); wxString destName; if ( !dest->GetDir().IsEmpty() ) destName = dest->GetDir() + sep; destName += fname.GetFullName(); if ( destName == GetFileName() ) return; // Same place ?? // Move the file on the disk: #if ( (wxMAJOR_VERSION < 2) || ((wxMAJOR_VERSION == 2)&& (wxMINOR_VERSION < 7)) ) if ( !wxRenameFile( GetFileName(), destName ) ) #else if ( !wxRenameFile( GetFileName(), destName, false ) ) #endif { wxMessageDialog( m_Parent, _( "Unable to move file ... "), _( "Permission error ?" ), wxICON_ERROR | wxOK ); return; } #ifdef KICAD_PYTHON object param = make_tuple( PyHandler::Convert( m_FileName ) , PyHandler::Convert( destName ) ); PyHandler::GetInstance()->TriggerEvent( wxT("kicad::MoveFile"), param ); #endif SetFileName( destName ); if ( TREE_DIRECTORY != GetType() ) { // Move the tree item itself now: wxTreeItemId oldId = GetId(); int i = m_Parent->GetItemImage( oldId ); wxString text = m_Parent->GetItemText( oldId ); // Bye bye old Id :'( wxTreeItemId newId = m_Parent->AppendItem( dest->GetId(), text, i ); m_Parent->SetItemData( newId, this ); m_Parent->SetItemData( oldId, NULL ); m_Parent->Delete( oldId ); } else { // We should move recursively all files, but that's quite boring // let's just refresh that's all ... TODO (change this to a better code ...) wxCommandEvent dummy; dynamic_cast<WinEDA_TreePrj*>(m_Parent)->GetParent()->m_Parent->OnRefresh(dummy); } } /****************************************************************/ bool TreePrjItemData::Rename( const wxString & name, bool check ) /****************************************************************/ /* rename the file checking if extension change occurs */ { if ( name.IsEmpty() ) return false; const wxString sep = wxFileName().GetPathSeparator(); wxString newFile; wxString dirs = GetDir(); if ( !dirs.IsEmpty() && GetType() != TREE_DIRECTORY ) newFile = dirs + sep + name; else newFile = name; if ( newFile == m_FileName ) return false; wxString ext = WinEDA_PrjFrame::GetFileExt( GetType() ); wxRegEx reg( wxT( "^.*\\" ) + ext + wxT( "$" ), wxRE_ICASE ); if ( check && !ext.IsEmpty() && !reg.Matches( newFile ) ) { wxMessageDialog dialog( m_Parent , _("Changing file extension will change file type.\n Do you want to continue ?") , _("Rename File") , wxYES_NO | wxICON_QUESTION ); if ( wxID_YES != dialog.ShowModal() ) return false; } #if ( (wxMAJOR_VERSION < 2) || ((wxMAJOR_VERSION == 2)&& (wxMINOR_VERSION < 7)) ) if (! wxRenameFile( m_FileName, newFile ) ) #else if (! wxRenameFile( m_FileName, newFile, false ) ) #endif { wxMessageDialog( m_Parent, _( "Unable to rename file ... "), _( "Permission error ?" ), wxICON_ERROR | wxOK ); return false; } SetFileName( newFile ); #ifdef KICAD_PYTHON object param = make_tuple( PyHandler::Convert( m_FileName ) , PyHandler::Convert( newFile ) ); PyHandler::GetInstance()->TriggerEvent( wxT("kicad::RenameFile"), param ); #endif return true; } /*******************************************/ bool TreePrjItemData::Delete( bool check ) /*******************************************/ /* delete a file */ { wxMessageDialog dialog( m_Parent, _("Do you really want to delete ") + GetFileName(), _("Delete File") , wxYES_NO | wxICON_QUESTION ); if ( !check || wxID_YES == dialog.ShowModal() ) { if ( !wxDirExists( m_FileName ) ) { wxRemoveFile( m_FileName ); } else { wxArrayString filelist; wxDir::GetAllFiles( m_FileName, &filelist ); for ( unsigned int i = 0; i < filelist.Count(); i++ ) wxRemoveFile( filelist[i] ); wxRmdir( m_FileName ); } m_Parent->Delete( GetId() ); #ifdef KICAD_PYTHON PyHandler::GetInstance()->TriggerEvent( wxT("kicad::DeleteFile"), PyHandler::Convert( m_FileName ) ); #endif return true; } return false; } /**********************************/ void TreePrjItemData::Activate() /**********************************/ /* Called under item activation */ { wxString FullFileName = wxGetCwd() + wxFileName().GetPathSeparator() + GetFileName(); switch ( GetType() ) { case TREE_PROJECT: break; case TREE_DIRECTORY: m_Parent->Toggle( GetId() ); break; case TREE_SCHEMA: { AddDelimiterString( FullFileName ); ExecuteFile(m_Parent, EESCHEMA_EXE, FullFileName); break; } case TREE_PCB: AddDelimiterString( FullFileName ); ExecuteFile(m_Parent, PCBNEW_EXE, FullFileName); break; #ifdef KICAD_PYTHON case TREE_PY: PyHandler::GetInstance()->RunScript( FullFileName ); break; #endif case TREE_GERBER: AddDelimiterString(FullFileName); ExecuteFile(m_Parent, GERBVIEW_EXE, FullFileName); break; case TREE_PDF: OpenPDF( FullFileName ); break; case TREE_NET: AddDelimiterString(FullFileName); ExecuteFile(m_Parent, CVPCB_EXE, FullFileName); break; case TREE_TXT: { wxString editorname = GetEditorName(); if ( !editorname.IsEmpty() ) ExecuteFile(m_Parent, editorname, FullFileName); break; } default: OpenFile( FullFileName ); break; } } /***************************************************/ TreePrjItemData * WinEDA_PrjFrame::GetSelectedData() /***************************************************/ { return dynamic_cast<TreePrjItemData*>( m_TreeProject->GetItemData(m_TreeProject->GetSelection()) ); }