Real3Proxy.h

// -*- tab-width: 2; indent-tabs-mode: nil; coding: utf-8-with-signature -*-
//-----------------------------------------------------------------------------
// Copyright 2000-2022 CEA (www.cea.fr) IFPEN (www.ifpenergiesnouvelles.com)
// See the top-level COPYRIGHT file for details.
// SPDX-License-Identifier: Apache-2.0
//-----------------------------------------------------------------------------
/*---------------------------------------------------------------------------*/
/* Real3Proxy.h                                                (C) 2000-2018 */
/*                                                                           */
/* Proxy d'un 'Real3'.                                                       */
/*---------------------------------------------------------------------------*/
#ifndef ARCANE_UTILS_REAL3PROXY_H
#define ARCANE_UTILS_REAL3PROXY_H
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#include "arcane/utils/Real3.h"
#include "arcane/utils/BuiltInProxy.h"
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
ARCANE_BEGIN_NAMESPACE
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
typedef BuiltInProxy<Real> RealProxy;
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
class ARCANE_UTILS_EXPORT Real3Proxy
: public Real3POD
{
 public:
 
  Real3Proxy(Real3& value,const MemoryAccessInfo& info)
  : x(value.x,info), y(value.y,info), z(value.z,info), m_value(value), m_info(info) {}
  Real3Proxy(const Real3Proxy& f)
  : x(f.x), y(f.y), z(f.z), m_value(f.m_value), m_info(f.m_info) {}
  Real3 operator=(const Real3Proxy f)
    { x = f.x; y = f.y; z = f.z; return m_value; }
  Real3 operator=(Real3 f)
    { x = f.x; y = f.y; z = f.z; return m_value; }
  Real3 operator= (Real v)
    { x = v; y = v; z = v; return m_value; }
  operator Real3() const
    {
      return getValue();
    }
  //operator Real3&()
  //{
  //  return getValueMutable();
  //}
 public:
 
  RealProxy x; 
  RealProxy y; 
  RealProxy z; 
 
 private:
 
  Real3& m_value;
  MemoryAccessInfo m_info;
 
 public:
 
  Real3 copy() const { return Real3(x,y,z); }
  Real3Proxy& reset() { x = y = z = 0.; return (*this); }
  Real3Proxy& assign(Real ax,Real ay,Real az)
    { x = ax; y = ay; z = az; return (*this); }
  Real3Proxy& assign(Real3 f)
    { x = f.x; y = f.y; z = f.z; return (*this); }
  bool isNearlyZero() const
    {
      return math::isNearlyZero(x.getValue()) && math::isNearlyZero(y.getValue()) && math::isNearlyZero(z.getValue());
    }
  Real abs2() const
    { return x*x + y*y + z*z; }
  Real abs() const
    { return _sqrt(abs2()); }
 
  std::istream& assign(std::istream& i);
  std::ostream& print(std::ostream& o) const;
  std::ostream& printXyz(std::ostream& o) const;
 
  Real3Proxy& add(Real3 b) { x+=b.x; y+=b.y; z+=b.z; return (*this); }
  Real3Proxy& sub(Real3 b) { x-=b.x; y-=b.y; z-=b.z; return (*this); }
  Real3Proxy& mul(Real3 b) { x*=b.x; y*=b.y; z*=b.z; return (*this); }
  Real3Proxy& div(Real3 b) { x/=b.x; y/=b.y; z/=b.z; return (*this); }
  Real3Proxy& addSame(Real b) { x+=b; y+=b; z+=b; return (*this); }
  Real3Proxy& subSame(Real b) { x-=b; y-=b; z-=b; return (*this); }
  Real3Proxy& mulSame(Real b) { x*=b; y*=b; z*=b; return (*this); }
  Real3Proxy& divSame(Real b) { x/=b; y/=b; z/=b; return (*this); }
  Real3Proxy& operator+= (Real3 b) { return add(b); }
  Real3Proxy& operator-= (Real3 b) { return sub(b); }
  Real3Proxy& operator*= (Real3 b) { return mul(b); }
  void operator*= (Real  b) { x*=b; y*=b; z*=b; }
  Real3Proxy& operator/= (Real3 b) { return div(b); }
  void operator/= (Real  b) { x/=b; y/=b; z/=b; }
  Real3 operator+  (Real3 b)  const { return Real3(x+b.x,y+b.y,z+b.z); }
  Real3 operator-  (Real3 b)  const { return Real3(x-b.x,y-b.y,z-b.z); }
  Real3 operator-() const { return Real3(-x,-y,-z); }
  Real3 operator*(Real3 b) const { return Real3(x*b.x,y*b.y,z*b.z); }
  Real3 operator/(Real3 b) const { return Real3(x/b.x,y/b.y,z/b.z); }
                    
  Real3Proxy& normalize()
    {
      Real d = abs();
      if (!math::isZero(d))
        divSame(d);
      return (*this);
    }
  
  friend bool operator==(Real3Proxy a,Real3Proxy b)
  { return  _eq(a.x,b.x) &&  _eq(a.y,b.y) && _eq(a.z,b.z); }
  friend bool operator!=(Real3Proxy a,Real3Proxy b)
  { return !(a==b); }
 
 public:
  Real3 getValue() const
    {
      m_info.setRead();
      return m_value;
    }
  Real3& getValueMutable()
    {
      m_info.setReadOrWrite();
      return m_value;
    }
 private:
  static bool _eq(Real a,Real b)
    { return math::isEqual(a,b); }
  static Real _sqrt(Real a)
    { return math::sqrt(a); }
};
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
inline Real3 operator*(Real sca,Real3Proxy vec)
{
  return Real3(vec.x*sca,vec.y*sca,vec.z*sca);
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
inline Real3
operator*(Real3Proxy vec,Real sca)
{
  return Real3(vec.x*sca,vec.y*sca,vec.z*sca);
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
inline Real3
operator/(Real3Proxy vec,Real sca)
{
  return Real3(vec.x/sca,vec.y/sca,vec.z/sca);
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
inline bool
operator<(const Real3Proxy v1,const Real3Proxy v2)
{
  return v1.getValue()<v2.getValue();
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
inline std::ostream&
operator<< (std::ostream& o,Real3Proxy t)
{
  return t.printXyz(o);
}
inline std::istream&
operator>> (std::istream& i,Real3Proxy& t)
{
  return t.assign(i);
}
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
ARCANE_END_NAMESPACE
 
/*---------------------------------------------------------------------------*/
/*---------------------------------------------------------------------------*/
 
#endif