Управление внешними ресурсами из MEX-функций

MEX-функции, которые соединяются с внешними ресурсами, такими как сокеты, датчики, файлы и базы данных, должны управлять этими ресурсами. Поскольку вы реализуете функции C++ MEX как классы, можно задать функции конструктора и деструктора, чтобы управлять этими ресурсами и переменными, которые сохраняются при повторных вызовах MEX-функции.

Когда вы вызываете MEX-функцию, MATLAB® создает образец MexFunction класс. Этот объект сохраняется для сеанса работы с MATLAB или до тех пор, пока вы не очистите объект с clear mex команда. Повторные вызовы MEX-функции могут обрабатывать входящие данные и освобождать ресурсы по завершении.

Чтение текстового файла

Эта MEX-функция открывает текстовый файл и читает по одному слову каждый раз, когда вы вызываете функцию. The MexFunction класс, который реализует MEX-функцию, задает конструктор и деструктор, чтобы открыть и закрыть файл. Каждое прочитанное слово сохранено в std::unordered_map для определения количества раз, когда слово встречается в файле.

Конструктор

The MexFunction конструктор выполняет следующие шаги:

Деструктор

Деструктор классов закрывает файл.

Оператор вызова функции ()

Каждый вызов MEX-функции читает слово из текстового файла и добавляет его к неупорядоченной карте или просто увеличивает количество слов для этого слова, если оно существует на карте. Текущее слово и его количество в командном окне MATLAB отображается в MEX-функцию с помощью std::ostringstream выходной поток. Чтобы разблокировать MEX-функцию, передайте аргумент (такой как 'unlock') к функции.

Отображение на MATLAB

The displayOnMATLAB member function использует matlab:: engine:: MATLABEngine:: feval, чтобы вызвать MATLABfprintf функция со строкой, записанной в выходной поток.

Описание кода

#include "mex.hpp"
#include "mexAdapter.hpp"
#include <unordered_map>
#include <fstream> 

using matlab::mex::ArgumentList;
using namespace matlab::data;

class MexFunction : public matlab::mex::Function {

    // Input stream to read words from file
    std::ifstream inFile;

    // Unordered map to keep track of word count
    std::unordered_map<std::string, int> wordCount;

    // Pointer to MATLAB engine
    std::shared_ptr<matlab::engine::MATLABEngine> matlabPtr = getEngine();

    // Factory to create MATLAB data arrays
    ArrayFactory factory;

public:
    MexFunction() {
        mexLock();  
        matlabPtr->eval(u"fname = fullfile(matlabroot, 'examples', 'matlab', 'sonnets.txt');");
        matlab::data::CharArray fileName = matlabPtr->getVariable(u"fname");
        inFile.open(fileName.toAscii());
        if (!inFile.is_open()) {
            std::ostringstream stream;
            stream << "Failed to open sonnets.txt" << std::endl;
            displayOnMATLAB(stream);
        }
    }

    ~MexFunction() {
        if (inFile.is_open())
            inFile.close();
    }

    void operator()(ArgumentList outputs, ArgumentList inputs) {
        if (inFile.is_open() && !inFile.eof()) {
            std::string word;
            inFile >> word;
            wordCount[word]++;
            std::ostringstream stream;
            stream << "Read : " << "\"" << word << "\"" 
                << ", current count: " << wordCount[word] << std::endl;
            displayOnMATLAB(stream);
        }
        if (!inputs.empty() || !inFile.is_open()) {
            mexUnlock();
        }
    }

    void displayOnMATLAB(const std::ostringstream& stream){
        matlabPtr->feval(u"fprintf", 0, 
            std::vector<Array>({ factory.createScalar(stream.str()) }));
    }
};

Создайте и запустите sonnetWordCount.cpp

Откройте файл исходного кода, sonnetWordCount.cpp, в редакторе и использовать mex команда для компиляции MEX-функции.

mex sonnetWordCount.cpp

Неоднократно вызывайте MEX-функцию, чтобы подсчитать использование слова.

>> sonnetWordCount
Read : "THE", current count: 1
>> sonnetWordCount
Read : "SONNETS", current count: 1
>> sonnetWordCount
Read : "by", current count: 1
>> sonnetWordCount
Read : "William", current count: 1
>> sonnetWordCount
Read : "Shakespeare", current count: 1
>> sonnetWordCount('unlock')

См. также

Похожие темы