TextureMapping/TextureMapper.h at master · AngYu/TextureMapping · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
#ifndef TEXTUREMAPPER_H
#define TEXTUREMAPPER_H

struct float3 {float x, y, z;};

class TextureMapper {

public:
    TextureMapper(std::string plyFilename, std::vector<cv::Mat> source, std::vector<cv::Mat> TcwPoses, int patchSize = 7);

private:
    int patchSize;
    std::vector<float3> vertices;
    std::vector<cv::Mat> TcwPoses;
    std::vector<cv::Mat> source;
    std::vector<cv::Mat> target;
    std::vector<cv::Mat> texture;

    void init();
    void read_ply_file(const std::string & filepath);
    void align(std::vector<cv::Mat> source, std::vector<cv::Mat> target);
    void reconstruct();
    int Mixi();
    cv::Mat patchSearch(std::vector<cv::Mat> source, std::vector<cv::Mat> target, int iterations);
    void vote(cv::Mat completenessPatchMatches, cv::Mat coherencePatchMatches);
    std::vector<std::vector<std::vector<int>>> findSourcePatches(cv::Mat completenessPatchMatches, cv::Mat coherencePatchMatches,
                                                                    int x, int y, int t);
    bool isInTargetPatch(cv::Vec<float, 4> targetMatch, int x, int y, int t);
    int Tixi(std::vector<std::vector<int>> completenessPatches, std::vector<std::vector<int>> coherencePatches, int c);
    float distance(int sx, int sy, int st,
                    int tx, int ty, int tt,
                    int patchSize, float threshold);
    int randomInt(int min, int max);
    std::vector<cv::Mat> getRGBD(std::vector<cv::Mat> target, std::vector<cv::Mat> TcwPoses, PlyModel model);
    bool projectToSurface();
};

#endif