BUG-2794 : refactored code to use BitArray
[bgpcep.git] / pcep / ietf-stateful07 / src / main / java / org / opendaylight / protocol / pcep / ietf / initiated00 / CInitiated00SrpObjectParser.java
1 /*
2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.protocol.pcep.ietf.initiated00;
9
10 import static org.opendaylight.protocol.util.ByteBufWriteUtil.writeUnsignedInt;
11
12 import com.google.common.base.Preconditions;
13 import io.netty.buffer.ByteBuf;
14 import io.netty.buffer.Unpooled;
15 import org.opendaylight.protocol.pcep.ietf.stateful07.Stateful07SrpObjectParser;
16 import org.opendaylight.protocol.pcep.spi.ObjectUtil;
17 import org.opendaylight.protocol.pcep.spi.PCEPDeserializerException;
18 import org.opendaylight.protocol.pcep.spi.TlvRegistry;
19 import org.opendaylight.protocol.pcep.spi.VendorInformationTlvRegistry;
20 import org.opendaylight.protocol.util.BitArray;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.Srp1;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.Srp1Builder;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.SrpIdNumber;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.srp.object.Srp;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.srp.object.SrpBuilder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.srp.object.srp.TlvsBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.ObjectHeader;
29
30 /**
31  * Parser for {@link Srp}
32  */
33 public class CInitiated00SrpObjectParser extends Stateful07SrpObjectParser {
34
35     private static final int REMOVE_FLAG = 31;
36
37     public CInitiated00SrpObjectParser(final TlvRegistry tlvReg, final VendorInformationTlvRegistry viTlvReg) {
38         super(tlvReg, viTlvReg);
39     }
40
41     @Override
42     public Srp parseObject(final ObjectHeader header, final ByteBuf bytes) throws PCEPDeserializerException {
43         Preconditions.checkArgument(bytes != null && bytes.isReadable(), "Array of bytes is mandatory. Can't be null or empty.");
44         if (bytes.readableBytes() < MIN_SIZE) {
45             throw new PCEPDeserializerException("Wrong length of array of bytes. Passed: " + bytes.readableBytes() + "; Expected: >="
46                     + MIN_SIZE + ".");
47         }
48         final SrpBuilder builder = new SrpBuilder();
49         builder.setIgnore(header.isIgnore());
50         builder.setProcessingRule(header.isProcessingRule());
51         final BitArray flags = BitArray.valueOf(bytes, FLAGS_SIZE);
52         builder.addAugmentation(Srp1.class, new Srp1Builder().setRemove(flags.get(REMOVE_FLAG)).build());
53         builder.setOperationId(new SrpIdNumber(bytes.readUnsignedInt()));
54         final TlvsBuilder tlvsBuilder = new TlvsBuilder();
55         parseTlvs(tlvsBuilder, bytes.slice());
56         builder.setTlvs(tlvsBuilder.build());
57         return builder.build();
58     }
59
60     @Override
61     public void serializeObject(final Object object, final ByteBuf buffer) {
62         Preconditions.checkArgument(object instanceof Srp, "Wrong instance of PCEPObject. Passed %s. Needed SrpObject.", object.getClass());
63         final Srp srp = (Srp) object;
64         final ByteBuf body = Unpooled.buffer();
65         final BitArray flags = new BitArray(FLAGS_SIZE);
66         if (srp.getAugmentation(Srp1.class) != null) {
67             flags.set(REMOVE_FLAG, srp.getAugmentation(Srp1.class).isRemove());
68         }
69         flags.toByteBuf(body);
70         Preconditions.checkArgument(srp.getOperationId() != null, "OperationId is mandatory.");
71         writeUnsignedInt(srp.getOperationId().getValue(), body);
72         serializeTlvs(srp.getTlvs(), body);
73         ObjectUtil.formatSubobject(TYPE, CLASS, object.isProcessingRule(), object.isIgnore(), body, buffer);
74     }
75 }