qBittorrent/src/misc.h

/*
 * Bittorrent Client using Qt4 and libtorrent.
 * Copyright (C) 2006  Christophe Dumez
 *
 * 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, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 *
 * Contact : chris@qbittorrent.org
 */

#ifndef MISC_H
#define MISC_H

#include <sstream>
#include <stdexcept>
#include <QObject>
#include <QString>
#include <QDir>
#include <QList>
#include <QPair>
#include <QThread>

#define MAX_CHAR_TMP 128

/*  Miscellaneaous functions that can be useful */
class misc : public QObject{
 Q_OBJECT

  public:
    // Convert any type of variable to C++ String
    // convert=true will convert -1 to 0
    template <class T> static std::string toString(const T& x, bool convert=false){
      std::ostringstream o;
      if(!(o<<x)) {
        throw std::runtime_error("::toString()");
      }
      if(o.str() == "-1" && convert)
        return "0";
      return o.str();
    }

    // Convert C++ string to any type of variable
    template <class T> static T fromString(const std::string& s){
      T x;
      std::istringstream i(s);
      if(!(i>>x)){
        throw std::runtime_error("::fromString()");
      }
      return x;
    }

    // return best userfriendly storage unit (B, KiB, MiB, GiB, TiB)
    // use Binary prefix standards from IEC 60027-2
    // see http://en.wikipedia.org/wiki/Kilobyte
    // value must be given in bytes
    template <class T> static QString friendlyUnit(const T& value){
      char tmp[MAX_CHAR_TMP];
      float val = value;
      if(val <= 0){
        return QString(tr("Unknown", "Unknown (size)"));
      }
      QString units[] = {tr("B", "bytes"), tr("KiB", "kibibytes (1024 bytes)"), tr("MiB", "mebibytes (1024 kibibytes)"), tr("GiB", "gibibytes (1024 mibibytes)")};
      for(int i=0; i<5; ++i){
        if (val < 1024.){
          snprintf(tmp, MAX_CHAR_TMP, "%.1f", (float)val);
          return QString(tmp) + " " + units[i];
        }
        val /= 1024.;
      }
      snprintf(tmp, MAX_CHAR_TMP, "%.1f", (float)val);
      return  QString(tmp) + " " + tr("TiB", "tebibytes (1024 gibibytes)");
    }

    // return qBittorrent config path
    static QString qBittorrentPath() {
      QString qBtPath = QDir::homePath();
      if(qBtPath.isNull()){
        return QString();
      }
      if(qBtPath[qBtPath.length()-1] == QDir::separator()){
        qBtPath = qBtPath + ".qbittorrent" + QDir::separator();
      }else{
        qBtPath = qBtPath + QDir::separator() + ".qbittorrent" + QDir::separator();
      }
      // Create dir if it does not exist
      QDir dir(qBtPath);
      if(!dir.exists()){
        dir.mkpath(qBtPath);
      }
      return qBtPath;
    }

    static bool removePath(QString path){
      if(!QFile::remove(path)){
        // Probably a folder
        QDir current_dir(path);
        if(current_dir.exists()){
          //Remove sub items
          QStringList subItems = current_dir.entryList();
          QString item;
          foreach(item, subItems){
            if(item != "." && item != ".."){
              qDebug("-> Removing "+(path+QDir::separator()+item).toUtf8());
              removePath(path+QDir::separator()+item);
            }
          }
          // Remove empty folder
          if(current_dir.rmpath(path)){
            return true;
          }else{
            return false;
          }
        }else{
          return false;
        }
      }
      return true;
    }

    // Function called by curl to write the data to the file
    static int my_fwrite(void *buffer, size_t size, size_t nmemb, void *stream){
      return fwrite(buffer, size, nmemb, (FILE*)stream);
    }

    static QString findFileInDir(const QString& dir_path, const QString& fileName){
      QDir dir(dir_path);
      if(dir.exists(fileName)){
        return dir.filePath(fileName);
      }
      QStringList subDirs = dir.entryList(QDir::Dirs);
      QString subdir_name;
      foreach(subdir_name, subDirs){
        QString result = findFileInDir(dir.path()+QDir::separator()+subdir_name, fileName);
        if(!result.isNull()){
          return result;
        }
      }
      return QString();
    }

    // Insertion sort, used instead of bubble sort because it is
    // approx. 5 times faster.
    template <class T> static void insertSort(QList<QPair<int, T> > &list, const QPair<int, T>& value, Qt::SortOrder sortOrder){
      int i = 0;
      if(sortOrder == Qt::AscendingOrder){
        while(i < list.size() and value.second > list.at(i).second){
          ++i;
        }
      }else{
        while(i < list.size() and value.second < list.at(i).second){
          ++i;
        }
      }
      list.insert(i, value);
    }

    template <class T> static void insertSort2(QList<QPair<int, T> > &list, const QPair<int, T>& value, Qt::SortOrder sortOrder){
      int i = 0;
      if(sortOrder == Qt::AscendingOrder){
        while(i < list.size() and value.first > list.at(i).first){
          ++i;
        }
      }else{
        while(i < list.size() and value.first < list.at(i).first){
          ++i;
        }
      }
      list.insert(i, value);
    }

    // Can't use template class for QString because >,< use unicode code for sorting
    // which is not what a human would expect when sorting strings.
    static void insertSortString(QList<QPair<int, QString> > &list, QPair<int, QString> value, Qt::SortOrder sortOrder){
      int i = 0;
      if(sortOrder == Qt::AscendingOrder){
        while(i < list.size() and QString::localeAwareCompare(value.second, list.at(i).second) > 0){
          ++i;
        }
      }else{
        while(i < list.size() and QString::localeAwareCompare(value.second, list.at(i).second) < 0){
          ++i;
        }
      }
      list.insert(i, value);
    }

    // Take a number of seconds and return an user-friendly
    // time duration like "1d 2h 10m".
    static QString userFriendlyDuration(const long int seconds){
      if(seconds <= 0){
        return QString::QString(tr("Unknown"));
      }
      if(seconds < 60){
        return QString::QString("< 1"+tr("m", "minutes"));
      }
      int minutes = seconds / 60;
      if(minutes < 60){
        return QString::QString(misc::toString(minutes).c_str())+tr("m", "minutes");
      }
      int hours = minutes / 60;
      minutes = minutes - hours*60;
      if(hours < 24){
        return QString::QString(misc::toString(hours).c_str())+tr("h", "hours")+" "+QString::QString(misc::toString(minutes).c_str())+tr("m", "minutes");
      }
      int days = hours / 24;
      hours = hours - days * 24;
      if(days < 100){
        return QString::QString(misc::toString(days).c_str())+tr("d", "days")+" "+QString::QString(misc::toString(hours).c_str())+tr("h ", "hours")+QString::QString(misc::toString(minutes).c_str())+tr("m", "minutes");
      }
      return QString::QString(tr("Unknown"));
    }
};

//  Trick to get a portable sleep() function
class SleeperThread : public QThread{
  public:
    static void msleep(unsigned long msecs)
    {
      QThread::msleep(msecs);
    }
};

#endif