|
#include <chrono>
|
|
#include <iomanip>
|
|
#include <iostream>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "AIO_ExecDriver.h"
|
|
#include "ExecDriver.h"
|
|
#include "Utilities/BufferAllocator.h"
|
|
#include "Utilities/FileUtils.h"
|
|
#include "Utilities/SearchLynxConfiguration.h"
|
|
|
|
std::vector<std::string> UNITS{"b/sec", "KB/s", "MB/s", "GB/s", "TB/s"};
|
|
|
|
double printRate(double rawRate, uint32_t& unitIdx)
|
|
{
|
|
uint32_t idx = 0;
|
|
double rate = rawRate;
|
|
while (rate > 1024)
|
|
{
|
|
rate /= 1024;
|
|
++idx;
|
|
}
|
|
|
|
unitIdx = idx;
|
|
return rate;
|
|
}
|
|
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
std::cout << "Read Test Utility\n" << std::endl;
|
|
|
|
if (argc < 2)
|
|
{
|
|
std::cerr << "ERROR: please provide an input file" << std::endl;
|
|
return -1;
|
|
}
|
|
|
|
// Start time measurement
|
|
auto start = std::chrono::system_clock::now();
|
|
|
|
// Prime the config object
|
|
auto slConfig = BlackLynx::SearchLynx::SearchLynxConfiguration::getInstance();
|
|
|
|
// Allocate buffers
|
|
auto bufAllocator = BlackLynx::SearchLynx::BufferAllocator::getInstance();
|
|
bufAllocator->initAllocation();
|
|
|
|
// Read in the file
|
|
std::string filename(argv[1]);
|
|
// BlackLynx::SearchLynx::ExecDriver driver(filename, 22);
|
|
BlackLynx::SearchLynx::AIO_ExecDriver driver(filename, 3);
|
|
driver.start();
|
|
driver.waitUntilComplete();
|
|
|
|
// Deallocate buffers
|
|
bufAllocator->deallocateBuffers();
|
|
|
|
// End time measurement
|
|
auto end = std::chrono::system_clock::now();
|
|
auto diff = end - start;
|
|
double duration = std::chrono::duration <double>(diff).count() ;
|
|
std::cout << "Duration: " << std::fixed << std::setprecision(3)
|
|
<< duration << " ms" << std::endl;
|
|
|
|
uint64_t fileSize = BlackLynx::SearchLynx::FileUtils::getFileSize(filename);
|
|
|
|
double rawRate = fileSize / duration;
|
|
std::cout << "Raw rate: " << rawRate << " b/sec" << std::endl;
|
|
|
|
uint32_t idx = 0;
|
|
double rate = printRate(rawRate, idx);
|
|
std::cout << "Rate: " << std::fixed << std::setprecision(3) << rate
|
|
<< " " << UNITS[idx] << std::endl;
|
|
|
|
return 0;
|
|
}
|