pygil.h

/* This file is part of the Tomographer project, which is distributed under the
 * terms of the MIT license.
 *
 * The MIT License (MIT)
 *
 * Copyright (c) 2016 ETH Zurich, Institute for Theoretical Physics, Philippe Faist
 * Copyright (c) 2017 Caltech, Institute for Quantum Information and Matter, Philippe Faist
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */


#ifndef TOMOGRAPHER_PY_PYGIL_H
#define TOMOGRAPHER_PY_PYGIL_H

#include <vector>
#include <algorithm>

#include <tomographerpy/common.h>

#include <tomographer/tools/cxxutil.h>
#include <tomographer/tools/loggers.h>

#include <tomographerpy/pylogger.h>


#define TPY_EXPR_WITH_GIL( x )                      \
  [&]() { py::gil_scoped_acquire _tpy_gil; return (x); } ()




namespace tpy {
  class GilProtectedPyLogger; // forward declaration
} // namespace tpy
namespace Tomographer { namespace Logger {
  // Traits for PyLogger
  template<>
  struct TOMOGRAPHER_EXPORT LoggerTraits<tpy::GilProtectedPyLogger> : public DefaultLoggerTraits
  {
    enum {
      // This logger is thread-safe because it makes sure that it acquires the GIL as
      // needed.
      IsThreadSafe = 1
    };
  };
} } // namespaces Tomographer::Logger


namespace tpy {

class TOMOGRAPHER_EXPORT GilProtectedPyLogger
  : public Tomographer::Logger::LoggerBase<GilProtectedPyLogger>
{
public:
  inline GilProtectedPyLogger(PyLogger & logger_, bool require_gil_acquisition_ = true)
    : LoggerBase<GilProtectedPyLogger>(logger_.level()), // freeze to current level
      logger(logger_),
      require_gil_acquisition(require_gil_acquisition_)
  {
  }

  inline PyLogger & getLogger() const { return logger; }

  inline bool getRequireGilAcquisition() const { return require_gil_acquisition; }

  inline void requireGilAcquisition(bool value) {
    tomographer_assert(require_gil_acquisition != value);
    require_gil_acquisition = value;
  }

  inline void emitLog(int level, const char * origin, const std::string & msg)
  {
    if (require_gil_acquisition) {
      py::gil_scoped_acquire gil_acquire;
      logger.emitLog(level, origin, msg);
    } else {
      logger.emitLog(level, origin, msg);
    }
  }

private:
  PyLogger & logger;
  bool require_gil_acquisition;
};

} // namespace tpy


#endif