ComplexVector.h

Go to the documentation of this file.

/*CXXR $Id: ComplexVector.h 1378 2013-04-26 13:39:20Z arr $
 *CXXR
 *CXXR This file is part of CXXR, a project to refactor the R interpreter
 *CXXR into C++.  It may consist in whole or in part of program code and
 *CXXR documentation taken from the R project itself, incorporated into
 *CXXR CXXR (and possibly MODIFIED) under the terms of the GNU General Public
 *CXXR Licence.
 *CXXR 
 *CXXR CXXR is Copyright (C) 2008-13 Andrew R. Runnalls, subject to such other
 *CXXR copyrights and copyright restrictions as may be stated below.
 *CXXR 
 *CXXR CXXR is not part of the R project, and bugs and other issues should
 *CXXR not be reported via r-bugs or other R project channels; instead refer
 *CXXR to the CXXR website.
 *CXXR */

/*
 *  R : A Computer Language for Statistical Data Analysis
 *  Copyright (C) 1995, 1996  Robert Gentleman and Ross Ihaka
 *  Copyright (C) 1999-2006   The R Development Core Team.
 *  Andrew Runnalls (C) 2008
 *
 *  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 2.1 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 Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, a copy is available at
 *  http://www.r-project.org/Licenses/
 */

#ifndef COMPLEXVECTOR_H
#define COMPLEXVECTOR_H

#include "CXXR/VectorBase.h"
#include "R_ext/Complex.h"

#ifdef __cplusplus

#include "R_ext/Arith.h"
#include "CXXR/FixedVector.hpp"
#include "CXXR/SEXP_downcast.hpp"

namespace CXXR {
    struct Complex : public Rcomplex {
        Complex()
        {}

        Complex(double rl, double im = 0.0)
        {
            r = rl;
            i = im;
        }

        Complex& operator=(double rhs)
        {
            r = rhs;
            i = 0;
            return *this;
        }

        template <class Archive>
        void serialize(Archive& ar, const unsigned int version)
        {
            ar & BOOST_SERIALIZATION_NVP(r);
            ar & BOOST_SERIALIZATION_NVP(i);
        }
    };

    // Template specializations:
    namespace ElementTraits {
        template <>
        struct NAFunc<Complex> {
            const Complex& operator()() const
            {
                static Complex na(NA_REAL, NA_REAL);
                return na;
            }
        };
    }

    template <>
    inline const char* FixedVector<Complex, CPLXSXP>::staticTypeName()
    {
        return "complex";
    }

    typedef CXXR::FixedVector<Complex, CPLXSXP> ComplexVector;
}  // namespace CXXR

BOOST_CLASS_EXPORT_KEY(CXXR::ComplexVector)

extern "C" {
#endif /* __cplusplus */

#ifndef __cplusplus
    Rboolean Rf_isComplex(SEXP s);
#else
    inline Rboolean Rf_isComplex(SEXP s)
    {
        return Rboolean(s && TYPEOF(s) == CPLXSXP);
    }
#endif

#ifndef __cplusplus
Rcomplex *COMPLEX(SEXP x);
#else
inline Rcomplex *COMPLEX(SEXP x)
{
    using namespace CXXR;
    return &(*SEXP_downcast<ComplexVector*>(x, false))[0];
}
#endif

#ifdef __cplusplus
}
#endif

#endif /* COMPLEXVECTOR_H */