cmd/src/YDWorkThread.h

Go to the documentation of this file.

#ifndef YDWORKTHREAD_H
#define YDWORKTHREAD_H

#include <QString>
#include <QThread>
#include <QWaitCondition>
#include <QList>
#include <QStringList>
#include <QTextStream>

#include "YDConfigFile.h"

#include <windows.h>
#include <stdio.h>
#include "JkDefragLib.h"

#define YD_MODE_MIN             0
#define YD_MODE_MAX             10
#define YD_MODE_DEFAULT         2
#define YD_MODE_INVALID         -1

#define YD_SPEED_MIN            1
#define YD_SPEED_MAX            100
#define YD_SPEED_DEFAULT        100
#define YD_SPEED_INVALID        -1

#define YD_FREESPACE_MIN        1
#define YD_FREESPACE_MAX        100
#define YD_FREESPACE_DEFAULT    1
#define YD_FREESPACE_INVALID    -1

#define YD_STARTFLAG_DEFAULT    true

class YDWorkThread: public QThread {

        Q_OBJECT

public:
        YDWorkThread(QString name, QObject *parent = NULL);
        ~YDWorkThread();

        QString diskState() const { return m_diskState; }
        QString diskName() const { return m_configFile.m_diskName; }

        void setExcludes(QStringList _list) { m_configFile.m_excludes = _list; }
        QStringList excludes() const { return m_configFile.m_excludes; }
        static QStringList defaultExcludes();

        void setSpaceHogs(QStringList _list) { m_configFile.m_spaceHogs = _list; }
        QStringList spaceHogs() const { return m_configFile.m_spaceHogs; }
        static QStringList defaultSpaceHogs();

        void setVerbose(int verbose) { m_verbose = verbose; }
        int verbose() const { return m_verbose; }

        void setMode(int mode) { m_configFile.m_mode = mode; }
        int mode() const { return m_configFile.m_mode; }

        void setSpeed(int speed) { m_configFile.m_speed = speed; }
        int speed() const { return m_configFile.m_speed; }

        void setFreeSpace(int freeSpace) { m_configFile.m_freeSpace = freeSpace; }
        int freeSpace() const { return m_configFile.m_freeSpace; }

        void setStartFlag(int startFlag) { m_configFile.m_startFlag = startFlag; }
        int startFlag() const { return m_configFile.m_startFlag; }

        void setLogStream(QTextStream *logStream) { m_logStream = logStream; }
        QTextStream *logStream() const { return m_logStream; }

        bool loadConfig(QString fileName = QString::null) { return m_configFile.load(fileName); }
        bool saveConfig(QString fileName = QString::null, bool hidden = true) { return m_configFile.save(fileName, hidden); }

public slots:
        void stop();

signals:
        void finished(QThread *thread);

protected:
        QString m_diskState;
        YDConfigFile m_configFile;
        int m_running;
        int m_redrawScreen;
        int m_verbose;
        QTextStream *m_logStream;
        
        void run();

private:
        virtual void showStatus(struct DefragDataStruct *data);
        virtual void showMove(struct ItemStruct *item, ULONG64 clusters, ULONG64 fromLcn, ULONG64 toLcn, ULONG64 fromVcn);
        virtual void showAnalyze(struct DefragDataStruct *data, struct ItemStruct *item);
        virtual void showDebug(int level, struct ItemStruct *item, WCHAR *message, va_list varArgs);
        virtual void drawCluster(struct DefragDataStruct *data, ULONG64 clusterStart, ULONG64 clusterEnd, int color);
        virtual void clearScreen(WCHAR *format, va_list varArgs);

        friend void __cdecl YDShowStatus(struct DefragDataStruct *data);
        friend void __cdecl YDShowMove(struct ItemStruct *item, ULONG64 clusters, ULONG64 fromLcn, ULONG64 toLcn, ULONG64 fromVcn);
        friend void __cdecl YDShowAnalyze(struct DefragDataStruct *data, struct ItemStruct *item);
        friend void __cdecl YDShowDebug(int level, struct ItemStruct *item, WCHAR *message, ...);
        friend void __cdecl YDDrawCluster(struct DefragDataStruct *data, ULONG64 clusterStart, ULONG64 clusterEnd, int color);
        friend void __cdecl YDClearScreen(WCHAR *format, ...);
};

typedef QList<YDWorkThread*> YDWorkThreadList;

#endif/*YDWORKTHREAD_H*/