mystring.h

Go to the documentation of this file.

/* Copyright (C) 2001-2012 Stefan Buehler <sbuehler@ltu.se>

   This program is free software; you can redistribute it and/or modify it
   under the terms of the GNU General Public License as published by the
   Free Software Foundation; either version 2, or (at your option) any
   later version.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
   USA. */

#ifndef mystring_h
#define mystring_h

#include <algorithm>
#include <cassert>
#include <climits>
#include <sstream>
#include <string>
#include "array.h"
#include "matpack.h"

template <class charT>
class my_basic_string : public std::basic_string<charT> {
 public:
  // Constructors:
  my_basic_string() = default;
  explicit my_basic_string(Index n, char c = ' ');
  my_basic_string(const std::basic_string<charT>& A,
                  Index pos = 0,
                  Index numpos = my_basic_string<charT>::npos);
  my_basic_string(const char A[]);

  // Insert string before all occurrences of the substring.
  void insert_substr(const my_basic_string<charT>& searchstr,
                     const my_basic_string<charT>& insstr);

  // Split string
  void split(Array<my_basic_string<charT> >& aos,
             const my_basic_string<charT>& delim) const;

  void toupper() {
    std::transform(this->begin(), this->end(), this->begin(), ::toupper);
  }

  my_basic_string toupper() const {
    my_basic_string s = *this;
    s.toupper();
    return s;
  }

  void tolower() {
    std::transform(this->begin(), this->end(), this->begin(), ::tolower);
  }

  my_basic_string tolower() const {
    my_basic_string s = *this;
    s.tolower();
    return s;
  }

  void trim();

  // Number of elements:
  Index nelem() const;

  // Index operators:
  char operator[](Index n) const;
  char& operator[](Index n);

  static const Index npos = static_cast<Index>(std::basic_string<charT>::npos);

  typedef Index size_type;
};

// Member functions for my_basic_string:

// Constructors:

template <class charT>
inline my_basic_string<charT>::my_basic_string(Index n, char c)
    : std::basic_string<charT>(n, c) { /* Nothing to do here. */
}

template <class charT>
inline my_basic_string<charT>::my_basic_string(
    const std::basic_string<charT>& A, Index pos, Index numpos) {
  // Range checks:
  assert(0 <= pos);  // Start index must be 0 or greater 0.

  if (!A.size()) return;

  //   cout << "A = " << A << "\n";
  //   cout << "pos = " << pos << "\n";
  //   cout << "size = " << A.size() << "\n";

  assert(static_cast<typename std::basic_string<charT>::size_type>(pos) <
         A.size());
  // At most the last element of the original string.

  assert(numpos == my_basic_string<charT>::npos ||
         ((numpos >= 0) &&
          (static_cast<typename std::basic_string<charT>::size_type>(numpos) <=
           (A.size() -
            pos))));  // Number of characters to copy must be at the most the
                      // number left. -1 means all remaining characters.

  // The assertions look complicated, because we have to cast pos and
  // npos to the unsigned size type of basic string to avoid warning
  // messages from the compiler. Both casts are save, because previous
  // assertions check that pos and npos are positive. (The allowed
  // case npos -1 (=my_basic_string<charT>::npos) is also handled
  // correctly.)

  std::basic_string<charT>::operator=(std::basic_string<charT>(A, pos, numpos));
}

template <class charT>
inline my_basic_string<charT>::my_basic_string(const char A[])
    : std::basic_string<charT>(A) { /* Nothing to do here. */
}

template <class charT>
inline void my_basic_string<charT>::insert_substr(
    const my_basic_string<charT>& searchstr,
    const my_basic_string<charT>& insstr) {
  size_t searchstr_size = searchstr.size();
  size_t insstr_size = insstr.size();
  size_t start_pos = 0;

  while (start_pos != std::string::npos) {
    start_pos = this->find(searchstr, start_pos);
    if (start_pos && start_pos != std::string::npos) {
      this->insert(start_pos, insstr);
      start_pos += searchstr_size + insstr_size;
    }
  }
}

template <class charT>
inline void my_basic_string<charT>::split(
    Array<my_basic_string<charT> >& aos,
    const my_basic_string<charT>& delim) const {
  size_t pos, oldpos;
  pos = oldpos = 0;
  aos.resize(0);

  while (oldpos < (size_t)this->nelem() &&
         (pos = this->find(delim, oldpos)) !=
             (size_t)my_basic_string<charT>::npos) {
    if (pos && pos - oldpos) aos.push_back(this->substr(oldpos, pos - oldpos));
    oldpos = pos + delim.nelem();
  }

  if (oldpos < (size_t)this->nelem()) aos.push_back(this->substr(oldpos));
}

template <class charT>
void my_basic_string<charT>::trim() {
  // Create ref to self for readability
  my_basic_string& this_string = *this;

  // Remove leading whitespace
  while (0 != this_string.nelem() &&
         (' ' == this_string[0] || '\t' == this_string[0] ||
          '\n' == this_string[0] || '\r' == this_string[0]))
    this_string.erase(0, 1);

  // Remove trailing whitespace
  while (0 != this_string.nelem() &&
         (' ' == this_string[this_string.nelem() - 1] ||
          '\t' == this_string[this_string.nelem() - 1] ||
          '\n' == this_string[this_string.nelem() - 1] ||
          '\r' == this_string[this_string.nelem() - 1]))
    this_string.erase(this_string.nelem() - 1);
}

template <class charT>
inline Index my_basic_string<charT>::nelem() const {
  size_t s = this->size();
  assert(s < LONG_MAX);
  return static_cast<long>(s);
}

template <class charT>
inline char my_basic_string<charT>::operator[](Index n) const {
  assert(0 <= n);
  assert(n < nelem());
  return std::basic_string<charT>::operator[](n);
}

template <class charT>
inline char& my_basic_string<charT>::operator[](Index n) {
  assert(0 <= n);
  assert(n < nelem());
  return std::basic_string<charT>::operator[](n);
}

typedef my_basic_string<char> String;

typedef Array<String> ArrayOfString;

typedef Array<Array<String> > ArrayOfArrayOfString;

template <class T>
void extract(T& x, String& line, Index n) {
  // Initialize output to zero! This is important, because otherwise
  // the output variable could `remember' old values.
  x = T(0);

  // This will contain the short subString with the item to extract.
  // Make it a String stream, for easy parsing,
  // extracting subString of width n from line:
  std::istringstream item(line.substr(0, n));

  //  cout << "line = '" << line << "'\n";
  //   cout << "line.substr(0,n) = " << line.substr(0,n) << endl;
  //   cout << "item = " << item.str() << endl;

  // Shorten line by n:
  line.erase(0, n);
  //  cout << "line = " << line << endl;

  // Convert with the aid of String stream item:
  item >> x;
}

#endif  // mystring_h