// Filename: "maincode.cpp"

#include "globals.h"
#include "cLVQ1.h"

int Pattern[T_PATTERNS][VEC_LEN] = {{0, 0, 1, 1, 0, 0, 0,
                                     0, 0, 0, 1, 0, 0, 0,
                                     0, 0, 0, 1, 0, 0, 0,
                                     0, 0, 1, 0, 1, 0, 0,
                                     0, 0, 1, 0, 1, 0, 0,
                                     0, 1, 1, 1, 1, 1, 0,
                                     0, 1, 0, 0, 0, 1, 0,
                                     0, 1, 0, 0, 0, 1, 0,
                                     1, 1, 1, 0, 1, 1, 1},

                                    {1, 1, 1, 1, 1, 1, 0,
                                     0, 1, 0, 0, 0, 0, 1,
                                     0, 1, 0, 0, 0, 0, 1,
                                     0, 1, 0, 0, 0, 0, 1,
                                     0, 1, 1, 1, 1, 1, 0,
                                     0, 1, 0, 0, 0, 0, 1,
                                     0, 1, 0, 0, 0, 0, 1,
                                     0, 1, 0, 0, 0, 0, 1,
                                     1, 1, 1, 1, 1, 1, 0},

                                    {0, 0, 1, 1, 1, 1, 1,
                                     0, 1, 0, 0, 0, 0, 1,
                                     1, 0, 0, 0, 0, 0, 0,
                                     1, 0, 0, 0, 0, 0, 0,
                                     1, 0, 0, 0, 0, 0, 0,
                                     1, 0, 0, 0, 0, 0, 0,
                                     1, 0, 0, 0, 0, 0, 0,
                                     0, 1, 0, 0, 0, 0, 1,
                                     0, 0, 1, 1, 1, 1, 0},

                                    {1, 1, 1, 1, 1, 0, 0,
                                     0, 1, 0, 0, 0, 1, 0,
                                     0, 1, 0, 0, 0, 0, 1,
                                     0, 1, 0, 0, 0, 0, 1,
                                     0, 1, 0, 0, 0, 0, 1,
                                     0, 1, 0, 0, 0, 0, 1,
                                     0, 1, 0, 0, 0, 0, 1,
                                     0, 1, 0, 0, 0, 1, 0,
                                     1, 1, 1, 1, 1, 0, 0},

                                    {1, 1, 1, 1, 1, 1, 1,
                                     0, 1, 0, 0, 0, 0, 1,
                                     0, 1, 0, 0, 0, 0, 0,
                                     0, 1, 0, 1, 0, 0, 0,
                                     0, 1, 1, 1, 0, 0, 0,
                                     0, 1, 0, 1, 0, 0, 0,
                                     0, 1, 0, 0, 0, 0, 0,
                                     0, 1, 0, 0, 0, 0, 1,
                                     1, 1, 1, 1, 1, 1, 1},

                                    {0, 0, 0, 1, 1, 1, 1,
                                     0, 0, 0, 0, 0, 1, 0,
                                     0, 0, 0, 0, 0, 1, 0,
                                     0, 0, 0, 0, 0, 1, 0,
                                     0, 0, 0, 0, 0, 1, 0,
                                     0, 0, 0, 0, 0, 1, 0,
                                     0, 1, 0, 0, 0, 1, 0,
                                     0, 1, 0, 0, 0, 1, 0,
                                     0, 0, 1, 1, 1, 0, 0},

                                    {1, 1, 1, 0, 0, 1, 1,
                                     0, 1, 0, 0, 1, 0, 0,
                                     0, 1, 0, 1, 0, 0, 0,
                                     0, 1, 1, 0, 0, 0, 0,
                                     0, 1, 1, 0, 0, 0, 0,
                                     0, 1, 0, 1, 0, 0, 0,
                                     0, 1, 0, 0, 1, 0, 0,
                                     0, 1, 0, 0, 0, 1, 0,
                                     1, 1, 1, 0, 0, 1, 1},

                                    {0, 0, 0, 1, 0, 0, 0,
                                     0, 0, 0, 1, 0, 0, 0,
                                     0, 0, 0, 1, 0, 0, 0,
                                     0, 0, 1, 0, 1, 0, 0,
                                     0, 0, 1, 0, 1, 0, 0,
                                     0, 1, 0, 0, 0, 1, 0,
                                     0, 1, 1, 1, 1, 1, 0,
                                     0, 1, 0, 0, 0, 1, 0,
                                     0, 1, 0, 0, 0, 1, 0},

                                    {1, 1, 1, 1, 1, 1, 0,
                                     1, 0, 0, 0, 0, 0, 1,
                                     1, 0, 0, 0, 0, 0, 1,
                                     1, 0, 0, 0, 0, 0, 1,
                                     1, 1, 1, 1, 1, 1, 0,
                                     1, 0, 0, 0, 0, 0, 1,
                                     1, 0, 0, 0, 0, 0, 1,
                                     1, 0, 0, 0, 0, 0, 1,
                                     1, 1, 1, 1, 1, 1, 0},

                                    {0, 0, 1, 1, 1, 0, 0,
                                     0, 1, 0, 0, 0, 1, 0,
                                     1, 0, 0, 0, 0, 0, 1,
                                     1, 0, 0, 0, 0, 0, 0,
                                     1, 0, 0, 0, 0, 0, 0,
                                     1, 0, 0, 0, 0, 0, 0,
                                     1, 0, 0, 0, 0, 0, 1,
                                     0, 1, 0, 0, 0, 1, 0,
                                     0, 0, 1, 1, 1, 0, 0},

                                    {1, 1, 1, 1, 1, 0, 0,
                                     1, 0, 0, 0, 0, 1, 0,
                                     1, 0, 0, 0, 0, 0, 1,
                                     1, 0, 0, 0, 0, 0, 1,
                                     1, 0, 0, 0, 0, 0, 1,
                                     1, 0, 0, 0, 0, 0, 1,
                                     1, 0, 0, 0, 0, 0, 1,
                                     1, 0, 0, 0, 0, 1, 0,
                                     1, 1, 1, 1, 1, 0, 0},
                                    
                                    {1, 1, 1, 1, 1, 1, 1,
                                     1, 0, 0, 0, 0, 0, 0,
                                     1, 0, 0, 0, 0, 0, 0,
                                     1, 0, 0, 0, 0, 0, 0,
                                     1, 1, 1, 1, 1, 0, 0,
                                     1, 0, 0, 0, 0, 0, 0,
                                     1, 0, 0, 0, 0, 0, 0,
                                     1, 0, 0, 0, 0, 0, 0,
                                     1, 1, 1, 1, 1, 1, 1},
                                    
                                    {0, 0, 0, 0, 0, 1, 0,
                                     0, 0, 0, 0, 0, 1, 0,
                                     0, 0, 0, 0, 0, 1, 0,
                                     0, 0, 0, 0, 0, 1, 0,
                                     0, 0, 0, 0, 0, 1, 0,
                                     0, 0, 0, 0, 0, 1, 0,
                                     0, 1, 0, 0, 0, 1, 0,
                                     0, 1, 0, 0, 0, 1, 0,
                                     0, 0, 1, 1, 1, 0, 0},
                                    
                                    {1, 0, 0, 0, 0, 1, 0,
                                     1, 0, 0, 0, 1, 0, 0,
                                     1, 0, 0, 1, 0, 0, 0,
                                     1, 0, 1, 0, 0, 0, 0,
                                     1, 1, 0, 0, 0, 0, 0,
                                     1, 0, 1, 0, 0, 0, 0,
                                     1, 0, 0, 1, 0, 0, 0,
                                     1, 0, 0, 0, 1, 0, 0,
                                     1, 0, 0, 0, 0, 1, 0},
                                    
                                    {0, 0, 0, 1, 0, 0, 0,
                                     0, 0, 0, 1, 0, 0, 0,
                                     0, 0, 1, 0, 1, 0, 0,
                                     0, 0, 1, 0, 1, 0, 0,
                                     0, 1, 0, 0, 0, 1, 0,
                                     0, 1, 1, 1, 1, 1, 0,
                                     1, 0, 0, 0, 0, 0, 1,
                                     1, 0, 0, 0, 0, 0, 1,
                                     1, 1, 0, 0, 0, 1, 1},
                                    
                                    {1, 1, 1, 1, 1, 1, 0,
                                     0, 1, 0, 0, 0, 0, 1,
                                     0, 1, 0, 0, 0, 0, 1,
                                     0, 1, 1, 1, 1, 1, 0,
                                     0, 1, 0, 0, 0, 0, 1,
                                     0, 1, 0, 0, 0, 0, 1,
                                     0, 1, 0, 0, 0, 0, 1,
                                     0, 1, 0, 0, 0, 0, 1,
                                     1, 1, 1, 1, 1, 1, 0},
                                    
                                    {0, 0, 1, 1, 1, 0, 1,
                                     0, 1, 0, 0, 0, 1, 1,
                                     1, 0, 0, 0, 0, 0, 1,
                                     1, 0, 0, 0, 0, 0, 0,
                                     1, 0, 0, 0, 0, 0, 0,
                                     1, 0, 0, 0, 0, 0, 0,
                                     1, 0, 0, 0, 0, 0, 1,
                                     0, 1, 0, 0, 0, 1, 0,
                                     0, 0, 1, 1, 1, 0, 0},
                                    
                                    {1, 1, 1, 1, 0, 0, 0,
                                     0, 1, 0, 0, 1, 0, 0,
                                     0, 1, 0, 0, 0, 1, 0,
                                     0, 1, 0, 0, 0, 1, 0,
                                     0, 1, 0, 0, 0, 1, 0,
                                     0, 1, 0, 0, 0, 1, 0,
                                     0, 1, 0, 0, 0, 1, 0,
                                     0, 1, 0, 0, 1, 0, 0,
                                     1, 1, 1, 1, 0, 0, 0},
                                    
                                    {1, 1, 1, 1, 1, 1, 1,
                                     0, 1, 0, 0, 0, 0, 1,
                                     0, 1, 0, 0, 1, 0, 0,
                                     0, 1, 1, 1, 1, 0, 0,
                                     0, 1, 0, 0, 1, 0, 0,
                                     0, 1, 0, 0, 0, 0, 0,
                                     0, 1, 0, 0, 0, 0, 0,
                                     0, 1, 0, 0, 0, 0, 1,
                                     1, 1, 1, 1, 1, 1, 1},
                                    
                                    {0, 0, 0, 0, 1, 1, 1,
                                     0, 0, 0, 0, 0, 1, 0,
                                     0, 0, 0, 0, 0, 1, 0,
                                     0, 0, 0, 0, 0, 1, 0,
                                     0, 0, 0, 0, 0, 1, 0,
                                     0, 0, 0, 0, 0, 1, 0,
                                     0, 0, 0, 0, 0, 1, 0,
                                     0, 1, 0, 0, 0, 1, 0,
                                     0, 0, 1, 1, 1, 0, 0},
                                    
                                    {1, 1, 1, 0, 0, 1, 1,
                                     0, 1, 0, 0, 0, 1, 0,
                                     0, 1, 0, 0, 1, 0, 0,
                                     0, 1, 0, 1, 0, 0, 0,
                                     0, 1, 1, 0, 0, 0, 0,
                                     0, 1, 0, 1, 0, 0, 0,
                                     0, 1, 0, 0, 1, 0, 0,
                                     0, 1, 0, 0, 0, 1, 0,
                                     1, 1, 1, 0, 0, 1, 1}};

char mFontNames[][3] = {"A1", "B1", "C1", "D1", "E1", "J1", "K1", 
                        "A2", "B2", "C2", "D2", "E2", "J2", "K2", 
                        "A3", "B3", "C3", "D3", "E3", "J3", "K3"};

int target[T_PATTERNS] = {0, 1, 2, 3, 4, 5, 6,
                          0, 1, 2, 3, 4, 5, 6,
                          0, 1, 2, 3, 4, 5, 6};

int main()
{
    cLVQ1 LVQ(Pattern, target);

    for(int i = 0; i < N_CLUSTERS; i++)
    {
        LVQ.initializeWeights(i, Pattern[i]);
        cout << "Weights for cluster " << i << " initialized to pattern " 
             << mFontNames[i] << endl;
    } // i

    LVQ.trainNetwork();

    for(int i = 0; i < T_PATTERNS; i++)
    {
        cout << "Pattern " << mFontNames[i] << " belongs to cluster " 
             << LVQ.getCluster(Pattern[i]) << endl;
    } // i

    return 0;

}