PvlContainer.cpp

Go to the documentation of this file.

#include <algorithm>

#include <QList>

#include "PvlContainer.h"
#include "Pvl.h"
#include "FileName.h"
#include "IException.h"
#include "Message.h"
#include "PvlFormat.h"

using namespace std;

namespace Isis {

  PvlContainer::PvlContainer(const QString &type) {
    init();
    m_name.setName(type);
  }


  PvlContainer::PvlContainer(const QString &type, const QString &name) {
    init();
    m_name.setName(type);
    setName(name);
  }


  PvlContainer::PvlContainer(const PvlContainer &other) {
    *this = other;
  }



  void PvlContainer::init() {
    m_filename = "";
    m_formatTemplate = NULL;
  }

  Isis::PvlKeyword &PvlContainer::findKeyword(const QString &name) {
    PvlKeywordIterator key = findKeyword(name, begin(), end());
    if(key == end()) {
      QString msg = "PVL Keyword [" + name + "] does not exist in [" +
                   type() + " = " + this->name() + "]";
      if(m_filename.size() > 0) msg += " in file [" + m_filename + "]";
      throw IException(IException::Unknown, msg, _FILEINFO_);
    }

    return *key;
  }

  const Isis::PvlKeyword &PvlContainer::findKeyword(const QString &name) const {
    ConstPvlKeywordIterator key = findKeyword(name, begin(), end());
    if(key == end()) {
      QString msg = "PVL Keyword [" + name + "] does not exist in [" +
                   type() + " = " + this->name() + "]";
      if(m_filename.size() > 0) msg += " in file [" + m_filename + "]";
      throw IException(IException::Unknown, msg, _FILEINFO_);
    }

    return *key;
  }

  void PvlContainer::deleteKeyword(const QString &name) {
    PvlKeywordIterator key = findKeyword(name, begin(), end());
    if(key == end()) {
      QString msg = "PVL Keyword [" + name + "] does not exist in [" +
                   type() + " = " + this->name() + "]";
      if(m_filename.size() > 0) msg += " in file [" + m_filename + "]";
      throw IException(IException::Unknown, msg, _FILEINFO_);
    }

    m_keywords.erase(key);
  }


  void PvlContainer::deleteKeyword(const int index) {
    if(index >= (int)m_keywords.size() || index < 0) {
      QString msg = "The specified index is out of bounds in PVL [" +
                   type() + " = " + name() + "]";
      if(m_filename.size() > 0) msg += " in file [" + m_filename + "]";
      throw IException(IException::Unknown, msg, _FILEINFO_);
    }

    PvlKeywordIterator key = begin();
    for(int i = 0; i < index; i++) key++;

    m_keywords.erase(key);
  }


  bool PvlContainer::cleanDuplicateKeywords() {
    bool keywordDeleted = false;

    for(int index = 0; index < m_keywords.size(); index ++) {
      PvlKeyword &current = m_keywords[index];

      for(PvlKeywordIterator key = begin() + index + 1; key < end(); key ++) {
        if(current == *key) {
          m_keywords.erase(key);
          keywordDeleted = true;
        }
      }
    }

    return keywordDeleted;
  }


  bool PvlContainer::hasKeyword(const QString &name) const {
    ConstPvlKeywordIterator key = findKeyword(name, begin(), end());
    if(key == end()) return false;
    return true;
  }


  PvlKeyword &PvlContainer::operator[](const int index) {
    if(index < 0 || index >= (int)m_keywords.size()) {
      QString msg = Message::ArraySubscriptNotInRange(index);
      throw IException(IException::Programmer, msg, _FILEINFO_);
    }
    return *(m_keywords.begin() + index);
  };


  const Isis::PvlKeyword &PvlContainer::operator[](const int index) const {
    if(index < 0 || index >= (int)m_keywords.size()) {
      QString msg = Message::ArraySubscriptNotInRange(index);
      throw IException(IException::Programmer, msg, _FILEINFO_);
    }
    return *(m_keywords.begin() + index);
  }

  void PvlContainer::addKeyword(const Isis::PvlKeyword &key,
                                const InsertMode mode) {
    if(mode == Append) {
      m_keywords.push_back(key);
    }
    else if(hasKeyword(key.name())) {
      Isis::PvlKeyword &outkey = findKeyword(key.name());
      outkey = key;
    }
    else {
      m_keywords.push_back(key);
    }
  }

