1 files changed, 10 insertions, 1 deletions
diff --git a/src/core/PassphraseGenerator.h b/src/core/PassphraseGenerator.h
index a4e0e426b..5868faab4 100644
--- a/src/core/PassphraseGenerator.h
+++ b/src/core/PassphraseGenerator.h
@@ -28,9 +28,17 @@ public:
     PassphraseGenerator();
     Q_DISABLE_COPY(PassphraseGenerator)
 
-    double calculateEntropy(const QString& passphrase);
+    enum PassphraseWordCase
+    {
+        LOWERCASE,
+        UPPERCASE,
+        TITLECASE
+    };
+
+    double estimateEntropy(int wordCount = 0);
     void setWordCount(int wordCount);
     void setWordList(const QString& path);
+    void setWordCase(PassphraseWordCase wordCase);
     void setDefaultWordList();
     void setWordSeparator(const QString& separator);
     bool isValid() const;
@@ -43,6 +51,7 @@ public:
 
 private:
     int m_wordCount;
+    PassphraseWordCase m_wordCase;
     QString m_separator;
     QVector<QString> m_wordlist;
 };