Team Fortress 2 Source Code as on 22/4/2020
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

1131 lines
27 KiB

  1. /* -----------------------------------------------------------------------------
  2. * See the LICENSE file for information on copyright, usage and redistribution
  3. * of SWIG, and the README file for authors - http://www.swig.org/release.html.
  4. *
  5. * rubycontainer.swg
  6. *
  7. * Ruby sequence <-> C++ container wrapper
  8. *
  9. * This wrapper, and its iterator, allows a general use (and reuse) of
  10. * the the mapping between C++ and Ruby, thanks to the C++
  11. * templates.
  12. *
  13. * Of course, it needs the C++ compiler to support templates, but
  14. * since we will use this wrapper with the STL containers, that should
  15. * be the case.
  16. * ----------------------------------------------------------------------------- */
  17. %{
  18. #include <iostream>
  19. %}
  20. #if !defined(SWIG_NO_EXPORT_ITERATOR_METHODS)
  21. # if !defined(SWIG_EXPORT_ITERATOR_METHODS)
  22. # define SWIG_EXPORT_ITERATOR_METHODS SWIG_EXPORT_ITERATOR_METHODS
  23. # endif
  24. #endif
  25. %include <rubyiterators.swg>
  26. /**** The RubySequence C++ Wrap ***/
  27. %insert(header) %{
  28. #include <stdexcept>
  29. %}
  30. %include <std_except.i>
  31. %fragment("RubySequence_Base","header")
  32. {
  33. %#include <functional>
  34. namespace swig {
  35. template < class T >
  36. struct yield : public std::unary_function< T, bool >
  37. {
  38. bool
  39. operator()( const T& v ) const
  40. {
  41. return RTEST( rb_yield( swig::from< T >(v) ) );
  42. }
  43. };
  44. inline size_t
  45. check_index(ptrdiff_t i, size_t size, bool insert = false) {
  46. if ( i < 0 ) {
  47. if ((size_t) (-i) <= size)
  48. return (size_t) (i + size);
  49. } else if ( (size_t) i < size ) {
  50. return (size_t) i;
  51. } else if (insert && ((size_t) i == size)) {
  52. return size;
  53. }
  54. throw std::out_of_range("index out of range");
  55. }
  56. inline size_t
  57. slice_index(ptrdiff_t i, size_t size) {
  58. if ( i < 0 ) {
  59. if ((size_t) (-i) <= size) {
  60. return (size_t) (i + size);
  61. } else {
  62. throw std::out_of_range("index out of range");
  63. }
  64. } else {
  65. return ( (size_t) i < size ) ? ((size_t) i) : size;
  66. }
  67. }
  68. template <class Sequence, class Difference>
  69. inline typename Sequence::iterator
  70. getpos(Sequence* self, Difference i) {
  71. typename Sequence::iterator pos = self->begin();
  72. std::advance(pos, check_index(i,self->size()));
  73. return pos;
  74. }
  75. template <class Sequence, class Difference>
  76. inline typename Sequence::const_iterator
  77. cgetpos(const Sequence* self, Difference i) {
  78. typename Sequence::const_iterator pos = self->begin();
  79. std::advance(pos, check_index(i,self->size()));
  80. return pos;
  81. }
  82. template <class Sequence, class Difference>
  83. inline Sequence*
  84. getslice(const Sequence* self, Difference i, Difference j) {
  85. typename Sequence::size_type size = self->size();
  86. typename Sequence::size_type ii = swig::check_index(i, size);
  87. typename Sequence::size_type jj = swig::slice_index(j, size);
  88. if (jj > ii) {
  89. typename Sequence::const_iterator vb = self->begin();
  90. typename Sequence::const_iterator ve = self->begin();
  91. std::advance(vb,ii);
  92. std::advance(ve,jj);
  93. return new Sequence(vb, ve);
  94. } else {
  95. return new Sequence();
  96. }
  97. }
  98. template <class Sequence, class Difference, class InputSeq>
  99. inline void
  100. setslice(Sequence* self, Difference i, Difference j, const InputSeq& v) {
  101. typename Sequence::size_type size = self->size();
  102. typename Sequence::size_type ii = swig::check_index(i, size, true);
  103. typename Sequence::size_type jj = swig::slice_index(j, size);
  104. if (jj < ii) jj = ii;
  105. size_t ssize = jj - ii;
  106. if (ssize <= v.size()) {
  107. typename Sequence::iterator sb = self->begin();
  108. typename InputSeq::const_iterator vmid = v.begin();
  109. std::advance(sb,ii);
  110. std::advance(vmid, jj - ii);
  111. self->insert(std::copy(v.begin(), vmid, sb), vmid, v.end());
  112. } else {
  113. typename Sequence::iterator sb = self->begin();
  114. typename Sequence::iterator se = self->begin();
  115. std::advance(sb,ii);
  116. std::advance(se,jj);
  117. self->erase(sb,se);
  118. self->insert(sb, v.begin(), v.end());
  119. }
  120. }
  121. template <class Sequence, class Difference>
  122. inline void
  123. delslice(Sequence* self, Difference i, Difference j) {
  124. typename Sequence::size_type size = self->size();
  125. typename Sequence::size_type ii = swig::check_index(i, size, true);
  126. typename Sequence::size_type jj = swig::slice_index(j, size);
  127. if (jj > ii) {
  128. typename Sequence::iterator sb = self->begin();
  129. typename Sequence::iterator se = self->begin();
  130. std::advance(sb,ii);
  131. std::advance(se,jj);
  132. self->erase(sb,se);
  133. }
  134. }
  135. }
  136. }
  137. %fragment("RubySequence_Cont","header",
  138. fragment="StdTraits",
  139. fragment="RubySequence_Base",
  140. fragment="ConstIterator_T")
  141. {
  142. namespace swig
  143. {
  144. /**
  145. * This class is a proxy class for references, used to return and set values
  146. * of an element of a Ruby Array of stuff.
  147. * It can be used by RubySequence_InputIterator to make it work with STL
  148. * algorithms.
  149. *
  150. */
  151. template <class T>
  152. struct RubySequence_Ref
  153. {
  154. RubySequence_Ref(VALUE seq, int index)
  155. : _seq(seq), _index(index)
  156. {
  157. }
  158. operator T () const
  159. {
  160. VALUE item = rb_ary_entry(_seq, _index );
  161. try {
  162. return swig::as<T>(item, true);
  163. } catch (std::exception& e) {
  164. char msg[1024];
  165. sprintf(msg, "in sequence element %d ", _index);
  166. VALUE lastErr = rb_gv_get("$!");
  167. if ( lastErr == Qnil ) {
  168. %type_error(swig::type_name<T>());
  169. }
  170. VALUE str = rb_str_new2(msg);
  171. str = rb_str_cat2( str, e.what() );
  172. SWIG_Ruby_ExceptionType( NULL, str );
  173. throw;
  174. }
  175. }
  176. RubySequence_Ref& operator=(const T& v)
  177. {
  178. rb_ary_set(_seq, _index, swig::from< T >(v));
  179. return *this;
  180. }
  181. private:
  182. VALUE _seq;
  183. int _index;
  184. };
  185. /**
  186. * This class is a proxy to return a pointer to a class, usually
  187. * RubySequence_Ref.
  188. * It can be used by RubySequence_InputIterator to make it work with STL
  189. * algorithms.
  190. *
  191. */
  192. template <class T>
  193. struct RubySequence_ArrowProxy
  194. {
  195. RubySequence_ArrowProxy(const T& x): m_value(x) {}
  196. const T* operator->() const { return &m_value; }
  197. operator const T*() const { return &m_value; }
  198. T m_value;
  199. };
  200. /**
  201. * Input Iterator. This adapator class is a random access iterator that
  202. * allows you to use STL algorithms with a Ruby class (a Ruby Array by default).
  203. *
  204. */
  205. template <class T, class Reference = RubySequence_Ref< T > >
  206. struct RubySequence_InputIterator
  207. {
  208. typedef RubySequence_InputIterator<T, Reference > self;
  209. typedef std::random_access_iterator_tag iterator_category;
  210. typedef Reference reference;
  211. typedef T value_type;
  212. typedef T* pointer;
  213. typedef ptrdiff_t difference_type;
  214. RubySequence_InputIterator()
  215. {
  216. }
  217. RubySequence_InputIterator(VALUE seq, int index)
  218. : _seq(seq), _index(index)
  219. {
  220. }
  221. reference operator*() const
  222. {
  223. return reference(_seq, _index);
  224. }
  225. RubySequence_ArrowProxy<T>
  226. operator->() const {
  227. return RubySequence_ArrowProxy<T>(operator*());
  228. }
  229. bool operator==(const self& ri) const
  230. {
  231. return (_index == ri._index) && (_seq == ri._seq);
  232. }
  233. bool operator!=(const self& ri) const
  234. {
  235. return !(operator==(ri));
  236. }
  237. self& operator ++ ()
  238. {
  239. ++_index;
  240. return *this;
  241. }
  242. self& operator -- ()
  243. {
  244. --_index;
  245. return *this;
  246. }
  247. self& operator += (difference_type n)
  248. {
  249. _index += n;
  250. return *this;
  251. }
  252. self operator +(difference_type n) const
  253. {
  254. return self(_seq, _index + n);
  255. }
  256. self& operator -= (difference_type n)
  257. {
  258. _index -= n;
  259. return *this;
  260. }
  261. self operator -(difference_type n) const
  262. {
  263. return self(_seq, _index - n);
  264. }
  265. difference_type operator - (const self& ri) const
  266. {
  267. return _index - ri._index;
  268. }
  269. bool operator < (const self& ri) const
  270. {
  271. return _index < ri._index;
  272. }
  273. reference
  274. operator[](difference_type n) const
  275. {
  276. return reference(_seq, _index + n);
  277. }
  278. private:
  279. VALUE _seq;
  280. difference_type _index;
  281. };
  282. /**
  283. * This adaptor class allows you to use a Ruby Array as if it was an STL
  284. * container, giving it begin(), end(), and iterators.
  285. *
  286. */
  287. template <class T>
  288. struct RubySequence_Cont
  289. {
  290. typedef RubySequence_Ref<T> reference;
  291. typedef const RubySequence_Ref<T> const_reference;
  292. typedef T value_type;
  293. typedef T* pointer;
  294. typedef int difference_type;
  295. typedef int size_type;
  296. typedef const pointer const_pointer;
  297. typedef RubySequence_InputIterator<T, reference> iterator;
  298. typedef RubySequence_InputIterator<T, const_reference> const_iterator;
  299. RubySequence_Cont(VALUE seq) : _seq(0)
  300. {
  301. if (!rb_obj_is_kind_of(seq, rb_cArray)) {
  302. throw std::invalid_argument("an Array is expected");
  303. }
  304. _seq = seq;
  305. }
  306. ~RubySequence_Cont()
  307. {
  308. }
  309. size_type size() const
  310. {
  311. return RARRAY_LEN(_seq);
  312. }
  313. bool empty() const
  314. {
  315. return size() == 0;
  316. }
  317. iterator begin()
  318. {
  319. return iterator(_seq, 0);
  320. }
  321. const_iterator begin() const
  322. {
  323. return const_iterator(_seq, 0);
  324. }
  325. iterator end()
  326. {
  327. return iterator(_seq, size());
  328. }
  329. const_iterator end() const
  330. {
  331. return const_iterator(_seq, size());
  332. }
  333. reference operator[](difference_type n)
  334. {
  335. return reference(_seq, n);
  336. }
  337. const_reference operator[](difference_type n) const
  338. {
  339. return const_reference(_seq, n);
  340. }
  341. bool check(bool set_err = false) const
  342. {
  343. int s = (int) size();
  344. for (int i = 0; i < s; ++i) {
  345. VALUE item = rb_ary_entry(_seq, i );
  346. if (!swig::check<value_type>(item)) {
  347. if (set_err) {
  348. char msg[1024];
  349. sprintf(msg, "in sequence element %d", i);
  350. SWIG_Error(SWIG_RuntimeError, msg);
  351. }
  352. return false;
  353. }
  354. }
  355. return true;
  356. }
  357. private:
  358. VALUE _seq;
  359. };
  360. }
  361. }
  362. /**
  363. * Macros used to typemap an STL iterator -> SWIGIterator conversion.
  364. *
  365. */
  366. %define %swig_sequence_iterator(Sequence...)
  367. #if defined(SWIG_EXPORT_ITERATOR_METHODS)
  368. %typemap(out,noblock=1,fragment="RubySequence_Cont")
  369. const_iterator, const_reverse_iterator {
  370. $result = SWIG_NewPointerObj(swig::make_const_iterator(%static_cast($1,const $type &),
  371. self),
  372. swig::ConstIterator::descriptor(),SWIG_POINTER_OWN);
  373. }
  374. %typemap(out,noblock=1,fragment="RubySequence_Cont")
  375. iterator, reverse_iterator {
  376. $result = SWIG_NewPointerObj(swig::make_nonconst_iterator(%static_cast($1,const $type &),
  377. self),
  378. swig::Iterator::descriptor(),SWIG_POINTER_OWN);
  379. }
  380. %typemap(out,noblock=1,fragment="RubySequence_Cont")
  381. std::pair<const_iterator, const_iterator> {
  382. $result = rb_ary_new2(2);
  383. RARRAY_PTR($result)[0] = SWIG_NewPointerObj(swig::make_const_iterator(%static_cast($1,const $type &).first),
  384. swig::ConstIterator::descriptor(),SWIG_POINTER_OWN);
  385. RARRAY_PTR($result)[1] = SWIG_NewPointerObj(swig::make_const_iterator(%static_cast($1,const $type &).second),
  386. swig::ConstIterator::descriptor(),SWIG_POINTER_OWN);
  387. RARRAY_LEN($result) = 2;
  388. }
  389. // std::map/multimap/set allow returning std::pair< iterator, iterator > from
  390. // equal_range, but we cannot still modify the key, so the iterator is
  391. // const.
  392. %typemap(out,noblock=1,fragment="RubySequence_Cont")
  393. std::pair<iterator, iterator> {
  394. $result = rb_ary_new2(2);
  395. RARRAY_PTR($result)[0] = SWIG_NewPointerObj(swig::make_const_iterator(%static_cast($1,const $type &).first),
  396. swig::ConstIterator::descriptor(),SWIG_POINTER_OWN);
  397. RARRAY_PTR($result)[1] = SWIG_NewPointerObj(swig::make_const_iterator(%static_cast($1,const $type &).second),
  398. swig::ConstIterator::descriptor(),SWIG_POINTER_OWN);
  399. RARRAY_LEN($result) = 2;
  400. }
  401. %typemap(in,noblock=1,fragment="RubySequence_Cont")
  402. const_iterator(swig::ConstIterator *iter = 0, int res),
  403. const_reverse_iterator(swig::ConstIterator *iter = 0, int res) {
  404. res = SWIG_ConvertPtr($input, %as_voidptrptr(&iter),
  405. swig::ConstIterator::descriptor(), 0);
  406. if (!SWIG_IsOK(res) || !iter) {
  407. %argument_fail(SWIG_TypeError, "$type", $symname, $argnum);
  408. } else {
  409. swig::ConstIterator_T<$type > *iter_t = dynamic_cast<swig::ConstIterator_T<$type > *>(iter);
  410. if (iter_t) {
  411. $1 = iter_t->get_current();
  412. } else {
  413. %argument_fail(SWIG_TypeError, "$type", $symname, $argnum);
  414. }
  415. }
  416. }
  417. %typemap(in,noblock=1,fragment="RubySequence_Cont")
  418. iterator(swig::Iterator *iter = 0, int res),
  419. reverse_iterator(swig::Iterator *iter = 0, int res) {
  420. res = SWIG_ConvertPtr($input, %as_voidptrptr(&iter), swig::Iterator::descriptor(), 0);
  421. if (!SWIG_IsOK(res) || !iter) {
  422. %argument_fail(SWIG_TypeError, "$type", $symname, $argnum);
  423. } else {
  424. swig::Iterator_T<$type > *iter_t = dynamic_cast<swig::Iterator_T<$type > *>(iter);
  425. if (iter_t) {
  426. $1 = iter_t->get_current();
  427. } else {
  428. %argument_fail(SWIG_TypeError, "$type", $symname, $argnum);
  429. }
  430. }
  431. }
  432. %typecheck(%checkcode(ITERATOR),noblock=1,fragment="RubySequence_Cont")
  433. const_iterator, const_reverse_iterator {
  434. swig::ConstIterator *iter = 0;
  435. int res = SWIG_ConvertPtr($input, %as_voidptrptr(&iter),
  436. swig::ConstIterator::descriptor(), 0);
  437. $1 = (SWIG_IsOK(res) && iter && (dynamic_cast<swig::ConstIterator_T<$type > *>(iter) != 0));
  438. }
  439. %typecheck(%checkcode(ITERATOR),noblock=1,fragment="RubySequence_Cont")
  440. iterator, reverse_iterator {
  441. swig::ConstIterator *iter = 0;
  442. int res = SWIG_ConvertPtr($input, %as_voidptrptr(&iter),
  443. swig::Iterator::descriptor(), 0);
  444. $1 = (SWIG_IsOK(res) && iter && (dynamic_cast<swig::Iterator_T<$type > *>(iter) != 0));
  445. }
  446. %fragment("RubySequence_Cont");
  447. // %newobject iterator;
  448. // %newobject const_iterator;
  449. // %extend {
  450. // swig::Iterator* iterator(VALUE* RUBY_SELF) {
  451. // return swig::make_nonconst_iterator($self->begin(), $self->begin(),
  452. // $self->end(), *RUBY_SELF);
  453. // }
  454. // swig::ConstIterator* const_iterator(VALUE* RUBY_SELF) {
  455. // return swig::make_const_iterator($self->begin(), $self->begin(),
  456. // $self->end(), *RUBY_SELF);
  457. // }
  458. // }
  459. #endif //SWIG_EXPORT_ITERATOR_METHODS
  460. %enddef
  461. /**** The Ruby container methods ****/
  462. %define %swig_container_methods(Container...)
  463. %extend {
  464. %newobject dup;
  465. Container* dup()
  466. {
  467. return new Container(*$self);
  468. }
  469. }
  470. %enddef
  471. /**
  472. * Macro used to define common Ruby printing methods for STL container
  473. *
  474. */
  475. %define %swig_sequence_printing_methods(Sequence...)
  476. %extend {
  477. VALUE inspect()
  478. {
  479. Sequence::const_iterator i = $self->begin();
  480. Sequence::const_iterator e = $self->end();
  481. VALUE str = rb_str_new2( swig::type_name< Sequence >() );
  482. str = rb_str_cat2( str, " [" );
  483. bool comma = false;
  484. VALUE tmp;
  485. for ( ; i != e; ++i, comma = true )
  486. {
  487. if (comma) str = rb_str_cat2( str, "," );
  488. tmp = swig::from< Sequence::value_type >( *i );
  489. tmp = rb_inspect( tmp );
  490. str = rb_str_buf_append( str, tmp );
  491. }
  492. str = rb_str_cat2( str, "]" );
  493. return str;
  494. }
  495. VALUE to_a()
  496. {
  497. Sequence::const_iterator i = $self->begin();
  498. Sequence::const_iterator e = $self->end();
  499. VALUE ary = rb_ary_new2( std::distance( i, e ) );
  500. VALUE tmp;
  501. for ( ; i != e; ++i )
  502. {
  503. tmp = swig::from< Sequence::value_type >( *i );
  504. rb_ary_push( ary, tmp );
  505. }
  506. return ary;
  507. }
  508. VALUE to_s()
  509. {
  510. Sequence::iterator i = $self->begin();
  511. Sequence::iterator e = $self->end();
  512. VALUE str = rb_str_new2( "" );
  513. VALUE tmp;
  514. for ( ; i != e; ++i )
  515. {
  516. tmp = swig::from< Sequence::value_type >( *i );
  517. tmp = rb_obj_as_string( tmp );
  518. str = rb_str_buf_append( str, tmp );
  519. }
  520. return str;
  521. }
  522. }
  523. %enddef
  524. /**
  525. * Macro used to add common methods to all STL sequence-type containers
  526. *
  527. */
  528. %define %swig_sequence_methods_common(Sequence...)
  529. %swig_container_methods(%arg(Sequence))
  530. %swig_sequence_iterator(%arg(Sequence))
  531. %swig_sequence_printing_methods(%arg(Sequence))
  532. %fragment("RubySequence_Base");
  533. %extend {
  534. VALUE slice( difference_type i, difference_type j )
  535. {
  536. if ( j <= 0 ) return Qnil;
  537. std::size_t len = $self->size();
  538. if ( i < 0 ) i = len - i;
  539. j += i;
  540. if ( static_cast<std::size_t>(j) >= len ) j = len-1;
  541. VALUE r = Qnil;
  542. try {
  543. r = swig::from< const Sequence* >( swig::getslice(self, i, j) );
  544. }
  545. catch( std::out_of_range )
  546. {
  547. }
  548. return r;
  549. }
  550. Sequence* each()
  551. {
  552. if ( !rb_block_given_p() )
  553. rb_raise( rb_eArgError, "no block given");
  554. VALUE r;
  555. Sequence::const_iterator i = self->begin();
  556. Sequence::const_iterator e = self->end();
  557. for ( ; i != e; ++i )
  558. {
  559. r = swig::from< Sequence::value_type >(*i);
  560. rb_yield(r);
  561. }
  562. return self;
  563. }
  564. %newobject select;
  565. Sequence* select() {
  566. if ( !rb_block_given_p() )
  567. rb_raise( rb_eArgError, "no block given" );
  568. Sequence* r = new Sequence;
  569. Sequence::const_iterator i = $self->begin();
  570. Sequence::const_iterator e = $self->end();
  571. for ( ; i != e; ++i )
  572. {
  573. VALUE v = swig::from< Sequence::value_type >(*i);
  574. if ( RTEST( rb_yield(v) ) )
  575. $self->insert( r->end(), *i);
  576. }
  577. return r;
  578. }
  579. %alias reject_bang "delete_if";
  580. Sequence* reject_bang() {
  581. if ( !rb_block_given_p() )
  582. rb_raise( rb_eArgError, "no block given" );
  583. Sequence::iterator i = self->begin();
  584. Sequence::iterator e = self->end();
  585. for ( ; i != e; )
  586. {
  587. VALUE r = swig::from< Sequence::value_type >(*i);
  588. if ( RTEST( rb_yield(r) ) )
  589. $self->erase(i++);
  590. else
  591. ++i;
  592. }
  593. return self;
  594. }
  595. VALUE delete_at(difference_type i) {
  596. VALUE r = Qnil;
  597. try {
  598. Sequence::iterator at = swig::getpos(self, i);
  599. r = swig::from< Sequence::value_type >( *(at) );
  600. $self->erase(at);
  601. }
  602. catch (std::out_of_range)
  603. {
  604. }
  605. return r;
  606. }
  607. VALUE __delete2__(const value_type& i) {
  608. VALUE r = Qnil;
  609. return r;
  610. }
  611. }
  612. %enddef
  613. /**
  614. * Macro used to add functions for back insertion of values in
  615. * STL Sequence containers
  616. *
  617. */
  618. %define %swig_sequence_back_inserters( Sequence... )
  619. %extend {
  620. VALUE pop() {
  621. if ($self->empty()) return Qnil;
  622. Sequence::value_type x = self->back();
  623. $self->pop_back();
  624. return swig::from< Sequence::value_type >( x );
  625. }
  626. %alias push "<<";
  627. const value_type push( const value_type& e ) {
  628. $self->push_back( e );
  629. return e;
  630. }
  631. %newobject reject;
  632. Sequence* reject() {
  633. if ( !rb_block_given_p() )
  634. rb_raise( rb_eArgError, "no block given" );
  635. Sequence* r = new Sequence;
  636. std::remove_copy_if( $self->begin(), $self->end(),
  637. std::back_inserter(*r),
  638. swig::yield< Sequence::value_type >() );
  639. return r;
  640. }
  641. }
  642. %enddef
  643. /**
  644. * Macro used to add functions for Sequences
  645. *
  646. */
  647. %define %swig_sequence_methods(Sequence...)
  648. %swig_sequence_methods_common(%arg(Sequence));
  649. %swig_sequence_back_inserters(%arg(Sequence));
  650. %extend {
  651. VALUE at(difference_type i) const {
  652. VALUE r = Qnil;
  653. try {
  654. r = swig::from< Sequence::value_type >( *(swig::cgetpos(self, i)) );
  655. }
  656. catch( std::out_of_range )
  657. {
  658. }
  659. return r;
  660. }
  661. VALUE __getitem__(difference_type i, difference_type j) const {
  662. if ( j <= 0 ) return Qnil;
  663. std::size_t len = $self->size();
  664. if ( i < 0 ) i = len - i;
  665. j += i; if ( static_cast<std::size_t>(j) >= len ) j = len-1;
  666. VALUE r = Qnil;
  667. try {
  668. r = swig::from< const Sequence* >( swig::getslice(self, i, j) );
  669. }
  670. catch( std::out_of_range )
  671. {
  672. }
  673. return r;
  674. }
  675. VALUE __getitem__(difference_type i) const {
  676. VALUE r = Qnil;
  677. try {
  678. r = swig::from< Sequence::value_type >( *(swig::cgetpos(self, i)) );
  679. }
  680. catch( std::out_of_range )
  681. {
  682. }
  683. return r;
  684. }
  685. VALUE __getitem__(VALUE i) const {
  686. if ( rb_obj_is_kind_of( i, rb_cRange ) == Qfalse )
  687. {
  688. rb_raise( rb_eTypeError, "not a valid index or range" );
  689. }
  690. VALUE r = Qnil;
  691. static ID id_end = rb_intern("end");
  692. static ID id_start = rb_intern("begin");
  693. static ID id_noend = rb_intern("exclude_end?");
  694. VALUE start = rb_funcall( i, id_start, 0 );
  695. VALUE end = rb_funcall( i, id_end, 0 );
  696. bool noend = ( rb_funcall( i, id_noend, 0 ) == Qtrue );
  697. int len = $self->size();
  698. int s = NUM2INT( start );
  699. if ( s < 0 ) s = len + s;
  700. else if ( s >= len ) return Qnil;
  701. int e = NUM2INT( end );
  702. if ( e < 0 ) e = len + e;
  703. if ( e < s ) return Qnil; //std::swap( s, e );
  704. if ( noend ) e -= 1;
  705. if ( e >= len ) e = len - 1;
  706. return swig::from< Sequence* >( swig::getslice(self, s, e+1) );
  707. }
  708. VALUE __setitem__(difference_type i, const value_type& x)
  709. {
  710. std::size_t len = $self->size();
  711. if ( i < 0 ) i = len - i;
  712. else if ( static_cast<std::size_t>(i) >= len )
  713. $self->resize( i+1, x );
  714. else
  715. *(swig::getpos(self,i)) = x;
  716. return swig::from< Sequence::value_type >( x );
  717. }
  718. VALUE __setitem__(difference_type i, difference_type j, const Sequence& v)
  719. throw (std::invalid_argument) {
  720. if ( j <= 0 ) return Qnil;
  721. std::size_t len = $self->size();
  722. if ( i < 0 ) i = len - i;
  723. j += i;
  724. if ( static_cast<std::size_t>(j) >= len ) {
  725. $self->resize( j+1, *(v.begin()) );
  726. j = len-1;
  727. }
  728. VALUE r = Qnil;
  729. swig::setslice(self, i, j, v);
  730. r = swig::from< const Sequence* >( &v );
  731. return r;
  732. }
  733. }
  734. %enddef
  735. // ..I don't think %swig_sequence_methods_val are really used at all anymore...
  736. %define %swig_sequence_methods_val(Sequence...)
  737. %swig_sequence_methods_common(%arg(Sequence))
  738. %extend {
  739. VALUE __getitem__(difference_type i) {
  740. VALUE r = Qnil;
  741. try {
  742. r = swig::from< Sequence::value_type >( *(swig::cgetpos(self, i)) );
  743. }
  744. catch( std::out_of_range )
  745. {
  746. }
  747. return r;
  748. }
  749. VALUE __setitem__(difference_type i, value_type x) {
  750. std::size_t len = $self->size();
  751. if ( i < 0 ) i = len - i;
  752. else if ( static_cast<std::size_t>(i) >= len )
  753. $self->resize( i+1, x );
  754. else *(swig::getpos(self,i)) = x;
  755. return swig::from< Sequence::value_type >( x );
  756. }
  757. }
  758. %enddef
  759. /**
  760. * Macro used to add functions for front insertion of
  761. * elements in STL sequence containers that support it.
  762. *
  763. */
  764. %define %swig_sequence_front_inserters( Sequence... )
  765. %extend {
  766. VALUE shift()
  767. {
  768. if ($self->empty()) return Qnil;
  769. Sequence::value_type x = self->front();
  770. $self->erase( $self->begin() );
  771. return swig::from< Sequence::value_type >( x );
  772. }
  773. %typemap(in) (int argc, VALUE* argv) {
  774. $1 = argc - 1;
  775. $2 = argv + 1;
  776. }
  777. Sequence* insert( difference_type pos, int argc, VALUE* argv, ... )
  778. {
  779. std::size_t len = $self->size();
  780. std::size_t i = swig::check_index( pos, len, true );
  781. Sequence::iterator start;
  782. VALUE elem = argv[0];
  783. int idx = 0;
  784. try {
  785. Sequence::value_type val = swig::as<Sequence::value_type>( elem, true );
  786. if ( i >= len ) {
  787. $self->resize(i-1, val);
  788. return $self;
  789. }
  790. start = $self->begin();
  791. std::advance( start, i );
  792. $self->insert( start++, val );
  793. for ( ++idx; idx < argc; ++idx )
  794. {
  795. elem = argv[idx];
  796. val = swig::as<Sequence::value_type>( elem );
  797. $self->insert( start++, val );
  798. }
  799. }
  800. catch( std::invalid_argument )
  801. {
  802. rb_raise( rb_eArgError,
  803. Ruby_Format_TypeError( "",
  804. swig::type_name<Sequence::value_type>(),
  805. __FUNCTION__, idx+2, elem ));
  806. }
  807. return $self;
  808. }
  809. %typemap(in) (int argc, VALUE* argv) {
  810. $1 = argc;
  811. $2 = argv;
  812. }
  813. Sequence* unshift( int argc, VALUE* argv, ... )
  814. {
  815. for ( int idx = argc-1; idx >= 0; --idx )
  816. {
  817. Sequence::iterator start = $self->begin();
  818. VALUE elem = argv[idx];
  819. try {
  820. Sequence::value_type val = swig::as<Sequence::value_type>( elem, true );
  821. $self->insert( start, val );
  822. }
  823. catch( std::invalid_argument )
  824. {
  825. rb_raise( rb_eArgError,
  826. Ruby_Format_TypeError( "",
  827. swig::type_name<Sequence::value_type>(),
  828. __FUNCTION__, idx+2, elem ));
  829. }
  830. }
  831. return $self;
  832. }
  833. }
  834. %enddef
  835. //
  836. // Common fragments
  837. //
  838. %fragment("StdSequenceTraits","header",
  839. fragment="StdTraits",
  840. fragment="RubySequence_Cont")
  841. {
  842. namespace swig {
  843. template <class RubySeq, class Seq>
  844. inline void
  845. assign(const RubySeq& rubyseq, Seq* seq) {
  846. %#ifdef SWIG_STD_NOASSIGN_STL
  847. typedef typename RubySeq::value_type value_type;
  848. typename RubySeq::const_iterator it = rubyseq.begin();
  849. for (;it != rubyseq.end(); ++it) {
  850. seq->insert(seq->end(),(value_type)(*it));
  851. }
  852. %#else
  853. seq->assign(rubyseq.begin(), rubyseq.end());
  854. %#endif
  855. }
  856. template <class Seq, class T = typename Seq::value_type >
  857. struct traits_asptr_stdseq {
  858. typedef Seq sequence;
  859. typedef T value_type;
  860. static int asptr(VALUE obj, sequence **seq) {
  861. if (rb_obj_is_kind_of(obj, rb_cArray) == Qtrue) {
  862. try {
  863. RubySequence_Cont<value_type> rubyseq(obj);
  864. if (seq) {
  865. sequence *pseq = new sequence();
  866. assign(rubyseq, pseq);
  867. *seq = pseq;
  868. return SWIG_NEWOBJ;
  869. } else {
  870. return rubyseq.check() ? SWIG_OK : SWIG_ERROR;
  871. }
  872. } catch (std::exception& e) {
  873. if (seq) {
  874. VALUE lastErr = rb_gv_get("$!");
  875. if (lastErr == Qnil) {
  876. rb_raise(rb_eTypeError, e.what());
  877. }
  878. }
  879. return SWIG_ERROR;
  880. }
  881. } else {
  882. sequence *p;
  883. if (SWIG_ConvertPtr(obj,(void**)&p,
  884. swig::type_info<sequence>(),0) == SWIG_OK) {
  885. if (seq) *seq = p;
  886. return SWIG_OLDOBJ;
  887. }
  888. }
  889. return SWIG_ERROR;
  890. }
  891. };
  892. // Partial specialization for GC_VALUE's. No need to typecheck each
  893. // element.
  894. template< class Seq >
  895. struct traits_asptr_stdseq< Seq, swig::GC_VALUE > {
  896. typedef Seq sequence;
  897. typedef swig::GC_VALUE value_type;
  898. static int asptr(VALUE obj, sequence **seq) {
  899. if (rb_obj_is_kind_of(obj, rb_cArray) == Qtrue) {
  900. try {
  901. if (seq) {
  902. RubySequence_Cont<value_type> rubyseq(obj);
  903. sequence *pseq = new sequence();
  904. assign(rubyseq, pseq);
  905. *seq = pseq;
  906. return SWIG_NEWOBJ;
  907. } else {
  908. return true;
  909. }
  910. } catch (std::exception& e) {
  911. if (seq) {
  912. VALUE lastErr = rb_gv_get("$!");
  913. if (lastErr == Qnil) {
  914. rb_raise(rb_eTypeError, e.what());
  915. }
  916. }
  917. return SWIG_ERROR;
  918. }
  919. } else {
  920. sequence *p;
  921. if (SWIG_ConvertPtr(obj,(void**)&p,
  922. swig::type_info<sequence>(),0) == SWIG_OK) {
  923. if (seq) *seq = p;
  924. return SWIG_OLDOBJ;
  925. }
  926. }
  927. return SWIG_ERROR;
  928. }
  929. };
  930. template <class Seq, class T = typename Seq::value_type >
  931. struct traits_from_stdseq {
  932. typedef Seq sequence;
  933. typedef T value_type;
  934. typedef typename Seq::size_type size_type;
  935. typedef typename sequence::const_iterator const_iterator;
  936. static VALUE from(const sequence& seq) {
  937. #ifdef SWIG_RUBY_EXTRA_NATIVE_CONTAINERS
  938. swig_type_info *desc = swig::type_info<sequence>();
  939. if (desc && desc->clientdata) {
  940. return SWIG_NewPointerObj(new sequence(seq), desc, SWIG_POINTER_OWN);
  941. }
  942. #endif
  943. size_type size = seq.size();
  944. if (size <= (size_type)INT_MAX) {
  945. VALUE obj = rb_ary_new2((int)size);
  946. int i = 0;
  947. for (const_iterator it = seq.begin();
  948. it != seq.end(); ++it, ++i) {
  949. RARRAY_PTR(obj)[i] = swig::from< value_type >(*it);
  950. }
  951. RARRAY_LEN(obj) = size;
  952. rb_obj_freeze(obj); // treat as immutable result
  953. return obj;
  954. } else {
  955. rb_raise(rb_eRangeError,"sequence size not valid in ruby");
  956. return Qnil;
  957. }
  958. }
  959. };
  960. }
  961. }
  962. %include <rubycontainer_extended.swg>