ImageDescriber_CCTAG.hpp

// This file is part of the AliceVision project.
// Copyright (c) 2016 AliceVision contributors.
// Copyright (c) 2012 openMVG contributors.
// This Source Code Form is subject to the terms of the Mozilla Public License,
// v. 2.0. If a copy of the MPL was not distributed with this file,
// You can obtain one at https://mozilla.org/MPL/2.0/.
 
#pragma once
 
#include <aliceVision/feature/imageDescriberCommon.hpp>
#include <aliceVision/feature/ImageDescriber.hpp>
#include <aliceVision/feature/regionsFactory.hpp>
#include <aliceVision/types.hpp>
 
#include <iostream>
#include <numeric>
 
namespace cctag {
struct Parameters;  // Hidden implementation
}
 
namespace aliceVision {
namespace feature {
 
class ImageDescriber_CCTAG : public ImageDescriber
{
  public:
    explicit ImageDescriber_CCTAG(const std::size_t nRings = 3);
 
    ~ImageDescriber_CCTAG() override = default;
 
    bool useCuda() const override;
 
    bool useFloatImage() const override { return false; }
 
    EImageDescriberType getDescriberType() const override;
 
    std::size_t getMemoryConsumption(std::size_t width, std::size_t height) const override { return 3 * width * height * sizeof(unsigned char); }
 
    void setUseCuda(bool) override;
 
    void setCudaPipe(int pipe) override { _cudaPipe = pipe; }
 
    void setConfigurationPreset(ConfigurationPreset preset) override;
 
    bool describe(const image::Image<unsigned char>& image,
                  std::unique_ptr<Regions>& regions,
                  const image::Image<unsigned char>* mask = nullptr) override;
 
    void allocate(std::unique_ptr<Regions>& regions) const override;
 
    struct CCTagParameters
    {
        explicit CCTagParameters(size_t nRings);
 
        ~CCTagParameters() = default;
 
        bool setPreset(EImageDescriberPreset preset);
 
        float _cannyThrLow;
        float _cannyThrHigh;
        std::unique_ptr<cctag::Parameters> _internalParams;
    };
 
  private:
    // CCTag parameters
    CCTagParameters _params;
    bool _doAppend = false;
    int _cudaPipe = 0;
};
 
template<class DescriptorT>
IndexT getCCTagId(const DescriptorT& desc)
{
    std::size_t cctagId = UndefinedIndexT;
    for (std::size_t i = 0; i < desc.size(); ++i)
    {
        if (desc.getData()[i] == (unsigned char)255)
        {
            if (cctagId != UndefinedIndexT)
            {
                return UndefinedIndexT;
            }
            cctagId = i;
        }
        else if (desc.getData()[i] != (unsigned char)0)
        {
            return UndefinedIndexT;
        }
    }
    return cctagId;
}
 
}  // namespace feature
}  // namespace aliceVision