SentimentLabels.cpp

#include "SentimentLabels.h"
#include <iostream>
#include <fstream>
#include<ctype.h>
#include <cstdio>
#include <string.h>

using namespace std;

/*
 * Maps phrase indexes with their sentiment score.
 */

SentimentLabels::SentimentLabels() {
    ifstream input("stanfordSentimentTreebank/sentiment_labels.txt");
    string line;
    bool character = false;
    bool first_line = true;
    long long first_number = 0;
    double second_number = 0;
    string word = "";
    while(getline( input, line ) ) {
         character = false;
         first_number = 0;
         second_number = 0;
         word.clear();
         if (!first_line) {
            for(char & c : line) {
                if (c == '|') {
                    character = true;
                } else {
                    if (!character) first_number = first_number*10 + c -'0';
                    else {
                        char aux = tolower(c);
                        word +=aux;
                    }
                }
            }
            second_number = atof(word.c_str());
            sentimentLabelsMap.insert(make_pair(first_number, second_number));
         }
         first_line = false;
    }
}

double SentimentLabels::getSentimentScore(long long phraseIndex) {
    unordered_map<long long, double>::const_iterator found_iter = sentimentLabelsMap.find(phraseIndex);
    if (found_iter == sentimentLabelsMap.end()) {
        cout<<"The given index: "<<phraseIndex<<" was not found as an index in the sentiment labels map."<<endl;
        return -1;
        }
    return found_iter->second;
}