LoopsDatabaseDefiner.cc

Go to the documentation of this file.

// -*- mode:c++;tab-width:2;indent-tabs-mode:t;show-trailing-whitespace:t;rm-trailing-spaces:t -*-
// vi: set ts=2 noet:
//
// (c) Copyright Rosetta Commons Member Institutions.
// (c) This file is part of the Rosetta software suite and is made available under license.
// (c) The Rosetta software is developed by the contributing members of the Rosetta Commons.
// (c) For more information, see http://www.rosettacommons.org. Questions about this can be
// (c) addressed to University of Washington UW TechTransfer, email: license@u.washington.edu.

/// @file   protocols/loops/loops_definers/LoopsDatabaseDefiner.cc
/// @brief  A loops definer is creates a serialized loops list
/// @author Matthew O'Meara (mattjomear@gmail.com)

// Unit Headers
#include <protocols/loops/loops_definers/LoopsDatabaseDefiner.hh>

// Package Headers
#include <protocols/loops/Loop.hh>

// Project Headers
#include <protocols/jd2/Job.hh>
#include <protocols/jd2/JobDistributor.hh>
#include <utility/tag/Tag.hh>
#include <protocols/moves/DataMap.hh>

// Utility Headers
#include <utility/sql_database/DatabaseSessionManager.hh>

// Basic Headers
#include <basic/database/sql_utils.hh>

// External Headers
#include <cppdb/frontend.h>

// C++ Headers
#include <string>
#include <utility/excn/Exceptions.hh>
#include <sstream>


using std::string;
using std::endl;
using std::stringstream;
using utility::tag::TagPtr;
using core::pose::Pose;
using protocols::moves::DataMap;
using utility::sql_database::sessionOP;
using protocols::jd2::JobDistributor;
using basic::database::parse_database_connection;
using cppdb::result;

namespace protocols {
namespace loops {
namespace loops_definers {

LoopsDatabaseDefiner::LoopsDatabaseDefiner() :
  db_session_(),
  database_table_()
{}

LoopsDatabaseDefiner::~LoopsDatabaseDefiner() {}

LoopsDatabaseDefiner::LoopsDatabaseDefiner(
  LoopsDatabaseDefiner const & src
) :
  db_session_(src.db_session_),
  database_table_(src.database_table_)
{}


/// @brief Create another loops definer of the type matching the most-derived
/// version of the class.
LoopsDefinerOP
LoopsDatabaseDefiner::clone(
) const {
  return new LoopsDatabaseDefiner(*this);
}

/// @brief Used to parse an xml-like tag to load parameters and properties.
void
LoopsDatabaseDefiner::parse_my_tag(
  TagPtr const tag,
  DataMap const &,
  Pose const &
) {

  db_session_ = parse_database_connection(tag);

  database_table_ =
    tag->getOption<string>("database_table", "loops");

  string const type(tag->getName());

  if(!tag->hasOption("name")){
    throw utility::excn::EXCN_RosettaScriptsOption(
      "Unable to create unnamed LoopsDefiner (type: " + type + ")" );
  }
  string const loops_name(tag->getOption<string>("name"));

}

SerializedLoopList
LoopsDatabaseDefiner::apply(
  Pose const &
) {

  string pose_tag(JobDistributor::get_instance()->current_job()->input_tag());

  stringstream sql_stmt;
  sql_stmt
    << "SELECT start, stop, cut, skip_rate, extended FROM " << database_table_
    << " WHERE tag='" << pose_tag << "';";
  result res = (*db_session_) << sql_stmt.str();

  SerializedLoopList loop_list;
  while(res.next()){
    SerializedLoop loop;
    int extended;
    res >> loop.start >> loop.stop >> loop.cut >> loop.skip_rate >> extended;
    loop.extended = extended;

    loop_list.push_back(loop);
  }

  if(loop_list.size() == 0){
    stringstream error_message;
    error_message
      << "Unable to locate loops for job distributor input tag '"
      << pose_tag << "' in database." << endl;
    utility_exit_with_message(error_message.str());
  }

  return loop_list;
}

} //namespace
} //namespace
} //namespace