  PvlContainer::PvlKeywordIterator PvlContainer::addKeyword(const Isis::PvlKeyword &key,
      PvlKeywordIterator pos) {
    return (m_keywords.insert(pos, key));
  }

  ostream &operator<<(std::ostream &os, PvlContainer &container) {

    // Set up a Formatter (This should not be necessary for a container because
    // Object or Group should have done this already, but just in case.
    bool removeFormatter = false;
    if(container.format() == NULL) {
      container.setFormat(new PvlFormat());
      removeFormatter = true;
    }

    Isis::PvlContainer outTemplate("DEFAULT_TEMPLATE");
    if(container.hasFormatTemplate()) outTemplate = *(container.formatTemplate());

    // Look for and process all include files inside the template
    Isis::PvlContainer newTemp(outTemplate.type());

    // Include files take precedence over all other objects and groups
    for(int i = 0; i < outTemplate.keywords(); i++) {
      if(outTemplate[i].isNamed("Isis:PvlTemplate:File")) {
        QString filename = outTemplate[i];
        Isis::FileName file(filename);
        if(!file.fileExists()) {
          QString message = "Could not open the template file [" + filename + "]";
          throw IException(IException::Io, message, _FILEINFO_);
        }
        Isis::Pvl include(file.expanded());

        for(int j = 0; j < include.keywords(); j++) {
          if(!newTemp.hasKeyword(include[j].name()))
            newTemp.addKeyword(include[j]);
        }
      }
      // If it is not an include file keyword add it in place
      else if(!newTemp.hasKeyword(outTemplate[i].name())) {
        newTemp.addKeyword(outTemplate[i]);
      }
    }

    outTemplate = newTemp;

    // Figure out the longest keyword
    int width = 0;
    for(int i = 0; i < container.keywords(); i++) {
      if(container[i].name().length() > width) width = container[i].name().length();
    }

    // This number keeps track of the number of keywords written
    int numKeywords = 0;

    // Write out the container using the output format template
    for(int i = 0; i < outTemplate.keywords(); i++) {
      for(int j = 0; j < container.keywords(); j++) {
        if(outTemplate[i].name() != container[j].name()) continue;
        container[j].setIndent(container.indent());
        container[j].setWidth(width);
        container[j].setFormat(container.format());
        // Add a blank line before keyword comments
        if(outTemplate[i].comments() + container[j].comments() > 0) os << container.format()->formatEOL();
        if(outTemplate[i].comments() > 0) {
          for(int k = 0; k < outTemplate[i].comments(); k++) {
            for(int l = 0; l < outTemplate[i].indent() + container[j].indent(); l++) os << " ";
            os << outTemplate[i].comment(k) << container.format()->formatEOL();
          }
        }
        os << container[j];
        container[j].setFormat(NULL);
        container[j].setIndent(0);
        container[j].setWidth(0);
        if(++numKeywords < container.keywords()) {
//          if (j+1 < container.Keywords() && container[j+1].comments() > 0) os << container.format()->formatEOL();
          os << container.format()->formatEOL();
        }
      }
    }

    // Output the keywords in the container that were not specified in the template
    for(int i = 0; i < container.keywords(); i++) {
      if(outTemplate.hasKeyword(container[i].name())) continue;
      container[i].setIndent(container.indent());
      container[i].setWidth(width);
      container[i].setFormat(container.format());
      os << container[i];
      container[i].setFormat(NULL);
      container[i].setIndent(0);
      container[i].setWidth(0);
      if(++numKeywords < container.keywords()) {
        if(i + 1 < container.keywords() && container[i+1].comments() > 0) os << container.format()->formatEOL();
        os << container.format()->formatEOL();
      }
    }

    if(removeFormatter) {
      delete container.format();
      container.setFormat(NULL);
    }

    return os;
  }


  PvlContainer::PvlKeywordIterator PvlContainer::findKeyword(const QString &name,
      PvlContainer::PvlKeywordIterator beg,
      PvlContainer::PvlKeywordIterator end) {
    PvlKeyword temp(name);
    return find(beg, end, temp);
  };


  PvlContainer::ConstPvlKeywordIterator PvlContainer::findKeyword(const QString &name,
      PvlContainer::ConstPvlKeywordIterator beg,
      PvlContainer::ConstPvlKeywordIterator end) const {
    PvlKeyword temp(name);
    return find(beg, end, temp);
  };


