src/dssp.cpp

   1 /*
   2 * This file is part of the GROMACS molecular simulation package.
   3 *
   4 * Copyright (c) 2021, by the GROMACS development team, led by
   5 * Mark Abraham, David van der Spoel, Berk Hess, and Erik Lindahl,
   6 * and including many others, as listed in the AUTHORS file in the
   7 * top-level source directory and at http://www.gromacs.org.
   8 *
   9 * GROMACS is free software; you can redistribute it and/or
  10 * modify it under the terms of the GNU Lesser General Public License
  11 * as published by the Free Software Foundation; either version 2.1
  12 * of the License, or (at your option) any later version.
  13 *
  14 * GROMACS is distributed in the hope that it will be useful,
  15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  17 * Lesser General Public License for more details.
  18 *
  19 * You should have received a copy of the GNU Lesser General Public
  20 * License along with GROMACS; if not, see
  21 * http://www.gnu.org/licenses, or write to the Free Software Foundation,
  22 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA.
  23 *
  24 * If you want to redistribute modifications to GROMACS, please
  25 * consider that scientific software is very special. Version
  26 * control is crucial - bugs must be traceable. We will be happy to
  27 * consider code for inclusion in the official distribution, but
  28 * derived work must not be called official GROMACS. Details are found
  29 * in the README & COPYING files - if they are missing, get the
  30 * official version at http://www.gromacs.org.
  31 *
  32 * To help us fund GROMACS development, we humbly ask that you cite
  33 * the research papers on the package. Check out http://www.gromacs.org.
  34 */
  35 /*! \internal \file
  36 * \brief
  37 * Implements gmx::analysismodules::Trajectory.
  38 *
  39 * \author Sergey Gorelov <gorelov_sv@pnpi.nrcki.ru>
  40 * \author Anatoly Titov <titov_ai@pnpi.nrcki.ru>
  41 * \author Alexey Shvetsov <alexxyum@gmail.com>
  42 * \ingroup module_trajectoryanalysis
  43 */
  44
  45 #include "dssp.h"
  46 #include "dssptools.h"
  47
  48 #include <algorithm>
  49
  50 #include <iostream> //remove
  51
  52 #include <gromacs/trajectoryanalysis.h>
  53
  54 namespace gmx
  55 {
  56
  57 namespace analysismodules
  58 {
  59
  60 class Dssp : public TrajectoryAnalysisModule
  61 {
  62 public:
  63    Dssp();
  64
  65    void initOptions(IOptionsContainer* options, TrajectoryAnalysisSettings* settings) override;
  66
  67    void optionsFinished(TrajectoryAnalysisSettings* settings) override;
  68
  69
  70    void initAnalysis(const TrajectoryAnalysisSettings& settings, const TopologyInformation& top) override;
  71
  72    void analyzeFrame(int frnr, const t_trxframe& fr, t_pbc* pbc, TrajectoryAnalysisModuleData* /* pdata */) override;
  73
  74    void finishAnalysis(int nframes) override;
  75    void writeOutput() override;
  76
  77 private:
  78    initParameters                                       initParams;
  79    std::string                                          filename_;
  80    int                                                  nres;
  81    DsspTool                                             DT;
  82    std::vector<std::pair<int, std::string>>             data;
  83 };
  84
  85 Dssp::Dssp() {}
  86
  87
  88 void Dssp::initOptions(IOptionsContainer* options, TrajectoryAnalysisSettings* settings)
  89 {
  90    static const char* const desc[] = {
  91        "[THISMODULE] todo",
  92    };
  93    options->addOption(
  94             StringOption("o").store(&filename_).required().defaultValue("SSP.dat").description("Filename for DSSP output"));
  95    options->addOption(RealOption("cutoff").store(&initParams.cutoff_).required().defaultValue(4.0).description(
  96            "cutoff for neighbour search"));
  97    options->addOption(
  98              SelectionOption("sel").store(&initParams.sel_).required().description(
  99                    "Group for DSSP"));
 100    options->addOption(
 101                EnumOption<NBSearchMethod>("algo").store(&initParams.NBS).required().defaultValue(NBSearchMethod::DSSP).enumValue(NBSearchMethodNames).description("Algorithm for search residues' neighbours"));
 102    options->addOption(
 103                BooleanOption("addh").store(&initParams.addHydrogens).defaultValue(false).description("Add hydrogens or not"));
 104    options->addOption(
 105                BooleanOption("p").store(&initParams.PPHelices).defaultValue(true).description("Prefer Pi Helices"));
 106    options->addOption(
 107                BooleanOption("v").store(&initParams.verbose).defaultValue(false).description(">:("));
 108    settings->setHelpText(desc);
 109 }
 110
 111
 112 void Dssp::optionsFinished(TrajectoryAnalysisSettings* /* settings */) {
 113 }
 114
 115
 116 void Dssp::initAnalysis(const TrajectoryAnalysisSettings &settings, const TopologyInformation& top)
 117 {
 118     DT.initAnalysis(top, initParams);
 119     nres = DT.nres;
 120 }
 121
 122 void Dssp::analyzeFrame(int frnr, const t_trxframe& fr, t_pbc* pbc, TrajectoryAnalysisModuleData* /* pdata*/)
 123 {
 124     DT.analyzeFrame(frnr, fr, pbc);
 125 }
 126
 127 void Dssp::finishAnalysis(int /*nframes*/) {
 128     data.resize(0);
 129     data = DT.getData();
 130 }
 131
 132
 133 void Dssp::writeOutput() {
 134
 135     std::ofstream dsspout;
 136     dsspout.open(filename_, std::ofstream::out | std::ofstream::trunc);
 137     dsspout << nres << "\n";
 138     for (std::size_t i{ 0 }; i < data.size(); ++i)
 139     {
 140         for(std::size_t j{0}; j < data[i].second.size(); ++j){
 141             dsspout << data[i].second[j];
 142         }
 143         dsspout << "\n";
 144     }
 145     dsspout.close();
 146     data.resize(0);
 147 }
 148
 149 const char DsspInfo::name[]             = "dssp";
 150 const char DsspInfo::shortDescription[] = "Calculate protein secondary structure via DSSP algo";
 151
 152 TrajectoryAnalysisModulePointer DsspInfo::create()
 153 {
 154    return TrajectoryAnalysisModulePointer(new Dssp);
 155 }
 156
 157 } // namespace analysismodules
 158
 159 } // namespace gmx
 160
 161 int main(int argc, char *argv[]) {
 162     return gmx::TrajectoryAnalysisCommandLineRunner::runAsMain<gmx::analysismodules::Dssp>(argc, argv);
 163 }