#include <cstdlib>
#include <ctime>

#include "encrypt.h"

// 印出參數
void PrintInt(int a) {
    cout << a << endl;
}

// 定義巨集
#define PrintInt(a) cout << a << endl

// 從標準程式庫中引入 string
#include <string>

// 使用 std 中的 string 名稱
using std::string;

// 定義兩個巨集常數
#define DIFF 97
#define N 26

// 宣告 Encrypt 類別
class Encrypt {
public:
    // 宣告建構函數
    Encrypt();
    // 宣告 setter 與 getter 成員函數
    void set_code_array();
    void set_code_array(string);
    string get_code_array();
    // 宣告編碼、解碼的成員函數
    string ToEncode(string);
    string ToDecode(string);

private:
    // 密碼表字串
    string code_array;
};

/* 《程式語言教學誌》的範例程式
   http://kaiching.org/
   檔名：encrypt.h
   功能：Encrypt 類別的標頭檔
   作者：張凱慶 */

// 進行解碼工作的成員函數
string Encrypt::ToDecode(string s) {
    // 暫存解碼結果的字串
    string r;
    int i, j;
    // 第一層迴圈逐一取得每一個字元
    for (i = 0; i < s.size(); i++) {
        // 判斷該字元是否為英文小寫字母，若是英文小寫字母就進行解碼轉換
        if (s.at(i) >= DIFF && s.at(i) < DIFF + N) {
            // 第二層迴圈尋找該字元在密碼表中的索引值
            // 該索引值加上 DIFF 就可轉換回原本的字元
            for (j = 0; j < N; j++) {
                if (s.at(i) == get_code_array().at(j)) {
                    r += (char) j + DIFF;
                    break;
                }
            }
        }
        else {
            r += s.at(i);
        }
    }

    // 結束回傳解碼過的字串
    return r;
}