class BedItem(QGraphicsRectItem):
    def __init__(self, parent, scene, 
                 bed, 
                 rect, pos, pen, 
                 columnWidths, columnSequence):
        QGraphicsRectItem.__init__(self, parent, scene)
        self.bed = bed
        self.p = QPen(Qt.NoPen)
        self.p.setStyle(Qt.DotLine)
        self.columnWidths = columnWidths
        self.columnSequence = columnSequence
        self.headerItems = dict()
        self.setRect(rect)
        self.setPos(pos)
        self.setPen(QPen(Qt.NoPen))
        self.showFields()
    def showFields(self):
        x = 0
        for headerClass in self.columnSequence:
            if headerClass == HeightHeaderItem:
                self.drawHeight(self.columnWidths[headerClass], x)
            if headerClass == BedHeaderItem:
                self.drawBedNumber(self.columnWidths[headerClass], x)
            if headerClass == LithologyHeaderItem:
                self.drawLithology(self.columnWidths[headerClass], x)
            if headerClass == GrainSizeHeaderItem:
                self.drawGrainSize(self.columnWidths[headerClass], x)
            if headerClass == ColorHeaderItem:
                self.drawColor(self.columnWidths[headerClass], x)
            if headerClass == FossilHeaderItem:
                self.drawFossils(self.columnWidths[headerClass], x)
            if headerClass == SedimentStructureHeaderItem:
                self.drawSedimentStructures(self.columnWidths[headerClass], x)
            if headerClass == CustomSymbolHeaderItem:
                self.drawCustomSymbols(self.columnWidths[headerClass], x)
            if headerClass == TectonicUnitHeaderItem:
                self.drawTectonicUnit(self.columnWidths[headerClass], x)
            if headerClass == LithologicalUnitHeaderItem:
                self.drawLithologicalUnit(self.columnWidths[headerClass], x)
            if headerClass == StratigraphicUnitHeaderItem:
                self.drawStratigraphicUnit(self.columnWidths[headerClass], x)
            if headerClass == FaciesHeaderItem:
                self.drawFacies(self.columnWidths[headerClass], x)
            if headerClass == OutcropTypeHeaderItem:
                self.drawOutcropType(self.columnWidths[headerClass], x)
            if headerClass == BeddingTypeHeaderItem:
                self.drawBeddingType(self.columnWidths[headerClass], x)
            self.drawBoundaryTypes()
            x += self.columnWidths[headerClass]
    def drawVerticalSeparator(self, x, pen = None):
        l = QGraphicsLineItem(self)
#        self.scene().addItem(l)
        l.setLine(QLineF(0, 0, 0, self.rect().height()))
        l.setPos(x, 0)
        if pen is None:
            l.setPen(self.p)
        else:
            l.setPen(pen)
    def drawHeight(self, w, y):
        pass
    def drawBedNumber(self, w, x):
        self.drawVerticalSeparator(x)
        self.bedNumberItm = BedNumberText(self, self.scene(), QFont(), self.bed, 0)
        self.bedNumberItm.adjustSize()
        self.bedNumberItm.setPos(QPointF(x, (self.rect().height() - self.bedNumberItm.boundingRect().height())/2))
    def drawLithology(self, w, x):
        self.lithologyItm = LithologyItem(self, self.scene(),
                                          QRectF(0, 0, w, self.rect().height()),
                                          self.p,
                                          self.bed)
        self.lithologyItm.setPos(QPointF(x, 0))
    def drawGrainSize(self, w, x):
        if len(self.bed.grainSizes) < 1:
            return
        self.grainSizeItm = GrainSizeItem(self, self.scene(),
                                          QRectF(0, 0, w, self.rect().height()),
                                          self.p,
                                          self.bed)
        self.grainSizeItm.setPos(x, 0)
    def drawColor(self, w, x):
        self.colorItm = ColorItem(self, self.scene(),
                                  QRectF(0, 0, w, self.rect().height()),
                                  self.p,
                                  self.bed)
        self.colorItm.setPos(QPointF(x, 0))
    def drawBeddingType(self, w, x):
        self.beddingTypeItm = BeddingTypeItem(self, self.scene(),
                                              QRectF(0, 0, w, self.rect().height()),
                                              self.p,
                                              self.bed)
        self.beddingTypeItm.setPos(QPointF(x, 0))
    def drawFossils(self, w, x):
        self.fossilItm = FossilItem(self, self.scene(),
                                    QRectF(0, 0, w, self.rect().height()),
                                    self.p,
                                    self.bed)
        self.fossilItm.setPos(QPointF(x, 0))
    def drawSedimentStructures(self, w, x):
        self.sedimentStructureItm = SedimentStructureItem(self, self.scene(),
                                                          QRectF(0, 0, w, self.rect().height()),
                                                          self.p,
                                                          self.bed)
        self.sedimentStructureItm.setPos(QPointF(x, 0))
    def drawCustomSymbols(self, w, x):
        self.customSymbolItm = CustomSymbolItem(self, self.scene(),
                                                QRectF(0, 0, w, self.rect().height()),
                                                self.p,
                                                self.bed)
        self.customSymbolItm.setPos(QPointF(x, 0))
    def drawTectonicUnit(self, w, x):
        self.tectonicUnitItm = TectonicUnitItem(self, self.scene(),
                                                QRectF(0, 0, w, self.rect().height()),
                                                self.p,
                                                self.bed)
        self.tectonicUnitItm.setPos(QPointF(x, 0))
    def drawLithologicalUnit(self, w, x):
        self.lithologicalUnitItm = LithologicalUnitItem(self, self.scene(),
                                                        QRectF(0, 0, w, self.rect().height()),
                                                        self.p,
                                                        self.bed)
        self.lithologicalUnitItm.setPos(QPointF(x, 0))
    def drawStratigraphicUnit(self, w, x):
        self.stratigraphicUnitItm = StratigraphicUnitItem(self, self.scene(),
                                                        QRectF(0, 0, w, self.rect().height()),
                                                        self.p,
                                                        self.bed)
        self.stratigraphicUnitItm.setPos(QPointF(x, 0))
    def drawFacies(self, w, x):
        self.faciesItm = FaciesItem(self, self.scene(),
                                    QRectF(0, 0, w, self.rect().height()),
                                    self.p,
                                    self.bed)
        self.faciesItm.setPos(QPointF(x, 0))
    def drawOutcropType(self, w, x):
        self.outcropTypeItm = OutcropTypeItem(self, self.scene(),
                                              QRectF(0, 0, w, self.rect().height()),
                                              self.p,
                                              self.bed)
        self.outcropTypeItm.setPos(QPointF(x, 0))

    def drawBoundaryTypes(self):
        f = SymbolFactory()
        for bt in self.bed.boundaryTypes:
            if bt.boundaryType.hasSvgItem():
                pm = f.pixmapFromSvgItem(bt.boundaryType.svgItem, self.rect().width(), Qt.IgnoreAspectRatio)
                itm = QGraphicsPixmapItem(self, self.scene())
                itm.setPixmap(pm)
                itm.setPos(0, self.rect().height() - bt.begin * self.rect().height() / 100)
 def drawTectonicUnit(self, w, x):
     self.tectonicUnitItm = TectonicUnitItem(self, self.scene(),
                                             QRectF(0, 0, w, self.rect().height()),
                                             self.p,
                                             self.bed)
     self.tectonicUnitItm.setPos(QPointF(x, 0))