  const PvlContainer &PvlContainer::operator=(const PvlContainer &other) {
    m_filename = other.m_filename;
    m_name = other.m_name;
    m_keywords = other.m_keywords;
    m_formatTemplate = other.m_formatTemplate;

    return *this;
  }

  void PvlContainer::validateAllKeywords(PvlContainer & pPvlCont)
  {
    // Validate the Keywords in the current Object
    int iTmplKeySize = keywords();
    for(int i=0; i<iTmplKeySize; i++) {
      PvlKeyword & pvlTmplKwrd = (*this)[i];
      QString sKeyName = pvlTmplKwrd.name();
      bool bKwrdFound = false;

      // These are reserved keywords for properties like "Range", "Value", "Type",
      // "Required" or "Repeated"
      if(sKeyName.contains("__Required") || sKeyName.contains("__Repeated") ||
         sKeyName.contains("__Range") || sKeyName.contains("__Value") ||
         sKeyName.contains("__Type")) {
        continue;
      }

      if(pPvlCont.hasKeyword(sKeyName)) {
        PvlKeyword & pvlKwrd = pPvlCont.findKeyword(sKeyName);
        QString sTmplKwrdRange = sKeyName + "__Range";
        QString sTmplKwrdValue = sKeyName + "__Value";
        QString sTmplKwrdType  = sKeyName + "__Type";
        QString sType="";
        PvlKeyword pvlTmplKwrdRange, pvlTmplKwrdValue;

        // Check if Type is specified (positive or negative for numbers)
        if(hasKeyword(sTmplKwrdType)) {
          sType = findKeyword(sTmplKwrdType)[0];
        }
        // Check for Range
        if(hasKeyword(sTmplKwrdRange)) {
          pvlTmplKwrdRange = findKeyword(sTmplKwrdRange);
          pvlTmplKwrd.validateKeyword(pvlKwrd, sType, &pvlTmplKwrdRange);
        }
        // Check for Value
        else if(hasKeyword(sTmplKwrdValue)) {
          pvlTmplKwrdValue = findKeyword(sTmplKwrdValue);
          pvlTmplKwrd.validateKeyword(pvlKwrd, sType, &pvlTmplKwrdValue);
        }
        else {
          pvlTmplKwrd.validateKeyword(pvlKwrd, sType);
        }
        pPvlCont.deleteKeyword(pvlKwrd.name());
        bKwrdFound = true;
      }
      else {
        bKwrdFound = true;
        QString sOption = sKeyName + "__Required";
        if(hasKeyword(sOption)) {
          PvlKeyword pvlKeyOption = findKeyword(sOption);
          if(pvlKeyOption[0] == "true") { // Required is true
            bKwrdFound = false;
          }
        }
      }
      if (bKwrdFound == false) {
        QString sErrMsg = "Keyword \"" + sKeyName + "\" Not Found in the Template File\n";
        throw IException(IException::User, sErrMsg, _FILEINFO_);
      }

      // Check for "Repeated" Option
      validateRepeatOption(pvlTmplKwrd, pPvlCont);
    }
  }

  void PvlContainer::validateRepeatOption(PvlKeyword & pPvlTmplKwrd, PvlContainer & pPvlCont)
  {
    QString sTmplKeyName = pPvlTmplKwrd.name();

    // Check for the Type
    QString sType = sTmplKeyName + "__Type";
    QString sValueType ="";
    if(hasKeyword(sType)) {
      sValueType = findKeyword(sType)[0];
    }
    QString sRepeatOption = sTmplKeyName + "__Repeated";
    bool bRepeat =false;
    if(hasKeyword(sRepeatOption)) {
      PvlKeyword pvlKeyOption = findKeyword(sRepeatOption);
      if(pvlKeyOption[0] == "true") { // Required is true
        bRepeat = true;
      }
    }
    if(bRepeat) {
      int iKeySize = pPvlCont.keywords();
      for(int j=(iKeySize-1); j>=0; j--) {
        PvlKeyword & pvlKwrd = pPvlCont[j];
        QString sKeyName = pvlKwrd.name();
        if(sTmplKeyName == sKeyName) {
          pPvlTmplKwrd.validateKeyword(pvlKwrd, sValueType);
          pPvlCont.deleteKeyword(pvlKwrd.name());
        }
      }
    }
  }

} // end namespace isis