52 lines
1.4 KiB
C++
52 lines
1.4 KiB
C++
#pragma once
|
|
|
|
#include <iostream>
|
|
#include <bitset>
|
|
#include <climits>
|
|
#include <set>
|
|
#include <boost/filesystem.hpp>
|
|
#include <boost/system/error_code.hpp>
|
|
#include <boost/filesystem/fstream.hpp>
|
|
|
|
#include <Magick++.h>
|
|
#include <string>
|
|
#include "Descriptor.h"
|
|
#include "Mutation.h"
|
|
|
|
using namespace std;
|
|
using namespace Magick;
|
|
|
|
//ENUM_STRING(descriptorss, (totalColors)(averageColor)(numDescriptors))
|
|
|
|
namespace fs = boost::filesystem;
|
|
|
|
//Manager class for
|
|
class DescriptorManager {
|
|
public:
|
|
DescriptorManager();
|
|
DescriptorManager(const fs::path & image_path);
|
|
~DescriptorManager();
|
|
void Init();
|
|
void rankImages(string image_name, int method);
|
|
multiset<pair<unsigned long long, string>> rankMutations(string image_name, int method);
|
|
void runExperiments();
|
|
void outputMap(int a);
|
|
void outputMap();
|
|
void outputRank(string imagename, int method);
|
|
private:
|
|
static const int numDescriptors = 8;
|
|
static const int numMutations = 7;
|
|
//TODO: automagically update this?
|
|
|
|
bool initialized = false;
|
|
int numImages;
|
|
|
|
map<string, bitset<64>> features[numDescriptors]; //array of maps that will hold the descriptors
|
|
map<string, bitset<64>> mutatedFeatures[numDescriptors][numMutations];
|
|
multiset<pair<unsigned long long, string> > distances; //multiset that stores the distances
|
|
fs::path img_path; //where the images are
|
|
Descriptor* descriptor_types[numDescriptors];
|
|
Mutation* mutators[numMutations];
|
|
};
|
|
|