query.hh

Go to the documentation of this file.

//------------------------------------------------------------------------------
// This file is part of the OpenStructure project <www.openstructure.org>
//
// Copyright (C) 2008-2020 by the OpenStructure authors
//
// This library is free software; you can redistribute it and/or modify it under
// the terms of the GNU Lesser General Public License as published by the Free
// Software Foundation; either version 3.0 of the License, or (at your option)
// any later version.
// This library 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 Lesser General Public License for more
// details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with this library; if not, write to the Free Software Foundation, Inc.,
// 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
//------------------------------------------------------------------------------
#ifndef OST_QUERY_HF
#define OST_QUERY_HF

/*
 Author: Marco Biasini
 */

#include <string>
#include <vector>
#include <set>

#include <boost/variant.hpp>
#include <boost/logic/tribool.hpp>

#include <ost/range.hh>

#include <ost/mol/module_config.hh>
#include <ost/mol/impl/query_impl_fw.hh>
#include <ost/mol/query_error.hh>
#include <ost/mol/atom_handle.hh>
#include <ost/mol/query_state.hh>

namespace ost { namespace mol {

struct DLLEXPORT_OST_MOL QueryFlag {
  typedef enum {
    EXCLUSIVE_BONDS=1,
    NO_BONDS=2,
    MATCH_RESIDUES=4
  } Flag;
};

typedef uint QueryFlags;

class DLLEXPORT_OST_MOL Query {
public:
  Query(const String& query_string="");
  
  Query(const impl::QueryImplP& impl);
  
  bool IsAtomSelected(const AtomHandle& ah) const;

  bool IsChainSelected(const ChainHandle& chain) const;
  
  bool IsResidueSelected(const ResidueHandle& residue) const;  
  
  const String& GetQueryString() const;
  
  const QueryErrorDesc& GetErrorDescription() const;

  QueryState CreateQueryState(const EntityView& view) const;
  
  QueryState CreateQueryState(const EntityHandle& view) const;
  
  // only valid if parsed without error
  bool IsValid() const;
  // an empty query String defaults to matching everything
  bool MatchAll() const;
  
  // for internal use only
  impl::QueryImplP Impl();
  const impl::QueryImplP& Impl() const;
private:  
  impl::QueryImplP   impl_;
};

// inlined helper function to quote strings for use in queries (e.g. cname=..).
// throws Error if string cannot be quoted
inline String DLLEXPORT_OST_MOL QueryQuoteName(const String& name) {
  // check what quotation marks to use
  char quote = '\'';
  if (name.find('\'') != String::npos) {
    if (name.find('"') != String::npos) {
      throw Error("Cannot quote chain name " + name + " because it contains "
                  "both ' and \" in its name.");
    }
    quote = '"';
  }
  // check problematic \ at end (escapes quotation mark and breaks logic)
  if (name[name.length() - 1] == '\\') {
    throw Error("Cannot quote chain name " + name + " because it ends in \\.");
  }
  return quote + name + quote;
}

}} // ns

#endif