MEX-функции, которые соединяются с внешними ресурсами, такими как сокеты, датчики, файлы и базы данных, должны управлять этими ресурсами. Поскольку вы реализуете функции C++ MEX как классы, можно задать функции конструктора и деструктора, чтобы управлять этими ресурсами и переменными, которые сохраняются при повторных вызовах MEX-функции.
Когда вы вызываете MEX-функцию, MATLAB® создает образец MexFunction
класс. Этот объект сохраняется для сеанса работы с MATLAB или до тех пор, пока вы не очистите объект с clear
mex
команда. Повторные вызовы MEX-функции могут обрабатывать входящие данные и освобождать ресурсы по завершении.
Эта MEX-функция открывает текстовый файл и читает по одному слову каждый раз, когда вы вызываете функцию. The MexFunction
класс, который реализует MEX-функцию, задает конструктор и деструктор, чтобы открыть и закрыть файл. Каждое прочитанное слово сохранено в std::unordered_map
для определения количества раз, когда слово встречается в файле.
The MexFunction
конструктор выполняет следующие шаги:
Функции mexLock
для предотвращения удаления MEX-функции из памяти.
Полный путь к текстовому файлу из MATLAB получите с помощью matlab:: engine:: MATLABEngine:: eval и matlab:: engine:: MATLABEngine:: getVariable.
Откройте текстовый файл в std::ifstream
входной поток.
Деструктор классов закрывает файл.
Каждый вызов MEX-функции читает слово из текстового файла и добавляет его к неупорядоченной карте или просто увеличивает количество слов для этого слова, если оно существует на карте. Текущее слово и его количество в командном окне MATLAB отображается в MEX-функцию с помощью std::ostringstream
выходной поток. Чтобы разблокировать MEX-функцию, передайте аргумент (такой как 'unlock'
) к функции.
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
, в редакторе и использовать 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')