Concat.vala 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. namespace Invercargill.Modifiers {
  2. public class Concat<T> : Enumerable<T> {
  3. private Enumerable<T> e1;
  4. private Enumerable<T> e2;
  5. public Concat(Enumerable<T> first, Enumerable<T> second) {
  6. e1 = first;
  7. e2 = second;
  8. }
  9. public override int? peek_count() {
  10. var c1 = e1.peek_count();
  11. var c2 = e2.peek_count();
  12. if(c1 == null || c2 == null)
  13. return null;
  14. return c1 + c2;
  15. }
  16. public override EnumerableInfo get_info() {
  17. return new EnumerableInfo.infer(this, EnumerableCategory.COMPUTED, Iterate.these(e1, e2));
  18. }
  19. public override Tracker<T> get_tracker() {
  20. var tracker1 = e1.get_tracker();
  21. var tracker2 = e2.get_tracker();
  22. return new LambdaTracker<T>(
  23. () => tracker1.has_next() || tracker2.has_next(),
  24. () => {
  25. if(tracker1.has_next()) {
  26. return tracker1.get_next();
  27. }
  28. return tracker2.get_next();
  29. });
  30. }
  31. }
  32. }