On Tue, Oct 22, 2024 at 5:31 PM Andrew Pinski <quic_apin...@quicinc.com> wrote:
>
> This adds quick_emplace_push and safe_emplace_push to vec.
> These are like std::vector's emplace_back so you don't need an extra
> copy of the struct around.
>
> Since we require C++11 and also support partial non-PODs for vec, these 
> functions
> can be added.
>
> I will be using them to cleanup some code and even improving how some places
> use vec.
>
> Bootstrapped and tested on x86_64-linux-gnu.
>
> gcc/ChangeLog:
>
>         * vec.h (vec<T,vl_embed>::quick_emplace_push): New function.
>         (vec<T,vl_ptr>::quick_emplace_push): New function.
>         (vec<T,vl_ptr>::safe_emplace_push): New function.
>         * vec.cc (test_init): Add test for safe_emplace_push.
>         (test_quick_emplace_push): New function.
>         (test_safe_emplace_push): New function.
>         (vec_cc_tests): Call test_quick_emplace_push and
>         test_safe_emplace_push.
>
> Signed-off-by: Andrew Pinski <quic_apin...@quicinc.com>
> ---
>  gcc/vec.cc | 41 +++++++++++++++++++++++++++++++++++++++++
>  gcc/vec.h  | 46 ++++++++++++++++++++++++++++++++++++++++++++++
>  2 files changed, 87 insertions(+)
>
> diff --git a/gcc/vec.cc b/gcc/vec.cc
> index ba963b96c6a..cd7cd3c156b 100644
> --- a/gcc/vec.cc
> +++ b/gcc/vec.cc
> @@ -304,6 +304,10 @@ test_init ()
>
>      ASSERT_EQ (2, v1.length ());
>      ASSERT_EQ (2, v2.length ());
> +    v2.safe_emplace_push (1);
> +
> +    ASSERT_EQ (3, v1.length ());
> +    ASSERT_EQ (3, v2.length ());
>      v1.release ();
>    }
>  }
> @@ -327,6 +331,25 @@ test_quick_push ()
>    ASSERT_EQ (7, v[2]);
>  }
>
> +/* Verify that vec::quick_emplace_push works correctly.  */
> +
> +static void
> +test_quick_emplace_push ()
> +{
> +  auto_vec <std::pair<int,int>> v;
> +  ASSERT_EQ (0, v.length ());
> +  v.reserve (3);
> +  ASSERT_EQ (0, v.length ());
> +  ASSERT_TRUE (v.space (3));
> +  v.quick_emplace_push (5, 5);
> +  v.quick_emplace_push (6, 6);
> +  v.quick_emplace_push (7, 7);
> +  ASSERT_EQ (3, v.length ());
> +  ASSERT_EQ (std::make_pair(5,5), v[0]);
> +  ASSERT_EQ (std::make_pair(6,6), v[1]);
> +  ASSERT_EQ (std::make_pair(7,7), v[2]);
> +}
> +
>  /* Verify that vec::safe_push works correctly.  */
>
>  static void
> @@ -343,6 +366,22 @@ test_safe_push ()
>    ASSERT_EQ (7, v[2]);
>  }
>
> +/* Verify that vec::safe_emplace_push works correctly.  */
> +
> +static void
> +test_safe_emplace_push ()
> +{
> +  auto_vec <std::pair<int,int>> v;
> +  ASSERT_EQ (0, v.length ());
> +  v.safe_emplace_push (5, 5);
> +  v.safe_emplace_push (6, 6);
> +  v.safe_emplace_push (7, 7);
> +  ASSERT_EQ (3, v.length ());
> +  ASSERT_EQ (std::make_pair(5,5), v[0]);
> +  ASSERT_EQ (std::make_pair(6,6), v[1]);
> +  ASSERT_EQ (std::make_pair(7,7), v[2]);
> +}
> +
>  /* Verify that vec::truncate works correctly.  */
>
>  static void
> @@ -591,7 +630,9 @@ vec_cc_tests ()
>  {
>    test_init ();
>    test_quick_push ();
> +  test_quick_emplace_push ();
>    test_safe_push ();
> +  test_safe_emplace_push ();
>    test_truncate ();
>    test_safe_grow_cleared ();
>    test_pop ();
> diff --git a/gcc/vec.h b/gcc/vec.h
> index b13c4716428..8277d156f05 100644
> --- a/gcc/vec.h
> +++ b/gcc/vec.h
> @@ -619,6 +619,8 @@ public:
>    void splice (const vec &);
>    void splice (const vec *src);
>    T *quick_push (const T &);
> +  template <typename ...Args>
> +  T *quick_emplace_push (Args&&... args);
>    using pop_ret_type
>      = typename std::conditional <std::is_trivially_destructible <T>::value,
>                                  T &, void>::type;
> @@ -1044,6 +1046,21 @@ vec<T, A, vl_embed>::quick_push (const T &obj)
>    return slot;
>  }
>
> +/* Push T(ARGS) (a new element) onto the end of the vector.  There must be
> +   sufficient space in the vector.  Return a pointer to the slot
> +   where T(ARGS) was inserted.  */
> +
> +template<typename T, typename A>
> +template<typename ...Args>
> +inline T *
> +vec<T, A, vl_embed>::quick_emplace_push (Args&&... args)
> +{
> +  gcc_checking_assert (space (1));
> +  T *slot = &address ()[m_vecpfx.m_num++];
> +  ::new (static_cast<void*>(slot)) T (std::forward<Args>(args)...);
> +  return slot;
> +}
> +
>
>  /* Pop and return a reference to the last element off the end of the
>     vector.  If T has non-trivial destructor, this method just pops
> @@ -1612,7 +1629,11 @@ public:
>    void splice (const vec &);
>    void safe_splice (const vec & CXX_MEM_STAT_INFO);
>    T *quick_push (const T &);
> +  template <typename ...Args>
> +  T *quick_emplace_push (Args&&... args);
>    T *safe_push (const T &CXX_MEM_STAT_INFO);
> +  template <typename ...Args>
> +  T *safe_emplace_push (Args&&... args CXX_MEM_STAT_INFO);
>    using pop_ret_type
>      = typename std::conditional <std::is_trivially_destructible <T>::value,
>                                  T &, void>::type;
> @@ -2070,6 +2091,18 @@ vec<T, va_heap, vl_ptr>::quick_push (const T &obj)
>    return m_vec->quick_push (obj);
>  }
>
> +/* Push OBJ (a new element) onto the end of the vector.  There must be
> +   sufficient space in the vector.  Return a pointer to the slot
> +   where OBJ was inserted.  */
> +
> +template<typename T>
> +template <typename ...Args>
> +inline T *
> +vec<T, va_heap, vl_ptr>::quick_emplace_push (Args&& ... args)
> +{
> +  return m_vec->quick_emplace_push (std::forward<Args>(args)...);
> +}
> +
>
>  /* Push a new element OBJ onto the end of this vector.  Reallocates
>     the embedded vector, if needed.  Return a pointer to the slot where
> @@ -2083,6 +2116,19 @@ vec<T, va_heap, vl_ptr>::safe_push (const T &obj 
> MEM_STAT_DECL)
>    return quick_push (obj);
>  }
>
> +/* Push a new element T(ARGS) onto the end of this vector.  Reallocates
> +   the embedded vector, if needed.  Return a pointer to the slot where
> +   OBJ was inserted.  */
> +
> +template<typename T>
> +template <typename ...Args>
> +inline T *
> +vec<T, va_heap, vl_ptr>::safe_emplace_push (Args&&... args MEM_STAT_DECL)

Did you check this works with --enable-gather-detailed-mem-stats?

> +{
> +  reserve (1, false PASS_MEM_STAT);
> +  return quick_emplace_push (std::forward<Args>(args)...);
> +}
> +
>
>  /* Pop and return a reference to the last element off the end of the
>     vector.  If T has non-trivial destructor, this method just pops
> --
> 2.43.0
>

Reply via email to