//
//  NameRecognizer.h
//  CardRecognizer
//
//  Created by Vladimir Tchernitski on 08/02/16.
//  Copyright © 2016 Vladimir Tchernitski. All rights reserved.
//

#ifndef NameRecognizer_h
#define NameRecognizer_h

#include "INameRecognizer.h"
#include <fstream>
#include <string>
#include <sstream>

class INeuralNetworkResultList;
class IServiceContainer;
class INeuralNetwork;
class INeuralNetworkObjectFactory;
class IRecognitionCoreDelegate;

class CNameRecognizer : public INameRecognizer
{
    
public:
    CNameRecognizer(const shared_ptr<IServiceContainer>& container);
    
    virtual ~CNameRecognizer();
    
public:
    
    virtual shared_ptr<INeuralNetworkResultList> Process(cv::Mat& matrix,
                                                         vector<cv::Mat>& digitsSamples, std::string& postprocessedName);
    
    virtual bool Deploy();
    
    virtual void SetRecognitionMode(PayCardsRecognizerMode flag);
    
    virtual void SetDelegate(const shared_ptr<IRecognitionCoreDelegate>& delegate);
    
    virtual void SetPathNameYLocalizationViola(const string& path);
    
    virtual void SetPathNameLocalizationXModel(const string& path);
    virtual void SetPathNameLocalizationXStruct(const string& path);
    
    virtual void SetPathNameSpaceCharModel(const string& path);
    virtual void SetPathNameSpaceCharStruct(const string& path);
    
    virtual void SetPathNameListTxt(const string& path);
    
private:
    
    void Predict(const vector<cv::Mat>& matrixes, shared_ptr<INeuralNetworkResultList>& neuralNetworkResultList, const shared_ptr<INeuralNetwork>& neuralNetwork);
    int FilterCoords(const std::vector<float>& xCoords, float modelDist, std::pair<int, int>& result);
    int LevenshteinDistance(const string& src, const string& dst);
private:
    
    weak_ptr<IServiceContainer> _container;
    weak_ptr<IRecognitionCoreDelegate> _delegate;
    weak_ptr<INeuralNetworkObjectFactory> _factory;
    
    string _pathNameYLocalizationViola;
    
    cv::CascadeClassifier _yCascade;
    
    shared_ptr<INeuralNetwork> _localizationXNeuralNetwork;
    shared_ptr<INeuralNetwork> _spaceCharNeuralNetwork;
    
    string _pathNameLocalizationXModel;
    string _pathNameLocalizationXStruct;
    
    string _pathNameSpaceCharModel;
    string _pathNameSpaceCharStruct;
    string _pathNameDictPath;
    std::istringstream _namesDict;
    
    PayCardsRecognizerMode _mode;
};


#endif /* NameRecognizer_h */