source: trunk/test/bam_read_filter.cc @ 2994

// $Id: bam_read_filter.cc 2994 2013-03-10 13:04:07Z peter $
//
// Copyright (C) 2013 Peter Johansson
//
// 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 3 of the License, 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, see <http://www.gnu.org/licenses/>.

#include <config.h>

#include "Suite.h"

#ifdef HAVE_LIBBAM
#include "yat/omic/BamFile.h"
#include "yat/omic/BamRead.h"
#include "yat/omic/BamReadFilter.h"
#include "yat/omic/BamReadIterator.h"
#endif

#include <boost/iterator/filter_iterator.hpp>

#include <algorithm>
#include <cassert>
#include <string>
#include <vector>

using namespace theplu::yat;

void test1(test::Suite& suite);

int main(int argc, char* argv[])
{
  test::Suite suite(argc, argv);
#ifndef HAVE_LIBBAM
  suite.out() << "no libbam\n";
  return EXIT_SKIP;
#endif
#ifndef HAVE_SAMTOOLS
  suite.out() << "no samtools\n";
  return EXIT_SKIP;
#endif
#ifdef HAVE_LIBBAM
  test1(suite);
#endif
  return suite.return_value();
}

#ifdef HAVE_LIBBAM
using namespace omic;

void test1(test::Suite& suite)
{
  std::string file = "../../data/foo.sorted.bam";

  InBamFile in;
  in.open(file);
  BamReadIterator begin(in);
  BamReadIterator end;

  BamReadFilter filter(20, BAM_DEF_MASK | BAM_FMUNMAP, BAM_FPROPER_PAIR);
  std::vector<BamRead> result;
  std::copy(boost::make_filter_iterator(filter, begin, end),
            boost::make_filter_iterator(filter, end),
            back_inserter(result));
  suite.out() << result.size() << "\n";
  suite.add(result.size()==710);
}
#endif