Scorer.h

#ifndef SCORER_H
#define SCORER_H

#include <iostream>
#include <vector>
#include <string>
#include <sstream>
#include <unordered_map>
#include "Buffer.h"

class Scorer {
protected:
    std::unordered_map<std::string, int> dictionary;

public:
    Scorer(const std::unordered_map<std::string, int>& dict) : dictionary(dict) {}
    virtual ~Scorer() {}
    virtual double calculateScore(const Buffer& buffer) = 0;
};

class WordCountScorer : public Scorer {
public:
    using Scorer::Scorer;
    double calculateScore(const Buffer& buffer) override {
        if (!buffer.getData()) return 0.0;
        double score = 0;
        std::stringstream ss(buffer.getData());
        std::string token;
        while (ss >> token) {
            if (dictionary.find(token) != dictionary.end()) {
                score += (dictionary.at(token) > 0) ? 1.0 : -1.0;
            }
        }
        return score;
    }
};

class WeightedScorer : public Scorer {
public:
    using Scorer::Scorer;
    double calculateScore(const Buffer& buffer) override {
        if (!buffer.getData()) return 0.0;
        double totalScore = 0;
        std::stringstream ss(buffer.getData());
        std::string token;
        while (ss >> token) {
            if (dictionary.find(token) != dictionary.end()) {
                totalScore += dictionary.at(token);
            }
        }
        return totalScore;
    }
};

#endif