ProcessByQuickFilter.cpp

Go to the documentation of this file.

#include "ProcessByQuickFilter.h"

#include "Application.h"
#include "FilterCachingAlgorithm.h"
#include "IException.h"
#include "LineManager.h"
#include "Preference.h"
#include "QuickFilter.h"

using namespace std;

namespace Isis {

  ProcessByQuickFilter::ProcessByQuickFilter() : Isis::Process() {
    SetFilterParameters(0, 0);
    p_getParametersFromUser = true;
  }

  void ProcessByQuickFilter::StartProcess(void
                                          funct(Isis::Buffer &in, Isis::Buffer &out, Isis::QuickFilter &filter)) {
    // Error checks ... there must be one input and output
    if(InputCubes.size() != 1) {
      string m = "StartProcess only supports exactly one input file";
      throw IException(IException::Programmer, m, _FILEINFO_);
    }

    if(OutputCubes.size() != 1) {
      string m = "StartProcess only supports exactly one output file";
      throw IException(IException::Programmer, m, _FILEINFO_);
    }

    // The lines in the input and output must match
    if(InputCubes[0]->lineCount() != OutputCubes[0]->lineCount()) {
      string m = "The lines in the input and output cube must match";
      throw IException(IException::Programmer, m, _FILEINFO_);
    }

    // The samples in the input and output must match
    if(InputCubes[0]->sampleCount() != OutputCubes[0]->sampleCount()) {
      string m = "The samples in the input and output cube must match";
      throw IException(IException::Programmer, m, _FILEINFO_);
    }

    // The bands in the input and output must match
    if(InputCubes[0]->bandCount() != OutputCubes[0]->bandCount()) {
      string m = "The bands in the input and output cube must match";
      throw IException(IException::Programmer, m, _FILEINFO_);
    }

    // Construct line buffer managers
    Isis::LineManager *topline = new Isis::LineManager(*InputCubes[0]);
    Isis::LineManager *iline = new Isis::LineManager(*InputCubes[0]);
    Isis::LineManager *botline = new Isis::LineManager(*InputCubes[0]);
    Isis::LineManager *oline = new Isis::LineManager(*OutputCubes[0]);

    int lines = InputCubes[0]->lineCount();
    int samples = InputCubes[0]->sampleCount();
    int bands = InputCubes[0]->bandCount();

    // See if we need to get parameters from the user
    if(p_getParametersFromUser) GetFilterParameters();

    // Make sure the boxcar widht and height aren't too big for the image
    if(lines * 2 - 1 < p_boxcarLines) {
      string msg = "Boxcar height is too big for cube size";
      throw IException(IException::User, msg, _FILEINFO_);
    }

    if(samples * 2 - 1 < p_boxcarSamples) {
      string msg = "Boxcar width is too big for cube size";
      throw IException(IException::User, msg, _FILEINFO_);
    }

    InputCubes[0]->addCachingAlgorithm(new FilterCachingAlgorithm(3));

    // Create the filter object
    Isis::QuickFilter filter(samples, p_boxcarSamples, p_boxcarLines);
    filter.SetMinMax(p_low, p_high);
    filter.SetMinimumPixels(p_minimum);

    // Loop for each band
    p_progress->SetMaximumSteps(lines * bands);
    p_progress->CheckStatus();
    for(int band = 1; band <= bands; band++) {
      // Preload the filter
      filter.Reset();
      int top = 1 - filter.HalfHeight();
      int bot;
      for(bot = top; bot <= (1 + filter.HalfHeight()); bot++) {
        int iline = bot;
        if(bot <= 0) iline = (-1 * bot + 2);
        botline->SetLine(iline, band);
        InputCubes[0]->read(*botline);
        filter.AddLine(botline->DoubleBuffer());
      }
      bot = 1 + filter.HalfHeight() + 1;

      // Loop for each line
      for(int line = 1; line <= lines; line++) {
        // Process a line
        iline->SetLine(line, band);
        oline->SetLine(line, band);

        InputCubes[0]->read(*iline);
        funct(*iline, *oline, filter);
        OutputCubes[0]->write(*oline);

        // Remove the top line
        if(top >= 1) {
          topline->SetLine(top, band);
        }
        else {
          topline->SetLine(-1 * top + 2, band);
        }

        InputCubes[0]->read(*topline);
        filter.RemoveLine(topline->DoubleBuffer());
        top++;

        // Add the next line
        p_progress->CheckStatus();
        if(line == lines) continue;

        if(bot <= InputCubes[0]->lineCount()) {
          botline->SetLine(bot, band);
        }
        else {
          botline->SetLine(lines - (bot - lines), band);
        }

        InputCubes[0]->read(*botline);
        filter.AddLine(botline->DoubleBuffer());
        bot++;

        // Update the progress
      }
    }

    // Free buffers before returning
    delete topline;
    delete iline;
    delete botline;
    delete oline;
  }

  void ProcessByQuickFilter::SetFilterParameters(int samples, int lines,
      double low, double high, int minimum) {
    p_getParametersFromUser = false;
    p_boxcarSamples = samples;
    p_boxcarLines = lines;
    p_low = low;
    p_high = high;
    p_minimum = minimum;
  }

  void ProcessByQuickFilter::GetFilterParameters() {
    // Get boxcar size
    p_boxcarSamples = Application::GetUserInterface().GetInteger("SAMPLES");
    p_boxcarLines = Application::GetUserInterface().GetInteger("LINES");

    // Get valid pixel range
    p_low = -DBL_MAX;
    p_high = DBL_MAX;
    if(Application::GetUserInterface().WasEntered("LOW")) {
      p_low = Application::GetUserInterface().GetDouble("LOW");
    }

    if(Application::GetUserInterface().WasEntered("HIGH")) {
      p_high = Application::GetUserInterface().GetDouble("HIGH");
    }

    // Get valid pixel count
    p_minimum = 0;
    if(Application::GetUserInterface().WasEntered("MINIMUM")) {
      p_minimum = Application::GetUserInterface().GetInteger("MINIMUM");
    }
  }
} // end namespace isis