Particles.hpp

Go to the documentation of this file.

/*
 *  This file is part of AQUAgpusph, a free CFD program based on SPH.
 *  Copyright (C) 2012  Jose Luis Cercos Pita <jl.cercos@upm.es>
 *
 *  AQUAgpusph 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.
 *
 *  AQUAgpusph 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 AQUAgpusph.  If not, see <http://www.gnu.org/licenses/>.
 */

 
#ifndef PARTICLES_H_INCLUDED
#define PARTICLES_H_INCLUDED
 
#include <vector>
#include <map>
#include "aquagpusph/sphPrerequisites.hpp"
#include "aquagpusph/ProblemSetup.hpp"
#include "Logger.hpp"
#include "InputOutput.hpp"
 
namespace Aqua {
namespace InputOutput {

class Particles : public InputOutput
{
  public:
        Particles(ProblemSetup& sim_data,
                  unsigned int iset,
                  size_t offset,
                  size_t n = 0);

        virtual ~Particles();

        inline const std::string& file() const { return _output_file; }

        inline size_t n() const { return _bounds.y - _bounds.x; }

        inline cl_event getUserEvent() const { return _user_event; }

        virtual inline void waitForSavers()
        {
                if (!_user_event)
                        return;
                cl_int err_code;
                err_code = clWaitForEvents(1, &_user_event);
                if (err_code != CL_SUCCESS) {
                        LOG(L_ERROR, "Failure waiting for a file writer\n");
                        Logger::singleton()->printOpenCLError(err_code);
                        throw std::runtime_error("OpenCL error");
                }
                err_code = clReleaseEvent(_user_event);
                if (err_code != CL_SUCCESS) {
                        LOG(L_ERROR, "Failure releasing the user event\n");
                        Logger::singleton()->printOpenCLError(err_code);
                        throw std::runtime_error("OpenCL error");
                }
                _user_event = NULL;
        }

        virtual void save(float t);

        virtual void print_file()
        {
                cl_int err_code;
                err_code = clSetUserEventStatus(_user_event, CL_COMPLETE);
                if (err_code != CL_SUCCESS) {
                        LOG(L_ERROR, "Failure setting the user event as completed.\n");
                        Logger::singleton()->printOpenCLError(err_code);
                        throw std::runtime_error("OpenCL error");
                }
                err_code = clReleaseEvent(_user_event);
                if (err_code != CL_SUCCESS) {
                        LOG(L_ERROR, "Failure releasing the user event.\n");
                        Logger::singleton()->printOpenCLError(err_code);
                        throw std::runtime_error("OpenCL error");
                }
        }
 
  protected:
        const ProblemSetup& simData() const { return _sim_data; }

        inline void n(const size_t& n)
        {
                const size_t offset = _bounds.x;
                _bounds.y = offset + n;
                for (auto const& mem : _data)
                        free(mem.second);
                _data.clear();
        }

        inline const ulvec2 bounds() const { return _bounds; }

        inline unsigned int setId() const { return _iset; }

        void loadDefault();

        inline void file(const std::string filename) { _output_file = filename; };

        unsigned int file(const std::string basename,
                          unsigned int start_index,
                          unsigned int digits = 5);

        inline float time() const { return _time; }

        cl_event download(std::vector<std::string> fields);

        inline std::map<std::string, void*> data() const { return _data; }
 
  private:
        ProblemSetup _sim_data;

        ulvec2 _bounds;

        unsigned int _iset;

        std::string _output_file;

        float _time;

        cl_event _user_event;

        std::map<std::string, void*> _data;
 
}; // class InputOutput
 
}
} // namespaces
 
#endif // PARTICLES_H_INCLUDED