#ifndef SEGMENT_LAYER_INC_CPP
#define SEGMENT_LAYER_INC_CPP


#include "segment.inc.cpp"
#include "layer.inc.cpp"


class SegmentLayer: public Segment {
public:
  Layer *fl;
  Layer *bl;
  
  SegmentLayer(Layer &fl, Layer &bl):
    Segment(fl.layout.getW(), fl.layout.getH(), fl.layout.getD(), fl.weightsCount, fl.weights)
  {
    assert(fl.layout.getW() == bl.layout.getW());
    assert(fl.layout.getH() == bl.layout.getH());
    assert(fl.layout.getD() == bl.layout.getD());
    filename = fl.filename;
  }
  
  SegmentLayer(Layer &layer): SegmentLayer(layer, layer.back()) { }
  
  
  Quality pass(Barrier &barrier, int x, int y, int z, NeuronReal trainRatio) override {
    check(x, y, z);
    
    // copy values
    
    // pass
    
    Layer *ffl = fl;
    for(Layer *l = fl->next; l; l = l->next) {
      if (!l->skipTrain) ffl = l;
      barrier.wait();
      l->pass(barrier);
      if (l == bl) break;
      if (!l->next) return Quality::nan();
    }
    
    // verify
    
    Quality q;
    
    if (trainRatio <= 0) return q;
    
    // back pass deltas
    
    for(Layer *l = bl; l != ffl; l = l->next) {
      barrier.wait();
      l->backpassDeltas(barrier);
    }
    
    // beack pass weights

    for(Layer *l = bl; l; l = l->next) {
      if (!l->skipTrain) {
        barrier.wait();
        l->backpassDeltas(barrier);
      }
    }
    
    return q;
  }
};




#endif