QtDirectory.cpp

Go to the documentation of this file.

/**************************************************************************************/
/*                                                                                    */
/*  Visualization Library                                                             */
/*  http://visualizationlibrary.org                                                   */
/*                                                                                    */
/*  Copyright (c) 2005-2020, Michele Bosi                                             */
/*  All rights reserved.                                                              */
/*                                                                                    */
/*  Redistribution and use in source and binary forms, with or without modification,  */
/*  are permitted provided that the following conditions are met:                     */
/*                                                                                    */
/*  - Redistributions of source code must retain the above copyright notice, this     */
/*  list of conditions and the following disclaimer.                                  */
/*                                                                                    */
/*  - Redistributions in binary form must reproduce the above copyright notice, this  */
/*  list of conditions and the following disclaimer in the documentation and/or       */
/*  other materials provided with the distribution.                                   */
/*                                                                                    */
/*  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND   */
/*  ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED     */
/*  WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE            */
/*  DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR  */
/*  ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES    */
/*  (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;      */
/*  LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON    */
/*  ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT           */
/*  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS     */
/*  SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.                      */
/*                                                                                    */
/**************************************************************************************/

#include <vlQt5/QtDirectory.hpp>
#include <vlQt5/QtFile.hpp>
#include <algorithm>
#include <QDir>

using namespace vl;

//-----------------------------------------------------------------------------
// QtDirectory
//-----------------------------------------------------------------------------
QtDirectory::QtDirectory( const String& name )
{
  setPath(name);
}
//-----------------------------------------------------------------------------
QtDirectory::QtDirectory()
{
}
//-----------------------------------------------------------------------------
void QtDirectory::listFilesRecursive(std::vector<String>& file_list) const
{
  file_list.clear();
  listFilesRecursive_internal(file_list);
}
//-----------------------------------------------------------------------------
ref<QtDirectory> QtDirectory::qtSubDir(const String& subdir_name) const
{
  if (path().empty())
  {
    Log::error( "QtDirectory::path() must not be empty!\n" );
    return NULL;
  }

  std::vector<String> dir_list;
  String p = translatePath(subdir_name).right(-path().length()-1);
  this->listSubDirs(dir_list);
  String cur_p = path();
  for(int i=0; i<(int)dir_list.size(); ++i)
  {
    dir_list[i] = dir_list[i].right(-cur_p.length()-1);
    if (p.startsWith(dir_list[i]+'/') || p == dir_list[i])
    {
      ref<QtDirectory> dir = new QtDirectory(cur_p + '/' + dir_list[i]);
      if (!dir)
        return NULL;
      cur_p = cur_p + '/' + dir_list[i];
      p = p.right(-dir_list[i].length()-1);
      dir->listSubDirs(dir_list);
      i=-1;
      if (p.empty())
        return dir;
    }
  }
  return NULL;
}
//-----------------------------------------------------------------------------
bool QtDirectory::exists() const
{
  if ( path().empty() )
  {
    Log::error( "QtDirectory::path() must not be empty!\n" );
    return false;
  }
  return QDir( QString( path().toStdString().c_str() ) ).exists();
}
//-----------------------------------------------------------------------------
void QtDirectory::listSubDirs(std::vector<String>& dirs_out, bool append) const
{
  if ( ! append ) {
    dirs_out.clear();
  }

  if ( path().empty() )
  {
    Log::error( "QtDirectory::path() must not be empty!\n" );
    return;
  }

  const char* p = path().toStdString().c_str();
  QStringList subdirs = QDir(QString(p)).entryList( QDir::Dirs | QDir::NoDotAndDotDot );

  for (int i = 0; i < subdirs.size(); ++i) {
    vl::String name = vl::String::fromStdString( subdirs.at(i).toStdString() );
    dirs_out.push_back( path() + name + '/' );
  }
}
//-----------------------------------------------------------------------------
void QtDirectory::listFiles(std::vector< ref<QtFile> >& file_list, bool append) const
{
  if (!append)
    file_list.clear();
  std::vector<String> file_names;
  listFiles(file_names,false);
  for(unsigned i=0; i<file_names.size(); ++i)
  {
    ref<QtFile> file = new QtFile;
    // file->setPath( path() + file_names[i] );
    file->setPath( file_names[i] );
    file_list.push_back(file);
  }
}
//-----------------------------------------------------------------------------
void QtDirectory::listFiles(std::vector<String>& files_out, bool append) const
{
  if ( ! append ) {
    files_out.clear();
  }

  if ( path().empty() )
  {
    Log::error( "QtDirectory::path() must not be empty!\n" );
    return;
  }

  const char* p = path().toStdString().c_str();
  QStringList files = QDir(QString(p)).entryList( QDir::Files );

  for (int i = 0; i < files.size(); ++i) {
    vl::String name = vl::String::fromStdString( files.at(i).toStdString() );
    files_out.push_back( path() + name + '/' );
  }
}
//-----------------------------------------------------------------------------
ref<VirtualFile> QtDirectory::file(const String& name) const
{
  return qtFile(name);
}
//-----------------------------------------------------------------------------
ref<QtFile> QtDirectory::qtFile(const String& name) const
{
  String p = translatePath(name);
  ref<QtFile> file = new QtFile(p);
  if (file->exists())
    return file;
  else
    return NULL;
}
//-----------------------------------------------------------------------------
void QtDirectory::listFilesRecursive_internal(std::vector<String>& file_list) const
{
  // add local child
  listFiles(file_list, true);
  // descend recursively
  std::vector<String> dir_list;
  listSubDirs(dir_list);
  for(unsigned i=0; i<dir_list.size(); ++i)
  {
    VL_CHECK(dir_list[i] != ".")
    VL_CHECK(dir_list[i] != "..")

    QtDirectory sub_dir( dir_list[i] );
    sub_dir.listFilesRecursive_internal(file_list);
  }
}
//-----------------------------------------------------------------------------