AtpGenerator.h

Go to the documentation of this file.

//**************************************************************************************************
//
//     OSSIM Open Source Geospatial Data Processing Library
//     See top level LICENSE.txt file for license information
//
//**************************************************************************************************
#ifndef AtpGeneratorBase_H_
#define AtpGeneratorBase_H_

#include <ossim/base/ossimFilename.h>
#include <ossim/base/ossimFilename.h>
#include <ossim/base/ossimRefPtr.h>
#include <ossim/imaging/ossimImageChain.h>
#include <ossim/imaging/ossimImageGeometry.h>
#include <ossim/imaging/ossimAnnotationSource.h>
#include <ossim/projection/ossimImageViewProjectionTransform.h>
#include <ossim/reg/Image.h>
#include "AtpTileSource.h"
#include "AutoTiePoint.h"
#include "AtpAnnotatedImage.h"
#include <vector>
#include <memory>

namespace ATP
{

class AtpGenerator : public std::enable_shared_from_this<AtpGenerator>
{
   friend class AtpTileSource;

public:
   enum Algorithm { ALGO_UNASSIGNED=0, CROSSCORR, DESCRIPTOR, NASA };

   AtpGenerator(Algorithm algo);

   virtual ~AtpGenerator();

   void setRefImage(std::shared_ptr<ossim::Image> ref_image);
   void setCmpImage(std::shared_ptr<ossim::Image> cmp_image);

   virtual void initialize();

   void setBandWeights(const std::vector<double> weights) { m_bandWeights = weights; }

   virtual bool generateTiePointList(ossim::TiePointList& tpList);

//   virtual bool renderTiePointDataAsImages(const ossimFilename& outBaseName,
//                                           const ossimFilename& lutFile=ossimFilename("")) = 0;

   static void writeTiePointList(ostream& out, const AtpList& tpList);

   ossimRefPtr<ossimImageViewProjectionTransform> getRefIVT() { return m_refIVT; }
   ossimRefPtr<ossimImageViewProjectionTransform> getCmpIVT() { return m_cmpIVT; }
   ossimRefPtr<ossimImageChain> getRefChain() { return m_refChain; }
   ossimRefPtr<ossimImageChain> getCmpChain() { return m_cmpChain; }

   std::string getRefImageID();
   std::string getCmpImageID();
   std::string getRefFilename();
   std::string getCmpFilename();

   ossimRefPtr<AtpTileSource> getAtpTileSource() { return m_atpTileSource; } // For engineering use
   ossimRefPtr<AtpAnnotatedImage> m_annotatedRefImage; // For engineering use
   ossimRefPtr<AtpAnnotatedImage> m_annotatedCmpImage; // For engineering use

protected:
   AtpGenerator() : m_algorithm (ALGO_UNASSIGNED), m_refEllipHgt(0) {}

   virtual ossimRefPtr<ossimImageChain>
   constructChain(std::shared_ptr<ossim::Image> image,
                  ossimRefPtr<ossimImageViewProjectionTransform>& ivt);

   //  @return true if successful
   bool getValidVertices(ossimRefPtr<ossimImageChain> chain,
                         ossimRefPtr<ossimImageViewProjectionTransform>& ivt,
                         std::vector<ossimDpt>& validVertices);

   void layoutSearchTileRects(ossimPolygon& overlapPoly);

   ossimImageHandler* getImageHandler(ossimRefPtr<ossimImageChain>& chain);

   Algorithm m_algorithm;
   std::shared_ptr<ossim::Image> m_refImage;
   std::shared_ptr<ossim::Image> m_cmpImage;
   ossimRefPtr<AtpTileSource> m_atpTileSource;
   ossimRefPtr<ossimImageChain> m_refChain;
   ossimRefPtr<ossimImageChain> m_cmpChain;
   ossimRefPtr<ossimImageViewProjectionTransform> m_refIVT;
   ossimRefPtr<ossimImageViewProjectionTransform> m_cmpIVT;
   ossimRefPtr<ossimImageGeometry> m_viewGeom;
   ossimIrect m_aoiView;
   std::vector<double> m_bandWeights;
   double m_refEllipHgt;
   std::vector<ossimIrect> m_searchTileRects;
   static std::shared_ptr<AutoTiePoint> s_referenceATP;

};
}
#endif /* AtpGeneratorBase_H_ */