main.cpp

#include <iostream>
#include <string>
#include <fstream>
#include <vector>
#include <unordered_map>
#include <getopt.h>
#include <cstring>
#include <numeric>
#include <cmath>
#include "./constants.h"
#include "./utilities.h"
#include "./Key.h"
#include "Hash_table.h"
#include "./data_point.h"

using namespace std;

int main(int argc, char** argv) {
    string input="", query="", output="";
    string func_name = "";
    bool read_query;
    int c,R=400;
    int k=-1;
    int L=-1;

    /* Reading Arguments from command line */
    while(( c = getopt(argc,argv,"d:q:k:L:o:f:h:R")) !=-1 ){
        switch (c) {
            case 'd':
                input = optarg;
                break;
            case 'q':
                query = optarg;
                break;
            case 'k':
                k = atoi(optarg);
                break;
            case 'L':
                L = atoi(optarg);
                break;
            case 'o':
                output = optarg;
                break;
            case 'R':
                R = atoi(optarg);
                break;
            case 'f':
                func_name = optarg;
                if ((func_name != "cosine") || (func_name!="euclidean")) {
                    cerr << "Invalid function name" << endl;
                }
                break;
            case 'h':
                cout << "\nlsh - Locality Sensitive Hashing\n" << endl <<
                     "Mandatory options\n\t-d data feed file\n\n"
                     "Optional flags\n"
                     "\t-q query data file\n"
                     "\t-k num of random probes\n"
                     "\t-L num of hash tables\n"
                     "\t-o output file\n"
                     "\t-f function to use (euclidean and cosine similarity available)" << endl;
                return 0;
            case '?':
                cerr << "Unknown option character " << char(optopt) << endl;
                return 1;
            default:
                abort();
        }
    }
    if (input.size()==0) {
        cerr << "Data file was not defined!" << endl ;
    }
    if (k==-1) k=const_lsh::k;
    if (L==-1) L=const_lsh::L;
    if (query.size()==0) read_query=true;
    if (output.size()==0) output = "./output.txt";
    if (func_name.size()==0) func_name = const_lsh::def_func;

    cout << "L: " << L << endl;
    cout << "k: " << k << endl;
    cout << "in: " << input  << endl;
    cout << "qr: " << query  << endl;
    cout << "ou: " << output << endl;
    cout << "fn: " << func_name << endl;

    ifstream inputfd;
    string line;
    int num_lines = 0;
    int d;
    inputfd.open (input);
    if (inputfd.is_open()){
        getline(inputfd,line);
        d = num_columns(line);
        num_lines++;
        while ( getline(inputfd,line) ){
            num_lines++;
        }
        inputfd.close();
    }else{
        cerr << "Unable to open file: " << input << endl;
    }

    cout << "d: " << d << endl;
    cout << "num lines: " << num_lines << endl;

    Hash_table ht = Hash_table(num_lines/const_lsh::table_size, d, k, func_name);
    /*vector <Hash_table> tables ;
    for (int i=0; i<L ; i++){
        tables.push_back(Hash_table(num_lines/const_lsh::table_size, d, k, func_name));
    }*/
    ht.print_stats();
    inputfd.open (input);
    vector<int> xx;
    char cc[1024] ;
    string name;
    for (int i=0 ; i< num_lines; i++) {
        getline(inputfd,line);
        strncpy(cc,line.c_str(),d*sizeof(int));
        char *pch = strtok (cc ," \t");
        while (pch != NULL)
        {
            xx.push_back(atoi(pch));
            pch = strtok (NULL, " \t");
        }
        ht.add_item(name,{xx,0},num_lines/const_lsh::table_size);
        xx.clear();
    }
    inputfd.close();

    //ht.print_stats();


    ifstream queryfd;
    queryfd.open(query);
    vector<data_point> neighboors,temp;
    while (getline(queryfd,line)) {
        strncpy(cc,line.c_str(),d*sizeof(int));
        char *pch = strtok (cc ," \t");
        while (pch != NULL)
        {
            xx.push_back(atoi(pch));
            pch = strtok (NULL, " \t");
        }
        temp = ht.get_bucket({xx,0});
        cout << "size: " << temp.size();
        neighboors.insert(neighboors.end(),temp.begin(),temp.end());
        for(int z=0;z<temp.size();z++){
            cout << "size: " << temp.size();
            for(int x=0;x<temp.size();x++){
                cout << temp[z].k.dim[x] << " ";
                //cout << neighboors[z].k.dim[x] << " ";
            }
            cout << endl << endl;
        }
        double result;
        for (int z=0;z<neighboors.size();z++){
            result = std::sqrt(std::inner_product(begin(xx), end(xx), begin(neighboors[z].k.dim), 0));
            cout <<"result: "<< result;
        }
        xx.clear();
    }

    //unordered_map<Key,string> hash_tb;
    /*Key a ,b , c1, d1;
    a.dim = { 1 , 2 , 3};
    hash_tb[a]="item1";
    b.dim = { 1 , 2 , 4};
    hash_tb[b]="item2";
    c1.dim = { 1 , 2 , 5};
    hash_tb[c1]="item3";
    d1.dim = { 1 , 2 , 6};
    hash_tb[d1]="item4";*/



    return 0;

}