QnetCubeDistanceFilter.cpp

#include "QnetCubeDistanceFilter.h"

#include <vector>
#include <cmath>

#include <QGridLayout>
#include <QLabel>
#include <QLineEdit>
#include <QMessageBox>
#include <QRadioButton>

#include "QnetNavTool.h"
#include "Camera.h"
#include "ControlNet.h"
#include "ControlPoint.h"
#include "ControlMeasure.h"
#include "Distance.h"
#include "Latitude.h"
#include "Longitude.h"
#include "SerialNumberList.h"
#include "SurfacePoint.h"

namespace Isis {
  QnetCubeDistanceFilter::QnetCubeDistanceFilter(QnetNavTool *navTool, QWidget *parent) :
      QnetFilter(navTool, parent) {
    p_lineEdit = NULL;
    p_pixels = NULL;
    p_meters = NULL;

    // Create components for the filter window
    QLabel *label = new QLabel("Filter by distance between points in cube");
    QLabel *lessThan = new QLabel("Contains points within ");
    p_lineEdit = new QLineEdit;
    p_pixels = new QRadioButton("pixels");
    p_meters = new QRadioButton("meters");
    p_pixels->setChecked(true);
    QLabel *pad = new QLabel();

    // create layout for radio buttons
    QVBoxLayout *units = new QVBoxLayout();
    units->addWidget(p_pixels);
    units->addWidget(p_meters);

    // Create the layout and add the components to it
    QGridLayout *gridLayout = new QGridLayout();
    gridLayout->addWidget(label, 0, 0, 1, 2);
    gridLayout->addWidget(lessThan, 1, 0);
    gridLayout->addWidget(p_lineEdit, 1, 1);
    gridLayout->addLayout(units, 1, 2);
    gridLayout->addWidget(pad, 2, 0);
    gridLayout->setRowStretch(2, 50);
    this->setLayout(gridLayout);
  }

