MSDefs.cpp

/*
 *  This file is part of the AiBO+ project
 *
 *  Copyright (C) 2005-2016 Csaba Kertész (csaba.kertesz@gmail.com)
 *
 *  AiBO+ 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.
 *
 *  AiBO+ 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, write to the Free Software
 *  Foundation, Inc., 59 Temple Street #330, Boston, MA 02111-1307, USA.
 *
 */

#include "MSDefs.hpp"

#include <qdir.h>
#include <qfile.h>
#include <QtGlobal>

#include <MCDefs.hpp>
#include <MCLog.hpp>

static QString BasePath;

void MSSetBasePath(const QString& path)
{
  if (path.isEmpty())
  {
    MC_WARNING("Empty path is not a valid base path.");
    return;
  }
  BasePath = path;
  MCCreatePath(path);
}


QString MSGetBasePath()
{
  if (BasePath.isEmpty())
  {
    MSSetBasePath(QDir::homePath());
  }
  return BasePath;
}


QString MSGetMlBasePath()
{
  return MSGetBasePath()+MC_DIR_SEPARATOR+"ml"+MC_DIR_SEPARATOR;
}


QString MSGetMlModelsBasePath()
{
  return MSGetMlBasePath()+"models"+MC_DIR_SEPARATOR;
}


QString MSGetMlSamplesBasePath()
{
  return MSGetMlBasePath()+"samples"+MC_DIR_SEPARATOR;
}


int MSDirectorySize(const QString& directory)
{
  if (directory.isEmpty())
    return 0;

  int Size = 0;
  QFileInfo FileInfo(directory);

  if (FileInfo.isDir())
  {
    QDir Dir(directory);
    QFileInfoList FileIntoList = Dir.entryInfoList(QDir::Files | QDir::Dirs |  QDir::Hidden |
                                                   QDir::NoSymLinks | QDir::NoDotAndDotDot);

    for (int i = 0; i < FileIntoList.size(); ++i)
    {
      QFileInfo FileInfo = FileIntoList.at(i);

      if (FileInfo.isDir())
        Size += MSDirectorySize(FileInfo.absoluteFilePath());
      else
        Size += FileInfo.size();
    }
  }
  return Size;
}


QStringList MSGetSubDirectories(const QString& directory)
{
  if (directory.isEmpty())
    return QStringList();

  QFileInfo FileInfo(directory);
  QStringList Dirs;

  if (!FileInfo.isDir())
    return Dirs;

  QDir Dir(directory);
  QFileInfoList DirList = Dir.entryInfoList(QDir::Dirs |  QDir::Hidden |
                                            QDir::NoSymLinks | QDir::NoDotAndDotDot);

  Dirs.push_back(FileInfo.absoluteFilePath());
  for (int i = 0; i < DirList.size(); ++i)
  {
    QFileInfo DirInfo = DirList.at(i);

    Dirs += MSGetSubDirectories(DirInfo.absoluteFilePath()+MC_DIR_SEPARATOR);
  }
  return Dirs;
}


QStringList MSGetSubDirectoryContents(const QString& directory)
{
  if (directory.isEmpty())
    return QStringList();

  QFileInfo FileInfo(directory);
  QStringList Files;

  if (!FileInfo.isDir())
    return Files;

  QDir Dir(directory);
  QFileInfoList FileIntoList = Dir.entryInfoList(QDir::Files | QDir::Dirs |  QDir::Hidden |
                                                 QDir::NoSymLinks | QDir::NoDotAndDotDot);

  for (int i = 0; i < FileIntoList.size(); ++i)
  {
    QFileInfo FileInfo = FileIntoList.at(i);

    if (FileInfo.isDir())
      Files += MSGetSubDirectoryContents(FileInfo.absoluteFilePath());
    else
      Files.push_back(FileInfo.absoluteFilePath());
  }
  return Files;
}