//***************************************************************************
//* Copyright (c) 2015 Saint Petersburg State University
//* Copyright (c) 2011-2014 Saint Petersburg Academic University
//* All Rights Reserved
//* See file LICENSE for details.
//***************************************************************************

#include "analyses.h"
#include <iostream>
#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <sys/mman.h>

void Analyses::init() {
    std::string key;

    struct stat sb;
    off_t len;
    char *p;
    int fd;

    fd = open (m_filename, O_RDONLY);
    if (fd == -1) {
        perror ("Error: could not open");
        return;
    }

    if (fstat (fd, &sb) == -1) {
        perror ("Error: have a fstat problem");
        return;
    }

    if (!S_ISREG (sb.st_mode)) {
        fprintf (stderr, "%s is not a file\n", m_filename);
        return;
    }

    p = (char *)mmap (0, sb.st_size, PROT_READ, MAP_SHARED, fd, 0);
    if (p == MAP_FAILED) {
        perror ("mmap");
        return;
    }

    if (close (fd) == -1) {
        perror ("close");
        return;
    }

    if (!m_isAlg) {
        for (len = 0; len < sb.st_size; len++) {
            if (p[len] == '\n') {
                continue;
            }
            key += p[len];
            if (key.length() == m_mer) {
                if (m_data.count(key) > 0) {
                    m_data[key] += 1;
                } else {
                    m_data[key] = 1;
                }
                key = key.substr(1);
            }
        }
        createDatFile();

    }
    else {
        m_algorithm[m_algname](p, sb.st_size, m_mer, 0.0, 0.0);
    }


    if (munmap (p, sb.st_size) == -1) {
        perror ("munmap");
        return;
    }
    paint();
}

void Analyses::initFastTq() {
    std::string key;

    struct stat sb;
    off_t len;
    char *p;
    int fd;

    fd = open (m_filename, O_RDONLY);
    if (fd == -1) {
        perror ("Error: could not open");
        return;
    }

    if (fstat (fd, &sb) == -1) {
        perror ("Error: have a fstat problem");
        return;
    }

    if (!S_ISREG (sb.st_mode)) {
        fprintf (stderr, "%s is not a file\n", m_filename);
        return;
    }

    p = (char *)mmap (0, sb.st_size, PROT_READ, MAP_SHARED, fd, 0);
    if (p == MAP_FAILED) {
        perror ("mmap");
        return;
    }

    if (close (fd) == -1) {
        perror ("close");
        return;
    }

    if (!m_isAlg) {
        int count = 1;
        for (len = 0; len < sb.st_size; len++) {
            if (p[len] == '\n') {
                continue;
            }
            if (count == 2) {
                key += p[len];
                if (key.length() == m_mer) {
                    if (m_data.count(key) > 0) {
                        m_data[key] += 1;
                    } else {
                        m_data[key] = 1;
                    }
                    key = key.substr(1);
                }
            }
            if (count <= 3) {
                ++count;
            } else {
                count = 1;
            }
        }
        createDatFile();

    } else {
        m_algorithm[m_algname](p, sb.st_size, m_mer, 0.0, 0.0);
    }

    if (munmap (p, sb.st_size) == -1) {
        perror ("munmap");
        return;
    }
    paint();
}

void Analyses::createDatFile() {
    std::ofstream out("daily.dat");
    if(!out) {
        std::cout << "Cannot open file.\n";
        return;
    }
    std::map <int, int> dt;

    for (std::map<std::string,int>::iterator it=m_data.begin() ; it != m_data.end(); ++it) {
        if (dt.find((*it).second) == dt.end()) {
            dt[(*it).second] = 1;
        } else {
            ++dt[(*it).second];
        }
    }

    for (std::map<int,int>::iterator it = dt.begin() ; it != dt.end(); ++it) {
        out << (*it).first << " " << (*it).second << "\n";
    }

    return;
}