  void QnetCubeDistanceFilter::filter() {
    // Make sure we have a list of files to filter
    if (serialNumberList() == NULL) {
      QMessageBox::information((QWidget *)parent(),
          "Error", "No cubes to filter");
      return;
    }

    // Make sure the user entered a filtering value
    if (p_lineEdit->text() == "") {
      QMessageBox::information((QWidget *)parent(),
          "Error", "Distance value must be entered");
      return;
    }

    // Get the user entered value for filtering
    int userEntered = p_lineEdit->text().toInt();

    // create temporary QList to contain new filtered images
    QList <int> temp;
    // Loop through each image in the currently filtered list
    // Loop in reverse order for consistency with other filters
    for (int i = filteredImages().size() - 1; i >= 0; i--) {

      // Loop through every control point in the control net
      for (int cp1 = 0; cp1 < controlNet()->GetNumPoints(); cp1++) {
        ControlPoint controlPt1 = *(*controlNet())[cp1];
        ControlMeasure controlMeas1;
        // Loop through each control measure of this point until we find this image
        for (int cm1 = 0; cm1 < controlPt1.GetNumMeasures(); cm1++) {
          // Check control measure to see if this point is in the current image
          if (controlPt1[cm1]->GetCubeSerialNumber() == serialNumberList()->serialNumber(filteredImages()[i])) {
            // this measure matches the image, set controlMeas1 variable and break out of loop
            controlMeas1 = *controlPt1[cm1];
            break;
          }
          // this measure does not match this cube, keep looking, continue to next measure
          else
            continue;
        }  // if no matching measure was found, controlMeas1 defaults to sample,line = 0

        // If the measure for this image has no line/sample values,
        // we can not measure distance, so continue to the next control point, cp1
        if (controlMeas1.GetSample() == 0 && controlMeas1.GetLine() == 0)
          continue;
        // if the user chooses distance in meters, create camera to find lat/lon for this measure
        double rad = 0, lat1 = 0, lon1 = 0;
        if (p_meters->isChecked()) {
          Camera *cam1;
          cam1  = controlNet()->Camera(filteredImages()[i]);
          // try to set image using sample/line values
          if (cam1->SetImage(controlMeas1.GetSample(), controlMeas1.GetLine())) {
            rad = cam1->LocalRadius().meters();
            lat1 = cam1->UniversalLatitude();
            lon1 = cam1->UniversalLongitude();
          }
          else {
            // if SetImage fails for this measure, don't calculate the distance from this point
            // continue to next control point, cp1
            continue;
          }
        }
        // Loop through the remaining control points to compute distances
        for (int cp2 = (cp1 + 1); cp2 < controlNet()->GetNumPoints(); cp2++) {
          ControlPoint &controlPt2 = *(*controlNet())[cp2];
          ControlMeasure controlMeas2;
          // Loop through each measure of the second point until we find this image
          for (int cm2 = 0; cm2 < controlPt2.GetNumMeasures(); cm2++) {
            // Check to see if this point is in the image
            if (controlPt2[cm2]->GetCubeSerialNumber() == serialNumberList()->serialNumber(filteredImages()[i])) {
              // this measure matches the image, set controlMeas2 variable and break out of loop
              controlMeas2 = *controlPt2[cm2];
              break;
            }
            // this measure does not match this cube, keep looking, continue to next measure
            else
              continue;
          }  // if no matching measure was found, controlMeas1 defaults to sample,line = 0

          // If the measure has no samp/line values, continue to next point, cp2
          if (controlMeas2.GetSample() == 0 && controlMeas2.GetLine() == 0)
            continue;

          // Now determine distance based on the units chosen
          double dist = 0;
          if (p_pixels->isChecked()) {
            double deltaSamp = controlMeas1.GetSample() - controlMeas2.GetSample();
            double deltaLine = controlMeas1.GetLine() - controlMeas2.GetLine();
            // use the distance formula for cartesian coordinates
            dist = sqrt((deltaSamp * deltaSamp) + (deltaLine * deltaLine));
          }
          else
            // meters is checked
          {
            // create camera to find lat/lon for this measure
            double lat2 = 0, lon2 = 0;
            Camera *cam2;
            cam2 = controlNet()->Camera(filteredImages()[i]);
            // try to set image using sample/line values
            if (cam2->SetImage(controlMeas2.GetSample(), controlMeas2.GetLine())) {
              lat2 = cam2->UniversalLatitude();
              lon2 = cam2->UniversalLongitude();
            }
            else {
              // if SetImage fails, don't calculate the distance to this point
              // continue to the next control point, cp2
              continue;
            }
            // Calculate the distance between the two points
            // Get the distance from the camera class
            SurfacePoint point1(
                Latitude(lat1, Angle::Degrees),
                Longitude(lon1, Angle::Degrees),
                Distance(rad, Distance::Meters));
            SurfacePoint point2(
                Latitude(lat2, Angle::Degrees),
                Longitude(lon2, Angle::Degrees),
                Distance(rad, Distance::Meters));
            dist = point1.GetDistanceToPoint(point2,
                Distance(rad, Distance::Meters)).meters();
          }

          if (dist == 0) {
            // distance not calculated, continue to next cp2
            continue;
          }

          // If the distance is less than the value entered by the user,
          // keep the cube in the list
          if (dist < userEntered) {
            if (!temp.contains(filteredImages()[i])) {
              // this image is added to the filtered list
              temp.push_back(filteredImages()[i]);
            }
          }
          if (temp.contains(filteredImages()[i])) {
            // (*filteredImages())[i] is already added to the filtered list,
            // no point in checking other distances,
            // break out of cp2 loop and continue to next cp1
            break;
          }
        } // this ends cp2 loop

        if (temp.contains(filteredImages()[i])) {
          // (*filteredImages())[i] is already added to the filtered list,
          // no point in checking other distances,
          // break out of cp1 loop and continue to next value of i
          break;
        }
      } // this ends cp1 loop
    } // this ends i loop

    // Sort QList of filtered points before displaying list to user
    qSort(temp.begin(), temp.end());
    // replace existing filter list with this one
    filteredImages() = temp;

    // Tell the nav tool that a list has been filtered and it needs to updated
    emit filteredListModified();
    return;
  }
}